feat: actions sdk

ubiquity-os · Aug 29, 2024 · c4160c6 · c4160c6
1 parent c91cce2
commit c4160c6
Show file tree

Hide file tree

Showing 6 changed files with 175 additions and 10 deletions.
diff --git a/bun.lockb b/bun.lockb
diff --git a/package.json b/package.json
@@ -48,6 +48,8 @@
     "open-source"
   ],
   "dependencies": {
+    "@actions/core": "^1.10.1",
+    "@actions/github": "^6.0.0",
     "@octokit/auth-app": "7.1.0",
     "@octokit/core": "6.1.2",
     "@octokit/plugin-paginate-rest": "11.3.3",

diff --git a/src/sdk/actions.ts b/src/sdk/actions.ts
@@ -0,0 +1,115 @@
+import * as core from "@actions/core";
+import * as github from "@actions/github";
+import { Context } from "./context";
+import { customOctokit } from "./octokit";
+import { EmitterWebhookEventName as WebhookEventName } from "@octokit/webhooks";
+import { Logs, LogLevel, LOG_LEVEL, LogReturn } from "@ubiquity-dao/ubiquibot-logger";
+import { config } from "dotenv";
+import { Type as T, TAnySchema } from "@sinclair/typebox";
+import { Value } from "@sinclair/typebox/value";
+config();
+
+interface Options {
+  logLevel?: LogLevel;
+  postCommentOnError?: boolean;
+  settingsSchema?: TAnySchema;
+  envSchema?: TAnySchema;
+}
+
+const inputSchema = T.Object({
+  stateId: T.String(),
+  eventName: T.String(),
+  eventPayload: T.String(),
+  authToken: T.String(),
+  settings: T.String(),
+  ref: T.String(),
+});
+
+export async function createActionsPlugin<TConfig = unknown, TEnv = unknown, TSupportedEvents extends WebhookEventName = WebhookEventName>(
+  handler: (context: Context<TConfig, TEnv, TSupportedEvents>) => Promise<Record<string, unknown> | undefined>,
+  options?: Options
+) {
+  const pluginOptions = {
+    logLevel: options?.logLevel || LOG_LEVEL.INFO,
+    postCommentOnError: options?.postCommentOnError || true,
+    settingsSchema: options?.settingsSchema,
+    envSchema: options?.envSchema,
+  };
+
+  const inputs = Value.Decode(inputSchema, github.context.payload.inputs);
+
+  let config: TConfig;
+  if (pluginOptions.settingsSchema) {
+    config = Value.Decode(pluginOptions.settingsSchema, JSON.parse(inputs.settings));
+  } else {
+    config = JSON.parse(inputs.settings) as TConfig;
+  }
+
+  let env: TEnv;
+  if (pluginOptions.envSchema) {
+    env = Value.Decode(pluginOptions.envSchema, process.env);
+  } else {
+    env = process.env as TEnv;
+  }
+
+  const context: Context<TConfig, TEnv, TSupportedEvents> = {
+    eventName: inputs.eventName as TSupportedEvents,
+    payload: JSON.parse(inputs.eventPayload),
+    octokit: new customOctokit({ auth: inputs.authToken }),
+    config: config,
+    env: env,
+    logger: new Logs(pluginOptions.logLevel),
+  };
+
+  try {
+    const result = await handler(context);
+    core.setOutput("result", result);
+    await returnDataToKernel(inputs.authToken, inputs.stateId, result);
+  } catch (error) {
+    console.error(error);
+
+    let loggerError: LogReturn | null;
+    if (error instanceof Error) {
+      core.setFailed(error);
+      loggerError = context.logger.error(`Error: ${error}`, { error: error });
+    } else if (error instanceof LogReturn) {
+      core.setFailed(error.logMessage.raw);
+      loggerError = error;
+    } else {
+      core.setFailed(`Error: ${error}`);
+      loggerError = context.logger.error(`Error: ${error}`);
+    }
+
+    if (pluginOptions.postCommentOnError && loggerError) {
+      await postComment(context, loggerError);
+    }
+  }
+}
+
+async function postComment(context: Context, error: LogReturn) {
+  if ("issue" in context.payload && context.payload.repository?.owner?.login) {
+    await context.octokit.rest.issues.createComment({
+      owner: context.payload.repository.owner.login,
+      repo: context.payload.repository.name,
+      issue_number: context.payload.issue.number,
+      body: `${error.logMessage.diff}\n<!--\n${getGithubWorkflowRunUrl()}\n${JSON.stringify(error.metadata, null, 2)}\n-->`,
+    });
+  }
+}
+
+function getGithubWorkflowRunUrl() {
+  return `${github.context.payload.repository?.html_url}/actions/runs/${github.context.runId}`;
+}
+
+async function returnDataToKernel(repoToken: string, stateId: string, output: object | undefined) {
+  const octokit = new customOctokit({ auth: repoToken });
+  await octokit.rest.repos.createDispatchEvent({
+    owner: github.context.repo.owner,
+    repo: github.context.repo.repo,
+    event_type: "return_data_to_ubiquibot_kernel",
+    client_payload: {
+      state_id: stateId,
+      output: output ? JSON.stringify(output) : null,
+    },
+  });
+}
diff --git a/src/sdk/constants.ts b/src/sdk/constants.ts
@@ -8,3 +8,4 @@ dkRj2Je2kag9b3FMxskv1npNSrPVcSc5lGNYlnZnfxIAnCknOC118JjitlrpT6wd
 8wIDAQAB
 -----END PUBLIC KEY-----
 `;
+export const KERNEL_APP_ID = 0;
diff --git a/src/sdk/index.ts b/src/sdk/index.ts
@@ -1,2 +1,4 @@
 export { createPlugin } from "./server";
+export { createActionsPlugin } from "./actions";
 export type { Context } from "./context";
+export { KERNEL_APP_ID, KERNEL_PUBLIC_KEY } from "./constants";
diff --git a/src/sdk/server.ts b/src/sdk/server.ts
@@ -5,19 +5,32 @@ import { customOctokit } from "./octokit";
 import { EmitterWebhookEventName as WebhookEventName } from "@octokit/webhooks";
 import { verifySignature } from "./signature";
 import { KERNEL_PUBLIC_KEY } from "./constants";
-import { Logs, LogLevel, LOG_LEVEL } from "@ubiquity-dao/ubiquibot-logger";
+import { Logs, LogLevel, LOG_LEVEL, LogReturn } from "@ubiquity-dao/ubiquibot-logger";
 import { Manifest } from "../types/manifest";
+import { TAnySchema } from "@sinclair/typebox";
+import { Value } from "@sinclair/typebox/value";
 
 interface Options {
   kernelPublicKey?: string;
   logLevel?: LogLevel;
+  postCommentOnError?: boolean;
+  settingsSchema?: TAnySchema;
+  envSchema?: TAnySchema;
 }
 
 export async function createPlugin<TConfig = unknown, TEnv = unknown, TSupportedEvents extends WebhookEventName = WebhookEventName>(
   handler: (context: Context<TConfig, TEnv, TSupportedEvents>) => Promise<Record<string, unknown> | undefined>,
   manifest: Manifest,
   options?: Options
 ) {
+  const pluginOptions = {
+    kernelPublicKey: options?.kernelPublicKey || KERNEL_PUBLIC_KEY,
+    logLevel: options?.logLevel || LOG_LEVEL.INFO,
+    postCommentOnError: options?.postCommentOnError || true,
+    settingsSchema: options?.settingsSchema,
+    envSchema: options?.envSchema,
+  };
+
   const app = new Hono();
 
   app.get("/manifest.json", (ctx) => {
@@ -32,34 +45,66 @@ export async function createPlugin<TConfig = unknown, TEnv = unknown, TSupported
     const payload = await ctx.req.json();
     const signature = payload.signature;
     delete payload.signature;
-    if (!(await verifySignature(options?.kernelPublicKey || KERNEL_PUBLIC_KEY, payload, signature))) {
+    if (!(await verifySignature(pluginOptions.kernelPublicKey, payload, signature))) {
       throw new HTTPException(400, { message: "Invalid signature" });
     }
 
-    try {
-      new customOctokit({ auth: payload.authToken });
-    } catch (error) {
-      console.error("SDK ERROR", error);
-      throw new HTTPException(500, { message: "Unexpected error" });
+    let config: TConfig;
+    if (pluginOptions.settingsSchema) {
+      config = Value.Decode(pluginOptions.settingsSchema, payload.settings);
+    } else {
+      config = payload.settings as TConfig;
+    }
+
+    let env: TEnv;
+    if (pluginOptions.envSchema) {
+      env = Value.Decode(pluginOptions.envSchema, process.env);
+    } else {
+      env = process.env as TEnv;
     }
 
     const context: Context<TConfig, TEnv, TSupportedEvents> = {
       eventName: payload.eventName,
       payload: payload.payload,
       octokit: new customOctokit({ auth: payload.authToken }),
-      config: payload.settings as TConfig,
-      env: ctx.env as TEnv,
-      logger: new Logs(options?.logLevel || LOG_LEVEL.INFO),
+      config: config,
+      env: env,
+      logger: new Logs(pluginOptions.logLevel),
     };
 
     try {
       const result = await handler(context);
       return ctx.json({ stateId: payload.stateId, output: result });
     } catch (error) {
       console.error(error);
+
+      let loggerError: LogReturn | null;
+      if (error instanceof Error) {
+        loggerError = context.logger.error(`Error: ${error}`, { error: error });
+      } else if (error instanceof LogReturn) {
+        loggerError = error;
+      } else {
+        loggerError = context.logger.error(`Error: ${error}`);
+      }
+
+      if (pluginOptions.postCommentOnError && loggerError) {
+        await postComment(context, loggerError);
+      }
+
       throw new HTTPException(500, { message: "Unexpected error" });
     }
   });
 
   return app;
 }
+
+async function postComment(context: Context, error: LogReturn) {
+  if ("issue" in context.payload && context.payload.repository?.owner?.login) {
+    await context.octokit.rest.issues.createComment({
+      owner: context.payload.repository.owner.login,
+      repo: context.payload.repository.name,
+      issue_number: context.payload.issue.number,
+      body: `${error.logMessage.diff}\n<!--\n${JSON.stringify(error.metadata, null, 2)}\n-->`,
+    });
+  }
+}
-Original file line number
+Diff line change
@@ Expand Up @@
 wIDAQAB
     -----END PUBLIC KEY-----
     `;
+    export const KERNEL_APP_ID = 0;