frostfs-node/cmd/frostfs-lens/internal/meta/tui.go

package meta

import (
	"context"
	"encoding/binary"
	"errors"
	"fmt"

	common "git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-lens/internal"
	schemaCommon "git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-lens/internal/schema/common"
	schema "git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-lens/internal/schema/metabase"
	"git.frostfs.info/TrueCloudLab/frostfs-node/cmd/frostfs-lens/internal/tui"
	"github.com/rivo/tview"
	"github.com/spf13/cobra"
	"go.etcd.io/bbolt"
)

var tuiCMD = &cobra.Command{
	Use:   "explore",
	Short: "Metabase exploration with a terminal UI",
	Long: `Launch a terminal UI to explore metabase and search for data.

Available search filters:
- cid  CID
- oid  OID
- addr CID/OID
- attr key[/value]
`,
	Run: tuiFunc,
}

var initialPrompt string

var parserPerSchemaVersion = map[uint64]schemaCommon.Parser{
	2: schema.MetabaseParserV2,
	3: schema.MetabaseParserV3,
}

func init() {
	common.AddComponentPathFlag(tuiCMD, &vPath)

	tuiCMD.Flags().StringVar(
		&initialPrompt,
		"filter",
		"",
		"Filter prompt to start with, format 'tag:value [+ tag:value]...'",
	)
}

func tuiFunc(cmd *cobra.Command, _ []string) {
	common.ExitOnErr(cmd, runTUI(cmd))
}

func runTUI(cmd *cobra.Command) error {
	db, err := tui.OpenDB(vPath, false)
	if err != nil {
		return fmt.Errorf("couldn't open database: %w", err)
	}
	defer db.Close()

	schemaVersion, hasVersion := lookupSchemaVersion(cmd, db)
	if !hasVersion {
		return errors.New("couldn't detect schema version")
	}

	metabaseParser, ok := parserPerSchemaVersion[schemaVersion]
	if !ok {
		return fmt.Errorf("unknown schema version %d", schemaVersion)
	}

	// Need if app was stopped with Ctrl-C.
	ctx, cancel := context.WithCancel(cmd.Context())
	defer cancel()

	app := tview.NewApplication()
	ui := tui.NewUI(ctx, app, db, metabaseParser, nil)

	_ = ui.AddFilter("cid", tui.CIDParser, "CID")
	_ = ui.AddFilter("oid", tui.OIDParser, "OID")
	_ = ui.AddCompositeFilter("addr", tui.AddressParser, "CID/OID")
	_ = ui.AddCompositeFilter("attr", tui.AttributeParser, "key[/value]")

	err = ui.WithPrompt(initialPrompt)
	if err != nil {
		return fmt.Errorf("invalid filter prompt: %w", err)
	}

	app.SetRoot(ui, true).SetFocus(ui)
	return app.Run()
}

var (
	shardInfoBucket = []byte{5}
	versionRecord   = []byte("version")
)

func lookupSchemaVersion(cmd *cobra.Command, db *bbolt.DB) (version uint64, ok bool) {
	err := db.View(func(tx *bbolt.Tx) error {
		bkt := tx.Bucket(shardInfoBucket)
		if bkt == nil {
			return nil
		}
		rec := bkt.Get(versionRecord)
		if rec == nil {
			return nil
		}

		version = binary.LittleEndian.Uint64(rec)
		ok = true

		return nil
	})
	if err != nil {
		common.ExitOnErr(cmd, fmt.Errorf("couldn't lookup version: %w", err))
	}

	return
}