feat(account): add login command for saving token to system keyring

Author: kangasta

CHANGELOG.md

@@ -10,6 +10,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ### Added
 
 - Experimental support for reading password or token from system keyring.
+- Experimental support from saving token to system keyring with `upctl account login --with-token`.
 
 ### Changed
 

internal/commands/account/login.go

@@ -0,0 +1,70 @@
+package account
+
+import (
+	"bufio"
+	"fmt"
+	"strings"
+
+	"github.com/UpCloudLtd/upcloud-cli/v3/internal/commands"
+	"github.com/UpCloudLtd/upcloud-cli/v3/internal/config"
+	"github.com/UpCloudLtd/upcloud-cli/v3/internal/output"
+	"github.com/spf13/pflag"
+)
+
+// LoginCommand creates the "account login" command
+func LoginCommand() commands.Command {
+	return &loginCommand{
+		BaseCommand: commands.New(
+			"login",
+			"Configure an authentication token to the system keyring.",
+			"upctl account login --with-token",
+		),
+	}
+}
+
+type loginCommand struct {
+	*commands.BaseCommand
+
+	withToken config.OptionalBoolean
+}
+
+// InitCommand implements Command.InitCommand
+func (s *loginCommand) InitCommand() {
+	fs := &pflag.FlagSet{}
+	config.AddToggleFlag(fs, &s.withToken, "with-token", false, "Read token from standard input.")
+	s.AddFlags(fs)
+
+	// Require the with-token flag until we support using browser to authenticate.
+	commands.Must(s.Cobra().MarkFlagRequired("with-token"))
+}
+
+// DoesNotUseServices implements commands.OfflineCommand as this command does not use services
+func (s *loginCommand) DoesNotUseServices() {}
+
+// ExecuteWithoutArguments implements commands.NoArgumentCommand
+func (s *loginCommand) ExecuteWithoutArguments(exec commands.Executor) (output.Output, error) {
+	if s.withToken.Value() {
+		return s.executeWithToken(exec)
+	}
+
+	return output.None{}, nil
+}
+
+func (s *loginCommand) executeWithToken(exec commands.Executor) (output.Output, error) {
+	in := bufio.NewReader(s.Cobra().InOrStdin())
+	token, err := in.ReadString('\n')
+	if err != nil {
+		return nil, fmt.Errorf("failed to read token from standard input: %w", err)
+	}
+
+	msg := "Saving provided token to the system keyring."
+	exec.PushProgressStarted(msg)
+	err = config.SaveToken(strings.TrimSpace(token))
+	if err != nil {
+		return commands.HandleError(exec, msg, err)
+	}
+
+	exec.PushProgressSuccess(msg)
+
+	return output.None{}, nil
+}

internal/commands/all/all.go

@@ -115,6 +115,7 @@ func BuildCommands(rootCmd *cobra.Command, conf *config.Config) {
 
 	// Account
 	accountCommand := commands.BuildCommand(account.BaseAccountCommand(), rootCmd, conf)
+	commands.BuildCommand(account.LoginCommand(), accountCommand.Cobra(), conf)
 	commands.BuildCommand(account.ShowCommand(), accountCommand.Cobra(), conf)
 	commands.BuildCommand(account.ListCommand(), accountCommand.Cobra(), conf)
 	commands.BuildCommand(account.DeleteCommand(), accountCommand.Cobra(), conf)

internal/config/config.go

@@ -249,6 +249,10 @@ func GetVersion() string {
 	return version
 }
 
+func SaveToken(token string) error {
+	return keyring.Set(keyringServiceName, "", token)
+}
+
 func getVersion() string {
 	// Version was overridden during the build
 	if Version != "dev" {