regex/windowChats.go

// Copyright 2017-2025 WIT.COM Inc. All rights reserved.
// Use of this source code is governed by the GPL 3.0

package main

import (
	"fmt"
	"time"

	"go.wit.com/lib/gadgets"
	"go.wit.com/lib/protobuf/chatpb"
	"go.wit.com/log"
)

/*
type stdReposTableWin struct {
	sync.Mutex
	win    *gadgets.GenericWindow // the machines gui window
	boxTB  *gui.Node              // the machines gui parent box widget
	TB     *gitpb.ReposTable      // the gui table buffer
	pb     *gitpb.Repos           // the current repos protobuf
	update bool                   // if the window should be updated
}

func (w *stdReposTableWin) Toggle() {
	if w == nil {
		return
	}
	if w.win == nil {
		return
	}
	w.win.Toggle()
}
*/

func makeChatsWindow() *gadgets.GenericWindow {
	insertWin := gadgets.NewGenericWindow("regex Chats", "Display Chats")
	insertWin.Win.Custom = func() {
		log.Info("test delete window here")
	}
	grid := insertWin.Group.RawGrid()

	var t *chatpb.ChatsTable
	grid.NewButton("show", func() {
		if t != nil {
			log.Info("Delete the table first")
			return
		}
		// display the protobuf
		t = addChatsPB(insertWin, me.chats)
		f := func(chat *chatpb.Chat) {
			log.Info("got to ChatTable.Custom() id =", chat.GetUuid(), chat.GetChatName())
		}
		t.Custom(f)
		log.Info("table has uuid", t.GetUuid())
	})

	grid.NewButton("delete PB table", func() {
		if t != nil {
			t.Delete()
			t = nil
			log.Info("Table should have been deleted")
		}
	})
	// display the protobuf
	t = addChatsPB(insertWin, me.chats)
	f := func(chat *chatpb.Chat) {
		log.Info("got to ChatTable.Custom() id =", chat.GetUuid(), chat.GetChatName())
	}
	t.Custom(f)
	log.Info("table has uuid", t.GetUuid())

	return insertWin
}

func addChatsPB(win *gadgets.GenericWindow, pb *chatpb.Chats) *chatpb.ChatsTable {
	t := pb.NewTable("testForgeRepos")
	t.NewUuid()
	tbox := win.Bottom.Box().SetProgName("TBOX")
	t.SetParent(tbox)

	sf := t.AddStringFunc("uuid", func(r *chatpb.Chat) string {
		return r.GetUuid()
	})
	sf.Custom = func(r *chatpb.Chat) {
		log.Info("todo: fix mouseClick() on stringFunc in GUI table", r.GetUuid())
	}

	// add a general show button
	bf := t.AddButtonFunc("cur version", func(chat *chatpb.Chat) string { return "show" })
	bf.Custom = func(r *chatpb.Chat) {
		log.Info("todo: show a chat window here", r.GetUuid())
	}

	// show the age of the chat
	t.AddTimeFunc("age", func(chat *chatpb.Chat) time.Time {
		return chat.GetCtime().AsTime()
	})
	t.AddChatName()

	// make a button to show the ChatEntries
	entryButton := t.AddButtonFunc("Entries", func(chat *chatpb.Chat) string {
		return fmt.Sprintf("%d", len(chat.GetEntries()))
	})
	entryButton.Custom = func(chat *chatpb.Chat) {
		log.Info("show entries for", chat.GetUuid())
		chat.PrintChatEntriesTable()
	}

	// make a button to show the Stats (old stuff from gemini-cli)
	statsButton := t.AddButtonFunc("Stats", func(chat *chatpb.Chat) string {
		return fmt.Sprintf("%d", len(chat.GetSession()))
	})
	statsButton.Custom = func(chat *chatpb.Chat) {
		log.Info("show gemini-cli /stats for", chat.GetUuid())
		chat.PrintChatStatsTable()
	}

	// make a button to show content in the *genai.GeminiRequest structures
	genaiButton := t.AddButtonFunc("# of genai.Req's", func(chat *chatpb.Chat) string {
		var counter int
		for _, entry := range chat.GetEntries() {
			if entry.GeminiRequest != nil {
				counter += 1
			}
		}
		return fmt.Sprintf("%d", counter)
	})
	genaiButton.Custom = func(chat *chatpb.Chat) {
		log.Info("show *genai.GeminiRequsts for", chat.GetUuid())
		chat.PrintChatGeminiTable()
		pb := makeBooksTable(chat)
		makeBookWindow(pb)
	}

	// draw the tabel (send the gui protobuf to the GO plugin)
	t.ShowTable()
	return t
}

func makeBooksTable(chat *chatpb.Chat) *chatpb.Books {
	var pb *chatpb.Books

	pb = chatpb.NewBooks()

	for _, entry := range chat.GetEntries() {
		if entry.GeminiRequest == nil {
			continue
		}
		newb := new(chatpb.Book)
		newb.Ctime = entry.Ctime
		newb.Uuid = entry.Uuid
		newb.From = entry.From
		newb.Title = entry.ContentFile
		newb.GeminiRequest = entry.GeminiRequest
		pb.Append(newb)
	}
	return pb
}