feat: Add "push" option.

darkobits · Aug 28, 2019 · d7bb087 · d7bb087
1 parent b8ee335
commit d7bb087
Show file tree

Hide file tree

Showing 3 changed files with 46 additions and 2 deletions.
diff --git a/src/bin/cli.ts b/src/bin/cli.ts
@@ -72,6 +72,14 @@ yargs.command({
       conflicts: ['dockerfile']
     });
 
+    command.option('push', {
+      group: 'Optional Arguments:',
+      description: 'Whether to push images to a registry.',
+      required: false,
+      type: 'boolean',
+      default: false
+    });
+
     command.example('$0', 'Dockerize the NodeJS project in the current directory using default options.');
     command.example('$0 --label="foo=bar" --label="baz=qux" --extra-args="--squash"', 'Dockerize the NodeJS project in the current directory, apply two labels, and pass the --squash argument to Docker.');
 

diff --git a/src/etc/types.ts b/src/etc/types.ts
@@ -57,4 +57,11 @@ export interface DockerizeArguments extends Arguments {
    * This file will be removed from the final image.
    */
   npmrc?: string;
+
+  /**
+   * Whether to run 'docker push' after building an image.
+   *
+   * Note: This option assumes 'docker login' has already been run.
+   */
+  push?: boolean;
 }
diff --git a/src/lib/dockerize.ts b/src/lib/dockerize.ts
@@ -34,6 +34,7 @@ export default async function dockerize(options: DockerizeArguments) {
   ow(options.extraArgs, 'extra Docker arguments', ow.any(ow.undefined, ow.string));
   ow(options.dockerfile, 'custom Dockerfile', ow.any(ow.undefined, ow.string));
   ow(options.npmrc, '.npmrc file', ow.any(ow.undefined, ow.string));
+  ow(options.push, 'push', ow.any(ow.undefined, ow.boolean));
 
 
   // ----- [2] Prepare Staging Area --------------------------------------------
@@ -181,7 +182,7 @@ export default async function dockerize(options: DockerizeArguments) {
 
   const buildTime = log.createTimer();
   const spinner = log.createSpinner();
-  const endInteractive = log.beginInteractive(() => log.info(`${spinner} Building image ${log.chalk.cyan.bold(tag)}...`));
+  const endBuildInteractive = log.beginInteractive(() => log.info(`${spinner} Building image ${log.chalk.cyan.bold(tag)}...`));
 
   await Promise.all([
     // Copy production-relevant package files to the staging directory.
@@ -218,5 +219,33 @@ export default async function dockerize(options: DockerizeArguments) {
     fs.remove(stagingDir)
   ]);
 
-  endInteractive(() => log.info(`🏁 Built image ${log.chalk.cyan.bold(tag)} ${log.chalk.dim(`(${imageSize})`)} in ${buildTime}.`));
+  endBuildInteractive(() => log.info(`🏁 Built image ${log.chalk.cyan.bold(tag)} ${log.chalk.dim(`(${imageSize})`)} in ${buildTime}.`));
+
+
+  // ----- [11] (Optional) Push Image ------------------------------------------
+
+  if (!options.push) {
+    return;
+  }
+
+  const pushTime = log.createTimer();
+  const endPushInteractive = log.beginInteractive(() => log.info(`${spinner} Pushing image ${log.chalk.cyan.bold(tag)}...`));
+
+  const pushProcess = execa('docker', ['push', tag], {
+    stdin: 'ignore',
+    stdout: log.isLevelAtLeast('silly') ? 'pipe' : 'ignore',
+    stderr: log.isLevelAtLeast('silly') ? 'pipe' : 'ignore',
+  });
+
+  if (pushProcess.stdout) {
+    pushProcess.stdout.pipe(log.createPipe('silly'));
+  }
+
+  if (pushProcess.stderr) {
+    pushProcess.stderr.pipe(log.createPipe('silly'));
+  }
+
+  await pushProcess;
+
+  endPushInteractive(() => log.info(`🚀 Pushed image ${log.chalk.cyan.bold(tag)} in ${pushTime}.`));
 }