source: backend/web3-Ednevnik-Backend/src/main/java/com/nikola/web3ednevnikbackend/controller/PaymentSoopstenieController.java

package com.nikola.web3ednevnikbackend.controller;

import com.nikola.web3ednevnikbackend.config.security.CustomUserDetails;
import com.nikola.web3ednevnikbackend.dto.payments.*;
import com.nikola.web3ednevnikbackend.service.payment.PaymentSoopstenieService;
import jakarta.validation.Valid;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.security.core.annotation.AuthenticationPrincipal;
import org.springframework.web.bind.annotation.*;
import com.nikola.web3ednevnikbackend.model.enums.soopstenieZaPlakjanjeStatus;

import java.util.List;
import java.util.UUID;

@RestController
@RequestMapping("/api/paymentsSoopstenija")
public class PaymentSoopstenieController {

    private final PaymentSoopstenieService paymentSoopstenieService;

    public PaymentSoopstenieController(PaymentSoopstenieService paymentSoopstenieService) {
        this.paymentSoopstenieService = paymentSoopstenieService;
    }


    @GetMapping("/roditel")
    @PreAuthorize("hasAnyRole('KLASEN_RAKOVODITEL','DIREKTOR_ADMIN', 'BLAGAJNIK', 'RODITEL')")
    public ResponseEntity<List<SoopstenieZaPlakjanjeResponseDTO_roditel>> getAllActiveSoopstenijaZaPlakjanje_RoditelView(@AuthenticationPrincipal CustomUserDetails userDetails) {


        UUID roditelId = userDetails.getId();

        List<SoopstenieZaPlakjanjeResponseDTO_roditel> soopstenijaZaPlakjanje = this.paymentSoopstenieService.getAllActiveSoopstenijaZaPlakjanje_RoditelView(roditelId);
        return new ResponseEntity<List<SoopstenieZaPlakjanjeResponseDTO_roditel>>(soopstenijaZaPlakjanje, HttpStatus.OK);
    }

    @GetMapping("/blagajnik")
    @PreAuthorize("hasAnyRole('KLASEN_RAKOVODITEL','DIREKTOR_ADMIN', 'BLAGAJNIK', 'RODITEL')")
    public ResponseEntity<List<SoopstenieZaPlakjanjeResponseDTO_blagajnik>> getAllActiveSoopstenijaZaPlakjanje_BlagajnikView(@AuthenticationPrincipal CustomUserDetails userDetails) {


        UUID blagajnikId = userDetails.getId();

        List<SoopstenieZaPlakjanjeResponseDTO_blagajnik> soopstenijaZaPlakjanje = this.paymentSoopstenieService.getAllActiveSoopstenijaZaPlakjanje_BlagajnikView(blagajnikId);
        return new ResponseEntity<List<SoopstenieZaPlakjanjeResponseDTO_blagajnik>>(soopstenijaZaPlakjanje, HttpStatus.OK);
    }

    @GetMapping("/direktorAdmin")
    @PreAuthorize("hasAnyRole('DIREKTOR_ADMIN','MON_ADMIN')")
    //ke se povikuva kako ?status=PENDING, ?status=REJECTED, ?status=APPROVED
    public ResponseEntity<List<SoopstenieZaPlakjanjeResponseDTO_direktorAdmin>> getAllSoopstenijaZaPlakjanje_direktorAdmin(@RequestParam(required = false, defaultValue = "PENDING") String status) {
        List<SoopstenieZaPlakjanjeResponseDTO_direktorAdmin> soopstenieZaPlakjanjeResponseDTODirektorAdmin = this.paymentSoopstenieService.getAllSoopstenijaZaPlakjanje_direktorAdmin(status);
        return new ResponseEntity<List<SoopstenieZaPlakjanjeResponseDTO_direktorAdmin>>(soopstenieZaPlakjanjeResponseDTODirektorAdmin, HttpStatus.OK);
    }

    @GetMapping("/approved")
    public ResponseEntity<List<SoopstenieZaPlakjanjeResponseDTO_roditel>> getApproved() {

        List<SoopstenieZaPlakjanjeResponseDTO_roditel> approvedSoopstenija = this.paymentSoopstenieService.getAllApprovedSoopstenija();

        return new ResponseEntity<>(approvedSoopstenija, HttpStatus.OK);
    }


    @PostMapping
    public ResponseEntity<Void> createSoopstenieZaPlakjanje(@Valid @RequestBody SoopstenieZaPlakjanjeRequestDTO soopstenieZaPlakjanjeRequestDTO) {
        this.paymentSoopstenieService.createSoopstenieZaPlakjanje(soopstenieZaPlakjanjeRequestDTO);

        return new ResponseEntity<>(HttpStatus.CREATED);
    }


    @PutMapping("/{id}/approve")
    @PreAuthorize("hasAnyRole('DIREKTOR_ADMIN','MON_ADMIN')")
    public ResponseEntity<Void> approve(@PathVariable("id") UUID soopstenieId,
                                        @Valid @RequestBody ApproveAndRejectSoopstenieZaPlakjanjeRequestDTO approveAndRejectSoopstenieZaPlakjanjeRequestDTO) {

        this.paymentSoopstenieService.approveSoopstenieByDirektorAdmin(soopstenieId, approveAndRejectSoopstenieZaPlakjanjeRequestDTO);

        return new ResponseEntity<>(HttpStatus.NO_CONTENT);
    }


    @PutMapping("/{id}/reject")
    @PreAuthorize("hasAnyRole('DIREKTOR_ADMIN','MON_ADMIN')")
    public ResponseEntity<Void> reject(@PathVariable("id") UUID soopstenieId,
                                       @Valid @RequestBody ApproveAndRejectSoopstenieZaPlakjanjeRequestDTO approveAndRejectSoopstenieZaPlakjanjeRequestDTO) {

        this.paymentSoopstenieService.rejectSoopstenieByDirektorAdmin(soopstenieId, approveAndRejectSoopstenieZaPlakjanjeRequestDTO);

        return ResponseEntity.ok().build();
    }
}