wcandillon/grunt-swagger-js-codegen

module.exports = function (grunt) {
    'use strict';
    var fs = require('fs');
    var yaml = require('yamljs');

    grunt.registerMultiTask('swagger-js-codegen', 'Swagger codegen for Javascript', function(){
        var request = require('request');
        var Q = require('q');
        var _ = require('lodash');

        var options = this.options();
        var dest = options.dest;
        var promises = [];
        var done = this.async();

        grunt.file.mkdir(dest);
        options.apis.forEach(function(api){
            var deferred = Q.defer();
            var fname = api.fileName || (api.moduleName || api.className) + '.js';
            if(api.swagger.substring(0, 'http://'.length) === 'http://' || api.swagger.substring(0, 'https://'.length) === 'https://') {
                request({
                    uri: api.swagger,
                    method: 'GET'
                }, function(error, response, body){
                    if(error || response.statusCode !== 200) {
                        deferred.reject('Error while fetching ' + api.swagger + ': ' + (error || body));
                    } else {
                        var swagger = JSON.parse(body);
                        generateApi(api, swagger, dest, fname);
                        deferred.resolve();
                    }
                });
            } else if(testIfYamlExtension(api.swagger)) {
                fs.readFile(api.swagger, 'UTF-8', function(err, data) {
                    if(err) {
                        deferred.reject(err);
                    } else {
                        var swagger = yaml.parse(data);
                        generateApi(api, swagger, dest, fname);
                        deferred.resolve();
                    }
                });
            } else {
                fs.readFile(api.swagger, 'UTF-8', function(err, data){
                    if(err) {
                        deferred.reject(err);
                    } else {
                        var swagger = JSON.parse(data);
                        generateApi(api, swagger, dest, fname);
                        deferred.resolve();
                    }
                });
            }
            promises.push(deferred.promise);
        });
        Q.all(promises).then(function(){
            done();
        }).catch(function(error){
            var e;
            if(_.isObject(error) && error.body) {
                e = JSON.stringify(error.body, null, 2);
            } else if(_.isObject(error)) {
                e = JSON.stringify(error, null, 2);
            } else if(error instanceof Error || _.isString(error)) {
                e = error;
            } else {
                e = new Error('Unknown error');
            }
            grunt.fail.fatal(e);
        });
    });

    function testIfYamlExtension(filename) {
        return filename.substring(filename.length - '.yaml'.length).toLowerCase() === '.yaml' || filename.substring(filename.length - '.yml'.length).toLowerCase() === '.yml';
    }

    function generateApi(api, swagger, dest, fname) {
        var CodeGen = require('swagger-js-codegen').CodeGen;
        var source = null;

        if (api.type === 'angular' || api.angularjs === true) {
            source = CodeGen.getAngularCode({ moduleName: api.moduleName, className: api.className, swagger: swagger });
        } else if (api.custom === true) {
            source = CodeGen.getCustomCode({ className: api.className, swagger: swagger, mustache: api.mustache,
                template: {
                    class: fs.readFileSync(api.template.class, 'utf-8'),
                    method: fs.readFileSync(api.template.method, 'utf-8'),
                    request: fs.readFileSync(api.template.request, 'utf-8')
                }
            });
        } else {
            source = CodeGen.getNodeCode({ className: api.className, swagger: swagger });
        }
        grunt.log.writeln('Generated ' + fname + ' from ' + api.swagger);
        fs.writeFileSync(dest + '/' + fname, source, 'UTF-8');
    }
};