import { ALL, Config, Middleware } from '@midwayjs/core';
import { NextFunction, Context } from '@midwayjs/koa';
import { IMiddleware, Init, Inject } from '@midwayjs/core';
import * as jwt from 'jsonwebtoken';
import * as _ from 'lodash';
import { CoolCommException, CoolUrlTagData, TagTypes } from '@cool-midway/core';
import { Utils } from '../../../comm/utils';
import { UserInfoService } from '../service/info';

/**
 * 用户
 */
@Middleware()
export class UserMiddleware implements IMiddleware<Context, NextFunction> {
  @Config(ALL)
  coolConfig;

  @Inject()
  coolUrlTagData: CoolUrlTagData;

  @Config('module.user.jwt')
  jwtConfig;

  ignoreUrls: string[] = [];

  @Config('koa.globalPrefix')
  prefix;

  @Inject()
  utils: Utils;

  @Init()
  async init() {
    this.ignoreUrls = this.coolUrlTagData.byKey(TagTypes.IGNORE_TOKEN, 'app');
  }

  resolve() {
    return async (ctx: Context, next: NextFunction) => {
      let { url } = ctx;
      url = url.replace(this.prefix, '').split('?')[0];
      if (_.startsWith(url, '/app/')) {
        let token = ctx.get('Authorization') || '';
        token = token.replace(/^Bearer\s+/i, '').trim();
        try {
          ctx.user = jwt.verify(token, this.jwtConfig.secret);
        } catch (error) {
          ctx.user = undefined;
          if (token) {
            console.error('[user] token无效:', (error as Error).message);
          }
        }
        if (ctx.user?.isRefresh) {
          ctx.user = undefined;
        } else if (ctx.user?.id) {
          try {
            // 令牌里的 id 是 unionid（手机号登录时就是手机号），整型 userId 列要用用户主键
            const userInfoService = await ctx.requestContext.getAsync(
              UserInfoService
            );
            const info = await userInfoService.person(ctx.user.id);
            if (info?.id) {
              ctx.user.userId = info.id;
            }
          } catch (error) {}
        }
        // 使用matchUrl方法来检查URL是否应该被忽略
        const isIgnored = this.ignoreUrls.some(pattern =>
          this.utils.matchUrl(pattern, url)
        );
        if (isIgnored) {
          await next();
          return;
        } else {
          if (!ctx.user) {
            ctx.status = 401;
            throw new CoolCommException('登录失效~');
          }
        }
      }
      await next();
    };
  }
}