// Copyright 2017-2025 WIT.COM Inc. All rights reserved. // Use of this source code is governed by the GPL 3.0 package main import ( "embed" "os" "unicode" "go.wit.com/lib/protobuf/argvpb" "go.wit.com/lib/protobuf/filepb" "go.wit.com/log" ) // sent via -ldflags var VERSION string var BUILDTIME string // used for shell auto completion var ARGNAME string = "wit" // todo: get this from $0 ? //go:embed resources/* var resources embed.FS func main() { me = new(mainType) // autocomplete must be processed before there is anything sent to STDOUT or STDERR me.argv = argvpb.Autocomplete(&argv) // adds shell auto complete to go-args me.homedir, _ = os.UserHomeDir() // store shortcut here todo: add better logic if me.argv.Cmd == "" { // user didn't enter a sub command // doDefaultBehavior() me.argv.GoodExit("do what?") } pwd, _ := os.Getwd() setTitle(log.Sprintf("wit %s %s", me.argv.Cmd, pwd)) // Standard subcommand handling starts here var s string var err error if argv.Upgrade != nil { doUpgrade() me.argv.GoodExit("") } if argv.Linux != nil { if argv.Linux.Rdate != nil { s, err = doRdate() } else { setTerminalTitle("pinging google", "ping", []string{"google.com"}) } } if argv.Build != nil { s, err = doBuild() } if argv.Git != nil { initMachine() s, err = doGit() } if argv.PB != nil { pbuuid, pbver, pberr := filepb.IdentifyPB(argv.PB.Identify) if pberr == nil { log.Info("pb version is:", pbver) log.Info("pb uuid is:", pbuuid) s = "pb identify worked" } else { s = "identify failed" err = pberr } } if argv.Clone != nil { doClone() me.argv.GoodExit("") } if argv.WITCOM { doWITCOM() me.argv.GoodExit("") } if argv.Clone != nil { doClone() me.argv.GoodExit("") } if argv.WITCOM { doWITCOM() me.argv.GoodExit("") } if argv.Upgrade != nil { doUpgrade() } if argv.Droplet != nil { s, err = doDroplet() } if argv.Publish != nil { if err := doPublish(); err != nil { me.argv.BadExit("doPublish failed", err) } me.argv.GoodExit("") } if argv.Zoo != nil { if areSuperuser() { exitOnErrorRealtime([]string{"journalctl", "-n", "100", "-f", "_SYSTEMD_UNIT=zood.service"}) } me.argv.GoodExit("do something here") } if err != nil { me.argv.BadExit(s, err) } me.argv.GoodExit(s) } // this is dumb. sync this with go-deb func trimNonNumericFromStart(s string) string { for i, r := range s { if unicode.IsDigit(r) { return s[i:] } } return "" }