Add deployment grab command
This commit is contained in:
balthazarbk
@@ -0,0 +1,414 @@
|
|||||||
|
import { GluegunMenuToolbox } from '@lenne.tech/gluegun-menu';
|
||||||
|
import chalk = require('chalk');
|
||||||
|
const { Confirm } = require('enquirer');
|
||||||
|
import { Octokit } from '@octokit/core';
|
||||||
|
import { exit } from 'process';
|
||||||
|
import {
|
||||||
|
defaultMenuSettings,
|
||||||
|
getCurrentBranch,
|
||||||
|
getSettings,
|
||||||
|
sleep,
|
||||||
|
} from '../../../../globals';
|
||||||
|
import { getTestStacksInfo, Stack } from '../../../../common/testservers';
|
||||||
|
import { getIssueFromRepo } from '../../../../services/github_rest';
|
||||||
|
const { spawn } = require('child_process');
|
||||||
|
|
||||||
|
const EXCLUDED_NAMED_STACKS = ['photowall-test-ingrid', 'photowall-test-usa'];
|
||||||
|
const EXCLUDED_STATUSES = [
|
||||||
|
'UPDATE_FAILED',
|
||||||
|
'CREATE_FAILED',
|
||||||
|
'ROLLBACK_COMPLETE',
|
||||||
|
'UPDATE_ROLLBACK_FAILED',
|
||||||
|
'UPDATE_ROLLBACK_COMPLETE',
|
||||||
|
'UPDATE_IN_PROGRESS',
|
||||||
|
'CREATE_IN_PROGRESS',
|
||||||
|
];
|
||||||
|
const ONE_MONTH_SECONDS = 2592000;
|
||||||
|
|
||||||
|
type IssueState = 'open' | 'closed' | 'unknown';
|
||||||
|
type Candidate = {
|
||||||
|
stack: Stack;
|
||||||
|
tier: number;
|
||||||
|
reason: string;
|
||||||
|
ageSeconds: number;
|
||||||
|
};
|
||||||
|
|
||||||
|
const previewUrl = (stackName: string): string => {
|
||||||
|
const suffix = stackName.split('photowall-test-')[1];
|
||||||
|
return `https://test-${suffix}.photowall-test.se`;
|
||||||
|
};
|
||||||
|
|
||||||
|
module.exports = {
|
||||||
|
name: 'grab',
|
||||||
|
alias: ['g'],
|
||||||
|
description:
|
||||||
|
'Auto-pick a free testserver and update it with current branch (g)',
|
||||||
|
hidden: false,
|
||||||
|
run: async (toolbox: GluegunMenuToolbox) => {
|
||||||
|
const { system, print } = toolbox;
|
||||||
|
|
||||||
|
const jsonOutput = toolbox.parameters.options.json;
|
||||||
|
const skipConfirm =
|
||||||
|
toolbox.parameters.options['skip-confirm'] ||
|
||||||
|
toolbox.parameters.options.y ||
|
||||||
|
!!jsonOutput;
|
||||||
|
const dryRun = toolbox.parameters.options['dry-run'];
|
||||||
|
|
||||||
|
const fail = (message: string) => {
|
||||||
|
if (jsonOutput) {
|
||||||
|
console.log(JSON.stringify({ success: false, message }, null, 2));
|
||||||
|
} else {
|
||||||
|
print.error(message);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
const config = await getSettings(toolbox);
|
||||||
|
if (!config || !config.photowall_repo || config.photowall_repo === '') {
|
||||||
|
fail(
|
||||||
|
'The grab command requires that you set an absolute path to your local photowall repository.',
|
||||||
|
);
|
||||||
|
if (!jsonOutput) {
|
||||||
|
await toolbox.menu.showMenu('setup', defaultMenuSettings);
|
||||||
|
}
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const currentBranch = await getCurrentBranch(
|
||||||
|
config.photowall_repo,
|
||||||
|
toolbox,
|
||||||
|
);
|
||||||
|
if (!currentBranch) {
|
||||||
|
fail('Could not determine current git branch.');
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const octokit = new Octokit({ auth: config.gh_token });
|
||||||
|
try {
|
||||||
|
await octokit.request('GET /repos/{owner}/{repo}/branches/{branch}', {
|
||||||
|
owner: 'photowall',
|
||||||
|
repo: 'photowall',
|
||||||
|
branch: currentBranch,
|
||||||
|
});
|
||||||
|
} catch (e) {
|
||||||
|
fail(
|
||||||
|
`Branch ${currentBranch} is not on the photowall remote — push it first.`,
|
||||||
|
);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const testStacks = await getTestStacksInfo(toolbox);
|
||||||
|
|
||||||
|
const alreadyDeployed = testStacks.find((s) => s.branch === currentBranch);
|
||||||
|
if (alreadyDeployed) {
|
||||||
|
const msg = `Your branch ${currentBranch} is already on ${alreadyDeployed.name} — nothing to do.`;
|
||||||
|
if (jsonOutput) {
|
||||||
|
console.log(
|
||||||
|
JSON.stringify(
|
||||||
|
{
|
||||||
|
success: false,
|
||||||
|
message: msg,
|
||||||
|
stack: alreadyDeployed.name,
|
||||||
|
branch: currentBranch,
|
||||||
|
url: previewUrl(alreadyDeployed.name),
|
||||||
|
},
|
||||||
|
null,
|
||||||
|
2,
|
||||||
|
),
|
||||||
|
);
|
||||||
|
} else {
|
||||||
|
print.info(
|
||||||
|
`Your branch ${chalk.green(currentBranch)} is already on ${chalk.yellow(
|
||||||
|
alreadyDeployed.name,
|
||||||
|
)} — nothing to do.`,
|
||||||
|
);
|
||||||
|
print.info(`Live at ${chalk.cyan(previewUrl(alreadyDeployed.name))}`);
|
||||||
|
}
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
const issueStateByNumber = new Map<string, IssueState>();
|
||||||
|
await Promise.all(
|
||||||
|
testStacks.map(async (stack) => {
|
||||||
|
const match = stack.branch.match(/^(\d+)/);
|
||||||
|
const issueNumber = match ? match[1] : null;
|
||||||
|
if (!issueNumber || issueStateByNumber.has(issueNumber)) return;
|
||||||
|
const issueData = await getIssueFromRepo(
|
||||||
|
toolbox,
|
||||||
|
config.gh_token,
|
||||||
|
'photowall',
|
||||||
|
issueNumber,
|
||||||
|
);
|
||||||
|
issueStateByNumber.set(
|
||||||
|
issueNumber,
|
||||||
|
issueData ? (issueData.state as IssueState) : 'unknown',
|
||||||
|
);
|
||||||
|
}),
|
||||||
|
);
|
||||||
|
|
||||||
|
const now = Date.now();
|
||||||
|
const candidates: Candidate[] = [];
|
||||||
|
for (const stack of testStacks) {
|
||||||
|
if (EXCLUDED_NAMED_STACKS.includes(stack.name)) continue;
|
||||||
|
if (EXCLUDED_STATUSES.includes(stack.status)) continue;
|
||||||
|
|
||||||
|
const updatedTs = new Date(stack.updated).getTime();
|
||||||
|
const ageSeconds = isNaN(updatedTs)
|
||||||
|
? Number.POSITIVE_INFINITY
|
||||||
|
: (now - updatedTs) / 1000;
|
||||||
|
const match = stack.branch.match(/^(\d+)/);
|
||||||
|
const issueNumber = match ? match[1] : null;
|
||||||
|
const issueState: IssueState = issueNumber
|
||||||
|
? (issueStateByNumber.get(issueNumber) ?? 'unknown')
|
||||||
|
: 'unknown';
|
||||||
|
|
||||||
|
if (stack.branch === 'master') {
|
||||||
|
candidates.push({
|
||||||
|
stack,
|
||||||
|
tier: 1,
|
||||||
|
reason: 'branch is master',
|
||||||
|
ageSeconds,
|
||||||
|
});
|
||||||
|
} else if (issueState === 'closed') {
|
||||||
|
candidates.push({
|
||||||
|
stack,
|
||||||
|
tier: 2,
|
||||||
|
reason: `issue #${issueNumber} closed`,
|
||||||
|
ageSeconds,
|
||||||
|
});
|
||||||
|
} else if (ageSeconds > ONE_MONTH_SECONDS) {
|
||||||
|
candidates.push({
|
||||||
|
stack,
|
||||||
|
tier: 3,
|
||||||
|
reason: 'last commit >1 month ago',
|
||||||
|
ageSeconds,
|
||||||
|
});
|
||||||
|
} else if (issueState === 'unknown') {
|
||||||
|
candidates.push({
|
||||||
|
stack,
|
||||||
|
tier: 4,
|
||||||
|
reason: 'no linked issue in branch name',
|
||||||
|
ageSeconds,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (candidates.length === 0) {
|
||||||
|
if (jsonOutput) {
|
||||||
|
console.log(
|
||||||
|
JSON.stringify(
|
||||||
|
{
|
||||||
|
success: false,
|
||||||
|
message: 'No free testservers — all are claimed.',
|
||||||
|
},
|
||||||
|
null,
|
||||||
|
2,
|
||||||
|
),
|
||||||
|
);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
print.info('No free testservers — all are claimed. Current state:');
|
||||||
|
await require('../status/status').run(toolbox);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
candidates.sort((a, b) => {
|
||||||
|
if (a.tier !== b.tier) return a.tier - b.tier;
|
||||||
|
return b.ageSeconds - a.ageSeconds;
|
||||||
|
});
|
||||||
|
|
||||||
|
const pick = candidates[0];
|
||||||
|
const stackSuffix = pick.stack.name.replace('photowall-', '');
|
||||||
|
const url = previewUrl(pick.stack.name);
|
||||||
|
|
||||||
|
if (dryRun) {
|
||||||
|
if (jsonOutput) {
|
||||||
|
console.log(
|
||||||
|
JSON.stringify(
|
||||||
|
{
|
||||||
|
success: true,
|
||||||
|
dryRun: true,
|
||||||
|
stack: pick.stack.name,
|
||||||
|
branch: currentBranch,
|
||||||
|
reason: pick.reason,
|
||||||
|
url,
|
||||||
|
},
|
||||||
|
null,
|
||||||
|
2,
|
||||||
|
),
|
||||||
|
);
|
||||||
|
} else {
|
||||||
|
print.info(
|
||||||
|
`Would deploy ${chalk.green(currentBranch)} → ${chalk.yellow(
|
||||||
|
pick.stack.name,
|
||||||
|
)}. Reason: ${pick.reason}.`,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
let shouldRun = skipConfirm;
|
||||||
|
if (!skipConfirm) {
|
||||||
|
const confirm = new Confirm({
|
||||||
|
name: 'confirm',
|
||||||
|
message: `Will deploy ${chalk.green(currentBranch)} → ${chalk.yellow(
|
||||||
|
pick.stack.name,
|
||||||
|
)}. Reason: ${pick.reason}.`,
|
||||||
|
});
|
||||||
|
try {
|
||||||
|
shouldRun = await confirm.run();
|
||||||
|
} catch (e) {
|
||||||
|
shouldRun = false;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!shouldRun) {
|
||||||
|
if (!jsonOutput) print.info('Grab cancelled');
|
||||||
|
if (toolbox.fromMenu()) {
|
||||||
|
await toolbox.menu.showMenu(
|
||||||
|
'deployment testservers',
|
||||||
|
defaultMenuSettings,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
try {
|
||||||
|
const cmd = `make update-stack STACK_ENVIRONMENT_NAME=${stackSuffix} GITHUB_BRANCH=${currentBranch}`;
|
||||||
|
if (!jsonOutput) {
|
||||||
|
print.fancy(
|
||||||
|
`Running [${chalk.yellow(
|
||||||
|
cmd,
|
||||||
|
)}] in folder [${`${config.photowall_repo}/cloudformation/`}]`,
|
||||||
|
);
|
||||||
|
}
|
||||||
|
const child = spawn(
|
||||||
|
'make',
|
||||||
|
[
|
||||||
|
'update-stack',
|
||||||
|
`STACK_ENVIRONMENT_NAME=${stackSuffix}`,
|
||||||
|
`GITHUB_BRANCH=${currentBranch}`,
|
||||||
|
],
|
||||||
|
{
|
||||||
|
cwd: `${config.photowall_repo}/cloudformation/`,
|
||||||
|
},
|
||||||
|
);
|
||||||
|
child.stdout.setEncoding('utf8');
|
||||||
|
child.stdout.on('data', (data) => {
|
||||||
|
if (!jsonOutput) {
|
||||||
|
process.stdout.write(data);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
child.stderr.on('data', (data) => {
|
||||||
|
if (!jsonOutput) {
|
||||||
|
process.stderr.write(data);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
await new Promise((resolve, reject) => {
|
||||||
|
child.on('close', (code) => {
|
||||||
|
if (code === 0) {
|
||||||
|
resolve(0);
|
||||||
|
} else {
|
||||||
|
reject(new Error(`make exited with code ${code}`));
|
||||||
|
}
|
||||||
|
});
|
||||||
|
});
|
||||||
|
|
||||||
|
const spinner = jsonOutput
|
||||||
|
? null
|
||||||
|
: print.spin(`Waiting for stack to update`);
|
||||||
|
let currentStackStatus = '';
|
||||||
|
while (currentStackStatus !== 'UPDATE_COMPLETE') {
|
||||||
|
const stackCmd = `aws cloudformation describe-stacks --stack-name ${pick.stack.name}`;
|
||||||
|
const response = await system.run(stackCmd, { trim: true });
|
||||||
|
const stackData = JSON.parse(response);
|
||||||
|
currentStackStatus = stackData.Stacks[0].StackStatus;
|
||||||
|
|
||||||
|
if (!jsonOutput) {
|
||||||
|
let statusOutput = '';
|
||||||
|
switch (currentStackStatus) {
|
||||||
|
case 'UPDATE_COMPLETE_CLEANUP_IN_PROGRESS':
|
||||||
|
statusOutput = `[${chalk.yellow(
|
||||||
|
currentStackStatus,
|
||||||
|
)}]: Almost done, cleanup in progress`;
|
||||||
|
break;
|
||||||
|
case 'UPDATE_COMPLETE':
|
||||||
|
statusOutput = `[${chalk.green(
|
||||||
|
currentStackStatus,
|
||||||
|
)}]: Update complete`;
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
statusOutput = `[${chalk.red(
|
||||||
|
currentStackStatus,
|
||||||
|
)}]: Updating the stack`;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
spinner.text = statusOutput;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (currentStackStatus !== 'UPDATE_COMPLETE') {
|
||||||
|
await sleep(1000);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!jsonOutput) {
|
||||||
|
spinner.text = `Will start the pipeline now [${chalk.yellow(
|
||||||
|
`${pick.stack.name}Pipeline`,
|
||||||
|
)}]`;
|
||||||
|
}
|
||||||
|
|
||||||
|
await system.run(
|
||||||
|
`aws codepipeline start-pipeline-execution --name ${pick.stack.name}Pipeline`,
|
||||||
|
);
|
||||||
|
await sleep(2000);
|
||||||
|
|
||||||
|
if (jsonOutput) {
|
||||||
|
console.log(
|
||||||
|
JSON.stringify(
|
||||||
|
{
|
||||||
|
success: true,
|
||||||
|
message: `Successfully updated ${pick.stack.name} with branch ${currentBranch} and started pipeline`,
|
||||||
|
stack: pick.stack.name,
|
||||||
|
branch: currentBranch,
|
||||||
|
reason: pick.reason,
|
||||||
|
pipeline: `${pick.stack.name}Pipeline`,
|
||||||
|
url,
|
||||||
|
},
|
||||||
|
null,
|
||||||
|
2,
|
||||||
|
),
|
||||||
|
);
|
||||||
|
} else {
|
||||||
|
spinner.stop();
|
||||||
|
print.fancy('Pipeline started, bye! 🚀');
|
||||||
|
print.info(`Will be deployed soon on ${chalk.cyan(url)}`);
|
||||||
|
}
|
||||||
|
} catch (e) {
|
||||||
|
if (jsonOutput) {
|
||||||
|
console.log(
|
||||||
|
JSON.stringify(
|
||||||
|
{
|
||||||
|
success: false,
|
||||||
|
message: `Error during grab: ${e.message || e}`,
|
||||||
|
error: e.message || String(e),
|
||||||
|
},
|
||||||
|
null,
|
||||||
|
2,
|
||||||
|
),
|
||||||
|
);
|
||||||
|
} else {
|
||||||
|
console.log(`e`, e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (jsonOutput) return;
|
||||||
|
|
||||||
|
if (toolbox.fromMenu()) {
|
||||||
|
await toolbox.menu.showMenu(
|
||||||
|
'deployment testservers',
|
||||||
|
defaultMenuSettings,
|
||||||
|
);
|
||||||
|
} else {
|
||||||
|
exit();
|
||||||
|
}
|
||||||
|
},
|
||||||
|
};
|
||||||
Reference in New Issue
Block a user