102 lines
3.1 KiB
Java
102 lines
3.1 KiB
Java
|
package net.geedge.confagent.controller;
|
|||
|
|
|
|||
|
|
import java.io.File;
|
|||
|
|
import java.util.HashMap;
|
|||
|
|
import java.util.Map;
|
|||
|
|
import java.util.Properties;
|
|||
|
|
|
|||
|
|
import org.springframework.beans.factory.annotation.Value;
|
|||
|
|
import org.springframework.web.bind.annotation.GetMapping;
|
|||
|
|
import org.springframework.web.bind.annotation.RequestMapping;
|
|||
|
|
import org.springframework.web.bind.annotation.RestController;
|
|||
|
|
|
|||
|
|
import cn.hutool.log.Log;
|
|||
|
|
import net.geedge.confagent.annotation.UnCheckToken;
|
|||
|
|
import net.geedge.confagent.entity.AuthEntity;
|
|||
|
|
import net.geedge.confagent.util.R;
|
|||
|
|
import net.geedge.confagent.util.RCode;
|
|||
|
|
import net.geedge.confagent.util.Tool;
|
|||
|
|
|
|||
|
|
@RestController
|
|||
|
|
@RequestMapping(value={"/auth"})
|
|||
|
|
public class AuthController extends BaseController{
|
|||
|
|
private final static Log log = Log.get();
|
|||
|
|
|
|||
|
|
@Value("${confagent.authFile:auth.yml}")
|
|||
|
|
private String authFile;
|
|||
|
|
|
|||
|
|
private static String rootPath = Tool.WebPathUtil.getRootPath();
|
|||
|
|
/**
|
|||
|
|
* @Description 每次请求auth接口重新生成 token并记录到文件
|
|||
|
|
* @Author rui
|
|||
|
|
* @Date 2021/3/23
|
|||
|
|
*/
|
|||
|
|
@RequestMapping
|
|||
|
|
@UnCheckToken
|
|||
|
|
public R auth(AuthEntity auth){
|
|||
|
|
|
|||
|
|
if(Tool.StrUtil.isBlank(auth.getName())||Tool.StrUtil.isBlank(auth.getPin())){
|
|||
|
|
return R.error(RCode.AUTH_NAME_PIN_ISNULL);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
File af = Tool.FileUtil.file(rootPath, authFile);
|
|||
|
|
log.debug("auth file path : {}" ,af.getAbsolutePath());
|
|||
|
|
Properties properties = Tool.YamlUtil.yamlToProperties(af.getAbsolutePath());
|
|||
|
|
|
|||
|
|
String name = properties.getProperty("auth.name","nezha");
|
|||
|
|
String pin = properties.getProperty("auth.pin","nezha");
|
|||
|
|
|
|||
|
|
if(Tool.StrUtil.equals(name,auth.getName())&& Tool.StrUtil.equals(pin,auth.getPin())){
|
|||
|
|
String token = Tool.IdUtil.fastShortUUID();
|
|||
|
|
log.debug("write token {} to {}",token,af.getAbsolutePath());
|
|||
|
|
writeToken(token);
|
|||
|
|
Map<String,String> map = new HashMap<>();
|
|||
|
|
map.put("token", token);
|
|||
|
|
return R.ok(map);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return R.error(RCode.AUTH_NAME_PIN_INVALID);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @Description 验证当前token是否和文件记录一致，一致后重新生成则更新文件并返回最新token
|
|||
|
|
* @Author rui
|
|||
|
|
* @Date 2021/3/24
|
|||
|
|
*/
|
|||
|
|
@GetMapping("/refresh")
|
|||
|
|
public R refresh(){
|
|||
|
|
String t = Tool.IdUtil.fastShortUUID();
|
|||
|
|
writeToken(t);
|
|||
|
|
Map<String,String> map = new HashMap<>();
|
|||
|
|
map.put("token", t);
|
|||
|
|
return R.ok(map);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/**
|
|||
|
|
* @Description 注销当前 token
|
|||
|
|
* @Author rui
|
|||
|
|
* @Date 2021/3/24
|
|||
|
|
*/
|
|||
|
|
@GetMapping("logout")
|
|||
|
|
public R logout(){
|
|||
|
|
File tf = Tool.FileUtil.file(rootPath, tokenFile);
|
|||
|
|
log.debug("token file path : {}" ,tf.getAbsolutePath());
|
|||
|
|
|
|||
|
|
if(Tool.FileUtil.exist(tf)){
|
|||
|
|
Tool.FileUtil.del(tf);
|
|||
|
|
}
|
|||
|
|
if(Tool.FileUtil.exist(promtailConfPath)){
|
|||
|
|
Tool.YamlUtil.writeAsMap(null,promtailConfPath);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return R.ok();
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
private void writeToken(String token){
|
|||
|
|
File tf = Tool.FileUtil.file(rootPath, tokenFile);
|
|||
|
|
log.debug("token file path : {}" ,tf.getAbsolutePath());
|
|||
|
|
Tool.FileUtil.writeUtf8String(token, tf);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|