mirror of
https://github.com/mickael-menu/zk
synced 2024-11-07 15:20:21 +00:00
107 lines
2.2 KiB
Go
107 lines
2.2 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
|
|
"github.com/mickael-menu/zk/adapter/fzf"
|
|
"github.com/mickael-menu/zk/adapter/sqlite"
|
|
"github.com/mickael-menu/zk/core/note"
|
|
"github.com/mickael-menu/zk/core/zk"
|
|
"github.com/mickael-menu/zk/util/errors"
|
|
)
|
|
|
|
// Edit opens notes matching a set of criteria with the user editor.
|
|
type Edit struct {
|
|
Force bool `short:f help:"Do not confirm before editing many notes at the same time."`
|
|
|
|
Filtering
|
|
Sorting
|
|
}
|
|
|
|
func (cmd *Edit) Run(container *Container) error {
|
|
zk, err := container.OpenZk()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
wd, err := os.Getwd()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
opts, err := NewFinderOpts(zk, cmd.Filtering, cmd.Sorting)
|
|
if err != nil {
|
|
return errors.Wrapf(err, "incorrect criteria")
|
|
}
|
|
|
|
db, err := container.Database(zk.DBPath())
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
var notes []note.Match
|
|
err = db.WithTransaction(func(tx sqlite.Transaction) error {
|
|
finder := container.NoteFinder(tx, fzf.NoteFinderOpts{
|
|
AlwaysFilter: true,
|
|
PreviewCmd: zk.Config.Tool.FzfPreview,
|
|
NewNoteDir: cmd.newNoteDir(zk),
|
|
BasePath: zk.Path,
|
|
CurrentPath: wd,
|
|
})
|
|
notes, err = finder.Find(*opts)
|
|
return err
|
|
})
|
|
if err != nil {
|
|
if err == note.ErrCanceled {
|
|
return nil
|
|
}
|
|
return err
|
|
}
|
|
|
|
count := len(notes)
|
|
|
|
if count > 0 {
|
|
if !cmd.Force && count > 5 {
|
|
confirmed, skipped := container.Terminal.Confirm(fmt.Sprintf("Are you sure you want to open %v notes in the editor?", count), false)
|
|
if skipped {
|
|
return fmt.Errorf("too many notes to be opened in the editor, aborting…")
|
|
} else if !confirmed {
|
|
return nil
|
|
}
|
|
}
|
|
paths := make([]string, 0)
|
|
for _, note := range notes {
|
|
absPath := filepath.Join(zk.Path, note.Path)
|
|
paths = append(paths, absPath)
|
|
}
|
|
|
|
note.Edit(zk, paths...)
|
|
|
|
} else {
|
|
fmt.Println("Found 0 note")
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
// newNoteDir returns the directory in which to create a new note when the fzf
|
|
// binding is triggered.
|
|
func (cmd *Edit) newNoteDir(zk *zk.Zk) *zk.Dir {
|
|
switch len(cmd.Path) {
|
|
case 0:
|
|
dir := zk.RootDir()
|
|
return &dir
|
|
case 1:
|
|
dir, err := zk.DirAt(cmd.Path[0])
|
|
if err != nil {
|
|
return nil
|
|
}
|
|
return dir
|
|
default:
|
|
// More than one directory, it's ambiguous for the "new note" fzf binding.
|
|
return nil
|
|
}
|
|
}
|