add config command

Author: savannahostrowski

.devcontainer/devcontainer.json

@@ -9,7 +9,8 @@
 			"extensions": [
 				"golang.go",
 				"github.vscode-github-actions",
-				"ms-azuretools.vscode-docker"
+				"ms-azuretools.vscode-docker",
+				"github.copilot"
 			]
 		}
 	}

README.md

@@ -5,10 +5,6 @@ Using file names and natural language, Ghost generates a GitHub Action skeleton.
 
 ## Getting started
 1. First, you'll need to set up an [OpenAI API key](https://platform.openai.com/account/api-keys).
-2. Set the API key in your environment variables as `OPENAI_API_KEY` (e.g. `export OPENAI_API_KEY=<your-key-here>`)
+2. Run `ghost config set OPENAI_API_KEY <your key here>` with your key from step 1
 3. Run `ghost run` to start project analysis of the current working directory.
 
-
-## Upcoming features
-- Ability to set other models (currently uses GPT 3.5 Turbo) using a config file
-- VS Code extension

cmd/config.go

@@ -0,0 +1,75 @@
+package cmd
+
+import (
+	"fmt"
+	"os"
+
+	"github.com/charmbracelet/log"
+	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
+)
+
+var configCmd = &cobra.Command{
+	Use:   "config",
+	Short: "Configure Ghost CLI",
+}
+
+var setCmd = &cobra.Command{
+	Use:   "set",
+	Short: "Set a configuration for Ghost",
+	Args: func(cmd *cobra.Command, args []string) error {
+		if err := cobra.MinimumNArgs(2)(cmd, args); err != nil {
+			return err
+		}
+		if args[0] == "OPENAI_API_KEY" {
+			viper.Set(args[0], args[1])
+			viper.WriteConfig()
+			return nil
+		}
+
+		return fmt.Errorf("invalid key: %s", args[0])
+	},
+	Run: func(cmd *cobra.Command, args []string) {
+
+	},
+}
+
+type Config struct {
+	OpenAIAPIKey string `mapstructure:"openai_api_key"`
+}
+
+var getCmd = &cobra.Command{
+	Use:   "get",
+	Short: "Get the configuration for Ghost",
+	Run: func(cmd *cobra.Command, args []string) {
+		home, _ := os.UserHomeDir()
+		path := home + "/.ghost.yaml"
+		config, err := loadConfig(path)
+
+		if err != nil {
+			log.Fatal("cannot load config:", err)
+		}
+		fmt.Println("OPENAI_API_KEY=",config.OpenAIAPIKey)
+
+	},
+}
+
+func init() {
+	configCmd.AddCommand(setCmd, getCmd)
+}
+
+func loadConfig(path string) (config Config, err error) {
+    viper.AddConfigPath(path)
+    viper.SetConfigName(".ghost")
+    viper.SetConfigType("yaml")
+
+    viper.AutomaticEnv()
+
+    err = viper.ReadInConfig()
+    if err != nil {
+        return
+    }
+
+    err = viper.Unmarshal(&config)
+    return
+}

cmd/root.go

@@ -1,13 +1,18 @@
 package cmd
 
 import (
+	"errors"
 	"fmt"
+	"os"
+	"path/filepath"
 
 	"github.com/enescakir/emoji"
 	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
 )
 
 var (
+	cfgFile string
 	rootCmd = &cobra.Command{
 		Use:   "ghost",
 		Short: fmt.Sprintf("\n%v Ghost is an experimental CLI that intelligently scaffolds a GitHub Action workflow based on your local application stack and natural language, using OpenAI.", emoji.Ghost),
@@ -19,6 +24,39 @@ func Execute() error {
 }
 
 func init() {
+	cobra.OnInitialize(InitConfig)
+
+	rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.cobra.yaml)")
+	rootCmd.PersistentFlags().Bool("viper", true, "use Viper for configuration")
+
 	rootCmd.CompletionOptions.HiddenDefaultCmd = true
-	rootCmd.AddCommand(runCmd)
+	rootCmd.AddCommand(runCmd, configCmd)
+}
+
+func InitConfig() {
+	if cfgFile != "" {
+		viper.SetConfigFile(cfgFile)
+	} else {
+		configHome, err := os.UserHomeDir()
+		configName := ".ghost"
+		configType := "yaml"
+
+		cobra.CheckErr(err)
+		viper.AddConfigPath(configHome)
+		viper.SetConfigType(configType)
+		viper.SetConfigName(configName)
+		configPath := filepath.Join(configHome, configName+"."+configType)
+
+
+		if _, err := os.Stat(configPath); err == nil {
+			viper.AutomaticEnv()
+		} else if errors.Is(err, os.ErrNotExist) {
+			if err := viper.SafeWriteConfig(); err != nil {
+				if err != nil {
+					str := fmt.Sprintf("could not write config file: %v", err)
+					panic(str)
+				}
+			}
+		}
+	}
 }

cmd/run.go

@@ -12,13 +12,15 @@ import (
 	"github.com/charmbracelet/bubbles/textinput"
 	"github.com/charmbracelet/bubbles/viewport"
 	tea "github.com/charmbracelet/bubbletea"
+
 	// "golang.org/x/term"
 
 	"github.com/charmbracelet/lipgloss"
 	"github.com/charmbracelet/log"
 	"github.com/enescakir/emoji"
 	"github.com/sashabaranov/go-openai"
 	"github.com/spf13/cobra"
+	"github.com/spf13/viper"
 )
 
 type model struct {
@@ -70,16 +72,24 @@ const (
 
 var runCmd = &cobra.Command{
 	Use:   "run",
-	Short: fmt.Sprintf("%v Run the Ghost CLI", emoji.Rocket),
+	Short: "Run Ghost CLI",
 	Run: func(cmd *cobra.Command, args []string) {
+		err := viper.ReadInConfig()
+		if err != nil {
+			panic(fmt.Errorf("fatal error config file: %w", err))
+		}
+
+		if !viper.IsSet("OPENAI_API_KEY") {
+			log.Error("Please set OPENAI_API_KEY in your environment using `ghost config set OPENAI_API_KEY <your key>`")
+			os.Exit(1)
+		}
+
 		m := initialModel()
 		p := tea.NewProgram(m, tea.WithAltScreen(), tea.WithMouseAllMotion())
-
 		if _, err := p.Run(); err != nil {
 			log.Fatal("Yikes! We've run into a problem: ", err)
 			os.Exit(1)
 		}
-
 	},
 }
 
@@ -98,7 +108,6 @@ func initialModel() model {
 	additionalInfo.CharLimit = 300
 	additionalInfo.Width = 300
 
-	// width, height, _ := term.GetSize(0)
 	vp := viewport.New(0, 0)
 	vp.Style = viewportStyle
 	vp.YPosition = 0
@@ -171,7 +180,7 @@ func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 					m.choice = "yes"
 				}
 			// View to allow the user to specify what tasks they'd like to do in their pipeline
-			case InputTasks, CorrectGHA :
+			case InputTasks, CorrectGHA:
 				if m.desiredTasks.Value() != "" {
 					m.currentView = LoadingGHA
 					cmds = append(cmds, func() tea.Msg {
@@ -202,6 +211,7 @@ func (m model) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 	default:
 		switch m.currentView {
 		case Preload:
+
 			if len(m.files) == 0 {
 				m.files = getFilesInCurrentDirAndSubDirs()
 			}
@@ -300,7 +310,8 @@ func (m model) View() string {
 type gptResponse string
 
 func chatGPTRequest(prompt string) (response gptResponse, err error) {
-	apiKey := os.Getenv("OPENAI_API_KEY")
+	viper.ReadInConfig()
+	apiKey := viper.GetString("openai_api_key")
 	client := openai.NewClient(apiKey)
 	resp, err := client.CreateChatCompletion(
 		context.Background(),

go.mod

@@ -4,9 +4,25 @@ go 1.20
 
 require github.com/sashabaranov/go-openai v1.9.5
 
-require github.com/enescakir/emoji v1.0.0
+require (
+	github.com/enescakir/emoji v1.0.0
+	github.com/spf13/viper v1.16.0
+)
 
-require github.com/atotto/clipboard v0.1.4 // indirect
+require (
+	github.com/atotto/clipboard v0.1.4 // indirect
+	github.com/fsnotify/fsnotify v1.6.0 // indirect
+	github.com/hashicorp/hcl v1.0.0 // indirect
+	github.com/magiconair/properties v1.8.7 // indirect
+	github.com/mitchellh/mapstructure v1.5.0 // indirect
+	github.com/pelletier/go-toml/v2 v2.0.8 // indirect
+	github.com/spf13/afero v1.9.5 // indirect
+	github.com/spf13/cast v1.5.1 // indirect
+	github.com/spf13/jwalterweatherman v1.1.0 // indirect
+	github.com/subosito/gotenv v1.4.2 // indirect
+	gopkg.in/ini.v1 v1.67.0 // indirect
+	gopkg.in/yaml.v3 v3.0.1 // indirect
+)
 
 require (
 	github.com/aymanbagabas/go-osc52/v2 v2.0.1 // indirect
@@ -29,7 +45,7 @@ require (
 	github.com/spf13/cobra v1.7.0
 	github.com/spf13/pflag v1.0.5 // indirect
 	golang.org/x/sync v0.1.0 // indirect
-	golang.org/x/sys v0.7.0 // indirect
+	golang.org/x/sys v0.8.0 // indirect
 	golang.org/x/term v0.7.0 // indirect
 	golang.org/x/text v0.9.0 // indirect
 )