151 lines
2.8 KiB
Go
151 lines
2.8 KiB
Go
// Copyright 2017-2025 WIT.COM Inc. All rights reserved.
|
|
// Use of this source code is governed by the GPL 3.0
|
|
|
|
package main
|
|
|
|
// An app to submit patches for the 30 GO GUI repos
|
|
|
|
import (
|
|
"embed"
|
|
"os"
|
|
"path/filepath"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/google/uuid"
|
|
"go.wit.com/dev/alexflint/arg"
|
|
"go.wit.com/gui"
|
|
"go.wit.com/lib/protobuf/chatpb"
|
|
"go.wit.com/log"
|
|
)
|
|
|
|
// sent via -ldflags
|
|
var VERSION string
|
|
var BUILDTIME string
|
|
|
|
// this optionally can store the GUI plugins
|
|
//
|
|
//go:embed resources/*
|
|
var resources embed.FS
|
|
|
|
// used for shell auto completion
|
|
var ARGNAME string = "regex"
|
|
|
|
// using this for now. triggers config save
|
|
var configSave bool
|
|
|
|
func main() {
|
|
// f, _ := os.OpenFile("/tmp/regex.secret.log", os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
|
|
// log.CaptureMode(f)
|
|
me = new(mainType)
|
|
gui.InitArg()
|
|
me.pp = arg.MustParse(&argv)
|
|
|
|
if argv.Bash {
|
|
argv.doBash()
|
|
os.Exit(0)
|
|
}
|
|
if len(argv.BashAuto) != 0 {
|
|
argv.doBashAuto()
|
|
os.Exit(0)
|
|
}
|
|
|
|
// load the default chat protobuf
|
|
me.chats = chatpb.NewChats()
|
|
if err := me.chats.ConfigLoad(); err != nil {
|
|
badExit(err)
|
|
}
|
|
|
|
// verify all the chats have Uuid's
|
|
if verifyUuids(me.chats) {
|
|
me.chats.ConfigSave()
|
|
}
|
|
|
|
aiClient, err := doConnect()
|
|
if err != nil {
|
|
badExit(err)
|
|
}
|
|
_ = aiClient
|
|
|
|
if argv.JsonFile != "" {
|
|
// now try to Marshal() into a protobuf
|
|
pb, err := parsePB(argv.JsonFile)
|
|
if err != nil {
|
|
badExit(err)
|
|
}
|
|
log.Info("GeminiContent pb.Marshal() worked pb.Contents len =", len(pb.Contents))
|
|
_, filename := filepath.Split(argv.JsonFile)
|
|
parts := strings.Split(filename, ".")
|
|
|
|
if len(parts) == 5 {
|
|
uuid := parts[1]
|
|
num, _ := strconv.Atoi(parts[3])
|
|
log.Info(uuid, parts)
|
|
if chat := me.chats.FindByUuid(uuid); chat != nil {
|
|
log.Info("FOUND CHAT", uuid, num)
|
|
newEntry := new(chatpb.ChatEntry)
|
|
newEntry.GeminiRequest = pb
|
|
newEntry.ContentFile = filename
|
|
newEntry.RequestCounter = int32(num)
|
|
chat.AppendEntry(newEntry)
|
|
me.chats.ConfigSave()
|
|
}
|
|
} else {
|
|
}
|
|
okExit("")
|
|
}
|
|
|
|
if argv.Interact != nil {
|
|
log.Info("testing AI client with simpleHello()")
|
|
err = simpleHello(aiClient)
|
|
if err != nil {
|
|
badExit(err)
|
|
}
|
|
|
|
doInteract()
|
|
okExit("")
|
|
}
|
|
|
|
if argv.Stats != "" {
|
|
doStats()
|
|
okExit("")
|
|
}
|
|
|
|
if argv.NewChat != nil {
|
|
doNewChat()
|
|
okExit("")
|
|
}
|
|
|
|
if argv.Playback != nil {
|
|
if argv.Uuid != "" {
|
|
showChat(argv.Uuid)
|
|
} else {
|
|
doPlayback()
|
|
}
|
|
okExit("")
|
|
}
|
|
|
|
// doGui()
|
|
|
|
// by default, start interacting with gemini-cli
|
|
me.pp.WriteHelp(os.Stdout)
|
|
okExit("")
|
|
}
|
|
|
|
func verifyUuids(chats *chatpb.Chats) bool {
|
|
var changed bool
|
|
for _, chat := range chats.GetChats() {
|
|
if chat.GetUuid() == "" {
|
|
chat.Uuid = uuid.New().String()
|
|
changed = true
|
|
}
|
|
for _, entry := range chat.GetEntries() {
|
|
if entry.GetUuid() == "" {
|
|
entry.Uuid = uuid.New().String()
|
|
changed = true
|
|
}
|
|
}
|
|
}
|
|
return changed
|
|
}
|