add fsm config load

pkg/deviceshifu/deviceshifubase/configmap_snippet.yaml

@@ -42,3 +42,24 @@ data:
           intervalMs: 1000
           pushSettings:
             pushToServer: false
+  fsm: |
+    states:
+      red: 
+        actions:
+          go:
+            next_state: green
+        forbid:
+          - caution    
+      green:
+        actions:
+          caution:
+            next_state: yellow
+        forbid:
+          - stop
+      yellow:
+        actions:
+          stop:
+            next_state: red     
+        forbid:
+          - go
+    startingState: red

pkg/deviceshifu/deviceshifubase/deviceshifubase.go

@@ -60,6 +60,7 @@ const (
 	PythonHandlersModuleName                     = "customized_handlers"
 	PythonScriptDir                              = "pythoncustomizedhandlers"
 	ControlMsgsConfigStr                         = "controlMsgs"
+	FSMStr                                       = "fsm"
 )
 
 var (

pkg/deviceshifu/deviceshifubase/deviceshifubase_config.go

@@ -24,6 +24,7 @@ type DeviceShifuConfig struct {
 	Telemetries              *DeviceShifuTelemetries
 	CustomInstructionsPython map[string]string `yaml:"customInstructionsPython"`
 	ControlMsgs              map[string]string `yaml:"controlMsgs,omitempty"`
+	FSM                      *DeviceShifuFSM   `yaml:"fsm,omitempty"`
 }
 
 // DeviceShifuDriverProperties properties of deviceshifuDriver
@@ -98,6 +99,20 @@ type EdgeDeviceConfig struct {
 	KubeconfigPath string
 }
 
+type DeviceShifuFSM struct {
+	States        map[string]DeviceShifuState `yaml:"states"`
+	StartingState string                      `yaml:"startingState"`
+}
+
+type DeviceShifuState struct {
+	Actions map[string]DeviceShifuAction `yaml:"actions"`
+	Forbid  []string                     `yaml:"forbid"`
+}
+
+type DeviceShifuAction struct {
+	NextState string `yaml:"next_state"`
+}
+
 // default value
 const (
 	DeviceInstructionInitialDelay            int64 = 3000
@@ -164,6 +179,14 @@ func NewDeviceShifuConfig(path string) (*DeviceShifuConfig, error) {
 		}
 	}
 
+	if fsm, ok := cfg[FSMStr]; ok {
+		err = yaml.Unmarshal([]byte(fsm), &dsc.FSM)
+		if err != nil {
+			logger.Errorf("Error parsing %v from ConfigMap, error: %v", FSMStr, err)
+			return nil, err
+		}
+	}
+
 	err = dsc.load()
 	return dsc, err
 }

pkg/deviceshifu/deviceshifubase/deviceshifubase_config_test.go

@@ -37,6 +37,7 @@ type ConfigMapData struct {
 		DriverProperties string `yaml:"driverProperties"`
 		Instructions     string `yaml:"instructions"`
 		Telemetries      string `yaml:"telemetries"`
+		FSM              string `yaml:"fsm"`
 	} `yaml:"data"`
 }
 
@@ -76,6 +77,30 @@ func TestNewDeviceShifuConfig(t *testing.T) {
 		},
 	}
 
+	var mockDeviceFSM = &DeviceShifuFSM{
+		States: map[string]DeviceShifuState{
+			"red": {
+				Actions: map[string]DeviceShifuAction{
+					"go": {NextState: "green"},
+				},
+				Forbid: []string{"caution"},
+			},
+			"green": {
+				Actions: map[string]DeviceShifuAction{
+					"caution": {NextState: "yellow"},
+				},
+				Forbid: []string{"stop"},
+			},
+			"yellow": {
+				Actions: map[string]DeviceShifuAction{
+					"stop": {NextState: "red"},
+				},
+				Forbid: []string{"go"},
+			},
+		},
+		StartingState: "red",
+	}
+
 	mockdsc, err := NewDeviceShifuConfig(MockDeviceConfigFolder)
 	if err != nil {
 		t.Errorf(err.Error())
@@ -97,6 +122,10 @@ func TestNewDeviceShifuConfig(t *testing.T) {
 		t.Errorf("Telemetries mismatch")
 	}
 
+	eq = reflect.DeepEqual(mockDeviceFSM, mockdsc.FSM)
+	if !eq {
+		t.Errorf("FSM mismatch")
+	}
 }
 
 func GenerateConfigMapFromSnippet(fileName string, folder string) error {
@@ -116,6 +145,7 @@ func GenerateConfigMapFromSnippet(fileName string, folder string) error {
 		path.Join(MockDeviceConfigFolder, ConfigmapDriverPropertiesStr): cmData.Data.DriverProperties,
 		path.Join(MockDeviceConfigFolder, ConfigmapInstructionsStr):     cmData.Data.Instructions,
 		path.Join(MockDeviceConfigFolder, ConfigmapTelemetriesStr):      cmData.Data.Telemetries,
+		path.Join(MockDeviceConfigFolder, FSMStr):                       cmData.Data.FSM,
 	}
 
 	err = os.MkdirAll(MockDeviceConfigFolder, os.ModePerm)

pkg/k8s/api/v1alpha1/edgedevice_types.go

@@ -102,6 +102,7 @@ type EdgeDeviceSpec struct {
 	ProtocolSettings *ProtocolSettings  `json:"protocolSettings,omitempty"`
 	CustomMetadata   *map[string]string `json:"customMetadata,omitempty"`
 
+	State *string `json:"state,omitempty"`
 	// TODO: add other fields like disconnectTimemoutInSeconds
 }
 

pkg/k8s/crd/config/edgedevice/bases/shifu.edgenesis.io_edgedevices.yaml

@@ -124,6 +124,8 @@ spec:
               sku:
                 description: Sku specifies the EdgeDevice's SKU.
                 type: string
+              state:
+                type: string
             type: object
           status:
             description: EdgeDeviceStatus defines the observed state of EdgeDevice

pkg/k8s/crd/install/config_crd.yaml

@@ -123,6 +123,8 @@ spec:
               sku:
                 description: Sku specifies the EdgeDevice's SKU.
                 type: string
+              state:
+                type: string
             type: object
           status:
             description: EdgeDeviceStatus defines the observed state of EdgeDevice

pkg/k8s/crd/install/config_default.yaml

@@ -130,6 +130,8 @@ spec:
               sku:
                 description: Sku specifies the EdgeDevice's SKU.
                 type: string
+              state:
+                type: string
             type: object
           status:
             description: EdgeDeviceStatus defines the observed state of EdgeDevice

pkg/k8s/crd/install/shifu_install.yml

@@ -130,6 +130,8 @@ spec:
               sku:
                 description: Sku specifies the EdgeDevice's SKU.
                 type: string
+              state:
+                type: string
             type: object
           status:
             description: EdgeDeviceStatus defines the observed state of EdgeDevice