|
|
@@ -2,14 +2,11 @@ package com.usoftchina.saas.auth.controller;
|
|
|
|
|
|
import com.github.pagehelper.PageInfo;
|
|
|
import com.usoftchina.saas.account.api.AccountApi;
|
|
|
+import com.usoftchina.saas.account.api.CompanyApi;
|
|
|
import com.usoftchina.saas.account.constant.AccountType;
|
|
|
-import com.usoftchina.saas.account.dto.AccountCopyDTO;
|
|
|
-import com.usoftchina.saas.account.dto.AccountDTO;
|
|
|
-import com.usoftchina.saas.account.dto.AccountUpdateDTO;
|
|
|
-import com.usoftchina.saas.account.dto.CompanyBaseDTO;
|
|
|
+import com.usoftchina.saas.account.dto.*;
|
|
|
import com.usoftchina.saas.auth.common.cookie.CookieHelper;
|
|
|
import com.usoftchina.saas.auth.common.cookie.CookieInfo;
|
|
|
-import com.usoftchina.saas.auth.common.cookie.CookieUtils;
|
|
|
import com.usoftchina.saas.auth.common.jwt.JwtHelper;
|
|
|
import com.usoftchina.saas.auth.common.jwt.JwtInfo;
|
|
|
import com.usoftchina.saas.auth.common.jwt.JwtToken;
|
|
|
@@ -34,8 +31,6 @@ import org.slf4j.LoggerFactory;
|
|
|
import org.springframework.beans.factory.annotation.Autowired;
|
|
|
import org.springframework.beans.factory.annotation.Value;
|
|
|
import org.springframework.boot.context.properties.EnableConfigurationProperties;
|
|
|
-import org.springframework.http.server.reactive.ServerHttpRequest;
|
|
|
-import org.springframework.http.server.reactive.ServerHttpResponse;
|
|
|
import org.springframework.web.bind.annotation.*;
|
|
|
|
|
|
import javax.servlet.http.HttpServletRequest;
|
|
|
@@ -57,6 +52,9 @@ public class AuthController {
|
|
|
@Autowired
|
|
|
private AccountApi accountApi;
|
|
|
|
|
|
+ @Autowired
|
|
|
+ private CompanyApi companyApi;
|
|
|
+
|
|
|
@Value("${auth.public-key}")
|
|
|
private String publicKeyPath;
|
|
|
|
|
|
@@ -66,7 +64,7 @@ public class AuthController {
|
|
|
@Value("${auth.header:Authorization}")
|
|
|
private String authHeader;
|
|
|
|
|
|
- @Value("${auth.expire:18000}")
|
|
|
+ @Value("${auth.expire:1800000}")
|
|
|
private int expire;
|
|
|
|
|
|
@Value("${auth.max-errors:5}")
|
|
|
@@ -203,7 +201,12 @@ public class AuthController {
|
|
|
JwtInfo jwtInfo = new JwtInfo(appId, companyId, accountDTO.getId(), accountDTO.getUsername(), accountDTO.getRealname());
|
|
|
JwtToken jwtToken = JwtHelper.generateToken(jwtInfo, privateKeyPath, expire);
|
|
|
TokenDTO tokenDTO = BeanMapper.map(jwtToken, TokenDTO.class);
|
|
|
- return Result.success(new AuthDTO(tokenDTO, accountDTO));
|
|
|
+ AuthDTO authDTO = new AuthDTO(tokenDTO, accountDTO);
|
|
|
+ CompanyDTO companyDTO = companyApi.getCompanyByUu(info.getSpaceUU()).getData();
|
|
|
+ if (null != companyDTO){
|
|
|
+ authDTO.setCompanyId(companyDTO.getId());
|
|
|
+ }
|
|
|
+ return Result.success(authDTO);
|
|
|
}
|
|
|
return Result.error(ExceptionCode.COOKIE_ILLEGAL_ARGUMENT);
|
|
|
}
|