@lcap/nasl 4.4.1-rc.5 → 4.4.2-rc.3

package/out/natural/transforms/transform2LogicItem.js

@@ -4,10 +4,12 @@ var __importDefault = (this && this.__importDefault) || function (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.transform2LogicItem = exports.transform2DefaultValue = exports.transformNode2Directory = exports.transformCall2Paginate = exports.transformCall2GetCustomConfig = exports.transformCall2LogicWithAuth = exports.transformCall2Logic = exports.transformCall2CallViewLogic = exports.transformCall2Extension = exports.transformCall2Connector = 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"));
@@ -80,23 +82,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 +423,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;
@@ -440,6 +435,9 @@ function transformArrayExpression(node, options) {
     });
 }
 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;
@@ -478,6 +476,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 +593,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 +638,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 +683,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 +1000,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 || ''),
     });
@@ -1148,6 +1151,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 +1264,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',
@@ -1680,11 +1694,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;
@@ -1859,8 +1868,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') {
@@ -1906,7 +1922,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),
         });
@@ -1918,7 +1936,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 做个区分
@@ -2177,6 +2195,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 || [];
@@ -2654,10 +2696,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);