2023-06-08 06:31:35 +02:00
|
|
|
const slugify = require("slugify");
|
|
|
|
const { Document } = require("./documents");
|
2023-06-15 08:12:59 +02:00
|
|
|
const { checkForMigrations } = require("../utils/database");
|
2023-06-04 04:28:07 +02:00
|
|
|
|
|
|
|
const Workspace = {
|
2023-06-08 06:31:35 +02:00
|
|
|
tablename: "workspaces",
|
2023-06-15 08:12:59 +02:00
|
|
|
writable: [
|
|
|
|
// Used for generic updates so we can validate keys in request body
|
|
|
|
"name",
|
|
|
|
"slug",
|
|
|
|
"vectorTag",
|
|
|
|
"openAiTemp",
|
2023-06-28 21:53:26 +02:00
|
|
|
"openAiHistory",
|
2023-06-15 08:12:59 +02:00
|
|
|
"lastUpdatedAt",
|
|
|
|
],
|
2023-06-04 04:28:07 +02:00
|
|
|
colsInit: `
|
|
|
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
2023-06-15 08:12:59 +02:00
|
|
|
name TEXT NOT NULL,
|
2023-06-04 04:28:07 +02:00
|
|
|
slug TEXT NOT NULL UNIQUE,
|
|
|
|
vectorTag TEXT DEFAULT NULL,
|
|
|
|
createdAt TEXT DEFAULT CURRENT_TIMESTAMP,
|
2023-06-15 08:12:59 +02:00
|
|
|
openAiTemp REAL DEFAULT NULL,
|
2023-06-28 21:53:26 +02:00
|
|
|
openAiHistory INTEGER DEFAULT 20,
|
2023-06-04 04:28:07 +02:00
|
|
|
lastUpdatedAt TEXT DEFAULT CURRENT_TIMESTAMP
|
|
|
|
`,
|
2023-06-15 08:12:59 +02:00
|
|
|
migrateTable: async function () {
|
|
|
|
console.log(`\x1b[34m[MIGRATING]\x1b[0m Checking for Workspace migrations`);
|
|
|
|
const db = await this.db(false);
|
|
|
|
await checkForMigrations(this, db);
|
|
|
|
},
|
|
|
|
migrations: function () {
|
|
|
|
return [
|
|
|
|
{
|
|
|
|
colName: "openAiTemp",
|
|
|
|
execCmd: `ALTER TABLE ${this.tablename} ADD COLUMN openAiTemp REAL DEFAULT NULL`,
|
|
|
|
doif: false,
|
|
|
|
},
|
|
|
|
{
|
|
|
|
colName: "id",
|
|
|
|
execCmd: `CREATE TRIGGER IF NOT EXISTS Trg_LastUpdated AFTER UPDATE ON ${this.tablename}
|
|
|
|
FOR EACH ROW
|
|
|
|
BEGIN
|
|
|
|
UPDATE ${this.tablename} SET lastUpdatedAt = CURRENT_TIMESTAMP WHERE id = old.id;
|
|
|
|
END`,
|
|
|
|
doif: true,
|
|
|
|
},
|
2023-06-28 21:53:26 +02:00
|
|
|
{
|
|
|
|
colName: "openAiHistory",
|
|
|
|
execCmd: `ALTER TABLE ${this.tablename} ADD COLUMN openAiHistory INTEGER DEFAULT 20`,
|
|
|
|
doif: false,
|
|
|
|
},
|
2023-06-15 08:12:59 +02:00
|
|
|
];
|
|
|
|
},
|
|
|
|
db: async function (tracing = true) {
|
2023-06-08 06:31:35 +02:00
|
|
|
const sqlite3 = require("sqlite3").verbose();
|
|
|
|
const { open } = require("sqlite");
|
2023-06-04 04:28:07 +02:00
|
|
|
|
|
|
|
const db = await open({
|
2023-06-08 23:00:43 +02:00
|
|
|
filename: `${
|
2023-06-13 20:26:11 +02:00
|
|
|
!!process.env.STORAGE_DIR ? `${process.env.STORAGE_DIR}/` : "storage/"
|
2023-06-08 23:00:43 +02:00
|
|
|
}anythingllm.db`,
|
2023-06-08 06:31:35 +02:00
|
|
|
driver: sqlite3.Database,
|
|
|
|
});
|
2023-06-04 04:28:07 +02:00
|
|
|
|
2023-06-08 06:31:35 +02:00
|
|
|
await db.exec(
|
|
|
|
`CREATE TABLE IF NOT EXISTS ${this.tablename} (${this.colsInit})`
|
|
|
|
);
|
2023-06-15 08:12:59 +02:00
|
|
|
|
|
|
|
if (tracing) db.on("trace", (sql) => console.log(sql));
|
2023-06-08 06:31:35 +02:00
|
|
|
return db;
|
2023-06-04 04:28:07 +02:00
|
|
|
},
|
|
|
|
new: async function (name = null) {
|
2023-06-08 06:31:35 +02:00
|
|
|
if (!name) return { result: null, message: "name cannot be null" };
|
2023-06-15 08:12:59 +02:00
|
|
|
var slug = slugify(name, { lower: true });
|
|
|
|
|
|
|
|
const existingBySlug = await this.get(`slug = '${slug}'`);
|
|
|
|
if (existingBySlug !== null) {
|
|
|
|
const slugSeed = Math.floor(10000000 + Math.random() * 90000000);
|
|
|
|
slug = slugify(`${name}-${slugSeed}`, { lower: true });
|
|
|
|
}
|
2023-06-04 04:28:07 +02:00
|
|
|
|
2023-06-08 06:31:35 +02:00
|
|
|
const db = await this.db();
|
|
|
|
const { id, success, message } = await db
|
|
|
|
.run(`INSERT INTO ${this.tablename} (name, slug) VALUES (?, ?)`, [
|
|
|
|
name,
|
2023-06-15 08:12:59 +02:00
|
|
|
slug,
|
2023-06-08 06:31:35 +02:00
|
|
|
])
|
2023-06-04 04:28:07 +02:00
|
|
|
.then((res) => {
|
2023-06-08 06:31:35 +02:00
|
|
|
return { id: res.lastID, success: true, message: null };
|
2023-06-04 04:28:07 +02:00
|
|
|
})
|
|
|
|
.catch((error) => {
|
2023-06-08 06:31:35 +02:00
|
|
|
return { id: null, success: false, message: error.message };
|
|
|
|
});
|
2023-06-15 08:12:59 +02:00
|
|
|
|
|
|
|
if (!success) {
|
|
|
|
db.close();
|
|
|
|
return { workspace: null, message };
|
|
|
|
}
|
2023-06-04 04:28:07 +02:00
|
|
|
|
2023-06-08 06:31:35 +02:00
|
|
|
const workspace = await db.get(
|
|
|
|
`SELECT * FROM ${this.tablename} WHERE id = ${id}`
|
|
|
|
);
|
2023-06-15 08:12:59 +02:00
|
|
|
db.close();
|
|
|
|
|
2023-06-08 06:31:35 +02:00
|
|
|
return { workspace, message: null };
|
2023-06-04 04:28:07 +02:00
|
|
|
},
|
2023-06-15 08:12:59 +02:00
|
|
|
update: async function (id = null, data = {}) {
|
|
|
|
if (!id) throw new Error("No workspace id provided for update");
|
|
|
|
|
|
|
|
const validKeys = Object.keys(data).filter((key) =>
|
|
|
|
this.writable.includes(key)
|
|
|
|
);
|
|
|
|
const values = Object.values(data);
|
|
|
|
if (validKeys.length === 0 || validKeys.length !== values.length)
|
|
|
|
return { workspace: { id }, message: "No valid fields to update!" };
|
2023-06-28 21:53:26 +02:00
|
|
|
|
2023-06-15 08:12:59 +02:00
|
|
|
const template = `UPDATE ${this.tablename} SET ${validKeys.map((key) => {
|
|
|
|
return `${key}=?`;
|
|
|
|
})} WHERE id = ?`;
|
|
|
|
const db = await this.db();
|
|
|
|
const { success, message } = await db
|
|
|
|
.run(template, [...values, id])
|
|
|
|
.then(() => {
|
|
|
|
return { success: true, message: null };
|
|
|
|
})
|
|
|
|
.catch((error) => {
|
|
|
|
return { success: false, message: error.message };
|
|
|
|
});
|
|
|
|
|
2023-06-28 21:53:26 +02:00
|
|
|
db.close();
|
2023-06-15 08:12:59 +02:00
|
|
|
if (!success) {
|
|
|
|
return { workspace: null, message };
|
|
|
|
}
|
|
|
|
|
|
|
|
const updatedWorkspace = await this.get(`id = ${id}`);
|
|
|
|
return { workspace: updatedWorkspace, message: null };
|
|
|
|
},
|
2023-06-08 06:31:35 +02:00
|
|
|
get: async function (clause = "") {
|
|
|
|
const db = await this.db();
|
|
|
|
const result = await db
|
|
|
|
.get(`SELECT * FROM ${this.tablename} WHERE ${clause}`)
|
|
|
|
.then((res) => res || null);
|
2023-06-04 04:28:07 +02:00
|
|
|
if (!result) return null;
|
2023-06-15 08:12:59 +02:00
|
|
|
db.close();
|
2023-06-04 04:28:07 +02:00
|
|
|
|
|
|
|
const documents = await Document.forWorkspace(result.id);
|
2023-06-08 06:31:35 +02:00
|
|
|
return { ...result, documents };
|
2023-06-04 04:28:07 +02:00
|
|
|
},
|
2023-06-08 06:31:35 +02:00
|
|
|
delete: async function (clause = "") {
|
|
|
|
const db = await this.db();
|
|
|
|
await db.get(`DELETE FROM ${this.tablename} WHERE ${clause}`);
|
2023-06-15 08:12:59 +02:00
|
|
|
db.close();
|
|
|
|
|
2023-06-08 06:31:35 +02:00
|
|
|
return true;
|
2023-06-04 04:28:07 +02:00
|
|
|
},
|
2023-06-08 06:31:35 +02:00
|
|
|
where: async function (clause = "", limit = null) {
|
|
|
|
const db = await this.db();
|
|
|
|
const results = await db.all(
|
|
|
|
`SELECT * FROM ${this.tablename} ${clause ? `WHERE ${clause}` : ""} ${
|
|
|
|
!!limit ? `LIMIT ${limit}` : ""
|
|
|
|
}`
|
|
|
|
);
|
2023-06-15 08:12:59 +02:00
|
|
|
db.close();
|
|
|
|
|
2023-06-08 06:31:35 +02:00
|
|
|
return results;
|
2023-06-04 04:28:07 +02:00
|
|
|
},
|
2023-06-08 06:31:35 +02:00
|
|
|
};
|
2023-06-04 04:28:07 +02:00
|
|
|
|
2023-06-08 06:31:35 +02:00
|
|
|
module.exports = { Workspace };
|