ansible-playable/server/api/custom_module/custom_module.controller.js

/**
 * Using Rails-like standard naming convention for endpoints.
 * GET     /api/custom_modules              ->  index
 * POST    /api/custom_modules              ->  create
 * GET     /api/custom_modules/:id          ->  show
 * PUT     /api/custom_modules/:id          ->  upsert
 * PATCH   /api/custom_modules/:id          ->  patch
 * DELETE  /api/custom_modules/:id          ->  destroy
 */

'use strict';

import jsonpatch from 'fast-json-patch';
import CustomModule from './custom_module.model';
import config from '../../config/environment';
const ssh2_exec = require('../../components/ssh/ssh2_exec');
const scp2_exec = require('../../components/scp/scp_exec');

const logger = require('../../components/logger/logger');

function respondWithResult(res, statusCode) {
  statusCode = statusCode || 200;
  return function(entity) {
    if(entity) {
      return res.status(statusCode).json(entity);
    }
    return null;
  };
}

function patchUpdates(patches) {
  return function(entity) {
    try {
      // eslint-disable-next-line prefer-reflect
      jsonpatch.apply(entity, patches, /*validate*/ true);
    } catch(err) {
      return Promise.reject(err);
    }

    return entity.save();
  };
}

function removeEntity(res) {
  return function(entity) {
    if(entity) {
      return entity.remove()
        .then(() => {
          res.status(204).end();
        });
    }
  };
}

function handleEntityNotFound(res) {
  return function(entity) {
    if(!entity) {
      res.status(404).end();
      return null;
    }
    return entity;
  };
}

function handleError(res, statusCode) {
  statusCode = statusCode || 500;
  return function(err) {
    res.status(statusCode).send(err);
  };
}

// Gets a list of CustomModules
export function index(req, res) {

  var ansibleEngine = req.body.ansibleEngine;

  if(!ansibleEngine.customModules){
    return res.status(500).send("Custom Modules Folder not defined in Ansible Engine")
  }

  var command = 'ls "' + ansibleEngine.customModules + '"';

  ssh2_exec.executeCommand(command,
    null,
    function(data){
      res.send(data)
    },
    function(data){
      res.status(500).send(data)
    },
    ansibleEngine
  );

}

// Gets a single CustomModule or a module_template from DB
export function show(req, res) {
  var ansibleEngine = req.body.ansibleEngine;

  if(!ansibleEngine.customModules){
    return res.status(500).send("Custom Modules Folder not defined in Ansible Engine")
  }

  logger.info('Show custom module ');

  var command = 'cat "' + ansibleEngine.customModules + '"/' + req.params.custom_module;

  // If request is for module template, return module_template from default path
  if(req.params.custom_module === 'template.py'){

    return require('fs').readFile('./helpers/module_template.py', (err, data) => {
      if (err) return res.status(500).send(data);
      res.send(data);
    });

  }else{
    ssh2_exec.executeCommand(command,
      null,
      function(data){
        res.send(data);
      },
      function(data){
        res.status(500).send(data)
      },
      ansibleEngine
    );
  }

}

// Test Module
export function testModule(req, res) {

  var ansibleEngine = req.body.ansibleEngine;

  var moduleArgs = req.body.moduleArgs;

  if(config.disablePlayboookExecution){
    return res.status(500).send('Testing has been disabled. Set environment variable DISABLE_PLAYBOOK_EXECUTION to "false" and restart web server')
  }

  if(!ansibleEngine.customModules){
    res.status(500).send("Custom Modules Folder not defined in Ansible Engine")
  }

  var test_module = '/tmp/test-module';

  var command = 'chmod 755 ' + test_module + '; python ' + test_module + ' -m "' + ansibleEngine.customModules + '/' + req.params.custom_module + "\" -a '" + JSON.stringify(moduleArgs) + "'";

  scp2_exec.copyFileToScriptEngine('./helpers/test-module',test_module,ansibleEngine,function(){
    console.log("Command=" + command);

    ssh2_exec.executeCommand(command,
      null,
      function(data){
        res.send(data);
      },
      function(data){
        res.status(500).send(data)
      },
      ansibleEngine
    );
  }, function(errorResponse){
    res.status(500).send(errorResponse)
  });


  /*return CustomModule.findById(req.params.custom_module).exec()
   .then(handleEntityNotFound(res))
   .then(respondWithResult(res))
   .catch(handleError(res));*/
}

// Creates a new CustomModule in the DB
export function create(req, res) {

  var custom_module_name =  req.params.custom_module;
  var custom_module_code =  req.body.custom_module_code;

  var ansibleEngine = req.body.ansibleEngine;

  if(!ansibleEngine.customModules){
    res.status(500).send("Custom Modules Folder not defined in Ansible Engine")
  }

  console.log("Custom module name " + "\"" + ansibleEngine.customModules + '/' + custom_module_name + "\"")

  scp2_exec.createFileOnScriptEngine(custom_module_code,ansibleEngine.customModules + '/' + custom_module_name,
    function(){
      res.send("Saved")
    },function(err){
      res.status(500).send("Failed to create file on target")
    },
    ansibleEngine
  );

  /*return CustomModule.create(req.body)
   .then(respondWithResult(res, 201))
   .catch(handleError(res));*/
}

// Upserts the given CustomModule in the DB at the specified ID
export function upsert(req, res) {
  if(req.body._id) {
    Reflect.deleteProperty(req.body, '_id');
  }
  return CustomModule.findOneAndUpdate({_id: req.params.id}, req.body, {new: true, upsert: true, setDefaultsOnInsert: true, runValidators: true}).exec()

    .then(respondWithResult(res))
    .catch(handleError(res));
}

// Updates an existing CustomModule in the DB
export function patch(req, res) {
  if(req.body._id) {
    Reflect.deleteProperty(req.body, '_id');
  }
  return CustomModule.findById(req.params.id).exec()
    .then(handleEntityNotFound(res))
    .then(patchUpdates(req.body))
    .then(respondWithResult(res))
    .catch(handleError(res));
}

// Deletes a CustomModule from the DB
export function destroy(req, res) {
  return CustomModule.findById(req.params.id).exec()
    .then(handleEntityNotFound(res))
    .then(removeEntity(res))
    .catch(handleError(res));
}