Organize structure and use json

Author: sinwoobang

.gitignore

@@ -1 +1,2 @@
 *.dump
+*.json

index_constraint_migration.mjs

@@ -4,48 +4,74 @@ $.verbose = false;
 const log = console.log;
 const error = chalk.bold.red;
 
+async function init() {
+	log();
+	log(chalk.cyan("Welcome to dms-psql-post-data"));
+	log();
+}
 
-log();
-log(chalk.green("Welcome to dms-psql-post-data"));
-log();
+async function configure() {
+	log(chalk.yellow("Reading your DB info..."));
 
-log(chalk.yellow('Input your Source DB info'));
+	const dbInfo = await fs.readJson("./info.json");
+	const { srcDBHost, srcDBUser, srcDBName, srcDBPass } = dbInfo;
+	const { tarDBHost, tarDBUser, tarDBName, tarDBPass } = dbInfo;
 
-let srcDBHost = await question('[1/8] Source DB Host: ');
-let srcDBUser = await question('[2/8] Source DB User: ');
-let srcDBName = await question('[3/8] Source DB Name: ');
-let srcDBPass = await question('[4/8] Source DB Password: ');
+	log(chalk.green("Done"));
+	log();
 
-let tarDBHost = await question('[5/8] Target DB Host: ');
-let tarDBUser = await question('[6/8] Target DB User: ');
-let tarDBName = await question('[7/8] Target DB Name: ');
-let tarDBPass = await question('[8/8] Target DB Password: ');
+	log(chalk.cyan("Input the number of job workers"));
+	log(
+		chalk.cyan(
+			"NOTE: Increasing the number makes it faster, however it may cause DB locks."
+		)
+	);
 
-log('Input the number of job workers');
-log(chalk.yellow('NOTE: Increasing the number makes it faster, however it may cause DB locks.'));
+	let restoreJobWorkerNum = await question(chalk.cyan("Default value[1]: "));
+	restoreJobWorkerNum = restoreJobWorkerNum ? restoreJobWorkerNum : 1;
 
-let restoreJobWorkerNum = await question('Default value[1]: ');
-restoreJobWorkerNum = restoreJobWorkerNum ? restoreJobWorkerNum : 1;
+	return {
+		...dbInfo,
+		restoreJobWorkerNum,
+	};
+}
 
-process.env.PGPASSWORD = srcDBPass
-log(chalk.yellow('Dumping Source DB Post-data(Index, Constraint)...'));
+async function dumpPostData({ srcDBHost, srcDBUser, srcDBName }) {
+	process.env.PGPASSWORD = srcDBPass;
+	log(chalk.yellow("Dumping Source DB Post-data(Index, Constraint)..."));
 
-try {
-  await $`pg_dump -h ${srcDBHost} -U ${srcDBUser} -Fc ${srcDBName} --section=post-data > postdata.dump`;
-} catch (p) {
-  log(error('Exception detected: Is your Source DB information correct?'));
-  log(error(`Error: ${p.stderr}`));
+	try {
+		await $`pg_dump -h ${srcDBHost} -U ${srcDBUser} -Fc ${srcDBName} --section=post-data > postdata.dump`;
+	} catch (p) {
+		log(error("Exception detected: Is your Source DB information correct?"));
+		log(error(`Error: ${p.stderr}`));
+	}
 }
 
-process.env.PGPASSWORD = tarDBPass
-log(chalk.yellow('Restoring Post-data on Target DB...'));
+async function restorePostData({
+	tarDBHost,
+	tarDBUser,
+	tarDBName,
+	tarDBPass,
+	restoreJobWorkerNum,
+}) {
+	process.env.PGPASSWORD = tarDBPass;
+	log(chalk.yellow("Restoring Post-data on Target DB..."));
 
-try {
-  await $`pg_restore -h ${tarDBHost} -U ${tarDBUser} -d ${tarDBName} -j ${restoreJobWorkerNum} -v postdata.dump`;
-} catch (p) {
-  log(error('Exception detected: Is your Target DB information correct?'));
-  log(error(`Error: ${p.stderr}`));
+	try {
+		await $`pg_restore -h ${tarDBHost} -U ${tarDBUser} -d ${tarDBName} -j ${restoreJobWorkerNum} -v postdata.dump`;
+	} catch (p) {
+		log(error("Exception detected: Is your Target DB information correct?"));
+		log(error(`Error: ${p.stderr}`));
+	}
 }
 
+await init();
+
+const config = await configure();
+
+await dumpPostData(config);
+await restorePostData(config);
+
 log();
-log(chalk.yellow('Finished'))
+log(chalk.yellow("Finished"));

info.json

@@ -0,0 +1,10 @@
+{
+  "srcDBHost": "",
+  "srcDBName": "",
+  "srcDBUser": "",
+  "srcDBPass": "",
+  "tarDBHost": "",
+  "tarDBName": "",
+  "tarDBUser": "",
+  "tarDBPass": ""
+}