Edwardsamaxl | f1bf7ad | 2025-06-03 23:52:16 +0800 | [diff] [blame] | 1 | package com.pt.controller; |
| 2 | |
| 3 | import com.pt.service.TrackerService; |
| 4 | import jakarta.servlet.http.HttpServletRequest; |
| 5 | import jakarta.servlet.http.HttpServletResponse; |
| 6 | import org.springframework.beans.factory.annotation.Autowired; |
| 7 | import org.springframework.web.bind.annotation.*; |
| 8 | |
| 9 | import java.io.IOException; |
22301102 | ca0fb2f | 2025-06-09 18:40:42 +0800 | [diff] [blame] | 10 | import java.util.HashMap; |
Edwardsamaxl | f1bf7ad | 2025-06-03 23:52:16 +0800 | [diff] [blame] | 11 | import java.util.Map; |
| 12 | |
| 13 | @RestController |
22301102 | 72bc8d0 | 2025-06-07 10:53:27 +0800 | [diff] [blame] | 14 | @RequestMapping("/api/tracker") |
22301102 | ca0fb2f | 2025-06-09 18:40:42 +0800 | [diff] [blame] | 15 | @CrossOrigin(origins = "*") |
Edwardsamaxl | f1bf7ad | 2025-06-03 23:52:16 +0800 | [diff] [blame] | 16 | public class TrackerController { |
| 17 | |
| 18 | @Autowired |
| 19 | private TrackerService trackerService; |
| 20 | |
22301102 | ca0fb2f | 2025-06-09 18:40:42 +0800 | [diff] [blame] | 21 | // tracker相应bt客户端的announce请求 |
| 22 | @PostMapping("/announce") |
| 23 | public void announceByPost( |
| 24 | @RequestParam("info_hash") String infoHash, |
| 25 | @RequestParam("peer_id") String peerId, |
| 26 | @RequestParam("port") int port, |
| 27 | HttpServletRequest request, |
| 28 | HttpServletResponse response |
| 29 | ) throws IOException { |
Edwardsamaxl | f1bf7ad | 2025-06-03 23:52:16 +0800 | [diff] [blame] | 30 | try { |
| 31 | String ip = request.getRemoteAddr(); |
Edwardsamaxl | f1bf7ad | 2025-06-03 23:52:16 +0800 | [diff] [blame] | 32 | |
22301102 | ca0fb2f | 2025-06-09 18:40:42 +0800 | [diff] [blame] | 33 | // 将参数封装为 Map 传给服务层(也可以直接传对象) |
| 34 | Map<String, String[]> params = new HashMap<>(); |
| 35 | params.put("info_hash", new String[]{infoHash}); |
| 36 | params.put("peer_id", new String[]{peerId}); |
| 37 | params.put("port", new String[]{String.valueOf(port)}); |
Edwardsamaxl | f1bf7ad | 2025-06-03 23:52:16 +0800 | [diff] [blame] | 38 | byte[] bencodedResponse = trackerService.handleAnnounce(params, ip); |
| 39 | |
| 40 | response.setContentType("application/x-bittorrent"); |
| 41 | response.getOutputStream().write(bencodedResponse); |
| 42 | } catch (Exception e) { |
| 43 | response.setStatus(HttpServletResponse.SC_INTERNAL_SERVER_ERROR); |
| 44 | response.setContentType("text/plain"); |
| 45 | response.getWriter().write("Tracker internal error: " + e.getMessage()); |
| 46 | } |
| 47 | } |
| 48 | |
| 49 | } |
| 50 | |