add env.go (#10)

JunNishimura · Jul 15, 2023 · f2a2b0d · f2a2b0d
1 parent 9fb6f4f
commit f2a2b0d
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/env.go b/env.go
@@ -0,0 +1,30 @@
+package main
+
+import (
+	"fmt"
+
+	"github.com/spf13/viper"
+)
+
+const (
+	configPath = "$HOME/.config/chatify"
+)
+
+func setEnv() {
+	// token setting
+	viper.AddConfigPath(configPath)
+	viper.SetConfigName("token")
+	viper.SetConfigType("json")
+
+	// client setting
+	viper.SetConfigName("client")
+	viper.SetConfigType("yaml")
+	viper.MergeInConfig()
+
+	// load
+	if err := viper.ReadInConfig(); err != nil {
+		if _, ok := err.(viper.ConfigFileNotFoundError); ok {
+			fmt.Println("config file does not exist")
+		}
+	}
+}