Add checks for resource group

Author: MDUYN

dist/index.js

@@ -14,7 +14,7 @@ __nccwpck_require__.r(__webpack_exports__);
 
 
 
-async function endpointExists(endpointName, resourceGroup, workspaceName) {
+async function checkIfEndpointExists(endpointName, resourceGroup, workspaceName) {
     try {
         let output = "";
         const options = {
@@ -27,14 +27,54 @@ async function endpointExists(endpointName, resourceGroup, workspaceName) {
         };
 
         // Check if the endpoint exists
-        await _actions_exec__WEBPACK_IMPORTED_MODULE_1__.exec(`az ml online-endpoint show --name ${endpointName} --resource-group ${resourceGroup} --workspace-name ${workspaceName}`, [], options);
+        await exec.exec(`az ml online-endpoint show --name ${endpointName} --resource-group ${resourceGroup} --workspace-name ${workspaceName}`, [], options);
 
         return true; // If the command succeeds, the endpoint exists
     } catch (error) {
         return false; // If the command fails, the endpoint does not exist
     }
 }
 
+async function checkIfResourceGroupExists(resourceGroup) {
+    try {
+        let output = "";
+        const options = {
+            listeners: {
+                stdout: (data) => {
+                    output += data.toString();
+                }
+            },
+            silent: true
+        };
+
+        // Check if the resource group exists
+        await _actions_exec__WEBPACK_IMPORTED_MODULE_1__.exec(`az group show --name ${resourceGroup}`, [], options);
+        return true; // If the command succeeds, the endpoint exists
+    } catch (error) {
+        return false; // If the command fails, the endpoint does not exist
+    }
+}
+
+async function checkIfWorkspaceExists(workspaceName, resourceGroup) {
+    try {
+        let output = "";
+        const options = {
+            listeners: {
+                stdout: (data) => {
+                    output += data.toString();
+                }
+            },
+            silent: true
+        };
+
+        // Check if the workspace exists
+        await _actions_exec__WEBPACK_IMPORTED_MODULE_1__.exec(`az ml workspace show --name ${workspaceName} --resource-group ${resourceGroup}`, [], options);
+        return true; // If the command succeeds, the endpoint exists
+    } catch (error) {
+        return false; // If the command fails, the endpoint does not exist
+    }
+}
+
 try {
     const endpointName = _actions_core__WEBPACK_IMPORTED_MODULE_0__.getInput("endpoint_name");
     const resourceGroup = _actions_core__WEBPACK_IMPORTED_MODULE_0__.getInput("resource_group");
@@ -53,6 +93,23 @@ try {
         throw new Error("Workspace name is required");
     }
 
+    // Check if the resource group exists
+    console.log(`🔹 Checking if resource group '${resourceGroup}' exists...`)
+    ;
+    const resourceGroupExists = await checkIfResourceGroupExists(resourceGroup);
+    if (!resourceGroupExists) {
+        throw new Error(`Resource group '${resourceGroup}' does not exist.`);
+    }
+
+    // Check if the workspace exists
+    console.log(`🔹 Checking if workspace '${workspaceName}' exists in resource group '${resourceGroup}'...`)
+    ;
+    const workspaceExists = await checkIfWorkspaceExists(workspaceName, resourceGroup);
+
+    if (!workspaceExists) {
+        throw new Error(`Workspace '${workspaceName}' does not exist in resource group '${resourceGroup}'.`);
+    }
+
     console.log(`🔹 Checking if endpoint '${endpointName}' exists...`);
     const exists = await endpointExists(
         endpointName, resourceGroup, workspaceName