wmtools/notes.nim

168 lines
4.3 KiB
Nim
Raw Normal View History

import base
import std/[os,strutils,sequtils,times]
const note_dir = WM_TOOLS_DIR & ".notes.dmenu/" # Putting it in Nextcloud so it can sync :-)
const default_bg = white
const default_fg = black
type
Note = object
id: string
text: string
proc startNotes()
proc displayOptionMenu(option: string)
var notes: seq[Note] = @[]
proc readNotes(): seq[Note] =
if len(notes) == 0:
if existsOrCreateDir(note_dir):
for note_file in walkDir(note_dir):
try:
var note = Note()
note.id = note_file[1]
2022-05-06 12:12:13 +02:00
note.text = stripQuotes(readFile(note_file[1]))
notes.add(note)
except:
echo "Unable to read note : ", note_file, " : ", getCurrentExceptionMsg()
return notes
proc removeNote(note: Note) =
try:
removeFile(note.id)
for idx, a_note in notes:
if note.id == a_note.id:
notes.delete(idx)
except:
echo "Unable to delete file : ", note.id, " : ", getCurrentExceptionMsg()
proc getNoteStrings(): seq[string] =
var note_list: seq[string] = @[]
if len(notes) == 0:
discard readNotes()
for note in notes:
note_list.add(note.text)
return note_list
proc writeNote(note: Note, is_new: bool = false) =
if note.text == "":
return
try:
2022-05-06 12:12:13 +02:00
writeFile(note.id, stripQuotes(note.text))
if is_new:
2022-05-06 12:12:13 +02:00
var new_note = note
new_note.text = stripQuotes(new_note.text)
notes.add(note)
2022-05-03 22:43:47 +02:00
else:
for idx, a_note in notes:
if a_note.id == note.id:
2022-05-06 12:12:13 +02:00
notes[idx].text = stripQuotes(note.text)
except:
echo "write_note, Unable to write note : ", note.id, " : ", getCurrentExceptionMsg()
2022-05-03 22:43:47 +02:00
proc getNotes(): (Info, seq[string]) =
var info = newInfo("Notes")
info.selected_bg = default_bg
info.selected_fg = default_fg
let notes = getNoteStrings()
return (info,notes)
proc getNote(text: string): Note =
for note in notes:
if note.text == text:
return note
return Note()
proc displayDeleteConfirmationMenu(note: string) =
var yes_no = newInfo("Confirm Delete note : " & note)
yes_no.selected_bg = default_bg
yes_no.selected_fg = default_fg
let args = @["yes", "no"]
let choice = outputData(yes_no, args)
case choice:
of "yes":
let rm_note = getNote(note)
if rm_note.text == note:
removeNote(rm_note)
of "no":
displayOptionMenu(note)
2022-05-04 23:44:41 +02:00
proc replaceNoteConfirmationMenu(note_idx: int, new_text: string): bool =
let old_note = notes[note_idx]
var diag = newInfo("Replace Note : '" & old_note.text & "', with '" & new_text & "' ?")
diag.selected_bg = default_bg
diag.selected_fg = default_fg
let args = @["yes", "no"]
let choice = outputData(diag,args)
case choice:
of "yes":
echo "here"
return true
else:
return false
proc replaceNote(new_text: string, old_text: string) =
for idx, note in notes:
if note.text == old_text:
2022-05-04 23:44:41 +02:00
if replaceNoteConfirmationMenu(idx,new_text):
var new_note = note
new_note.text = new_text
notes[idx] = new_note
writeNote(new_note)
return
proc displayOptionMenu(option: string) =
var select = newInfo("Note")
select.selected_bg = default_bg
select.selected_fg = default_fg
let note_choices = @["rm","back"]
2022-05-03 22:43:47 +02:00
let args = concat(@[option],note_choices)
let chosen = outputData(select,args)
if chosen in note_choices:
case chosen:
of "rm":
displayDeleteConfirmationMenu(option)
startNotes()
of "back":
startNotes()
2022-05-03 22:43:47 +02:00
elif chosen != "" and chosen != option:
replaceNote(chosen, option)
displayOptionMenu(chosen)
else:
displayOptionMenu(option)
proc newNote(text: string) =
var new_note = Note()
2022-05-04 23:44:41 +02:00
let note_id = times.now().format("'{'yyyyMMdd-HHmmss-ffffff'}'")
new_note.id = note_dir & note_id
new_note.text = text
writeNote(new_note, is_new = true)
proc startNotes() =
let (info,all_notes) = getNotes()
let all_choices = @["exit"]
let args = concat(all_notes, all_choices)
let option = outputData(info, args)
if option in all_choices:
case option:
of "exit":
return
if option in all_notes:
displayOptionMenu(option)
elif option != "":
newNote(option)
startNotes()
proc main() =
echo "Note dir : ", note_dir
if tool != "dmenu" and tool != "rofi":
echo "Can only be run in dmenu or rofi mode. Exiting..."
return
startNotes()
if isMainModule:
main()