keymap.go

// keymap.go contains the KeyMap struct which is used to define the key bindings for the application.
package main

import (
	"github.com/charmbracelet/bubbles/key"
)

type KeyMap struct {
	Space            key.Binding
	Delete           key.Binding
	SortByName       key.Binding
	SortBySize       key.Binding
	SortByModified   key.Binding
	SortByQuant      key.Binding
	SortByFamily     key.Binding
	RunModel         key.Binding
	ConfirmYes       key.Binding
	ConfirmNo        key.Binding
	LinkModel        key.Binding
	LinkAllModels    key.Binding
	ClearScreen      key.Binding
	CompareModelfile key.Binding
	InspectModel     key.Binding
	Quit             key.Binding
	CopyModel        key.Binding
	PushModel        key.Binding
	PullModel        key.Binding
	Top              key.Binding
	AltScreen        key.Binding
	EditModel        key.Binding
	UnloadModels     key.Binding
	Help             key.Binding
	RenameModel      key.Binding
	PullNewModel     key.Binding
	SortOrder        string
}

func (k KeyMap) ShortHelp() []key.Binding {
	return []key.Binding{k.Help, k.Quit}
}

func NewKeyMap() *KeyMap {
	return &KeyMap{
		Space:            key.NewBinding(key.WithKeys("space"), key.WithHelp("space", "select")),
		AltScreen:        key.NewBinding(key.WithKeys("A")),
		ClearScreen:      key.NewBinding(key.WithKeys("C")),
		ConfirmNo:        key.NewBinding(key.WithKeys("n")),
		ConfirmYes:       key.NewBinding(key.WithKeys("y")),
		CompareModelfile: key.NewBinding(key.WithKeys("M"), key.WithHelp("M", "compare modelfile")),
		CopyModel:        key.NewBinding(key.WithKeys("c"), key.WithHelp("c", "copy")),
		RenameModel:      key.NewBinding(key.WithKeys("r"), key.WithHelp("r", "rename")),
		Delete:           key.NewBinding(key.WithKeys("D"), key.WithHelp("D", "delete")),
		Help:             key.NewBinding(key.WithKeys("h"), key.WithHelp("h", "help")),
		InspectModel:     key.NewBinding(key.WithKeys("i"), key.WithHelp("i", "inspect")),
		EditModel:        key.NewBinding(key.WithKeys("e"), key.WithHelp("e", "edit model")),
		LinkAllModels:    key.NewBinding(key.WithKeys("L"), key.WithHelp("L", "link all")),
		LinkModel:        key.NewBinding(key.WithKeys("l"), key.WithHelp("l", "link (L=all)")),
		PushModel:        key.NewBinding(key.WithKeys("P"), key.WithHelp("P", "push")),
		PullModel:        key.NewBinding(key.WithKeys("p"), key.WithHelp("p", "pull")),
		PullNewModel:     key.NewBinding(key.WithKeys("g"), key.WithHelp("g", "get")),
		Quit:             key.NewBinding(key.WithKeys("q")),
		RunModel:         key.NewBinding(key.WithKeys("enter"), key.WithHelp("enter", "run")),
		SortByFamily:     key.NewBinding(key.WithKeys("f"), key.WithHelp("f", "^family")),
		SortByModified:   key.NewBinding(key.WithKeys("m"), key.WithHelp("m", "^modified")),
		SortByName:       key.NewBinding(key.WithKeys("n"), key.WithHelp("n", "^name")),
		SortByQuant:      key.NewBinding(key.WithKeys("K"), key.WithHelp("K", "^quant")),
		SortBySize:       key.NewBinding(key.WithKeys("s"), key.WithHelp("s", "^size")),
		Top:              key.NewBinding(key.WithKeys("t"), key.WithHelp("t", "top")),
		UnloadModels:     key.NewBinding(key.WithKeys("U"), key.WithHelp("U", "unload all")),
	}
}

func (k *KeyMap) GetSortOrder() string {
	return k.SortOrder
}