@dataramen/cli 0.0.10 → 0.0.11

package/dist/code/api/project/router.js

@@ -1,100 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const createRouter_1 = require("../../utils/createRouter");
-const request_1 = require("../../utils/request");
-const db_1 = require("../../repository/db");
-const typeorm_1 = require("typeorm");
-exports.default = (0, createRouter_1.createRouter)((instance) => {
-    instance.get("/team/:teamId/files", async (request, reply) => {
-        const { teamId } = (0, request_1.getRequestParams)(request);
-        const query = {
-            where: {
-                team: {
-                    id: teamId,
-                },
-            },
-            order: {
-                name: 'ASC',
-            },
-            select: {
-                id: true,
-                name: true,
-                updatedAt: true,
-            }
-        };
-        const [dataSources = [], workbooks = [], queries = []] = await Promise.all([
-            db_1.DataSourceRepository.find(query),
-            db_1.WorkbookRepository.find({
-                ...query,
-                where: {
-                    ...query.where,
-                    isTrash: false,
-                }
-            }),
-            db_1.QueriesRepository.find({
-                ...query,
-                where: {
-                    ...query.where,
-                    isTrash: false,
-                }
-            })
-        ]);
-        return {
-            data: [
-                ...dataSources.map((d) => ({ ...d, type: "dataSource" })),
-                ...workbooks.map((w) => ({ ...w, type: "workbook" })),
-                ...queries.map((w) => ({ ...w, type: "query" })),
-            ],
-        };
-    });
-    instance.get("/team/:teamId/trash", async (request) => {
-        const { teamId } = (0, request_1.getRequestParams)(request);
-        const workbooks = await db_1.AppDataSource.query(`
-        select id, name, updatedAt
-        from workbooks
-        where teamId = '${teamId}' AND (isTrash = true OR isTrash IS NULL)
-        order by name asc
-        limit 100;
-    `);
-        return {
-            data: workbooks,
-        };
-    });
-    instance.get("/team/:teamId/tables", async (request) => {
-        const { teamId } = (0, request_1.getRequestParams)(request);
-        const { search, size } = (0, request_1.getRequestQuery)(request);
-        const tables = await db_1.DatabaseInspectionRepository.find({
-            where: {
-                tableName: (0, typeorm_1.Like)(`%${search}%`),
-                datasource: {
-                    team: {
-                        id: teamId,
-                    }
-                }
-            },
-            relations: {
-                datasource: true,
-            },
-            select: {
-                id: true,
-                tableName: true,
-                datasource: {
-                    name: true,
-                    id: true,
-                },
-            },
-            order: {
-                tableName: "ASC",
-            },
-            take: parseInt(size) || 20,
-        });
-        return {
-            data: tables.map(t => ({
-                name: t.tableName,
-                id: t.id,
-                dataSourceName: t.datasource?.name || '--',
-                dataSourceId: t.datasource?.id || '--',
-            })),
-        };
-    });
-});

package/dist/code/api/queries/router.js

@@ -1,122 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const createRouter_1 = require("../../utils/createRouter");
-const request_1 = require("../../utils/request");
-const queryUtils_1 = require("../../utils/queryUtils");
-const httpError_1 = require("../../utils/httpError");
-const db_1 = require("../../repository/db");
-const typeorm_1 = require("typeorm");
-exports.default = (0, createRouter_1.createRouter)((instance) => {
-    instance.get("/", async (request) => {
-        const { dataSourceId, teamId, limit, orderBy, name } = (0, request_1.getRequestQuery)(request);
-        if (!dataSourceId && !teamId) {
-            throw new httpError_1.HttpError(400, "Either dsId or teamId is required");
-        }
-        const where = {
-            isTrash: false,
-        };
-        if (dataSourceId) {
-            where.dataSource = {
-                id: dataSourceId,
-            };
-        }
-        if (teamId) {
-            where.team = {
-                id: teamId,
-            };
-        }
-        if (name) {
-            where.name = (0, typeorm_1.Like)(`%${name}%`);
-        }
-        const queries = await db_1.QueriesRepository.find({
-            where,
-            take: limit,
-            order: (0, queryUtils_1.parseOrderQueryParam)(orderBy, {
-                createdAt: "DESC",
-            })
-        });
-        return {
-            data: queries,
-        };
-    });
-    instance.get("/:id", async (request) => {
-        const { id } = (0, request_1.getRequestParams)(request);
-        const query = await db_1.QueriesRepository.findOne({
-            where: {
-                id,
-            },
-            select: {
-                dataSource: {
-                    id: true,
-                },
-            },
-            relations: {
-                dataSource: true,
-            }
-        });
-        if (!query) {
-            return {
-                status: 404,
-                data: "Query not found",
-            };
-        }
-        return {
-            data: query,
-        };
-    });
-    instance.post("/", async (request) => {
-        const payload = (0, request_1.getRequestPayload)(request);
-        const dataSource = await db_1.DataSourceRepository.findOne({
-            where: {
-                id: payload.dataSourceId,
-            },
-            relations: {
-                team: true,
-            },
-        });
-        const query = await db_1.QueriesRepository.save(db_1.QueriesRepository.create({
-            name: payload.name,
-            isTrash: false,
-            opts: payload.opts,
-            team: {
-                id: dataSource?.team.id,
-            },
-            dataSource: {
-                id: payload.dataSourceId,
-            }
-        }));
-        return {
-            data: query,
-        };
-    });
-    // update workbook
-    instance.patch("/:id", async (request) => {
-        const { id } = (0, request_1.getRequestParams)(request);
-        const payload = (0, request_1.getRequestPayload)(request);
-        const result = await db_1.QueriesRepository.update(id, payload);
-        if (!result.affected) {
-            throw new httpError_1.HttpError(404, "Workbook not found");
-        }
-        const query = await db_1.QueriesRepository.findOneBy({
-            id,
-        });
-        return {
-            data: query,
-        };
-    });
-    // delete query
-    instance.delete("/:id", async (request) => {
-        return db_1.AppDataSource.transaction(async () => {
-            const { id } = (0, request_1.getRequestParams)(request);
-            const result = await db_1.QueriesRepository.delete({
-                id
-            });
-            if (!result.affected) {
-                return {
-                    status: 404,
-                    data: "Workbook not found",
-                };
-            }
-        });
-    });
-});

package/dist/code/api/runner/router.js

@@ -1,22 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const createRouter_1 = require("../../utils/createRouter");
-const request_1 = require("../../utils/request");
-const userSqlPromptRunner_1 = require("../../services/userSqlPromptRunner");
-exports.default = (0, createRouter_1.createRouter)((instance) => {
-    // run sql
-    instance.post("/sql", async (request) => {
-        const payload = (0, request_1.getRequestPayload)(request);
-        const result = await (0, userSqlPromptRunner_1.runUserSqlQuery)(request, payload);
-        return {
-            data: result,
-        };
-    });
-    instance.post("/raw", async (request) => {
-        const payload = (0, request_1.getRequestPayload)(request);
-        const result = await (0, userSqlPromptRunner_1.runRawSqlQuery)(request, payload);
-        return {
-            data: result,
-        };
-    });
-});

package/dist/code/api/status/router.js

@@ -1,17 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const createRouter_1 = require("../../utils/createRouter");
-const node_child_process_1 = require("node:child_process");
-exports.default = (0, createRouter_1.createRouter)((instance) => {
-    instance.get("/", async (request, reply) => {
-        return {
-            data: {
-                active: true,
-                version: process.env.SERVER_VERSION,
-            },
-        };
-    });
-    instance.get("/logs", async () => {
-        (0, node_child_process_1.spawn)(`echo "${process.env.SERVER_VERSION}"`);
-    });
-});

package/dist/code/api/teams/router.js

@@ -1,35 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const createRouter_1 = require("../../utils/createRouter");
-const request_1 = require("../../utils/request");
-const db_1 = require("../../repository/db");
-exports.default = (0, createRouter_1.createRouter)((instance) => {
-    instance.get("/", async (request, reply) => {
-        const userId = (0, request_1.getRequestUserId)(request);
-        const teams = await db_1.TeamRepository.find({
-            where: {
-                users: {
-                    id: userId,
-                }
-            },
-        });
-        return {
-            data: teams,
-        };
-    });
-    // post
-    instance.post("/", async (request, reply) => {
-        return db_1.AppDataSource.transaction(async () => {
-            const userId = (0, request_1.getRequestUserId)(request);
-            const teamPayload = (0, request_1.getRequestPayload)(request);
-            const user = db_1.UserRepository.create();
-            user.id = userId;
-            const team = db_1.TeamRepository.create(teamPayload);
-            team.users = [user];
-            await db_1.TeamRepository.save(team);
-            return {
-                data: team,
-            };
-        });
-    });
-});

package/dist/code/api/userSettings/router.js

@@ -1,54 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const createRouter_1 = require("../../utils/createRouter");
-const request_1 = require("../../utils/request");
-const httpError_1 = require("../../utils/httpError");
-const token_1 = require("../../utils/token");
-const db_1 = require("../../repository/db");
-function sanitizeFeSettings(settings) {
-    const newSettings = db_1.UserSettingsRepository.create(settings);
-    if (settings.openAiToken) {
-        newSettings.openAiToken = (0, token_1.sanitizeOpenAIToken)(settings.openAiToken);
-    }
-    return newSettings;
-}
-exports.default = (0, createRouter_1.createRouter)((instance) => {
-    // get user settings by user id
-    instance.get("/", async (req) => {
-        const userId = (0, request_1.getRequestUserId)(req);
-        let settings = await db_1.UserSettingsRepository.findOneBy({
-            user: {
-                id: userId,
-            }
-        });
-        if (!settings) {
-            // create default settings if not found
-            settings = await db_1.UserSettingsRepository.save(db_1.UserSettingsRepository.create({
-                user: {
-                    id: userId,
-                },
-                model: "gpt-4o",
-            }));
-        }
-        return {
-            data: sanitizeFeSettings(settings),
-        };
-    });
-    // update user settings
-    instance.patch("/", async (req) => {
-        const { settings } = (0, request_1.getRequestPayload)(req);
-        if (!settings.id) {
-            throw new httpError_1.HttpError(400, "Settings id is required!");
-        }
-        const result = await db_1.UserSettingsRepository.update(settings.id, settings);
-        if (!result.affected) {
-            throw new httpError_1.HttpError(404, "You do not own these settings!");
-        }
-        const updatedSettings = await db_1.UserSettingsRepository.findOneBy({
-            id: settings.id,
-        });
-        return {
-            data: updatedSettings,
-        };
-    });
-});

package/dist/code/api/users/router.js

@@ -1,91 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const createRouter_1 = require("../../utils/createRouter");
-const request_1 = require("../../utils/request");
-const httpError_1 = require("../../utils/httpError");
-const oauth2Client_1 = require("../../services/oauthClient/oauth2Client");
-const db_1 = require("../../repository/db");
-async function createUser() {
-    const team = db_1.TeamRepository.create({
-        name: "Personal",
-    });
-    const user = await db_1.UserRepository.save(db_1.UserRepository.create({}));
-    team.users = [user];
-    await db_1.TeamRepository.save(team);
-    await db_1.UserRepository.update(user.id, {
-        currentTeam: {
-            id: team.id,
-        }
-    });
-    return db_1.UserRepository.findOne({
-        where: {},
-        relations: {
-            currentTeam: true,
-        },
-    });
-}
-exports.default = (0, createRouter_1.createRouter)((instance) => {
-    instance.post("/register", async (request, reply) => {
-        const userInfo = await (0, oauth2Client_1.getTokenInfoFromRequest)(request);
-        if (!userInfo?.email || !userInfo?.sub) {
-            throw new httpError_1.HttpError(400, "Invalid token");
-        }
-        const user = await createUser();
-        return {
-            data: user,
-        };
-    });
-    // get user
-    instance.get("/", async (request, reply) => {
-        const user = await db_1.UserRepository.findOne({
-            where: {}, // select first user
-            relations: {
-                currentTeam: true,
-            }
-        });
-        if (!user) {
-            const user = await createUser();
-            return {
-                data: user,
-            };
-        }
-        return {
-            data: user,
-        };
-    });
-    // update user
-    instance.put("/", async (request, reply) => {
-        const userId = (0, request_1.getRequestUserId)(request);
-        const payload = (0, request_1.getRequestPayload)(request);
-        const result = await db_1.UserRepository.update(userId, payload);
-        if (!result.affected) {
-            throw new httpError_1.HttpError(404, "User not found");
-        }
-        const user = db_1.UserRepository.findOneBy({
-            id: userId,
-        });
-        return {
-            data: user,
-        };
-    });
-    instance.put('/team', async (request, reply) => {
-        const userId = (0, request_1.getRequestUserId)(request);
-        const payload = (0, request_1.getRequestPayload)(request);
-        await db_1.UserRepository.update(userId, {
-            currentTeam: {
-                id: payload.teamId,
-            }
-        });
-        const user = await db_1.UserRepository.findOne({
-            where: {
-                id: userId,
-            },
-            relations: {
-                currentTeam: true,
-            },
-        });
-        return {
-            data: user,
-        };
-    });
-});

package/dist/code/api/workbooks/router.js

@@ -1,123 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const createRouter_1 = require("../../utils/createRouter");
-const request_1 = require("../../utils/request");
-const queryUtils_1 = require("../../utils/queryUtils");
-const httpError_1 = require("../../utils/httpError");
-const db_1 = require("../../repository/db");
-const files_1 = require("../../services/files");
-exports.default = (0, createRouter_1.createRouter)((instance) => {
-    instance.get("/", async (request) => {
-        const { dsId, teamId, limit, orderBy } = (0, request_1.getRequestQuery)(request);
-        if (!dsId && !teamId) {
-            throw new httpError_1.HttpError(400, "Either dsId or teamId is required");
-        }
-        const workbooks = await db_1.WorkbookRepository.find({
-            where: {
-                team: {
-                    id: teamId,
-                },
-                isTrash: false,
-            },
-            take: limit,
-            order: (0, queryUtils_1.parseOrderQueryParam)(orderBy, {
-                createdAt: "DESC",
-            })
-        });
-        return {
-            data: workbooks,
-        };
-    });
-    instance.get("/:id", async (request) => {
-        const { id } = (0, request_1.getRequestParams)(request);
-        const workbook = await db_1.WorkbookRepository.findOneBy({
-            id,
-        });
-        if (!workbook) {
-            return {
-                status: 404,
-                data: "Workbook not found",
-            };
-        }
-        return {
-            data: workbook,
-        };
-    });
-    instance.post("/", async (request) => {
-        const payload = (0, request_1.getRequestPayload)(request);
-        const workbook = await db_1.WorkbookRepository.save(db_1.WorkbookRepository.create({
-            name: payload.name,
-            isTrash: false,
-            team: {
-                id: payload.teamId,
-            }
-        }));
-        return {
-            data: workbook,
-        };
-    });
-    // update workbook
-    instance.patch("/:id", async (request) => {
-        const { id } = (0, request_1.getRequestParams)(request);
-        const payload = (0, request_1.getRequestPayload)(request);
-        const result = await db_1.WorkbookRepository.update(id, payload);
-        if (!result.affected) {
-            throw new httpError_1.HttpError(404, "Workbook not found");
-        }
-        const workbook = await db_1.WorkbookRepository.findOneBy({
-            id,
-        });
-        return {
-            data: workbook,
-        };
-    });
-    // delete workbook
-    instance.delete("/:id", async (request) => {
-        const { id } = (0, request_1.getRequestParams)(request);
-        // todo: delete file
-        const result = await db_1.WorkbookRepository.delete({
-            id
-        });
-        if (!result.affected) {
-            return {
-                status: 404,
-                data: "Workbook not found",
-            };
-        }
-    });
-    instance.get("/:id/content", async (request) => {
-        const { id } = (0, request_1.getRequestParams)(request);
-        const workbook = await db_1.WorkbookRepository.findOneBy({
-            id,
-        });
-        if (!workbook) {
-            throw new httpError_1.HttpError(404, "Workbook not found");
-        }
-        try {
-            const file = await (0, files_1.getFile)(workbook?.path);
-            if (file) {
-                return {
-                    data: file,
-                };
-            }
-        }
-        catch (e) { }
-        return {
-            data: "",
-        };
-    });
-    instance.patch("/:id/content", async (request) => {
-        const { id } = (0, request_1.getRequestParams)(request);
-        const body = (0, request_1.getRequestPayload)(request);
-        const workbook = await db_1.WorkbookRepository.findOneBy({
-            id,
-        });
-        if (!workbook?.path) {
-            throw new httpError_1.HttpError(404, "Workbook not found");
-        }
-        await (0, files_1.storeFile)(workbook?.path, body.content);
-        await db_1.WorkbookRepository.update(workbook.id, {
-            updatedAt: new Date()
-        });
-    });
-});

package/dist/code/api/workbooks/types.js

@@ -1,2 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });

package/dist/code/env.js

@@ -1,25 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-const dotenv_1 = require("dotenv");
-const node_path_1 = require("node:path");
-const node_fs_1 = require("node:fs");
-const packageJson = (() => {
-    try {
-        const file = (0, node_fs_1.readFileSync)((0, node_path_1.join)(__dirname, "..", "package.json"), "utf8");
-        return JSON.parse(file);
-    }
-    catch (e) {
-        return {
-            version: "0.0.0",
-        };
-    }
-})();
-(0, dotenv_1.config)({
-    path: [
-        (0, node_path_1.join)(__dirname, "..", "env", ".env"),
-        (0, node_path_1.join)(__dirname, "..", "env", ".env.default"),
-    ],
-});
-(0, dotenv_1.populate)(process.env, {
-    SERVER_VERSION: packageJson.version,
-});

package/dist/code/index.js

@@ -1,86 +0,0 @@
-"use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-require("./env");
-const fastify_1 = __importDefault(require("fastify"));
-const cors_1 = __importDefault(require("@fastify/cors"));
-const httpError_1 = require("./utils/httpError");
-const db_1 = require("./repository/db");
-const files_1 = require("./services/files");
-const router_1 = __importDefault(require("./api/chat/router"));
-const router_2 = __importDefault(require("./api/dataSources/router"));
-const router_3 = __importDefault(require("./api/project/router"));
-const router_4 = __importDefault(require("./api/queries/router"));
-const router_5 = __importDefault(require("./api/runner/router"));
-const router_6 = __importDefault(require("./api/status/router"));
-const router_7 = __importDefault(require("./api/teams/router"));
-const router_8 = __importDefault(require("./api/users/router"));
-const router_9 = __importDefault(require("./api/userSettings/router"));
-const router_10 = __importDefault(require("./api/workbooks/router"));
-const server = (0, fastify_1.default)();
-const PORT = process.env.PORT ? parseInt(process.env.PORT) : 4466;
-function registerRouter(fn, prefix) {
-    server.register(fn, { prefix });
-    console.log("Registered " + prefix);
-}
-async function initialize() {
-    await (0, files_1.setupProjectFolders)(); // creates .dataramen/files
-    // const routers = glob.sync("./api/**/router.js", {
-    //   cwd: __dirname,
-    // });
-    //
-    // routers.forEach((file) => {
-    //   const apiName = file.split("/")[2];
-    //   const prefix = "/api/" + folderNameToSnakeCase(apiName);
-    //   server.register(require(file), { prefix });
-    //   console.log(`${prefix} -> ${apiName} controller `);
-    // });
-    registerRouter(router_1.default, `/api/chat`);
-    registerRouter(router_2.default, `/api/data-sources`);
-    registerRouter(router_3.default, `/api/project`);
-    registerRouter(router_4.default, `/api/queries`);
-    registerRouter(router_5.default, `/api/runner`);
-    registerRouter(router_6.default, `/api/status`);
-    registerRouter(router_7.default, `/api/teams`);
-    registerRouter(router_8.default, `/api/users`);
-    registerRouter(router_9.default, `/api/user-settings`);
-    registerRouter(router_10.default, `/api/workbooks`);
-    server.register(cors_1.default, {
-        origin: "*",
-        methods: "*",
-    });
-    server.addHook("onResponse", async (request) => {
-        if (request.__connections) {
-            request.__connections.forEach((connection) => {
-                connection.close();
-            });
-        }
-    });
-    server.setErrorHandler((error, request, reply) => {
-        if (error instanceof httpError_1.HttpError) {
-            console.error(error);
-            reply.status(error.status).send({
-                error: error.message,
-            });
-            return;
-        }
-        else {
-            console.error(error);
-            reply.status(500).send({
-                error: "Internal Server Error",
-            });
-        }
-    });
-    await server.after();
-    await (0, db_1.init)();
-    server.listen({ port: PORT, host: '0.0.0.0' }, (err, address) => {
-        if (err) {
-            console.error(err);
-            process.exit(1);
-        }
-        console.log(`Server listening at ${address}`);
-    });
-}
-initialize();