midway-fatcms 0.0.1-beta.55 → 0.0.1-beta.56

package/src/service/AuthService.ts

@@ -0,0 +1,284 @@
+import { Inject, Provide, Config } from '@midwayjs/core';
+import { Context } from '@midwayjs/koa';
+import { KeysOfSimpleSQL } from '@/libs/crud-pro/models/keys';
+import { CurdMixService } from './curd/CurdMixService';
+import * as md5 from 'md5';
+import { UserAccountService } from './UserAccountService';
+import { createUniqueId } from '@/libs/utils/functions';
+import { IConsumerUserInfo, ISessionInfo, SYS_ACCOUNT_TYPE } from '@/models/userSession';
+import { SystemTables } from '@/models/SystemTables';
+import { UserSessionService } from './UserSessionService';
+import { CommonResult } from '@/libs/utils/common-dto';
+import { isEnableSuperAdmin } from '@/libs/utils/fatcms-request';
+import { CommonException } from '@/libs/crud-pro/exceptions';
+import { GLOBAL_STATIC_CONFIG } from '@/libs/global-config/global-config';
+
+@Provide()
+export class AuthService {
+  @Inject()
+  private ctx: Context;
+
+  @Config('superAdminList')
+  private superAdminList: any[];
+
+  @Inject()
+  private curdMixService: CurdMixService;
+
+  @Inject()
+  private userService: UserAccountService;
+
+  @Inject()
+  private userSessionService: UserSessionService;
+
+  /**
+   * 明文密码转unsaltedPwd密码
+   * @param plainPwd
+   * @param loginName
+   */
+  public toUnsaltedPwd(plainPwd: string, loginName: string) {
+    const pw2 = md5(md5(plainPwd) + '_' + loginName);
+    return pw2;
+  }
+
+  /**
+   * front: pw1 = md5(md5(input_pw) + "_" + loginName)
+   * db: pw2 = md5(pwd_salt + pw1)
+   * 检查用户名密码是否正确
+   * @param loginName
+   * @param unsaltedPwd
+   */
+  public async checkLoginPassword(loginName: string, unsaltedPwd: string): Promise<CommonResult> {
+    const message = '用户名或密码错误';
+    const errorCode = 'USERNAME_OR_PASSWORD_IS_INCORRECT';
+
+    if (!loginName) {
+      return CommonResult.errorRes(message, errorCode, '!loginName');
+    }
+
+    if (!unsaltedPwd) {
+      return CommonResult.errorRes(message, errorCode, '!unsaltedPwd');
+    }
+
+    const userAccount = await this.queryUserAccountByLoginName(loginName);
+    if (!userAccount) {
+      return CommonResult.errorRes(message, errorCode, '!userAccount');
+    }
+
+    if (userAccount.status !== 1) {
+      return CommonResult.errorRes(message, errorCode, 'userAccount.status !== 1');
+    }
+
+    if (!userAccount.pwd_md5) {
+      return CommonResult.errorRes(message, errorCode, '!userAccount.pwd_md5');
+    }
+
+    const pwd_salt = userAccount.pwd_salt;
+    const pwd_md5 = md5(unsaltedPwd + pwd_salt);
+
+    if (userAccount['pwd_md5'] !== pwd_md5) {
+      return CommonResult.errorRes(message, errorCode, "userAccount['pwd_md5'] !== pwd_md5");
+    }
+
+    return CommonResult.successRes();
+  }
+
+  /**
+   * 创建一个toC端的UserSession
+   * @param consumerUserInfo
+   * @param loadRoleCodes
+   */
+  public async createUserSessionForConsumer(consumerUserInfo: IConsumerUserInfo, loadRoleCodes?: boolean ) {
+    const sessionId = createUniqueId();
+
+    const accountId = consumerUserInfo.accountId;
+
+    let roleCodes = [];
+    let functionCodes = [];
+
+    if(loadRoleCodes === true) {
+      roleCodes = await this.queryUserRoleCodeList(accountId);
+      functionCodes = await this.queryFunctionCodeList(roleCodes);
+    }
+
+
+    const sessionInfo: ISessionInfo = {
+      nickName: consumerUserInfo.nickName,
+      avatar: consumerUserInfo.avatar,
+      roleCodes: roleCodes,
+      functionCodes: functionCodes,
+      loginName: consumerUserInfo.loginName,
+      sessionId,
+      accountId: consumerUserInfo.accountId,
+      workbenchCode: consumerUserInfo.workbenchCode,
+      accountType: consumerUserInfo.accountType,
+    };
+
+    await this.userSessionService.saveUserSession(sessionInfo);
+
+    return {
+      sessionId,
+      loginName: sessionInfo.loginName,
+      accountId: sessionInfo.accountId,
+      nickName: sessionInfo.nickName,
+      avatar: sessionInfo.avatar,
+      workbenchCode: sessionInfo.workbenchCode,
+      accountType: sessionInfo.accountType,
+    };
+  }
+
+  /**
+   * 创建一个Session
+   * @param loginName
+   * @param workbenchCode
+   */
+  public async createUserSession(loginName: string, workbenchCode: string) {
+    const userAccount = await this.queryUserAccountByLoginName(loginName);
+
+    if (!userAccount) {
+      throw new CommonException('USER_ACCOUNT_NOT_EXIST', '用户账号不存在');
+    }
+
+    const accountId = userAccount.account_id;
+    const sessionId = createUniqueId();
+    const roleCodes = await this.queryUserRoleCodeList(accountId);
+    const functionCodes = await this.queryFunctionCodeList(roleCodes);
+
+    const sessionInfo: ISessionInfo = {
+      nickName: userAccount.nick_name,
+      avatar: userAccount.avatar,
+      roleCodes,
+      functionCodes,
+      loginName,
+      sessionId,
+      accountId,
+      workbenchCode,
+      accountType: SYS_ACCOUNT_TYPE,
+    };
+
+    await this.userSessionService.saveUserSession(sessionInfo);
+
+    return {
+      sessionId,
+      loginName,
+      accountId,
+      nickName: userAccount.nick_name,
+      avatar: userAccount.avatar,
+      workbenchCode: sessionInfo.workbenchCode,
+      accountType: sessionInfo.accountType,
+    };
+  }
+
+  /**
+   * 获取用户
+   * @param loginName
+   * @private
+   */
+  private async queryUserAccountByLoginName(loginName: string) {
+    const superAdminList = this.superAdminList;
+    const findFromSuperAdmin = () => {
+      if (!isEnableSuperAdmin(this.ctx)) {
+        return null;
+      }
+
+      if (!Array.isArray(superAdminList)) {
+        return null;
+      }
+
+      for (let i = 0; i < superAdminList.length; i++) {
+        const element = superAdminList[i];
+        if (element && element.login_name === loginName) {
+          const nick_name = 'SA' + i;
+          return {
+            nick_name: element.nick_name || nick_name,
+            avatar: '',
+            account_id: nick_name,
+            status: 1,
+            login_name: element.login_name,
+            pwd_salt: element.pwd_salt,
+            pwd_md5: element.pwd_md5,
+          };
+        }
+      }
+    };
+    const superAdmin = findFromSuperAdmin();
+    if (superAdmin) {
+      return superAdmin;
+    }
+    return this.userService.queryUserAccountByLoginName(loginName);
+  }
+
+  /**
+   * 查询此accountId下关联的角色code列表
+   * @param accountId
+   * @private
+   */
+  private async queryUserRoleCodeList(accountId: string): Promise<string[]> {
+    const { SystemDbName, SystemDbType } = GLOBAL_STATIC_CONFIG.getConfig();
+
+    const res = await this.curdMixService.executeCrudByCfg(
+      {
+        condition: { account_id: accountId },
+        limit: 10000,
+      },
+      {
+        sqlTable: SystemTables.sys_perm_user_role,
+        sqlSimpleName: KeysOfSimpleSQL.SIMPLE_QUERY,
+        sqlDatabase: SystemDbName,
+        sqlDbType: SystemDbType,
+      }
+    );
+    const rows = res.getResModel().rows || [];
+    const roleCodes: string[] = rows.map(({ role_code }) => {
+      return role_code as string;
+    });
+    return [...new Set(roleCodes)];
+  }
+
+  /**
+   * 查询此roleCodeList下面关联的所有功能点列表
+   * @param roleCodeList
+   * @private
+   */
+  private async queryFunctionCodeList(roleCodeList: string[]): Promise<string[]> {
+    if (!roleCodeList || roleCodeList.length === 0) {
+      return [];
+    }
+
+    const { SystemDbName, SystemDbType } = GLOBAL_STATIC_CONFIG.getConfig();
+
+    const res = await this.curdMixService.executeCrudByCfg(
+      {
+        condition: { role_code: { $in: roleCodeList } },
+        limit: 10000,
+      },
+      {
+        sqlTable: SystemTables.sys_perm_role_func,
+        sqlSimpleName: KeysOfSimpleSQL.SIMPLE_QUERY,
+        sqlDatabase: SystemDbName,
+        sqlDbType: SystemDbType,
+      }
+    );
+    const rows = res.getResModel().rows || [];
+    const funcCodes: string[] = rows.map(({ func_code }) => {
+      return func_code;
+    });
+    return [...new Set(funcCodes)];
+  }
+
+  async removeUserSession(sessionId: string) {
+    return this.userSessionService.removeUserSession(sessionId);
+  }
+
+  async refreshSession(sessionInfo: ISessionInfo): Promise<ISessionInfo> {
+    const oldSessionId = sessionInfo.sessionId;
+
+    const sessionId = createUniqueId();
+    const newSessionInfo: ISessionInfo = { ...sessionInfo, sessionId };
+
+    await this.userSessionService.saveUserSession(newSessionInfo);
+
+    await this.removeUserSession(oldSessionId);
+
+    return newSessionInfo;
+  }
+}

package/src/service/EnumInfoService.ts

@@ -0,0 +1,130 @@
+import { Inject, Provide } from '@midwayjs/core';
+import { Context } from '@midwayjs/koa';
+import { KeysOfSimpleSQL } from '@/libs/crud-pro/models/keys';
+import { CurdProService } from './curd/CurdProService';
+import { RelatedType } from './curd/CurdMixUtils';
+import { SystemTables } from '@/models/SystemTables';
+import { parseConfigContentToEnumInfo, IEnumInfo } from '@/libs/utils/parseConfig';
+import { RedisCacheService } from './base/RedisCacheService';
+import { CurdMixByLinkToCustomService } from './curd/CurdMixByLinkToCustomService';
+import { GLOBAL_STATIC_CONFIG } from '@/libs/global-config/global-config';
+import { RedisKeys } from '@/models/RedisKeys';
+
+interface IQueryEnumInfo {
+  type: RelatedType;
+  code: string;
+}
+interface IQueryEnumResult {
+  type: RelatedType;
+  code: string;
+  values: IEnumInfo[];
+}
+
+function filterEmpty(e: IEnumInfo): boolean {
+  return e.label && typeof e.value !== 'undefined';
+}
+
+@Provide()
+export class EnumInfoService {
+  @Inject()
+  ctx: Context;
+  @Inject()
+  private curdProService: CurdProService;
+
+  @Inject()
+  private curdMixByLinkToCustomService: CurdMixByLinkToCustomService;
+
+  @Inject()
+  private redisCacheService: RedisCacheService;
+
+  async queryEnumInfo(codeList: IQueryEnumInfo[], refreshCache: boolean): Promise<IQueryEnumResult[]> {
+    const expireSecond = 2 * 60;
+
+    const result = [];
+    for (let i = 0; i < codeList.length; i++) {
+      const { code, type } = codeList[i];
+      const cacheKey = `${RedisKeys.QUERY_ENUM_INFO_PREFIX}${type}@@${code}`;
+      const emums = refreshCache ? null : await this.redisCacheService.getJsonObject(cacheKey);
+      if (!Array.isArray(emums) || emums.length === 0) {
+        let values;
+        if (type === RelatedType.sysCfgEnum) {
+          values = await this.queryEnumInfoBySysCfgEnum(code);
+          await this.redisCacheService.setJsonObject(cacheKey, values, expireSecond);
+        } else if (type === RelatedType.dict) {
+          values = await this.queryEnumInfoItemByDict(code);
+          await this.redisCacheService.setJsonObject(cacheKey, values, expireSecond);
+        } else if (type === RelatedType.linkToCustom) {
+          values = await this.queryEnumInfoByLinkToCustom(code);
+          // await this.redisCacheService.setJsonObject(cacheKey, values, expireSecond);
+        }
+        result.push({ type, code, values });
+      } else {
+        result.push({ type, code, values: emums });
+      }
+    }
+    return result;
+  }
+
+  /**
+   * code形如：fatcms~~~sys_perm_role~~~role_code,role_name,xxx_name
+   * code形如：fatcms~~~sys_perm_role~~~id,*
+   * 【数据库】~~~【表名】~~~【label】,【value】，【style】
+   * @param code
+   * @private
+   */
+  private async queryEnumInfoByLinkToCustom(code: string): Promise<IEnumInfo[]> {
+    const cc = await this.curdMixByLinkToCustomService.queryInfoByLinkToCustom(code);
+    if (!cc) {
+      return [];
+    }
+    const keys = [...cc.keys()];
+    return keys.map(value => {
+      const label = cc.get(value);
+      return { label, value };
+    });
+  }
+
+  private async queryEnumInfoBySysCfgEnum(code: string): Promise<IEnumInfo[]> {
+    const { SystemDbName, SystemDbType } = GLOBAL_STATIC_CONFIG.getConfig();
+
+    const res1 = await this.curdProService.executeCrudByCfg(
+      {
+        condition: { config_code: code },
+      },
+      {
+        sqlTable: SystemTables.sys_configs,
+        sqlSimpleName: KeysOfSimpleSQL.SIMPLE_QUERY_ONE,
+        sqlDatabase: SystemDbName,
+        sqlDbType: SystemDbType,
+      }
+    );
+    const obj = res1.getOneObj();
+    if (!obj) {
+      return null;
+    }
+
+    const rows = parseConfigContentToEnumInfo(obj);
+    return rows.map(e => ({ label: e.label, value: e.value, children: e.children })).filter(filterEmpty);
+  }
+
+  private async queryEnumInfoItemByDict(code: string): Promise<IEnumInfo[]> {
+    const { SystemDbName, SystemDbType } = GLOBAL_STATIC_CONFIG.getConfig();
+
+    const res1 = await this.curdProService.executeCrudByCfg(
+      {
+        condition: { dict_code: code },
+      },
+      {
+        sqlTable: SystemTables.sys_data_dict_item,
+        sqlSimpleName: KeysOfSimpleSQL.SIMPLE_QUERY,
+        sqlDatabase: SystemDbName,
+        sqlDbType: SystemDbType,
+      }
+    );
+    const obj = res1.getResRows();
+    if (!obj) {
+      return null;
+    }
+    return obj.map(e => ({ label: e.label, value: e.value })).filter(filterEmpty);
+  }
+}