kythia-core 0.9.5-beta → 0.10.1-beta

package/src/database/KythiaSequelize.js

@@ -4,11 +4,12 @@
  * @file src/database/KythiaSequelize.js
  * @copyright © 2025 kenndeclouv
  * @assistant chaa & graa
- * @version 0.9.5-beta
+ * @version 0.10.0-beta
  *
  * @description
  * Main Sequelize connection factory for the application.
  */
+
 const { Sequelize } = require('sequelize');
 
 /**
@@ -20,100 +21,108 @@ const { Sequelize } = require('sequelize');
  * @returns {Sequelize} Configured Sequelize instance
  */
 function createSequelizeInstance(config, logger) {
-    const dbConfig = config.db || {};
-    if (!config.db) config.db = dbConfig;
-
-    let driver = dbConfig.driver || process.env.DB_DRIVER;
-    let name = dbConfig.name || process.env.DB_NAME;
-
-    if (!driver || driver === '') {
-        driver = 'sqlite';
-        if (logger) logger.info('💡 DB driver not specified. Defaulting to: sqlite');
-    } else {
-        driver = driver.toLowerCase();
-    }
-
-    if (driver === 'sqlite') {
-        if (!name || name === '') {
-            name = 'kythiadata.sqlite';
-            if (logger) logger.info('💡 DB name for sqlite not specified. Defaulting to: kythiadata.sqlite');
-        }
-    }
-
-    dbConfig.driver = driver;
-    dbConfig.name = name;
-
-    const dialect = dbConfig.driver;
-    const dbName = dbConfig.name;
-    const dbUser = dbConfig.user || process.env.DB_USER;
-    const dbPassword = dbConfig.password || process.env.DB_PASSWORD;
-    const dbHost = dbConfig.host || process.env.DB_HOST;
-    const dbPort = dbConfig.port || process.env.DB_PORT;
-
-    const dbSocket = dbConfig.socketPath || process.env.DB_SOCKET_PATH;
-    const dbSsl = dbConfig.ssl || process.env.DB_SSL;
-    const dbDialectOptions = dbConfig.dialectOptions || process.env.DB_DIALECT_OPTIONS;
-
-    const seqConfig = {
-        database: dbName,
-        username: dbUser,
-        password: dbPassword,
-        dialect: dialect,
-        logging: (sql) => {
-            logger.debug(sql);
-        },
-        define: {
-            charset: 'utf8mb4',
-            collate: 'utf8mb4_unicode_ci',
-        },
-    };
-
-    if (dialect !== 'sqlite') {
-        seqConfig.timezone = dbConfig.timezone || '+00:00';
-    }
-
-    switch (dialect) {
-        case 'sqlite':
-            seqConfig.storage = dbConfig.name;
-            delete seqConfig.database;
-            break;
-
-        case 'mysql':
-        case 'mariadb':
-            seqConfig.host = dbHost;
-            seqConfig.port = dbPort;
-            if (dbSocket) {
-                seqConfig.dialectOptions = { socketPath: dbSocket };
-            }
-            break;
-
-        case 'postgres':
-            seqConfig.host = dbHost;
-            seqConfig.port = dbPort;
-            if (dbSsl === 'true' || dbSsl === true) {
-                seqConfig.dialectOptions = {
-                    ssl: { require: true, rejectUnauthorized: false },
-                };
-            }
-            break;
-
-        case 'mssql':
-            seqConfig.host = dbHost;
-            seqConfig.port = dbPort;
-            if (dbDialectOptions) {
-                try {
-                    seqConfig.dialectOptions = typeof dbDialectOptions === 'string' ? JSON.parse(dbDialectOptions) : dbDialectOptions;
-                } catch (e) {
-                    logger.error('Error parsing dialect options:', e.message);
-                }
-            }
-            break;
-
-        default:
-            throw new Error(`${dialect} is not supported or not configured.`);
-    }
-
-    return new Sequelize(seqConfig);
+	const dbConfig = config.db || {};
+	if (!config.db) config.db = dbConfig;
+
+	let driver = dbConfig.driver || process.env.DB_DRIVER;
+	let name = dbConfig.name || process.env.DB_NAME;
+
+	if (!driver || driver === '') {
+		driver = 'sqlite';
+		if (logger)
+			logger.info('💡 DB driver not specified. Defaulting to: sqlite');
+	} else {
+		driver = driver.toLowerCase();
+	}
+
+	if (driver === 'sqlite') {
+		if (!name || name === '') {
+			name = 'kythiadata.sqlite';
+			if (logger)
+				logger.info(
+					'💡 DB name for sqlite not specified. Defaulting to: kythiadata.sqlite',
+				);
+		}
+	}
+
+	dbConfig.driver = driver;
+	dbConfig.name = name;
+
+	const dialect = dbConfig.driver;
+	const dbName = dbConfig.name;
+	const dbUser = dbConfig.user || process.env.DB_USER;
+	const dbPassword = dbConfig.password || process.env.DB_PASSWORD;
+	const dbHost = dbConfig.host || process.env.DB_HOST;
+	const dbPort = dbConfig.port || process.env.DB_PORT;
+
+	const dbSocket = dbConfig.socketPath || process.env.DB_SOCKET_PATH;
+	const dbSsl = dbConfig.ssl || process.env.DB_SSL;
+	const dbDialectOptions =
+		dbConfig.dialectOptions || process.env.DB_DIALECT_OPTIONS;
+
+	const seqConfig = {
+		database: dbName,
+		username: dbUser,
+		password: dbPassword,
+		dialect: dialect,
+		logging: (sql) => {
+			logger.debug(sql);
+		},
+		define: {
+			charset: 'utf8mb4',
+			collate: 'utf8mb4_unicode_ci',
+		},
+	};
+
+	if (dialect !== 'sqlite') {
+		seqConfig.timezone = dbConfig.timezone || '+00:00';
+	}
+
+	switch (dialect) {
+		case 'sqlite':
+			seqConfig.storage = dbConfig.name;
+			delete seqConfig.database;
+			break;
+
+		case 'mysql':
+		case 'mariadb':
+			seqConfig.host = dbHost;
+			seqConfig.port = dbPort;
+			if (dbSocket) {
+				seqConfig.dialectOptions = { socketPath: dbSocket };
+			}
+			break;
+
+		case 'postgres':
+			seqConfig.host = dbHost;
+			seqConfig.port = dbPort;
+			if (dbSsl === 'true' || dbSsl === true) {
+				seqConfig.dialectOptions = {
+					ssl: { require: true, rejectUnauthorized: false },
+				};
+			}
+			break;
+
+		case 'mssql':
+			seqConfig.host = dbHost;
+			seqConfig.port = dbPort;
+			if (dbDialectOptions) {
+				try {
+					seqConfig.dialectOptions =
+						typeof dbDialectOptions === 'string'
+							? JSON.parse(dbDialectOptions)
+							: dbDialectOptions;
+				} catch (e) {
+					logger.error('Error parsing dialect options:', e.message);
+				}
+			}
+			break;
+
+		default:
+			throw new Error(`${dialect} is not supported or not configured.`);
+	}
+
+	return new Sequelize(seqConfig);
 }
 
 module.exports = createSequelizeInstance;

package/src/database/KythiaStorage.js

@@ -0,0 +1,117 @@
+/**
+ * 🗄️ Laravel-Style Migration Storage Adapter
+ *
+ * @file src/database/KythiaStorage.js
+ * @copyright © 2025 kenndeclouv
+ * @assistant chaa & graa
+ * @version 0.10.0-beta
+ *
+ * @description
+ * Custom storage adapter for Umzug that mimics Laravel's migration table structure.
+ * It adds a 'batch' column to the migrations table, allowing for smart rollbacks
+ * (undoing only the last batch of migrations instead of just the last file).
+ *
+ * ✨ Core Features:
+ * -  Batch Tracking: Stores migration batch numbers.
+ * -  Auto-Setup: Creates the migrations table automatically if missing.
+ * -  Smart Rollback Support: Methods to fetch the last batch ID and files.
+ */
+
+const { DataTypes } = require('sequelize');
+
+class KythiaStorage {
+	constructor({ sequelize, tableName = 'migrations' }) {
+		this.sequelize = sequelize;
+		this.tableName = tableName;
+		this.currentBatch = 1;
+	}
+
+	async ensureTable() {
+		const queryInterface = this.sequelize.getQueryInterface();
+
+		const tables = await queryInterface.showAllTables();
+		if (!tables.includes(this.tableName)) {
+			await queryInterface.createTable(this.tableName, {
+				id: {
+					type: DataTypes.INTEGER,
+					primaryKey: true,
+					autoIncrement: true,
+				},
+				name: {
+					type: DataTypes.STRING,
+					allowNull: false,
+				},
+				batch: {
+					type: DataTypes.INTEGER,
+					allowNull: false,
+					defaultValue: 1,
+				},
+				migration_time: {
+					type: DataTypes.DATE,
+					defaultValue: DataTypes.NOW,
+				},
+			});
+		}
+	}
+
+	async executed() {
+		await this.ensureTable();
+		const [results] = await this.sequelize.query(
+			`SELECT name FROM ${this.tableName} ORDER BY id ASC`,
+		);
+		return results.map((r) => r.name);
+	}
+
+	async logMigration({ name }) {
+		await this.ensureTable();
+
+		await this.sequelize.query(
+			`INSERT INTO ${this.tableName} (name, batch, migration_time) VALUES (?, ?, ?)`,
+			{
+				replacements: [name, this.currentBatch, new Date()],
+				type: this.sequelize.QueryTypes.INSERT,
+			},
+		);
+	}
+
+	async unlogMigration({ name }) {
+		await this.ensureTable();
+		await this.sequelize.query(`DELETE FROM ${this.tableName} WHERE name = ?`, {
+			replacements: [name],
+			type: this.sequelize.QueryTypes.DELETE,
+		});
+	}
+
+	setBatch(batchNumber) {
+		this.currentBatch = batchNumber;
+	}
+
+	async getLastBatchNumber() {
+		await this.ensureTable();
+		const [result] = await this.sequelize.query(
+			`SELECT MAX(batch) as max_batch FROM ${this.tableName}`,
+			{
+				type: this.sequelize.QueryTypes.SELECT,
+			},
+		);
+		return result ? result.max_batch || 0 : 0;
+	}
+
+	async getLastBatchMigrations() {
+		await this.ensureTable();
+		const lastBatch = await this.getLastBatchNumber();
+		if (lastBatch === 0) return [];
+
+		const migrations = await this.sequelize.query(
+			`SELECT name FROM ${this.tableName} WHERE batch = ? ORDER BY id DESC`,
+			{
+				replacements: [lastBatch],
+				type: this.sequelize.QueryTypes.SELECT,
+			},
+		);
+
+		return migrations.map((m) => m.name);
+	}
+}
+
+module.exports = KythiaStorage;

package/src/database/ModelLoader.js

@@ -0,0 +1,79 @@
+/**
+ * 🤖 Automatic Model Loader & Bootstrapper
+ *
+ * @file src/loaders/ModelLoader.js
+ * @copyright © 2025 kenndeclouv
+ * @assistant chaa & graa
+ * @version 0.10.0-beta
+ *
+ * @description
+ * Scans the `addons` directory for KythiaModel definitions, requires them,
+ * and executes their `autoBoot` static method for database introspection.
+ * Finally, it links Sequelize associations after all models are loaded.
+ *
+ * ✨ Core Features:
+ * -  Addon Scanning: Finds models in `addons/{AddonName}/database/models`.
+ * -  AutoBoot Execution: Triggers DB introspection (reading columns from DB).
+ * -  Association Linking: Automatically calls `.associate()` on models if defined.
+ * -  Container Registration: Registers booted models into the global container.
+ */
+
+const path = require('node:path');
+const fs = require('node:fs');
+
+async function bootModels(kythiaInstance, sequelize) {
+	const { container, logger } = kythiaInstance;
+	const rootDir = container.appRoot;
+	const addonsDir = path.join(rootDir, 'addons');
+
+	if (!fs.existsSync(addonsDir)) return;
+
+	const loadedModels = [];
+	const addonFolders = fs
+		.readdirSync(addonsDir)
+		.filter((f) => fs.statSync(path.join(addonsDir, f)).isDirectory());
+
+	logger.info('📂 Scanning & Booting Models...');
+
+	for (const addon of addonFolders) {
+		const modelsDir = path.join(addonsDir, addon, 'database', 'models');
+		if (fs.existsSync(modelsDir)) {
+			const files = fs.readdirSync(modelsDir).filter((f) => f.endsWith('.js'));
+
+			for (const file of files) {
+				const modelPath = path.join(modelsDir, file);
+				try {
+					const ModelClass = require(modelPath);
+
+					if (ModelClass.autoBoot) {
+						loadedModels.push(ModelClass);
+					}
+				} catch (err) {
+					logger.error(`❌ Failed to require model ${file}:`, err);
+				}
+			}
+		}
+	}
+
+	for (const ModelClass of loadedModels) {
+		try {
+			await ModelClass.autoBoot(sequelize);
+
+			container.models[ModelClass.name] = ModelClass;
+			logger.info(
+				`   ✨ Booted: ${ModelClass.name} -> ${ModelClass.tableName}`,
+			);
+		} catch (err) {
+			logger.error(`❌ AutoBoot Failed for ${ModelClass.name}:`, err);
+		}
+	}
+
+	logger.info('🔗 Linking Associations...');
+	Object.values(container.models).forEach((model) => {
+		if (model.associate) {
+			model.associate(container.models);
+		}
+	});
+}
+
+module.exports = bootModels;