1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
package com.dji.sample.droneairport.controller;
 
import com.dji.sample.common.model.CustomClaim;
import com.dji.sample.common.model.ResponseResult;
import com.dji.sample.droneairport.dao.DeviceExpanSionMapper;
import com.dji.sample.droneairport.model.param.ReturnTaskParam;
import com.dji.sample.droneairport.service.RegistService;
import com.dji.sample.droneairport.utils.SM2.Utils;
import com.dji.sample.droneairport.utils.SM4Util;
import com.dji.sample.manage.service.impl.DeviceRedisServiceImpl;
import org.bouncycastle.crypto.InvalidCipherTextException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
 
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
 
import java.io.IOException;
import java.io.UnsupportedEncodingException;
import java.util.Map;
 
import static com.dji.sample.component.AuthInterceptor.TOKEN_CLAIM;
 
@RequestMapping("/droneAirport")
@RestController
public class RegistController {
    @Autowired
    private RegistService registService;
    @Autowired
    private DeviceExpanSionMapper expanSionDao;
 
    @Autowired
    private DeviceRedisServiceImpl deviceRedisService;
 
    //    @PostMapping("/RegistService")
//    public ResponseResult resgitePort(@RequestBody RegistDto param) {
//        return ResponseResult.success(registService.registPort(param));
//    }
//    @PostMapping("/authorization")
//    public ResponseResult checkDeviceOnline(@RequestBody String body, HttpServletRequest request) throws Exception {
//        String headerValue = request.getHeader("x-lc-secret");
//        return registService.authorization(body, headerValue);
//
//    }
 
    @PostMapping("/addDevice/{workspace_id}")
    public ResponseResult registeDrone(@PathVariable(name = "workspace_id") String workspaceId) {
        return ResponseResult.success(registService.addDrone(workspaceId));
    }
 
    @GetMapping("/getDroneStatus")
    public ResponseResult droneStatus(@RequestParam String deviceid, HttpServletResponse response) throws Exception {
        Map<String, String> info=registService.getDroneState(deviceid);
        response.setHeader("x-lc-secret", info.get("serect"));
        return ResponseResult.success(info.get("text"));
    }
 
    @PostMapping("/AddTask")
    public ResponseResult AddTask(HttpServletRequest request, @RequestBody String body,HttpServletResponse response) throws Exception {
        String headerValue = request.getHeader("x-lc-secret");
        CustomClaim customClaim = (CustomClaim) request.getAttribute(TOKEN_CLAIM);
        String creator = customClaim.getUsername();
        Map<String, String> info = registService.AddTask(body, headerValue, creator);
        response.setHeader("x-lc-secret", info.get("serect"));
        return ResponseResult.success(200, "无人机机场任务添加成功", info.get("traceId"), info.get("jobId"));
    }
 
    @PostMapping("/updateTask")
    public ResponseResult updateTask(HttpServletRequest request, @RequestBody String body,HttpServletResponse response) throws Exception {
        String headerValue = request.getHeader("x-lc-secret");
        CustomClaim customClaim = (CustomClaim) request.getAttribute(TOKEN_CLAIM);
        String creator = customClaim.getUsername();
        Map<String, String>  info = registService.updateTask(body, headerValue, creator);
        response.setHeader("x-lc-secret", info.get("serect"));
        return ResponseResult.success(200, "无人机机场任务更新成功", info.get("traceId"), info.get("jobId"));
    }
 
    @GetMapping("/cancelTask")
    public ResponseResult cancelTask(@RequestParam String taskid) {
        return ResponseResult.success(registService.cancelTask(taskid));
    }
    @PostMapping("/queryTaskStatus")
    public ResponseResult queryTaskStatus(HttpServletRequest request, @RequestParam String taskid) throws Exception {
        return registService.queryTaskStatus(taskid);
    }
    @GetMapping("/getResult")
    public ResponseResult getResult(@RequestParam String taskid,HttpServletResponse response) throws InvalidCipherTextException, IOException {
        Map<String, String>  info = registService.findDbFilesByTaskId(taskid);
        response.setHeader("x-lc-secret", info.get("serect"));
        return ResponseResult.success(200, "无人机成果获取成功!",info.get("data"), info.get("traceId"));
    }
    @PostMapping("/test")
    public ResponseResult test(HttpServletRequest request,@RequestBody String param,HttpServletResponse response) throws Exception {
        String headerValue = request.getHeader("x-lc-secret");
        headerValue=Utils.convertBase64ToHex(headerValue);
        headerValue=Utils.decrypt(headerValue, "00D631FD5615416EAB63D33A9E66E801F95DE840567504210080006081DE877AE3");
        System.out.println(headerValue);
        String value=SM4Util.decrypt(headerValue,param);
        System.out.println(value);
        String sm= SM4Util.generateSM4Key();
        String values=SM4Util.encrypt(sm,value);
        String pub=Utils.encrypt(sm,"04225AACF606D800EA3C2C31FCF8FB161B15F7A8D0460DEB91013D4F228C455E76A2ED8D71BC6525B5DC5CC015C155479D8839950344AEE438A3A6305C90F8269F");
        pub=Utils.hexToBase64(pub);
        response.setHeader("x-lc-secret", pub);
        return ResponseResult.success(values);
    }
}