Skip to content

Commit a3fdc4e

Browse files
bramburnbramburn
committed
Merge branch 'main' of https://github.com/RooVetGit/Roo-Code into refactor/cline.ts/list_files
2 parents 222b804 + e7e5511 commit a3fdc4e

File tree

10 files changed

+426
-417
lines changed

10 files changed

+426
-417
lines changed

‎package.json‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -185,19 +185,19 @@
185185
],
186186
"roo-code.contextMenu": [
187187
{
188-
"command": "roo-cline.explainCode",
188+
"command": "roo-cline.addToContext",
189189
"group": "1_actions@1"
190190
},
191191
{
192-
"command": "roo-cline.fixCode",
192+
"command": "roo-cline.explainCode",
193193
"group": "1_actions@2"
194194
},
195195
{
196-
"command": "roo-cline.improveCode",
196+
"command": "roo-cline.fixCode",
197197
"group": "1_actions@3"
198198
},
199199
{
200-
"command": "roo-cline.addToContext",
200+
"command": "roo-cline.improveCode",
201201
"group": "1_actions@4"
202202
}
203203
],

‎src/activate/registerCodeActions.ts‎

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,20 +53,24 @@ const registerCodeAction = (
5353
// Handle both code action and direct command cases.
5454
let filePath: string
5555
let selectedText: string
56+
let startLine: number | undefined
57+
let endLine: number | undefined
5658
let diagnostics: any[] | undefined
5759

5860
if (args.length > 1) {
5961
// Called from code action.
60-
;[filePath, selectedText, diagnostics] = args
62+
;[filePath, selectedText, startLine, endLine, diagnostics] = args
6163
} else {
6264
// Called directly from command palette.
6365
const context = EditorUtils.getEditorContext()
6466
if (!context) return
65-
;({ filePath, selectedText, diagnostics } = context)
67+
;({ filePath, selectedText, startLine, endLine, diagnostics } = context)
6668
}
6769

6870
const params = {
6971
...{ filePath, selectedText },
72+
...(startLine !== undefined ? { startLine: startLine.toString() } : {}),
73+
...(endLine !== undefined ? { endLine: endLine.toString() } : {}),
7074
...(diagnostics ? { diagnostics } : {}),
7175
...(userInput ? { userInput } : {}),
7276
}

‎src/core/Cline.ts‎

Lines changed: 3 additions & 147 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,7 @@ import {
3131
import { countFileLines } from "../integrations/misc/line-counter"
3232
import { fetchInstructionsTool } from "./tools/fetchInstructionsTool"
3333
import { listFilesTool } from "./tools/listFilesTool"
34+
import { readFileTool } from "./tools/readFileTool"
3435
import { ExitCodeDetails } from "../integrations/terminal/TerminalProcess"
3536
import { Terminal } from "../integrations/terminal/Terminal"
3637
import { TerminalRegistry } from "../integrations/terminal/TerminalRegistry"
@@ -83,9 +84,7 @@ import { insertGroups } from "./diff/insert-groups"
8384
import { telemetryService } from "../services/telemetry/TelemetryService"
8485
import { validateToolUse, isToolAllowedForMode, ToolName } from "./mode-validator"
8586
import { parseXml } from "../utils/xml"
86-
import { readLines } from "../integrations/misc/read-lines"
8787
import { getWorkspacePath } from "../utils/path"
88-
import { isBinaryFile } from "isbinaryfile"
8988

9089
export type ToolResponse = string | Array<Anthropic.TextBlockParam | Anthropic.ImageBlockParam>
9190
type UserContent = Array<Anthropic.Messages.ContentBlockParam>
@@ -2257,151 +2256,8 @@ export class Cline extends EventEmitter<ClineEvents> {
22572256
}
22582257

22592258
case "read_file": {
2260-
const relPath: string | undefined = block.params.path
2261-
const startLineStr: string | undefined = block.params.start_line
2262-
const endLineStr: string | undefined = block.params.end_line
2263-
2264-
// Get the full path and determine if it's outside the workspace
2265-
const fullPath = relPath ? path.resolve(this.cwd, removeClosingTag("path", relPath)) : ""
2266-
const isOutsideWorkspace = isPathOutsideWorkspace(fullPath)
2267-
2268-
const sharedMessageProps: ClineSayTool = {
2269-
tool: "readFile",
2270-
path: getReadablePath(this.cwd, removeClosingTag("path", relPath)),
2271-
isOutsideWorkspace,
2272-
}
2273-
try {
2274-
if (block.partial) {
2275-
const partialMessage = JSON.stringify({
2276-
...sharedMessageProps,
2277-
content: undefined,
2278-
} satisfies ClineSayTool)
2279-
await this.ask("tool", partialMessage, block.partial).catch(() => {})
2280-
break
2281-
} else {
2282-
if (!relPath) {
2283-
this.consecutiveMistakeCount++
2284-
pushToolResult(await this.sayAndCreateMissingParamError("read_file", "path"))
2285-
break
2286-
}
2287-
2288-
// Check if we're doing a line range read
2289-
let isRangeRead = false
2290-
let startLine: number | undefined = undefined
2291-
let endLine: number | undefined = undefined
2292-
2293-
// Check if we have either range parameter
2294-
if (startLineStr || endLineStr) {
2295-
isRangeRead = true
2296-
}
2297-
2298-
// Parse start_line if provided
2299-
if (startLineStr) {
2300-
startLine = parseInt(startLineStr)
2301-
if (isNaN(startLine)) {
2302-
// Invalid start_line
2303-
this.consecutiveMistakeCount++
2304-
await this.say("error", `Failed to parse start_line: ${startLineStr}`)
2305-
pushToolResult(formatResponse.toolError("Invalid start_line value"))
2306-
break
2307-
}
2308-
startLine -= 1 // Convert to 0-based index
2309-
}
2310-
2311-
// Parse end_line if provided
2312-
if (endLineStr) {
2313-
endLine = parseInt(endLineStr)
2314-
2315-
if (isNaN(endLine)) {
2316-
// Invalid end_line
2317-
this.consecutiveMistakeCount++
2318-
await this.say("error", `Failed to parse end_line: ${endLineStr}`)
2319-
pushToolResult(formatResponse.toolError("Invalid end_line value"))
2320-
break
2321-
}
2322-
2323-
// Convert to 0-based index
2324-
endLine -= 1
2325-
}
2326-
2327-
const accessAllowed = this.rooIgnoreController?.validateAccess(relPath)
2328-
if (!accessAllowed) {
2329-
await this.say("rooignore_error", relPath)
2330-
pushToolResult(formatResponse.toolError(formatResponse.rooIgnoreError(relPath)))
2331-
2332-
break
2333-
}
2334-
2335-
this.consecutiveMistakeCount = 0
2336-
const absolutePath = path.resolve(this.cwd, relPath)
2337-
const completeMessage = JSON.stringify({
2338-
...sharedMessageProps,
2339-
content: absolutePath,
2340-
} satisfies ClineSayTool)
2341-
2342-
const didApprove = await askApproval("tool", completeMessage)
2343-
if (!didApprove) {
2344-
break
2345-
}
2346-
2347-
// Get the maxReadFileLine setting
2348-
const { maxReadFileLine = 500 } = (await this.providerRef.deref()?.getState()) ?? {}
2349-
2350-
// Count total lines in the file
2351-
let totalLines = 0
2352-
try {
2353-
totalLines = await countFileLines(absolutePath)
2354-
} catch (error) {
2355-
console.error(`Error counting lines in file ${absolutePath}:`, error)
2356-
}
2357-
2358-
// now execute the tool like normal
2359-
let content: string
2360-
let isFileTruncated = false
2361-
let sourceCodeDef = ""
2362-
2363-
const isBinary = await isBinaryFile(absolutePath).catch(() => false)
2364-
2365-
if (isRangeRead) {
2366-
if (startLine === undefined) {
2367-
content = addLineNumbers(await readLines(absolutePath, endLine, startLine))
2368-
} else {
2369-
content = addLineNumbers(
2370-
await readLines(absolutePath, endLine, startLine),
2371-
startLine + 1,
2372-
)
2373-
}
2374-
} else if (!isBinary && maxReadFileLine >= 0 && totalLines > maxReadFileLine) {
2375-
// If file is too large, only read the first maxReadFileLine lines
2376-
isFileTruncated = true
2377-
2378-
const res = await Promise.all([
2379-
maxReadFileLine > 0 ? readLines(absolutePath, maxReadFileLine - 1, 0) : "",
2380-
parseSourceCodeDefinitionsForFile(absolutePath, this.rooIgnoreController),
2381-
])
2382-
2383-
content = res[0].length > 0 ? addLineNumbers(res[0]) : ""
2384-
const result = res[1]
2385-
if (result) {
2386-
sourceCodeDef = `\n\n${result}`
2387-
}
2388-
} else {
2389-
// Read entire file
2390-
content = await extractTextFromFile(absolutePath)
2391-
}
2392-
2393-
// Add truncation notice if applicable
2394-
if (isFileTruncated) {
2395-
content += `\n\n[Showing only ${maxReadFileLine} of ${totalLines} total lines. Use start_line and end_line if you need to read more]${sourceCodeDef}`
2396-
}
2397-
2398-
pushToolResult(content)
2399-
break
2400-
}
2401-
} catch (error) {
2402-
await handleError("reading file", error)
2403-
break
2404-
}
2259+
readFileTool(this, block, askApproval, handleError, pushToolResult, removeClosingTag)
2260+
break
24052261
}
24062262

24072263
case "fetch_instructions": {

‎src/core/CodeActionProvider.ts‎

Lines changed: 26 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -56,10 +56,26 @@ export class CodeActionProvider implements vscode.CodeActionProvider {
5656
const filePath = EditorUtils.getFilePath(document)
5757
const actions: vscode.CodeAction[] = []
5858

59+
actions.push(
60+
this.createAction(
61+
ACTION_NAMES.ADD_TO_CONTEXT,
62+
vscode.CodeActionKind.QuickFix,
63+
COMMAND_IDS.ADD_TO_CONTEXT,
64+
[
65+
filePath,
66+
effectiveRange.text,
67+
effectiveRange.range.start.line + 1,
68+
effectiveRange.range.end.line + 1,
69+
],
70+
),
71+
)
72+
5973
actions.push(
6074
...this.createActionPair(ACTION_NAMES.EXPLAIN, vscode.CodeActionKind.QuickFix, COMMAND_IDS.EXPLAIN, [
6175
filePath,
6276
effectiveRange.text,
77+
effectiveRange.range.start.line + 1,
78+
effectiveRange.range.end.line + 1,
6379
]),
6480
)
6581

@@ -74,6 +90,8 @@ export class CodeActionProvider implements vscode.CodeActionProvider {
7490
...this.createActionPair(ACTION_NAMES.FIX, vscode.CodeActionKind.QuickFix, COMMAND_IDS.FIX, [
7591
filePath,
7692
effectiveRange.text,
93+
effectiveRange.range.start.line + 1,
94+
effectiveRange.range.end.line + 1,
7795
diagnosticMessages,
7896
]),
7997
)
@@ -83,6 +101,8 @@ export class CodeActionProvider implements vscode.CodeActionProvider {
83101
...this.createActionPair(ACTION_NAMES.FIX_LOGIC, vscode.CodeActionKind.QuickFix, COMMAND_IDS.FIX, [
84102
filePath,
85103
effectiveRange.text,
104+
effectiveRange.range.start.line + 1,
105+
effectiveRange.range.end.line + 1,
86106
]),
87107
)
88108
}
@@ -92,16 +112,12 @@ export class CodeActionProvider implements vscode.CodeActionProvider {
92112
ACTION_NAMES.IMPROVE,
93113
vscode.CodeActionKind.RefactorRewrite,
94114
COMMAND_IDS.IMPROVE,
95-
[filePath, effectiveRange.text],
96-
),
97-
)
98-
99-
actions.push(
100-
this.createAction(
101-
ACTION_NAMES.ADD_TO_CONTEXT,
102-
vscode.CodeActionKind.QuickFix,
103-
COMMAND_IDS.ADD_TO_CONTEXT,
104-
[filePath, effectiveRange.text],
115+
[
116+
filePath,
117+
effectiveRange.text,
118+
effectiveRange.range.start.line + 1,
119+
effectiveRange.range.end.line + 1,
120+
],
105121
),
106122
)
107123

‎src/core/EditorUtils.ts‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,10 @@ export interface EditorContext {
3838
filePath: string
3939
/** The effective text selected or derived from the document. */
4040
selectedText: string
41+
/** The starting line number of the selected text (1-based). */
42+
startLine: number
43+
/** The ending line number of the selected text (1-based). */
44+
endLine: number
4145
/** Optional list of diagnostics associated with the effective range. */
4246
diagnostics?: DiagnosticData[]
4347
}
@@ -194,6 +198,8 @@ export class EditorUtils {
194198
return {
195199
filePath,
196200
selectedText: effectiveRange.text,
201+
startLine: effectiveRange.range.start.line + 1, // Convert to 1-based line numbers
202+
endLine: effectiveRange.range.end.line + 1, // Convert to 1-based line numbers
197203
...(diagnostics.length > 0 ? { diagnostics } : {}),
198204
}
199205
} catch (error) {

‎src/core/__tests__/CodeActionProvider.test.ts‎

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -75,13 +75,13 @@ describe("CodeActionProvider", () => {
7575
const actions = provider.provideCodeActions(mockDocument, mockRange, mockContext)
7676

7777
expect(actions).toHaveLength(7) // 2 explain + 2 fix logic + 2 improve + 1 add to context
78-
expect((actions as any)[0].title).toBe(`${ACTION_NAMES.EXPLAIN} in New Task`)
79-
expect((actions as any)[1].title).toBe(`${ACTION_NAMES.EXPLAIN} in Current Task`)
80-
expect((actions as any)[2].title).toBe(`${ACTION_NAMES.FIX_LOGIC} in New Task`)
81-
expect((actions as any)[3].title).toBe(`${ACTION_NAMES.FIX_LOGIC} in Current Task`)
82-
expect((actions as any)[4].title).toBe(`${ACTION_NAMES.IMPROVE} in New Task`)
83-
expect((actions as any)[5].title).toBe(`${ACTION_NAMES.IMPROVE} in Current Task`)
84-
expect((actions as any)[6].title).toBe(ACTION_NAMES.ADD_TO_CONTEXT)
78+
expect((actions as any)[0].title).toBe(ACTION_NAMES.ADD_TO_CONTEXT)
79+
expect((actions as any)[1].title).toBe(`${ACTION_NAMES.EXPLAIN} in New Task`)
80+
expect((actions as any)[2].title).toBe(`${ACTION_NAMES.EXPLAIN} in Current Task`)
81+
expect((actions as any)[3].title).toBe(`${ACTION_NAMES.FIX_LOGIC} in New Task`)
82+
expect((actions as any)[4].title).toBe(`${ACTION_NAMES.FIX_LOGIC} in Current Task`)
83+
expect((actions as any)[5].title).toBe(`${ACTION_NAMES.IMPROVE} in New Task`)
84+
expect((actions as any)[6].title).toBe(`${ACTION_NAMES.IMPROVE} in Current Task`)
8585
})
8686

8787
it("should provide fix action instead of fix logic when diagnostics exist", () => {

0 commit comments

Comments
 (0)