BeWoPlaner/BeWoPlanerMobil/node_modules/wd/lib/callbacks.js

var utils = require("./utils"),
    newError = utils.newError,
    getJsonwireError = utils.getJsonwireError,
    isWebDriverException = utils.isWebDriverException;

var cbStub = function() {};

// just calls the callback when there is no result
exports.simpleCallback = function(cb) {
  cb = cb || cbStub;
  return function(err, data) {
    if(err) { return cb(err); }
    if((data === '') || (data === 'OK')) {
      // expected behaviour when not returning JsonWire response
      cb(null);
    } else {
      // looking for JsonWire response
      var jsonWireRes;
      var jsonwireError;
      var errorMessage;
      var error;
      try{jsonWireRes = JSON.parse(data);}catch(ign){}
      if (jsonWireRes && typeof jsonWireRes.status !== "undefined") {
        // valid JsonWire response
        if(jsonWireRes.status === 0) {
          cb(null);
        } else {
          jsonwireError  = getJsonwireError(jsonWireRes.status);
          errorMessage = 'Error response status: ' + jsonWireRes.status;
          if(jsonwireError) {
            errorMessage += ", " + jsonwireError.summary + " - " + jsonwireError.detail;
          }
          if(jsonWireRes.value && jsonWireRes.value.message) {
            errorMessage += " Selenium error: " + jsonWireRes.value.message;
          }
          error = newError(
            { message: errorMessage
              , status:jsonWireRes.status
              , cause:jsonWireRes });
          if(jsonwireError){ error['jsonwire-error'] = jsonwireError; }
          cb(error);
        }
      } else if (jsonWireRes && typeof jsonWireRes.value !== "undefined") {
        // valid W3C draft spec response
        if (jsonWireRes.value === null || Object.keys(jsonWireRes.value).length === 0) {
          cb(null);
        } else {
          jsonwireError = jsonWireRes.value;
          errorMessage = 'Error response: ' + jsonwireError.error + ' - ' +
                             jsonwireError.message;
          if (jsonwireError.stacktrace) {
            errorMessage += ' - Stack trace:\n' + jsonwireError.stacktrace;
          }
          error = newError(
            { message: errorMessage
              , cause: jsonwireError });
          cb(error);
        }
      } else {
        // something wrong
        cb(newError(
          {message:'Unexpected data in simpleCallback.', data: jsonWireRes || data}) );
      }
    }
  };
};

// base for all callback handling data
var callbackWithDataBase = function(cb) {
  cb = cb || cbStub;
  return function(err, data) {
    if(err) { return cb(err); }
    var obj,
        alertText;
    try {
      obj = JSON.parse(data);
    } catch (e) {
      return cb(newError({message:'Not JSON response', data:data}));
    }
    try {
        alertText = obj.value.alert.text;
    } catch (e) {
        alertText = '';
    }
    if (obj.status > 0) {
      var jsonwireError  = getJsonwireError(obj.status);
      var errorMessage = 'Error response status: ' + obj.status + ", " +alertText;
      if(jsonwireError) {
        errorMessage += ", " + jsonwireError.summary + " - " + jsonwireError.detail;
      }
      if(obj.value && obj.value.message) {
        errorMessage += " Selenium error: " + obj.value.message;
      }
      var error = newError(
        { message: errorMessage
          , status:obj.status
          , cause:obj });
      if(jsonwireError){ error['jsonwire-error'] = jsonwireError; }
      cb(error);
    } else {
      cb(null, obj);
    }
  };
};

// retrieves field value from result
exports.callbackWithData = function(cb, browser) {
  cb = cb || cbStub;
  return callbackWithDataBase(function(err,obj) {
    if(err) {return cb(err);}
    if(isWebDriverException(obj.value)) {return cb(newError(
      {message:obj.value.message,cause:obj.value}));}
    // we might get a WebElement back as part of executeScript, so let's
    // check to make sure we convert if necessary to element objects
    var elId = utils.getElementId(obj.value);
    if(obj.value && typeof elId !== "undefined") {
        obj.value = browser.newElement(elId);
    } else if (Object.prototype.toString.call(obj.value) === "[object Array]") {
        for (var i = 0; i < obj.value.length; i++) {
            elId = utils.getElementId(obj.value[i]);
            if (obj.value[i] && typeof elId !== "undefined") {
                obj.value[i] = browser.newElement(elId);
            }
        }
    }
    cb(null, obj.value);
  });
};

// retrieves ONE element
exports.elementCallback = function(cb, browser) {
  cb = cb || cbStub;
  return callbackWithDataBase(function(err, obj) {
    if(err) {return cb(err);}
    if(isWebDriverException(obj.value)) {return cb(newError(
      {message:obj.value.message,cause:obj.value}));}
    var elId = utils.getElementId(obj.value);
    if (!elId) {
      cb(newError(
        {message:"no ELEMENT in response value field.",cause:obj}));
    } else {
      var el = browser.newElement(elId);
      cb(null, el);
    }
  });
};

// retrieves SEVERAL elements
exports.elementsCallback = function(cb, browser) {
  cb = cb || cbStub;
  return callbackWithDataBase(function(err, obj) {
    if(err) {return cb(err);}
    if(isWebDriverException(obj.value)) {return cb(newError(
      {message:obj.value.message,cause:obj.value}));}
    if (!Array.isArray(obj.value)) {return cb(newError(
      {message:"Response value field is not an Array.", cause:obj.value}));}
    var i, elements = [];
    for (i = 0; i < obj.value.length; i++) {
      var value = obj.value[i];
      var elId = utils.getElementId(value);
      var el = browser.newElement(elId);
      elements.push(el);
    }
    cb(null, elements);
  });
};

// retrieves ONE or SEVERAL elements
exports.elementOrElementsCallback = function(cb, browser) {
  cb = cb || cbStub;
  return callbackWithDataBase(function(err, obj) {
    if(err) {return cb(err);}
    if(isWebDriverException(obj.value)) {return cb(newError(
      {message:obj.value.message,cause:obj.value}));}
    var el;
    var elId = utils.getElementId(obj.value);
    if (elId){
      el = browser.newElement(elId);
      cb(null, el);
    } else if (Array.isArray(obj.value)){
      var i, elements = [];
      for (i = 0; i < obj.value.length; i++) {
        elId = utils.getElementId(obj.value[i]);
        el = browser.newElement(elId);
        elements.push(el);
      }
      cb(null, elements);
    } else {
      cb(newError(
        {message:"no element or element array in response value field.",cause:obj}));
    }
  });
};