package com.trafficaudit.security.captcha;
|
|
import com.trafficaudit.common.Result;
|
import org.springframework.web.bind.annotation.GetMapping;
|
import org.springframework.web.bind.annotation.PostMapping;
|
import org.springframework.web.bind.annotation.RequestBody;
|
import org.springframework.web.bind.annotation.RequestMapping;
|
import org.springframework.web.bind.annotation.RestController;
|
|
import javax.annotation.Resource;
|
import java.util.Collections;
|
import java.util.List;
|
import java.util.Map;
|
|
@RestController
|
@RequestMapping("/api/auth")
|
public class CaptchaController {
|
|
@Resource
|
private CaptchaService captchaService;
|
|
@GetMapping("/captcha")
|
public Result<Map<String, Object>> captcha() {
|
return Result.ok(captchaService.generate());
|
}
|
|
@PostMapping("/captcha/verify")
|
public Result<Map<String, Object>> verify(@RequestBody Map<String, Object> params) {
|
String captchaId = (String) params.get("captchaId");
|
List<String> clickIds = strList(params.get("captchaClickIds"));
|
CaptchaService.VerifyResult r = captchaService.verify(captchaId, clickIds);
|
switch (r) {
|
case OK:
|
return Result.ok(Collections.singletonMap("ok", true));
|
case WRONG:
|
return Result.error(4001, "点击顺序错误，请重新点击");
|
case RETRY_LIMIT:
|
return Result.error(4002, "错误次数过多，已更换题目");
|
default:
|
return Result.error(4003, "验证码不存在或已过期，请刷新");
|
}
|
}
|
|
@SuppressWarnings("unchecked")
|
private List<String> strList(Object o) {
|
if (o instanceof List) {
|
return (List<String>) o;
|
}
|
return Collections.emptyList();
|
}
|
}
|
