💄 Optimized CLI

pkg/cmd/rds/conn/commands.go

@@ -0,0 +1,89 @@
+package conn
+
+import (
+	"encoding/json"
+	"fmt"
+
+	"github.com/rs/zerolog/log"
+	"github.com/samber/lo"
+	"github.com/spf13/viper"
+	"github.com/urfave/cli/v2"
+)
+
+var CliCommands = []*cli.Command{
+	{
+		Name:        "list",
+		Aliases:     []string{"ls"},
+		Description: "List all connected remote server",
+		Action: func(ctx *cli.Context) error {
+			var servers []CliConnection
+			raw, _ := json.Marshal(viper.Get("servers"))
+			_ = json.Unmarshal(raw, &servers)
+
+			log.Info().Msgf("There are %d server(s) connected in total.", len(servers))
+			for idx, server := range servers {
+				log.Info().Msgf("%d) %s: %s", idx+1, server.ID, server.Url)
+			}
+
+			return nil
+		},
+	},
+	{
+		Name:        "connect",
+		Aliases:     []string{"add"},
+		Description: "Connect and save configuration of remote server",
+		ArgsUsage:   "<id> <server url> <credential>",
+		Action: func(ctx *cli.Context) error {
+			if ctx.Args().Len() < 3 {
+				return fmt.Errorf("must have three arguments: <id> <server url> <credential>")
+			}
+
+			c := CliConnection{
+				ID:         ctx.Args().Get(0),
+				Url:        ctx.Args().Get(1),
+				Credential: ctx.Args().Get(2),
+			}
+
+			if err := c.GetConnectivity(); err != nil {
+				return fmt.Errorf("couldn't connect server: %s", err.Error())
+			} else {
+				var servers []CliConnection
+				raw, _ := json.Marshal(viper.Get("servers"))
+				_ = json.Unmarshal(raw, &servers)
+				viper.Set("servers", append(servers, c))
+
+				if err := viper.WriteConfig(); err != nil {
+					return err
+				} else {
+					log.Info().Msg("Successfully connected a new remote server, enter \"rds ls\" to get more info.")
+					return nil
+				}
+			}
+		},
+	},
+	{
+		Name:        "disconnect",
+		Aliases:     []string{"remove"},
+		Description: "Remove a remote server configuration",
+		ArgsUsage:   "<id>",
+		Action: func(ctx *cli.Context) error {
+			if ctx.Args().Len() < 1 {
+				return fmt.Errorf("must have more one arguments: <server url>")
+			}
+
+			var servers []CliConnection
+			raw, _ := json.Marshal(viper.Get("servers"))
+			_ = json.Unmarshal(raw, &servers)
+			viper.Set("servers", lo.Filter(servers, func(item CliConnection, idx int) bool {
+				return item.ID != ctx.Args().Get(0)
+			}))
+
+			if err := viper.WriteConfig(); err != nil {
+				return err
+			} else {
+				log.Info().Msg("Successfully disconnected a remote server, enter \"rds ls\" to get more info.")
+				return nil
+			}
+		},
+	},
+}

pkg/cmd/rds/conn/connect.go

@@ -0,0 +1,37 @@
+package conn
+
+import (
+	"encoding/json"
+	"fmt"
+
+	roadsign "code.smartsheep.studio/goatworks/roadsign/pkg"
+	"github.com/gofiber/fiber/v2"
+	"github.com/rs/zerolog/log"
+)
+
+type CliConnection struct {
+	ID         string `json:"id"`
+	Url        string `json:"url"`
+	Credential string `json:"credential"`
+}
+
+func (v CliConnection) GetConnectivity() error {
+	client := fiber.Get(v.Url + "/cgi/connectivity")
+	client.BasicAuth("RoadSign CLI", v.Credential)
+
+	if status, data, err := client.Bytes(); len(err) > 0 {
+		return fmt.Errorf("couldn't connect to server: %q", err)
+	} else if status != 200 {
+		return fmt.Errorf("server rejected request, may cause by invalid credential")
+	} else {
+		var resp fiber.Map
+		if err := json.Unmarshal(data, &resp); err != nil {
+			return err
+		} else if resp["server"] != "RoadSign" {
+			return fmt.Errorf("remote server isn't roadsign")
+		} else if resp["version"] != roadsign.AppVersion {
+			log.Warn().Msg("Server connected successfully, but remote server version mismatch than CLI version, some features may buggy or completely unusable.")
+		}
+	}
+	return nil
+}

pkg/cmd/rds/main.go

@@ -0,0 +1,47 @@
+package main
+
+import (
+	"os"
+
+	roadsign "code.smartsheep.studio/goatworks/roadsign/pkg"
+	"code.smartsheep.studio/goatworks/roadsign/pkg/cmd/rds/conn"
+	"github.com/rs/zerolog"
+	"github.com/rs/zerolog/log"
+	"github.com/spf13/viper"
+	"github.com/urfave/cli/v2"
+)
+
+func init() {
+	zerolog.TimeFieldFormat = zerolog.TimeFormatUnix
+	log.Logger = log.Output(zerolog.ConsoleWriter{Out: os.Stdout})
+}
+
+func main() {
+	// Configure settings
+	viper.AddConfigPath("$HOME")
+	viper.SetConfigName(".roadsignrc")
+	viper.SetConfigType("yaml")
+
+	// Load settings
+	if err := viper.ReadInConfig(); err != nil {
+		if _, ok := err.(viper.ConfigFileNotFoundError); ok {
+			viper.SafeWriteConfig()
+			viper.ReadInConfig()
+		} else {
+			log.Panic().Err(err).Msg("An error occurred when loading settings.")
+		}
+	}
+
+	// Configure CLI
+	app := &cli.App{
+		Name:     "RoadSign CLI",
+		Version:  roadsign.AppVersion,
+		Suggest:  true,
+		Commands: append([]*cli.Command{}, conn.CliCommands...),
+	}
+
+	// Run CLI
+	if err := app.Run(os.Args); err != nil {
+		log.Fatal().Err(err).Msg("An error occurred when running cli.")
+	}
+}