package jnpf.message.controller;
|
|
import io.swagger.v3.oas.annotations.Operation;
|
import io.swagger.v3.oas.annotations.Parameter;
|
import io.swagger.v3.oas.annotations.tags.Tag;
|
import jakarta.servlet.http.HttpServletResponse;
|
import jnpf.base.ActionResult;
|
import jnpf.base.controller.SuperController;
|
import jnpf.config.ConfigValueUtil;
|
import jnpf.constant.MsgCode;
|
import jnpf.consts.DeviceType;
|
import jnpf.database.util.TenantDataSourceUtil;
|
import jnpf.exception.LoginException;
|
import jnpf.message.entity.ShortLinkEntity;
|
import jnpf.message.service.ShortLinkService;
|
import jnpf.util.*;
|
import lombok.RequiredArgsConstructor;
|
import lombok.extern.slf4j.Slf4j;
|
import org.springframework.web.bind.annotation.GetMapping;
|
import org.springframework.web.bind.annotation.PathVariable;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.RestController;
|
|
import java.io.IOException;
|
|
@Slf4j
|
@RestController
|
@RequiredArgsConstructor
|
@Tag(name = "短链接跳转", description = "message")
|
@RequestMapping("/ShortLink")
|
public class ShortLinkController extends SuperController<ShortLinkService, ShortLinkEntity> {
|
|
private final ShortLinkService shortLinkService;
|
private final ConfigValueUtil configValueUtil;
|
protected final AuthUtil authUtil;
|
|
/**
|
* 消息发送配置弹窗列表
|
*
|
* @return
|
*/
|
@NoDataSourceBind
|
@Operation(summary = "根据短链接获取实际链接地址")
|
@Parameter(name = "shortLink", description = "短链接", required = true)
|
@Parameter(name = "tenant", description = "租户")
|
@GetMapping(value = {"/{shortLink}/{tenant}", "/{shortLink}"})
|
public ActionResult<Object> getShortUrl(@PathVariable("shortLink") String shortLink, @PathVariable(value = "tenant", required = false) String tenant, HttpServletResponse response) throws LoginException, IOException {
|
String tokenStr = "&token=";
|
if (configValueUtil.isMultiTenancy()) {
|
if (StringUtil.isNotEmpty(tenant)) {
|
//切换成租户库
|
TenantDataSourceUtil.switchTenant(tenant);
|
} else {
|
return ActionResult.fail(MsgCode.LOG115.get());
|
}
|
}
|
String link;
|
ShortLinkEntity entity = shortLinkService.getInfoByLink(shortLink);
|
if (entity == null) {
|
return ActionResult.fail(MsgCode.FA039.get());
|
}
|
String frontDomain = configValueUtil.getFrontDomain();
|
String appDomain = configValueUtil.getAppDomain();
|
String realPcLink = entity.getRealPcLink();
|
String realAppLink = entity.getRealAppLink();
|
if (!realPcLink.contains("http")) {
|
realPcLink = frontDomain + realPcLink;
|
}
|
if (!realAppLink.contains("http")) {
|
realAppLink = appDomain + realAppLink;
|
}
|
DeviceType type = UserProvider.getDeviceForAgent();
|
String token = AuthUtil.loginTempUser(entity.getUserId(), tenant);
|
if (StringUtil.isEmpty(token)) {
|
return ActionResult.fail(MsgCode.AD104.get());
|
}
|
|
if (entity.getIsUsed() == 1) {
|
if (entity.getClickNum() < entity.getUnableNum() && entity.getUnableTime().after(DateUtil.getNowDate())) {
|
if (DeviceType.PC.equals(type)) {
|
link = realPcLink + tokenStr + token;
|
entity.setClickNum(entity.getClickNum() + 1);
|
shortLinkService.updateById(entity);
|
} else {
|
link = realAppLink + tokenStr + token;
|
entity.setClickNum(entity.getClickNum() + 1);
|
shortLinkService.updateById(entity);
|
}
|
} else {
|
return ActionResult.fail(MsgCode.FA039.get());
|
}
|
} else {
|
if (entity.getUnableTime().after(DateUtil.getNowDate())) {
|
if (DeviceType.PC.equals(type)) {
|
link = realPcLink + tokenStr + token;
|
entity.setClickNum(entity.getClickNum() + 1);
|
shortLinkService.updateById(entity);
|
} else {
|
link = realAppLink + tokenStr + token;
|
entity.setClickNum(entity.getClickNum() + 1);
|
shortLinkService.updateById(entity);
|
}
|
} else {
|
return ActionResult.fail(MsgCode.FA039.get());
|
}
|
}
|
|
response.sendRedirect(link);
|
return ActionResult.success("");
|
}
|
|
}
|
