Merge pull request raykrueger#8 from skrueger-ftc/stop-command

feat: Add Discord command to manually stop Game Server
seakrueger · Jun 12, 2023 · d79b53c · d79b53c
2 parents 217927f + 6515b6b
commit d79b53c
Show file tree

Hide file tree

Showing 3 changed files with 35 additions and 3 deletions.
diff --git a/resources/discord-bot-asl.json b/resources/discord-bot-asl.json
@@ -13,7 +13,12 @@
         {
           "Variable": "$.SubCommand",
           "StringEquals": "start",
-          "Next": "UpdateService"
+          "Next": "StartService"
+        },
+        {
+          "Variable": "$.SubCommand",
+          "StringEquals": "stop",
+          "Next": "StopService"
         }
       ],
       "Default": "WutMessage"
@@ -26,7 +31,7 @@
       },
       "ResultPath": "$.Discord"
     },
-    "UpdateService": {
+    "StartService": {
       "Type": "Task",
       "Next": "RunningCheck",
       "Parameters": {
@@ -68,6 +73,21 @@
       },
       "ResultPath": "$.Discord"
     },
+    "StopService": {
+      "Type": "Task",
+      "Next": "DescribeServices",
+      "Parameters": {
+        "Cluster": "",
+        "Service": "",
+        "DesiredCount": 0
+      },
+      "Resource": "arn:aws:states:::aws-sdk:ecs:updateService",
+      "ResultSelector": {
+        "DesiredCount.$": "$.Service.DesiredCount",
+        "RunningCount.$": "$.Service.RunningCount"
+      },
+      "ResultPath": "$.Service"
+    },
     "DescribeServices": {
       "Type": "Task",
       "Parameters": {

diff --git a/resources/functions/discord_provider/discord_provider.py b/resources/functions/discord_provider/discord_provider.py
@@ -35,6 +35,11 @@ def on_create(event):
                 "description": "Start the server, if it isn't running",
                 "type": 1,
             },
+            {
+                "name": "stop",
+                "description": "Stop the server, if it is running",
+                "type": 1,
+            },
         ],
     }
 

diff --git a/src/discord-state-machine.ts b/src/discord-state-machine.ts
@@ -51,13 +51,20 @@ export class DiscordStateMachine extends Construct {
             ],
           },
         },
-        UpdateService: {
+        StartService: {
           Parameters: {
             Cluster: props.service.cluster.clusterArn,
             Service: props.service.serviceArn,
             DesiredCount: 1,
           },
         },
+        StopService: {
+          Parameters: {
+            Cluster: props.service.cluster.clusterArn,
+            Service: props.service.serviceArn,
+            DesiredCount: 0,
+          },
+        },
         SendDiscordResponse: {
           Parameters: {
             FunctionName: f.functionArn,