From f4dcb4bb1dc3570559395016f11d0962e2e20117 Mon Sep 17 00:00:00 2001
From: liabru <liambrum@gmail.com>
Date: Sun, 23 Mar 2014 23:07:04 +0000
Subject: [PATCH] updated doc build task

---
 Gruntfile.js | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/Gruntfile.js b/Gruntfile.js
index b266393..9ef441d 100644
--- a/Gruntfile.js
+++ b/Gruntfile.js
@@ -138,10 +138,14 @@ module.exports = function(grunt) {
     }
   });
 
-  grunt.registerTask('doc', function() {
-    var dev = grunt.option('dev');
-    if (dev)
+  grunt.registerTask('doc', function(mode) {
+    var isDev = (mode === 'dev'),
+        isRelease = (mode === 'release'),
+        isEdge = (mode === 'edge');
+
+    if (isEdge)
       grunt.config.set('docVersion', 'edge version (master)');
+    
     grunt.task.run('yuidoc');
   });