From c8f45f04715e7abaca6e1c47794c647cb9b526ac Mon Sep 17 00:00:00 2001
From: gentlementlegen <fernand.veyrier@epitech.eu>
Date: Fri, 20 Dec 2024 16:18:43 +0900
Subject: [PATCH] feat: runtime info populated for worker and node environments

---
 src/comment.ts              |  8 ++++---
 src/helpers/get-run-info.ts | 45 -----------------------------------
 src/helpers/runtime-info.ts | 47 +++++++++++++++++++++++++++++++++++++
 src/server.ts               | 11 +++++++++
 4 files changed, 63 insertions(+), 48 deletions(-)
 delete mode 100644 src/helpers/get-run-info.ts
 create mode 100644 src/helpers/runtime-info.ts

diff --git a/src/comment.ts b/src/comment.ts
index 11260ba..77fdc1c 100644
--- a/src/comment.ts
+++ b/src/comment.ts
@@ -1,6 +1,6 @@
-import { Context } from "./context";
 import { LogReturn, Metadata } from "@ubiquity-os/ubiquity-os-logger";
-import { getPluginName, getRunUrl, getVersion } from "./helpers/get-run-info";
+import { Context } from "./context";
+import { PluginRuntimeInfo } from "./helpers/runtime-info";
 import { sanitizeMetadata } from "./util";
 
 const HEADER_NAME = "UbiquityOS";
@@ -67,8 +67,10 @@ async function createStructuredMetadataWithMessage(context: Context, message: Lo
   } else {
     instigatorName = context.payload.sender?.login || HEADER_NAME;
   }
+  const runUrl = PluginRuntimeInfo.getInstance().runUrl;
+  const version = await PluginRuntimeInfo.getInstance().version;
 
-  const ubiquityMetadataHeader = `<!-- ${HEADER_NAME} - ${await getPluginName()} - @${instigatorName} - ${getRunUrl()} - ${callingFnName} - ${await getVersion(context)}`;
+  const ubiquityMetadataHeader = `<!-- ${HEADER_NAME} - @${instigatorName} - ${runUrl} - ${callingFnName} - ${version}`;
 
   let metadataSerialized: string;
   const metadataSerializedVisible = ["```json", jsonPretty, "```"].join("\n");
diff --git a/src/helpers/get-run-info.ts b/src/helpers/get-run-info.ts
deleted file mode 100644
index e109b2e..0000000
--- a/src/helpers/get-run-info.ts
+++ /dev/null
@@ -1,45 +0,0 @@
-import github from "@actions/github";
-import { getRuntimeKey } from "hono/adapter";
-import simpleGit from "simple-git";
-import { Context } from "../context";
-
-export async function getVersion(context: Context) {
-  const env = context.env as Record<string, string>;
-  switch (getRuntimeKey()) {
-    case "workerd":
-      return env.CF_VERSION_METADATA ?? "[missing CF_VERSION_METADATA]";
-    case "node":
-    case "bun":
-      return simpleGit().revparse(["--short", "HEAD"]);
-    default:
-      return "unknown_version";
-  }
-}
-
-export function getRunUrl() {
-  switch (getRuntimeKey()) {
-    case "workerd":
-      return "worker url with time";
-    case "node":
-      return `${github.context.payload.repository?.html_url}/actions/runs/${github.context.runId}`;
-    case "bun":
-      return "http://localhost";
-    default:
-      return "unknown_run_url";
-  }
-}
-
-export async function getPluginName() {
-  switch (getRuntimeKey()) {
-    case "workerd":
-      return "worker url with time";
-    case "node":
-    case "bun": {
-      const currentHash = await simpleGit().revparse(["--short", "HEAD"]);
-      const manifest = JSON.parse(await simpleGit().show([`${currentHash}:manifest.json`]));
-      return `"${manifest.name}"`;
-    }
-    default:
-      return "unknown_run_url";
-  }
-}
diff --git a/src/helpers/runtime-info.ts b/src/helpers/runtime-info.ts
new file mode 100644
index 0000000..0310808
--- /dev/null
+++ b/src/helpers/runtime-info.ts
@@ -0,0 +1,47 @@
+import github from "@actions/github";
+import { getRuntimeKey } from "hono/adapter";
+import simpleGit from "simple-git";
+
+export abstract class PluginRuntimeInfo {
+  private static _instance: PluginRuntimeInfo | null = null;
+  protected _env: Record<string, string> = {};
+
+  protected constructor(env?: Record<string, string>) {
+    if (env) {
+      this._env = env;
+    }
+  }
+
+  public static getInstance(env?: Record<string, string>) {
+    if (!PluginRuntimeInfo._instance) {
+      PluginRuntimeInfo._instance = getRuntimeKey() === "workerd" ? new CfRuntimeInfo(env) : new NodeRuntimeInfo(env);
+    }
+    return PluginRuntimeInfo._instance;
+  }
+
+  public abstract get version(): Promise<string>;
+  public abstract get runUrl(): string;
+}
+
+export class CfRuntimeInfo extends PluginRuntimeInfo {
+  public get version(): Promise<string> {
+    return Promise.resolve(this._env.CF_VERSION_METADATA ?? "[missing CF_VERSION_METADATA]");
+  }
+  public get runUrl(): string {
+    const accountId = this._env.CF_ACCOUNT_ID;
+    const workerName = this._env.CF_WORKER_NAME;
+    const toTime = Date.now() + 60000;
+    const fromTime = Date.now() - 60000;
+    const timeParam = encodeURIComponent(`{"type":"absolute","to":${toTime},"from":${fromTime}}`);
+    return `https://dash.cloudflare.com/${accountId}/workers/services/view/${workerName}/production/observability/logs?granularity=0&time=${timeParam}`;
+  }
+}
+
+export class NodeRuntimeInfo extends PluginRuntimeInfo {
+  public get version() {
+    return simpleGit().revparse(["--short", "HEAD"]);
+  }
+  public get runUrl() {
+    return github.context.payload.repository ? `${github.context.payload.repository?.html_url}/actions/runs/${github.context.runId}` : "http://localhost";
+  }
+}
diff --git a/src/server.ts b/src/server.ts
index ab867c0..770f616 100644
--- a/src/server.ts
+++ b/src/server.ts
@@ -7,6 +7,7 @@ import { env as honoEnv } from "hono/adapter";
 import { HTTPException } from "hono/http-exception";
 import { postComment } from "./comment";
 import { Context } from "./context";
+import { PluginRuntimeInfo } from "./helpers/runtime-info";
 import { customOctokit } from "./octokit";
 import { verifySignature } from "./signature";
 import { Manifest } from "./types/manifest";
@@ -79,6 +80,16 @@ export function createPlugin<TConfig = unknown, TEnv = unknown, TCommand = unkno
       env = ctx.env as TEnv;
     }
 
+    const workerUrl = new URL(inputs.ref).origin;
+    let workerName;
+
+    if (workerUrl.includes("localhost")) {
+      workerName = "localhost";
+    } else {
+      workerName = `${workerUrl.split("//")[1].split(".")[0]}`;
+    }
+    PluginRuntimeInfo.getInstance({ ...env, CF_WORKER_NAME: workerName });
+
     let command: TCommand | null = null;
     if (inputs.command && pluginOptions.commandSchema) {
       try {