feat(decorator): Add prompt if file already exists

Author: robinboehm

decorator/USAGE

@@ -6,12 +6,3 @@ Example:
 
     This will create:
         app/scripts/decorators/serviceNameDecorator.js
-
-
-    If you want to use multiple decorators for one service,
-    you are able to use the second argument for the filename
-
-    yo angular:decorator serviceName serviceNameLogging [--coffee]
-
-    This will create:
-            app/scripts/decorators/serviceNameLoggingDecorator.js

decorator/index.js

@@ -1,30 +1,74 @@
 'use strict';
-var path = require('path');
 var util = require('util');
 var ScriptBase = require('../script-base.js');
-var angularUtils = require('../util.js');
+var fs = require('fs');
 
+var Generator = module.exports = function Generator(args, options) {
+    ScriptBase.apply(this, arguments);
+    this.fileName = this.name;
+}
 
-module.exports = Generator;
+util.inherits(Generator, ScriptBase);
 
-function Generator() {
-    ScriptBase.apply(this, arguments);
-    var postFix = "Decorator";
+Generator.prototype.askForOverwrite = function askForOverwrite() {
+    var cb = this.async();
+
+    // TODO: Any yeoman.util function to handle this?
+    var fileExists = fs.existsSync(this.env.cwd+'/app/scripts/'+buildRelativePath(this.fileName)+".js");
+    if (fileExists) {
+        var prompts = [{
+            name: 'overwriteDecorator',
+            message: 'Would you like to overwrite existing decorator?',
+            default: 'Y/n',
+            warning: 'Yes: Decorator will be replaced..'
+        }];
 
-    //TODO: Any better way in yeoman to get this value?
-    var fileName = arguments[0][1];
-    if(fileName === undefined){
-        fileName = this.name+postFix;
+        this.prompt(prompts, function (err, props) {
+            if (err) {
+                return this.emit('error', err);
+            }
+
+            this.overwriteDecorator = (/y/i).test(props.overwriteDecorator);
+
+            cb();
+        }.bind(this));
     }
     else{
-        fileName += postFix;
+        cb();
+        return;
     }
-    this.fileName = fileName;
-}
+};
 
-util.inherits(Generator, ScriptBase);
+Generator.prototype.askForNewName = function askForNewName() {
+    var cb = this.async();
+
+    if(this.overwriteDecorator === undefined || this.overwriteDecorator === true){
+        cb();
+        return;
+    }
+    else{
+        var prompts = new Array();
+        prompts.push({
+            name: 'decortatorName',
+            message: 'Alternative name for the decorator:'
+        });
+
+        this.prompt(prompts, function (err, props) {
+            if (err) {
+                return this.emit('error', err);
+            }
+            this.fileName = props.decortatorName;
+
+            cb();
+        }.bind(this));
+    }
+};
 
 Generator.prototype.createDecoratorFiles = function createDecoratorFiles() {
-    this.appTemplate('decorator', 'scripts/decorators/' + this.fileName);
-    this.addScriptToIndex('decorators/' + this.fileName);
+    this.appTemplate('decorator', 'scripts/'+buildRelativePath(this.fileName));
+    this.addScriptToIndex(buildRelativePath(this.fileName));
 };
+
+function buildRelativePath(fileName){
+    return 'decorators/' + fileName+"Decorator";
+}