refactor: update mobius server cli

mobius/cmd/cli.go

@@ -5,13 +5,29 @@ import (
 	"github.com/spf13/cobra"
 )
 
+const Description = `
+
+
+███▄ ▄███▓ ▒█████   ▄▄▄▄    ██▓ █    ██   ██████ 
+▓██▒▀█▀ ██▒▒██▒  ██▒▓█████▄ ▓██▒ ██  ▓██▒▒██    ▒ 
+▓██    ▓██░▒██░  ██▒▒██▒ ▄██▒██▒▓██  ▒██░░ ▓██▄   
+▒██    ▒██ ▒██   ██░▒██░█▀  ░██░▓▓█  ░██░  ▒   ██▒
+▒██▒   ░██▒░ ████▓▒░░▓█  ▀█▓░██░▒▒█████▓ ▒██████▒▒
+░ ▒░   ░  ░░ ▒░▒░▒░ ░▒▓███▀▒░▓  ░▒▓▒ ▒ ▒ ▒ ▒▓▒ ▒ ░
+░  ░      ░  ░ ▒ ▒░ ▒░▒   ░  ▒ ░░░▒░ ░ ░ ░ ░▒  ░ ░
+░      ░   ░ ░ ░ ▒   ░    ░  ▒ ░ ░░░ ░ ░ ░  ░  ░  
+       ░       ░ ░   ░       ░     ░           ░  
+                          ░                       
+                                                  
+                                                     
+🌈 Mobius is an open-source, lightweight, high-performance ai studio gateway
+`
+
 func New(cliConfig *Config) *cobra.Command {
 	cmd := &cobra.Command{
-		Use:   "mobius <command> <subcommand> [flags]",
-		Short: "🌈 Mobius is an open-source, lightweight, high-performance ai studio gateway",
-		Long: heredoc.Doc(`
-      Mobius is an open-source, lightweight, high-performance ai studio gateway.
-    `),
+		Use:           "mobius <command> <subcommand> [flags]",
+		Short:         "🌈 Mobius is an open-source, lightweight, high-performance ai studio gateway",
+		Long:          heredoc.Doc(Description),
 		SilenceUsage:  true,
 		SilenceErrors: true,
 	}

mobius/cmd/serve.go

@@ -0,0 +1,25 @@
+package cmd
+
+import (
+	"context"
+	"os/signal"
+	"syscall"
+
+	"github.com/missingstudio/studio/backend/config"
+	"github.com/missingstudio/studio/backend/pkg/server"
+	"github.com/missingstudio/studio/common/logger"
+)
+
+func Serve(cfg *config.Config) error {
+	logger := logger.New(cfg.LogFormatJson, nil)
+
+	ctx, cancelFunc := signal.NotifyContext(context.Background(), syscall.SIGTERM, syscall.SIGINT)
+	defer cancelFunc()
+
+	if err := server.Serve(ctx, logger, cfg); err != nil {
+		logger.Error("error starting server", "error", err)
+		return err
+	}
+
+	return nil
+}

mobius/cmd/server.go

@@ -1,15 +1,10 @@
 package cmd
 
 import (
-	"context"
 	"fmt"
-	"os/signal"
-	"syscall"
 
 	"github.com/MakeNowJust/heredoc"
 	"github.com/missingstudio/studio/backend/config"
-	"github.com/missingstudio/studio/backend/pkg/server"
-	"github.com/missingstudio/studio/common/logger"
 
 	"github.com/spf13/cobra"
 )
@@ -66,17 +61,7 @@ func serverStartCommand() *cobra.Command {
 			if err != nil {
 				panic(err)
 			}
-
-			logger := logger.New(appConfig.LogFormatJson, nil)
-
-			ctx, cancelFunc := signal.NotifyContext(context.Background(), syscall.SIGTERM, syscall.SIGINT)
-			defer cancelFunc()
-
-			if err = server.Serve(ctx, logger, appConfig); err != nil {
-				logger.Error("error starting server", "error", err)
-				return err
-			}
-			return nil
+			return Serve(appConfig)
 		},
 	}
 

mobius/main.go

@@ -7,6 +7,7 @@ import (
 )
 
 func main() {
+	// Cli config to connect with GRPC server
 	cliConfig, err := cmd.LoadConfig()
 	if err != nil {
 		cliConfig = &cmd.Config{}