automa/src/background/workflow-engine/blocks-handler/handler-loop-data.js

import { generateJSON } from '@/utils/data-exporter';
import { getBlockConnection } from '../helper';

function loopData(block) {
  return new Promise((resolve, reject) => {
    const { data } = block;
    const nextBlockId = getBlockConnection(block);

    if (this.loopList[data.loopId]) {
      this.loopList[data.loopId].index += 1;

      let currentLoopData;

      if (data.loopThrough === 'numbers') {
        currentLoopData = this.loopData[data.loopId] + 1;
      } else {
        currentLoopData =
          this.loopList[data.loopId].data[this.loopList[data.loopId].index];
      }

      this.loopData[data.loopId] = currentLoopData;
    } else {
      let currLoopData;

      switch (data.loopThrough) {
        case 'numbers':
          currLoopData = data.fromNumber;
          break;
        case 'data-columns':
          currLoopData = generateJSON(Object.keys(this.data), this.data);
          break;
        case 'google-sheets':
          currLoopData = this.googleSheets[data.referenceKey];
          break;
        case 'custom-data':
          currLoopData = JSON.parse(data.loopData);
          break;
        default:
      }

      if (data.loopThrough !== 'number' && !Array.isArray(currLoopData)) {
        const error = new Error('invalid-loop-data');
        error.nextBlockId = nextBlockId;

        reject(error);
        return;
      }

      this.loopList[data.loopId] = {
        index: 0,
        data: currLoopData,
        id: data.loopId,
        blockId: block.id,
        type: data.loopThrough,
        maxLoop:
          data.loopThrough === 'numbers'
            ? data.toNumber + 1 - data.fromNumber
            : data.maxLoop || currLoopData.length,
      };
      /* eslint-disable-next-line */
      this.loopData[data.loopId] = data.loopThrough === 'numbers' ? data.fromNumber : currLoopData[0];
    }

    resolve({
      nextBlockId,
      data: this.loopData[data.loopId],
    });
  });
}

export default loopData;