Database

src/database/dialects.js

@@ -0,0 +1,42 @@
+module.exports = {
+	sqlite: {
+		name: 'SQLite',
+		dialect: 'sqlite',
+		packages: ['sqlite3'],
+	},
+	mysql: {
+		name: 'MySQL',
+		dialect: 'mysql',
+		packages: ['mysql2']
+	},
+	maria: {
+		name: 'MariaDB',
+		dialect: 'mariadb',
+		package: ['mariadb']
+	},
+	mariadb: {
+		name: 'MariaDB',
+		dialect: 'mariadb',
+		package: ['mariadb']
+	},
+	postgre: {
+		name: 'PostgreSQL',
+		dialect: 'postgres',
+		packages: ['pg', 'pg-hstore']
+	},
+	postgres: {
+		name: 'PostgreSQL',
+		dialect: 'postgres',
+		packages: ['pg', 'pg-hstore']
+	},
+	postgresql: {
+		name: 'PostgreSQL',
+		dialect: 'postgres',
+		packages: ['pg', 'pg-hstore']
+	},
+	microsoft: {
+		name: 'Microsoft SQL',
+		dialect: 'mssql',
+		packages: ['tedious']
+	},
+};

src/database/index.js

@@ -0,0 +1,126 @@
+const { 
+	Sequelize,
+	DataTypes
+} = require('sequelize');
+const { path } = require('../utils/fs');
+const config = require('../../user/config');
+const types = require('./dialects');
+const supported = Object.keys(types);
+
+module.exports = async (log) => {
+
+	const {
+		DB_TYPE,
+		DB_HOST,
+		DB_USER,
+		DB_PASS,
+		DB_NAME
+	} = process.env;
+
+	let type = (DB_TYPE || 'sqlite').toLowerCase();
+
+	if (!supported.includes(type)) {
+		log.report('Invalid database type');
+		throw new Error(`DB_TYPE (${type}) is not a valid type`);
+	}
+
+	try {
+		types[type].packages.forEach(pkg => require(pkg));
+	} catch {
+		log.report('Specified database type is not installed');
+		let required = types[type].packages.map(i => `"${i}"`).join(' and ');
+		throw new Error(`Please install the package(s) for your selected database type: ${required}`);
+	}
+
+	let sequelize;
+
+	if (type === 'sqlite') {
+		log.info('Using SQLite storage');
+		sequelize = new Sequelize({
+			dialect: types[type].dialect,
+			storage: path('./user/database.sqlite'),
+			logging: log.debug
+		});
+	} else {
+		log.info(`Connecting to ${types[type].name} database...`);
+		sequelize = new Sequelize(DB_NAME, DB_USER, DB_PASS, {
+			dialect: types[type].dialect,
+			host: DB_HOST,
+			logging: log.debug
+		});
+	}
+
+	try {
+		await sequelize.authenticate();
+		log.success('Connected to database successfully');
+	} catch (error) {
+		log.warn('Unable to connect to database');
+		log.error(error);
+	}
+
+	/* let models = {};
+	let files = fs.readdirSync(path('src/database/models/')).filter(file => file.endsWith('.js'));
+
+	for (let file of files) {
+		let table = require(`./models/${file}`);
+		let model = sequelize.define(table.name, table.model);
+		models[table.name] = model;
+	} */
+
+	const Guild = sequelize.define('Guild', {
+		id: {
+			type: DataTypes.CHAR(18),
+			primaryKey: true
+		},
+		prefix: {
+			type: DataTypes.STRING, // STRING(255) = VARCHAR(255)
+			defaultValue: config.defaults.prefix
+		},
+		locale: {
+			type: DataTypes.STRING,
+			defaultValue: config.defaults.locale
+		}
+	}, {
+		tableName: 'guilds'
+	});
+
+
+	const Ticket = sequelize.define('Ticket', {
+		id: {
+			type: DataTypes.CHAR(18),
+			primaryKey: true
+		},
+		guild: {
+			type: DataTypes.STRING,
+			allowNull: false,
+			references: {
+				model: Guild,
+				key: 'id'
+			},	
+		}
+	}, {
+		tableName: 'tickets'
+	});
+
+	// eslint-disable-next-line no-unused-vars
+	const Message = sequelize.define('Message', {
+		id: {
+			type: DataTypes.CHAR(18),
+			primaryKey: true
+		},
+		ticket: {
+			type: DataTypes.STRING,
+			allowNull: false,
+			references: {
+				model: Ticket,
+				key: 'id'
+			},
+		}
+	}, {
+		tableName: 'messages'
+	});
+
+	sequelize.sync();
+
+	return sequelize.models;
+};