import { Variables, Procedures } from 'blockly/core';

export const procedures_defreturn = function (_, generator) {
    // Define a procedure with a return value.
    var funcName = generator.variableDB_.getName(this.getFieldValue('NAME'),
        Procedures.NAME_TYPE);
    var branch = (this.getInput('STACK') && generator.statementToCode(this, 'STACK'));
    if (generator.INFINITE_LOOP_TRAP) {
        branch = generator.INFINITE_LOOP_TRAP.replace(/%1/g,
            '\'' + this.id + '\'') + branch;
    }
    var returnValue = generator.valueToCode(this, 'RETURN',
        generator.ORDER_NONE) || '';
    var type = this.getFieldValue('TYPE');
    if (type === 'CUSTOM') {
        if (this.getField('RETRUN_TYPE')) {
            type = this.getFieldValue('RETRUN_TYPE');
        } else {
            type = 'void';
        }
    }
    if (returnValue) {
        returnValue = '  return ' + returnValue + ';\n';
    }
    var returnType = type ? type : 'void';
    var args = [];
    for (var x = 0; x < this.arguments_.length; x++) {
        args[x] = this.argumentstype_[x] + ' ' + generator.variableDB_.getName(this.arguments_[x],
            Variables.NAME_TYPE);
    }
    var code = returnType + ' ' + funcName + '(' + args.join(', ') + ') {\n' +
        branch + returnValue + '}\n';
    code = generator.scrub_(this, code);
    generator.definitions_[funcName] = code;
    return null;
}

export const procedures_defnoreturn = function (_, generator) {
    // Define a procedure with a return value.
    var funcName = generator.variableDB_.getName(this.getFieldValue('NAME'),
        Procedures.NAME_TYPE);
    var branch = (this.getInput('STACK') && generator.statementToCode(this, 'STACK'));
    if (generator.INFINITE_LOOP_TRAP) {
        branch = generator.INFINITE_LOOP_TRAP.replace(/%1/g,
            '\'' + this.id + '\'') + branch;
    }
    var returnType = 'void';
    var args = [];
    for (var x = 0; x < this.arguments_.length; x++) {
        args[x] = this.argumentstype_[x] + ' ' + generator.variableDB_.getName(this.arguments_[x],
            Variables.NAME_TYPE);
    }
    var code = returnType + ' ' + funcName + '(' + args.join(', ') + ') {\n' +
        branch + '}\n';
    code = generator.scrub_(this, code);
    generator.definitions_[funcName] = code;
    return null;
}

export const procedures_callreturn = function (_, generator) {
    // Call a procedure with a return value.
    var funcName = generator.variableDB_.getName(this.getFieldValue('NAME'),
        Procedures.NAME_TYPE);
    var args = [];
    for (var x = 0; x < this.arguments_.length; x++) {
        args[x] = generator.valueToCode(this, 'ARG' + x,
            generator.ORDER_NONE) || 'null';
    }
    var code = funcName + '(' + args.join(', ') + ')';
    return [code, generator.ORDER_UNARY_POSTFIX];
}

export const procedures_callnoreturn = function (_, generator) {
    // Call a procedure with no return value.
    var funcName = generator.variableDB_.getName(this.getFieldValue('NAME'),
        Procedures.NAME_TYPE);
    var args = [];
    for (var x = 0; x < this.arguments_.length; x++) {
        args[x] = generator.valueToCode(this, 'ARG' + x,
            generator.ORDER_NONE) || 'null';
    }
    var code = funcName + '(' + args.join(', ') + ');\n';
    return code;
}

export const procedures_ifreturn = function (_, generator) {
    // Conditionally return value from a procedure.
    var condition = generator.valueToCode(this, 'CONDITION',
        generator.ORDER_NONE) || 'false';
    var code = 'if (' + condition + ') {\n';
    if (this.hasReturnValue_) {
        var value = generator.valueToCode(this, 'VALUE',
            generator.ORDER_NONE) || '';
        code += '  return ' + value + ';\n';
    } else {
        code += '  return;\n';
    }
    code += '}\n';
    return code;
}

export const procedures_return = function (_, generator) {
    // Conditionally return value from a procedure.
    var code = ""
    if (this.hasReturnValue_) {
        var value = generator.valueToCode(this, 'VALUE',
            generator.ORDER_NONE) || 'None';
        code += 'return ' + value + ';\n';
    } else {
        code += 'return;\n';
    }
    code += '\n';
    return code;
}