blockly/generators/lua/logic.js

/**
 * @license
 * Copyright 2016 Google LLC
 * SPDX-License-Identifier: Apache-2.0
 */

/**
 * @fileoverview Generating Lua for logic blocks.
 */

import * as goog from '../../closure/goog/goog.js';
goog.declareModuleId('Blockly.Lua.logic');

import {luaGenerator, Order} from '../lua.js';


luaGenerator.forBlock['controls_if'] = function(block) {
  // If/elseif/else condition.
  let n = 0;
  let code = '';
  if (luaGenerator.STATEMENT_PREFIX) {
    // Automatic prefix insertion is switched off for this block.  Add manually.
    code += luaGenerator.injectId(luaGenerator.STATEMENT_PREFIX, block);
  }
  do {
    const conditionCode =
        luaGenerator.valueToCode(block, 'IF' + n, Order.NONE) || 'false';
    let branchCode = luaGenerator.statementToCode(block, 'DO' + n);
    if (luaGenerator.STATEMENT_SUFFIX) {
      branchCode = luaGenerator.prefixLines(
          luaGenerator.injectId(luaGenerator.STATEMENT_SUFFIX, block),
          luaGenerator.INDENT) + branchCode;
    }
    code +=
        (n > 0 ? 'else' : '') + 'if ' + conditionCode + ' then\n' + branchCode;
    n++;
  } while (block.getInput('IF' + n));

  if (block.getInput('ELSE') || luaGenerator.STATEMENT_SUFFIX) {
    let branchCode = luaGenerator.statementToCode(block, 'ELSE');
    if (luaGenerator.STATEMENT_SUFFIX) {
      branchCode =
          luaGenerator.prefixLines(
            luaGenerator.injectId(
              luaGenerator.STATEMENT_SUFFIX, block),
            luaGenerator.INDENT) +
          branchCode;
    }
    code += 'else\n' + branchCode;
  }
  return code + 'end\n';
};

luaGenerator.forBlock['controls_ifelse'] = luaGenerator.forBlock['controls_if'];

luaGenerator.forBlock['logic_compare'] = function(block) {
  // Comparison operator.
  const OPERATORS =
      {'EQ': '==', 'NEQ': '~=', 'LT': '<', 'LTE': '<=', 'GT': '>', 'GTE': '>='};
  const operator = OPERATORS[block.getFieldValue('OP')];
  const argument0 =
        luaGenerator.valueToCode(block, 'A', Order.RELATIONAL) || '0';
  const argument1 =
        luaGenerator.valueToCode(block, 'B', Order.RELATIONAL) || '0';
  const code = argument0 + ' ' + operator + ' ' + argument1;
  return [code, Order.RELATIONAL];
};

luaGenerator.forBlock['logic_operation'] = function(block) {
  // Operations 'and', 'or'.
  const operator = (block.getFieldValue('OP') === 'AND') ? 'and' : 'or';
  const order = (operator === 'and') ? Order.AND : Order.OR;
  let argument0 = luaGenerator.valueToCode(block, 'A', order);
  let argument1 = luaGenerator.valueToCode(block, 'B', order);
  if (!argument0 && !argument1) {
    // If there are no arguments, then the return value is false.
    argument0 = 'false';
    argument1 = 'false';
  } else {
    // Single missing arguments have no effect on the return value.
    const defaultArgument = (operator === 'and') ? 'true' : 'false';
    if (!argument0) {
      argument0 = defaultArgument;
    }
    if (!argument1) {
      argument1 = defaultArgument;
    }
  }
  const code = argument0 + ' ' + operator + ' ' + argument1;
  return [code, order];
};

luaGenerator.forBlock['logic_negate'] = function(block) {
  // Negation.
  const argument0 =
        luaGenerator.valueToCode(block, 'BOOL', Order.UNARY) || 'true';
  const code = 'not ' + argument0;
  return [code, Order.UNARY];
};

luaGenerator.forBlock['logic_boolean'] = function(block) {
  // Boolean values true and false.
  const code = (block.getFieldValue('BOOL') === 'TRUE') ? 'true' : 'false';
  return [code, Order.ATOMIC];
};

luaGenerator.forBlock['logic_null'] = function(block) {
  // Null data type.
  return ['nil', Order.ATOMIC];
};

luaGenerator.forBlock['logic_ternary'] = function(block) {
  // Ternary operator.
  const value_if = luaGenerator.valueToCode(block, 'IF', Order.AND) || 'false';
  const value_then =
        luaGenerator.valueToCode(block, 'THEN', Order.AND) || 'nil';
  const value_else = luaGenerator.valueToCode(block, 'ELSE', Order.OR) || 'nil';
  const code = value_if + ' and ' + value_then + ' or ' + value_else;
  return [code, Order.OR];
};