@lcap/nasl 3.2.0-beta.4 → 3.3.0-alpha

package/src/generator/genReleaseBody.ts

@@ -1,83 +1,109 @@
-import { App, Assignment, BatchAssignment, CallFunction, CallInterface, CallLogic, ForEachStatement, IfStatement, Logic, LogicItem, Match, MatchCase, MemberExpression, Role, SwitchCase, SwitchStatement, View, ViewElement, WhileStatement, StringInterpolation, Frontend, ExternalDestination, NewComposite, SelectMembers, NewList } from '../concepts';
+import {
+  App,
+  Assignment,
+  BatchAssignment,
+  CallFunction,
+  CallInterface,
+  CallLogic,
+  ForEachStatement,
+  IfStatement,
+  Logic,
+  LogicItem,
+  Match,
+  MatchCase,
+  MemberExpression,
+  Role,
+  SwitchCase,
+  SwitchStatement,
+  View,
+  ViewElement,
+  WhileStatement,
+  StringInterpolation,
+  Frontend,
+  ExternalDestination,
+  NewComposite,
+  SelectMembers,
+  NewList,
+} from '../concepts';
 import { genFrontendBundleFiles } from './genBundleFiles';
 import { genPermissionData, genLogicAuthFlag } from './permission';
 import * as utils from '../utils';
 
 interface ServerValidationData {
-    rules: object[];
-    failureRules: object[];
-    uuid: string;
+  rules: object[];
+  failureRules: object[];
+  uuid: string;
+}
+
+interface FileData {
+  name: string;
+  content: string;
 }
 
 interface ReleaseData {
-    env: string,
-    debug: boolean,
-    replicas: number,
-    realRelease: boolean,
-    appInfo: {
-        tenant: string,
-        nuimsDomain: string,
-        envNuimsDomain: string,
-        STATIC_URL: string,
-        USER_STATIC_URL: string,
-        extendedConfig: string,
-        lowcodeDomain: string,
-        envLcpDomain: string,
-        tenantType: string,
-        tenantLabel: string,
-    },
-    staticUrl?: string,
-    frontends: Array<Frontend>,
-    validations: InputServerValidationData,
-    logPublishFunc: Function,
+  env: string;
+  debug: boolean;
+  replicas: number;
+  realRelease: boolean;
+  appInfo: {
+    tenant: string;
+    nuimsDomain: string;
+    envNuimsDomain: string;
+    STATIC_URL: string;
+    USER_STATIC_URL: string;
+    extendedConfig: string;
+    lowcodeDomain: string;
+    envLcpDomain: string;
+    tenantType: string;
+    tenantLabel: string;
+  };
+  staticUrl?: string;
+  frontends: Array<Frontend>;
+  validations: InputServerValidationData;
+  logPublishFunc: Function;
 }
 
 type InputServerValidationData = { [key in string]: ServerValidationData[] };
 
 function getValidationMap(serverData: InputServerValidationData) {
-    const map = new Map<string, ServerValidationData>();
+  const map = new Map<string, ServerValidationData>();
 
-    Object.values(serverData).forEach((logic) => {
-        logic.forEach((item) => {
-            map.set(item.uuid, item);
-        });
+  Object.values(serverData).forEach((logic) => {
+    logic.forEach((item) => {
+      map.set(item.uuid, item);
     });
+  });
 
-    return map;
+  return map;
 }
 
 function clearValidationMap(serverData: InputServerValidationData) {
-    Object.values(serverData).forEach((logic) => {
-        logic.forEach((item) => {
-            item.rules = [];
-            item.failureRules = [];
-        });
+  Object.values(serverData).forEach((logic) => {
+    logic.forEach((item) => {
+      item.rules = [];
+      item.failureRules = [];
     });
+  });
 
-    return serverData;
+  return serverData;
 }
 
 function mergeValidation(serverData: InputServerValidationData, nasl: App) {
-    const map = getValidationMap(serverData);
+  const map = getValidationMap(serverData);
 
-    if (map.size === 0) {
-        return {};
-    }
+  if (map.size === 0) {
+    return {};
+  }
 
-    nasl.traverseChildren((node) => {
-        if (
-            node.concept === 'CallLogic' &&
-            'uuid' in node &&
-            node.validation &&
-            map.has(node.uuid)
-        ) {
-            const data = map.get(node.uuid);
-            const naslRules = node.validation.rules.map((rule) => rule.toJSON());
-            data.rules = [...naslRules, ...data.rules];
-        }
-    });
+  nasl.traverseChildren((node) => {
+    if (node.concept === 'CallLogic' && 'uuid' in node && node.validation && map.has(node.uuid)) {
+      const data = map.get(node.uuid);
+      const naslRules = node.validation.rules.map((rule) => rule.toJSON());
+      data.rules = [...naslRules, ...data.rules];
+    }
+  });
 
-    return serverData;
+  return serverData;
 }
 
 /**
@@ -85,278 +111,273 @@ function mergeValidation(serverData: InputServerValidationData, nasl: App) {
  * @description 之后可能移除
  */
 function removePasswordValidation(serverData: InputServerValidationData, nasl: App) {
-    const validationMap = getValidationMap(serverData);
-
-    if (validationMap.size === 0) {
-        return {};
+  const validationMap = getValidationMap(serverData);
+
+  if (validationMap.size === 0) {
+    return {};
+  }
+
+  const passwordCallLogicUuid = new Set<string>();
+  const removeRuleNames = [
+    'rangeLength',
+    'pattern',
+    '^az',
+    '^az09',
+    '^az09_',
+    '^azAZ',
+    '^azAZ09',
+    '^azAZ09_',
+    'az09$',
+    'azAZ09$',
+    '^09$',
+    '^az09$',
+    '^az09-$',
+    '^az09-.$',
+    '^azAZ09$',
+    '^azAZ09-$',
+    '^azAZ09_$',
+    '^azAZ09-_$',
+  ];
+
+  // 获取所有调用加密函数的调用节点 uuid
+  nasl.traverseChildren((node) => {
+    if (node.concept !== 'CallLogic' || !('uuid' in node)) {
+      return;
     }
 
-    const passwordCallLogicUuid = new Set<string>();
-    const removeRuleNames = [
-        "rangeLength",
-        "pattern",
-        "^az",
-        "^az09",
-        "^az09_",
-        "^azAZ",
-        "^azAZ09",
-        "^azAZ09_",
-        "az09$",
-        "azAZ09$",
-        "^09$",
-        "^az09$",
-        "^az09-$",
-        "^az09-.$",
-        "^azAZ09$",
-        "^azAZ09-$",
-        "^azAZ09_$",
-        "^azAZ09-_$",
-    ];
-
-    // 获取所有调用加密函数的调用节点 uuid
-    nasl.traverseChildren((node) => {
-        if (node.concept !== 'CallLogic' || !('uuid' in node)) {
-            return;
-        }
-
-        const calledLogicPath = `${node.calleeNamespace}.${node.calleeName}`;
-
-        if (
-            Boolean(node.uuid) &&
-            (
-                calledLogicPath === 'app.dataSources.defaultDS.entities.LCAPUser.logics.update' ||
-                calledLogicPath === 'app.dataSources.defaultDS.entities.LCAPUser.logics.create' ||
-                calledLogicPath.includes('encrypt')
-            )
-        ) {
-            passwordCallLogicUuid.add(node.uuid);
-        }
-    });
+    const calledLogicPath = `${node.calleeNamespace}.${node.calleeName}`;
 
-    if (passwordCallLogicUuid.size === 0) {
-        return serverData;
+    if (
+      Boolean(node.uuid) &&
+      (calledLogicPath === 'app.dataSources.defaultDS.entities.LCAPUser.logics.update' ||
+        calledLogicPath === 'app.dataSources.defaultDS.entities.LCAPUser.logics.create' ||
+        calledLogicPath.includes('encrypt'))
+    ) {
+      passwordCallLogicUuid.add(node.uuid);
     }
+  });
 
-    passwordCallLogicUuid.forEach((uuid) => {
-        const validation = validationMap.get(uuid);
+  if (passwordCallLogicUuid.size === 0) {
+    return serverData;
+  }
 
-        if (!validation) {
-            return;
-        }
+  passwordCallLogicUuid.forEach((uuid) => {
+    const validation = validationMap.get(uuid);
 
-        for (let i = 0; i < validation.rules.length; i++) {
-            const current = validation.rules[i];
+    if (!validation) {
+      return;
+    }
 
-            if (removeRuleNames.includes((current as any).calleeName)) {
-                validation.rules.splice(i, 1);
-                i--;
-            }
-        }
-    });
+    for (let i = 0; i < validation.rules.length; i++) {
+      const current = validation.rules[i];
 
-    return serverData;
+      if (removeRuleNames.includes((current as any).calleeName)) {
+        validation.rules.splice(i, 1);
+        i--;
+      }
+    }
+  });
+
+  return serverData;
 }
 
 export async function genReleaseBody(app: App, releaseData: ReleaseData) {
-    const {
-        env,
-        debug,
-        replicas,
-        realRelease,
-        appInfo,
-        staticUrl,
-        frontends = [],
-        validations = {},
-        logPublishFunc = (): void => undefined,
-    } = releaseData;
-    const {
-        preferenceMap: {
-            serverValidationRules,
-        },
-    } = app;
-
-    const authReport: ReturnType<typeof getAuthReport> = getAuthReport(app, frontends);
-    logPublishFunc(`[权限] "[${app.title}]" 分析权限数据成功！`);
-    const logicPageResourceDtoList = await genPermissionData(app);
-
-    const releaseFrontends = app.frontends.map((frontend) => ({
-        name: frontend.name,
-        type: frontend.type,
-        path: frontend.path,
-        selected: frontends.includes(frontend),
-        title: frontend.title,
-    }));
-
-    const serverValidationRulesBySwitch = (
-        serverValidationRules === 'custom'
-            // 自定义
-            ? mergeValidation(validations, app)
-            : serverValidationRules === 'auto'
-                // 自动
-                ? validations
-                // 关闭
-                : clearValidationMap(validations)
-    );
-    const removedPasswordValidation = removePasswordValidation(serverValidationRulesBySwitch, app);
-    const body = {
-        ...authReport, // 如果发布需要上报权限，导出源码不需要
-        logicPageResourceDtoList,
-        logicAuthFlag: genLogicAuthFlag(app),
-        reportIdList: findAllReportIdList(app, frontends),
-        appId: app.id,
-        env,
-        replicas,
-        callbackLogicsName: app.getExtensionsCallbackLogics(),
-        frontends: releaseFrontends,
-        callLogicValidations: removedPasswordValidation,
-        files: [] as Array<{ name: string, content: string }>,
-    };
-
-    if (logicPageResourceDtoList) {
-        Object.assign(body, { logicPageResourceDtoList });
+  const {
+    env,
+    debug,
+    replicas,
+    realRelease,
+    appInfo,
+    staticUrl,
+    frontends = [],
+    validations = {},
+    logPublishFunc = (): void => undefined,
+  } = releaseData;
+  const {
+    preferenceMap: { serverValidationRules },
+  } = app;
+
+  const authReport: ReturnType<typeof getAuthReport> = getAuthReport(app, frontends);
+  logPublishFunc(`[权限] "[${app.title}]" 分析权限数据成功！`);
+  const logicPageResourceDtoList = await genPermissionData(app);
+
+  const releaseFrontends = app.frontends.map((frontend) => ({
+    name: frontend.name,
+    type: frontend.type,
+    path: frontend.path,
+    selected: frontends.includes(frontend),
+    title: frontend.title,
+  }));
+
+  const serverValidationRulesBySwitch =
+    serverValidationRules === 'custom'
+      ? // 自定义
+        mergeValidation(validations, app)
+      : serverValidationRules === 'auto'
+      ? // 自动
+        validations
+      : // 关闭
+        clearValidationMap(validations);
+  const removedPasswordValidation = removePasswordValidation(serverValidationRulesBySwitch, app);
+  const body = {
+    ...authReport, // 如果发布需要上报权限，导出源码不需要
+    logicPageResourceDtoList,
+    logicAuthFlag: genLogicAuthFlag(app),
+    reportIdList: findAllReportIdList(app, frontends),
+    appId: app.id,
+    env,
+    replicas,
+    callbackLogicsName: app.getExtensionsCallbackLogics(),
+    frontends: releaseFrontends,
+    callLogicValidations: removedPasswordValidation,
+    files: [] as Array<{ name: string; content: string }>,
+  };
+
+  if (logicPageResourceDtoList) {
+    Object.assign(body, { logicPageResourceDtoList });
+  }
+
+  let NaslAnnotatedJSON;
+  try {
+    NaslAnnotatedJSON = await (app as any).naslServer.getNaslAnnotatedJSON(app, true);
+    logPublishFunc(`[语言] "[${app.title}]" 获取类型标注成功！`);
+  } catch (err) {
+    console.error(err);
+    throw new Error('代码标注失败，请稍后重试！或刷新后重试！');
+  }
+
+  // 可能所有的前端都不选
+  let files: FileData[] = [];
+  if (frontends && frontends.length) {
+    logPublishFunc(`[代码] "[${app.title}]" 生成页面代码：开始`);
+    files = genFrontendBundleFiles(app, frontends, {
+      ...appInfo,
+      env,
+      debug,
+      appid: app.id,
+      devDnsAddr: (app as any).devDnsAddr,
+      miniEnable: (app as any).miniEnable,
+    } as any);
+    if (files && files.length === 0) {
+      throw new Error('前端代码生成失败');
     }
 
-    let NaslAnnotatedJSON;
-    try {
-        NaslAnnotatedJSON = await (app as any).naslServer.getNaslAnnotatedJSON(app, true);
-        logPublishFunc(`[语言] "[${app.title}]" 获取类型标注成功！`);
-    } catch (err) {
-        console.error(err);
-        throw new Error('代码标注失败，请稍后重试！或刷新后重试！');
-    }
+    if (staticUrl) files = replaceAssetUrl(files, appInfo.USER_STATIC_URL, staticUrl);
+    logPublishFunc(`[代码] "[${app.title}]" 生成页面代码：完成！`);
+  }
 
-    // 可能所有的前端都不选
-    let files = [];
-    if (frontends && frontends.length) {
-        logPublishFunc(`[代码] "[${app.title}]" 生成页面代码：开始`);
-        files = genFrontendBundleFiles(app, frontends, Object.assign({}, appInfo, {
-            env,
-            debug,
-            appid: app.id,
-            devDnsAddr: (app as any).devDnsAddr,
-            miniEnable: (app as any).miniEnable,
-        }) as any);
-        if (files && files.length === 0) {
-            throw new Error('前端代码生成失败');
-        }
-
-        if (staticUrl)
-            files = replaceAssetUrl(files, appInfo.USER_STATIC_URL, staticUrl);
-        logPublishFunc(`[代码] "[${app.title}]" 生成页面代码：完成！`);
-    }
+  // 如果是发布，就排除views，导出源码就不排除
+  if (realRelease) {
+    NaslAnnotatedJSON.frontends.forEach((frontend: Frontend) => {
+      delete frontend.views;
+    });
+  }
+  const jsonString = JSON.stringify(NaslAnnotatedJSON);
 
-    // 如果是发布，就排除views，导出源码就不排除
-    if (realRelease) {
-        NaslAnnotatedJSON.frontends.forEach((frontend: Frontend) => {
-            delete frontend.views;
-        });
-    }
-    const jsonString = JSON.stringify(NaslAnnotatedJSON);
+  // 打印传给服务端的nasl,为了服务端排查问题方便
+  if (globalThis.window) {
+    console.info(NaslAnnotatedJSON, '标注后的nasl');
+  }
 
-    // 打印传给服务端的nasl,为了服务端排查问题方便
-    if (globalThis.window) {
-        console.info(NaslAnnotatedJSON, '标注后的nasl');
-    }
+  files.push({ name: 'nasl-annotated.json', content: jsonString });
+  body.files = files;
 
-    files.push({ name: 'nasl-annotated.json', content: jsonString });
-    body.files = files;
-
-    return body;
+  return body;
 }
 
 /**
  * 权限上报数据
  */
 function getAuthReport(app: App, frontends: Array<Frontend>) {
-    const result = {
-        roleList: [] as Array<Role>,
-        resourceList: [] as Array<string>,
-        resourceDataList: [] as Array<{ value: string, description: string, type: 'page' | 'component' | 'logic', clientType?: string }>,
-        roleResourceMappingList: [] as Array<{ roleName: string, resourcePath: string }>,
-    };
-
-    result.roleList = app.roles;
-
-    const getAuth = (target: ViewElement) => {
-        const authDirective = target.bindDirectives.find((item) => item.name === 'auth');
-        return authDirective?.value;
-    };
-
-    const loopViews = (view: View | ViewElement, isMain: boolean, frontend: Frontend) => {
-        if (view.children) {
-            view.children.forEach((item) => loopViews(item, false, frontend));
-        }
-        if (view.concept === 'View' && view.elements) {
-            view.elements.forEach((item) => loopViews(item, false, frontend));
-        }
+  const result = {
+    roleList: [] as Array<Role>,
+    resourceList: [] as Array<string>,
+    resourceDataList: [] as Array<{ value: string; description: string; type: 'page' | 'component' | 'logic'; clientType?: string }>,
+    roleResourceMappingList: [] as Array<{ roleName: string; resourcePath: string }>,
+  };
+
+  result.roleList = app.roles;
+
+  const getAuth = (target: ViewElement) => {
+    const authDirective = target.bindDirectives.find((item) => item.name === 'auth');
+    return authDirective?.value;
+  };
+
+  const loopViews = (view: View | ViewElement, isMain: boolean, frontend: Frontend) => {
+    if (view.children) {
+      view.children.forEach((item) => loopViews(item, false, frontend));
+    }
+    if (view.concept === 'View' && view.elements) {
+      view.elements.forEach((item) => loopViews(item, false, frontend));
+    }
 
-        if ((view.concept === 'View' && view.auth)
-            || (view.concept === 'View' && !isMain && view.parentAuth)
-            || (view.concept === 'ViewElement' && getAuth(view))
-        ) {
-            result.resourceList.push(view.authPath);
-            result.resourceDataList.push({
-                value: view.authPath,
-                description: view.authDescription,
-                type: view.concept === 'View' ? 'page' : 'component',
-                clientType: frontend.name,
-            });
-            if (view.bindRoles) {
-                const _arr = view.bindRoles.map((it) => ({
-                    roleName: it,
-                    resourcePath: view.authPath,
-                }));
-
-                result.roleResourceMappingList.push(..._arr);
-            }
-        }
-    };
-    frontends.forEach((frontend) => {
-        frontend.views.forEach((view) => loopViews(view, true, frontend));
-    });
+    if (
+      (view.concept === 'View' && view.auth) ||
+      (view.concept === 'View' && !isMain && view.parentAuth) ||
+      (view.concept === 'ViewElement' && getAuth(view))
+    ) {
+      result.resourceList.push(view.authPath);
+      result.resourceDataList.push({
+        value: view.authPath,
+        description: view.authDescription,
+        type: view.concept === 'View' ? 'page' : 'component',
+        clientType: frontend.name,
+      });
+      if (view.bindRoles) {
+        const _arr = view.bindRoles.map((it) => ({
+          roleName: it,
+          resourcePath: view.authPath,
+        }));
+
+        result.roleResourceMappingList.push(..._arr);
+      }
+    }
+  };
+  frontends.forEach((frontend) => {
+    frontend.views.forEach((view) => loopViews(view, true, frontend));
+  });
 
-    return result;
+  return result;
 }
 
 /**
  * 获取页面中所有 分析报告组件 的reportId
  */
 function findAllReportIdList(app: App, frontends: Array<Frontend>) {
-    const reportIds: string[] = [];
-    frontends.forEach((frontend) => {
-        frontend.views.forEach((view) => loop(view));
-    });
-
-    function loop(root: View | ViewElement) {
-        if (root.concept === 'ViewElement') {
-            root.bindAttrs.forEach((attr) => {
-                if (attr.name === 'reportId') {
-                    reportIds.push(JSON.parse(attr.value || '{}').id);
-                }
-            });
-        }
-        if (root.children && root.children.length) {
-            root.children.forEach((item) => loop(item));
-        }
-        if (root.concept === 'View' && root.elements && root.elements.length) {
-            root.elements.forEach((item) => loop(item));
+  const reportIds: string[] = [];
+  frontends.forEach((frontend) => {
+    frontend.views.forEach((view) => loop(view));
+  });
+
+  function loop(root: View | ViewElement) {
+    if (root.concept === 'ViewElement') {
+      root.bindAttrs.forEach((attr) => {
+        if (attr.name === 'reportId') {
+          reportIds.push(JSON.parse(attr.value || '{}').id);
         }
+      });
+    }
+    if (root.children && root.children.length) {
+      root.children.forEach((item) => loop(item));
     }
-    return [...new Set(reportIds)].filter(Boolean);
+    if (root.concept === 'View' && root.elements && root.elements.length) {
+      root.elements.forEach((item) => loop(item));
+    }
+  }
+  return [...new Set(reportIds)].filter(Boolean);
 }
 
-function replaceAssetUrl(files: Array<{ name: string, content: string }>, oldUrlWithoutProtocol: string, newUrlWithProtocol: string) {
-    const newUrlWithoutProtocol = newUrlWithProtocol.replace(/^https?:/, '');
-    if (newUrlWithoutProtocol === oldUrlWithoutProtocol)
-        return files;
-
-    const reg = new RegExp(`${oldUrlWithoutProtocol}[^?'"\\s]*\\?[^?'"\\s]*(?=("|'))`, 'g');
-    const reg2 = new RegExp(`(https?:)?${oldUrlWithoutProtocol}`, 'g');
-    for (const file of files) {
-        // 清除 query 参数
-        file.content = file.content.replace(reg, (...args) => args[0].split('?')[0]);
-        file.content = file.content.replace(reg2, (whole, protocol) => (protocol ? newUrlWithProtocol : newUrlWithoutProtocol));
-        file.name = file.name.replace(reg2, (whole, protocol) => (protocol ? newUrlWithProtocol : newUrlWithoutProtocol));
-    }
-    return files;
+function replaceAssetUrl(files: Array<{ name: string; content: string }>, oldUrlWithoutProtocol: string, newUrlWithProtocol: string) {
+  const newUrlWithoutProtocol = newUrlWithProtocol.replace(/^https?:/, '');
+  if (newUrlWithoutProtocol === oldUrlWithoutProtocol) return files;
+
+  const reg = new RegExp(`${oldUrlWithoutProtocol}[^?'"\\s]*\\?[^?'"\\s]*(?=("|'))`, 'g');
+  const reg2 = new RegExp(`(https?:)?${oldUrlWithoutProtocol}`, 'g');
+  for (const file of files) {
+    // 清除 query 参数
+    file.content = file.content.replace(reg, (...args) => args[0].split('?')[0]);
+    file.content = file.content.replace(reg2, (whole, protocol) => (protocol ? newUrlWithProtocol : newUrlWithoutProtocol));
+    file.name = file.name.replace(reg2, (whole, protocol) => (protocol ? newUrlWithProtocol : newUrlWithoutProtocol));
+  }
+  return files;
 }