mctl/cmd/login.go

/*
Copyright © 2025 Jake jake.young.dev@gmail.com
*/
package cmd

import (
	"bufio"
	"fmt"
	"os"

	"code.jakeyoungdev.com/jake/mctl/client"
	"code.jakeyoungdev.com/jake/mctl/database"
	"github.com/spf13/cobra"
)

var (
	server string
)

// loginCmd represents the login command
var loginCmd = &cobra.Command{
	Use:          "login",
	Example:      "mctl login -s <server>",
	SilenceUsage: true,
	Short:        "Login to server and send commands",
	Long: `Login to default server and enter command loop. The default server
			is used unless the server flag is set`,
	Run: func(cmd *cobra.Command, args []string) {
		db, err := database.New()
		cobra.CheckErr(err)
		defer db.Close()

		//get default server
		if server == "" {
			ds, err := db.GetServer("default")
			cobra.CheckErr(err)

			server = ds.Name
		}

		cli, err := client.New(server)
		cobra.CheckErr(err)
		defer cli.Close()

		//start command loop
		fmt.Println("Connected! Type 'mctl' to close")
		scanner := bufio.NewScanner(os.Stdin)

		var runningCmd string
		for runningCmd != "mctl" {
			fmt.Printf("RCON@%s /> ", server)

			if scanner.Scan() {
				runningCmd = scanner.Text()

				if runningCmd == "" {
					continue
				}

				//mctl exits command terminal
				if runningCmd == "mctl" {
					break
				}

				res, err := cli.Command(runningCmd)
				cobra.CheckErr(err)
				fmt.Printf("\n%s\n", res)
			}
		}

		fmt.Printf("Disconnected from %s\n", server)
	},
}

func init() {
	loginCmd.Flags().StringVarP(&server, "server", "s", "", "server alias")
	rootCmd.AddCommand(loginCmd)
}