/* jslint node: true */ 'use strict'; var async = require('async'); var path = require('path'); var db = require('./database'); var utils = require('../public/src/utils'); var Upgrade = { available: [ { version: "1.2.0", upgrades: ['category_recent_tids'] }, { version: "1.3.0", upgrades: ['favourites_to_bookmarks', 'sorted_sets_for_post_replies'] }, { version: "1.4.0", upgrades: ['global_and_user_language_keys', 'sorted_set_for_pinned_topics'] }, { version: "1.5.0", upgrades: ['flags_refactor'] } ] }; Upgrade.run = function (callback) { process.stdout.write('\nParsing upgrade scripts... '); var queue = []; var skipped = 0; // Retrieve list of upgrades that have already been run db.getSortedSetRange('schemaLog', 0, -1, function (err, completed) { if (err) { return callback(err); } queue = Upgrade.available.reduce(function (memo, cur) { cur.upgrades.forEach(function (filename) { if (completed.indexOf(filename) === -1) { memo.push(path.join(__dirname, './upgrades', filename)); } else { ++skipped; } }); return memo; }, queue); Upgrade.process(queue, skipped, callback); }); }; Upgrade.runSingle = function (query, callback) { process.stdout.write('\nParsing upgrade scripts... '); async.waterfall([ async.apply(utils.walk, path.join(__dirname, './upgrades')), function (files, next) { next(null, files.filter(function (file) { return file.search(new RegExp(query)) !== -1; })); } ], function (err, files) { if (err) { return callback(err); } Upgrade.process(files, 0, callback); }); }; Upgrade.process = function (files, skipCount, callback) { process.stdout.write('OK'.green + ' | '.reset + String(files.length).cyan + ' script(s) found'.cyan + (skipCount > 0 ? ', '.cyan + String(skipCount).cyan + ' skipped'.cyan : '') + '\n'.reset); // Do I need to sort the files here? we'll see. // sort(); async.eachSeries(files, function (file, next) { var scriptExport = require(file); var date = new Date(scriptExport.timestamp); process.stdout.write(' → '.white + String('[' + [date.getFullYear(), date.getMonth() + 1, date.getDate() + 1].join('/') + '] ').gray + String(scriptExport.name).reset + '... '); // Do the upgrade... scriptExport.method(function (err) { if (err) { process.stdout.write('error\n'.red); return next(err); } // Record success in schemaLog db.sortedSetAdd('schemaLog', Date.now(), path.basename(file, '.js')); process.stdout.write('OK\n'.green); next(); }); }, function (err) { if (err) { return callback(err); } process.stdout.write('Upgrade complete!\n\n'.green); callback(); }); }; module.exports = Upgrade;