获取用户镜像token

netflix-service/src/main/java/com/cyksj/service/chatgpt/ChatGptAccountService.java

@@ -90,4 +90,6 @@ public interface ChatGptAccountService {
     String trialGpt(String clientIP, String trialToken, String userAgent);
 
     JSONObject getGptUseInfo(String userToken);
+
+	String getMirrorToken(long userId, Long relationId);
 }

netflix-service/src/main/java/com/cyksj/service/chatgpt/impl/ChatGptAccountServiceImpl.java

@@ -109,6 +109,14 @@ public class ChatGptAccountServiceImpl implements ChatGptAccountService {
 
     private final ChatgptUserRestrictionMapper chatgptUserRestrictionMapper;
 
+    private final MidjourneyUserMapper midjourneyUserMapper;
+
+    private final ClaudeUserMapper claudeUserMapper;
+
+    private final LumaUserMapper lumaUserMapper;
+
+    private final GrokUserMapper grokUserMapper;
+
     @Override
     public String getGptSession(String account, String password) {
         String officialSession = "";
@@ -841,6 +849,63 @@ public class ChatGptAccountServiceImpl implements ChatGptAccountService {
         }
     }
 
+    @Override
+    public String getMirrorToken(long userId, Long relationId) {
+        List<Long> userIdList = userBindRelationService.getRelationUserIdList(userId, null);
+        GroupsRelation relation = groupsRelationMapper.selectOne(Wrappers.lambdaQuery(GroupsRelation.class)
+                .eq(GroupsRelation::getId, relationId)
+                .gt(GroupsRelation::getExpiryTime, DateTime.now())
+                .in(GroupsRelation::getUserId, userIdList)
+                .last("limit 1"));
+        if (relation == null) {
+            throw BusinessRuntimeException.getInstance("车票不存在");
+        }
+        GroupsTrips groupsTrips = groupsMapper.selectById(relation.getGroupsId());
+        GoodsDonSku sku = skuMapper.selectById(groupsTrips.getSkuId());
+        Long goodsId = sku.getGoodsId();
+        if (goodsId == Constant.MJ_GID) {
+            MidjourneyUser user = midjourneyUserMapper.selectOne(Wrappers.lambdaQuery(MidjourneyUser.class)
+                    .eq(MidjourneyUser::getRelationId, relationId)
+                    .last("limit 1"));
+            if (user != null) {
+                return user.getUserToken();
+            }
+        }
+        if (goodsId == Constant.GPT_PLUS_GID) {
+            ChatgptUser chatgptUser = chatgptUserMapper.selectOne(Wrappers.lambdaQuery(ChatgptUser.class)
+                    .eq(ChatgptUser::getRelationId, relationId)
+                    .last("limit 1"));
+            if (chatgptUser != null) {
+                return chatgptUser.getUserToken();
+            }
+        }
+        if (goodsId == Constant.CALUDE_GOODS_ID) {
+            ClaudeUser claudeUser = claudeUserMapper.selectOne(Wrappers.lambdaQuery(ClaudeUser.class)
+                    .eq(ClaudeUser::getRelationId, relationId)
+                    .last("limit 1"));
+            if (claudeUser != null) {
+                return claudeUser.getUserToken();
+            }
+        }
+        if (goodsId == Constant.GROK_GOODS_ID) {
+            GrokUser grokUser = grokUserMapper.selectOne(Wrappers.lambdaQuery(GrokUser.class)
+                    .eq(GrokUser::getRelationId, relationId)
+                    .last("limit 1"));
+            if (grokUser != null) {
+                return grokUser.getUserToken();
+            }
+        }
+        if (goodsId == Constant.LUMA_GOODS_ID) {
+            LumaUser lumaUser = lumaUserMapper.selectOne(Wrappers.lambdaQuery(LumaUser.class)
+                    .eq(LumaUser::getRelationId, relationId)
+                    .last("limit 1"));
+            if (lumaUser != null) {
+                return lumaUser.getUserToken();
+            }
+        }
+        throw BusinessRuntimeException.getInstance("车票不存在");
+    }
+
     private synchronized ChatgptUser createTrialUser(String clientIP) {
         //固定 sku id
         GoodsDonSku goodsDonSku = skuMapper.selectById(535L);

netflix-web/src/main/java/com/cyksj/web/controller/mirror/MirrorController.java

@@ -212,6 +212,16 @@ public class MirrorController {
         return GatewayResponse.SUCCESS.newBuilder().toResult(re.toString());
     }
 
+    /**
+     * 获取用户镜像token
+     */
+    @GetMapping("/get/mirrorToken/{relationId}")
+    public Result getMirrorToken(@PathVariable Long relationId) {
+        long userId = StpUserUtil.getLoginIdAsLong();
+        String token = chatGptAccountService.getMirrorToken(userId, relationId);
+        return GatewayResponse.SUCCESS.newBuilder().toResult(token);
+    }
+
     /**
      * 获取车队历史记录
      */