grok

netflix-common/src/main/java/com/cyksj/common/constant/Constant.java

@@ -319,6 +319,11 @@ public interface Constant {
 	 */
 	String LUMA_MIRROR_ONLY_ACCOUNT = "luma123";
 
+	/**
+	 * grok
+	 */
+	String GROK_MIRROR_ONLY_ACCOUNT = "grok123";
+
 	/**
 	 * 免登录渠道
 	 */

netflix-dao/src/main/java/com/cyksj/mapper/GrokUserMapper.java

@@ -0,0 +1,11 @@
+package com.cyksj.mapper;
+
+import com.baomidou.mybatisplus.core.mapper.BaseMapper;
+import com.cyksj.model.entity.GrokUser;
+
+/**
+ * @author chan
+ * @date 2025/3/20 18:32
+ */
+public interface GrokUserMapper extends BaseMapper<GrokUser> {
+}

netflix-dao/src/main/java/com/cyksj/model/entity/GrokUser.java

@@ -0,0 +1,53 @@
+package com.cyksj.model.entity;
+
+import com.baomidou.mybatisplus.annotation.FieldStrategy;
+import com.baomidou.mybatisplus.annotation.TableField;
+import com.ejlchina.searcher.bean.DbField;
+import lombok.Data;
+
+import java.util.Date;
+
+/**
+ * @author chan
+ * @date 2025/3/20 18:04
+ */
+@Data
+public class GrokUser extends BaseEntity {
+
+    private String name;
+
+    @DbField("userToken")
+    @TableField("userToken")
+    private String userToken;
+
+    @TableField("expireTime")
+    private Date expireTime;
+
+    @TableField("isPlus")
+    private Boolean isPlus;
+
+    private String remark;
+
+    /**
+     * 车票id
+     */
+    @TableField("relationId")
+    private Long relationId;
+
+    /**
+     * 限制次数
+     */
+    private Integer limitNum;
+
+    /**
+     * 限制时间
+     */
+    private Long limitTime;
+
+
+    /**
+     * 是否封禁
+     */
+    private Boolean isBlock;
+
+}

netflix-dao/src/main/java/com/cyksj/model/request/GrokOauthReq.java

@@ -0,0 +1,17 @@
+package com.cyksj.model.request;
+
+import lombok.Data;
+
+/**
+ * @author chan
+ * @date 2025/3/20 18:50
+ */
+@Data
+public class GrokOauthReq {
+
+    /**
+     * 用户token
+     */
+    private String userToken;
+
+}

netflix-dao/src/main/java/com/cyksj/model/views/GrokUserView.java

@@ -0,0 +1,41 @@
+package com.cyksj.model.views;
+
+import lombok.Builder;
+import lombok.Data;
+
+import java.util.Date;
+
+/**
+ * @author chan
+ * @date 2025/3/20 18:22
+ */
+@Data
+@Builder
+public class GrokUserView {
+
+    /**
+     * 规格名称
+     */
+    private String skuName;
+
+    /**
+     * 过期时间
+     */
+    private Date expireTime;
+
+    /**
+     * 限制时间
+     */
+    private Long limitTime;
+
+    /**
+     * 限制次数
+     */
+    private Integer limitNum;
+
+    /**
+     * 使用次数
+     */
+    private Long use;
+
+}

netflix-service/src/main/java/com/cyksj/service/grok/GrokService.java

@@ -0,0 +1,22 @@
+package com.cyksj.service.grok;
+
+import com.cyksj.model.entity.GrokUser;
+import com.cyksj.model.views.GrokUserView;
+
+import java.time.LocalDateTime;
+
+/**
+ * @author chan
+ * @date 2025/3/20 18:20
+ */
+public interface GrokService {
+    GrokUserView getUserInfo(long userId, Long relationId);
+
+    GrokUser findByUserTokenAndExpireTimeAfter(String userToken, LocalDateTime now);
+
+    GrokUser checkCarAccount(Long userId, Long relationId);
+
+    Long getConversationCount(String userToken, Long limitTime);
+
+    String getCarLoginUrl(String domain, Long userId, Long relationId);
+}

netflix-service/src/main/java/com/cyksj/service/grok/impl/GrokServiceImpl.java

@@ -0,0 +1,215 @@
+package com.cyksj.service.grok.impl;
+
+import cn.hutool.core.date.DateTime;
+import cn.hutool.core.lang.UUID;
+import cn.hutool.core.util.RandomUtil;
+import cn.hutool.core.util.StrUtil;
+import cn.hutool.http.HttpUtil;
+import cn.hutool.json.JSONObject;
+import cn.hutool.json.JSONUtil;
+import com.baomidou.mybatisplus.core.toolkit.Wrappers;
+import com.cyksj.common.exception.BusinessRuntimeException;
+import com.cyksj.common.util.StringUtil;
+import com.cyksj.mapper.*;
+import com.cyksj.model.entity.*;
+import com.cyksj.model.views.GrokUserView;
+import com.cyksj.service.grok.GrokService;
+import com.cyksj.service.sys.SysConfigService;
+import com.cyksj.service.user.UserBindRelationService;
+import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.dao.DuplicateKeyException;
+import org.springframework.stereotype.Service;
+
+import java.math.BigDecimal;
+import java.math.RoundingMode;
+import java.time.LocalDateTime;
+import java.util.List;
+import java.util.Optional;
+
+/**
+ * @author chan
+ * @date 2025/3/20 18:20
+ */
+@Service
+@Slf4j
+@RequiredArgsConstructor
+public class GrokServiceImpl implements GrokService {
+
+    private final GoodsDonSkuMapper skuMapper;
+
+    private final GrokUserMapper grokUserMapper;
+
+    private final UserBindRelationService userBindRelationService;
+
+    private final GroupsRelationMapper groupsRelationMapper;
+
+    private final UserMapper userMapper;
+
+    private final GroupsMapper groupsMapper;
+
+    private final SysConfigService sysConfigService;
+
+    @Override
+    public GrokUserView getUserInfo(long userId, Long relationId) {
+        GrokUser grokUser = checkCarAccount(userId, relationId);
+        if (grokUser != null) {
+
+            GroupsRelation groupsRelation = getGroupsRelation(userId, relationId);
+            GroupsTrips groupsTrips = getGroupsTrips(groupsRelation);
+            GoodsDonSku goodsDonSku = skuMapper.selectById(groupsTrips.getSkuId());
+
+           GrokUserView view =GrokUserView.builder()
+                    .skuName(goodsDonSku.getSubTitle())
+                    .expireTime(grokUser.getExpireTime())
+                    .limitNum(grokUser.getLimitNum())
+                    .limitTime(grokUser.getLimitTime())
+
+                    //.use(Math.min(grokUser.getLimitNum(), getConversationCount(grokUser.getUserToken(),grokUser.getLimitTime())))
+                    .build();
+            return view;
+        } else {
+            throw BusinessRuntimeException.getInstance("您还未购买该车票");
+        }
+    }
+
+    @Override
+    public GrokUser findByUserTokenAndExpireTimeAfter(String userToken, LocalDateTime now) {
+        return grokUserMapper.selectOne(Wrappers.lambdaQuery(GrokUser.class).eq(GrokUser::getUserToken, userToken).gt(GrokUser::getExpireTime, now));
+    }
+
+    @Override
+    public GrokUser checkCarAccount(Long userId, Long relationId) {
+        GroupsRelation groupsRelation = getGroupsRelation(userId, relationId);
+        GroupsTrips groupsTrips = getGroupsTrips(groupsRelation);
+        GoodsDonSku goodsDonSku = skuMapper.selectById(groupsTrips.getSkuId());
+
+        if (goodsDonSku != null && goodsDonSku.getIsMirror()) {
+            return getGrokUser(userId, relationId, groupsRelation, goodsDonSku);
+        } else {
+            return null;
+        }
+    }
+
+
+    @Override
+    public Long getConversationCount(String userToken, Long limitTime) {
+        JSONObject conversationCount = getConversationCount(userToken);
+        Long count = Optional.ofNullable(conversationCount.getLong("count")).orElse(0l);
+        return new BigDecimal(count).divide(BigDecimal.valueOf(4000),0, RoundingMode.HALF_UP).longValue();
+    }
+
+    @Override
+    public String getCarLoginUrl(String domain, Long userId, Long relationId) {
+
+        GroupsRelation groupsRelation = getGroupsRelation(userId, relationId);
+        GroupsTrips groupsTrips = getGroupsTrips(groupsRelation);
+        GoodsDonSku goodsDonSku = skuMapper.selectById(groupsTrips.getSkuId());
+        if (goodsDonSku != null && goodsDonSku.getIsMirror() && goodsDonSku.getIsCar()) {
+            GrokUser grokUser = getGrokUser(userId, relationId, groupsRelation, goodsDonSku);
+
+            String DOMAIN = domain;
+            if (StrUtil.isEmpty(DOMAIN)) {
+                DOMAIN = "https://grok.bestaistore.com";
+                try {
+                    SysConfig sysConfig = sysConfigService.getOne(Wrappers.lambdaQuery(SysConfig.class).eq(SysConfig::getSysKey, "mirror_grok_domain"));
+                    if (sysConfig != null) {
+                        String sysValue = sysConfig.getSysValue();
+                        if (JSONUtil.isJsonArray(sysValue)) {
+                            List<String> domainList = JSONUtil.parseArray(sysValue).toList(String.class);
+                            DOMAIN = domainList.get(RandomUtil.randomInt(domainList.size()));
+                        }
+                    }
+                } catch (Exception e) {
+                    log.error("随机抽取域名错误. msg:{}", StringUtil.getErrorText(e));
+                }
+            }
+            log.info("domain:{}，用户:{},所在车次:{},座位id:{},在{}获取跳转Grok镜像的登录url", DOMAIN, userId, groupsTrips.getId(), relationId, DateTime.now());
+            return DOMAIN + "/signtoken" + "?userToken=" + grokUser.getUserToken();
+        } else {
+            throw BusinessRuntimeException.getInstance("服务器出了点问题");
+        }
+    }
+
+    /**
+     * 获取车位信息
+     *
+     * @param userId
+     * @param relationId
+     * @return
+     */
+    private GroupsRelation getGroupsRelation(Long userId, Long relationId) {
+        List<Long> userIdList = userBindRelationService.getRelationUserIdList(userId, null);
+        GroupsRelation groupsRelation = groupsRelationMapper.selectOne(Wrappers.lambdaQuery(GroupsRelation.class).in(GroupsRelation::getUserId, userIdList).eq(GroupsRelation::getId, relationId));
+        if (groupsRelation == null) {
+            throw BusinessRuntimeException.getInstance("车票不存在");
+        }
+        return groupsRelation;
+    }
+
+    /**
+     * 获取车队信息
+     *
+     * @param groupsRelation
+     * @return
+     */
+    private GroupsTrips getGroupsTrips(GroupsRelation groupsRelation) {
+        GroupsTrips groupsTrips = groupsMapper.selectById(groupsRelation.getGroupsId());
+        if (groupsTrips == null) {
+            throw BusinessRuntimeException.getInstance("车队异常");
+        }
+        return groupsTrips;
+    }
+
+    private GrokUser getGrokUser(Long userId, Long relationId, GroupsRelation groupsRelation, GoodsDonSku goodsDonSku) {
+        GrokUser grokUser = grokUserMapper.selectOne(Wrappers.lambdaQuery(GrokUser.class).eq(GrokUser::getRelationId, relationId));
+        User user = userMapper.selectById(userId);
+        if (grokUser == null) {
+            grokUser = createGrokUser(groupsRelation, user, goodsDonSku);
+        } else {
+            updateGrokUser(groupsRelation, user, grokUser);
+        }
+        return grokUser;
+    }
+
+
+    /**
+     * 创建chatgptUser
+     */
+    private synchronized GrokUser createGrokUser(GroupsRelation groupsRelation, User user, GoodsDonSku goodsDonSku) {
+        GrokUser grokUser = new GrokUser();
+        if (goodsDonSku.getIsCar()) {
+            grokUser.setLimitNum(goodsDonSku.getGptLimitNum());
+            grokUser.setLimitTime(goodsDonSku.getGptLimitTime());
+        }
+        try {
+            grokUser.setExpireTime(groupsRelation.getExpiryTime());
+            grokUser.setIsPlus(true);
+            grokUser.setName(user.getNickname());
+            grokUser.setRelationId(groupsRelation.getId());
+            grokUser.setUserToken(UUID.randomUUID().toString());
+            grokUserMapper.insert(grokUser);
+
+        } catch (DuplicateKeyException e) {
+            log.error("重复插入Grok镜像用户:{}token,errmsg:{}", user.getNickname(), StringUtil.getErrorText(e));
+        }
+        return grokUser;
+    }
+
+
+    private void updateGrokUser(GroupsRelation groupsRelation, User user, GrokUser grokUser) {
+        grokUser.setName(user.getNickname());
+        grokUser.setExpireTime(groupsRelation.getExpiryTime());
+        grokUserMapper.updateById(grokUser);
+    }
+
+    public JSONObject getConversationCount(String userToken) {
+        try {
+            String res = HttpUtil.get("http://147.79.20.183:9611/grok/getConversationCount?userToken=" + userToken);
+            return new JSONObject(res);
+        }catch (Exception e){
+            return new JSONObject("{\"count\": 0}");
+        }
+    }
+
+}

netflix-service/src/main/java/com/cyksj/service/groups/impl/GroupFunServiceImpl.java

@@ -100,7 +100,7 @@ public class GroupFunServiceImpl implements GroupsFuncService {
 		Boolean isMirror = sku.getIsMirror();
 		Boolean isCar = sku.getIsCar();
 		String account = null;
-		if ((goodsId == 87 && isMirror) || (goodsId == 18 && isCar) || (goodsId == 26 && isMirror) || goodsId == 75 && isCar) {
+		if ((goodsId == 88 && isMirror) || (goodsId == 87 && isMirror) || (goodsId == 18 && isCar) || (goodsId == 26 && isMirror) || (goodsId == 75 && isCar)) {
 			if (goodsId == 18) {
 				account = Constant.GPT_CAR_ONLY_ACCOUNT;
 			}
@@ -113,6 +113,9 @@ public class GroupFunServiceImpl implements GroupsFuncService {
 			if (goodsId == 87) {
 				account = Constant.LUMA_MIRROR_ONLY_ACCOUNT;
 			}
+			if (goodsId == 88) {
+				account = Constant.GROK_MIRROR_ONLY_ACCOUNT;
+			}
 			AccountView accountView = beanSearcher.searchFirst(AccountView.class, MapUtils.builder()
 					.field(AccountView::getGoodsId, goodsId)
 					.field(AccountView::getAccount, account).

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

@@ -21,6 +21,7 @@ import com.cyksj.model.entity.*;
 import com.cyksj.model.views.*;
 import com.cyksj.service.chatgpt.ChatGptAccountService;
 import com.cyksj.service.claude.ClaudeService;
+import com.cyksj.service.grok.GrokService;
 import com.cyksj.service.luma.LumaService;
 import com.cyksj.service.midjourney.MidjourneyAccountService;
 import com.cyksj.service.sys.SysConfigService;
@@ -44,7 +45,7 @@ import java.time.LocalDateTime;
 import java.util.List;
 
 /**
- * /server/镜像服务/claude
+ * /server/镜像服务/grok
  *
  * @author chan
  * @date 2024/3/19 14:41
@@ -95,6 +96,8 @@ public class MirrorController {
 
     private final SysConfigService sysConfigService;
 
+    private final GrokService grokService;
+
 
     /**
      * GPT车票跳转登录
@@ -510,4 +513,63 @@ public class MirrorController {
             response.sendRedirect(DOMAIN + "/codetoken?logintoken=" + userToken);
         }
     }
+
+
+    //------------------------ grok -------------------------------
+
+
+    /**
+     * Grok车票跳转登录
+     *
+     */
+    @RequestMapping("/grok/oauth")
+    public String grokOauth(String userToken) {
+        try {
+            log.info("用户通过userToken:{}访问grok镜像", userToken);
+            GrokUser user = grokService.findByUserTokenAndExpireTimeAfter(userToken, LocalDateTime.now());
+            JSONObject result = new JSONObject();
+
+            if (user == null) {
+                result.putOpt("code",0);
+                result.putOpt("msg","用户不存在或已过期");
+                return result.toString();
+            }
+            result.putOpt("code",1);
+            result.putOpt("msg","登陆成功");
+            result.putOpt("isPro",true);
+            result.putOpt("usertoken",userToken);
+            result.putOpt("userToken",userToken);
+            result.putOpt("expireTime", new DateTime(user.getExpireTime()).toString());
+            return result.toString();
+        } catch (Exception e) {
+            log.error("服务器错误", e);
+            JSONObject result = new JSONObject();
+            result.putOpt("code",0);
+            result.putOpt("msg","服务器错误");
+            return result.toString();
+        }
+    }
+
+    /**
+     * 获取grok车票信息
+     */
+    @GetMapping("/grok/userInfo")
+    public Result<GrokUserView> getGrokUserInfo(Long relationId) {
+        long userId = StpUserUtil.getLoginIdAsLong();
+        GrokUserView user = grokService.getUserInfo(userId, relationId);
+        return GatewayResponse.SUCCESS.newBuilder().toResult(user);
+    }
+
+    /**
+     * grok车队跳转登录
+     *
+     */
+    @GetMapping("/grok/{relationId}/")
+    public Result<String> grokCarLogin(String domain, @PathVariable Long relationId) throws IOException {
+        Long userId = StpUserUtil.getLoginIdAsLong();
+        grokService.checkCarAccount(userId, relationId);
+        String url = grokService.getCarLoginUrl(domain, userId, relationId);
+        return GatewayResponse.SUCCESS.newBuilder().toResult(url);
+    }
+
 }