node-sqlite/dist/commands/schema.js

53 lines
2.8 KiB
JavaScript

"use strict";
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
return new (P || (P = Promise))(function (resolve, reject) {
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
step((generator = generator.apply(thisArg, _arguments || [])).next());
});
};
var __importDefault = (this && this.__importDefault) || function (mod) {
return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.default = default_1;
const commander_1 = require("commander");
const db_schema_manager_1 = require("../lib/sqlite/db-schema-manager");
const init_1 = __importDefault(require("../functions/init"));
const grab_dir_names_1 = __importDefault(require("../data/grab-dir-names"));
const path_1 = __importDefault(require("path"));
const schema_to_typedef_1 = __importDefault(require("../lib/sqlite/schema-to-typedef"));
const append_default_fields_to_db_schema_1 = __importDefault(require("../utils/append-default-fields-to-db-schema"));
const chalk_1 = __importDefault(require("chalk"));
function default_1() {
return new commander_1.Command("schema")
.description("Build DB From Schema")
.option("-v, --vector", "Recreate Vector Tables. This will drop and rebuild all vector tables")
.option("-t, --typedef", "Generate typescript type definitions")
.action((opts) => __awaiter(this, void 0, void 0, function* () {
console.log(`Starting process ...`);
const { config, dbSchema } = yield (0, init_1.default)();
const { ROOT_DIR } = (0, grab_dir_names_1.default)();
const isVector = Boolean(opts.vector || opts.v);
const isTypeDef = Boolean(opts.typedef || opts.t);
const finaldbSchema = (0, append_default_fields_to_db_schema_1.default)({ dbSchema });
const manager = new db_schema_manager_1.SQLiteSchemaManager({
schema: finaldbSchema,
recreate_vector_table: isVector,
});
yield manager.syncSchema();
manager.close();
if (isTypeDef && config.typedef_file_path) {
const out_file = path_1.default.resolve(ROOT_DIR, config.typedef_file_path);
(0, schema_to_typedef_1.default)({
dbSchema: finaldbSchema,
dst_file: out_file,
});
}
console.log(`${chalk_1.default.bold(chalk_1.default.green(`DB Schema setup success!`))}`);
process.exit();
}));
}