feat: add PR link as a resource when PR is merged

package.json

@@ -31,6 +31,7 @@
     "@actions/github": "^5.1.1",
     "@vercel/ncc": "^0.34.0",
     "dotenv": "^16.0.3",
-    "node-fetch": "^3.3.0"
+    "node-fetch": "^3.3.0",
+    "uuid": "^9.0.0"
   }
 }

src/api/create-resource.js

@@ -0,0 +1,50 @@
+import { v4 as uuidv4 } from "uuid";
+import fetch from "node-fetch";
+import core from "@actions/core";
+import dotenv from "dotenv";
+
+dotenv.config();
+
+const ATLAN_INSTANCE_URL =
+  core.getInput("ATLAN_INSTANCE_URL") || process.env.ATLAN_INSTANCE_URL;
+const ATLAN_API_TOKEN =
+  core.getInput("ATLAN_API_TOKEN") || process.env.ATLAN_API_TOKEN;
+
+export default async function createResource(guid, name, link) {
+  var myHeaders = {
+    Authorization: `Bearer ${ATLAN_API_TOKEN}`,
+    "Content-Type": "application/json",
+  };
+
+  var raw = JSON.stringify({
+    entities: [
+      {
+        typeName: "Link",
+        attributes: {
+          qualifiedName: uuidv4(),
+          name,
+          link,
+          tenantId: "default",
+        },
+        relationshipAttributes: {
+          asset: {
+            guid,
+          },
+        },
+      },
+    ],
+  });
+
+  var requestOptions = {
+    method: "POST",
+    headers: myHeaders,
+    body: raw,
+  };
+
+  var response = await fetch(
+    `${ATLAN_INSTANCE_URL}/api/meta/entity/bulk`,
+    requestOptions
+  ).then((e) => e.json());
+
+  return response;
+}

src/api/index.js

@@ -1,2 +1,3 @@
 export { default as getDownstreamAssets } from "./get-downstream-assets.js";
 export { default as getAsset } from "./get-asset.js";
+export { default as createResource } from "./create-resource.js";

src/index.js

@@ -2,7 +2,7 @@ import dotenv from "dotenv";
 import core from "@actions/core";
 import github from "@actions/github";
 
-import { printDownstreamAssets } from "./main/index.js";
+import { printDownstreamAssets, setResourceOnAsset } from "./main/index.js";
 
 dotenv.config();
 
@@ -12,15 +12,15 @@ async function run() {
   const { context } = github;
   const octokit = github.getOctokit(GITHUB_TOKEN);
   const { pull_request } = context.payload;
-  const { state } = pull_request;
-  console.log(pull_request);
+  const { state, merged } = pull_request;
 
   switch (state) {
     case "open":
       await printDownstreamAssets({ octokit, context });
       break;
     case "closed":
-      console.log(pull_request);
+      if (merged) await setResourceOnAsset({ octokit, context });
+      break;
   }
 }
 

src/main/index.js

@@ -1 +1,2 @@
 export { default as printDownstreamAssets } from "./print-downstream-assets.js";
+export { default as setResourceOnAsset } from "./set-resource-on-asset.js";

src/main/print-downstream-assets.js

@@ -8,6 +8,8 @@ import {
 export default async function printDownstreamAssets({ octokit, context }) {
   const changedFiles = await getChangedFiles(octokit, context);
 
+  if (changedFiles.length === 0) return;
+
   changedFiles.forEach(async ({ name, filePath }) => {
     const assetName = await getAssetName(octokit, context, name, filePath);
     const asset = await getAsset({ name: assetName });

src/main/set-resource-on-asset.js

@@ -0,0 +1,44 @@
+import { getAsset, createResource } from "../api/index.js";
+import {
+  createCustomComment,
+  getChangedFiles,
+  getAssetName,
+} from "../utils/index.js";
+
+export default async function setResourceOnAsset({ octokit, context }) {
+  const changedFiles = await getChangedFiles(octokit, context);
+  const { pull_request } = context.payload;
+
+  if (changedFiles.length === 0) return;
+
+  changedFiles.forEach(async ({ name, filePath }) => {
+    const assetName = await getAssetName(octokit, context, name, filePath);
+    const asset = await getAsset({ name: assetName });
+
+    if (!asset) return;
+
+    const { guid: modelGuid } = asset;
+    const { guid: tableAssetGuid } = asset.attributes.sqlAsset;
+
+    await createResource(
+      modelGuid,
+      "Pull Request on GitHub",
+      pull_request.html_url
+    );
+    await createResource(
+      tableAssetGuid,
+      "Pull Request on GitHub",
+      pull_request.html_url
+    );
+  });
+
+  const comment = await createCustomComment(
+    octokit,
+    context,
+    `🎊 Congrats on the merge!
+  
+This pull request has been added as a resource to all the assets modified. ✅
+`
+  );
+  console.log(comment);
+}

src/utils/create-comment.js

@@ -73,3 +73,15 @@ export default async function createComment(
   if (IS_DEV) return comment;
   return octokit.rest.issues.createComment(commentObj);
 }
+
+export async function createCustomComment(octokit, context, content) {
+  const { pull_request } = context.payload;
+  const commentObj = {
+    ...context.repo,
+    issue_number: pull_request.number,
+    body: content,
+  };
+
+  if (IS_DEV) return content;
+  return octokit.rest.issues.createComment(commentObj);
+}

src/utils/index.js

@@ -4,7 +4,10 @@ export {
   getCertificationImage,
 } from "./get-image-url.js";
 export { default as hostedImages } from "./hosted-images.js";
-export { default as createComment } from "./create-comment.js";
+export {
+  default as createComment,
+  createCustomComment,
+} from "./create-comment.js";
 export {
   getFileContents,
   getChangedFiles,