feat: config file utils

Author: traviolus

cmd/root.go

@@ -3,12 +3,11 @@ package cmd
 import (
 	"context"
 	"github.com/initia-labs/weave/models"
+	"github.com/initia-labs/weave/utils"
 
 	tea "github.com/charmbracelet/bubbletea"
 	"github.com/spf13/cobra"
 	"github.com/spf13/viper"
-
-	"github.com/initia-labs/weave/states"
 )
 
 func Execute() error {
@@ -17,13 +16,12 @@ func Execute() error {
 		Use:     "weave",
 		Long:    "Weave is a CLI for managing Initia deployments.",
 		PersistentPreRunE: func(cmd *cobra.Command, args []string) error {
-			if err := viper.BindPFlags(cmd.Flags()); err != nil {
-				return err
-			}
 			viper.AutomaticEnv()
 			viper.SetEnvPrefix("weave")
 
-			states.GetGlobalStorage() // Ensure global storage is initialized
+			if err := utils.InitializeConfig(); err != nil {
+				return err
+			}
 
 			return nil
 		},

models/homepage.go

@@ -1,6 +1,7 @@
 package models
 
 import (
+	"fmt"
 	tea "github.com/charmbracelet/bubbletea"
 
 	"github.com/initia-labs/weave/models/weaveinit"
@@ -9,6 +10,8 @@ import (
 
 type Homepage struct {
 	utils.Selector[HomepageOption]
+	TextInput        utils.TextInput
+	isFirstTimeSetup bool
 }
 
 type HomepageOption string
@@ -25,6 +28,8 @@ func NewHomepage() tea.Model {
 			},
 			Cursor: 0,
 		},
+		isFirstTimeSetup: utils.IsFirstTimeSetup(),
+		TextInput:        "",
 	}
 }
 
@@ -33,6 +38,19 @@ func (m *Homepage) Init() tea.Cmd {
 }
 
 func (m *Homepage) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
+	if m.isFirstTimeSetup {
+		input, done := m.TextInput.Update(msg)
+		if done {
+			err := utils.SetConfig("common.gas_station_mnemonic", string(input))
+			if err != nil {
+				return nil, nil
+			}
+			return NewHomepage(), nil
+		}
+		m.TextInput = input
+		return m, nil
+	}
+
 	selected, cmd := m.Select(msg)
 	if selected != nil {
 		switch *selected {
@@ -46,12 +64,24 @@ func (m *Homepage) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 
 func (m *Homepage) View() string {
 	view := "Hi 👋🏻 Weave is a CLI for managing Initia deployments.\n"
-	for i, option := range m.Options {
-		if i == m.Cursor {
-			view += "(•) " + string(option) + "\n"
-		} else {
-			view += "( ) " + string(option) + "\n"
+
+	if m.isFirstTimeSetup {
+		view += fmt.Sprintf(
+			"It looks like this is your first time using Weave. Let's get started!\n"+
+				"Please set up a Gas Station account (The account that will hold the funds required by the "+
+				"OPinit-bots or relayer to send transactions):\n> %s\n", m.TextInput,
+		)
+	} else {
+		view += "What would you like to do today?\n"
+		for i, option := range m.Options {
+			if i == m.Cursor {
+				view += "(•) " + string(option) + "\n"
+			} else {
+				view += "( ) " + string(option) + "\n"
+			}
 		}
+		view += "\nPress Enter to select, or q to quit."
 	}
-	return view + "\nPress Enter to select, or q to quit."
+
+	return view
 }

models/weaveinit/run_l1_node.go

@@ -7,14 +7,28 @@ import (
 )
 
 type RunL1Node struct {
-	utils.TextInput
+	utils.Selector[L1NodeNetworkOption]
 	state *RunL1NodeState
 }
 
+type L1NodeNetworkOption string
+
+const (
+	Mainnet L1NodeNetworkOption = "Mainnet"
+	Testnet L1NodeNetworkOption = "Testnet"
+	Local   L1NodeNetworkOption = "Local"
+)
+
 func NewRunL1Node(state *RunL1NodeState) *RunL1Node {
 	return &RunL1Node{
-		TextInput: "",
-		state:     state,
+		Selector: utils.Selector[L1NodeNetworkOption]{
+			Options: []L1NodeNetworkOption{
+				Mainnet,
+				Testnet,
+				Local,
+			},
+		},
+		state: state,
 	}
 }
 
@@ -23,16 +37,24 @@ func (m *RunL1Node) Init() tea.Cmd {
 }
 
 func (m *RunL1Node) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
-	input, done := m.TextInput.Update(msg)
-	if done {
-		m.state.gasStationMnemonic = string(input)
+	selected, cmd := m.Select(msg)
+	if selected != nil {
+		m.state.network = string(*selected)
 		fmt.Println("[info] state ", m.state)
 		return m, tea.Quit
 	}
-	m.TextInput = input
-	return m, nil
+
+	return m, cmd
 }
 
 func (m *RunL1Node) View() string {
-	return fmt.Sprintf("? Please set up a Gas Station account (The account that will hold the funds required by the OPinit-bots or relayer to send transactions)\nYou can also set this up later. Weave will not send any transactions without your confirmation.\n> %s\n", m.TextInput)
+	view := "? Which network will your node participate in?\n"
+	for i, option := range m.Options {
+		if i == m.Cursor {
+			view += "(•) " + string(option) + "\n"
+		} else {
+			view += "( ) " + string(option) + "\n"
+		}
+	}
+	return view + "\nPress Enter to select, or q to quit."
 }

models/weaveinit/state.go

@@ -1,5 +1,5 @@
 package weaveinit
 
 type RunL1NodeState struct {
-	gasStationMnemonic string
+	network string
 }

utils/config.go

@@ -0,0 +1,75 @@
+package utils
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+
+	"github.com/spf13/viper"
+)
+
+func InitializeConfig() error {
+	homeDir, err := os.UserHomeDir()
+	if err != nil {
+		return fmt.Errorf("failed to get user home directory: %v", err)
+	}
+
+	configPath := filepath.Join(homeDir, ".weave", "config.json")
+	if err := os.MkdirAll(filepath.Dir(configPath), os.ModePerm); err != nil {
+		return fmt.Errorf("failed to create config directory: %v", err)
+	}
+
+	viper.SetConfigFile(configPath)
+	viper.SetConfigType("json")
+
+	if _, err := os.Stat(configPath); os.IsNotExist(err) {
+		if err := createEmptyConfigFile(configPath); err != nil {
+			return fmt.Errorf("failed to create default config file: %v", err)
+		}
+	}
+
+	return LoadConfig()
+}
+
+func createEmptyConfigFile(filePath string) error {
+	file, err := os.Create(filePath)
+	if err != nil {
+		return fmt.Errorf("failed to create config file: %v", err)
+	}
+	defer file.Close()
+
+	// Initialize with an empty JSON object
+	_, err = file.WriteString("{}")
+	if err != nil {
+		return fmt.Errorf("failed to write to config file: %v", err)
+	}
+
+	return nil
+}
+
+func LoadConfig() error {
+	if err := viper.ReadInConfig(); err != nil {
+		return fmt.Errorf("failed to read config file: %v", err)
+	}
+	return nil
+}
+
+func GetConfig(key string) interface{} {
+	return viper.Get(key)
+}
+
+func SetConfig(key string, value interface{}) error {
+	viper.Set(key, value)
+	return WriteConfig()
+}
+
+func WriteConfig() error {
+	if err := viper.WriteConfig(); err != nil {
+		return fmt.Errorf("failed to write config file: %v", err)
+	}
+	return nil
+}
+
+func IsFirstTimeSetup() bool {
+	return viper.Get("common.gas_station_mnemonic") == nil
+}