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