henrikrudstrom/oce-wrap

const features = require('../features.js');
const testLib = require('../testLib.js');
const camelCase = require('camel-case');

//
// swig rendering
//

function renderArg(arg, index, indexes) {
  var res = arg.decl + ' ' + arg.name;

  if (indexes.indexOf(index) !== -1) {
    res += '_out';
  }

  if (arg.default) {
    res += '=' + arg.default;
  }
  return res;
}

function renderMemberFunction(decl) {
  if (decl.declType !== 'constructor' && decl.declType !== 'memfun' || decl.custom)
    return false;

  var indexes = decl.argouts ? decl.argouts.map(argout => argout.index) : [];

  var args = decl.origArguments.map((arg, index) => renderArg(arg, index, indexes))
    .join(', ');

  var stat = decl.static ? 'static ' : '';
  var cons = decl.const ? 'const ' : '';

  return {
    name: decl.parent + 'MemberFunctions',
    src: `
    %feature("compactdefaultargs") ${decl.origName};
    ${stat}${cons}${decl.origReturnType + ' '}${decl.origName}(${args});`
  };
}

features.registerRenderer('swig', 40, renderMemberFunction);

//
// test rendering
//

function argValues(args) {
  return args.map(arg => testLib.createValue(arg.type)).join(', ');
}

function renderMemberFunctionTest(calldef, parts) {
  if (calldef.declType !== 'memfun' || calldef.static === '1')
    return false;

  var cls = calldef.getParent();
  var args = argValues(calldef.arguments.filter(arg => !arg.outArg));
  var createFunc = cls.name.slice(0, 1).toLowerCase() + cls.name.slice(1);
  var testSrc = `\
    var obj = create.${cls.parent}.${createFunc}();
    var res = obj.${calldef.name}(${args});`;

  return {
    name: cls.name + 'MemberSpecs',
    src: testLib.renderTest(calldef, testSrc, parts)
  };
}

function renderConstructorTest(calldef, parts) {
  if (calldef.declType !== 'constructor')
    return false;

  var cls = calldef.getParent();
  var args = argValues(calldef.arguments.filter(arg => !arg.outArg));

  var src = `\
    var res = new ${cls.parent}.${calldef.name}(${args});`;

  return {
    name: cls.name + 'MemberSpecs',
    src: testLib.renderTest(calldef, src, parts)
  };
}

function renderStaticFunctionTest(calldef, parts) {
  if ((calldef.declType !== 'staticfunc' && calldef.static !== '1') ||
    calldef.getParent().declType !== 'class')
    return false;

  var cls = calldef.getParent();
  var args = argValues(calldef.arguments.filter(arg => !arg.outArg));

  var testSrc = `\
    var res = ${cls.parent}.${cls.name}.${calldef.name}(${args});`;

  return {
    name: cls.name + 'MemberSpecs',
    src: testLib.renderTest(calldef, testSrc, parts)
  };
}

function renderFreeFunctionTest(calldef, parts) {
  if (calldef.declType !== 'staticfunc' || calldef.getParent().declType !== 'module')
    return false;

  var mod = calldef.getParent();
  var args = argValues(calldef.arguments.filter(arg => !arg.outArg));

  var testSrc = `\
    var res = ${mod.name}.${calldef.name}(${args});`;
  return {
    name: mod.name + 'ModuleSpecs',
    src: testLib.renderTest(calldef, testSrc, parts)
  };
}

features.registerRenderer('spec', 50, renderMemberFunctionTest);
features.registerRenderer('spec', 50, renderStaticFunctionTest);
features.registerRenderer('spec', 50, renderFreeFunctionTest);
features.registerRenderer('spec', 50, renderConstructorTest);