@lcap/nasl 2.16.1 → 2.17.1

package/src/concepts/AnonymousFunction__.ts

@@ -0,0 +1,551 @@
+import Namespace from './Namespace__';
+import { TranslatorState, shiftState, withSourceMap, indent } from '../translator';
+import { ElementToVueOptions } from './ViewElement__';
+import ForEachStatement from './ForEachStatement__';
+//================================================================================
+// 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
+// ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
+//================================================================================
+import { EventPayload, Params } from '../common/EventEmitter';
+import { concept, excludedInJSON, property } from '../decorators';
+
+import * as utils from '../utils';
+import { v4 as uuidv4 } from 'uuid';
+import BaseNode from '../common/BaseNode';
+import classMap from '../common/classMap';
+import Param from './Param__';
+import LogicItem from './LogicItem__';
+import TypeAnnotation from './TypeAnnotation__';
+import Module from './Module__';
+import App from './App__';
+
+/**
+ * 匿名函数
+ */
+@concept('匿名函数')
+export class AnonymousFunction extends BaseNode {
+    /**
+     * 产品概念
+     */
+    @property()
+    concept: 'AnonymousFunction' = 'AnonymousFunction';
+
+    /**
+     * 匿名函数描述
+     */
+    @property()
+    description: string = undefined;
+
+    /**
+     * 输入参数列表
+     */
+    @property('Param')
+    params: Array<Param> = [];
+
+    /**
+     * 逻辑项
+     */
+    @property('LogicItem')
+    body: LogicItem = undefined;
+
+    /**
+     * 类型
+     */
+    @property('TypeAnnotation')
+    typeAnnotation: TypeAnnotation = undefined;
+
+    /**
+     * 祖先 Module
+     */
+    get module() {
+        return this.getAncestor('Module') as Module;
+    }
+    /**
+     * 祖先 App
+     */
+    get app() {
+        return this.getAncestor('App') as App;
+    }
+
+    /**
+     * @param source 需要合并的部分参数
+     */
+    constructor(source?: Partial<AnonymousFunction>) {
+        source = Object.assign({}, AnonymousFunction.getDefaultOptions(), source);
+        super(source);
+        super.subConstructor(source);
+    }
+
+    static from(source: any, parentNode?: any, parentKey?: string): AnonymousFunction {
+        return super.from(source, parentNode, parentKey) as AnonymousFunction;
+    }
+
+    /**
+     * 从父级删除该节点
+     * @internal
+     */
+    _delete() {
+        let params: Params = null;
+        if (this.parentNode) {
+            params = (this.parentNode as any)?.__removeAnonymousFunction?.(this);
+        }
+        return params;
+    }
+    /**
+     * 设置匿名函数描述
+     */
+    setDescription(description: string) {
+        const object = {
+            description,
+        };
+        this.update({
+            ...object,
+        });
+    }
+
+    /**
+     * 设置逻辑项
+     */
+    setBody(body: LogicItem) {
+        body._delete();
+        const relationOptions = { parentNode: this, parentKey: 'body' };
+        Object.assign(body, relationOptions);
+        const object = {
+            body,
+        };
+        this.update({
+            ...object,
+        });
+    }
+
+        getParamUniqueName(name = 'param1') {
+        return utils.unique(name, this.getVarExistingNames(), undefined, false);
+    }
+
+    /**
+    * 插入输入参数
+    * @internal
+    * @param name 输入参数名称，如果不填会自动生成一个唯一名称
+    */
+    _insertParamAt(name: string, index: number): Param;
+
+    /**
+    * 插入输入参数
+    * @internal
+    * @param paramOptions 输入参数参数
+    */
+    _insertParamAt(paramOptions: Partial<Param>, index: number): Param;
+
+    /**
+    * 插入输入参数
+    * @internal
+    * @param param 已有的输入参数实例
+    */
+    _insertParamAt(param: Param, index: number): Param;
+
+    _insertParamAt(options: string | Partial<Param> | Param, index: number) {
+        const paramOptions: any = {};
+        const relationOptions = { parentNode: this, parentKey: 'params' };
+        let param: Param;
+        if (!options) {
+            param = Param.from({
+                ...paramOptions,
+                name: this.getParamUniqueName(),
+            }, this, 'params');
+        } else if (typeof options === 'string') {
+            param = Param.from({
+                ...paramOptions,
+                name: options,
+            }, this, 'params');
+        } else if (options instanceof Param) {
+            options.ensureDelete(); // 同一实例不支持多处存在
+            param = options;
+            Object.assign(param, relationOptions);
+        } else {
+            param = Param.from({
+                ...paramOptions,
+                ...options,
+            }, this, 'params');
+        }
+        this.params.splice(index, 0, param);
+        return param;
+    }
+
+    /**
+     * 插入输入参数
+     * @param name 输入参数名称，如果不填会自动生成一个唯一名称
+     */
+    insertParamAt(name: string, index: number): Param;
+
+    /**
+     * 插入输入参数
+     * @param paramOptions 输入参数参数
+     */
+    insertParamAt(paramOptions: Partial<Param>, index: number): Param;
+
+    /**
+     * 插入输入参数
+     * @param param 已有的输入参数实例
+     */
+    insertParamAt(param: Param, index: number): Param;
+
+    insertParamAt(options: string | Partial<Param> | Param, index: number) {
+        const node = this._insertParamAt(options as any, index);
+        node.create({
+            index,
+            parentNode: this,
+            parentKey: 'params',
+        });
+        return node;
+    }
+
+    /**
+     * 添加输入参数
+     * @internal
+     * @param name 输入参数名称，如果不填会自动生成一个唯一名称
+     */
+    _addParam(name?: string): Param;
+
+    /**
+     * 添加输入参数
+     * @internal
+     * @param paramOptions 输入参数参数
+     */
+    _addParam(paramOptions: Partial<Param>): Param;
+
+    /**
+     * 添加输入参数
+     * @internal
+     * @param param 已有的输入参数实例
+     */
+    _addParam(param: Param): Param;
+
+    _addParam(options?: string | Partial<Param> | Param) {
+        const index = this.params.length;
+        return this._insertParamAt(options as any, index);
+    }
+
+    /**
+     * 添加输入参数
+     * @internal
+     * @param name 输入参数名称，如果不填会自动生成一个唯一名称
+     */
+    addParam(name?: string): Param;
+
+    /**
+     * 添加输入参数
+     * @param paramOptions 输入参数参数
+     */
+    addParam(paramOptions: Partial<Param>): Param;
+
+    /**
+     * 添加输入参数
+     * @param param 已有的输入参数实例
+     */
+    addParam(param: Param): Param;
+
+    addParam(options?: string | Partial<Param> | Param) {
+        const node = this._addParam(options as any);
+        const index = this.params.indexOf(node);
+        node.create({
+            index,
+            parentNode: this,
+            parentKey: 'params',
+        });
+        return node;
+    }
+
+
+    
+
+    /**
+    * 删除输入参数
+    * @param name 输入参数名称
+    */
+    removeParam(name: string): void;
+
+    /**
+     * 删除输入参数
+     * @param param 已有的输入参数实例
+     */
+    removeParam(param: Param): void;
+
+    removeParam(options: string | Param) {
+        let param: Param;
+        if (typeof options === 'string') {
+            param = this.params.find((item) => item.name === options);
+            if (!param) {
+                throw new Error('找不到输入参数 ' + options);
+            }
+        } else {
+            param = options;
+        }
+        return param.delete();
+    }
+
+    __removeParam(param: Param) {
+        const parentKey = param.parentKey;
+        const params: Params = {
+            parentNode: this,
+            parentKey,
+            index: -1,
+            object: null,
+            oldObject: param,
+        };
+        if (parentKey) {
+            params.parentKey = parentKey;
+            if (Array.isArray((this as any)[parentKey])) {
+                const index = (this as any)[parentKey].indexOf(param);
+                ~index && (this as any)[parentKey].splice(index, 1);
+                params.index = index;
+            } else if ((this as any)[parentKey] === param) {
+                params.index = 0;
+                (this as any)[parentKey] = undefined;
+            }
+        }
+        return params;
+    }
+
+
+
+    /**
+    * 删除逻辑项
+    * @param name 逻辑项名称
+    */
+    removeBody(name: string): void;
+
+    /**
+     * 删除逻辑项
+     * @param body 已有的逻辑项实例
+     */
+    removeBody(body: LogicItem): void;
+
+    removeBody(options: string | LogicItem) {
+        let body: LogicItem;
+        if (typeof options === 'string') {
+            body = this.body;
+            if (!body) {
+                throw new Error('找不到逻辑项 ' + options);
+            }
+        } else {
+            body = options;
+        }
+        return body.delete();
+    }
+
+    __removeLogicItem(logicItem: LogicItem) {
+        const parentKey = logicItem.parentKey;
+        const params: Params = {
+            parentNode: this,
+            parentKey,
+            index: -1,
+            object: null,
+            oldObject: logicItem,
+        };
+        if (parentKey) {
+            params.parentKey = parentKey;
+            if (Array.isArray((this as any)[parentKey])) {
+                const index = (this as any)[parentKey].indexOf(logicItem);
+                ~index && (this as any)[parentKey].splice(index, 1);
+                params.index = index;
+            } else if ((this as any)[parentKey] === logicItem) {
+                params.index = 0;
+                (this as any)[parentKey] = undefined;
+            }
+        }
+        return params;
+    }
+
+
+
+    /**
+    * 删除类型标注
+    * @param name 类型标注名称
+    */
+    removeTypeAnnotation(name: string): void;
+
+    /**
+     * 删除类型标注
+     * @param typeAnnotation 已有的类型标注实例
+     */
+    removeTypeAnnotation(typeAnnotation: TypeAnnotation): void;
+
+    removeTypeAnnotation(options: string | TypeAnnotation) {
+        let typeAnnotation: TypeAnnotation;
+        if (typeof options === 'string') {
+            typeAnnotation = this.typeAnnotation;
+            if (!typeAnnotation) {
+                throw new Error('找不到类型标注 ' + options);
+            }
+        } else {
+            typeAnnotation = options;
+        }
+        return typeAnnotation.delete();
+    }
+
+    __removeTypeAnnotation(typeAnnotation: TypeAnnotation) {
+        const parentKey = typeAnnotation.parentKey;
+        const params: Params = {
+            parentNode: this,
+            parentKey,
+            index: -1,
+            object: null,
+            oldObject: typeAnnotation,
+        };
+        if (parentKey) {
+            params.parentKey = parentKey;
+            if (Array.isArray((this as any)[parentKey])) {
+                const index = (this as any)[parentKey].indexOf(typeAnnotation);
+                ~index && (this as any)[parentKey].splice(index, 1);
+                params.index = index;
+            } else if ((this as any)[parentKey] === typeAnnotation) {
+                params.index = 0;
+                (this as any)[parentKey] = undefined;
+            }
+        }
+        return params;
+    }
+
+    //================================================================================
+    // ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
+    // 自动生成的代码已结束。下面可以手动编写。
+    //================================================================================
+
+    toJS(state?: TranslatorState): String {
+        let code = '';
+        code += '(';
+        this.params.forEach((param, index) => {
+            code += param.toJS(shiftState(state, code));
+            if (index !== this.params.length - 1) {
+                code += ',';
+            }
+        });
+        code += ')';
+        code += ' => (';
+        if (this.body) {
+            code += this.body.toJS(shiftState(state, code));
+        }
+        code += ')';
+        return code;
+    }
+
+    getVarExistingNames(excludedList: Array<Param> = []) {
+        const excludedSet = new Set(excludedList);
+        return [...this.params].filter((item) => !excludedSet.has(item)).map((item) => item.name);
+    }
+    getNamespace(): string {
+        if (this.parentNode) {
+            let parentNodeNamespace = '';
+            if ((this.parentNode as Namespace).getNamespace) {
+                parentNodeNamespace = [(this.parentNode as Namespace).getNamespace(), this.parentNode.name].join('.');
+            }
+            return `${parentNodeNamespace}`;
+        } else
+            throw new Error('无法获取命名空间，请设置 parentNode！');
+    }
+    @withSourceMap()
+    toEmbeddedTS(state?: TranslatorState): string {
+        let code = '';
+        if (this.body) {
+            code += `(`;
+            this.params.forEach((param, index) => {
+                code += param.toEmbeddedTS(shiftState(state, code, { inline: true }));
+                if (index !== this.params.length - 1)
+                    code += ', ';
+            });
+            code += `)`;
+            code += ` => (`;
+
+            if (this.body) {
+                code += this.body.toEmbeddedTS && this.body.toEmbeddedTS(shiftState(state, code, { tabSize: state.tabSize + 1 }));
+            }
+            code += indent(state.tabSize) + ')';
+        } else {
+            code += '__IDENTIFIER__';
+        }
+
+        return code;
+    }
+
+    toVue(options?: ElementToVueOptions): string {
+        let code = '(';
+        this.params.forEach((param, index) => {
+            code += param.name;
+            if (index !== this.params.length - 1)
+                code += ', ';
+        });
+        code += `)`;
+        code += ` => (`;
+
+        if (this.body) {
+            code += this.body.toVue && this.body.toVue();
+        }
+        code += ')';
+        return code;
+    }
+
+    toUI(state?: TranslatorState): string {
+        let code = '(';
+        this.params.forEach((param, index) => {
+            code += param.name;
+            if (index !== this.params.length - 1)
+                code += ', ';
+        });
+        code += `)`;
+        code += ` => (`;
+
+        if (this.body) {
+            code += this.body.toUI && this.body.toUI();
+        }
+        code += ')';
+        return code;
+    }
+
+    toEmbeddedTSDefinition(state?: TranslatorState): string {
+        let code = '(';
+        this.params.forEach((param, index) => {
+            code += param.toEmbeddedTS(shiftState(state, code, { inline: true }));
+            if (index !== this.params.length - 1)
+                code += ', ';
+        });
+        code += `) => (`;
+        code += 'any';
+        code += indent(state.tabSize) + ')';
+        return code;
+    }
+
+    checkParamName() {
+        let node: BaseNode = this;
+        const names: Set<string> = new Set();
+        while(node && node.concept !== 'Logic') {
+            node = node.parentNode;
+            if(node instanceof AnonymousFunction) {
+                node.params.forEach((param) => {
+                    names.add(param.name);
+                });
+            }
+
+            if(node instanceof ForEachStatement) {
+                names.add(node.item.name);
+                names.add(node.index.name);
+            }
+        }
+
+        this.params.forEach((param) => {
+            const name = utils.unique(param.name, names);
+            param.name = name;
+        });
+    }
+
+    //================================================================================
+    // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
+    // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
+    //================================================================================
+}
+
+classMap.AnonymousFunction = AnonymousFunction;
+export default AnonymousFunction;
+//================================================================================
+// ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑   ↑
+// 自动生成的代码已结束。下面可以手动编写。
+//================================================================================

package/src/concepts/App__.ts

@@ -6,7 +6,6 @@ export interface AssetsInfo {
 import { queryNodeByPath } from '../service/storage/jsoner';
 import { getNaslNodeByNodeCallee } from '../automate/engine/utils';
 import ConfigProperty from './ConfigProperty__';
-
 //================================================================================
 // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
 // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
@@ -2865,11 +2864,6 @@ ${keys.map((key) => `    ${key}: ${this.theme?.variableMap[key]};`).join('\n')}
         };
     }
 
-    /**
-     * 标注全量类型后的json
-     */
-    @excludedInJSON()
-    NaslAnnotatedString: App = undefined;
     //================================================================================
     // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
     // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓

package/src/concepts/Argument__.ts

@@ -3,6 +3,7 @@ import { ElementToVueOptions } from './ViewElement__';
 import StringLiteral from './StringLiteral__';
 import TypeAnnotation from './TypeAnnotation__';
 
+import CallFunction from './CallFunction__';
 //================================================================================
 // 从这里开始到结尾注释之间的代码由 NASL Workbench 自动生成，请不手动修改！
 // ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓   ↓
@@ -133,6 +134,13 @@ export class Argument extends LogicItem {
     // 自动生成的代码已结束。下面可以手动编写。
     //================================================================================
 
+    preSetExpression(expression: LogicItem) {
+        (this.parentNode as CallFunction).detectArgs?.({
+            expression,
+        });
+        this.setExpression(expression);
+    }
+
     /**
      * 设置keyword
      */
@@ -196,9 +204,9 @@ export class Argument extends LogicItem {
         return code;
     }
 
-    toBrief(): string { 
+    toBrief(): string {
         const expr = utils.toBriefExpr(this.expression);
-        return `<span style="color: #AAAAAA">${this.keyword} = </span><span style="color: #FFFFFF">${expr}</span>`
+        return `<span style="color: #AAAAAA">${this.keyword} = </span><span style="color: #FFFFFF">${expr}</span>`;
     }
 
     /**