Add support to importing of value from a file

This commit is contained in:
James Hewitt 2023-05-12 14:27:33 +01:00
parent ab21f99955
commit 8a508a45ee
No known key found for this signature in database
GPG Key ID: EA6C3C654B6193E4
3 changed files with 79 additions and 15 deletions

View File

@ -756,3 +756,36 @@ func writeToFile(fn, s string, mode int) error {
}
return nil
}
func importValue(path []string, fName string) error {
if AppArgs.ReadOnly {
return errors.New("DB is in Read-Only Mode")
}
return db.Update(func(tx *bolt.Tx) error {
// len(b.GetPath())-1 is the key for the pair we're updating,
// the rest are buckets leading to that key
b := tx.Bucket([]byte(path[0]))
if b == nil {
// Invalid path, try for the root bucket
b = tx.Cursor().Bucket()
}
if b != nil {
if len(path) > 0 {
for i := range path[1 : len(path)-1] {
b = b.Bucket([]byte(path[i+1]))
if b == nil {
return errors.New("updatePairValue: Invalid Path")
}
}
}
// Now update the last key in the path
bk := []byte(path[len(path)-1])
v, err := os.ReadFile(fName)
if err != nil {
return err
}
return b.Put(bk, v)
}
return errors.New("importValue: Invalid Bucket")
})
}

View File

@ -3,7 +3,7 @@ package main
import (
"fmt"
"github.com/br0xen/termbox-util"
termboxUtil "github.com/br0xen/termbox-util"
"github.com/nsf/termbox-go"
)
@ -111,9 +111,9 @@ func (screen *AboutScreen) drawScreen(style Style) {
{"e", "edit value of pair"},
{"r", "rename pair/bucket"},
{"", ""},
{"", ""},
{"D", "delete item"},
{"x,X", "export as string/json to file"},
{"i", "import file to value of pair"},
{"", ""},
{"?", "this screen"},
{"q", "quit program"},

View File

@ -61,9 +61,10 @@ const (
modeInsertPairVal = 70 // 0000 0100 0110
modeDelete = 256 // 0001 0000 0000
modeModToParent = 8 // 0000 0000 1000
modeExport = 512 // 0010 0000 0000
modeExportValue = 513 // 0010 0000 0001
modeExportJSON = 514 // 0010 0000 0010
modeIO = 512 // 0010 0000 0000
modeIOExportValue = 513 // 0010 0000 0001
modeIOExportJSON = 514 // 0010 0000 0010
modeIOImportValue = 516 // 0010 0000 0100
)
/*
@ -88,8 +89,8 @@ func (screen *BrowserScreen) handleKeyEvent(event termbox.Event) int {
return screen.handleInsertKeyEvent(event)
} else if screen.mode == modeDelete {
return screen.handleDeleteKeyEvent(event)
} else if screen.mode&modeExport == modeExport {
return screen.handleExportKeyEvent(event)
} else if screen.mode&modeIO == modeIO {
return screen.handleIOKeyEvent(event)
}
return BrowserScreenIndex
}
@ -204,11 +205,14 @@ func (screen *BrowserScreen) handleBrowseKeyEvent(event termbox.Event) int {
} else if event.Ch == 'D' {
screen.startDeleteItem()
} else if event.Ch == 'x' {
// Export Value
// Export Value to a file
screen.startExportValue()
} else if event.Ch == 'X' {
// Export Key/Value (or Bucket) as JSON
screen.startExportJSON()
} else if event.Ch == 'i' {
// Import value from a file
screen.startImportValue()
}
return BrowserScreenIndex
}
@ -374,7 +378,7 @@ func (screen *BrowserScreen) handleInsertKeyEvent(event termbox.Event) int {
return BrowserScreenIndex
}
func (screen *BrowserScreen) handleExportKeyEvent(event termbox.Event) int {
func (screen *BrowserScreen) handleIOKeyEvent(event termbox.Event) int {
if event.Key == termbox.KeyEsc {
screen.mode = modeBrowse
screen.inputModal.Clear()
@ -383,24 +387,33 @@ func (screen *BrowserScreen) handleExportKeyEvent(event termbox.Event) int {
if screen.inputModal.IsDone() {
b, p, _ := screen.db.getGenericFromPath(screen.currentPath)
fileName := screen.inputModal.GetValue()
if screen.mode&modeExportValue == modeExportValue {
if screen.mode&modeIOExportValue == modeIOExportValue {
// Exporting the value
if p != nil {
if err := exportValue(screen.currentPath, fileName); err != nil {
//screen.setMessage("Error Exporting to file " + fileName + ".")
//screen.setMessage("Error exporting to file " + fileName + ".")
screen.setMessage(err.Error())
} else {
screen.setMessage("Value exported to file: " + fileName)
}
}
} else if screen.mode&modeExportJSON == modeExportJSON {
} else if screen.mode&modeIOExportJSON == modeIOExportJSON {
if b != nil || p != nil {
if exportJSON(screen.currentPath, fileName) != nil {
screen.setMessage("Error Exporting to file " + fileName + ".")
screen.setMessage("Error exporting to file " + fileName + ".")
} else {
screen.setMessage("Value exported to file: " + fileName)
}
}
} else if screen.mode&modeIOImportValue == modeIOImportValue {
if p != nil {
if err := importValue(screen.currentPath, fileName); err != nil {
screen.setMessage(err.Error())
} else {
screen.setMessage("Value imported from file: " + fileName)
screen.refreshDatabase()
}
}
}
screen.mode = modeBrowse
screen.inputModal.Clear()
@ -907,7 +920,7 @@ func (screen *BrowserScreen) startExportValue() bool {
mod.SetValue("")
mod.Show()
screen.inputModal = mod
screen.mode = modeExportValue
screen.mode = modeIOExportValue
return true
}
screen.setMessage("Couldn't do string export on " + screen.currentPath[len(screen.currentPath)-1] + "(did you mean 'X'?)")
@ -930,12 +943,30 @@ func (screen *BrowserScreen) startExportJSON() bool {
}
mod.Show()
screen.inputModal = mod
screen.mode = modeExportJSON
screen.mode = modeIOExportJSON
return true
}
return false
}
func (screen *BrowserScreen) startImportValue() bool {
_, p, e := screen.db.getGenericFromPath(screen.currentPath)
if e == nil && p != nil {
w, h := termbox.Size()
inpW, inpH := (w / 2), 6
inpX, inpY := ((w / 2) - (inpW / 2)), ((h / 2) - inpH)
mod := termboxUtil.CreateInputModal("", inpX, inpY, inpW, inpH, termbox.ColorWhite, termbox.ColorBlack)
mod.SetTitle(termboxUtil.AlignText(fmt.Sprintf("Import value of '%s' from:", p.key), inpW, termboxUtil.AlignCenter))
mod.SetValue("")
mod.Show()
screen.inputModal = mod
screen.mode = modeIOImportValue
return true
}
screen.setMessage("Couldn't do import on " + screen.currentPath[len(screen.currentPath)-1] + ", must be a pair.")
return false
}
func (screen *BrowserScreen) setMessage(msg string) {
screen.message = msg
screen.messageTime = time.Now()