1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
|
// This method is called when your extension is deactivated
export function deactivate() {}
import * as vscode from "vscode"
const TERMINAL_NAME = "opencode"
export function activate(context: vscode.ExtensionContext) {
const openNewTerminalDisposable = vscode.commands.registerCommand("opencode.openNewTerminal", async () => {
await openTerminal()
})
const openTerminalDisposable = vscode.commands.registerCommand("opencode.openTerminal", async () => {
// An opencode terminal already exists => focus it
const existingTerminal = vscode.window.terminals.find((t) => t.name === TERMINAL_NAME)
if (existingTerminal) {
existingTerminal.show()
return
}
await openTerminal()
})
let addFilepathDisposable = vscode.commands.registerCommand("opencode.addFilepathToTerminal", async () => {
const fileRef = getActiveFile()
if (!fileRef) {
return
}
const terminal = vscode.window.activeTerminal
if (!terminal) {
return
}
if (terminal.name === TERMINAL_NAME) {
// @ts-ignore
const port = terminal.creationOptions.env?.["_EXTENSION_OPENCODE_PORT"]
port ? await appendPrompt(parseInt(port), fileRef) : terminal.sendText(fileRef, false)
terminal.show()
}
})
context.subscriptions.push(openNewTerminalDisposable, openTerminalDisposable, addFilepathDisposable)
async function openTerminal() {
// Create a new terminal in split screen
const port = Math.floor(Math.random() * (65535 - 16384 + 1)) + 16384
const terminal = vscode.window.createTerminal({
name: TERMINAL_NAME,
iconPath: {
light: vscode.Uri.file(context.asAbsolutePath("images/button-dark.svg")),
dark: vscode.Uri.file(context.asAbsolutePath("images/button-light.svg")),
},
location: {
viewColumn: vscode.ViewColumn.Beside,
preserveFocus: false,
},
env: {
_EXTENSION_OPENCODE_PORT: port.toString(),
OPENCODE_CALLER: "vscode",
},
})
terminal.show()
terminal.sendText(`opencode --port ${port}`)
const fileRef = getActiveFile()
if (!fileRef) {
return
}
// Wait for the terminal to be ready
let tries = 10
let connected = false
do {
await new Promise((resolve) => setTimeout(resolve, 200))
try {
await fetch(`http://localhost:${port}/app`)
connected = true
break
} catch {}
tries--
} while (tries > 0)
// If connected, append the prompt to the terminal
if (connected) {
await appendPrompt(port, `In ${fileRef}`)
terminal.show()
}
}
async function appendPrompt(port: number, text: string) {
await fetch(`http://localhost:${port}/tui/append-prompt`, {
method: "POST",
headers: {
"Content-Type": "application/json",
},
body: JSON.stringify({ text }),
})
}
function getActiveFile() {
const activeEditor = vscode.window.activeTextEditor
if (!activeEditor) {
return
}
const document = activeEditor.document
const workspaceFolder = vscode.workspace.getWorkspaceFolder(document.uri)
if (!workspaceFolder) {
return
}
// Get the relative path from workspace root
const relativePath = vscode.workspace.asRelativePath(document.uri)
let filepathWithAt = `@${relativePath}`
// Check if there's a selection and add line numbers
const selection = activeEditor.selection
if (!selection.isEmpty) {
// Convert to 1-based line numbers
const startLine = selection.start.line + 1
const endLine = selection.end.line + 1
if (startLine === endLine) {
// Single line selection
filepathWithAt += `#L${startLine}`
} else {
// Multi-line selection
filepathWithAt += `#L${startLine}-${endLine}`
}
}
return filepathWithAt
}
}
|
