Implement 'useradd' CLI command for basic auth

Author: AmirAbrams

command.go

@@ -8,6 +8,8 @@ import (
 	"time"
 
 	"dyn-https/blockchain/dynamic"
+	"dyn-https/configs"
+	"dyn-https/configs/settings"
 	"dyn-https/https/rest"
 	"dyn-https/util"
 
@@ -29,7 +31,7 @@ func unlockWallet(d *dynamic.Dynamicd) bool {
 
 func appCommandLoop(acc *[]dynamic.Account, al *dynamic.ActiveLinks,
 	shutdown *rest.AppShutdown, d *dynamic.Dynamicd,
-	status *dynamic.SyncStatus, sync bool) {
+	status *dynamic.SyncStatus, sync bool, c *settings.Configuration) {
 	go func() {
 		var err error
 		var unlocked = false
@@ -74,6 +76,13 @@ func appCommandLoop(acc *[]dynamic.Account, al *dynamic.ActiveLinks,
 					}
 				} else if strings.HasPrefix(cmdText, "restart") {
 					rest.RestartWebServiceRouter()
+				} else if strings.HasPrefix(cmdText, "useradd") {
+					user, err := configs.AdminUserCommand(c, cmdText)
+					if err != nil {
+						util.Error.Println("User add command failed. %v\n", err)
+					} else {
+						util.Info.Printf("User %v successfully added or updated\n", user)
+					}
 				} else {
 					util.Warning.Println("Invalid command", cmdText)
 					status, err = d.GetSyncStatus()

configs/adminuser.go

@@ -0,0 +1,38 @@
+package configs
+
+import (
+	"crypto/sha256"
+	"encoding/base64"
+	"errors"
+	"fmt"
+	"strings"
+
+	"dyn-https/configs/settings"
+	"dyn-https/models"
+)
+
+func AdminUserCommand(c *settings.Configuration, cmd string) (string, error) {
+	cmd = strings.Replace(cmd, "useradd ", "", -1)
+	spaceindex := strings.Index(cmd, " ")
+	if spaceindex <= 0 {
+		return "", errors.New("Invalid paramaters")
+	}
+	params := strings.Split(cmd, " ")
+	if len(params) != 2 {
+		return "", fmt.Errorf("Invalid number of paramaters %d", len(params))
+	}
+	username := params[0]
+	password := params[1]
+	if len(password) < 8 {
+		return username, fmt.Errorf("Invalid password length %d. Minimum password must be at least 8 charactors.", len(password))
+	}
+	// todo: check password complexity
+	hash := sha256.Sum256([]byte(username + password))
+	expectedHash := base64.StdEncoding.EncodeToString(hash[:])
+	admin := models.Admin{UserName: username, ExpectedHash: expectedHash}
+	if len(admin.ExpectedHash) == 0 || len(admin.UserName) == 0 {
+		return username, fmt.Errorf("Invalid number of paramaters %d", len(params))
+	}
+	c.UpdateAdmins(admin)
+	return username, nil
+}

configs/settings/configuration.go

@@ -118,8 +118,9 @@ func (c *Configuration) Load(dir, seperator string) error {
 		util.Info.Println("Configuration loaded successfully.")
 	}
 	if c.configFile.Admins == nil {
-		admin := models.Admin{UserName: "admin", ExpectedHash: "u9cYLNDulUiPGh5vP+DY+U7Q0U5NsdznE/6CoyMcUj0="}
-		admins := []models.Admin{admin}
+		// admin with password test
+		//admin := models.Admin{UserName: "admin", ExpectedHash: "u9cYLNDulUiPGh5vP+DY+U7Q0U5NsdznE/6CoyMcUj0="}
+		admins := []models.Admin{}
 		c.configFile.Admins = admins
 		c.updateFile()
 	}
@@ -141,3 +142,23 @@ func (c *Configuration) AdminArrayToMap() map[string]string {
 	}
 	return admins
 }
+
+func (c *Configuration) UpdateAdmins(admin models.Admin) {
+	c.mut.Lock()
+	defer c.mut.Unlock()
+	index := -1
+	for i, value := range c.configFile.Admins {
+		if value.UserName == admin.UserName {
+			index = i
+			break
+		}
+	}
+	if index != -1 {
+		c.configFile.Admins[index] = admin
+	} else {
+		c.configFile.Admins = append(c.configFile.Admins, admin)
+	}
+	// update config file
+	c.updateFile()
+	return
+}

init.go

@@ -148,7 +148,7 @@ func Init(version, githash string) error {
 			util.Info.Printf("Found %v links\n", len(al.Links))
 		}
 
-		appCommandLoop(acc, al, &shutdown, dynamicd, status, sync)
+		appCommandLoop(acc, al, &shutdown, dynamicd, status, sync, &config)
 
 		for {
 			select {