wuchimedes | 223bfab | 2025-04-04 17:16:05 +0800 | [diff] [blame] | 1 | package com.example.g8backend.controller; |
| 2 | |
wuchimedes | db9fe68 | 2025-04-22 19:24:11 +0800 | [diff] [blame] | 3 | import com.example.g8backend.dto.AnnounceRequestDTO; |
| 4 | import com.example.g8backend.dto.AnnounceResponseDTO; |
| 5 | import jakarta.servlet.http.HttpServletRequest; |
wuchimedes | 223bfab | 2025-04-04 17:16:05 +0800 | [diff] [blame] | 6 | import org.springframework.beans.factory.annotation.Autowired; |
| 7 | import org.springframework.http.ResponseEntity; |
wuchimedes | a0649c6 | 2025-04-05 15:53:39 +0800 | [diff] [blame] | 8 | import org.springframework.web.bind.annotation.*; |
wuchimedes | 223bfab | 2025-04-04 17:16:05 +0800 | [diff] [blame] | 9 | import com.example.g8backend.service.ITrackerService; |
| 10 | |
| 11 | @RestController |
wuchimedes | a0649c6 | 2025-04-05 15:53:39 +0800 | [diff] [blame] | 12 | @RequestMapping("/tracker") |
wuchimedes | 223bfab | 2025-04-04 17:16:05 +0800 | [diff] [blame] | 13 | public class TrackerController { |
| 14 | |
wuchimedes | a0649c6 | 2025-04-05 15:53:39 +0800 | [diff] [blame] | 15 | @Autowired |
| 16 | private ITrackerService trackerService; |
| 17 | |
wuchimedes | b06716b | 2025-04-08 19:48:24 +0800 | [diff] [blame] | 18 | @GetMapping("/announce/{passkey}") |
wuchimedes | db9fe68 | 2025-04-22 19:24:11 +0800 | [diff] [blame] | 19 | public ResponseEntity<AnnounceResponseDTO> getAnnouncements( |
| 20 | HttpServletRequest request, |
| 21 | @RequestParam("info_hash") String infoHash, |
| 22 | @RequestParam("peer_id") String peerId, |
| 23 | @RequestParam("port") int port, |
| 24 | @RequestParam("uploaded") double uploaded, |
| 25 | @RequestParam("downloaded") double downloaded, |
| 26 | @RequestParam(value = "event", required = false) String event, |
| 27 | @RequestParam(value = "left", required = false) Double left, |
| 28 | @RequestParam(value = "compact", required = false) Integer compact, |
wuchimedes | a0649c6 | 2025-04-05 15:53:39 +0800 | [diff] [blame] | 29 | @PathVariable String passkey) { |
| 30 | |
wuchimedes | db9fe68 | 2025-04-22 19:24:11 +0800 | [diff] [blame] | 31 | AnnounceRequestDTO requestDTO = new AnnounceRequestDTO(); |
| 32 | requestDTO.setPasskey(passkey); |
| 33 | requestDTO.setInfoHash(infoHash); |
| 34 | requestDTO.setPeerId(peerId); |
| 35 | requestDTO.setPort(port); |
| 36 | requestDTO.setUploaded(uploaded); |
| 37 | requestDTO.setDownloaded(downloaded); |
| 38 | requestDTO.setEvent(event); |
| 39 | requestDTO.setLeft(left); |
| 40 | requestDTO.setCompact(compact); |
| 41 | |
| 42 | String ipAddress = request.getHeader("X-Forwarded-For"); |
| 43 | if (ipAddress == null || ipAddress.isEmpty() || "unknown".equalsIgnoreCase(ipAddress)) { |
| 44 | ipAddress = request.getHeader("Proxy-Client-IP"); |
| 45 | } |
| 46 | if (ipAddress == null || ipAddress.isEmpty() || "unknown".equalsIgnoreCase(ipAddress)) { |
| 47 | ipAddress = request.getHeader("WL-Proxy-Client-IP"); |
| 48 | } |
| 49 | if (ipAddress == null || ipAddress.isEmpty() || "unknown".equalsIgnoreCase(ipAddress)) { |
| 50 | ipAddress = request.getRemoteAddr(); |
| 51 | } |
| 52 | requestDTO.setIp(ipAddress.split(",")[0]); |
| 53 | return ResponseEntity.ok(trackerService.handleAnnounce(requestDTO)); |
wuchimedes | a0649c6 | 2025-04-05 15:53:39 +0800 | [diff] [blame] | 54 | } |
wuchimedes | 223bfab | 2025-04-04 17:16:05 +0800 | [diff] [blame] | 55 | } |