420 lines
12 KiB
JavaScript
420 lines
12 KiB
JavaScript
|
|
||
|
|
const MyVC = require('./index');
|
||
|
|
const fs = require('fs-extra');
|
||
|
|
|
||
|
|
const typeMap = {
|
||
|
|
events: {
|
||
|
|
name: 'EVENT',
|
||
|
|
abbr: 'EVNT',
|
||
|
|
color: 'cyan'
|
||
|
|
},
|
||
|
|
functions: {
|
||
|
|
name: 'FUNCTION',
|
||
|
|
abbr: 'FUNC',
|
||
|
|
color: 'cyan'
|
||
|
|
},
|
||
|
|
procedures: {
|
||
|
|
name: 'PROCEDURE',
|
||
|
|
abbr: 'PROC',
|
||
|
|
color: 'yellow'
|
||
|
|
},
|
||
|
|
triggers: {
|
||
|
|
name: 'TRIGGER',
|
||
|
|
abbr: 'TRIG',
|
||
|
|
color: 'blue'
|
||
|
|
},
|
||
|
|
views: {
|
||
|
|
name: 'VIEW',
|
||
|
|
abbr: 'VIEW',
|
||
|
|
color: 'magenta'
|
||
|
|
},
|
||
|
|
};
|
||
|
|
|
||
|
|
class Routine {
|
||
|
|
construct(path, mark) {
|
||
|
|
const path = path
|
||
|
|
const split = path.split('/');
|
||
|
|
|
||
|
|
const fullPath = `${this.opts.workspace}/routines/${path}.sql`;
|
||
|
|
const schema = split[0];
|
||
|
|
const type = typeMap[split[1]];
|
||
|
|
const name = split[2];
|
||
|
|
|
||
|
|
Object.assign(this, {
|
||
|
|
path,
|
||
|
|
mark: mark,
|
||
|
|
exists: await fs.pathExists(fullPath),
|
||
|
|
type,
|
||
|
|
schema,
|
||
|
|
name,
|
||
|
|
fullName: `${schema}.${name}`,
|
||
|
|
isRoutine: ['FUNC', 'PROC'].indexOf(type.abbr) !== -1
|
||
|
|
});
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
const tokens = {
|
||
|
|
string: {
|
||
|
|
start: '\'',
|
||
|
|
end: '\'',
|
||
|
|
escape: char => char == '\'' || char == '\\'
|
||
|
|
},
|
||
|
|
id: {
|
||
|
|
start: '`',
|
||
|
|
end: '`',
|
||
|
|
escape: char => char == '`'
|
||
|
|
},
|
||
|
|
multiComment: {
|
||
|
|
start: '/*',
|
||
|
|
end: '*/',
|
||
|
|
escape: () => false
|
||
|
|
},
|
||
|
|
singleComment: {
|
||
|
|
start: '-- ',
|
||
|
|
end: '\n',
|
||
|
|
escape: () => false
|
||
|
|
}
|
||
|
|
};
|
||
|
|
|
||
|
|
const tokenIndex = new Map();
|
||
|
|
for (const tokenId in tokens) {
|
||
|
|
const token = tokens[tokenId];
|
||
|
|
tokenIndex.set(token.start[0], token);
|
||
|
|
}
|
||
|
|
|
||
|
|
class Push {
|
||
|
|
get myOpts() {
|
||
|
|
return {
|
||
|
|
alias: {
|
||
|
|
force: 'f',
|
||
|
|
user: 'u',
|
||
|
|
applyUncommited: 'a'
|
||
|
|
}
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
async run(myvc, opts) {
|
||
|
|
const conn = await myvc.dbConnect();
|
||
|
|
this.conn = conn;
|
||
|
|
|
||
|
|
const version = await myvc.fetchDbVersion() || {};
|
||
|
|
|
||
|
|
console.log(
|
||
|
|
`Database information:`
|
||
|
|
+ `\n -> Version: ${version.number}`
|
||
|
|
+ `\n -> Commit: ${version.gitCommit}`
|
||
|
|
);
|
||
|
|
|
||
|
|
if (!/^[0-9]*$/.test(version.number))
|
||
|
|
throw new Error('Wrong database version');
|
||
|
|
if (!version.number)
|
||
|
|
version.number = '00000';
|
||
|
|
|
||
|
|
if (opts.user) {
|
||
|
|
const [[user]] = conn.query(
|
||
|
|
`SELECT LEFT(USER(), INSTR(USER(), '@') - 1)`
|
||
|
|
);
|
||
|
|
let [[userVersion]] = await conn.query(
|
||
|
|
`SELECT number, gitCommit
|
||
|
|
FROM versionUser
|
||
|
|
WHERE code = ? AND user = ?`,
|
||
|
|
[opts.code, user]
|
||
|
|
);
|
||
|
|
userVersion = userVersion || {};
|
||
|
|
console.log(
|
||
|
|
`User information:`
|
||
|
|
+ `\n -> User: ${user}`
|
||
|
|
+ `\n -> Version: ${userVersion.number}`
|
||
|
|
+ `\n -> Commit: ${userVersion.gitCommit}`
|
||
|
|
);
|
||
|
|
|
||
|
|
if (userVersion.number > version.number)
|
||
|
|
version = userVersion;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (opts.env == 'production') {
|
||
|
|
console.log(
|
||
|
|
'\n ( ( ) ( ( ) ) '
|
||
|
|
+ '\n )\\ ))\\ ) ( /( )\\ ) ( ))\\ ) ( /( ( /( '
|
||
|
|
+ '\n(()/(()/( )\\()|()/( ( )\\ ) /(()/( )\\()) )\\())'
|
||
|
|
+ '\n /(_))(_)|(_)\\ /(_)) )\\ (((_) ( )(_))(_)|(_)\\ ((_)\\ '
|
||
|
|
+ '\n(_))(_)) ((_|_))_ _ ((_))\\___(_(_()|__)) ((_) _((_)'
|
||
|
|
+ '\n| _ \\ _ \\ / _ \\| \\| | | ((/ __|_ _|_ _| / _ \\| \\| |'
|
||
|
|
+ '\n| _/ /| (_) | |) | |_| || (__ | | | | | (_) | . |'
|
||
|
|
+ '\n|_| |_|_\\ \\___/|___/ \\___/ \\___| |_| |___| \\___/|_|\\_|'
|
||
|
|
+ '\n'
|
||
|
|
);
|
||
|
|
|
||
|
|
if (!opts.force) {
|
||
|
|
const readline = require('readline');
|
||
|
|
const rl = readline.createInterface({
|
||
|
|
input: process.stdin,
|
||
|
|
output: process.stdout
|
||
|
|
});
|
||
|
|
const answer = await new Promise(resolve => {
|
||
|
|
rl.question('Are you sure? (Default: no) [yes|no] ', resolve);
|
||
|
|
});
|
||
|
|
rl.close();
|
||
|
|
|
||
|
|
if (answer !== 'yes')
|
||
|
|
throw new Error('Changes aborted');
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
console.log('Applying versions.');
|
||
|
|
|
||
|
|
let nChanges = 0;
|
||
|
|
const versionsDir = `${opts.workspace}/versions`;
|
||
|
|
|
||
|
|
function logVersion(type, version, name) {
|
||
|
|
console.log('', type.bold, `[${version.bold}]`, name);
|
||
|
|
}
|
||
|
|
|
||
|
|
if (await fs.pathExists(versionsDir)) {
|
||
|
|
const versionDirs = await fs.readdir(versionsDir);
|
||
|
|
|
||
|
|
for (const versionDir of versionDirs) {
|
||
|
|
if (versionDir == 'README.md')
|
||
|
|
continue;
|
||
|
|
|
||
|
|
const match = versionDir.match(/^([0-9]{5})-([a-zA-Z0-9]+)?$/);
|
||
|
|
if (!match) {
|
||
|
|
logVersion('[W]'.yellow, '?????', versionDir);
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
const dirVersion = match[1];
|
||
|
|
const versionName = match[2];
|
||
|
|
|
||
|
|
if (version.number >= dirVersion) {
|
||
|
|
logVersion('[I]'.blue, dirVersion, versionName);
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
logVersion('[+]'.green, dirVersion, versionName);
|
||
|
|
const scriptsDir = `${versionsDir}/${versionDir}`;
|
||
|
|
const scripts = await fs.readdir(scriptsDir);
|
||
|
|
|
||
|
|
for (const script of scripts) {
|
||
|
|
if (!/^[0-9]{2}-[a-zA-Z0-9_]+\.sql$/.test(script)) {
|
||
|
|
console.log(` - Ignoring wrong file name: ${script}`);
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
console.log(` - ${script}`);
|
||
|
|
await this.queryFromFile(pushConn, `${scriptsDir}/${script}`);
|
||
|
|
nChanges++;
|
||
|
|
}
|
||
|
|
|
||
|
|
//await this.updateVersion(nChanges, 'number', dirVersion);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
const pushConn = await myvc.createConnection();
|
||
|
|
|
||
|
|
console.log('Applying changed routines.');
|
||
|
|
|
||
|
|
let nRoutines = 0;
|
||
|
|
let changes = await fs.pathExists(`${opts.workspace}/.git`)
|
||
|
|
? await myvc.changedRoutines(version.gitCommit)
|
||
|
|
: await myvc.cachedChanges();
|
||
|
|
changes = await this.parseChanges(changes);
|
||
|
|
|
||
|
|
await conn.query(
|
||
|
|
`CREATE TEMPORARY TABLE tProcsPriv
|
||
|
|
ENGINE = MEMORY
|
||
|
|
SELECT * FROM mysql.procs_priv LIMIT 0`
|
||
|
|
);
|
||
|
|
|
||
|
|
const routines = [];
|
||
|
|
for (const change of changes)
|
||
|
|
if (change.isRoutine)
|
||
|
|
routines.push([change.schema, change.name]);
|
||
|
|
|
||
|
|
if (routines.length) {
|
||
|
|
await conn.query(
|
||
|
|
`DROP TEMPORARY TABLE IF EXISTS tProcsPriv`
|
||
|
|
);
|
||
|
|
await conn.query(
|
||
|
|
`CREATE TEMPORARY TABLE tProcsPriv
|
||
|
|
ENGINE = MEMORY
|
||
|
|
SELECT * FROM mysql.procs_priv
|
||
|
|
WHERE (Db, Routine_name) IN (?)`,
|
||
|
|
[routines]
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
for (const change of changes) {
|
||
|
|
const actionMsg = change.exists ? '[+]'.green : '[-]'.red;
|
||
|
|
const typeMsg = `[${change.type.abbr}]`[change.type.color];
|
||
|
|
|
||
|
|
console.log('', actionMsg.bold, typeMsg.bold, change.fullName);
|
||
|
|
|
||
|
|
if (change.exists)
|
||
|
|
await this.queryFromFile(pushConn, `routines/${change.path}.sql`);
|
||
|
|
else {
|
||
|
|
const escapedName =
|
||
|
|
conn.escapeId(change.schema, true) + '.' +
|
||
|
|
conn.escapeId(change.name, true);
|
||
|
|
|
||
|
|
const query = `DROP ${change.type.name} IF EXISTS ${escapedName}`;
|
||
|
|
await conn.query(query);
|
||
|
|
}
|
||
|
|
|
||
|
|
nRoutines++;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (routines.length) {
|
||
|
|
await conn.query(
|
||
|
|
`INSERT IGNORE INTO mysql.procs_priv
|
||
|
|
SELECT * FROM tProcsPriv`
|
||
|
|
);
|
||
|
|
await conn.query(
|
||
|
|
`DROP TEMPORARY TABLE tProcsPriv`
|
||
|
|
);
|
||
|
|
}
|
||
|
|
|
||
|
|
await pushConn.end();
|
||
|
|
|
||
|
|
if (nRoutines > 0) {
|
||
|
|
await conn.query('FLUSH PRIVILEGES');
|
||
|
|
//await this.updateVersion(nRoutines, 'gitCommit', version.gitCommit);
|
||
|
|
|
||
|
|
console.log(` -> ${nRoutines} routines have changed.`);
|
||
|
|
} else
|
||
|
|
console.log(` -> No routines changed.`);
|
||
|
|
}
|
||
|
|
|
||
|
|
async parseChanges(changes) {
|
||
|
|
const routines = [];
|
||
|
|
for (const change of changes)
|
||
|
|
routines.push(new Routine(change));
|
||
|
|
return routines;
|
||
|
|
}
|
||
|
|
|
||
|
|
async updateVersion(nChanges, column, value) {
|
||
|
|
if (nChanges == 0) return;
|
||
|
|
const {opts} = this;
|
||
|
|
|
||
|
|
column = this.conn.escapeId(column, true);
|
||
|
|
|
||
|
|
if (opts.user) {
|
||
|
|
await this.conn.query(
|
||
|
|
`INSERT INTO versionUser
|
||
|
|
SET code = ?,
|
||
|
|
user = ?,
|
||
|
|
${column} = ?
|
||
|
|
ON DUPLICATE KEY UPDATE
|
||
|
|
${column} = VALUES(${column})`,
|
||
|
|
[
|
||
|
|
opts.code,
|
||
|
|
user,
|
||
|
|
value
|
||
|
|
]
|
||
|
|
);
|
||
|
|
} else {
|
||
|
|
await this.conn.query(
|
||
|
|
`INSERT INTO version
|
||
|
|
SET code = ?,
|
||
|
|
${column} = ?
|
||
|
|
ON DUPLICATE KEY UPDATE
|
||
|
|
${column} = VALUES(${column})`,
|
||
|
|
[
|
||
|
|
opts.code,
|
||
|
|
value
|
||
|
|
]
|
||
|
|
);
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Executes an SQL script.
|
||
|
|
*
|
||
|
|
* @param {String} file Path to the SQL script
|
||
|
|
* @returns {Array<Result>} The resultset
|
||
|
|
*/
|
||
|
|
async queryFromFile(conn, file) {
|
||
|
|
let results = [];
|
||
|
|
const stmts = this.querySplit(await fs.readFile(file, 'utf8'));
|
||
|
|
|
||
|
|
for (const stmt of stmts)
|
||
|
|
results = results.concat(await conn.query(stmt));
|
||
|
|
|
||
|
|
return results;
|
||
|
|
}
|
||
|
|
|
||
|
|
/**
|
||
|
|
* Splits an SQL muti-query into a single-query array, it does an small
|
||
|
|
* parse to correctly handle the DELIMITER statement.
|
||
|
|
*
|
||
|
|
* @param {Array<String>} stmts The splitted SQL statements
|
||
|
|
*/
|
||
|
|
querySplit(sql) {
|
||
|
|
const stmts = [];
|
||
|
|
let i,
|
||
|
|
char,
|
||
|
|
token,
|
||
|
|
escaped,
|
||
|
|
stmtStart;
|
||
|
|
|
||
|
|
let delimiter = ';';
|
||
|
|
const delimiterRe = /\s*delimiter\s+(\S+)[^\S\r\n]*(?:\r?\n|\r)/yi;
|
||
|
|
|
||
|
|
function begins(str) {
|
||
|
|
let j;
|
||
|
|
for (j = 0; j < str.length; j++)
|
||
|
|
if (sql[i + j] != str[j])
|
||
|
|
return false;
|
||
|
|
i += j;
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
for (i = 0; i < sql.length;) {
|
||
|
|
stmtStart = i;
|
||
|
|
|
||
|
|
delimiterRe.lastIndex = i;
|
||
|
|
const match = sql.match(delimiterRe);
|
||
|
|
if (match) {
|
||
|
|
delimiter = match[1];
|
||
|
|
i += match[0].length;
|
||
|
|
continue;
|
||
|
|
}
|
||
|
|
|
||
|
|
while (i < sql.length) {
|
||
|
|
char = sql[i];
|
||
|
|
|
||
|
|
if (token) {
|
||
|
|
if (!escaped && begins(token.end))
|
||
|
|
token = null;
|
||
|
|
else {
|
||
|
|
escaped = !escaped && token.escape(char);
|
||
|
|
i++;
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
if (begins(delimiter)) break;
|
||
|
|
|
||
|
|
const tok = tokenIndex.get(char);
|
||
|
|
if (tok && begins(tok.start))
|
||
|
|
token = tok;
|
||
|
|
else
|
||
|
|
i++;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
const len = i - stmtStart - delimiter.length;
|
||
|
|
stmts.push(sql.substr(stmtStart, len));
|
||
|
|
}
|
||
|
|
|
||
|
|
const len = stmts.length;
|
||
|
|
if (len > 1 && /^\s*$/.test(stmts[len - 1]))
|
||
|
|
stmts.pop();
|
||
|
|
|
||
|
|
return stmts;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
module.exports = Push;
|
||
|
|
|
||
|
|
if (require.main === module)
|
||
|
|
new MyVC().run(Push);
|