@lcap/nasl 4.5.0-beta.1 → 4.5.0-beta.2

package/out/natural/transforms/transform2LogicItem.js

@@ -1,16 +1,15 @@
 "use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.transformCall2Interface = exports.transformAlert2ShowMessage = exports.transformForEach2ForEachStatement = exports.transformForOf2ForEachStatement = exports.transform2ForEachStatement = exports.transformMap2ListTransform = exports.transform2UIFunction = exports.transformCall2CallFunction = exports.transform2CallFunction = exports.transform2CallOtherNASLLogic = exports.transform2CallEventLogic = exports.transformCall2JSONFunctions = exports.transformCall2Logging = exports.transformCall2ConsoleLog = exports.transformCall2CallDataSourceLogic = exports.transformCall2CallEntityLogic = exports.transformCall2CallViewElementLogic = exports.transformCallFunctionExpression2Match = exports.transformMemberExpression = exports.transformNewExpression = exports.transform2NewRegex = exports.transform2PartialNewComposite = exports.transform2Comment = exports.transform2StringInterpolation = exports.transformReturnStatement2AssignmentOrEnd = exports.transform2NullLiteral = exports.transform2NumericLiteral = exports.transform2DirectiveLiteral = exports.transform2StringLiteral = exports.transform2BooleanLiteral = exports.transformUpdateExpression2Assignment = exports.transform2Assignment = exports.transform2AnonymousFunction = exports.transformIdentifier = exports.transformSwitchStatement2Match = exports.transform2WhileStatement = exports.transform2Break = exports.transform2Continue = exports.transform2IfStatement = exports.transform2BinaryExpression = exports.transform2VariadicExpression = exports.transformCallExpressionToBinaryExpression = exports.fixLogicNode = exports.flatIfStatementForMatch = exports.flatMatchPatternsExpression = exports.getSelectMembers = exports.unflatMemberExpression = exports.flatMemberExpression = exports.fixExpression = exports.TransformManager = void 0;
-exports.transform2LogicItem = exports.transform2DefaultValue = exports.transformNode2Directory = exports.transformCall2Paginate = exports.transformCall2LogicWithAuth = exports.transformCall2Logic = exports.transformCall2CallViewLogic = exports.transformCall2Extension = exports.transformCall2Connector = void 0;
+exports.transformAlert2ShowMessage = exports.transformForEach2ForEachStatement = exports.transformForOf2ForEachStatement = exports.transform2ForEachStatement = exports.transformMap2ListTransform = exports.transform2UIFunction = exports.transformCall2CallFunction = exports.transform2CallFunction = exports.transform2CallOtherNASLLogic = exports.transform2CallEventLogic = exports.transformCall2JSONFunctions = exports.transformCall2Logging = exports.transformCall2ConsoleLog = exports.transformCall2CallDataSourceLogic = exports.transformCall2CallEntityLogic = exports.transformCall2CallViewElementLogic = exports.transformCallFunctionExpression2Match = exports.transformMemberExpression = exports.transformNewExpression = exports.transform2NewRegex = exports.transform2PartialNewComposite = exports.transform2Comment = exports.transform2StringInterpolation = exports.transformReturnStatement2AssignmentOrEnd = exports.transform2NullLiteral = exports.transform2NumericLiteral = exports.transform2DirectiveLiteral = exports.transform2StringLiteral = exports.transform2BooleanLiteral = exports.transformUpdateExpression2Assignment = exports.transform2Assignment = exports.transform2AnonymousFunction = exports.transformIdentifier = exports.transformSwitchStatement2Match = exports.transform2WhileStatement = exports.transform2Break = exports.transform2Continue = exports.transform2IfStatement = exports.transformArrayExpression = exports.transform2BinaryExpression = exports.transform2VariadicExpression = exports.transformCallExpressionToBinaryExpression = exports.fixLogicNode = exports.flatIfStatementForMatch = exports.flatMatchPatternsExpression = exports.getSelectMembers = exports.unflatMemberExpression = exports.flatMemberExpression = exports.fixExpression = exports.TransformManager = void 0;
+exports.transform2LogicItem = exports.transform2DefaultValue = exports.transformNode2Directory = exports.transformCall2Paginate = exports.transformCall2GetCustomConfig = exports.transformCall2LogicWithAuth = exports.transformCall2Logic = exports.transformCall2CallViewLogic = exports.transformCall2Extension = exports.transformCall2Connector = exports.transformCall2Interface = void 0;
 const utils_1 = require("../utils");
 const misc_1 = require("./utils/misc");
+const flattenStatementsWithComments_1 = require("./utils/flattenStatementsWithComments");
 const checkDataQueryCalleeNameOrder_1 = require("./checker/checkDataQueryCalleeNameOrder");
+const checkIdentifierName_1 = require("./checker/checkIdentifierName");
 const transform2TypeAnnotation_1 = require("./transform2TypeAnnotation");
 const transform2Param_1 = require("./transform2Param");
-const server_1 = __importDefault(require("../../server"));
+const service_1 = require("@lcap/nasl-concepts/service");
 const parseDecorator_1 = require("./utils/parseDecorator");
 ;
 class TransformManager {
@@ -80,23 +79,13 @@ function transformTSCode2SubLogic(node, options) {
         statements = [...node?.body?.directives, ...node?.body?.body];
     }
     parameters.forEach((node) => {
-        const param = (0, transform2Param_1.transform2Param)(node, options);
+        const param = (0, transform2Param_1.transform2Param)(node, options, undefined, true);
         if (param && !param?.typeAnnotation) {
             param.typeAnnotation = null;
         }
         param && logic.addParam(param);
     });
-    const newStatements = [];
-    statements.forEach((node, index) => {
-        if (node?.leadingComments?.length) {
-            newStatements.push(...node?.leadingComments);
-        }
-        if (node?.trailingComments?.length && index === statements.length - 1) {
-            newStatements.push(...node?.trailingComments);
-        }
-        newStatements.push(node);
-    });
-    statements = newStatements;
+    statements = (0, flattenStatementsWithComments_1.flattenStatementsWithComments)(statements);
     const body = statements.map((node) => transformNodeFunction(node, {
         ...options,
         logic,
@@ -431,6 +420,9 @@ function transformConditionalExpression(node, options) {
     });
 }
 function transformArrayExpression(node, options) {
+    if (options?.inCallQueryComponentField) {
+        return (0, utils_1.throwError)(options, '数据查询中不支持使用 NewList(直接声明数组)，请将对应操作放在数据查询外部进行处理', node);
+    }
     let typeAnnotation = options.typeAnnotation;
     delete options.typeAnnotation;
     const transformNodeFunction = options?.transformNodeFunction || transform2LogicItem;
@@ -439,7 +431,11 @@ function transformArrayExpression(node, options) {
         items: node?.elements?.map((arg) => transformNodeFunction(arg, { ...options, typeAnnotation: undefined })).filter(Boolean),
     });
 }
+exports.transformArrayExpression = transformArrayExpression;
 function transform2NewAnonymousStructure(node, options) {
+    if (options?.inCallQueryComponentField) {
+        return (0, utils_1.throwError)(options, '数据查询中不支持使用 NewComposite(直接声明匿名结构)，请将对应操作放在数据查询外部进行处理', node);
+    }
     for (let i = 0; i < node.properties.length; i++) {
         const item = node.properties[i];
         const itemValue = item?.value;
@@ -462,7 +458,7 @@ function transform2NewAnonymousStructure(node, options) {
     const structureProperties = [];
     newProperties.forEach((arg, index) => {
         structureProperties.push(new utils_1.naslTypes.StructureProperty({
-            name: arg.key.name,
+            name: arg.key.name ?? arg.key.value,
             typeAnnotation: arg.naslType || null,
         }));
     });
@@ -478,6 +474,9 @@ function transform2NewStructure(namespaceAndName, objectExpression, options) {
     return newComposite;
 }
 function transform2NewExpression(calleeName, node, options) {
+    if (options?.inCallQueryComponentField) {
+        return (0, utils_1.throwError)(options, `数据查询中不支持使用 ${calleeName}，请将对应操作放在数据查询外部进行处理`, node);
+    }
     const transformNodeFunction = options?.transformNodeFunction || transform2LogicItem;
     if (calleeName === 'NewList') {
         let typeAnnotation;
@@ -592,6 +591,7 @@ function transform2ArrayPattern(decl, options) {
         if (element.type === 'Identifier') {
             const varTypeAnnotation = element.typeAnnotation?.typeAnnotation;
             if (options.transformType === 'logic') {
+                (0, checkIdentifierName_1.checkIdentifierName)(element.name, '变量', options, element);
                 options.logic.variables.push(new utils_1.naslTypes.Variable({
                     name: element.name,
                     typeAnnotation: varTypeAnnotation && (0, transform2TypeAnnotation_1.transform2TypeAnnotation)(varTypeAnnotation, options) || undefined,
@@ -636,6 +636,7 @@ function transform2VariableAndAddAssignment(node, options) {
         }
         let typeAnnotation;
         if (options.transformType === 'logic' && ![...options.logic.variables, ...options.logic.returns].some((variable) => variable.name === variableName)) {
+            (0, checkIdentifierName_1.checkIdentifierName)(variableName, '变量', options, decl.id);
             const varTypeAnnotation = decl.id.typeAnnotation?.typeAnnotation;
             typeAnnotation = varTypeAnnotation && (0, transform2TypeAnnotation_1.transform2TypeAnnotation)(varTypeAnnotation, options) || undefined;
             const variable = new utils_1.naslTypes.Variable({
@@ -680,15 +681,7 @@ function flatCommentWithNode(node) {
         newNodeList = [...node?.innerComments];
     }
     if (node?.body?.length) {
-        node?.body.forEach((item, index) => {
-            if (item?.leadingComments?.length) {
-                newNodeList.push(...item?.leadingComments);
-            }
-            if (item?.trailingComments?.length && index === node?.body.length - 1) {
-                newNodeList.push(...item?.trailingComments);
-            }
-            newNodeList.push(item);
-        });
+        newNodeList.push(...(0, flattenStatementsWithComments_1.flattenStatementsWithComments)(node.body));
     }
     return newNodeList;
 }
@@ -1005,16 +998,24 @@ function transform2StringInterpolation(node, options) {
     });
 }
 exports.transform2StringInterpolation = transform2StringInterpolation;
+function transformRequire2JSBlock(node, calleeName, callee, options) {
+    const arg0 = node.arguments[0].value;
+    const filename = arg0.replace('./tmp/', '');
+    return new utils_1.naslTypes.JSBlock({ code: options.externalFiles?.get(filename) ?? '' });
+}
 function transform2Comment(node, options) {
     if (node.type === 'EmptyStatement') {
         if (node?.leadingComments) {
-            const newValue = node?.leadingComments?.[0]?.value?.replace(/\n/g, '<br>');
+            // trim() 防止 Babel 解析 /* ... */ 时将内侧空格存入 value，
+            // 导致每次 files2nasl → nasl2files 轮回后注释空格累积 +1
+            const newValue = node?.leadingComments?.[0]?.value?.trim().replace(/\n/g, '<br>');
             return new utils_1.naslTypes.Comment({
                 value: String(newValue),
             });
         }
     }
-    const newValue = node?.value?.replace(/\n/g, '<br>');
+    // trim() 同上，防止空格在每次 prepare-src 轮回中递增
+    const newValue = node?.value?.trim().replace(/\n/g, '<br>');
     return new utils_1.naslTypes.Comment({
         value: String(newValue || ''),
     });
@@ -1037,7 +1038,16 @@ function transform2PartialNewComposite(node, options) {
     const newProperties = (node?.properties || node?.arguments?.[0]?.properties);
     let idx = 0;
     newProperties?.forEach((arg, index) => {
-        properties.push(transformNodeFunction(arg.key, options));
+        // StringLiteral 类型的键（如 JSON 格式的 "name"）统一转换为 Identifier 处理，并校验命名规则
+        let keyNode;
+        if (arg.key?.type === 'StringLiteral') {
+            (0, checkIdentifierName_1.checkIdentifierName)(arg.key.value, '属性', options, arg.key);
+            keyNode = utils_1.babelTypes.identifier(arg.key.value);
+        }
+        else {
+            keyNode = arg.key;
+        }
+        properties.push(transformNodeFunction(keyNode, options));
         const calleeName = (0, utils_1.generate)(arg.value).code;
         let sliceMembers = false;
         if (arg.value.type === 'MemberExpression') {
@@ -1103,7 +1113,7 @@ function transform2PartialNewComposite(node, options) {
             }
         }
         structureProperties.push(new utils_1.naslTypes.StructureProperty({
-            name: arg?.key?.name,
+            name: arg?.key?.name ?? arg?.key?.value,
             typeAnnotation: null
         }));
     });
@@ -1148,6 +1158,9 @@ function transformNewExpression(node, options) {
         return transform2NewRegex(node, options);
     }
     else if (calleeName.includes('.entities.') || calleeName.includes('.structures.')) {
+        if (options?.inCallQueryComponentField) {
+            return (0, utils_1.throwError)(options, `数据查询中不支持使用 new ${calleeName}，请将对应操作放在数据查询外部进行处理`, node);
+        }
         return transform2NewStructure(calleeName, node.arguments[0], options);
     }
     return (0, utils_1.throwError)(options, '在新建表达式中不支持节点类型', node.type, node);
@@ -1258,6 +1271,14 @@ function transformMemberExpression(node, options) {
             return newNode;
         }
     }
+    const configPropertyRegex = /^(app|extensions\.\w+)\.configuration\.(?:groups\.)?(\w+)\.(?:properties\.)?(\w+)$/;
+    if (configPropertyRegex.test(calleeName)) {
+        const [, prefix, groupName, propName] = calleeName.match(configPropertyRegex);
+        return new utils_1.naslTypes.Identifier({
+            namespace: `${prefix}.configuration.groups.${groupName}.properties`,
+            name: propName,
+        });
+    }
     if (node.property.type === 'NumericLiteral' || node?.property?.type === 'BinaryExpression') {
         return new utils_1.naslTypes.CallFunction({
             calleeNamespace: 'nasl.util',
@@ -1470,6 +1491,7 @@ function transformCall2CallDataSourceLogic(node, calleeName, callee, options) {
             if (newArguments?.[0])
                 newArguments[0].keyword = 'body';
             if (newArguments?.[1]) {
+                newArguments[1].keyword = 'properties';
                 if (newArguments[1]?.expression.concept === 'AnonymousFunction' && newArguments[1]?.expression?.params?.[0]?.name) {
                     newArguments[1].expression.params[0].typeAnnotation = utils_1.naslTypes.TypeAnnotation.createReference(entityName, {
                         typeNamespace: calleeNamespace,
@@ -1497,6 +1519,7 @@ function transformCall2CallDataSourceLogic(node, calleeName, callee, options) {
             }
             const expressionJson = expression?.toJSON();
             const argument2 = newArguments?.[1];
+            const argument3 = newArguments?.[2];
             newArguments = [new utils_1.naslTypes.Argument({
                     expression: new utils_1.naslTypes.Function({
                         name: "filter",
@@ -1523,6 +1546,10 @@ function transformCall2CallDataSourceLogic(node, calleeName, callee, options) {
                 else {
                     newArguments.push(new utils_1.naslTypes.Argument({}));
                 }
+                if (argument3?.expression?.concept === 'AnonymousFunction') {
+                    argument3.keyword = 'properties';
+                    newArguments.push(argument3);
+                }
             }
         }
         return new utils_1.naslTypes.CallLogic({
@@ -1633,7 +1660,7 @@ function transform2CallOtherNASLLogic(node, calleeName, callee, options) {
 exports.transform2CallOtherNASLLogic = transform2CallOtherNASLLogic;
 function transform2CallFunction(node, calleeName, argument, options) {
     calleeName = misc_1.replaceCallFunctionMap?.[calleeName] || calleeName;
-    const func = server_1.default ? server_1.default.getBuildInFunction(calleeName) : null;
+    const func = (0, service_1.getBuildInFunction)(calleeName);
     let newArguments = argument.map((arg, index) => {
         let supportSubLogic = false;
         if (arg?.type === 'FunctionExpression' || (arg?.type === 'ArrowFunctionExpression' && arg?.body?.type === 'BlockStatement')) {
@@ -1674,11 +1701,6 @@ function transform2CallFunction(node, calleeName, argument, options) {
             });
         }
     }
-    if (calleeName === 'Convert' && !newTypeArguments) {
-        newTypeArguments = [
-            new utils_1.naslTypes.TypeAnnotation()
-        ];
-    }
     if (calleeName === 'FromString') {
         if (newTypeArguments?.[0]?.typeName === 'Time' && newArguments?.[0]?.expression?.concept === 'StringLiteral') {
             const timeValue = newArguments?.[0]?.expression?.value;
@@ -1772,16 +1794,16 @@ function transform2CallFunction(node, calleeName, argument, options) {
     else {
         // 最后判断一下参数数量是否匹配
         // 这两个参数比较特殊，可以无限加参数，不再判断参数数量
-        const excludes = ['HasValue', 'ListSort'];
-        if (func.params?.length !== newArguments?.length && !excludes.includes(calleeName)) {
-            return (0, utils_1.throwError)(options, `函数${calleeName}生成不符合标准的函数调用，参数数量不匹配`, node.callee, node);
-        }
-        else {
-            newArguments?.forEach((arg, idx) => {
-                if (func?.params[idx]?.name)
-                    arg.keyword = func?.params[idx]?.name;
-            });
-        }
+        // 走LS检查，不在transform中判断了
+        // const excludes = ['HasValue', 'ListSort']
+        // if (func.params?.length !== newArguments?.length && !excludes.includes(calleeName)) {
+        //   return throwError(options, `函数${calleeName}生成不符合标准的函数调用，参数数量不匹配`, node.callee, node);
+        // } else {
+        // }
+        newArguments?.forEach((arg, idx) => {
+            if (func?.params[idx]?.name)
+                arg.keyword = func?.params[idx]?.name;
+        });
     }
     const nodeElement = new utils_1.naslTypes.CallFunction({
         calleeNamespace: 'nasl.util',
@@ -1809,7 +1831,7 @@ function transform2SqlFunction(node, calleeName, argument, options) {
     }
     const specFunList = ['COUNT', 'COUNTD', 'SUM', 'AVG', 'MAX', 'MIN'];
     // TODO: 先写死了 mysql，后续需要传入: 参考 requestSqlFunctions
-    const func = server_1.default ? server_1.default?.getSqlFunction('mysql', calleeName) : null;
+    const func = (0, service_1.getSqlFunction)('mysql', calleeName);
     let addArguments = [];
     if (specFunList.includes(calleeName)) {
         const value = calleeName === 'COUNTD' ? 'DISTINCT' : 'ALL';
@@ -1853,8 +1875,15 @@ function transformCall2CallFunction(node, calleeName, callee, options) {
     return transform2CallFunction(node, curCalleeName, argument, options);
 }
 exports.transformCall2CallFunction = transformCall2CallFunction;
+const SUPPORTED_NASL_UI_FUNCTIONS = ['showMessage', 'destination', 'gotoLink'];
 function transform2UIFunction(node, calleeName, callee, options) {
-    if (callee[2].name === 'destination') {
+    const uiFunctionName = callee[2].name;
+    if (!SUPPORTED_NASL_UI_FUNCTIONS.includes(uiFunctionName)) {
+        return (0, utils_1.throwError)(options, `nasl.ui 仅支持 ${SUPPORTED_NASL_UI_FUNCTIONS.map((f) => `nasl.ui.${f}`).join('、')}，不支持 nasl.ui.${uiFunctionName}`, node);
+    }
+    if (uiFunctionName === 'destination') {
+        if (!node.arguments[0])
+            return (0, utils_1.throwError)(options, 'nasl.ui.destination 缺少目标页面参数', node);
         let namespaceNode = node.arguments[0];
         let namespace = '';
         if (namespaceNode.type === 'TemplateLiteral') {
@@ -1900,7 +1929,9 @@ function transform2UIFunction(node, calleeName, callee, options) {
             arguments: argument,
         });
     }
-    if (callee[2].name === 'gotoLink') {
+    if (uiFunctionName === 'gotoLink') {
+        if (!node.arguments[0])
+            return (0, utils_1.throwError)(options, 'nasl.ui.gotoLink 缺少链接参数', node);
         return new utils_1.naslTypes.ExternalDestination({
             link: fixExpression(node?.arguments[0], options),
         });
@@ -1912,7 +1943,7 @@ function transform2UIFunction(node, calleeName, callee, options) {
     // }
     return new utils_1.naslTypes.CallLogic({
         calleeNamespace: 'nasl.ui',
-        calleeName: callee[2].name,
+        calleeName: uiFunctionName,
         shortcut: true,
         /**
          * 这个函数是修复 AI 翻译出错的情况，与正常的 transform 做个区分
@@ -2140,7 +2171,7 @@ function transformCall2Logic(node, calleeName, callee, options) {
         // 未带前缀，补一下
         return transform2CallFunction(node, calleeName, node?.arguments || [], options);
     }
-    else if (server_1.default.getSqlFunction('mysql', calleeName) || calleeName === 'COUNTD') {
+    else if ((0, service_1.getSqlFunction)('mysql', calleeName) || calleeName === 'COUNTD') {
         return transform2SqlFunction(node, calleeName, node?.arguments || [], options);
     }
     let newCalleeName = calleeName?.includes('__IDENTIFIER__') ? undefined : calleeName;
@@ -2171,6 +2202,30 @@ function transformCall2LogicWithAuth(node, calleeName, callee, options) {
     });
 }
 exports.transformCall2LogicWithAuth = transformCall2LogicWithAuth;
+function transformCall2GetCustomConfig(node, calleeName, callee, options) {
+    const transformNodeFunction = options?.transformNodeFunction || transform2LogicItem;
+    const arg = node.arguments?.[0];
+    if (!arg || arg.type !== 'MemberExpression') {
+        return (0, utils_1.throwError)(options, 'nasl.configuration.getCustomConfig 的参数必须是 {app|extensions.xxx}.configuration.{group}.{name} 形式的成员访问', node);
+    }
+    const flatCallee = flatMemberExpression(arg);
+    const argName = flatCallee.map((n) => n.name).join('.');
+    const configPropertyRegex = /^(app|extensions\.\w+)\.configuration\.(\w+)\.(\w+)$/;
+    if (!configPropertyRegex.test(argName)) {
+        return (0, utils_1.throwError)(options, `nasl.configuration.getCustomConfig 参数路径格式不正确，期望 {app|extensions.xxx}.configuration.{group}.{name}，实际为 ${argName}`, node);
+    }
+    return new utils_1.naslTypes.CallLogic({
+        calleeNamespace: 'nasl.configuration',
+        calleeName: 'getCustomConfig',
+        arguments: [
+            new utils_1.naslTypes.Argument({
+                keyword: 'name',
+                expression: transformNodeFunction(arg, options),
+            }),
+        ],
+    });
+}
+exports.transformCall2GetCustomConfig = transformCall2GetCustomConfig;
 function transformCall2Paginate(node, calleeName, callee, options) {
     const transformNodeFunction = options?.transformNodeFunction || transform2LogicItem;
     const newCallee = node.arguments || [];
@@ -2648,10 +2703,12 @@ function transform2LogicItem(node, options) {
     });
     transformManager.register('MemberExpression', transformMemberExpression);
     transformManager.register('OptionalMemberExpression', transformMemberExpression);
+    transformManager.registerCallExpression((node, calleeName) => calleeName === 'require' && node.arguments[0]?.value?.startsWith('./tmp/'), transformRequire2JSBlock);
     transformManager.registerCallExpression((node, calleeName, callee) => callee.length === 1 && (node.callee.type === 'FunctionExpression' && node.callee.id?.name === 'match' || node.callee.type === 'ArrowFunctionExpression'), transformCallFunctionExpression2Match);
     transformManager.registerCallExpression((node, calleeName) => /^\$refs\./.test(calleeName), transformCall2CallViewElementLogic);
     transformManager.registerCallExpression((node, calleeName, callee) => callee.length === 2 && callee[0].type === 'Identifier' && callee[0].name.endsWith('Entity'), transformCall2CallEntityLogic);
     transformManager.registerCallExpression('nasl.oql.query', transformCall2OQLQuery);
+    transformManager.registerCallExpression('nasl.configuration.getCustomConfig', transformCall2GetCustomConfig);
     transformManager.registerCallExpression('nasl.util.consoleLog', transformCall2ConsoleLog);
     transformManager.registerCallExpression('console.log', transformCall2ConsoleLog);
     transformManager.registerCallExpression((node, calleeName) => calleeName.startsWith('nasl.logging.'), transformCall2Logging);