temp

src/auth/models/user-secret.ts

@@ -8,7 +8,14 @@ import { OauthUser } from '../oauth/oauth.ts';
 export const redis = useContextKey<Redis>('redis');
 
 const UserSecretStatus = ['active', 'inactive', 'expired'] as const;
-
+const randomString = (length: number) => {
+  const chars = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789';
+  let result = '';
+  for (let i = 0; i < length; i++) {
+    result += chars.charAt(Math.floor(Math.random() * chars.length));
+  }
+  return result;
+};
 type Data = {
   [key: string]: any;
   /**
@@ -45,10 +52,12 @@ export class UserSecret extends Model {
     if (!oauth.isSecretKey(token)) {
       return await oauth.verifyToken(token);
     }
-    // const secretToken = await oauth.verifyToken(token);
-    // if (secretToken) {
-    //   return secretToken;
-    // }
+    const secretToken = await oauth.verifyToken(token);
+    if (secretToken) {
+      console.log('verifyToken: verified as normal token');
+      return secretToken;
+    }
+    console.log('verifyToken: try to verify as secret key');
     const userSecret = await UserSecret.findOne({
       where: { token },
     });
@@ -66,7 +75,7 @@ export class UserSecret extends Model {
     if (!oauthUser) {
       return null; // 如果没有找到对应的oauth用户，则返回null
     }
-    // await oauth.saveSecretKey(oauthUser, userSecret.token);
+    await oauth.saveSecretKey(oauthUser, userSecret.token);
     // 存储到oauth中的token store中
     return oauthUser;
   }
@@ -74,10 +83,10 @@ export class UserSecret extends Model {
    * owner 组织用户的 oauthUser
    * @returns
    */
-  async getOauthUser() {
+  async getOauthUser(opts?: { wx?: boolean }) {
     const user = await User.findOne({
       where: { id: this.userId },
-      attributes: ['id', 'username', 'type', 'owner'],
+      attributes: ['id', 'username', 'type', 'owner', 'data'],
     });
     let org: User = null;
     if (!user) {
@@ -117,6 +126,44 @@ export class UserSecret extends Model {
     const expiredTime = new Date(this.expiredTime);
     return now > expiredTime.getTime(); // 如果当前时间大于过期时间，则认为已过期
   }
+  /**
+   *  检查是否过期，如果过期则更新状态为expired
+   *  
+   * @returns 
+   */
+  async checkOnUse() {
+    if (!this.expiredTime) {
+      return {
+        code: 200
+      }
+    }
+    try {
+
+      const now = Date.now();
+      const expiredTime = new Date(this.expiredTime);
+      const isExpired = now > expiredTime.getTime(); // 如果当前时间大于过期时间，则认为已过期
+      if (isExpired) {
+        this.status = 'active';
+        const expireTime = UserSecret.getExpiredTime();
+        this.expiredTime = expireTime;
+        await this.save()
+      }
+      if (this.status !== 'active') {
+        this.status = 'active';
+        await this.save()
+      }
+      return {
+        code: 200
+      };
+    }
+    catch (e) {
+      console.error('checkExpiredAndUpdate error', this.id, this.title);
+      return {
+        code: 500,
+        message: 'checkExpiredAndUpdate error'
+      }
+    }
+  }
   async createNewToken() {
     if (this.token) {
       await oauth.delToken(this.token);
@@ -134,8 +181,21 @@ export class UserSecret extends Model {
     }
     return token;
   }
-  static async createSecret(tokenUser: { id: string; uid?: string }, expireDay = 365) {
-    const expireTime = expireDay * 24 * 60 * 60 * 1000; // 转换为毫秒
+  /**
+   *  根据 unionid 生成redis的key
+   * `wxmp:unionid:token:${unionid}`
+   * @param unionid 
+   * @returns 
+   */
+  static wxRedisKey(unionid: string) {
+    return `wxmp:unionid:token:${unionid}`;
+  }
+  static getExpiredTime(expireDays?: number) {
+    const defaultExpireDays = expireDays || 365;
+    const expireTime = defaultExpireDays * 24 * 60 * 60 * 1000;
+    return new Date(Date.now() + expireTime)
+  }
+  static async createSecret(tokenUser: { id: string; uid?: string, title?: string }, expireDays = 365) {
     const token = await UserSecret.createToken();
     let userId = tokenUser.id;
     let orgId: string = null;
@@ -147,11 +207,13 @@ export class UserSecret extends Model {
       userId,
       orgId,
       token,
-      expiredTime: new Date(Date.now() + expireTime),
+      title: tokenUser.title || randomString(6),
+      expiredTime: UserSecret.getExpiredTime(expireDays),
     });
 
     return userSecret;
   }
+
   async getPermission(opts: { id: string; uid?: string }) {
     const { id, uid } = opts;
     let userId: string = id;