Run blueapi in debug mode and run setup prior to attach

Author: DiamondJoseph

.devcontainer/devcontainer.json

@@ -29,7 +29,8 @@
                 "charliermarsh.ruff",
                 "ms-pyright.pyright",
                 "ms-azuretools.vscode-docker",
-                "ms-kubernetes-tools.vscode-kubernetes-tools"
+                "ms-kubernetes-tools.vscode-kubernetes-tools",
+                "rioj7.command-variable",
             ]
         }
     },

.vscode/launch.json

@@ -24,19 +24,22 @@
                 "host": "0.0.0.0",
                 "port": 5678,
             },
+            "preLaunchTask": "kubectl port-forward",
+            "postDebugTask": "pv-unmount",
+            "subProcess": true,
             "pathMappings": [
                 {
-                    "localRoot": "${input:mount_path}",
-                    "remoteRoot": "/workspaces_git"
+                    "localRoot": "${input:Mount Point}",
+                    "remoteRoot": "${input:PVC Name}"
                 },
                 {
-                    "localRoot": "./src/blueapi",
-                    "remoteRoot": "/venv/lib/python3.1*/site-packages/blueapi"
+                    "localRoot": "${workspaceFolder}/src/blueapi",
+                    "remoteRoot": "/venv/lib/python3.11/site-packages/blueapi"
                 },
                 {
-                    "localRoot": ".",
-                    "remoteRoot": "/blueapi"
-                },
+                    "localRoot": "${workspaceFolder}",
+                    "remoteRoot": "/workspaces/blueapi"
+                }
             ]
         },
         {
@@ -86,10 +89,24 @@
     ],
     "inputs": [
         {
-            "id": "mount_path",
-            "type": "promptString",
-            "description": "Mounted location of PVC containing /workspace_git/",
-            "default": ""
+            "id": "PVC Name",
+            "type": "command",
+            "command": "extension.commandvariable.promptStringRemember",
+            "args": {
+                "key": "PVC Name",
+                "description": "Name of the PVC to mount locally",
+                "default": "scratch-1.0.0"
+            }
+        },
+        {
+            "id": "Mount Point",
+            "type": "command",
+            "command": "extension.commandvariable.promptStringRemember",
+            "args": {
+                "key": "Mount Point",
+                "description": "Location to mount PVC locally, must exist as an empty directory",
+                "default": "/scratch/mountPoint"
+            }
         },
         {
             "id": "controller_args",

.vscode/tasks.json

@@ -11,6 +11,79 @@
                 "cwd": "${workspaceRoot}"
             },
             "problemMatcher": [],
+        },
+        {
+            "type": "shell",
+            "label": "kubectl pv-mounter",
+            "command": "kubectl",
+            "args": [
+                "pv-mounter",
+                "mount",
+                "${input:namespace}",
+                "${input:PVC Name}",
+                "${input:Mount Point}",
+                "--cpu-limit",
+                "100m"
+            ],
+            "problemMatcher": [],
+        },
+        {
+            "dependsOn": "kubectl pv-mounter",
+            "type": "shell",
+            "label": "kubectl port-forward",
+            "command": "kubectl",
+            "args": [
+                "port-forward",
+                "-n",
+                "${input:namespace}",
+                "${input:pod}",
+                "5678",
+                "&"
+            ],
+            "problemMatcher": [],
+        },
+        {
+            "type": "shell",
+            "label": "pv-unmount",
+            "command": "kubectl",
+            "args": [
+                "pv-mounter",
+                "clean",
+                "${input:namespace}",
+                "${promptStringRemember:PVC Name}",
+                "${promptStringRemember:Mount Point}",
+            ],
+            "problemMatcher": [],
         }
+    ],
+    "inputs": [
+        {
+            "id": "namespace",
+            "type": "promptString",
+            "description": "Beamline namespace in the cluster",
+            "default": "ixx-beamline"
+        },
+        {
+            "id": "pod",
+            "type": "promptString",
+            "description": "Blueapi pod name",
+            "default": "ixx-blueapi-0"
+        },
+        {
+            "id": "PVC Name",
+            "type": "command",
+            "command": "extension.commandvariable.remember",
+            "args": {
+                "key": "PVC Name"
+            }
+        },
+        {
+            "id": "Mount Point",
+            "type": "command",
+            "command": "extension.commandvariable.remember",
+            "args": {
+                "key": "Mount Point"
+            }
+        },
     ]
 }

helm/blueapi/templates/statefulset.yaml

@@ -155,6 +155,18 @@ spec:
               name: nslcd
             {{- end }}
           {{- if not .Values.debug.enabled }}
+          command: [
+            "python",
+            "-Xfrozen_modules=off",
+            "-m",
+            "debugpy",
+            "--listen",
+            "0.0.0.0:5678",
+            "--configure-subProcess",
+            "true", 
+            "-m",
+            "blueapi"
+          ]
           args:
             - "-c"
             - "/config/config.yaml"