updates

Signed-off-by: Jess Frazelle <[email protected]>

Author: jessfraz

bambulabs/src/message.rs

@@ -93,6 +93,8 @@ pub enum Result {
 #[derive(Debug, Clone, PartialEq, Serialize, Deserialize, JsonSchema)]
 #[serde(rename_all = "snake_case", tag = "command")]
 pub enum Print {
+    /// Ams control.
+    AmsControl(AmsControl),
     /// The status of the print.
     PushStatus(PushStatus),
     /// The gcode line.
@@ -113,6 +115,7 @@ impl Print {
     /// Returns the sequence id of the message.
     pub fn sequence_id(&self) -> SequenceId {
         match self {
+            Print::AmsControl(ams_ctrl) => ams_ctrl.sequence_id.clone(),
             Print::PushStatus(push_status) => push_status.sequence_id.clone(),
             Print::GcodeLine(gcode_line) => gcode_line.sequence_id.clone(),
             Print::ProjectFile(project_file) => project_file.sequence_id.clone(),
@@ -124,6 +127,21 @@ impl Print {
     }
 }
 
+/// An ams control command.
+#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, JsonSchema)]
+pub struct AmsControl {
+    /// The sequence id.
+    pub sequence_id: SequenceId,
+    /// The reason for the message.
+    pub reason: Reason,
+    /// The result of the command.
+    pub result: Result,
+    /// The param.
+    pub param: Option<String>,
+    #[serde(flatten)]
+    other: BTreeMap<String, Value>,
+}
+
 /// A gcode line.
 #[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize, JsonSchema)]
 pub struct GcodeLine {

openapi/api.json

@@ -464,6 +464,54 @@
       "Print": {
         "description": "A print command.",
         "oneOf": [
+          {
+            "additionalProperties": true,
+            "description": "Ams control.",
+            "properties": {
+              "command": {
+                "enum": [
+                  "ams_control"
+                ],
+                "type": "string"
+              },
+              "param": {
+                "description": "The param.",
+                "nullable": true,
+                "type": "string"
+              },
+              "reason": {
+                "allOf": [
+                  {
+                    "$ref": "#/components/schemas/Reason"
+                  }
+                ],
+                "description": "The reason for the message."
+              },
+              "result": {
+                "allOf": [
+                  {
+                    "$ref": "#/components/schemas/Result"
+                  }
+                ],
+                "description": "The result of the command."
+              },
+              "sequence_id": {
+                "allOf": [
+                  {
+                    "$ref": "#/components/schemas/SequenceId"
+                  }
+                ],
+                "description": "The sequence id."
+              }
+            },
+            "required": [
+              "command",
+              "reason",
+              "result",
+              "sequence_id"
+            ],
+            "type": "object"
+          },
           {
             "additionalProperties": true,
             "description": "The status of the print.",