123 lines
No EOL
4.5 KiB
JavaScript
123 lines
No EOL
4.5 KiB
JavaScript
"use strict";
|
|
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
|
|
if (k2 === undefined) k2 = k;
|
|
var desc = Object.getOwnPropertyDescriptor(m, k);
|
|
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
|
|
desc = { enumerable: true, get: function() { return m[k]; } };
|
|
}
|
|
Object.defineProperty(o, k2, desc);
|
|
}) : (function(o, m, k, k2) {
|
|
if (k2 === undefined) k2 = k;
|
|
o[k2] = m[k];
|
|
}));
|
|
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
|
|
Object.defineProperty(o, "default", { enumerable: true, value: v });
|
|
}) : function(o, v) {
|
|
o["default"] = v;
|
|
});
|
|
var __importStar = (this && this.__importStar) || (function () {
|
|
var ownKeys = function(o) {
|
|
ownKeys = Object.getOwnPropertyNames || function (o) {
|
|
var ar = [];
|
|
for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
|
|
return ar;
|
|
};
|
|
return ownKeys(o);
|
|
};
|
|
return function (mod) {
|
|
if (mod && mod.__esModule) return mod;
|
|
var result = {};
|
|
if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
|
|
__setModuleDefault(result, mod);
|
|
return result;
|
|
};
|
|
})();
|
|
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
};
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
exports.DisclawDatabase = void 0;
|
|
const better_sqlite3_1 = __importDefault(require("better-sqlite3"));
|
|
const fs = __importStar(require("fs"));
|
|
const path = __importStar(require("path"));
|
|
const SCHEMA_SQL = `
|
|
CREATE TABLE IF NOT EXISTS workspaces (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
channel_id TEXT NOT NULL UNIQUE,
|
|
guild_id TEXT NOT NULL,
|
|
agent_name TEXT NOT NULL UNIQUE,
|
|
workspace_path TEXT NOT NULL,
|
|
created_at TEXT NOT NULL DEFAULT (datetime('now'))
|
|
);
|
|
|
|
CREATE TABLE IF NOT EXISTS conversations (
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
|
workspace_id INTEGER NOT NULL REFERENCES workspaces(id),
|
|
discord_msg_id TEXT NOT NULL UNIQUE,
|
|
role TEXT NOT NULL,
|
|
content TEXT NOT NULL,
|
|
author_name TEXT,
|
|
created_at TEXT NOT NULL DEFAULT (datetime('now'))
|
|
);
|
|
|
|
CREATE INDEX IF NOT EXISTS idx_conversations_workspace
|
|
ON conversations(workspace_id, created_at);
|
|
`;
|
|
class DisclawDatabase {
|
|
db;
|
|
constructor(dbPath) {
|
|
const dir = path.dirname(dbPath);
|
|
if (!fs.existsSync(dir)) {
|
|
fs.mkdirSync(dir, { recursive: true });
|
|
}
|
|
this.db = new better_sqlite3_1.default(dbPath);
|
|
this.db.pragma("journal_mode = WAL");
|
|
this.db.pragma("foreign_keys = ON");
|
|
this.runMigrations();
|
|
}
|
|
runMigrations() {
|
|
this.db.exec(SCHEMA_SQL);
|
|
}
|
|
// --- Workspace queries ---
|
|
getWorkspaceByChannelId(channelId) {
|
|
return this.db
|
|
.prepare("SELECT * FROM workspaces WHERE channel_id = ?")
|
|
.get(channelId);
|
|
}
|
|
getWorkspaceByName(name) {
|
|
return this.db
|
|
.prepare("SELECT * FROM workspaces WHERE agent_name = ?")
|
|
.get(name);
|
|
}
|
|
getAllWorkspaces() {
|
|
return this.db
|
|
.prepare("SELECT * FROM workspaces ORDER BY created_at")
|
|
.all();
|
|
}
|
|
createWorkspace(channelId, guildId, agentName, workspacePath) {
|
|
const stmt = this.db.prepare("INSERT INTO workspaces (channel_id, guild_id, agent_name, workspace_path) VALUES (?, ?, ?, ?)");
|
|
const result = stmt.run(channelId, guildId, agentName, workspacePath);
|
|
return this.db
|
|
.prepare("SELECT * FROM workspaces WHERE id = ?")
|
|
.get(result.lastInsertRowid);
|
|
}
|
|
// --- Conversation queries ---
|
|
addConversation(workspaceId, discordMsgId, role, content, authorName) {
|
|
this.db
|
|
.prepare("INSERT OR IGNORE INTO conversations (workspace_id, discord_msg_id, role, content, author_name) VALUES (?, ?, ?, ?, ?)")
|
|
.run(workspaceId, discordMsgId, role, content, authorName);
|
|
}
|
|
getConversationHistory(workspaceId, limit = 50) {
|
|
return this.db
|
|
.prepare(`SELECT * FROM conversations
|
|
WHERE workspace_id = ?
|
|
ORDER BY created_at DESC
|
|
LIMIT ?`)
|
|
.all(workspaceId, limit);
|
|
}
|
|
close() {
|
|
this.db.close();
|
|
}
|
|
}
|
|
exports.DisclawDatabase = DisclawDatabase;
|
|
//# sourceMappingURL=database.js.map
|