From 65e6fbd54bb83ea3f0c7456ab8c5080ac4fb759a Mon Sep 17 00:00:00 2001
From: Ali BARIN <ali.barin53@gmail.com>
Date: Sat, 23 Apr 2022 01:21:51 +0200
Subject: [PATCH] feat: add new watcher trigger in GitHub (#302)

* feat: add new watcher trigger in GitHub
---
 packages/backend/src/apps/github/info.json    | 39 +++++++++++++++
 packages/backend/src/apps/github/triggers.ts  |  3 ++
 .../apps/github/triggers/new-pull-request.ts  |  8 ++--
 .../src/apps/github/triggers/new-watcher.ts   | 48 +++++++++++++++++++
 4 files changed, 95 insertions(+), 3 deletions(-)
 create mode 100644 packages/backend/src/apps/github/triggers/new-watcher.ts

diff --git a/packages/backend/src/apps/github/info.json b/packages/backend/src/apps/github/info.json
index 73595f68..b0f8a9db 100644
--- a/packages/backend/src/apps/github/info.json
+++ b/packages/backend/src/apps/github/info.json
@@ -365,6 +365,45 @@
           "name": "Test trigger"
         }
       ]
+    },
+    {
+      "name": "New watcher",
+      "key": "newWatcher",
+      "interval": "15m",
+      "description": "Triggers when a new watcher is added to a repo",
+      "substeps": [
+        {
+          "key": "chooseAccount",
+          "name": "Choose account"
+        },
+        {
+          "key": "chooseTrigger",
+          "name": "Set up a trigger",
+          "arguments": [
+            {
+              "label": "Repo",
+              "key": "repo",
+              "type": "dropdown",
+              "required": true,
+              "variables": false,
+              "source": {
+                "type": "query",
+                "name": "getData",
+                "arguments": [
+                  {
+                    "name": "key",
+                    "value": "listRepos"
+                  }
+                ]
+              }
+            }
+          ]
+        },
+        {
+          "key": "testStep",
+          "name": "Test trigger"
+        }
+      ]
     }
   ]
 }
diff --git a/packages/backend/src/apps/github/triggers.ts b/packages/backend/src/apps/github/triggers.ts
index 4b00f576..2e680f72 100644
--- a/packages/backend/src/apps/github/triggers.ts
+++ b/packages/backend/src/apps/github/triggers.ts
@@ -4,6 +4,7 @@ import NewOrganization from './triggers/new-organization';
 import NewBranch from './triggers/new-branch';
 import NewNotification from './triggers/new-notification';
 import NewPullRequest from './triggers/new-pull-request';
+import NewWatcher from './triggers/new-watcher';
 
 export default class Triggers {
   newRepository: NewRepository;
@@ -11,6 +12,7 @@ export default class Triggers {
   newBranch: NewBranch;
   newNotification: NewNotification;
   newPullRequest: NewPullRequest;
+  newWatcher: NewWatcher;
 
   constructor(connectionData: IJSONObject, parameters: IJSONObject) {
     this.newRepository = new NewRepository(connectionData);
@@ -18,5 +20,6 @@ export default class Triggers {
     this.newBranch = new NewBranch(connectionData, parameters);
     this.newNotification = new NewNotification(connectionData, parameters);
     this.newPullRequest = new NewPullRequest(connectionData, parameters);
+    this.newWatcher = new NewWatcher(connectionData, parameters);
   }
 }
diff --git a/packages/backend/src/apps/github/triggers/new-pull-request.ts b/packages/backend/src/apps/github/triggers/new-pull-request.ts
index eb1a249a..4bf907c8 100644
--- a/packages/backend/src/apps/github/triggers/new-pull-request.ts
+++ b/packages/backend/src/apps/github/triggers/new-pull-request.ts
@@ -32,13 +32,15 @@ export default class NewPullRequest {
   }
 
   async run(startTime: Date) {
-    const iterator = await this.client.paginate.iterator(this.client.rest.pulls.list, this.options);
+    const iterator = await this.client.paginate.iterator(
+      this.client.rest.pulls.list,
+      this.options
+    );
     const newPullRequests = [];
 
     const startTimeDateObject = DateTime.fromJSDate(startTime);
 
-    pullRequestIterator:
-    for await (const { data: pullRequests } of iterator) {
+    pullRequestIterator: for await (const { data: pullRequests } of iterator) {
       for (const pullRequest of pullRequests) {
         const createdAtDateObject = DateTime.fromISO(pullRequest.created_at);
 
diff --git a/packages/backend/src/apps/github/triggers/new-watcher.ts b/packages/backend/src/apps/github/triggers/new-watcher.ts
new file mode 100644
index 00000000..bba1dd1d
--- /dev/null
+++ b/packages/backend/src/apps/github/triggers/new-watcher.ts
@@ -0,0 +1,48 @@
+import { Octokit } from 'octokit';
+import { IJSONObject } from '@automatisch/types';
+
+export default class NewWatcher {
+  client?: Octokit;
+  repoOwner?: string;
+  repo?: string;
+
+  constructor(connectionData: IJSONObject, parameters: IJSONObject) {
+    if (connectionData.accessToken) {
+      this.client = new Octokit({
+        auth: connectionData.accessToken as string,
+      });
+    }
+
+    if (parameters?.repo) {
+      const [owner, repo] = (parameters.repo as string).split('/');
+
+      this.repoOwner = owner;
+      this.repo = repo;
+    }
+  }
+
+  get options() {
+    return {
+      owner: this.repoOwner,
+      repo: this.repo,
+    };
+  }
+
+  async run() {
+    // TODO: implement pagination on undated entries
+    return await this.client.paginate(
+      this.client.rest.activity.listWatchersForRepo,
+      this.options
+    );
+  }
+
+  async testRun() {
+    return await this.run();
+    const options = {
+      ...this.options,
+      per_page: 1,
+    };
+
+    return (await this.client.rest.activity.listWatchersForRepo(options)).data;
+  }
+}