@lcap/builder/lib/nasl/story-nasl-block.js

lcap builder utils

"use strict";
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    var desc = Object.getOwnPropertyDescriptor(m, k);
    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
      desc = { enumerable: true, get: function() { return m[k]; } };
    }
    Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
    if (k2 === undefined) k2 = k;
    o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
    Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
    o["default"] = v;
});
var __importStar = (this && this.__importStar) || function (mod) {
    if (mod && mod.__esModule) return mod;
    var result = {};
    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
    __setModuleDefault(result, mod);
    return result;
};
var __importDefault = (this && this.__importDefault) || function (mod) {
    return (mod && mod.__esModule) ? mod : { "default": mod };
};
Object.defineProperty(exports, "__esModule", { value: true });
const parser = __importStar(require("@babel/parser"));
const traverse_1 = __importDefault(require("@babel/traverse"));
const ts2nasl_1 = require("../ts2nasl");
const babel_utils_1 = require("../utils/babel-utils");
const getBlocksFromStory = (code, framework) => {
    const ast = parser.parse(code, {
        sourceType: 'module',
        plugins: ['typescript', 'jsx'],
    });
    const blocks = [];
    (0, traverse_1.default)(ast, {
        VariableDeclarator(path) {
            const hasExporedName = path.findParent((p) => p.type === 'ExportNamedDeclaration');
            if (!hasExporedName || !path.node.id || !path.node.init
                || path.node.id.type !== 'Identifier'
                || path.node.init.type !== 'ObjectExpression'
                || !path.node.init.properties
                || path.node.init.properties.length === 0) {
                return;
            }
            const block = {
                name: path.node.id.name,
                template: '',
            };
            if (framework.startsWith('vue')) {
                let nameSeted = false;
                path.traverse({
                    ObjectProperty: (p) => {
                        if (p.node.key && p.node.key.type === 'Identifier' && p.node.key.name === 'name' && p.node.value && p.node.value.type === 'StringLiteral' && !nameSeted) {
                            block.name = p.node.value.value;
                            nameSeted = true;
                        }
                        else if (p.node.key && p.node.key.type === 'Identifier' && p.node.key.name === 'template') {
                            // eslint-disable-next-line no-eval
                            block.template = `<template>${eval((0, babel_utils_1.getNodeCode)(p.node.value))}</template>`;
                        }
                        else if (nameSeted && block.template) {
                            p.stop();
                        }
                    },
                });
            }
            else {
                path.traverse({
                    ObjectProperty: (p) => {
                        if (p.node.key && p.node.key.type === 'Identifier' && p.node.key.name === 'name' && p.node.value && p.node.value.type === 'StringLiteral') {
                            block.name = p.node.value.value;
                            p.skip();
                        }
                    },
                    JSXElement: (p) => {
                        const inRenderProps = p.findParent((parentPath) => parentPath.type === 'ObjectProperty' && parentPath.node.key && parentPath.node.key.name === 'render');
                        if (inRenderProps && ['ReturnStatement', 'ArrowFunctionExpression'].indexOf(p.parent.type) === -1) {
                            return;
                        }
                        block.template = JSON.stringify((0, ts2nasl_1.transformTSX2Nasl)(p.node));
                        p.skip();
                    },
                });
            }
            blocks.push(block);
        },
    });
    return blocks;
};
exports.default = getBlocksFromStory;