added mode to SC register request

Author: YourTechBud

model/validate.go

@@ -5,6 +5,7 @@ type RegisterRequest struct {
 	ID      string `json:"id"` // This is the space cloud id
 	Key     string `json:"key"`
 	Account string `json:"account"`
+	Mode    int    `json:"mode"`
 }
 
 // RegisterResponse is the response to the register request

utils/admin/admin.go

@@ -42,7 +42,7 @@ func (m *Manager) SetOperationMode(op *config.OperationConfig) error {
 
 	if op.Mode > 0 {
 		// Start the validation process for higher op modes
-		if err := m.validator.startValidation(m.nodeID, op.Email, op.Key); err != nil {
+		if err := m.validator.startValidation(m.nodeID, op.Email, op.Key, op.Mode); err != nil {
 			return err
 		}
 	} else {

utils/admin/validate.go

@@ -24,13 +24,13 @@ func newValidator(cb func()) *validator {
 }
 
 // Start starts the validation process
-func (v *validator) startValidation(id, account, key string) error {
+func (v *validator) startValidation(id, account, key string, mode int) error {
 	// Set validation status to active
 	v.lock.Lock()
 	v.active = true
 	v.lock.Unlock()
 
-	if err := v.registerSpaceCloud(id, account, key); err != nil {
+	if err := v.registerSpaceCloud(id, account, key, mode); err != nil {
 		return err
 	}
 
@@ -56,7 +56,7 @@ func (v *validator) startValidation(id, account, key string) error {
 				return
 			}
 
-			if err := v.registerSpaceCloud(id, account, key); err != nil {
+			if err := v.registerSpaceCloud(id, account, key, mode); err != nil {
 				log.Println("Validate: Error -", err)
 			} else {
 				timer = time.Now()

utils/admin/websocket.go

@@ -35,8 +35,8 @@ func (v *validator) connect(url string) error {
 	return nil
 }
 
-func (v *validator) write(id, account, key string) error {
-	req := &model.RegisterRequest{ID: id, Account: account, Key: key}
+func (v *validator) write(id, account, key string, mode int) error {
+	req := &model.RegisterRequest{ID: id, Account: account, Key: key, Mode: mode}
 	msg := &model.Message{Type: utils.TypeRegisterRequest, Data: req}
 	return v.socket.WriteJSON(msg)
 }
@@ -69,13 +69,13 @@ func (v *validator) routineRead() error {
 	}
 }
 
-func (v *validator) registerSpaceCloud(id, account, secret string) error {
+func (v *validator) registerSpaceCloud(id, account, secret string, mode int) error {
 	err := v.connect(url)
 	if err != nil {
 		return err
 	}
 
-	err = v.write(id, account, secret)
+	err = v.write(id, account, secret, mode)
 	if err != nil {
 		return err
 	}