summaryrefslogtreecommitdiff
path: root/argv.go
blob: c8fdbafb8c395b77c87ba18dd22dc3532db9bb29 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
// Copyright 2017-2025 WIT.COM Inc. All rights reserved.
// Use of this source code is governed by the GPL 3.0

package main

import (
	"os"

	"go.wit.com/lib/protobuf/argvpb"
)

/*
	this parses the command line arguements using alex flint's go-arg
*/

var argv args

type args struct {
	Uuid     string       `arg:"--uuid"                               help:"look at this uuid"`
	Topic    string       `arg:"--topic"                              help:"set the topic"`
	JsonFile string       `arg:"--json"                               help:"import a JSON file from gemini-cli"`
	Interact *EmptyCmd    `arg:"subcommand:interact"                  help:"open env EDITOR"`
	Playback *PlaybackCmd `arg:"subcommand:playback"                  help:"dump your prior conversations to the terminal'"`
	NewChat  *PlaybackCmd `arg:"subcommand:newchat"                   help:"used by gemini-cli on startup"`
	Clean    *CleanCmd    `arg:"subcommand:clean"                     help:"cleanup the files in /tmp"`
	Stats    string       `arg:"--stats"                              help:"add stats to a chat"`
	Force    bool         `arg:"--force"                              help:"try to strong arm things"`
	Verbose  bool         `arg:"--verbose"                            help:"show more output"`
	Bash     bool         `arg:"--bash"                               help:"generate bash completion"`
	BashAuto []string     `arg:"--auto-complete"                      help:"todo: move this to go-arg"`
}

type EmptyCmd struct {
}

type CleanCmd struct {
	Match string `arg:"positional"`
}

type PlaybackCmd struct {
	Last   *EmptyCmd `arg:"subcommand:last"                      help:"dump the last one"`
	List   *EmptyCmd `arg:"subcommand:list"                      help:"list memories"`
	Long   *EmptyCmd `arg:"subcommand:long"                      help:"show info on each chat"`
	Purge  *EmptyCmd `arg:"subcommand:purge"                     help:"verify chat uuids & purge empty chats"`
	Submit *EmptyCmd `arg:"subcommand:submit"                    help:"convert and submit the last entry"`
}

func (a args) Description() string {
	return `
regex -- interact with Googles' Gemini AI
	`
}

/*
	handles shell autocomplete
*/

func (args) Appname() string {
	return ARGNAME
}

func (args) Buildtime() (string, string) {
	return BUILDTIME, VERSION
}

func (args) Version() string {
	return argvpb.StandardVersion(ARGNAME, VERSION, BUILDTIME)
}

// sends the strings to bash or zsh that will be your options
func (a args) SendCompletionStrings(pb *argvpb.Argv) {
	if pb.Cmd == "" {
		base := []string{"--bash", "interact", "playback", "clean", "--version"}
		pb.SendStrings(base)
	} else {
		pb.SubCommand(pb.Goargs...)
	}
	os.Exit(0)
}