zk/cmd/index.go

package cmd

import (
	"github.com/mickael-menu/zk/adapter/sqlite"
	"github.com/mickael-menu/zk/core/zk"
)

// Index indexes the content of all the notes in the slip box.
type Index struct {
	Directory string `arg optional type:"path" default:"." help:"Directory containing the notes to index"`
}

func (cmd *Index) Run(container *Container) error {
	z, err := zk.Open(".")
	if err != nil {
		return err
	}

	dir, err := z.RequireDirAt(cmd.Directory)
	if err != nil {
		return err
	}

	db, err := container.Database(z)
	if err != nil {
		return err
	}
	tx, err := db.Begin()
	defer tx.Rollback()
	if err != nil {
		return err
	}
	indexer, err := sqlite.NewIndexer(tx, z.Path, container.Logger)
	if err != nil {
		return err
	}
	err = zk.Index(*dir, indexer, container.Logger)
	if err != nil {
		return err
	}

	return tx.Commit()
}