Created InitialSetup method that handles all steps required for starting the server for the first time

2020-01-20 15:17:43 -05:00
parent 398dfd04fc
commit 9e5ffaaff4
6 changed files with 108 additions and 42 deletions
@@ -0,0 +1,69 @@
+package server
+
+import (
+	"fmt"
+	"time"
+
+	"github.com/cloudsonic/sonic-server/consts"
+	"github.com/cloudsonic/sonic-server/log"
+	"github.com/cloudsonic/sonic-server/model"
+	"github.com/google/uuid"
+)
+
+func initialSetup(ds model.DataStore) {
+	_ = ds.WithTx(func(tx model.DataStore) error {
+		_, err := ds.Property().Get(consts.InitialSetupFlagKey)
+		if err == nil {
+			return nil
+		}
+		log.Warn("Running initial setup")
+		if err = createDefaultUser(ds); err != nil {
+			return err
+		}
+		if err = createJWTSecret(ds); err != nil {
+			return err
+		}
+
+		err = ds.Property().Put(consts.InitialSetupFlagKey, time.Now().String())
+		return err
+	})
+}
+
+func createJWTSecret(ds model.DataStore) error {
+	_, err := ds.Property().Get(consts.JWTSecretKey)
+	if err == nil {
+		return nil
+	}
+	jwtSecret, _ := uuid.NewRandom()
+	log.Warn("Creating JWT secret, used for encrypting UI sessions")
+	err = ds.Property().Put(consts.JWTSecretKey, jwtSecret.String())
+	if err != nil {
+		log.Error("Could not save JWT secret in DB", err)
+	}
+	return err
+}
+
+func createDefaultUser(ds model.DataStore) error {
+	c, err := ds.User().CountAll()
+	if err != nil {
+		panic(fmt.Sprintf("Could not access User table: %s", err))
+	}
+	if c == 0 {
+		id, _ := uuid.NewRandom()
+		initialPassword, _ := uuid.NewRandom()
+		log.Warn("Creating initial user. Please change the password!", "user", consts.InitialUserName, "password", initialPassword)
+		initialUser := model.User{
+			ID:       id.String(),
+			UserName: consts.InitialUserName,
+			Name:     consts.InitialName,
+			Email:    "",
+			Password: initialPassword.String(),
+			IsAdmin:  true,
+		}
+		err := ds.User().Put(&initialUser)
+		if err != nil {
+			log.Error("Could not create initial user", "user", initialUser, err)
+		}
+	}
+	return err
+}