From 24819804aa2d37dc8b0d3e8220b285f2766a9aca Mon Sep 17 00:00:00 2001
From: David Calavera <david.calavera@gmail.com>
Date: Thu, 1 Sep 2016 07:59:24 -0700
Subject: [PATCH] Add deploy preview context.

So we can see content from drafts and the future in deploy previews.

Signed-off-by: David Calavera <david.calavera@gmail.com>
---
 gulpfile.babel.js | 24 +++++++++++++-----------
 netlify.toml      |  7 +++++--
 package.json      |  1 +
 3 files changed, 19 insertions(+), 13 deletions(-)

diff --git a/gulpfile.babel.js b/gulpfile.babel.js
index 62e65d0..b3c942c 100644
--- a/gulpfile.babel.js
+++ b/gulpfile.babel.js
@@ -10,20 +10,13 @@ import webpackConfig from "./webpack.conf";
 
 const browserSync = BrowserSync.create();
 const hugoBin = "hugo";
+const defaultArgs = ["-d", "../dist", "-s", "site", "-v"];
 
-gulp.task("hugo", (cb) => {
-  const args = ["-d", "../dist", "-s", "site", "-v"];
-  return cp.spawn(hugoBin, args, {stdio: "inherit"}).on("close", (code) => {
-    if (code === 0) {
-      browserSync.reload();
-    } else {
-      browserSync.notify("Hugo build failed :(");
-    }
-    cb();
-  });
-});
+gulp.task("hugo", (cb) => buildSite(cb));
+gulp.task("hugo-preview", (cb) => buildSite(cb, ["--buildDrafts", "--buildFuture"]));
 
 gulp.task("build", ["css", "js", "hugo"]);
+gulp.task("build-preview", ["css", "js", "hugo-preview"]);
 
 gulp.task("css", () => (
   gulp.src("./src/css/*.css")
@@ -56,3 +49,12 @@ gulp.task("server", ["hugo", "css", "js"], () => {
   gulp.watch("./src/css/**/*.css", ["css"]);
   gulp.watch("./site/**/*", ["hugo"]);
 });
+
+function buildSite(cb, options) {
+  const args = options ? defaultArgs.concat(options) : defaultArgs;
+
+  return cp.spawn(hugoBin, args, {stdio: "inherit"}).on("close", () => {
+    browserSync.reload();
+    cb();
+  });
+}
diff --git a/netlify.toml b/netlify.toml
index efcfd06..dd1ef8e 100644
--- a/netlify.toml
+++ b/netlify.toml
@@ -1,3 +1,6 @@
 [build]
-command = "npm run build"
-publish = "dist"
+  command = "npm run build"
+  publish = "dist"
+
+[context.deploy-preview]
+  command = "npm run build-preview"
diff --git a/package.json b/package.json
index 67a3ab0..f82f02b 100644
--- a/package.json
+++ b/package.json
@@ -7,6 +7,7 @@
     "hugo": "gulp hugo",
     "webpack": "gulp webpack",
     "build": "gulp build",
+    "build-preview": "gulp build-preview",
     "start": "gulp server",
     "lint": "eslint src"
   },