@lcap/nasl 4.4.1-rc.4 → 4.4.2-rc.2

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,
@@ -592,6 +584,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 +629,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 +674,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 +991,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 || ''),
     });
@@ -1258,6 +1252,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 +1472,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 +1500,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 +1527,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({
@@ -1674,11 +1682,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 +1775,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',
@@ -1853,8 +1856,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 +1910,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 +1924,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 做个区分
@@ -2171,6 +2183,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 +2684,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);