124 lines
2.3 KiB
Go
124 lines
2.3 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"
|
|
|
|
"github.com/google/uuid"
|
|
"go.wit.com/dev/alexflint/arg"
|
|
"go.wit.com/lib/gui/prep"
|
|
"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() {
|
|
me = new(mainType)
|
|
prep.Bash(ARGNAME, argv.DoAutoComplete) // this line should be: prep.Bash(argv)
|
|
me.myGui = prep.Gui() // prepares the GUI package for go-args
|
|
me.pp = arg.MustParse(&argv)
|
|
|
|
var err error
|
|
// 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()
|
|
}
|
|
|
|
// Get the last chat
|
|
numChats := len(me.chats.GetChats())
|
|
if numChats > 0 {
|
|
me.lastChat = me.chats.GetChats()[numChats-1]
|
|
log.Printf("The current Gemini API session is UUID: %s\n", me.lastChat.GetUuid())
|
|
}
|
|
|
|
err = initGeminiAPI()
|
|
if err != nil {
|
|
badExit(err)
|
|
}
|
|
|
|
if argv.JsonFile != "" {
|
|
doJSON()
|
|
okExit("")
|
|
}
|
|
|
|
if argv.Interact != nil {
|
|
log.Info("testing AI client with simpleHello()")
|
|
err = simpleHello()
|
|
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("")
|
|
}
|
|
|
|
if argv.Clean != nil {
|
|
doClean()
|
|
okExit("")
|
|
}
|
|
|
|
me.myGui.Start() // loads the GUI toolkit
|
|
doGui() // start making our forge GUI
|
|
debug() // sits here forever
|
|
}
|
|
|
|
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
|
|
}
|