tbenandClaude Opus 5.5
864dcce30a
A workspace's task now carries the bot that has it there (bot_id with workspace_id), or none while it's shared. A bot reaches its own list and, in its workspaces, its own tasks plus the shared ones; a workspace admin every task there. Bots create workspace tasks for themselves by default, take a shared one by starting it, and hand-overs record the new bot. Removing or deleting a bot makes its workspace tasks shared again. The conversation's Tasks modal shows that bot's tasks in the workspace. Co-Authored-By: Claude Opus 5.5 <[email protected]>
88 lines
3.2 KiB
TypeScript
88 lines
3.2 KiB
TypeScript
import BunSQLite from "@moduletrace/bun-sqlite";
|
|
import deleteMedia from "@/src/functions/backend/db/media/delete-media";
|
|
import purgeStorage, { grabStorageMediaIds } from "@/src/functions/backend/db/bot-files/purge-storage";
|
|
import { grabTaskMediaIds } from "@/src/functions/backend/db/tasks/purge-task-media";
|
|
import AppError from "@/src/utils/app-error";
|
|
import { clearEnvCache } from "../envs/grab-all-envs";
|
|
import grabUserBot from "./grab-user-bot";
|
|
import assertCrewManager from "@/src/functions/backend/db/access/assert-crew-manager";
|
|
|
|
type Params = {
|
|
user_id: number | string;
|
|
bot_id: number | string;
|
|
};
|
|
|
|
/**
|
|
* Deletes one of the user's worker bots with its conversations, tasks,
|
|
* jobs, models, envs (users' personal secrets for it too) and workspace
|
|
* places. A job it's running stops: it loses
|
|
* its row. The coordinator can't be deleted.
|
|
*/
|
|
export default async function deleteBot({ user_id, bot_id }: Params) {
|
|
await assertCrewManager({ user_id });
|
|
|
|
const bot = await grabUserBot({ bot_id, user_id });
|
|
|
|
if (!bot?.id) {
|
|
throw new AppError(`Bot not found.`);
|
|
}
|
|
|
|
if (bot.kind == "coordinator") {
|
|
throw new AppError(`The coordinator can't be deleted.`);
|
|
}
|
|
|
|
/**
|
|
* Read before the delete: the files' rows go with the bot, and so do its
|
|
* tasks and the links to what they filed.
|
|
*/
|
|
const file_media_ids = await grabStorageMediaIds({ scope: "bot", owner_id: bot.id! }).catch(
|
|
() => [],
|
|
);
|
|
const task_media_ids = await grabTaskMediaIds({ bot_id: bot.id! }).catch(() => []);
|
|
|
|
/**
|
|
* Explicitly, since foreign keys are off outside the server process.
|
|
*/
|
|
await BunSQLite.sql({ sql: `DELETE FROM bot_types WHERE bot_id = ?`, values: [bot.id] });
|
|
|
|
/**
|
|
* Its tasks in workspaces belong to the workspace: they stay, shared
|
|
* again, instead of cascading away with its own list.
|
|
*/
|
|
await BunSQLite.sql({
|
|
sql: `UPDATE tasks SET bot_id = NULL, updated_at = ? WHERE bot_id = ? AND workspace_id IS NOT NULL`,
|
|
values: [Date.now(), bot.id],
|
|
});
|
|
await BunSQLite.sql({ sql: `DELETE FROM envs WHERE bot_id = ?`, values: [bot.id] });
|
|
await BunSQLite.sql({ sql: `DELETE FROM secret_requests WHERE bot_id = ?`, values: [bot.id] });
|
|
await BunSQLite.sql({ sql: `DELETE FROM bot_graphics WHERE bot_id = ?`, values: [bot.id] });
|
|
|
|
const res = await BunSQLite.sql({ sql: `DELETE FROM bots WHERE id = ?`, values: [bot.id] });
|
|
|
|
if (!res.success) {
|
|
throw new Error(`Couldn't delete the bot: ${res.error}`);
|
|
}
|
|
|
|
clearEnvCache();
|
|
|
|
if (bot.image_media_id) {
|
|
await deleteMedia({ id: bot.image_media_id, user_id: Number(user_id) }).catch(() => {});
|
|
}
|
|
|
|
/**
|
|
* What the bot's tasks filed and no folder owns (a chat upload): the
|
|
* links went with the tasks, so these bytes would be stranded.
|
|
*/
|
|
await deleteMedia({ ids: task_media_ids, user_id: Number(user_id), can_delete_all_media: true }).catch(
|
|
() => {},
|
|
);
|
|
|
|
/**
|
|
* Its folder goes with it: the bytes, their `media` rows and the search
|
|
* vectors.
|
|
*/
|
|
await purgeStorage({ scope: "bot", owner_id: bot.id!, media_ids: file_media_ids });
|
|
|
|
return bot;
|
|
}
|