Split into category packages; module github.com/ZacharyZhang-NY/MujicaUI

chat/project_knowledge_panel.go

@@ -0,0 +1,124 @@
+package chat
+
+import (
+	"fmt"
+	"log/slog"
+
+	"github.com/egoist/mygo/ui"
+
+	"github.com/ZacharyZhang-NY/MujicaUI/core"
+	"github.com/ZacharyZhang-NY/MujicaUI/icons"
+	"github.com/ZacharyZhang-NY/MujicaUI/input"
+	"github.com/ZacharyZhang-NY/MujicaUI/internal"
+	"github.com/ZacharyZhang-NY/MujicaUI/theme"
+)
+
+// KnowledgeFile is a project knowledge file, sized in tokens.
+type KnowledgeFile struct {
+	ID, Name string
+	Tokens   int
+}
+
+// ProjectKnowledgePanelOptions configure a ProjectKnowledgePanel. Capacity is required.
+type ProjectKnowledgePanelOptions struct {
+	// Capacity is how many tokens the project's knowledge holds.
+	Capacity int
+	// Title replaces the built-in "Project knowledge".
+	Title    string
+	Disabled bool
+}
+
+// ProjectKnowledgePanelView is a built panel and its requests.
+type ProjectKnowledgePanelView struct {
+	Element *ui.Element
+	// Add asks the caller to pick files to add.
+	Add bool
+	// Removed is the ID of a file to remove.
+	Removed string
+}
+
+// ProjectKnowledgePanel lists knowledge files with add, remove and usage.
+func ProjectKnowledgePanel(c *ui.Context, files []KnowledgeFile, opts ProjectKnowledgePanelOptions) ProjectKnowledgePanelView {
+	if opts.Capacity <= 0 {
+		panic("mujica: ProjectKnowledgePanel needs a positive Capacity")
+	}
+	used := 0
+	for _, f := range files {
+		if f.Tokens < 0 {
+			panic("mujica: KnowledgeFile with negative Tokens")
+		}
+		used += f.Tokens
+	}
+	k := core.Tokens(c)
+	d := core.Density(c)
+	col := ui.Column(c).Gap(d.Unit()*2).Padding(d.Unit()*3).Background(k.Surface).Border(theme.BorderWidth, k.Border).
+		Radius(theme.CardRadius).Disabled(opts.Disabled)
+	v := ProjectKnowledgePanelView{Element: col}
+	col.Children(func() {
+		ui.Row(c).Gap(d.Unit() * 2).AlignItems(ui.Center).Children(func() {
+			title := opts.Title
+			if title == "" {
+				title = core.Msg(c, "project_knowledge_panel.title")
+			}
+			ui.Text(c, title).Font(theme.SerifFont).FontSize(core.FontSize(c, theme.H3Size)).TextColor(k.Text).Grow(1)
+			v.Add = input.Button(c, core.Msg(c, "project_knowledge_panel.add"), input.ButtonOptions{Variant: input.Secondary, Icon: icons.Must("plus"), Disabled: opts.Disabled}).Clicked()
+		})
+		ui.Box(c).FillWidth().Height(9).Draw(func(p *ui.Painter, r ui.Rect) {
+			internal.RuleDiamond(p, r, 6, 5, k.Ornament)
+		})
+		projectKnowledgeMeter(c, k, used, opts.Capacity)
+		if len(files) == 0 {
+			ui.Text(c, core.Msg(c, "project_knowledge_panel.empty")).TextColor(k.TextMuted).Padding(d.Unit()*2, 0)
+			return
+		}
+		ui.Column(c).Gap(2).FocusGroup(ui.Vertical).Children(func() {
+			for _, f := range files {
+				ui.Row(c).Key(f.ID).MinHeight(d.ControlHeight()).Gap(d.Unit()*2).AlignItems(ui.Center).Padding(0, d.Unit()).Children(func() {
+					ui.Icon(c, icons.Must("file-text")).Size(16, 16).TextColor(k.TextMuted)
+					ui.Text(c, f.Name).SingleLine().Grow(1).MinWidth(0)
+					ui.Text(c, fmt.Sprintf(core.Msg(c, "project_knowledge_panel.tokens"), f.Tokens)).
+						FontSize(core.FontSize(c, theme.CaptionSize)).TextColor(k.TextMuted)
+					x := core.IconAction(c, icons.Must("x"), fmt.Sprintf(core.Msg(c, "project_knowledge_panel.remove"), f.Name))
+					if x.Clicked() && !x.IsDisabled() {
+						v.Removed = f.ID
+					}
+				})
+			}
+		})
+	})
+	if v.Add || v.Removed != "" {
+		slog.Debug("mujica: project_knowledge_panel action", "add", v.Add, "remove", v.Removed)
+	}
+	return v
+}
+
+// projectKnowledgeMeter shows usage; limits get words and icons.
+func projectKnowledgeMeter(c *ui.Context, k theme.Tokens, used, capacity int) {
+	d := core.Density(c)
+	frac := float64(used) / float64(capacity)
+	ink, note, icon := k.AccentText, "", ""
+	switch {
+	case used > capacity:
+		ink, note, icon = k.Danger, core.Msg(c, "project_knowledge_panel.over"), "circle-alert"
+	case frac >= 0.8:
+		ink, note, icon = k.Warning, core.Msg(c, "project_knowledge_panel.near"), "triangle-alert"
+	}
+	pct := int(frac * 100)
+	label := fmt.Sprintf(core.Msg(c, "project_knowledge_panel.usage"), used, capacity, pct)
+	ui.Column(c).Gap(d.Unit()).Role(ui.RoleMeter).Label(label).Children(func() {
+		ui.Box(c).FillWidth().Height(6).Draw(func(p *ui.Painter, r ui.Rect) {
+			p.Fill(r, k.Border, 3)
+			p.Fill(ui.Rect{X: r.X, Y: r.Y, W: r.W * float32(min(frac, 1)), H: r.H}, ink, 3)
+		})
+		ui.Row(c).Gap(d.Unit()).AlignItems(ui.Center).Children(func() {
+			if icon != "" {
+				ui.Icon(c, icons.Must(icon)).Size(14, 14).TextColor(ink)
+			}
+			t := ui.Text(c, label).FontSize(core.FontSize(c, theme.CaptionSize)).TextColor(k.TextMuted)
+			if note != "" {
+				t.TextColor(ink)
+				ui.Text(c, note).FontSize(core.FontSize(c, theme.CaptionSize)).TextColor(ink).Role(ui.RoleStatus)
+			}
+		})
+	})
+}