feat(engine): handle assignee and reviewer events

.vscode/settings.json

@@ -55,5 +55,7 @@
   "github-actions.workflows.pinned.workflows": [
     ".github/workflows/server-unit-tests.yml",
     ".github/workflows/lint-server.yml"
-  ]
+  ],
+  "github-actions.workflows.pinned.refresh.enabled": true,
+  "github-actions.workflows.pinned.refresh.interval": 30
 }

server/src/api/pull-request/pull-request.service.ts

@@ -4,7 +4,7 @@ import { ReturnModelType } from '@typegoose/typegoose';
 
 import { BaseService } from '@kb-abstracts';
 import { IGithubChanges } from '@kb-interfaces';
-import { PullRequest } from '@kb-models';
+import { PullRequest, User } from '@kb-models';
 
 export interface INewData {
   title?: string;
@@ -57,4 +57,23 @@ export class PullRequestService extends BaseService<PullRequest> {
       ...historyUpdate
     });
   }
+
+  async updateAssignees(prid: string, assignees: User[]) {
+    await this.prModel.findOneAndUpdate({ prid }, {
+      assignees: assignees.map((user) => user.username)
+    });
+  }
+
+  async updateReviewers(prid: string, reviewer: User, forRemoval?: boolean) {
+    const changeQuery = {
+      $pull: {} as any,
+      $addToSet: {} as any
+    };
+    const addDelAttr = forRemoval ? '$pull' : '$addToSet';
+    changeQuery[addDelAttr].reviewers = reviewer.username;
+    if (forRemoval) {
+      changeQuery['$addToSet']['history.deletedReviewers'] = reviewer.username;
+    }
+    await this.prModel.findOneAndUpdate({ prid }, changeQuery);
+  }
 }

server/src/dev-tools/captured-events/pull-request-review-request-added.event.ts

@@ -3,7 +3,7 @@ export const pullReuqestReviewRequestAddedEvent = {
   event: 'pull_request',
   payload: {
     action: 'review_requested',
-    number: 2,
+    number: 1,
     pull_request: {
       url: 'https://api.github.com/repos/Thatkookooguy/test-new-achievibit-events/pulls/2',
       id: 353189935,
@@ -12,7 +12,7 @@ export const pullReuqestReviewRequestAddedEvent = {
       diff_url: 'https://github.com/Thatkookooguy/test-new-achievibit-events/pull/2.diff',
       patch_url: 'https://github.com/Thatkookooguy/test-new-achievibit-events/pull/2.patch',
       issue_url: 'https://api.github.com/repos/Thatkookooguy/test-new-achievibit-events/issues/2',
-      number: 2,
+      number: 1,
       state: 'open',
       locked: false,
       title: 'Update README.md',

server/src/dev-tools/captured-events/pull-request-review-request-removed.event.ts

@@ -3,7 +3,7 @@ export const pullRequestReviewRequestRemovedEvent = {
   event: 'pull_request',
   payload: {
     action: 'review_request_removed',
-    number: 2,
+    number: 1,
     pull_request: {
       url: 'https://api.github.com/repos/Thatkookooguy/test-new-achievibit-events/pulls/2',
       id: 353189935,
@@ -12,7 +12,7 @@ export const pullRequestReviewRequestRemovedEvent = {
       diff_url: 'https://github.com/Thatkookooguy/test-new-achievibit-events/pull/2.diff',
       patch_url: 'https://github.com/Thatkookooguy/test-new-achievibit-events/pull/2.patch',
       issue_url: 'https://api.github.com/repos/Thatkookooguy/test-new-achievibit-events/issues/2',
-      number: 2,
+      number: 1,
       state: 'open',
       locked: false,
       title: 'Update README.md',

server/src/engines/github.engine.ts

@@ -135,25 +135,86 @@ export class GithubEngine extends Engine<IGithubPullRequestEvent> {
       eventData.changes
     );
   }
-  handlePullRequestAssigneeAdded(
+  async handlePullRequestAssigneeAdded(
     eventData: IGithubPullRequestEvent
   ): Promise<void> {
-    throw new Error('Method not implemented.');
+    const {
+      githubCreator,
+      githubOwner,
+      githubPR
+    } = this.extractGithubEntities(eventData);
+    const pr = this.extractPullRequest(
+      githubPR,
+      this.extractUser(githubCreator),
+      this.extractRepo(eventData.repository),
+      this.extractUser(githubOwner)
+    );
+
+    const githubAssignees = eventData.pull_request.assignees;
+    const assignees = githubAssignees
+      .map((assignee) => this.extractUser(assignee));
+
+    // TODO: need to save these users!!!
+
+    await this.pullRequestsService.updateAssignees(pr.prid, assignees);
   }
-  handlePullRequestAssigneeRemoved(
+  async handlePullRequestAssigneeRemoved(
     eventData: IGithubPullRequestEvent
   ): Promise<void> {
-    throw new Error('Method not implemented.');
+    const {
+      githubCreator,
+      githubOwner,
+      githubPR
+    } = this.extractGithubEntities(eventData);
+    const pr = this.extractPullRequest(
+      githubPR,
+      this.extractUser(githubCreator),
+      this.extractRepo(eventData.repository),
+      this.extractUser(githubOwner)
+    );
+
+    const githubAssignees = eventData.pull_request.assignees;
+    const assignees = githubAssignees
+      .map((assignee) => this.extractUser(assignee));
+
+    await this.pullRequestsService.updateAssignees(pr.prid, assignees);
   }
-  handlePullRequestReviewRequestAdded(
+  async handlePullRequestReviewRequestAdded(
     eventData: IGithubPullRequestEvent
   ): Promise<void> {
-    throw new Error('Method not implemented.');
+    const {
+      githubCreator,
+      githubOwner,
+      githubPR
+    } = this.extractGithubEntities(eventData);
+    const pr = this.extractPullRequest(
+      githubPR,
+      this.extractUser(githubCreator),
+      this.extractRepo(eventData.repository),
+      this.extractUser(githubOwner)
+    );
+    const reviewer = this.extractUser(eventData.requested_reviewer);
+
+    // TODO: need to save these users!!!
+
+    await this.pullRequestsService.updateReviewers(pr.prid, reviewer);
   }
-  handlePullRequestReviewRequestRemoved(
+  async handlePullRequestReviewRequestRemoved(
     eventData: IGithubPullRequestEvent
   ): Promise<void> {
-    throw new Error('Method not implemented.');
+    const {
+      githubCreator,
+      githubOwner,
+      githubPR
+    } = this.extractGithubEntities(eventData);
+    const pr = this.extractPullRequest(
+      githubPR,
+      this.extractUser(githubCreator),
+      this.extractRepo(eventData.repository),
+      this.extractUser(githubOwner)
+    );
+    const reviewer = this.extractUser(eventData.requested_reviewer);
+    await this.pullRequestsService.updateReviewers(pr.prid, reviewer, true);
   }
   handlePullRequestReviewCommentAdded(
     eventData: IGithubPullRequestEvent

server/src/interfaces/github-pr-payload.model.ts

@@ -111,6 +111,7 @@ export interface IGithubPullRequest {
   merged_at: string;
   merge_commit_sha: string;
   assignee: string;
+  assignees: IGithubUser[];
   milestone: string;
   commits_url: string;
   review_comments_url: string;
@@ -189,4 +190,6 @@ export interface IGithubPullRequestEvent {
   sender: IGithubUser;
   label?: { name: string };
   changes?: IGithubChanges;
+  requested_reviewer?: IGithubUser;
+  // assignees?: IGithubUser[];
 }

server/src/models/pull-request.model.ts

@@ -73,6 +73,11 @@ export class PullRequest extends BaseModel {
   @PersistInDb()
   assignees?: string[];
 
+  @Expose()
+  @IsOptional()
+  @PersistInDb()
+  reviewers?: string[];
+
   constructor(partial: Partial<PullRequest> = {}) {
     super();
     Object.assign(this, partial);