Add temporary ID support to assign_milestone safe output

actions/setup/js/assign_milestone.cjs

@@ -8,6 +8,7 @@
 const { getErrorMessage } = require("./error_helpers.cjs");
 const { logStagedPreviewInfo } = require("./staged_preview.cjs");
 const { createAuthenticatedGitHubClient } = require("./handler_auth.cjs");
+const { loadTemporaryIdMapFromResolved, resolveRepoIssueTarget } = require("./temporary_id.cjs");
 
 /** @type {string} Safe output type handled by this module */
 const HANDLER_TYPE = "assign_milestone";
@@ -57,17 +58,37 @@ async function main(config = {}) {
 
     const item = message;
 
-    const issueNumber = Number(item.issue_number);
-    const milestoneNumber = Number(item.milestone_number);
+    // Convert resolvedTemporaryIds to a normalized Map for resolveRepoIssueTarget
+    const temporaryIdMap = loadTemporaryIdMapFromResolved(resolvedTemporaryIds);
+
+    // Resolve issue_number, which may be a temporary ID (e.g. "aw_abc123") or a plain number
+    const resolvedIssueTarget = resolveRepoIssueTarget(item.issue_number, temporaryIdMap, context.repo.owner, context.repo.repo);
 
-    if (isNaN(issueNumber) || issueNumber <= 0) {
+    // If the issue_number is a temporary ID that hasn't been resolved yet, defer processing
+    if (resolvedIssueTarget.wasTemporaryId && !resolvedIssueTarget.resolved) {
+      core.info(`Deferring assign_milestone: unresolved temporary ID (${item.issue_number})`);
+      return {
+        success: false,
+        deferred: true,
+        error: resolvedIssueTarget.errorMessage || `Unresolved temporary ID: ${item.issue_number}`,
+      };
+    }
+
+    if (resolvedIssueTarget.errorMessage || !resolvedIssueTarget.resolved) {
       core.error(`Invalid issue_number: ${item.issue_number}`);
       return {
         success: false,
         error: `Invalid issue_number: ${item.issue_number}`,
       };
     }
 
+    const issueNumber = resolvedIssueTarget.resolved.number;
+    if (resolvedIssueTarget.wasTemporaryId) {
+      core.info(`Resolved temporary ID '${item.issue_number}' to issue #${issueNumber}`);
+    }
+
+    const milestoneNumber = Number(item.milestone_number);
+
     if (isNaN(milestoneNumber) || milestoneNumber <= 0) {
       core.error(`Invalid milestone_number: ${item.milestone_number}`);
       return {

actions/setup/js/assign_milestone.test.cjs

@@ -209,4 +209,45 @@ describe("assign_milestone (Handler Factory Architecture)", () => {
     expect(result.success).toBe(false);
     expect(result.error).toContain("Invalid milestone_number");
   });
+
+  it("should resolve a temporary ID for issue_number", async () => {
+    mockGithub.rest.issues.update.mockResolvedValue({});
+
+    const message = {
+      type: "assign_milestone",
+      issue_number: "aw_abc123",
+      milestone_number: 5,
+    };
+
+    const resolvedTemporaryIds = {
+      aw_abc123: { repo: "test-owner/test-repo", number: 42 },
+    };
+
+    const result = await handler(message, resolvedTemporaryIds);
+
+    expect(result.success).toBe(true);
+    expect(result.issue_number).toBe(42);
+    expect(result.milestone_number).toBe(5);
+    expect(mockGithub.rest.issues.update).toHaveBeenCalledWith({
+      owner: "test-owner",
+      repo: "test-repo",
+      issue_number: 42,
+      milestone: 5,
+    });
+  });
+
+  it("should defer when temporary ID is not yet resolved", async () => {
+    const message = {
+      type: "assign_milestone",
+      issue_number: "aw_pending1",
+      milestone_number: 5,
+    };
+
+    // No resolved temporary IDs provided
+    const result = await handler(message, {});
+
+    expect(result.success).toBe(false);
+    expect(result.deferred).toBe(true);
+    expect(mockGithub.rest.issues.update).not.toHaveBeenCalled();
+  });
 });

actions/setup/js/safe_outputs_tools.json

@@ -567,7 +567,7 @@
       "properties": {
         "issue_number": {
           "type": ["number", "string"],
-          "description": "Issue number to assign to the milestone. This is the numeric ID from the GitHub URL (e.g., 567 in github.com/owner/repo/issues/567)."
+          "description": "Issue number to assign to the milestone. This is the numeric ID from the GitHub URL (e.g., 567 in github.com/owner/repo/issues/567). Can also be a temporary_id (e.g., 'aw_abc123') from a previously created issue in the same workflow run."
         },
         "milestone_number": {
           "type": ["number", "string"],

pkg/workflow/js/safe_outputs_tools.json

@@ -663,7 +663,7 @@
             "number",
             "string"
           ],
-          "description": "Issue number to assign to the milestone. This is the numeric ID from the GitHub URL (e.g., 567 in github.com/owner/repo/issues/567)."
+          "description": "Issue number to assign to the milestone. This is the numeric ID from the GitHub URL (e.g., 567 in github.com/owner/repo/issues/567). Can also be a temporary_id (e.g., 'aw_abc123') from a previously created issue in the same workflow run."
         },
         "milestone_number": {
           "type": [

pkg/workflow/safe_outputs_validation_config.go

@@ -99,7 +99,7 @@ var ValidationConfig = map[string]TypeValidationConfig{
 	"assign_milestone": {
 		DefaultMax: 1,
 		Fields: map[string]FieldValidation{
-			"issue_number":     {IssueOrPRNumber: true},
+			"issue_number":     {IssueNumberOrTemporaryID: true},
 			"milestone_number": {Required: true, PositiveInteger: true},
 			"repo":             {Type: "string", MaxLength: 256}, // Optional: target repository in format "owner/repo"
 		},

pkg/workflow/testdata/wasm_golden/TestWasmGolden_CompileFixtures/basic-copilot.golden

@@ -54,6 +54,8 @@ jobs:
         uses: actions/github-script@ed597411d8f924073f98dfc5c65a23a2325f34cd # v8
         with:
           script: |
+            const { setupGlobals } = require('${{ runner.temp }}/gh-aw/actions/setup_globals.cjs');
+            setupGlobals(core, github, context, exec, io);
             const { main } = require('${{ runner.temp }}/gh-aw/actions/generate_aw_info.cjs');
             await main(core, context);
       - name: Validate COPILOT_GITHUB_TOKEN secret

pkg/workflow/testdata/wasm_golden/TestWasmGolden_CompileFixtures/smoke-copilot.golden

@@ -68,6 +68,8 @@ jobs:
         uses: actions/github-script@ed597411d8f924073f98dfc5c65a23a2325f34cd # v8
         with:
           script: |
+            const { setupGlobals } = require('${{ runner.temp }}/gh-aw/actions/setup_globals.cjs');
+            setupGlobals(core, github, context, exec, io);
             const { main } = require('${{ runner.temp }}/gh-aw/actions/generate_aw_info.cjs');
             await main(core, context);
       - name: Validate COPILOT_GITHUB_TOKEN secret

pkg/workflow/testdata/wasm_golden/TestWasmGolden_CompileFixtures/with-imports.golden

@@ -54,6 +54,8 @@ jobs:
         uses: actions/github-script@ed597411d8f924073f98dfc5c65a23a2325f34cd # v8
         with:
           script: |
+            const { setupGlobals } = require('${{ runner.temp }}/gh-aw/actions/setup_globals.cjs');
+            setupGlobals(core, github, context, exec, io);
             const { main } = require('${{ runner.temp }}/gh-aw/actions/generate_aw_info.cjs');
             await main(core, context);
       - name: Validate COPILOT_GITHUB_TOKEN secret