shellphone.app/db/backup.ts
2021-10-13 21:42:23 +02:00

130 lines
3.6 KiB
TypeScript

import url from "url";
import querystring from "querystring";
import { spawn } from "child_process";
import { PassThrough } from "stream";
import { sendEmail } from "integrations/aws-ses";
import { s3 } from "integrations/aws-s3";
export default async function backup(schedule: "daily" | "weekly" | "monthly") {
const s3Bucket = `shellphone-${schedule}-backup`;
const { database, host, port, user, password } = parseDatabaseUrl(process.env.DATABASE_URL!);
const fileName = getFileName(database);
console.log(`Dumping database ${database}`);
const pgDumpChild = spawn("pg_dump", [`-U${user}`, `-d${database}`], {
env: {
...process.env,
PGPASSWORD: password,
PGHOST: host,
PGPORT: port.toString(),
},
stdio: ["ignore", "pipe", "inherit"],
});
console.log(`Compressing dump "${fileName}"`);
const gzippedDumpStream = new PassThrough();
const gzipChild = spawn("gzip", { stdio: ["pipe", "pipe", "inherit"] });
gzipChild.on("exit", (code) => {
if (code !== 0) {
return sendEmail({
body: `${schedule} backup failed: gzip: Bad exit code (${code})`,
subject: `${schedule} backup failed: gzip: Bad exit code (${code})`,
recipients: ["error@shellphone.app"],
});
}
});
pgDumpChild.stdout.pipe(gzipChild.stdin);
gzipChild.stdout.pipe(gzippedDumpStream);
pgDumpChild.on("exit", (code) => {
if (code !== 0) {
console.log("pg_dump failed, upload aborted");
return sendEmail({
body: `${schedule} backup failed: pg_dump: Bad exit code (${code})`,
subject: `${schedule} backup failed: pg_dump: Bad exit code (${code})`,
recipients: ["error@shellphone.app"],
});
}
console.log(`Uploading "${fileName}" to S3 bucket "${s3Bucket}"`);
const uploadPromise = s3
.upload({
Bucket: s3Bucket,
Key: fileName,
ACL: "private",
ContentType: "text/plain",
ContentEncoding: "gzip",
Body: gzippedDumpStream,
})
.promise();
uploadPromise
.then(() => console.log(`Successfully uploaded "${fileName}"`))
.catch((error) =>
sendEmail({
body: `${schedule} backup failed: ${error}`,
subject: `${schedule} backup failed: ${error}`,
recipients: ["error@shellphone.app"],
}),
);
});
}
function pad(number: number) {
return number.toString().padStart(2, "0");
}
function getFileName(database: string) {
const now = new Date();
const year = now.getUTCFullYear();
const month = pad(now.getUTCMonth() + 1);
const day = pad(now.getUTCDate());
const hours = pad(now.getUTCHours());
const minutes = pad(now.getUTCMinutes());
const seconds = pad(now.getUTCSeconds());
return `${database}-${year}-${month}-${day}_${hours}-${minutes}-${seconds}.sql.gz`; // 2021-09-15_16-00-02.sql.gz
}
type DatabaseUrl = {
readonly user: string;
readonly password: string;
readonly host: string;
readonly port: number;
readonly database: string;
};
function parseDatabaseUrl(databaseUrl: string): DatabaseUrl {
const parsedUrl = url.parse(databaseUrl, false, true);
const config = querystring.parse(parsedUrl.query!);
if (parsedUrl.auth) {
const userPassword = parsedUrl.auth.split(":", 2);
config.user = userPassword[0];
if (userPassword.length > 1) {
config.password = userPassword[1];
}
}
if (parsedUrl.pathname) {
config.database = parsedUrl.pathname.replace(/^\//, "").replace(/\/$/, "");
}
if (parsedUrl.hostname) {
config.host = parsedUrl.hostname;
}
if (parsedUrl.port) {
config.port = parsedUrl.port;
}
return {
user: config.user as string,
password: config.password as string,
host: config.host as string,
port: Number.parseInt(config.port as string, 10),
database: config.database as string,
};
}