server.js

/*
 * Copyright (c) 2011 Vinay Pulim <vinay@milewise.com>
 * MIT Licensed
 */

"use strict";

function getDateString(d) {
  function pad(n) {
    return n < 10 ? '0' + n : n;
  }
  return d.getUTCFullYear() + '-'
    + pad(d.getUTCMonth() + 1) + '-'
    + pad(d.getUTCDate()) + 'T'
    + pad(d.getUTCHours()) + ':'
    + pad(d.getUTCMinutes()) + ':'
    + pad(d.getUTCSeconds()) + 'Z';
}

var url = require('url'),
  zlib = null,
  events = require('events'),
  util = require('util'),
  findPrefix = require('./utils').findPrefix;

try {
  zlib = require("zlib");
} catch (error) {
}

var Server = function (server, path, services, wsdl, options) {
  var self = this;

  events.EventEmitter.call(this);

  options = options || {};
  this.path = path;
  this.services = services;
  this.wsdl = wsdl;
  this.suppressStack = options && options.suppressStack;
  this.returnFault = options && options.returnFault;
  this.onewayOptions = options && options.oneWay || {};

  if (path[path.length - 1] !== '/')
    path += '/';
  wsdl.onReady(function (err) {
    if (typeof server.route === 'function' && typeof server.use === 'function') {
      //handle only the required URL path for express server
      server.route(path).all(function (req, res, next) {
        if (typeof self.authorizeConnection === 'function') {
          if (!self.authorizeConnection(req, res)) {
            res.end();
            return;
          }
        }
        self._requestListener(req, res);
      });
    } else {
      var listeners = server.listeners('request').slice();
      server.removeAllListeners('request');
      server.addListener('request', function (req, res) {
        if (typeof self.authorizeConnection === 'function') {
          if (!self.authorizeConnection(req, res)) {
            res.end();
            return;
          }
        }
        var reqPath = url.parse(req.url).pathname;
        if (reqPath[reqPath.length - 1] !== '/') {
          reqPath += '/';
        }
        if (path === reqPath) {
          self._requestListener(req, res);
        } else {
          for (var i = 0, len = listeners.length; i < len; i++) {
            listeners[i].call(this, req, res);
          }
        }
      });
    }
  });

  this._initializeOptions(options);
};
util.inherits(Server, events.EventEmitter);

Server.prototype._processSoapHeader = function (soapHeader, name, namespace, xmlns) {
  var self = this;

  switch (typeof soapHeader) {
  case 'object':
    return this.wsdl.objectToXML(soapHeader, name, namespace, xmlns, true);
  case 'function':
    return function() {
      var result = soapHeader.apply(null, arguments);

      if (typeof result === 'object') {
        return self.wsdl.objectToXML(result, name, namespace, xmlns, true);
      } else {
        return result;
      }
    };
  default:
    return soapHeader;
  }
};

Server.prototype.addSoapHeader = function (soapHeader, name, namespace, xmlns) {
  if (!this.soapHeaders) {
    this.soapHeaders = [];
  }
  soapHeader = this._processSoapHeader(soapHeader, name, namespace, xmlns);
  return this.soapHeaders.push(soapHeader) - 1;
};

Server.prototype.changeSoapHeader = function (index, soapHeader, name, namespace, xmlns) {
  if (!this.soapHeaders) {
    this.soapHeaders = [];
  }
  soapHeader = this._processSoapHeader(soapHeader, name, namespace, xmlns);
  this.soapHeaders[index] = soapHeader;
};

Server.prototype.getSoapHeaders = function () {
  return this.soapHeaders;
};

Server.prototype.clearSoapHeaders = function () {
  this.soapHeaders = null;
};

Server.prototype._initializeOptions = function (options) {
  this.wsdl.options.attributesKey = options.attributesKey || 'attributes';
  this.onewayOptions.statusCode = this.onewayOptions.responseCode || 200;
  this.onewayOptions.emptyBody = !!this.onewayOptions.emptyBody;
};

Server.prototype._processRequestXml = function (req, res, xml) {
  var self = this;
  var result;
  var error;
  try {
    if (typeof self.log === 'function') {
      self.log("received", xml);
    }
    self._process(xml, req, function (result, statusCode) {
      if (statusCode) {
        res.statusCode = statusCode;
      }
      res.write(result);
      res.end();
      if (typeof self.log === 'function') {
        self.log("replied", result);
      }
    });
  } catch (err) {
    if (err.Fault !== undefined) {
      return self._sendError(err.Fault, function (result, statusCode) {
        res.statusCode = statusCode || 500;
        res.write(result);
        res.end();
        if (typeof self.log === 'function') {
          self.log("error", err);
        }
      }, new Date().toISOString());
    } else {
      error = err.stack ? (self.suppressStack === true ? err.message : err.stack) : err;
      res.statusCode = 500;
      res.write(error);
      res.end();
      if (typeof self.log === 'function') {
        self.log("error", error);
      }
    }
  }
};

Server.prototype._requestListener = function (req, res) {
  var self = this;
  var reqParse = url.parse(req.url);
  var reqPath = reqParse.pathname;
  var reqQuery = reqParse.search;

  if (typeof self.log === 'function') {
    self.log("info", "Handling " + req.method + " on " + req.url);
  }

  if (req.method === 'GET') {
    if (reqQuery && reqQuery.toLowerCase() === '?wsdl') {
      if (typeof self.log === 'function') {
        self.log("info", "Wants the WSDL");
      }
      res.setHeader("Content-Type", "application/xml");
      res.write(self.wsdl.toXML());
    }
    res.end();
  } else if (req.method === 'POST') {
    if (typeof req.headers['content-type'] !== "undefined") {
      res.setHeader('Content-Type', req.headers['content-type']);
    } else {
      res.setHeader('Content-Type', "application/xml");
    }

    //request body is already provided by an express middleware
    //in this case unzipping should also be done by the express middleware itself
    if (req.body && Object.keys(req.body).length > 0) {
      return self._processRequestXml(req, res, req.body.toString());
    }

    var chunks = [], gunzip, source = req;
    if (req.headers["content-encoding"] === "gzip") {
      gunzip = zlib.createGunzip();
      req.pipe(gunzip);
      source = gunzip;
    }
    source.on('data', function (chunk) {
      chunks.push(chunk);
    });
    source.on('end', function () {
      var xml = Buffer.concat(chunks).toString();
      var result;
      var error;
      self._processRequestXml(req, res, xml);
    });
  }
  else {
    res.end();
  }
};

Server.prototype._process = function (input, req, callback) {
  var self = this,
    pathname = url.parse(req.url).pathname.replace(/\/$/, ''),
    obj = this.wsdl.xmlToObject(input),
    body = obj.Body,
    headers = obj.Header,
    bindings = this.wsdl.definitions.bindings, binding,
    method, methodName,
    serviceName, portName,
    includeTimestamp = obj.Header && obj.Header.Security && obj.Header.Security.Timestamp,
    authenticate = self.authenticate || function defaultAuthenticate() { return true; };

  function process() {

    if (typeof self.log === 'function') {
      self.log("info", "Attempting to bind to " + pathname);
    }

    //Avoid Cannot convert undefined or null to object due to Object.keys(body)
    //and throw more meaningful error
    if (!body) {
      throw new Error('Failed to parse the SOAP Message body');
    }

    // use port.location and current url to find the right binding
    binding = (function () {
      var services = self.wsdl.definitions.services;
      var firstPort;
      var name;
      for (name in services) {
        serviceName = name;
        var service = services[serviceName];
        var ports = service.ports;
        for (name in ports) {
          portName = name;
          var port = ports[portName];
          var portPathname = url.parse(port.location).pathname.replace(/\/$/, '');

          if (typeof self.log === 'function') {
            self.log("info", "Trying " + portName + " from path " + portPathname);
          }

          if (portPathname === pathname)
            return port.binding;

          // The port path is almost always wrong for generated WSDLs
          if (!firstPort) {
            firstPort = port;
          }
        }
      }
      return !firstPort ? void 0 : firstPort.binding;
    })();

    if (!binding) {
      throw new Error('Failed to bind to WSDL');
    }

    try {
      if (binding.style === 'rpc') {
        methodName = Object.keys(body)[0];

        self.emit('request', obj, methodName);
        if (headers)
          self.emit('headers', headers, methodName);

        self._executeMethod({
          serviceName: serviceName,
          portName: portName,
          methodName: methodName,
          outputName: methodName + 'Response',
          args: body[methodName],
          headers: headers,
          style: 'rpc'
        }, req, callback);
      } else {
        var messageElemName = (Object.keys(body)[0] === 'attributes' ? Object.keys(body)[1] : Object.keys(body)[0]);
        var pair = binding.topElements[messageElemName];

        self.emit('request', obj, pair.methodName);
        if (headers)
          self.emit('headers', headers, pair.methodName);

        self._executeMethod({
          serviceName: serviceName,
          portName: portName,
          methodName: pair.methodName,
          outputName: pair.outputName,
          args: body[messageElemName],
          headers: headers,
          style: 'document'
        }, req, callback, includeTimestamp);
      }
    }
    catch (error) {
      if (error.Fault !== undefined) {
        return self._sendError(error.Fault, callback, includeTimestamp);
      }

      throw error;
    }
  }

  // Authentication

  if (typeof authenticate === 'function') {

    var authResultProcessed = false,
      processAuthResult = function (authResult) {

        if (!authResultProcessed && (authResult || authResult === false)) {

          authResultProcessed = true;

          if (authResult) {

            try {
              process();
            } catch (error) {

              if (error.Fault !== undefined) {
                return self._sendError(error.Fault, callback, includeTimestamp);
              }

              return self._sendError({
                Code: {
                  Value: 'SOAP-ENV:Server',
                  Subcode: { value: 'InternalServerError' }
                },
                Reason: { Text: error.toString() },
                statusCode: 500
              }, callback, includeTimestamp);
            }

          } else {

            return self._sendError({
              Code: {
                Value: 'SOAP-ENV:Client',
                Subcode: { value: 'AuthenticationFailure' }
              },
              Reason: { Text: 'Invalid username or password' },
              statusCode: 401
            }, callback, includeTimestamp);
          }
        }
      };

    processAuthResult(authenticate(obj.Header && obj.Header.Security, processAuthResult));

  } else {
    throw new Error('Invalid authenticate function (not a function)');
  }
};

Server.prototype._executeMethod = function (options, req, callback, includeTimestamp) {
  options = options || {};
  var self = this,
    method, body, headers,
    serviceName = options.serviceName,
    portName = options.portName,
    methodName = options.methodName,
    outputName = options.outputName,
    args = options.args,
    style = options.style,
    handled = false;

  if (this.soapHeaders) {
    headers = this.soapHeaders.map(function(header) {
      if (typeof header === 'function') {
        return header(methodName, args, options.headers, req);
      } else {
        return header;
      }
    }).join("\n");
  }

  try {
    method = this.services[serviceName][portName][methodName];
  } catch (error) {
    return callback(this._envelope('', headers, includeTimestamp));
  }

  function handleResult(error, result) {
    if (handled)
      return;
    handled = true;

    if (error && error.Fault !== undefined) {
      return self._sendError(error.Fault, callback, includeTimestamp);
    }
    else if (result === undefined) {
      // Backward compatibility to support one argument callback style
      result = error;
    }

    if (style === 'rpc') {
      body = self.wsdl.objectToRpcXML(outputName, result, '', self.wsdl.definitions.$targetNamespace);
    } else {
      var element = self.wsdl.definitions.services[serviceName].ports[portName].binding.methods[methodName].output;
      body = self.wsdl.objectToDocumentXML(outputName, result, element.targetNSAlias, element.targetNamespace);
    }
    callback(self._envelope(body, headers, includeTimestamp));
  }

  if (!self.wsdl.definitions.services[serviceName].ports[portName].binding.methods[methodName].output) {
    // no output defined = one-way operation so return empty response
    handled = true;
    body = '';
    if (this.onewayOptions.emptyBody) {
      body = self._envelope('', headers, includeTimestamp);
    }
    callback(body, this.onewayOptions.responseCode);
  }

  var result = method(args, handleResult, options.headers, req);
  if (typeof result !== 'undefined') {
    handleResult(result);
  }
};

Server.prototype._envelope = function (body, headers, includeTimestamp) {
  var defs = this.wsdl.definitions,
    ns = defs.$targetNamespace,
    encoding = '',
    alias = findPrefix(defs.xmlns, ns);

  var envelopeDefinition = this.wsdl.options.forceSoap12Headers
    ? "http://www.w3.org/2003/05/soap-envelope"
    : "http://schemas.xmlsoap.org/soap/envelope/";

  var xml = "<?xml version=\"1.0\" encoding=\"utf-8\"?>" +
    "<soap:Envelope xmlns:soap=\"" + envelopeDefinition + "\" " +
    encoding +
    this.wsdl.xmlnsInEnvelope + '>';

  headers = headers || '';

  if (includeTimestamp) {
    var now = new Date();
    var created = getDateString(now);
    var expires = getDateString(new Date(now.getTime() + (1000 * 600)));

    headers += "<o:Security soap:mustUnderstand=\"1\" " +
      "xmlns:o=\"http://docs.oasis-open.org/wss/2004/01/oasis-200401-wss-wssecurity-secext-1.0.xsd\" " +
      "xmlns:u=\"http://docs.oasis-open.org/wss/2004/01/oasis-200401-wss-wssecurity-utility-1.0.xsd\">" +
      "    <u:Timestamp u:Id=\"_0\">" +
      "      <u:Created>" + created + "</u:Created>" +
      "      <u:Expires>" + expires + "</u:Expires>" +
      "    </u:Timestamp>" +
      "  </o:Security>\n";
  }

  if (headers !== '') {
    xml += "<soap:Header>" + headers + "</soap:Header>";
  }

  xml += body ? "<soap:Body>" + body + "</soap:Body>" : "<soap:Body/>";

  xml += "</soap:Envelope>";
  return xml;
};

Server.prototype._sendError = function (soapFault, callback, includeTimestamp) {
  var self = this,
    fault;

  var statusCode;
  if (soapFault.statusCode) {
    statusCode = soapFault.statusCode;
    soapFault.statusCode = undefined;
  }

  if (soapFault.faultcode) {
    // Soap 1.1 error style
    // Root element will be prependend with the soap NS
    // It must match the NS defined in the Envelope (set by the _envelope method)
    fault = self.wsdl.objectToDocumentXML("soap:Fault", soapFault, undefined);
  }
  else {
    // Soap 1.2 error style.
    // 3rd param is the NS prepended to all elements
    // It must match the NS defined in the Envelope (set by the _envelope method)
    fault = self.wsdl.objectToDocumentXML("Fault", soapFault, "soap");
  }

  return callback(self._envelope(fault, '', includeTimestamp), statusCode);
};

exports.Server = Server;