You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
zk/cmd/edit.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.Zk()
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, false)
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
}
}