Enhance GitHub PR functionality by adding labels and assignees support, and update package.json keywords for better searchability

Author: zaknafeyn

package.json

@@ -13,7 +13,11 @@
     "git",
     "checkout",
     "stash",
-    "auto stash"
+    "auto stash",
+    "pull request",
+    "PR",
+    "pr clone",
+    "clone pull request"
   ],
   "license": "MIT",
   "version": "0.0.1",

src/common/api/ghClient.ts

@@ -2,7 +2,7 @@ import * as https from 'https';
 import { authentication } from 'vscode';
 
 import { EXTENSION_NAME } from '../../const';
-import { GitHubCommit, GitHubCommitFile, GitHubPR } from '../../types/dataTypes';
+import { GitHubCommit, GitHubCommitFile, GitHubLabel, GitHubPR, GitHubUser } from '../../types/dataTypes';
 
 export class GitHubClient {
   private static readonly BASE_URL = 'https://api.github.com';
@@ -72,9 +72,21 @@ export class GitHubClient {
           data += chunk;
         });
 
-        res.on('end', () => {
+        res.on('end', async () => {
           if (res.statusCode && res.statusCode === 403) {
-            // todo: add reauthenticate method and retry
+            // Check if this is a SAML enforcement error and we haven't already retried with re-auth
+            if (!reAuthenticate && data.includes('Resource protected by organization SAML enforcement')) {
+              try {
+                // Retry with re-authentication
+                const result = await this.makeRequest<T>(endpoint, method, body, true);
+                resolve(result);
+                return;
+              } catch (retryError) {
+                // If re-authentication fails, throw the retry error instead of the original
+                reject(retryError);
+                return;
+              }
+            }
           }
 
           if (res.statusCode && res.statusCode >= 200 && res.statusCode < 300) {
@@ -195,17 +207,35 @@ export class GitHubClient {
     body: string,
     head: string,
     base: string,
-    isDraft: boolean = false
+    isDraft: boolean = false,
+    labels?: string[],
+    assignees?: string[]
   ): Promise<GitHubPR> {
     const endpoint = `/repos/${this.owner}/${this.repo}/pulls`;
-    const requestBody = {
+    const requestBody: any = {
       title,
       body,
       head,
       base,
       draft: isDraft,
     };
 
+    // Add labels and assignees if provided
+    if (labels && labels.length > 0) {
+      requestBody.labels = labels;
+    }
+    if (assignees && assignees.length > 0) {
+      requestBody.assignees = assignees;
+    }
+
     return this.makeRequest<GitHubPR>(endpoint, 'POST', requestBody);
   }
+
+  /**
+   * Fetch all labels from the repository
+   */
+  public async fetchLabels(): Promise<GitHubLabel[]> {
+    const endpoint = `/repos/${this.owner}/${this.repo}/labels`;
+    return this.makeRequest<GitHubLabel[]>(endpoint);
+  }
 }

src/services/prCloneInPlaceService.ts

@@ -273,13 +273,19 @@ export class PrCloneInPlaceService extends PrCloneServiceBase {
   ): Promise<GitHubPR> {
     const prBody = description;
 
+    // Extract labels and assignees from original PR
+    const labels = originalPr.labels?.map(label => label.name) || [];
+    const assignees = originalPr.assignees?.map(assignee => assignee.login) || [];
+
     // Create PR using the GitHub API
     const newPr = await this.ghClient.createPullRequest(
       originalPr.title, // Use original PR title
       prBody,
       featureBranch,
       targetBranch,
-      isDraft
+      isDraft,
+      labels,
+      assignees
     );
 
     this.loggingService.info(`Created PR #${newPr.number}: ${newPr.title}`);

src/types/dataTypes.ts

@@ -1,3 +1,9 @@
+export interface GitHubUser {
+  id: number;
+  login: string;
+  avatar_url: string;
+}
+
 export interface GitHubPR {
   number: number;
   title: string;
@@ -10,6 +16,8 @@ export interface GitHubPR {
     ref: string;
   };
   html_url: string;
+  labels: GitHubLabel[];
+  assignees: GitHubUser[];
 }
 
 export interface GitHubCommitFile {
@@ -29,6 +37,14 @@ export interface GitHubCommit {
   files?: GitHubCommitFile[];
 }
 
+export interface GitHubLabel {
+  id: number;
+  name: string;
+  description: string | null;
+  color: string;
+  default: boolean;
+}
+
 export interface AppState {
   view: 'input' | 'clone';
   prData?: GitHubPR;