All checks were successful
build and deploy kleinTodo / build (push) Successful in 15s
117 lines
2.8 KiB
Go
117 lines
2.8 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"io"
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
|
|
"gitea.kleinsense.nl/DariusKlein/kleinTodo/client/todo/clientCommon/config"
|
|
"gitea.kleinsense.nl/DariusKlein/kleinTodo/client/todo/httpClient"
|
|
"gitea.kleinsense.nl/DariusKlein/kleinTodo/common"
|
|
"github.com/BurntSushi/toml"
|
|
"github.com/urfave/cli/v3"
|
|
)
|
|
|
|
// Register Command
|
|
func Register() *cli.Command {
|
|
return &cli.Command{
|
|
Name: "register",
|
|
Usage: "Register new user to kleinTodo server",
|
|
Action: registerAction,
|
|
Flags: registerFlags(),
|
|
}
|
|
}
|
|
|
|
// registerFlags Register cli flags
|
|
func registerFlags() []cli.Flag {
|
|
return []cli.Flag{
|
|
&cli.StringFlag{
|
|
Name: usernameFlagName,
|
|
Aliases: []string{"u"},
|
|
Required: true,
|
|
},
|
|
&cli.StringFlag{
|
|
Name: passwordFlagName,
|
|
Aliases: []string{"p"},
|
|
Required: true,
|
|
},
|
|
&cli.StringFlag{
|
|
Name: serverUrlFlagName,
|
|
Aliases: []string{"url"},
|
|
Required: true,
|
|
},
|
|
}
|
|
}
|
|
|
|
// registerAction logic for Template
|
|
func registerAction(context context.Context, c *cli.Command) error {
|
|
var username = c.String(usernameFlagName)
|
|
var password = c.String(passwordFlagName)
|
|
var serverUrl = c.String(serverUrlFlagName)
|
|
|
|
err := registerAndGetToken(serverUrl, username, password)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if (cfg.Server.Credentials.Username != username || cfg.Server.Credentials.Password != password) &&
|
|
common.AskUserBool("Do you wish to save your credentials? (WARNING: stored in plaintext)") {
|
|
cfg.Server.Credentials.Username = username
|
|
cfg.Server.Credentials.Password = password
|
|
}
|
|
|
|
if cfg.Server.Url != serverUrl && common.AskUserBool("Do you wish to save your chosen server url?") {
|
|
cfg.Server.Url = serverUrl
|
|
}
|
|
|
|
path, configPath, err := config.GetConfigPath()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if err = os.MkdirAll(path, 0770); err != nil {
|
|
return err
|
|
}
|
|
configBytes, err := toml.Marshal(cfg)
|
|
err = os.WriteFile(configPath, configBytes, 0644)
|
|
|
|
log.Println("Registered new user to kleinTodo server successfully")
|
|
|
|
return nil
|
|
}
|
|
|
|
func registerAndGetToken(url, username, password string) error {
|
|
credentials := common.Credentials{
|
|
Username: username,
|
|
Password: password,
|
|
}
|
|
|
|
payload, err := json.Marshal(credentials)
|
|
if err != nil {
|
|
return fmt.Errorf("error marshaling credentials: %w", err)
|
|
}
|
|
|
|
req, err := http.NewRequest("POST", url+"/register", bytes.NewBuffer(payload))
|
|
if err != nil {
|
|
return fmt.Errorf("error creating request: %w", err)
|
|
}
|
|
req.Header.Set("Content-Type", "application/json")
|
|
|
|
resp, err := httpClient.GetHttpClient("").Do(req)
|
|
if err != nil {
|
|
return fmt.Errorf("error sending request: %w", err)
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
body, _ := io.ReadAll(resp.Body)
|
|
return fmt.Errorf("register failed with status %d: %s", resp.StatusCode, string(body))
|
|
}
|
|
|
|
return nil
|
|
}
|