separate into two files, add file_command capabilities
This commit is contained in:
485
main.go
485
main.go
@ -2,12 +2,17 @@ package main
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"encoding/base64"
|
||||
"encoding/json"
|
||||
"errors"
|
||||
"html/template"
|
||||
"io"
|
||||
"io/fs"
|
||||
"log"
|
||||
"net/http"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
@ -40,6 +45,8 @@ type ChatCompletionResponse struct {
|
||||
var (
|
||||
openaiAPIKey string
|
||||
indexTmpl *template.Template
|
||||
workDir string // Path to the workspace directory.
|
||||
absWorkDir string // Absolute workspace path.
|
||||
)
|
||||
|
||||
func init() {
|
||||
@ -47,16 +54,42 @@ func init() {
|
||||
if openaiAPIKey == "" {
|
||||
log.Fatal("Environment variable OPENAI_API_KEY is not set")
|
||||
}
|
||||
// Parse the HTML template with responsive layout updates.
|
||||
indexTmpl = template.Must(template.New("index").Parse(indexHTML))
|
||||
workDir = os.Getenv("WORK_DIR")
|
||||
if workDir == "" {
|
||||
workDir = "./workspace"
|
||||
}
|
||||
var err error
|
||||
absWorkDir, err = filepath.Abs(workDir)
|
||||
if err != nil {
|
||||
log.Fatalf("Error getting absolute path for workspace: %v", err)
|
||||
}
|
||||
if err := os.MkdirAll(absWorkDir, os.ModePerm); err != nil {
|
||||
log.Fatalf("Error creating workspace directory: %v", err)
|
||||
}
|
||||
|
||||
// Load the HTML template from an external file.
|
||||
tmplBytes, err := os.ReadFile("index.html")
|
||||
if err != nil {
|
||||
log.Fatalf("Error reading index.html: %v", err)
|
||||
}
|
||||
indexTmpl, err = template.New("index").Parse(string(tmplBytes))
|
||||
if err != nil {
|
||||
log.Fatalf("Error parsing index.html template: %v", err)
|
||||
}
|
||||
}
|
||||
|
||||
func main() {
|
||||
http.HandleFunc("/", indexHandler)
|
||||
http.HandleFunc("/chat", chatHandler)
|
||||
http.HandleFunc("/title", titleHandler)
|
||||
|
||||
// File system endpoints (sandboxed within absWorkDir)
|
||||
http.HandleFunc("/files", filesListHandler) // GET: list files.
|
||||
http.HandleFunc("/file", fileHandler) // GET for read; POST for write.
|
||||
|
||||
addr := ":8080"
|
||||
log.Println("Server starting on", addr)
|
||||
log.Println("Workspace directory:", absWorkDir)
|
||||
log.Fatal(http.ListenAndServe(addr, nil))
|
||||
}
|
||||
|
||||
@ -141,7 +174,7 @@ func titleHandler(w http.ResponseWriter, r *http.Request) {
|
||||
titleRequestMessages := []Message{
|
||||
{
|
||||
Role: "system",
|
||||
Content: "You are an AI that generates conversation titles. Based on the following conversation, please provide a short, descriptive title of no more than five words with no extra commentary.",
|
||||
Content: "You are an AI that generates conversation titles. Based on the conversation, please provide a short, descriptive title (five words maximum) with no extra commentary.",
|
||||
},
|
||||
}
|
||||
titleRequestMessages = append(titleRequestMessages, clientRequest.Messages...)
|
||||
@ -193,333 +226,119 @@ func titleHandler(w http.ResponseWriter, r *http.Request) {
|
||||
json.NewEncoder(w).Encode(responsePayload)
|
||||
}
|
||||
|
||||
const indexHTML = `
|
||||
<!DOCTYPE html>
|
||||
<html lang="en">
|
||||
<head>
|
||||
<meta charset="UTF-8">
|
||||
<title>ChatGPT Clone - Conversations</title>
|
||||
<meta name="viewport" content="width=device-width, initial-scale=1">
|
||||
<!-- Bootstrap 5 CSS and offcanvas component support -->
|
||||
<link href="https://cdn.jsdelivr.net/npm/bootstrap@5.3.0/dist/css/bootstrap.min.css" rel="stylesheet">
|
||||
<!-- Highlight.js CSS for code highlighting -->
|
||||
<link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/highlight.js/11.7.0/styles/default.min.css">
|
||||
<style>
|
||||
/* Use full viewport height */
|
||||
html, body {
|
||||
height: 100%;
|
||||
margin: 0;
|
||||
}
|
||||
/* Main container uses flex layout in row direction on md+ and column on small screens */
|
||||
.container-main {
|
||||
height: 100vh;
|
||||
display: flex;
|
||||
flex-direction: row;
|
||||
}
|
||||
/* Sidebar styles (hidden on small screens) */
|
||||
.sidebar {
|
||||
width: 25%;
|
||||
max-width: 300px;
|
||||
background-color: #fff;
|
||||
padding: 0.5rem;
|
||||
box-shadow: 0 0 10px rgba(0,0,0,0.1);
|
||||
overflow-y: auto;
|
||||
}
|
||||
/* On small devices, hide the sidebar and use offcanvas instead */
|
||||
@media (max-width: 767.98px) {
|
||||
.sidebar {
|
||||
display: none;
|
||||
}
|
||||
}
|
||||
/* Chat container fills remaining space and uses column layout */
|
||||
.chat-container {
|
||||
flex: 1;
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
background-color: #fff;
|
||||
padding: 1rem;
|
||||
box-shadow: 0 0 10px rgba(0,0,0,0.1);
|
||||
}
|
||||
.chat-header {
|
||||
flex-shrink: 0;
|
||||
}
|
||||
.chat-log {
|
||||
flex: 1;
|
||||
min-height: 0;
|
||||
overflow-y: auto;
|
||||
background-color: #eee;
|
||||
border: 1px solid #ddd;
|
||||
border-radius: 5px;
|
||||
padding: 0.5rem;
|
||||
margin: 0.5rem 0;
|
||||
}
|
||||
.chat-input {
|
||||
flex-shrink: 0;
|
||||
}
|
||||
.message {
|
||||
padding: 0.75rem;
|
||||
border-radius: 5px;
|
||||
margin-bottom: 0.5rem;
|
||||
white-space: pre-wrap;
|
||||
}
|
||||
.user { background-color: #d1e7dd; text-align: right; }
|
||||
.assistant { background-color: #f8d7da; text-align: left; }
|
||||
pre {
|
||||
max-width: 100%;
|
||||
box-sizing: border-box;
|
||||
white-space: pre-wrap;
|
||||
overflow-x: auto;
|
||||
background-color: #2d2d2d;
|
||||
color: #f8f8f2;
|
||||
padding: 0.5rem;
|
||||
border-radius: 5px;
|
||||
}
|
||||
.conversation-item {
|
||||
padding: 0.5rem;
|
||||
cursor: pointer;
|
||||
border-bottom: 1px solid #ddd;
|
||||
position: relative;
|
||||
}
|
||||
.conversation-item:hover,
|
||||
.conversation-item.active { background-color: #f0f0f0; }
|
||||
.delete-btn {
|
||||
position: absolute;
|
||||
right: 5px;
|
||||
top: 5px;
|
||||
background: transparent;
|
||||
border: none;
|
||||
color: #dc3545;
|
||||
font-size: 1rem;
|
||||
}
|
||||
textarea#messageInput {
|
||||
resize: none;
|
||||
height: 80px;
|
||||
overflow-y: auto;
|
||||
}
|
||||
</style>
|
||||
</head>
|
||||
<body>
|
||||
<!-- Offcanvas for mobile conversation list -->
|
||||
<div class="offcanvas offcanvas-start" tabindex="-1" id="offcanvasConversations" aria-labelledby="offcanvasConversationsLabel">
|
||||
<div class="offcanvas-header">
|
||||
<h5 class="offcanvas-title" id="offcanvasConversationsLabel">Conversations</h5>
|
||||
<button type="button" class="btn-close" data-bs-dismiss="offcanvas" aria-label="Close"></button>
|
||||
</div>
|
||||
<div class="offcanvas-body" id="offcanvasConversationList" style="overflow-y:auto;"></div>
|
||||
</div>
|
||||
|
||||
<div class="container-main">
|
||||
<!-- Sidebar for md+ screens -->
|
||||
<div class="sidebar d-none d-md-block" id="conversationListContainer">
|
||||
<h5>Conversations</h5>
|
||||
<button id="newConvoBtn" class="btn btn-sm btn-primary mb-2">New Conversation</button>
|
||||
<div id="conversationList"></div>
|
||||
</div>
|
||||
<!-- Chat container -->
|
||||
<div class="chat-container">
|
||||
<!-- On small screens, a button to open conversation offcanvas -->
|
||||
<div class="d-block d-md-none mb-2">
|
||||
<button class="btn btn-sm btn-secondary" type="button" data-bs-toggle="offcanvas" data-bs-target="#offcanvasConversations">
|
||||
Open Conversations
|
||||
</button>
|
||||
<button id="newConvoBtnMobile" class="btn btn-sm btn-primary ms-2">New Conversation</button>
|
||||
</div>
|
||||
<div class="chat-header">
|
||||
<h4 id="chatTitle">Conversation</h4>
|
||||
</div>
|
||||
<div id="chatLog" class="chat-log"></div>
|
||||
<div class="chat-input">
|
||||
<form id="chatForm">
|
||||
<div class="input-group">
|
||||
<textarea id="messageInput" class="form-control" placeholder="Type your message here" required></textarea>
|
||||
<button class="btn btn-primary" type="submit">Send</button>
|
||||
</div>
|
||||
</form>
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
|
||||
<!-- Include Bootstrap 5 JS bundle (includes Popper and offcanvas support) -->
|
||||
<script src="https://cdn.jsdelivr.net/npm/bootstrap@5.3.0/dist/js/bootstrap.bundle.min.js"></script>
|
||||
<!-- Marked & Highlight.js Libraries -->
|
||||
<script src="https://cdnjs.cloudflare.com/ajax/libs/marked/4.3.0/marked.min.js"></script>
|
||||
<script src="https://cdnjs.cloudflare.com/ajax/libs/highlight.js/11.7.0/highlight.min.js"></script>
|
||||
<script>
|
||||
marked.setOptions({
|
||||
highlight: function(code, lang) {
|
||||
if (lang && hljs.getLanguage(lang)) {
|
||||
return hljs.highlight(code, { language: lang }).value;
|
||||
}
|
||||
return hljs.highlightAuto(code).value;
|
||||
}
|
||||
});
|
||||
|
||||
// --- Conversation Management ---
|
||||
let conversations = [];
|
||||
let currentConversation = null;
|
||||
function generateId() {
|
||||
return 'c-' + Date.now() + '-' + Math.floor(Math.random() * 1000);
|
||||
}
|
||||
function loadConversations() {
|
||||
const loaded = localStorage.getItem('conversations');
|
||||
conversations = loaded ? JSON.parse(loaded) : [];
|
||||
renderConversationList();
|
||||
}
|
||||
function saveConversations() {
|
||||
localStorage.setItem('conversations', JSON.stringify(conversations));
|
||||
}
|
||||
function createNewConversation() {
|
||||
const newConvo = {
|
||||
id: generateId(),
|
||||
title: 'New Conversation',
|
||||
messages: [
|
||||
{ role: "system", content: "You are ChatGPT, a helpful assistant. When providing code, please always wrap the code in three backticks so that it renders correctly." }
|
||||
]
|
||||
};
|
||||
conversations.push(newConvo);
|
||||
currentConversation = newConvo;
|
||||
saveConversations();
|
||||
renderConversationList();
|
||||
renderChatLog();
|
||||
document.getElementById('chatTitle').textContent = "Conversation (" + newConvo.id + ")";
|
||||
}
|
||||
function deleteConversation(id) {
|
||||
if (!confirm("Are you sure you want to delete this conversation?")) return;
|
||||
conversations = conversations.filter(convo => convo.id !== id);
|
||||
if (currentConversation && currentConversation.id === id) {
|
||||
currentConversation = conversations.length > 0 ? conversations[0] : null;
|
||||
if (!currentConversation) {
|
||||
createNewConversation();
|
||||
} else {
|
||||
document.getElementById('chatTitle').textContent = "Conversation (" + currentConversation.id + ")";
|
||||
}
|
||||
}
|
||||
saveConversations();
|
||||
renderConversationList();
|
||||
renderChatLog();
|
||||
}
|
||||
function renderConversationList() {
|
||||
// Render for sidebar (md+)
|
||||
const listElem = document.getElementById('conversationList');
|
||||
listElem.innerHTML = "";
|
||||
// Also render for offcanvas (mobile)
|
||||
const offcanvasElem = document.getElementById('offcanvasConversationList');
|
||||
offcanvasElem.innerHTML = "";
|
||||
conversations.forEach(convo => {
|
||||
const createConvoItem = (container) => {
|
||||
const div = document.createElement("div");
|
||||
div.className = "conversation-item" + (currentConversation && currentConversation.id === convo.id ? " active" : "");
|
||||
div.textContent = convo.title || convo.id;
|
||||
const deleteBtn = document.createElement("button");
|
||||
deleteBtn.className = "delete-btn";
|
||||
deleteBtn.innerHTML = "×";
|
||||
deleteBtn.onclick = function(e) { e.stopPropagation(); deleteConversation(convo.id); };
|
||||
div.appendChild(deleteBtn);
|
||||
div.onclick = () => {
|
||||
currentConversation = convo;
|
||||
document.getElementById('chatTitle').textContent = "Conversation (" + convo.id + ")";
|
||||
renderConversationList();
|
||||
renderChatLog();
|
||||
// Hide offcanvas after selection on mobile
|
||||
var offcanvas = bootstrap.Offcanvas.getInstance(document.getElementById('offcanvasConversations'));
|
||||
if(offcanvas) offcanvas.hide();
|
||||
};
|
||||
container.appendChild(div);
|
||||
};
|
||||
createConvoItem(listElem);
|
||||
createConvoItem(offcanvasElem);
|
||||
});
|
||||
}
|
||||
function renderChatLog() {
|
||||
const chatLog = document.getElementById('chatLog');
|
||||
chatLog.innerHTML = "";
|
||||
if (!currentConversation) return;
|
||||
currentConversation.messages.forEach(msg => {
|
||||
appendMessage(msg.role, msg.content, false);
|
||||
});
|
||||
chatLog.scrollTop = chatLog.scrollHeight;
|
||||
}
|
||||
function appendMessage(role, content, update = true) {
|
||||
const chatLog = document.getElementById('chatLog');
|
||||
const messageElem = document.createElement("div");
|
||||
messageElem.className = "message " + role;
|
||||
messageElem.innerHTML = marked.parse(content);
|
||||
chatLog.appendChild(messageElem);
|
||||
chatLog.scrollTop = chatLog.scrollHeight;
|
||||
if (update && currentConversation) {
|
||||
currentConversation.messages.push({ role: role, content: content });
|
||||
if (role === "assistant" && currentConversation.title === "New Conversation") {
|
||||
autoTitleConversation();
|
||||
}
|
||||
saveConversations();
|
||||
}
|
||||
return messageElem;
|
||||
}
|
||||
async function autoTitleConversation() {
|
||||
try {
|
||||
const response = await fetch("/title", {
|
||||
method: "POST",
|
||||
headers: { "Content-Type": "application/json" },
|
||||
body: JSON.stringify({ messages: currentConversation.messages })
|
||||
});
|
||||
if (!response.ok) throw new Error("Title API error: " + response.status);
|
||||
const data = await response.json();
|
||||
const newTitle = data.title.trim();
|
||||
if(newTitle) {
|
||||
currentConversation.title = newTitle;
|
||||
document.getElementById('chatTitle').textContent = "Conversation (" + currentConversation.id + "): " + newTitle;
|
||||
renderConversationList();
|
||||
saveConversations();
|
||||
}
|
||||
} catch (error) {
|
||||
console.error("Error auto-titling conversation:", error);
|
||||
}
|
||||
}
|
||||
const chatForm = document.getElementById('chatForm');
|
||||
const messageInput = document.getElementById('messageInput');
|
||||
messageInput.addEventListener('keydown', function(e) {
|
||||
if (e.key === "Enter" && !e.shiftKey) {
|
||||
e.preventDefault();
|
||||
chatForm.requestSubmit();
|
||||
}
|
||||
});
|
||||
chatForm.addEventListener('submit', async function(e) {
|
||||
e.preventDefault();
|
||||
const userMessage = messageInput.value.trim();
|
||||
if (!userMessage || !currentConversation) return;
|
||||
appendMessage("user", userMessage);
|
||||
messageInput.value = "";
|
||||
const typingIndicator = appendMessage("assistant", "Typing...", false);
|
||||
try {
|
||||
const response = await fetch("/chat", {
|
||||
method: "POST",
|
||||
headers: { "Content-Type": "application/json" },
|
||||
body: JSON.stringify({ messages: currentConversation.messages })
|
||||
});
|
||||
if (!response.ok) throw new Error("Server error: " + response.status);
|
||||
const data = await response.json();
|
||||
typingIndicator.remove();
|
||||
appendMessage("assistant", data.reply);
|
||||
} catch (error) {
|
||||
typingIndicator.remove();
|
||||
appendMessage("assistant", "Error: " + error.message);
|
||||
}
|
||||
});
|
||||
document.getElementById('newConvoBtn').addEventListener("click", function() {
|
||||
createNewConversation();
|
||||
});
|
||||
document.getElementById('newConvoBtnMobile').addEventListener("click", function() {
|
||||
createNewConversation();
|
||||
});
|
||||
loadConversations();
|
||||
if(conversations.length === 0) {
|
||||
createNewConversation();
|
||||
} else {
|
||||
currentConversation = conversations[0];
|
||||
document.getElementById('chatTitle').textContent = "Conversation (" + currentConversation.id + ")";
|
||||
renderChatLog();
|
||||
}
|
||||
</script>
|
||||
</body>
|
||||
</html>
|
||||
`
|
||||
//
|
||||
// File system endpoints
|
||||
//
|
||||
|
||||
// filesListHandler lists all files (recursively) in the workspace.
|
||||
func filesListHandler(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodGet {
|
||||
http.Error(w, "Only GET allowed", http.StatusMethodNotAllowed)
|
||||
return
|
||||
}
|
||||
var files []string
|
||||
err := filepath.WalkDir(absWorkDir, func(path string, d fs.DirEntry, err error) error {
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
if !d.IsDir() {
|
||||
rel, err := filepath.Rel(absWorkDir, path)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
files = append(files, rel)
|
||||
}
|
||||
return nil
|
||||
})
|
||||
if err != nil {
|
||||
http.Error(w, "Error listing files: "+err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(map[string]interface{}{"files": files})
|
||||
}
|
||||
|
||||
// fileHandler supports file read (GET) and write (POST) operations.
|
||||
func fileHandler(w http.ResponseWriter, r *http.Request) {
|
||||
switch r.Method {
|
||||
case http.MethodGet:
|
||||
handleFileRead(w, r)
|
||||
case http.MethodPost:
|
||||
handleFileWrite(w, r)
|
||||
default:
|
||||
http.Error(w, "Only GET and POST allowed", http.StatusMethodNotAllowed)
|
||||
}
|
||||
}
|
||||
|
||||
func handleFileRead(w http.ResponseWriter, r *http.Request) {
|
||||
filename := r.URL.Query().Get("filename")
|
||||
if filename == "" {
|
||||
http.Error(w, "Missing filename", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
fullPath, err := secureFilePath(filename)
|
||||
if err != nil {
|
||||
http.Error(w, "Invalid filename: "+err.Error(), http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
data, err := os.ReadFile(fullPath)
|
||||
if err != nil {
|
||||
http.Error(w, "Error reading file: "+err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
encoded := base64.StdEncoding.EncodeToString(data)
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(map[string]string{"content_b64": encoded})
|
||||
}
|
||||
|
||||
func handleFileWrite(w http.ResponseWriter, r *http.Request) {
|
||||
var payload struct {
|
||||
Filename string `json:"filename"`
|
||||
ContentB64 string `json:"content_b64"`
|
||||
}
|
||||
if err := json.NewDecoder(r.Body).Decode(&payload); err != nil {
|
||||
http.Error(w, "Invalid JSON: "+err.Error(), http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
if payload.Filename == "" || payload.ContentB64 == "" {
|
||||
http.Error(w, "Filename and content_b64 are required", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
decoded, err := base64.StdEncoding.DecodeString(payload.ContentB64)
|
||||
if err != nil {
|
||||
http.Error(w, "Error decoding Base64 content: "+err.Error(), http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
fullPath, err := secureFilePath(payload.Filename)
|
||||
if err != nil {
|
||||
http.Error(w, "Invalid filename: "+err.Error(), http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
dir := filepath.Dir(fullPath)
|
||||
if err := os.MkdirAll(dir, os.ModePerm); err != nil {
|
||||
http.Error(w, "Error creating directory: "+err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
if err := os.WriteFile(fullPath, decoded, 0644); err != nil {
|
||||
http.Error(w, "Error writing file: "+err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
json.NewEncoder(w).Encode(map[string]string{"status": "success"})
|
||||
}
|
||||
|
||||
func secureFilePath(relPath string) (string, error) {
|
||||
relPath = filepath.ToSlash(relPath)
|
||||
if strings.Contains(relPath, "..") {
|
||||
return "", errors.New("relative paths outside the workspace are not allowed")
|
||||
}
|
||||
fullPath := filepath.Join(absWorkDir, relPath)
|
||||
absFullPath, err := filepath.Abs(fullPath)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
if !strings.HasPrefix(absFullPath, absWorkDir) {
|
||||
return "", errors.New("attempt to access files outside the workspace")
|
||||
}
|
||||
return absFullPath, nil
|
||||
}
|
||||
|
Reference in New Issue
Block a user