goldenChart/backend/src/main/java/com/goldenchart/controller/PaperTradingController.java

package com.goldenchart.controller;

import com.goldenchart.dto.*;
import com.goldenchart.service.PaperTradingService;
import lombok.RequiredArgsConstructor;
import org.springframework.format.annotation.DateTimeFormat;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;

import java.time.LocalDate;
import java.time.LocalDateTime;
import java.util.List;
import java.util.Map;

@RestController
@RequestMapping("/paper")
@RequiredArgsConstructor
public class PaperTradingController {

    private final PaperTradingService paperTradingService;

    @GetMapping("/summary")
    public ResponseEntity<PaperSummaryDto> summary(@RequestHeader Map<String, String> h) {
        long uid = TradingControllerSupport.requireRegisteredUser(h);
        return ResponseEntity.ok(paperTradingService.getSummary(uid, null));
    }

    @PostMapping("/summary")
    public ResponseEntity<PaperSummaryDto> summaryWithMarks(
            @RequestHeader Map<String, String> h,
            @RequestBody(required = false) Map<String, Double> markPrices) {
        long uid = TradingControllerSupport.requireRegisteredUser(h);
        return ResponseEntity.ok(paperTradingService.getSummary(uid, markPrices));
    }

    @GetMapping("/allocations")
    public ResponseEntity<List<PaperAllocationDto>> allocations(@RequestHeader Map<String, String> h) {
        long uid = TradingControllerSupport.requireRegisteredUser(h);
        return ResponseEntity.ok(paperTradingService.listAllocations(uid, null));
    }

    @PutMapping("/allocations/bulk")
    public ResponseEntity<List<PaperAllocationDto>> bulkAllocations(
            @RequestHeader Map<String, String> h,
            @RequestBody PaperAllocationBulkRequest body) {
        long uid = TradingControllerSupport.requireRegisteredUser(h);
        return ResponseEntity.ok(paperTradingService.bulkAllocations(uid, body));
    }

    @PatchMapping("/allocations/{symbol}")
    public ResponseEntity<PaperAllocationDto> patchAllocation(
            @RequestHeader Map<String, String> h,
            @PathVariable String symbol,
            @RequestBody PaperAllocationPatchRequest body) {
        long uid = TradingControllerSupport.requireRegisteredUser(h);
        return ResponseEntity.ok(paperTradingService.patchAllocation(uid, symbol, body));
    }

    @GetMapping("/trades")
    public ResponseEntity<?> trades(
            @RequestHeader Map<String, String> h,
            @RequestParam(required = false) String symbol,
            @RequestParam(required = false) String side,
            @RequestParam(required = false) String source,
            @RequestParam(required = false) @DateTimeFormat(iso = DateTimeFormat.ISO.DATE_TIME) LocalDateTime from,
            @RequestParam(required = false) @DateTimeFormat(iso = DateTimeFormat.ISO.DATE_TIME) LocalDateTime to,
            @RequestParam(defaultValue = "0") int page,
            @RequestParam(required = false) Integer size) {
        long uid = TradingControllerSupport.requireRegisteredUser(h);
        if (size == null && symbol == null && side == null && source == null && from == null && to == null) {
            return ResponseEntity.ok(paperTradingService.listTradesLegacy(uid));
        }
        int pageSize = size != null ? size : 50;
        return ResponseEntity.ok(paperTradingService.listTrades(
                uid, symbol, side, source, from, to, page, pageSize));
    }

    @DeleteMapping("/trades/batch")
    public ResponseEntity<Map<String, Integer>> deleteTradesBatch(
            @RequestHeader Map<String, String> h,
            @RequestBody PaperTradeDeleteRequest body) {
        long uid = TradingControllerSupport.requireRegisteredUser(h);
        int deleted = paperTradingService.deleteTradesBatch(uid, body != null ? body.getIds() : null);
        return ResponseEntity.ok(Map.of("deleted", deleted));
    }

    @GetMapping("/ledger")
    public ResponseEntity<PaperPageDto<PaperLedgerEntryDto>> ledger(
            @RequestHeader Map<String, String> h,
            @RequestParam(required = false) @DateTimeFormat(iso = DateTimeFormat.ISO.DATE_TIME) LocalDateTime from,
            @RequestParam(required = false) @DateTimeFormat(iso = DateTimeFormat.ISO.DATE_TIME) LocalDateTime to,
            @RequestParam(defaultValue = "0") int page,
            @RequestParam(defaultValue = "50") int size) {
        long uid = TradingControllerSupport.requireRegisteredUser(h);
        return ResponseEntity.ok(paperTradingService.listLedger(uid, from, to, page, size));
    }

    @GetMapping("/snapshots/daily")
    public ResponseEntity<List<PaperDailySnapshotDto>> dailySnapshots(
            @RequestHeader Map<String, String> h,
            @RequestParam(required = false) @DateTimeFormat(iso = DateTimeFormat.ISO.DATE) LocalDate from,
            @RequestParam(required = false) @DateTimeFormat(iso = DateTimeFormat.ISO.DATE) LocalDate to) {
        long uid = TradingControllerSupport.requireRegisteredUser(h);
        return ResponseEntity.ok(paperTradingService.listDailySnapshots(uid, from, to));
    }

    @GetMapping("/orders")
    public ResponseEntity<List<PaperOrderDto>> pendingOrders(@RequestHeader Map<String, String> h) {
        long uid = TradingControllerSupport.requireRegisteredUser(h);
        return ResponseEntity.ok(paperTradingService.listPendingOrders(uid));
    }

    @PostMapping("/orders")
    public ResponseEntity<PaperPlaceOrderResult> placeOrder(
            @RequestHeader Map<String, String> h,
            @RequestBody PaperOrderRequest body) {
        long uid = TradingControllerSupport.requireRegisteredUser(h);
        return ResponseEntity.ok(paperTradingService.placeOrder(uid, body));
    }

    @PostMapping("/orders/{id}/cancel")
    public ResponseEntity<PaperOrderDto> cancelOrder(
            @RequestHeader Map<String, String> h,
            @PathVariable Long id) {
        long uid = TradingControllerSupport.requireRegisteredUser(h);
        return ResponseEntity.ok(paperTradingService.cancelOrder(uid, id));
    }

    @PostMapping("/reset")
    public ResponseEntity<PaperSummaryDto> reset(@RequestHeader Map<String, String> h) {
        long uid = TradingControllerSupport.requireRegisteredUser(h);
        return ResponseEntity.ok(paperTradingService.resetAccount(uid));
    }
}