mirror of
https://github.com/Iheuzio/gpt-contextfiles.git
synced 2025-07-18 14:00:48 +00:00
572 lines
19 KiB
JavaScript
572 lines
19 KiB
JavaScript
const vscode = require('vscode');
|
|
const { Configuration, OpenAIApi } = require("openai");
|
|
|
|
// move these into the script so that instead of echoing the question and the contents,
|
|
// it will echo the question, followed by the answer from the response when the submit button is pressed.
|
|
const configuration = new Configuration({
|
|
apiKey: process.env.OPENAI_API_KEY,
|
|
});
|
|
|
|
const openai = new OpenAIApi(configuration);
|
|
|
|
// Represents a file item in the file explorer
|
|
class FileItem {
|
|
constructor(uri, selected = false) {
|
|
this.uri = uri;
|
|
this.selected = selected;
|
|
}
|
|
|
|
toggleSelected() {
|
|
this.selected = !this.selected;
|
|
}
|
|
}
|
|
|
|
// Represents the selected files in the file explorer
|
|
const selectedFiles = [];
|
|
|
|
// Tree data provider for the selected files
|
|
class FileDataProvider {
|
|
constructor() {
|
|
this._onDidChangeTreeData = new vscode.EventEmitter();
|
|
this.onDidChangeTreeData = this._onDidChangeTreeData.event;
|
|
}
|
|
|
|
refresh() {
|
|
this._onDidChangeTreeData.fire();
|
|
}
|
|
|
|
getTreeItem(element) {
|
|
return {
|
|
label: element.uri.fsPath,
|
|
collapsibleState: vscode.TreeItemCollapsibleState.None
|
|
};
|
|
}
|
|
|
|
getChildren(element) {
|
|
if (element) {
|
|
return [];
|
|
}
|
|
|
|
// Return only the selected files
|
|
return selectedFiles.filter(file => file.selected);
|
|
}
|
|
}
|
|
|
|
// Command for adding files to gpt-contextfiles
|
|
const addFilesCommand = vscode.commands.registerCommand('extension.addFilesToGPTContext', () => {
|
|
const editor = vscode.window.activeTextEditor;
|
|
if (editor) {
|
|
const uri = editor.document.uri;
|
|
const existingFileIndex = selectedFiles.findIndex(file => file.uri.fsPath === uri.fsPath);
|
|
|
|
if (existingFileIndex !== -1) {
|
|
// File already exists, remove it from the list
|
|
selectedFiles.splice(existingFileIndex, 1);
|
|
} else {
|
|
// Add the file to the list with selected state
|
|
selectedFiles.push(new FileItem(uri, true));
|
|
}
|
|
|
|
fileDataProvider.refresh();
|
|
}
|
|
});
|
|
|
|
const fileDataProvider = new FileDataProvider();
|
|
|
|
// Function to handle question submission
|
|
async function handleQuestionSubmission(panel, question, selectedUris) {
|
|
// Update the selectedFiles array based on the selectedUris
|
|
selectedFiles.forEach(file => {
|
|
file.selected = selectedUris.includes(file.uri.fsPath);
|
|
});
|
|
|
|
fileDataProvider.refresh();
|
|
|
|
const fileContents = selectedFiles
|
|
.filter(file => file.selected)
|
|
.map(file => {
|
|
const document = vscode.workspace.textDocuments.find(doc => doc.uri.fsPath === file.uri.fsPath);
|
|
if (document) {
|
|
const lines = document.getText().split('\n');
|
|
const formattedLines = lines.map(line => `\t${line}`).join('\n');
|
|
return `${file.uri.fsPath}:\n\`\`\`\n${formattedLines}\n\`\`\``;
|
|
}
|
|
return '';
|
|
})
|
|
.join('\n\n');
|
|
|
|
// Call OpenAI API with the question and file contents
|
|
try {
|
|
const chatCompletion = await openai.createChatCompletion({
|
|
model: "gpt-3.5-turbo-16k",
|
|
messages: [
|
|
{ role: "system", content: "Answer the coding questions, only provide the code and documentation, explaining the solution after providing the code." },
|
|
{ role: "user", content: question + "\n" + fileContents},
|
|
],
|
|
});
|
|
|
|
// Extract the answer from the OpenAI response
|
|
const answer = chatCompletion.data.choices[0].message.content;
|
|
|
|
// Update the webview content to display only the OpenAI response
|
|
panel.webview.html = getWebviewContent(answer, question);
|
|
} catch (error) {
|
|
// Handle any errors from the OpenAI API
|
|
console.error("Failed to get OpenAI response:", error);
|
|
panel.webview.html = getWebviewContent(`Failed to get response from OpenAI API. Error: ${error.message}`, question);
|
|
}
|
|
}
|
|
|
|
// Command for displaying the webview panel
|
|
const openGPTContextPanelCommand = vscode.commands.registerCommand('extension.openGPTContextPanel', () => {
|
|
const panel = vscode.window.createWebviewPanel(
|
|
'gptContextPanel',
|
|
'GPT Context',
|
|
vscode.ViewColumn.One,
|
|
{
|
|
enableScripts: true
|
|
}
|
|
);
|
|
|
|
panel.webview.html = getWebviewContent();
|
|
|
|
panel.webview.onDidReceiveMessage(async message => {
|
|
if (message.command === 'submitQuestion') {
|
|
await handleQuestionSubmission(panel, message.text, message.selectedUris);
|
|
} else if (message.command === 'toggleFileSelection') {
|
|
const uri = message.uri;
|
|
const file = selectedFiles.find(file => file.uri.fsPath === uri);
|
|
if (file) {
|
|
file.toggleSelected();
|
|
fileDataProvider.refresh();
|
|
}
|
|
} else if (message.command === 'clearSelectedFiles') {
|
|
const clearedFiles = selectedFiles.filter(file => file.selected === false);
|
|
selectedFiles.length = 0; // Clear the array
|
|
clearedFiles.forEach(file => {
|
|
fileDataProvider.refresh();
|
|
});
|
|
panel.webview.html = getWebviewContent();
|
|
} else if (message.command === 'refreshFiles') {
|
|
fileDataProvider.refresh();
|
|
panel.webview.html = getWebviewContent();
|
|
}
|
|
});
|
|
});
|
|
|
|
|
|
// Command for refreshing the selected files
|
|
const refreshSelectedFilesCommand = vscode.commands.registerCommand('extension.refreshSelectedFiles', () => {
|
|
fileDataProvider.refresh();
|
|
});
|
|
|
|
// Command for clearing the selected files
|
|
const clearSelectedFilesCommand = vscode.commands.registerCommand('extension.clearSelectedFiles', () => {
|
|
selectedFiles.forEach(file => {
|
|
file.selected = false;
|
|
});
|
|
fileDataProvider.refresh();
|
|
});
|
|
|
|
// Command for refreshing all files
|
|
const refreshFilesCommand = vscode.commands.registerCommand('extension.refreshFiles', () => {
|
|
fileDataProvider.refresh();
|
|
});
|
|
|
|
// Helper function to generate the HTML content for the webview panel
|
|
function getWebviewContent(apiResponse = '', question = '') {
|
|
const fileList = selectedFiles
|
|
.map(
|
|
file =>
|
|
`<div><input type="checkbox" data-uri="${file.uri.fsPath}" ${file.selected ? 'checked' : ''
|
|
} onchange="toggleFileSelection('${file.uri.fsPath}')" /> ${file.uri.fsPath}</div>`
|
|
)
|
|
.join('');
|
|
|
|
return `
|
|
<html>
|
|
<head>
|
|
<style>
|
|
.panel {
|
|
display: flex;
|
|
flex-direction: column;
|
|
align-items: center;
|
|
justify-content: center;
|
|
width: 50%;
|
|
height: 100%;
|
|
margin: 0 auto;
|
|
background-color: #1e1e1e;
|
|
color: #d4d4d4;
|
|
font-family: 'Segoe UI', Tahoma, Geneva, Verdana, sans-serif;
|
|
}
|
|
|
|
.textbox {
|
|
width: 100%;
|
|
height: 200px;
|
|
resize: both;
|
|
padding: 10px;
|
|
}
|
|
|
|
.buttons {
|
|
display: flex;
|
|
justify-content: space-between;
|
|
width: 50%;
|
|
}
|
|
|
|
.button {
|
|
flex-grow: 1;
|
|
background-color: #007acc;
|
|
color: #fff;
|
|
padding: 10px;
|
|
text-align: center;
|
|
cursor: pointer;
|
|
border: none;
|
|
outline: none;
|
|
font-size: 14px;
|
|
}
|
|
|
|
#response {
|
|
border: 1px solid #000;
|
|
background-color: #e0e0e0;
|
|
padding: 10px;
|
|
margin-top: 20px;
|
|
overflow-wrap: break-word;
|
|
}
|
|
|
|
#file-list {
|
|
margin-top: 20px;
|
|
border
|
|
}
|
|
|
|
.form-group {
|
|
display: flex;
|
|
flex-direction: column;
|
|
margin-bottom: 20px;
|
|
width: 100%;
|
|
}
|
|
|
|
.form-group label {
|
|
margin-bottom: 5px;
|
|
font-size: 14px;
|
|
}
|
|
|
|
.form-group input[type="text"] {
|
|
padding: 10px;
|
|
font-size: 14px;
|
|
border: none;
|
|
outline: none;
|
|
background-color: #2d2d2d;
|
|
color: #d4d4d4;
|
|
}
|
|
|
|
.form-group input[type="text"]::placeholder {
|
|
color: #d4d4d4;
|
|
}
|
|
|
|
.form-group .button-options {
|
|
display: flex;
|
|
justify-content: space-between;
|
|
margin-top: 10px;
|
|
}
|
|
|
|
.form-group .button-options button {
|
|
padding: 10px;
|
|
font-size: 14px;
|
|
border: none;
|
|
outline: none;
|
|
cursor: pointer;
|
|
background-color: #007acc;
|
|
color: #fff;
|
|
}
|
|
|
|
.form-group .button-options button:hover {
|
|
background-color: #005f8c;
|
|
}
|
|
|
|
.form-group .button-options button:active {
|
|
background-color: #004d73;
|
|
}
|
|
|
|
.form-group .button-options button:focus {
|
|
box-shadow: 0 0 0 2px rgba(0, 122, 204, 0.5);
|
|
}
|
|
|
|
.file-list {
|
|
margin-top: 20px;
|
|
width: 100%;
|
|
}
|
|
|
|
.file-list h2 {
|
|
margin-bottom: 10px;
|
|
font-size: 14px;
|
|
}
|
|
|
|
.file-list .file-item {
|
|
display: flex;
|
|
align-items: center;
|
|
margin-bottom: 5px;
|
|
font-size: 14px;
|
|
}
|
|
|
|
.file-list .file-item input[type="checkbox"] {
|
|
margin-right: 5px;
|
|
}
|
|
|
|
.file-list .file-item label {
|
|
margin-bottom: 0;
|
|
}
|
|
|
|
.file-list .file-item .file-path {
|
|
overflow-wrap: break-word;
|
|
}
|
|
|
|
.file-list .file-item .file-path:hover {
|
|
text-decoration: underline;
|
|
cursor: pointer;
|
|
}
|
|
|
|
.file-list .file-item .file-path:active {
|
|
color: #007acc;
|
|
}
|
|
|
|
.file-list .file-item .file-path:focus {
|
|
box-shadow: 0 0 0 2px rgba(0, 122, 204, 0.5);
|
|
}
|
|
|
|
.collapsible {
|
|
background-color: #2d2d2d;
|
|
color: #d4d4d4;
|
|
cursor: pointer;
|
|
padding: 10px;
|
|
width: 100%;
|
|
border: none;
|
|
outline: none;
|
|
text-align: left;
|
|
font-size: 14px;
|
|
}
|
|
|
|
.collapsible:hover {
|
|
background-color: #3c3c3c;
|
|
}
|
|
|
|
.collapsible:active {
|
|
background-color: #4c4c4c;
|
|
}
|
|
|
|
.collapsible:focus {
|
|
box-shadow: 0 0 0 2px rgba(0, 122, 204, 0.5);
|
|
}
|
|
|
|
.content {
|
|
padding: 0 10px;
|
|
display: none;
|
|
overflow: hidden;
|
|
background-color: #f1f1f1;
|
|
width:100%;
|
|
}
|
|
|
|
.content p {
|
|
margin-top: 0;
|
|
font-size: 14px;
|
|
}
|
|
|
|
.active, .collapsible:hover {
|
|
background-color: #555;
|
|
}
|
|
|
|
.active:after {
|
|
content: "\\2212";
|
|
}
|
|
|
|
.collapsible:after {
|
|
content: "\\002B";
|
|
color: #d4d4d4;
|
|
font-weight: bold;
|
|
float: right;
|
|
margin-left: 5px;
|
|
}
|
|
|
|
.active:after {
|
|
content: "\\2212";
|
|
}
|
|
|
|
.collapsible:after {
|
|
content: "\\002B";
|
|
color: #d4d4d4;
|
|
font-weight: bold;
|
|
float: right;
|
|
margin-left: 5px;
|
|
}
|
|
|
|
#rendered {
|
|
background-color: #2d2d2d;
|
|
word-wrap: wrap;
|
|
margin-top: 20px;
|
|
border: 1px solid white;
|
|
border-radius: 5px;
|
|
padding: 10px;
|
|
}
|
|
|
|
#question-rep {
|
|
font-weight: bold;
|
|
background-color: #2d2d2d;
|
|
word-wrap: wrap;
|
|
border: 1px solid white;
|
|
border-radius: 5px;
|
|
}
|
|
|
|
div#api-response.content.active {
|
|
background-color: #313131;
|
|
}
|
|
|
|
</style>
|
|
</head>
|
|
<body class="panel">
|
|
<h1>GPT Context</h1>
|
|
<form id="questionForm">
|
|
<div class="form-group">
|
|
<label for="question">Enter your question:</label>
|
|
<input type="text" id="question" name="question" placeholder="Enter your question" required>
|
|
<div class="button-options">
|
|
<button type="submit">Submit</button>
|
|
<button type="button" onclick="clearSelectedFiles()">Clear</button>
|
|
<button type="button" onclick="refreshSelectedFiles()">Refresh</button>
|
|
</div>
|
|
</div>
|
|
<div class="form-group">
|
|
<div class="collapsible" onclick="toggleApiResponse()">
|
|
API Response
|
|
</div>
|
|
<div class="content" id="api-response">
|
|
<div id="question-rep">
|
|
<p>${question ? '> ' + question : null}</p>
|
|
</div>
|
|
${
|
|
apiResponse ? `
|
|
<div id="rendered">
|
|
<p id="responses">${apiResponse}</p>
|
|
</div>
|
|
` : null
|
|
}
|
|
</div>
|
|
</div>
|
|
<div class="form-group">
|
|
<label for="selected-files">Selected Files:</label>
|
|
<div class="file-list">
|
|
${fileList ? fileList : '<p class="no-files">No files selected</p>'}
|
|
</div>
|
|
</div>
|
|
<script>
|
|
const vscode = acquireVsCodeApi();
|
|
|
|
function toggleFileSelection(uri) {
|
|
vscode.postMessage({
|
|
command: 'toggleFileSelection',
|
|
uri: uri
|
|
});
|
|
}
|
|
|
|
function clearSelectedFiles() {
|
|
vscode.postMessage({
|
|
command: 'clearSelectedFiles'
|
|
});
|
|
}
|
|
|
|
function refreshSelectedFiles() {
|
|
vscode.postMessage({
|
|
command: 'refreshFiles'
|
|
});
|
|
}
|
|
|
|
function toggleApiResponse() {
|
|
const apiResponse = document.getElementById('api-response');
|
|
var response = document.getElementById('responses');
|
|
if(response === null || response === 'null') {
|
|
return;
|
|
}
|
|
apiResponse.classList.toggle('active');
|
|
const content = apiResponse;
|
|
const collapsible = apiResponse.previousElementSibling;
|
|
if (content.style.display === 'block') {
|
|
content.style.display = 'none';
|
|
collapsible.classList.remove('active');
|
|
} else {
|
|
content.style.display = 'block';
|
|
collapsible.classList.add('active');
|
|
}
|
|
}
|
|
|
|
|
|
const form = document.getElementById('questionForm');
|
|
form.addEventListener('submit', event => {
|
|
event.preventDefault();
|
|
const question = document.getElementById('question').value;
|
|
const checkboxes = document.querySelectorAll('input[type="checkbox"]');
|
|
const selectedUris = [];
|
|
checkboxes.forEach(checkbox => {
|
|
if (checkbox.checked) {
|
|
const uri = checkbox.getAttribute('data-uri');
|
|
selectedUris.push(uri);
|
|
}
|
|
});
|
|
vscode.postMessage({
|
|
command: 'submitQuestion',
|
|
text: question,
|
|
selectedUris: selectedUris
|
|
});
|
|
});
|
|
</script>
|
|
</body>
|
|
</html>
|
|
`;
|
|
}
|
|
|
|
|
|
|
|
// Activates the extension
|
|
function activate(context) {
|
|
context.subscriptions.push(addFilesCommand);
|
|
context.subscriptions.push(openGPTContextPanelCommand);
|
|
context.subscriptions.push(refreshSelectedFilesCommand);
|
|
context.subscriptions.push(clearSelectedFilesCommand);
|
|
context.subscriptions.push(refreshFilesCommand);
|
|
vscode.window.registerTreeDataProvider('selectedFiles', fileDataProvider);
|
|
|
|
const provider = {
|
|
resolveWebviewView(webviewView) {
|
|
webviewView.webview.options = {
|
|
enableScripts: true
|
|
};
|
|
webviewView.webview.html = getWebviewContent();
|
|
webviewView.webview.onDidReceiveMessage(async message => {
|
|
if (message.command === 'toggleFileSelection') {
|
|
const uri = message.uri;
|
|
const file = selectedFiles.find(file => file.uri.fsPath === uri);
|
|
if (file) {
|
|
file.toggleSelected();
|
|
fileDataProvider.refresh();
|
|
}
|
|
} else if (message.command === 'clearSelectedFiles') {
|
|
const clearedFiles = selectedFiles.filter(file => file.selected === false);
|
|
selectedFiles.length = 0; // Clear the array
|
|
clearedFiles.forEach(file => {
|
|
fileDataProvider.refresh();
|
|
});
|
|
webviewView.webview.html = getWebviewContent();
|
|
} else if (message.command === 'refreshFiles') {
|
|
fileDataProvider.refresh();
|
|
webviewView.webview.html = getWebviewContent();
|
|
} else if (message.command === 'submitQuestion') {
|
|
await handleQuestionSubmission(webviewView, message.text, message.selectedUris);
|
|
}
|
|
});
|
|
}
|
|
};
|
|
|
|
context.subscriptions.push(vscode.window.registerWebviewViewProvider('gpt-context-sidebar', provider));
|
|
|
|
}
|
|
|
|
|
|
exports.activate = activate;
|