automa/src/utils/workflow-data.js

import Workflow from '@/models/workflow';
import { parseJSON, fileSaver, openFilePicker, isObject } from './helper';

export function importWorkflow(attrs = {}) {
  openFilePicker(['application/json'], attrs)
    .then((files) => {
      const getDrawflow = ({ drawflow }) => {
        if (isObject(drawflow)) return JSON.stringify(drawflow);

        return drawflow;
      };
      const handleOnLoadReader = ({ target }) => {
        const workflow = JSON.parse(target.result);

        if (workflow.includedWorkflows) {
          Object.keys(workflow.includedWorkflows).forEach((workflowId) => {
            const isWorkflowExists = Workflow.query()
              .where('id', workflowId)
              .exists();

            if (isWorkflowExists) return;

            const currentWorkflow = workflow.includedWorkflows[workflowId];
            currentWorkflow.table =
              currentWorkflow.table || currentWorkflow.dataColumns;
            delete currentWorkflow.dataColumns;

            Workflow.insert({
              data: {
                ...currentWorkflow,
                drawflow: getDrawflow(workflow.includedWorkflows[workflowId]),
                id: workflowId,
                createdAt: Date.now(),
              },
            });
          });

          delete workflow.includedWorkflows;
        }

        workflow.table = workflow.table || workflow.dataColumns;
        delete workflow.dataColumns;

        Workflow.insert({
          data: {
            ...workflow,
            drawflow: getDrawflow(workflow),
            createdAt: Date.now(),
          },
        });
      };

      files.forEach((file) => {
        const reader = new FileReader();

        reader.onload = handleOnLoadReader;
        reader.readAsText(file);
      });
    })
    .catch((error) => {
      console.error(error);
    });
}

export function convertWorkflow(workflow, additionalKeys = []) {
  if (!workflow) return null;

  const keys = [
    'name',
    'icon',
    'table',
    'version',
    'drawflow',
    'settings',
    'globalData',
    'description',
    ...additionalKeys,
  ];
  const content = {
    extVersion: chrome.runtime.getManifest().version,
  };

  keys.forEach((key) => {
    content[key] = workflow[key];
  });

  return content;
}
function findIncludedWorkflows({ drawflow }, maxDepth = 3, workflows = {}) {
  if (maxDepth === 0) return workflows;

  const blocks = parseJSON(drawflow, null)?.drawflow.Home.data;

  if (!blocks) return workflows;

  Object.values(blocks).forEach(({ data, name }) => {
    if (name !== 'execute-workflow' || workflows[data.workflowId]) return;

    const workflow = Workflow.find(data.workflowId);

    if (workflow && !workflow.isProtected) {
      workflows[data.workflowId] = convertWorkflow(workflow);
      findIncludedWorkflows(workflow, maxDepth - 1, workflows);
    }
  });

  return workflows;
}
export function exportWorkflow(workflow) {
  if (workflow.isProtected) return;

  const includedWorkflows = findIncludedWorkflows(workflow);
  const content = convertWorkflow(workflow);

  content.includedWorkflows = includedWorkflows;

  const blob = new Blob([JSON.stringify(content)], {
    type: 'application/json',
  });
  const url = URL.createObjectURL(blob);

  fileSaver(`${workflow.name}.json`, url);
}

export default {
  export: exportWorkflow,
  import: importWorkflow,
};