feat(codereview): add JetBrains integration for code review functionality (#593)

Author: unknownzjc

packages/types/src/vscode.ts

@@ -74,6 +74,11 @@ export const costrictCommandIds = [
 	// code review
 	"codeReviewButtonClicked",
 	"codeReview",
+	"codeReviewJetbrains",
+	"reviewFilesAndFoldersJetbrains",
+	"askReviewSuggestionWithAIJetbrains",
+	"acceptIssueJetbrains",
+	"rejectIssueJetbrains",
 	"reviewFilesAndFolders",
 	"reviewRepo",
 	"askReviewSuggestionWithAI",

src/core/costrict/code-review/codeReviewService.ts

@@ -48,8 +48,8 @@ import type { ClineProvider } from "../../webview/ClineProvider"
 import { TelemetryService } from "@roo-code/telemetry"
 import { CodeReviewErrorType, type TelemetryErrorType } from "../telemetry"
 import { COSTRICT_DEFAULT_HEADERS } from "../../../shared/headers"
-import { zgsmCodebaseIndexManager } from "../codebase-index"
 import { fileExistsAtPath } from "../../../utils/fs"
+import { isJetbrainsPlatform } from "../../../utils/platform"
 /**
  * Code Review Service - Singleton
  *
@@ -793,7 +793,7 @@ export class CodeReviewService {
 				vscode.CommentMode.Preview,
 				{ name: "CoStrict", iconPath },
 				undefined,
-				"Intial",
+				isJetbrainsPlatform() ? issue.id : "Intial",
 			),
 		}
 	}

src/core/costrict/code-review/index.ts

@@ -120,6 +120,115 @@ export function initCodeReview(
 				reviewInstance.askWithAI(comment.id)
 			}
 		},
+		codeReviewJetbrains: async (args: any) => {
+			const visibleProvider = await ClineProvider.getInstance()
+			if (!visibleProvider) {
+				return
+			}
+			visibleProvider.log(`[CodeReview] start review ${args}`)
+
+			const data = args?.[0]?.[0]
+			if (!data) {
+				visibleProvider.log("[CodeReview] Invalid args structure")
+				return
+			}
+
+			const { startLine, endLine, filePath, selectedText } = data
+			visibleProvider.log(
+				`[CodeReview] extracted data: filePath=${filePath}, startLine=${startLine}, endLine=${endLine}`,
+			)
+
+			const cwd = visibleProvider.cwd.toPosix()
+			reviewInstance.setProvider(visibleProvider)
+			const params = {
+				filePath,
+				endLine: endLine + "",
+				startLine: startLine + "",
+				selectedText: selectedText,
+			}
+			const prompt = supportPrompt.create("ADD_TO_CONTEXT", params)
+			reviewInstance.createReviewTask(prompt, [
+				{
+					type: ReviewTargetType.CODE,
+					file_path: toRelativePath(filePath.toPosix(), cwd),
+					line_range: [startLine, endLine],
+				},
+			])
+		},
+		reviewFilesAndFoldersJetbrains: async (args: any) => {
+			const visibleProvider = await ClineProvider.getInstance()
+			if (!visibleProvider) {
+				return
+			}
+			visibleProvider.log(`[CodeReview] start review ${JSON.stringify(args)}`)
+			const data = args?.[0]?.[0]
+			if (!data) {
+				visibleProvider.log("[CodeReview] Invalid args structure")
+				return
+			}
+			const cwd = visibleProvider.cwd.toPosix()
+			const { filePaths } = data
+			const targets: ReviewTarget[] = await Promise.all(
+				filePaths.map(async (filePath: string) => {
+					return {
+						type: ReviewTargetType.FILE,
+						file_path: toRelativePath(filePath.toPosix(), cwd),
+					}
+				}),
+			)
+			reviewInstance.setProvider(visibleProvider)
+			reviewInstance.startReview(targets)
+		},
+		acceptIssueJetbrains: async (args: any) => {
+			const visibleProvider = await ClineProvider.getInstance()
+			if (!visibleProvider) {
+				return
+			}
+			reviewInstance.setProvider(visibleProvider)
+			visibleProvider.log(`[CodeReview] accept issue ${JSON.stringify(args)}`)
+			const data = args?.[0]?.[0]
+			if (!data) {
+				visibleProvider.log("[CodeReview] Invalid args structure")
+				return
+			}
+
+			const { id } = data
+			reviewInstance.updateIssueStatus(id, IssueStatus.ACCEPT)
+		},
+		rejectIssueJetbrains: async (args: any) => {
+			const visibleProvider = await ClineProvider.getInstance()
+			if (!visibleProvider) {
+				return
+			}
+			reviewInstance.setProvider(visibleProvider)
+			visibleProvider.log(`[CodeReview] reject issue ${JSON.stringify(args)}`)
+			const data = args?.[0]?.[0]
+			if (!data) {
+				visibleProvider.log("[CodeReview] Invalid args structure")
+				return
+			}
+
+			const { id } = data
+			reviewInstance.updateIssueStatus(id, IssueStatus.REJECT)
+		},
+		askReviewSuggestionWithAIJetbrains: async (args: any) => {
+			const visibleProvider = await ClineProvider.getInstance()
+			if (!visibleProvider) {
+				return
+			}
+			visibleProvider.log(`[CodeReview] ask review suggestion with AI ${JSON.stringify(args)}`)
+			reviewInstance.setProvider(visibleProvider)
+			const data = args?.[0]?.[0]
+			if (!data) {
+				visibleProvider.log("[CodeReview] Invalid args structure")
+				return
+			}
+
+			const { id } = data
+			if (id) {
+				reviewInstance.askWithAI(id)
+			}
+		},
 	}
 	for (const [id, callback] of Object.entries(commandMap)) {
 		const command = getCommand(id as CostrictCommandId)

src/package.json

@@ -195,6 +195,26 @@
 				"command": "zgsm.reviewRepo",
 				"title": "%command.reviewRepo.title%"
 			},
+			{
+				"command": "zgsm.codeReviewJetbrains",
+				"title": "%command.codeReview.title%"
+			},
+			{
+				"command": "zgsm.reviewFilesAndFoldersJetbrains",
+				"title": "%command.codeReview.title%"
+			},
+			{
+				"command": "zgsm.askReviewSuggestionWithAIJetbrains",
+				"title": "%command.codeReview.title%"
+			},
+			{
+				"command": "zgsm.acceptIssueJetbrains",
+				"title": "%command.codeReview.title%"
+			},
+			{
+				"command": "zgsm.rejectIssueJetbrains",
+				"title": "%command.codeReview.title%"
+			},
 			{
 				"command": "zgsm.acceptIssue",
 				"title": "Accept",

src/utils/platform.ts

@@ -0,0 +1,4 @@
+import * as vscode from "vscode"
+export function isJetbrainsPlatform(): boolean {
+	return ["intellij-machine", "development-machine"].includes(vscode?.env?.machineId)
+}

webview-ui/src/components/code-review/FileIssueList.tsx

@@ -29,14 +29,16 @@ const FileIssueList: React.FC<FileIssueListProps> = ({ fileName, issues, onIssue
 	return (
 		<div className="w-full" style={{ opacity: allIssuesProcessed ? 0.4 : 1 }}>
 			<div className="flex justify-between pr-5">
-				<div className="flex items-center gap-2">
-					<SetiFileIcon fileName={fileName} size={16} />
-					<span>{fileName}</span>
-					<div onClick={() => setIsExpanded(!isExpanded)} className="cursor-pointer">
+				<div className="flex items-center gap-2 min-w-0 flex-1 overflow-hidden">
+					<SetiFileIcon fileName={fileName} size={16} className="flex-shrink-0" />
+					<span className="truncate" title={fileName}>
+						{fileName}
+					</span>
+					<div onClick={() => setIsExpanded(!isExpanded)} className="cursor-pointer flex-shrink-0">
 						{isExpanded ? <ChevronUpIcon /> : <ChevronDownIcon />}
 					</div>
 				</div>
-				<div className="flex items-center gap-1">
+				<div className="flex items-center gap-1 flex-shrink-0">
 					<span className="codicon codicon-bug"></span>
 					<span>{issues.length}</span>
 				</div>