From b02d1049d5eb7c175fdbec981d0b19ab312bc188 Mon Sep 17 00:00:00 2001
From: whilefoo <whilefoo@users.noreply.github.com>
Date: Thu, 22 Feb 2024 23:03:50 +0100
Subject: [PATCH] feat: ref

---
 src/github/handlers/index.ts          | 12 ++++++++----
 src/github/types/config.ts            |  2 +-
 src/github/utils/workflow-dispatch.ts |  2 +-
 3 files changed, 10 insertions(+), 6 deletions(-)

diff --git a/src/github/handlers/index.ts b/src/github/handlers/index.ts
index e8a3c44..f57a619 100644
--- a/src/github/handlers/index.ts
+++ b/src/github/handlers/index.ts
@@ -3,7 +3,7 @@ import { GitHubEventHandler } from "../github-event-handler";
 import { getConfig } from "../utils/config";
 import { issueCommentCreated } from "./issue-comment/created";
 import { repositoryDispatch } from "./repository-dispatch";
-import { dispatchWorkflow } from "../utils/workflow-dispatch";
+import { dispatchWorkflow, getDefaultBranch } from "../utils/workflow-dispatch";
 
 function tryCatchWrapper(fn: (event: EmitterWebhookEvent) => unknown) {
   return async (event: EmitterWebhookEvent) => {
@@ -46,14 +46,15 @@ async function handleEvent(event: EmitterWebhookEvent, eventHandler: InstanceTyp
   for (const { workflow, settings } of handler) {
     console.log(`Calling handler for event ${event.name} and workflow ${workflow}`);
 
+    const ref = workflow.ref ?? (await getDefaultBranch(context, workflow.owner, workflow.repository));
     const token = await eventHandler.getToken(event.payload.installation.id);
-    const inputs = new DelegatedComputeInputs(context.key, event, settings, token);
+    const inputs = new DelegatedComputeInputs(context.key, event, settings, token, ref);
 
     await dispatchWorkflow(context, {
       owner: workflow.owner,
       repository: workflow.repository,
       workflowId: workflow.workflowId,
-      ref: workflow.branch,
+      ref: workflow.ref,
       inputs: inputs.getInputs(),
     });
   }
@@ -64,12 +65,14 @@ class DelegatedComputeInputs<T extends EmitterWebhookEventName = EmitterWebhookE
   public event: EmitterWebhookEvent<T>;
   public settings: unknown;
   public authToken: string;
+  public ref: string;
 
-  constructor(eventName: T, event: EmitterWebhookEvent<T>, settings: unknown, authToken: string) {
+  constructor(eventName: T, event: EmitterWebhookEvent<T>, settings: unknown, authToken: string, ref: string) {
     this.eventName = eventName;
     this.event = event;
     this.settings = settings;
     this.authToken = authToken;
+    this.ref = ref;
   }
 
   public getInputs() {
@@ -78,6 +81,7 @@ class DelegatedComputeInputs<T extends EmitterWebhookEventName = EmitterWebhookE
       event: JSON.stringify(this.event),
       settings: JSON.stringify(this.settings),
       authToken: this.authToken,
+      ref: this.ref,
     };
   }
 }
diff --git a/src/github/types/config.ts b/src/github/types/config.ts
index 9176d12..a93b4be 100644
--- a/src/github/types/config.ts
+++ b/src/github/types/config.ts
@@ -13,7 +13,7 @@ const handlerSchema = T.Array(
       owner: T.String(),
       repository: T.String(),
       workflowId: T.String(),
-      branch: T.Optional(T.String()),
+      ref: T.Optional(T.String()),
     }),
     settings: T.Optional(T.Unknown()),
   }),
diff --git a/src/github/utils/workflow-dispatch.ts b/src/github/utils/workflow-dispatch.ts
index 036c6e5..d10b280 100644
--- a/src/github/utils/workflow-dispatch.ts
+++ b/src/github/utils/workflow-dispatch.ts
@@ -32,7 +32,7 @@ export async function dispatchWorkflow(context: GitHubContext, options: Workflow
   });
 }
 
-async function getDefaultBranch(context: GitHubContext, owner: string, repository: string) {
+export async function getDefaultBranch(context: GitHubContext, owner: string, repository: string) {
   const repo = await context.octokit.repos.get({
     owner: owner,
     repo: repository,