Database migration and seeding

2 files changed, 101 insertions, 0 deletions

diff --git a/src/api/database/migrations/20190221225812_initialMigration.js b/src/api/database/migrations/20190221225812_initialMigration.js
new file mode 100644
index 0000000..455d5c5
--- /dev/null
+++ b/src/api/database/migrations/20190221225812_initialMigration.js
@@ -0,0 +1,69 @@
+exports.up = async knex => {
+	await knex.schema.createTable('users', table => {
+		table.increments();
+		table.string('username');
+		table.string('password');
+		table.boolean('enabled').defaultTo(true);
+		table.boolean('isAdmin').defaultTo(false);
+		table.string('apiKey');
+		table.timestamp('passwordEditedAt');
+		table.timestamp('apiKeyEditedAt');
+		table.timestamp('createdAt');
+		table.timestamp('editedAt');
+	});
+
+	await knex.schema.createTable('albums', table => {
+		table.increments();
+		table.integer('userId');
+		table.string('name');
+		table.timestamp('zippedAt');
+		table.timestamp('createdAt');
+		table.timestamp('editedAt');
+	});
+
+	await knex.schema.createTable('files', table => {
+		table.increments();
+		table.integer('userId');
+		table.string('name');
+		table.string('original');
+		table.string('type');
+		table.integer('size');
+		table.string('hash');
+		table.string('ip');
+		table.timestamp('createdAt');
+		table.timestamp('editedAt');
+	});
+
+	await knex.schema.createTable('links', table => {
+		table.increments();
+		table.integer('userId');
+		table.integer('albumId');
+		table.string('identifier');
+		table.integer('views').defaultTo(0);
+		table.boolean('enabled').defaultTo(true);
+		table.boolean('enableDownload').defaultTo(true);
+		table.timestamp('expiresAt');
+		table.timestamp('createdAt');
+		table.timestamp('editedAt');
+	});
+
+	await knex.schema.createTable('albumsFiles', table => {
+		table.increments();
+		table.integer('albumId');
+		table.integer('fileId');
+	});
+
+	await knex.schema.createTable('albumsLinks', table => {
+		table.increments();
+		table.integer('albumId');
+		table.integer('linkId');
+	});
+};
+exports.down = async knex => {
+	await knex.schema.dropTableIfExists('users');
+	await knex.schema.dropTableIfExists('albums');
+	await knex.schema.dropTableIfExists('files');
+	await knex.schema.dropTableIfExists('links');
+	await knex.schema.dropTableIfExists('albumsFiles');
+	await knex.schema.dropTableIfExists('albumsLinks');
+};
diff --git a/src/api/database/seeds/initial.js b/src/api/database/seeds/initial.js
new file mode 100644
index 0000000..d4a343c
--- /dev/null
+++ b/src/api/database/seeds/initial.js
@@ -0,0 +1,32 @@
+const bcrypt = require('bcrypt');
+const moment = require('moment');
+const randomstring = require('randomstring');
+
+exports.seed = async db => {
+	const now = moment.utc().toDate();
+	const user = await db.table('users').where({ username: 'root' }).first();
+	if (user) return;
+	try {
+		const hash = await bcrypt.hash(process.env.ADMIN_PASSWORD, 10);
+		await db.table('users').insert({
+			username: process.env.ADMIN_ACCOUNT,
+			password: hash,
+			apiKey: randomstring.generate(64),
+			passwordEditedAt: now,
+			apiKeyEditedAt: now,
+			createdAt: now,
+			editedAt: now,
+			isAdmin: true
+		});
+		console.log();
+		console.log('====================================================');
+		console.log('==    Successfully created the admin account.     ==');
+		console.log('====================================================');
+		console.log('==      Run `yarn api` and `yarn site` next       ==');
+		console.log('== preferably with pm2 or tmux to keep them alive ==');
+		console.log('====================================================');
+		console.log();
+	} catch (error) {
+		console.error(error);
+	}
+}