/*---------------------------------------------------------------------------------------------

* Copyright (c) Microsoft Corporation. All rights reserved.

* Licensed under the MIT License. See License.txt in the project root for license information.

*--------------------------------------------------------------------------------------------*/

'use strict';

import * as crypto from 'crypto';

import * as OctokitTypes from '@octokit/types';

import * as vscode from 'vscode';

import { OctokitCommon } from './common';

import { FolderRepositoryManager, PullRequestDefaults } from './folderRepositoryManager';

import { GitHubRepository, ViewerPermission } from './githubRepository';

import * as GraphQL from './graphql';

import {

AccountType,

GithubItemStateEnum,

IAccount,

IActor,

IGitHubRef,

IIssueComment,

ILabel,

IMilestone,

IProjectItem,

Issue,

ISuggestedReviewer,

ITeam,

MergeMethod,

MergeQueueEntry,

MergeQueueState,

Notification,

NotificationSubjectType,

PullRequest,

PullRequestMergeability,

Reaction,

reviewerId,

reviewerLabel,

ReviewState,

toAccountType,

User,

} from './interface';

import { IssueModel } from './issueModel';

import { GHPRComment, GHPRCommentThread } from './prComment';

import { PullRequestModel } from './pullRequestModel';

import { RemoteInfo } from '../../common/types';

import { Repository } from '../api/api';

import { GitApiImpl } from '../api/api1';

import { AuthProvider, GitHubServerType } from '../common/authentication';

import { COPILOT_ACCOUNTS, IComment, IReviewThread, SubjectType } from '../common/comment';

import { COPILOT_REVIEWER, COPILOT_SWE_AGENT } from '../common/copilot';

import { DiffHunk, parseDiffHunk } from '../common/diffHunk';

import { emojify } from '../common/emoji';

import { GitHubRef } from '../common/githubRef';

import Logger from '../common/logger';

import { Remote } from '../common/remote';

import { GITHUB_ENTERPRISE, OVERRIDE_DEFAULT_BRANCH, PR_SETTINGS_NAMESPACE, URI } from '../common/settingKeys';

import * as Common from '../common/timelineEvent';

import { DataUri, toOpenIssueWebviewUri, toOpenPullRequestWebviewUri } from '../common/uri';

import { escapeRegExp, gitHubLabelColor, stringReplaceAsync, uniqBy } from '../common/utils';

export const ISSUE_EXPRESSION = /(([A-Za-z0-9_.\-]+)\/([A-Za-z0-9_.\-]+))?(#|GH-)([1-9][0-9]*)($|\b)/;

export const ISSUE_OR_URL_EXPRESSION = /(https?:\/\/github\.com\/(([^\s]+)\/([^\s]+))\/([^\s]+\/)?(issues|pull)\/([0-9]+)(#issuecomment\-([0-9]+))?)|(([A-Za-z0-9_.\-]+)\/([A-Za-z0-9_.\-]+))?(#|GH-)([1-9][0-9]*)($|\b)/;

export interface CommentReactionHandler {

toggleReaction(comment: vscode.Comment, reaction: vscode.CommentReaction): Promise<void>;

}

export type ParsedIssue = {

owner: string | undefined;

name: string | undefined;

issueNumber: number;

commentNumber?: number;

};

export function parseIssueExpressionOutput(output: RegExpMatchArray | null): ParsedIssue | undefined {

if (!output) {

return undefined;

}

const issue: ParsedIssue = { owner: undefined, name: undefined, issueNumber: 0 };

if (output.length === 7) {

issue.owner = output[2];

issue.name = output[3];

issue.issueNumber = parseInt(output[5]);

return issue;

} else if (output.length === 16) {

issue.owner = output[3] || output[11];

issue.name = output[4] || output[12];

issue.issueNumber = parseInt(output[7] || output[14]);

issue.commentNumber = output[9] !== undefined ? parseInt(output[9]) : undefined;

return issue;

} else {

return undefined;

}

}

export function threadRange(startLine: number, endLine: number, endCharacter?: number): vscode.Range {

if ((startLine !== endLine) && (endCharacter === undefined)) {

endCharacter = 300; // 300 is a "large" number that will select a lot of the line since don't know anything about the line length

} else if (!endCharacter) {

endCharacter = 0;

}

return new vscode.Range(startLine, 0, endLine, endCharacter);

}

export async function setReplyAuthor(thread: vscode.CommentThread | vscode.CommentThread2, currentUser: IAccount, context: vscode.ExtensionContext) {

if (currentUser.avatarUrl) {

const thread2 = thread as vscode.CommentThread2;

thread2.canReply = { name: currentUser.name ?? currentUser.login, iconPath: vscode.Uri.parse(currentUser.avatarUrl) };

const uri = await DataUri.avatarCirclesAsImageDataUris(context, [currentUser], 28, 28);

thread2.canReply = { name: currentUser.name ?? currentUser.login, iconPath: uri[0] };

} else {

thread.canReply = true;

}

}

export function createVSCodeCommentThreadForReviewThread(

context: vscode.ExtensionContext,

uri: vscode.Uri,

range: vscode.Range | undefined,

thread: IReviewThread,

commentController: vscode.CommentController,

currentUser: IAccount,

githubRepositories?: GitHubRepository[]

): GHPRCommentThread {

const vscodeThread = commentController.createCommentThread(uri, range, []);

(vscodeThread as GHPRCommentThread).gitHubThreadId = thread.id;

vscodeThread.comments = thread.comments.map(comment => new GHPRComment(context, comment, vscodeThread as GHPRCommentThread, githubRepositories));

const resolved = isResolvedToResolvedState(thread.isResolved);

let applicability = vscode.CommentThreadApplicability.Current;

if (thread.viewerCanResolve && !thread.isResolved) {

vscodeThread.contextValue = 'canResolve';

} else if (thread.viewerCanUnresolve && thread.isResolved) {

vscodeThread.contextValue = 'canUnresolve';

}

if (thread.isOutdated) {

vscodeThread.contextValue += 'outdated';

applicability = vscode.CommentThreadApplicability.Outdated;

}

vscodeThread.state = { resolved, applicability };

updateCommentThreadLabel(vscodeThread as GHPRCommentThread);

vscodeThread.collapsibleState = getCommentCollapsibleState(thread, undefined, currentUser.login);

setReplyAuthor(vscodeThread, currentUser, context);

return vscodeThread as GHPRCommentThread;

}

function isResolvedToResolvedState(isResolved: boolean) {

return isResolved ? vscode.CommentThreadState.Resolved : vscode.CommentThreadState.Unresolved;

}

export const COMMENT_EXPAND_STATE_SETTING = 'commentExpandState';

export const COMMENT_EXPAND_STATE_COLLAPSE_VALUE = 'collapseAll';

export const COMMENT_EXPAND_STATE_COLLAPSE_PREEXISTING_VALUE = 'collapsePreexisting';

export const COMMENT_EXPAND_STATE_EXPAND_VALUE = 'expandUnresolved';

export function getCommentCollapsibleState(thread: IReviewThread, expand?: boolean, currentUser?: string, isNewlyAdded?: boolean) {

const config = vscode.workspace.getConfiguration(PR_SETTINGS_NAMESPACE)?.get(COMMENT_EXPAND_STATE_SETTING);

const isFromCurrent = (currentUser && (thread.comments[thread.comments.length - 1].user?.login === currentUser));

const isJustSuggestion = thread.comments.length === 1 && thread.comments[0].body.startsWith('```suggestion') && thread.comments[0].body.endsWith('```');

// When collapsePreexisting is set, keep newly added comments expanded

if (config === COMMENT_EXPAND_STATE_COLLAPSE_PREEXISTING_VALUE && isNewlyAdded && !isJustSuggestion) {

return vscode.CommentThreadCollapsibleState.Expanded;

}

if (thread.isResolved || (!thread.isOutdated && isFromCurrent && !isJustSuggestion)) {

return vscode.CommentThreadCollapsibleState.Collapsed;

}

if (expand === undefined) {

expand = config === COMMENT_EXPAND_STATE_EXPAND_VALUE;

}

return expand

? vscode.CommentThreadCollapsibleState.Expanded : vscode.CommentThreadCollapsibleState.Collapsed;

}

export function updateThreadWithRange(context: vscode.ExtensionContext, vscodeThread: GHPRCommentThread, reviewThread: IReviewThread, githubRepositories?: GitHubRepository[], expand?: boolean, isNewlyAdded?: boolean) {

if (!vscodeThread.range) {

return;

}

const editors = vscode.window.visibleTextEditors;

for (let editor of editors) {

if (editor.document.uri.toString() === vscodeThread.uri.toString()) {

const endLine = editor.document.lineAt(vscodeThread.range.end.line);

const range = new vscode.Range(vscodeThread.range.start.line, 0, vscodeThread.range.end.line, endLine.text.length);

updateThread(context, vscodeThread, reviewThread, githubRepositories, expand, range, undefined, isNewlyAdded);

break;

}

}

}

export function updateThread(context: vscode.ExtensionContext, vscodeThread: GHPRCommentThread, reviewThread: IReviewThread, githubRepositories?: GitHubRepository[], expand?: boolean, range?: vscode.Range, currentUser?: string, isNewlyAdded?: boolean) {

if (reviewThread.viewerCanResolve && !reviewThread.isResolved) {

vscodeThread.contextValue = 'canResolve';

} else if (reviewThread.viewerCanUnresolve && reviewThread.isResolved) {

vscodeThread.contextValue = 'canUnresolve';

}

if (reviewThread.isOutdated) {

vscodeThread.contextValue += 'outdated';

}

const newResolvedState = isResolvedToResolvedState(reviewThread.isResolved);

const newApplicabilityState = reviewThread.isOutdated ? vscode.CommentThreadApplicability.Outdated : vscode.CommentThreadApplicability.Current;

if ((vscodeThread.state?.resolved !== newResolvedState) || (vscodeThread.state?.applicability !== newApplicabilityState)) {

vscodeThread.state = {

resolved: newResolvedState,

applicability: newApplicabilityState

};

}

vscodeThread.collapsibleState = getCommentCollapsibleState(reviewThread, expand, currentUser, isNewlyAdded);

if (range) {

vscodeThread.range = range;

}

if ((vscodeThread.comments.length === reviewThread.comments.length) && vscodeThread.comments.every((vscodeComment, index) => vscodeComment.commentId === `${reviewThread.comments[index].id}`)) {

// The comments all still exist. Update them instead of creating new ones. This allows the UI to be more stable.

let index = 0;

for (const comment of vscodeThread.comments) {

if (comment instanceof GHPRComment) {

comment.update(reviewThread.comments[index]);

}

index++;

}

} else {

vscodeThread.comments = reviewThread.comments.map(c => new GHPRComment(context, c, vscodeThread, githubRepositories));

}

updateCommentThreadLabel(vscodeThread);

}

export function updateCommentThreadLabel(thread: GHPRCommentThread) {

if (thread.state?.resolved === vscode.CommentThreadState.Resolved) {

thread.label = vscode.l10n.t('Marked as resolved');

return;

}

if (thread.comments.length) {

const participantsList = uniqBy(thread.comments, comment => comment.originalAuthor.name)

.map(comment => `@${comment.originalAuthor.name}`)

.join(', ');

thread.label = vscode.l10n.t('Participants: {0}', participantsList);

} else {

thread.label = vscode.l10n.t('Start discussion');

}

}

export function updateCommentReactions(comment: vscode.Comment, reactions: Reaction[] | undefined) {

let reactionsHaveUpdates = false;

const previousReactions = comment.reactions;

const newReactions = getReactionGroup().map((reaction, index) => {

if (!reactions) {

return { label: reaction.label, authorHasReacted: false, count: 0, iconPath: reaction.icon || '' };

}

const matchedReaction = reactions.find(re => re.label === reaction.label);

let newReaction: vscode.CommentReaction;

if (matchedReaction) {

newReaction = {

label: matchedReaction.label,

authorHasReacted: matchedReaction.viewerHasReacted,

count: matchedReaction.count,

iconPath: reaction.icon || '',

reactors: matchedReaction.reactors.map(reactor => ({ name: reactor }))

};

} else {

newReaction = { label: reaction.label, authorHasReacted: false, count: 0, iconPath: reaction.icon || '' };

}

if (!reactionsHaveUpdates && (!previousReactions || (previousReactions[index].authorHasReacted !== newReaction.authorHasReacted) || (previousReactions[index].count !== newReaction.count))) {

reactionsHaveUpdates = true;

}

return newReaction;

});

comment.reactions = newReactions;

return reactionsHaveUpdates;

}

export function updateCommentReviewState(thread: GHPRCommentThread, newDraftMode: boolean) {

if (newDraftMode) {

return;

}

thread.comments = thread.comments.map(comment => {

if (comment instanceof GHPRComment) {

comment.rawComment.isDraft = false;

}

comment.label = undefined;

return comment;

});

}

export function isEnterprise(provider: AuthProvider): boolean {

return provider === AuthProvider.githubEnterprise;

}

export function convertRESTUserToAccount(

user: OctokitCommon.PullsListResponseItemUser,

githubRepository?: GitHubRepository,

): IAccount {

return parseAccount(user, githubRepository);

}

export function convertRESTHeadToIGitHubRef(head: OctokitCommon.PullsListResponseItemHead): IGitHubRef {

return {

label: head.label,

ref: head.ref,

sha: head.sha,

repo: {

cloneUrl: head.repo.clone_url,

isInOrganization: head.repo.owner.type === 'Organization',

owner: head.repo.owner!.login,

name: head.repo.name

},

};

}

async function transformHtmlUrlsToExtensionUrls(body: string, githubRepository: GitHubRepository): Promise<string> {

const issueRegex = new RegExp(

`href="https?:\/\/${escapeRegExp(githubRepository.remote.gitProtocol.url.authority)}\\/${escapeRegExp(githubRepository.remote.owner)}\\/${escapeRegExp(githubRepository.remote.repositoryName)}\\/(issues|pull)\\/([0-9]+)"`, 'g');

return stringReplaceAsync(body, issueRegex, async (match: string, issuesOrPull: string, number: string) => {

if (issuesOrPull === 'issues') {

return `href="${(await toOpenIssueWebviewUri({ owner: githubRepository.remote.owner, repo: githubRepository.remote.repositoryName, issueNumber: Number(number) })).toString()}""`;

} else {

return `href="${(await toOpenPullRequestWebviewUri({ owner: githubRepository.remote.owner, repo: githubRepository.remote.repositoryName, pullRequestNumber: Number(number) })).toString()}"`;

}

});

}

export function convertRESTPullRequestToRawPullRequest(

pullRequest:

| OctokitCommon.PullsGetResponseData

| OctokitCommon.PullsListResponseItem,

githubRepository: GitHubRepository,

): PullRequest {

const {

number,

body,

title,

html_url,

user,

state,

assignees,

created_at,

updated_at,

head,

base,

labels,

node_id,

id,

draft,

} = pullRequest;

const item: PullRequest = {

id,

graphNodeId: node_id,

number,

body: body ?? '',

title,

titleHTML: title,

url: html_url,

user: convertRESTUserToAccount(user!, githubRepository),

state,

merged: (pullRequest as OctokitCommon.PullsGetResponseData).merged || false,

assignees: assignees

? assignees.map(assignee => convertRESTUserToAccount(assignee!, githubRepository))

: undefined,

createdAt: created_at,

updatedAt: updated_at,

viewerCanUpdate: false,

head: head.repo ? convertRESTHeadToIGitHubRef(head as OctokitCommon.PullsListResponseItemHead) : undefined,

base: convertRESTHeadToIGitHubRef(base),

labels: labels.map<ILabel>(l => ({ name: '', color: '', ...l })),

isDraft: draft,

suggestedReviewers: [], // suggested reviewers only available through GraphQL API

projectItems: [], // projects only available through GraphQL API

commits: [], // commits only available through GraphQL API

reactionCount: 0, // reaction count only available through GraphQL API

reactions: [], // reactions only available through GraphQL API

commentCount: 0 // comment count only available through GraphQL API

};

// mergeable is not included in the list response, will need to fetch later

if ((pullRequest as OctokitCommon.PullsGetResponseData).mergeable !== undefined) {

item.mergeable = (pullRequest as OctokitCommon.PullsGetResponseData).mergeable

? PullRequestMergeability.Mergeable

: PullRequestMergeability.NotMergeable;

}

return item;

}

export function convertRESTIssueToRawPullRequest(

pullRequest: OctokitCommon.IssuesCreateResponseData,

githubRepository: GitHubRepository,

): Issue {

const {

number,

body,

title,

html_url,

user,

state,

assignees,

created_at,

updated_at,

labels,

node_id,

id,

comments

} = pullRequest;

const item: Issue = {

id,

graphNodeId: node_id,

number,

body: body ?? '',

title,

titleHTML: title,

url: html_url,

user: convertRESTUserToAccount(user!, githubRepository),

state,

assignees: assignees

? assignees.map(assignee => convertRESTUserToAccount(assignee!, githubRepository))

: undefined,

createdAt: created_at,

updatedAt: updated_at,

labels: labels.map<ILabel>(l =>

typeof l === 'string' ? { name: l, color: '' } : { name: l.name ?? '', color: l.color ?? '', description: l.description ?? undefined },

),

projectItems: [], // projects only available through GraphQL API

reactionCount: 0, // reaction count only available through GraphQL API

reactions: [], // reactions only available through GraphQL API

commentCount: comments

};

return item;

}

export function convertRESTReviewEvent(

review: OctokitCommon.PullsCreateReviewResponseData,

githubRepository: GitHubRepository,

): Common.ReviewEvent {

return {

event: Common.EventType.Reviewed,

comments: [],

submittedAt: review.submitted_at,

body: review.body,

bodyHTML: review.body,

htmlUrl: review.html_url,

user: convertRESTUserToAccount(review.user!, githubRepository),

authorAssociation: review.user!.type,

state: review.state as 'COMMENTED' | 'APPROVED' | 'CHANGES_REQUESTED' | 'PENDING',

id: review.id,

reactions: undefined // reactions only available through GraphQL API

};

}

export function parseCommentDiffHunk(comment: IComment): DiffHunk[] {

const diffHunks: DiffHunk[] = [];

const diffHunkReader = parseDiffHunk(comment.diffHunk);

let diffHunkIter = diffHunkReader.next();

while (!diffHunkIter.done) {

const diffHunk = diffHunkIter.value;

diffHunks.push(diffHunk);

diffHunkIter = diffHunkReader.next();

}

return diffHunks;

}

export function convertGraphQLEventType(text: string) {

switch (text) {

case 'PullRequestCommit':

return Common.EventType.Committed;

case 'LabeledEvent':

return Common.EventType.Labeled;

case 'MilestonedEvent':

return Common.EventType.Milestoned;

case 'AssignedEvent':

return Common.EventType.Assigned;

case 'UnassignedEvent':

return Common.EventType.Unassigned;

case 'HeadRefDeletedEvent':

return Common.EventType.HeadRefDeleted;

case 'IssueComment':

return Common.EventType.Commented;

case 'PullRequestReview':

return Common.EventType.Reviewed;

case 'MergedEvent':

return Common.EventType.Merged;

case 'CrossReferencedEvent':

return Common.EventType.CrossReferenced;

case 'ClosedEvent':

return Common.EventType.Closed;

case 'ReopenedEvent':

return Common.EventType.Reopened;

case 'BaseRefChangedEvent':

return Common.EventType.BaseRefChanged;

default:

return Common.EventType.Other;

}

}

export function parseGraphQLReviewThread(thread: GraphQL.ReviewThread, githubRepository: GitHubRepository): IReviewThread {

return {

id: thread.id,

prReviewDatabaseId: thread.comments.edges && thread.comments.edges.length ?

thread.comments.edges[0].node.pullRequestReview?.databaseId :

undefined,

isResolved: thread.isResolved,

viewerCanResolve: thread.viewerCanResolve,

viewerCanUnresolve: thread.viewerCanUnresolve,

path: thread.path,

startLine: thread.startLine ?? thread.line,

endLine: thread.line,

originalStartLine: thread.originalStartLine ?? thread.originalLine,

originalEndLine: thread.originalLine,

diffSide: thread.diffSide,

isOutdated: thread.isOutdated,

comments: thread.comments.nodes.map(comment => parseGraphQLComment(comment, thread.isResolved, thread.isOutdated, githubRepository)),

subjectType: thread.subjectType ?? SubjectType.LINE

};

}

export function parseGraphQLComment(comment: GraphQL.ReviewComment, isResolved: boolean, isOutdated: boolean, githubRepository: GitHubRepository): IComment {

const specialAuthor = COPILOT_ACCOUNTS[comment.author?.login ?? ''];

const c: IComment = {

id: comment.databaseId,

url: comment.url,

body: comment.body,

specialDisplayBodyPostfix: specialAuthor?.postComment,

bodyHTML: comment.bodyHTML,

path: comment.path,

canEdit: comment.viewerCanDelete,

canDelete: comment.viewerCanDelete,

pullRequestReviewId: comment.pullRequestReview && comment.pullRequestReview.databaseId,

diffHunk: comment.diffHunk,

position: comment.position,

commitId: comment.commit.oid,

originalPosition: comment.originalPosition,

originalCommitId: comment.originalCommit && comment.originalCommit.oid,

user: comment.author ? parseAccount(comment.author, githubRepository) : undefined,

createdAt: comment.createdAt,

htmlUrl: comment.url,

graphNodeId: comment.id,

isDraft: comment.state === 'PENDING',

inReplyToId: comment.replyTo && comment.replyTo.databaseId,

reactions: parseGraphQLReaction(comment.reactionGroups),

isResolved,

isOutdated

};

const diffHunks = parseCommentDiffHunk(c);

c.diffHunks = diffHunks;

return c;

}

export function parseGraphQlIssueComment(comment: GraphQL.IssueComment, githubRepository: GitHubRepository): IComment {

return {

id: comment.databaseId,

url: comment.url,

body: comment.body,

specialDisplayBodyPostfix: COPILOT_ACCOUNTS[comment.author?.login ?? '']?.postComment,

bodyHTML: comment.bodyHTML,

canEdit: comment.viewerCanDelete,

canDelete: comment.viewerCanDelete,

user: parseAccount(comment.author, githubRepository),

createdAt: comment.createdAt,

htmlUrl: comment.url,

graphNodeId: comment.id,

diffHunk: '',

reactions: parseGraphQLReaction(comment.reactionGroups)

};

}

export function parseGraphQLReaction(reactionGroups: GraphQL.ReactionGroup[]): Reaction[] {

const reactionContentEmojiMapping = getReactionGroup().reduce((prev, curr) => {

prev[curr.title] = curr;

return prev;

}, {} as { [key: string]: { title: string; label: string; icon?: string } });

const reactions = reactionGroups

.filter(group => group.reactors.totalCount > 0)

.map(group => {

const reaction: Reaction = {

label: reactionContentEmojiMapping[group.content].label,

count: group.reactors.totalCount,

icon: reactionContentEmojiMapping[group.content].icon,

viewerHasReacted: group.viewerHasReacted,

reactors: group.reactors.nodes.map(node => COPILOT_ACCOUNTS[node.login]?.name ?? node.login)

};

return reaction;

});

return reactions;

}

function parseRef(refName: string, oid: string, repository?: GraphQL.RefRepository): IGitHubRef | undefined {

if (!repository) {

return undefined;

}

return {

label: `${repository.owner.login}:${refName}`,

ref: refName,

sha: oid,

repo: {

cloneUrl: repository.url,

isInOrganization: repository.isInOrganization,

owner: repository.owner.login,

name: refName

},

};

}

export interface RestAccount {

login: string;

html_url: string;

avatar_url: string;

email?: string | null;

node_id: string;

name?: string | null;

type: string;

}

export interface GraphQLAccount {

login: string;

url: string;

avatarUrl: string;

email?: string;

id: string;

name?: string;

__typename: string;

}

export function parseAccount(

author: GraphQLAccount | RestAccount | null,

githubRepository?: GitHubRepository,

): IAccount {

if (author) {

let avatarUrl: string;

let id: string;

let url: string;

let accountType: string;

if ((author as RestAccount).html_url) {

const asRestAccount = author as RestAccount;

avatarUrl = asRestAccount.avatar_url;

id = asRestAccount.node_id;

url = asRestAccount.html_url;

accountType = asRestAccount.type;

} else {

const asGraphQLAccount = author as GraphQLAccount;

avatarUrl = asGraphQLAccount.avatarUrl;

id = asGraphQLAccount.id;

url = asGraphQLAccount.url;

accountType = asGraphQLAccount.__typename;

}

// In some places, Copilot comes in as a user, and in others as a bot

const finalAvatarUrl = githubRepository ? getAvatarWithEnterpriseFallback(avatarUrl, undefined, githubRepository.remote.isEnterprise) : avatarUrl;

return {

login: author.login,

url: COPILOT_ACCOUNTS[author.login]?.url ?? url,

avatarUrl: finalAvatarUrl,

email: author.email ?? undefined,

id,

name: author.name ?? COPILOT_ACCOUNTS[author.login]?.name ?? undefined,

specialDisplayName: COPILOT_ACCOUNTS[author.login] ? (author.name ?? COPILOT_ACCOUNTS[author.login].name) : undefined,

accountType: toAccountType(accountType),

};

} else {

return {

login: '',

url: '',

id: '',

accountType: AccountType.User

};

}

}

function parseTeam(team: GraphQL.Team, githubRepository: GitHubRepository): ITeam {

return {

name: team.name,

url: team.url,

avatarUrl: getAvatarWithEnterpriseFallback(team.avatarUrl, undefined, githubRepository.remote.isEnterprise),

id: team.id,

org: githubRepository.remote.owner,

slug: team.slug

};

}

export function parseGraphQLReviewers(data: GraphQL.GetReviewRequestsResponse, repository: GitHubRepository): (IAccount | ITeam)[] {

if (!data.repository) {

return [];

}

const reviewers: (IAccount | ITeam)[] = [];

for (const reviewer of data.repository.pullRequest.reviewRequests.nodes) {

if (GraphQL.isTeam(reviewer.requestedReviewer)) {

const team: ITeam = parseTeam(reviewer.requestedReviewer, repository);

reviewers.push(team);

} else if (GraphQL.isAccount(reviewer.requestedReviewer) || GraphQL.isBot(reviewer.requestedReviewer)) {

const account: IAccount = parseAccount(reviewer.requestedReviewer, repository);

reviewers.push(account);

}

}

return reviewers;

}

function parseActor(

author: { login: string; url: string; avatarUrl: string; } | null,

githubRepository: GitHubRepository,

): IActor {

if (author) {

return {

login: author.login,

url: author.url,

avatarUrl: getAvatarWithEnterpriseFallback(author.avatarUrl, undefined, githubRepository.remote.isEnterprise),

};

} else {

return {

login: '',

url: '',

};

}

}

export function parseProjectItems(projects: { id: string; project: { id: string; title: string; } }[] | undefined): IProjectItem[] | undefined {

if (!projects) {

return undefined;

}

return projects.map(project => {

return {

id: project.id,

project: project.project

};

});

}

export function parseMilestone(

milestone: { title: string; dueOn?: string; createdAt: string; id: string, number: number } | undefined,

): IMilestone | undefined {

if (!milestone) {

return undefined;

}

return {

title: milestone.title,

dueOn: milestone.dueOn,

createdAt: milestone.createdAt,

id: milestone.id,

number: milestone.number

};

}

export function parseMergeQueueEntry(mergeQueueEntry: GraphQL.MergeQueueEntry | null | undefined): MergeQueueEntry | undefined | null {

if (!mergeQueueEntry) {

return null;

}

let state: MergeQueueState;

switch (mergeQueueEntry.state) {

case 'AWAITING_CHECKS': {

state = MergeQueueState.AwaitingChecks;

break;

}

case 'LOCKED': {

state = MergeQueueState.Locked;

break;

}

case 'QUEUED': {

state = MergeQueueState.Queued;

break;

}

case 'MERGEABLE': {

state = MergeQueueState.Mergeable;

break;

}

case 'UNMERGEABLE': {

state = MergeQueueState.Unmergeable;

break;

}

}

return { position: mergeQueueEntry.position, state, url: mergeQueueEntry.mergeQueue.url };

}

export function parseMergeMethod(mergeMethod: GraphQL.MergeMethod | undefined): MergeMethod | undefined {

switch (mergeMethod) {

case 'MERGE': return 'merge';

case 'REBASE': return 'rebase';

case 'SQUASH': return 'squash';

}

}

export function parseMergeability(mergeability: 'UNKNOWN' | 'MERGEABLE' | 'CONFLICTING' | undefined,

mergeStateStatus: 'BEHIND' | 'BLOCKED' | 'CLEAN' | 'DIRTY' | 'HAS_HOOKS' | 'UNKNOWN' | 'UNSTABLE' | undefined): PullRequestMergeability {

let parsed: PullRequestMergeability;

switch (mergeability) {

case undefined:

case 'UNKNOWN':

parsed = PullRequestMergeability.Unknown;

break;

case 'MERGEABLE':

parsed = PullRequestMergeability.Mergeable;

break;

case 'CONFLICTING':

parsed = PullRequestMergeability.Conflict;

break;

}

if (parsed !== PullRequestMergeability.Conflict) {

if (mergeStateStatus === 'BLOCKED') {

parsed = PullRequestMergeability.NotMergeable;

} else if (mergeStateStatus === 'BEHIND') {

parsed = PullRequestMergeability.Behind;

}

}

return parsed;

}

export async function parseGraphQLPullRequest(

graphQLPullRequest: GraphQL.PullRequest,

githubRepository: GitHubRepository,

): Promise<PullRequest> {

const pr: PullRequest = {

id: graphQLPullRequest.databaseId,

graphNodeId: graphQLPullRequest.id,

url: graphQLPullRequest.url,

number: graphQLPullRequest.number,

state: graphQLPullRequest.state,

body: graphQLPullRequest.body,

bodyHTML: await transformHtmlUrlsToExtensionUrls(graphQLPullRequest.bodyHTML, githubRepository),

title: graphQLPullRequest.title,

titleHTML: graphQLPullRequest.titleHTML,

createdAt: graphQLPullRequest.createdAt,

updatedAt: graphQLPullRequest.updatedAt,

isRemoteHeadDeleted: !graphQLPullRequest.headRef,

head: parseRef(graphQLPullRequest.headRef?.name ?? graphQLPullRequest.headRefName, graphQLPullRequest.headRefOid, graphQLPullRequest.headRepository),

isRemoteBaseDeleted: !graphQLPullRequest.baseRef,

base: parseRef(graphQLPullRequest.baseRef?.name ?? graphQLPullRequest.baseRefName, graphQLPullRequest.baseRefOid, graphQLPullRequest.baseRepository),

user: parseAccount(graphQLPullRequest.author, githubRepository),

merged: graphQLPullRequest.merged,

mergeable: parseMergeability(graphQLPullRequest.mergeable, graphQLPullRequest.mergeStateStatus),

mergeQueueEntry: parseMergeQueueEntry(graphQLPullRequest.mergeQueueEntry),

hasComments: graphQLPullRequest.reviewThreads.totalCount > 0,

autoMerge: !!graphQLPullRequest.autoMergeRequest,

autoMergeMethod: parseMergeMethod(graphQLPullRequest.autoMergeRequest?.mergeMethod),

allowAutoMerge: graphQLPullRequest.viewerCanEnableAutoMerge || graphQLPullRequest.viewerCanDisableAutoMerge,

viewerCanUpdate: graphQLPullRequest.viewerCanUpdate,

labels: graphQLPullRequest.labels.nodes,

isDraft: graphQLPullRequest.isDraft,

suggestedReviewers: parseSuggestedReviewers(graphQLPullRequest.suggestedReviewers),

comments: parseComments(graphQLPullRequest.comments?.nodes, githubRepository),

projectItems: parseProjectItems(graphQLPullRequest.projectItems?.nodes),

milestone: parseMilestone(graphQLPullRequest.milestone),

assignees: graphQLPullRequest.assignees?.nodes.map(assignee => parseAccount(assignee, githubRepository)),

commits: parseCommits(graphQLPullRequest.commits.nodes),

reactionCount: graphQLPullRequest.reactions.totalCount,

reactions: parseGraphQLReaction(graphQLPullRequest.reactionGroups),

commentCount: graphQLPullRequest.comments.totalCount,

additions: graphQLPullRequest.additions,

deletions: graphQLPullRequest.deletions,

};

pr.mergeCommitMeta = parseCommitMeta(graphQLPullRequest.baseRepository.mergeCommitTitle, graphQLPullRequest.baseRepository.mergeCommitMessage, pr);

pr.squashCommitMeta = parseCommitMeta(graphQLPullRequest.baseRepository.squashMergeCommitTitle, graphQLPullRequest.baseRepository.squashMergeCommitMessage, pr);

return pr;

}

function parseCommitMeta(titleSource: GraphQL.DefaultCommitTitle | undefined, descriptionSource: GraphQL.DefaultCommitMessage | undefined, pullRequest: PullRequest): { title: string, description: string } | undefined {

if (titleSource === undefined || descriptionSource === undefined) {

return undefined;

}

let title = '';

let description = '';

const prNumberPostfix = `(#${pullRequest.number})`;

switch (titleSource) {

case GraphQL.DefaultCommitTitle.prTitle: {

title = `${pullRequest.title} ${prNumberPostfix}`;

break;

}

case GraphQL.DefaultCommitTitle.mergeMessage: {

title = `Merge pull request #${pullRequest.number} from ${pullRequest.head?.label ?? ''}`;

break;

}

case GraphQL.DefaultCommitTitle.commitOrPrTitle: {

if (pullRequest.commits.length === 1) {

title = `${pullRequest.commits[0].message.split('\n')[0]} ${prNumberPostfix}`;

} else {

title = `${pullRequest.title} ${prNumberPostfix}`;

}

break;

}

}

switch (descriptionSource) {

case GraphQL.DefaultCommitMessage.prBody: {

description = pullRequest.body;

break;

}

case GraphQL.DefaultCommitMessage.commitMessages: {

if ((pullRequest.commits.length === 1) && (titleSource === GraphQL.DefaultCommitTitle.commitOrPrTitle)) {

const split = pullRequest.commits[0].message.split('\n');

description = split.length > 1 ? split.slice(1).join('\n').trim() : '';

} else {

description = pullRequest.commits.map(commit => `* ${commit.message}`).join('\n\n');

}

break;

}

case GraphQL.DefaultCommitMessage.prTitle: {

description = pullRequest.title;

break;

}

}

return { title, description };

}

function parseCommits(commits: { commit: { message: string; }; }[]): { message: string; }[] {

return commits.map(commit => {

return {

message: commit.commit.message

};

});

}

function parseComments(comments: GraphQL.AbbreviatedIssueComment[] | undefined, githubRepository: GitHubRepository) {

if (!comments) {

return;

}

const parsedComments: {

author: IAccount;

body: string;

databaseId: number;

reactionCount: number;

createdAt: string;

}[] = [];

for (const comment of comments) {

parsedComments.push({

author: parseAccount(comment.author, githubRepository),

body: comment.body,

databaseId: comment.databaseId,

reactionCount: comment.reactions.totalCount,

createdAt: comment.createdAt

});

}

return parsedComments;

}

export async function parseGraphQLIssue(issue: GraphQL.Issue, githubRepository: GitHubRepository): Promise<Issue> {

return {

id: issue.databaseId,

graphNodeId: issue.id,

url: issue.url,

number: issue.number,

state: issue.state,

stateReason: issue.stateReason,

body: issue.body,

bodyHTML: await transformHtmlUrlsToExtensionUrls(issue.bodyHTML, githubRepository),

title: issue.title,

titleHTML: issue.titleHTML,

createdAt: issue.createdAt,

updatedAt: issue.updatedAt,

assignees: issue.assignees?.nodes.map(assignee => parseAccount(assignee, githubRepository)),

user: parseAccount(issue.author, githubRepository),

labels: issue.labels.nodes,

milestone: parseMilestone(issue.milestone),

repositoryName: issue.repository?.name ?? githubRepository.remote.repositoryName,

repositoryOwner: issue.repository?.owner.login ?? githubRepository.remote.owner,

repositoryUrl: issue.repository?.url ?? githubRepository.remote.url,

projectItems: parseProjectItems(issue.projectItems?.nodes),

comments: issue.comments.nodes?.map(comment => parseIssueComment(comment, githubRepository)),

reactionCount: issue.reactions.totalCount,

reactions: parseGraphQLReaction(issue.reactionGroups),

commentCount: issue.comments.totalCount

};

}

function parseIssueComment(comment: GraphQL.AbbreviatedIssueComment, githubRepository: GitHubRepository): IIssueComment {

return {

author: parseAccount(comment.author, githubRepository),

body: comment.body,

databaseId: comment.databaseId,

reactionCount: comment.reactions.totalCount,

createdAt: comment.createdAt,

};

}

function parseSuggestedReviewers(

suggestedReviewers: GraphQL.SuggestedReviewerResponse[] | undefined,

): ISuggestedReviewer[] {

if (!suggestedReviewers) {

return [];

}

const ret: ISuggestedReviewer[] = suggestedReviewers.map(suggestedReviewer => {