fix get licenseid

Author: diamonwiggins

.github/workflows/mlflow-ci.yml

@@ -203,7 +203,7 @@ jobs:
             exit 1
           fi
           
-          echo "License ID: $INSTALLATION_ID"
+          echo "License ID retrieved successfully"
           echo "license_id=$INSTALLATION_ID" >> $GITHUB_OUTPUT
         env:
           REPLICATED_API_TOKEN: ${{ secrets.REPLICATED_PLATFORM_EXAMPLES_TOKEN }}

applications/mlflow/Taskfile.yml

@@ -254,60 +254,22 @@ tasks:
           exit 1
         fi
         
-        # Create a temporary file to store the vendor-cli output
-        OUTPUT_FILE=$(mktemp)
-        
-        # Run vendor-cli to inspect the customer and get the installation ID
+        # Run vendor-cli to inspect the customer and get the installation ID as JSON
         echo "Running vendor-cli to inspect customer..."
-        docker run --rm \
+        CUSTOMER_JSON=$(docker run --rm \
           -e REPLICATED_API_TOKEN=$REPLICATED_API_TOKEN \
           -e REPLICATED_APP={{.APP_NAME}} \
           replicated/vendor-cli:latest \
-          customer inspect --customer "{{.CUSTOMER_NAME}}" > $OUTPUT_FILE 2>&1
+          customer inspect --customer "{{.CUSTOMER_NAME}}" --output json)
         
-        # Check if the command succeeded
-        if [ $? -ne 0 ]; then
-          echo "ERROR: vendor-cli command failed. Output:"
-          cat $OUTPUT_FILE
-          rm $OUTPUT_FILE
-          exit 1
-        fi
-        
-        # Try direct extraction using grep pattern for the installationId
-        INSTALLATION_ID=$(grep -o '"installationId": *"[^"]*"' $OUTPUT_FILE | grep -o '"[^"]*"$' | tr -d '"' || true)
-        
-        # If grep didn't find it, try as a fallback to parse via jq if the file looks like JSON
-        if [ -z "$INSTALLATION_ID" ] && grep -q "^{" $OUTPUT_FILE; then
-          echo "Attempting to extract installation ID with jq..."
-          set +e # Don't exit on error
-          if command -v jq >/dev/null 2>&1; then
-            INSTALLATION_ID=$(cat $OUTPUT_FILE | jq -r '.installationId // empty' 2>/dev/null || true)
-          fi
-          set -e
-        fi
+        # Use jq to properly extract the installationId
+        INSTALLATION_ID=$(echo "$CUSTOMER_JSON" | jq -r '.installationId')
         
         # Check if we got a valid ID
-        if [ -z "$INSTALLATION_ID" ]; then
-          echo "Failed to extract installationId from vendor-cli output"
-          echo "Output content:" 
-          cat $OUTPUT_FILE | head -n 20
-          
-          # Special case: If the output appears to be just a plain string already, use that
-          if [ $(wc -l < $OUTPUT_FILE) -eq 1 ] && [ $(wc -w < $OUTPUT_FILE) -eq 1 ]; then
-            echo "Output appears to be a single token, using it directly as the license ID"
-            INSTALLATION_ID=$(cat $OUTPUT_FILE | tr -d '[:space:]')
-          else
-            rm $OUTPUT_FILE
-            exit 1
-          fi
-        fi
-        
-        # Clean up the temporary file
-        rm $OUTPUT_FILE
-        
-        # Validate the ID format (alphanumeric)
-        if ! [[ $INSTALLATION_ID =~ ^[a-zA-Z0-9]+$ ]]; then
-          echo "ERROR: Extracted installation ID doesn't match expected format: '$INSTALLATION_ID'"
+        if [ -z "$INSTALLATION_ID" ] || [ "$INSTALLATION_ID" = "null" ]; then
+          echo "Failed to extract installationId from customer JSON"
+          echo "JSON structure:" 
+          echo "$CUSTOMER_JSON" | jq 'del(.installationId)' # Print JSON without the license ID
           exit 1
         fi