main.go

package main

import (
	"cody-gateway-cli/app"
	"cody-gateway-cli/config"
	"os"

	"github.com/fatih/color"
	"github.com/howeyc/gopass"
	"github.com/jessevdk/go-flags"
)

// options are command-line options that are provided by the user.
type options struct {
	Verbose                bool   `short:"V" long:"verbose" description:"Enable verbose output"`
	GatewayHost            string `long:"host" description:"Define an alternate SSH Port" default:"22"`
	GatewayToken           bool   `long:"accesstoken" description:"Ask for a password to use for cody gateway authentication"`
	DebugSecretToken       string `short:"s" long:"debugtoken" description:"Define a bearer secret token to use" env:"SECRET_TOKEN"`
	CompletionMode         string `short:"m" long:"mode" description:"Define chat or code completion mode"`
	VersionAPI             bool   `long:"versionapi" description:"Invoke Cody Gateway version api call"`
	HealthCheckAPI         bool   `long:"healthcheckapi" description:"Invoke Cody Gateway health check api call"`
	AnthropicCompletionAPI bool   `long:"anthropicapi" description:"Invoke Anthropic Completion API"`
	EmbeddingsAPI          bool   `long:"embeddingsapi" description:"Invoke Embeddings API call"`
	OpenAICompletionAPI    bool   `long:"openaiapi" description:"Invoke OpenAI Completions API call"`
}

func main() {
	var opts options
	_, err := flags.ParseArgs(&opts, os.Args[1:])

	if err != nil {
		os.Exit(1)
	}

	// Convert to internal config
	cfg := config.New()
	cfg.Verbose = opts.Verbose
	cfg.GatewayHost = opts.GatewayHost

	if opts.DebugSecretToken != "" {
		cfg.DebugSecretToken = opts.DebugSecretToken
	}

	if opts.HealthCheckAPI {
		cfg.HealthCheckAPI = opts.HealthCheckAPI
	}

	if opts.VersionAPI {
		cfg.VersionAPI = opts.VersionAPI
	}

	if opts.GatewayToken {
		color.White("Enter Cody Gateway Access Token for ")
		p, err := gopass.GetPasswd()
		if err != nil {
			color.Red("Unable to obtain Cody Gateway Access Token: %s", err)
		}
		cfg.GatewayToken = string(p)
	}

	if opts.AnthropicCompletionAPI || opts.OpenAICompletionAPI {
		if opts.CompletionMode != "" {
			cfg.CompletionMode = opts.CompletionMode
		} else {
			color.Red("Require --mode chat | code when calling Anthropic or OpenAI APIs")
		}
		cfg.AnthropicCompletionAPI = opts.AnthropicCompletionAPI
		cfg.OpenAICompletionAPI = opts.OpenAICompletionAPI
	}

	if opts.EmbeddingsAPI {
		cfg.EmbeddingsAPI = opts.EmbeddingsAPI
	}

	// Run the App
	err = app.Run(cfg)
	if err != nil {
		if cfg.Verbose {
			color.Red("Error executing: %s", err)
		}
		os.Exit(1)
	}
	if cfg.Verbose {
		color.Green("Execution completed successfully")
	}
}