no-copy/src/main/java/ru/soune/nocopy/controller/AuthController.java

package ru.soune.nocopy.controller;

import jakarta.validation.Valid;
import lombok.RequiredArgsConstructor;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;
import ru.soune.nocopy.dto.AuthResponse;
import ru.soune.nocopy.dto.LoginRequest;
import ru.soune.nocopy.dto.LoginResponse;
import ru.soune.nocopy.dto.RegisterRequest;
import ru.soune.nocopy.entity.AuthToken;
import ru.soune.nocopy.exception.TokenNotFoundException;
import ru.soune.nocopy.repository.AuthTokenRepository;
import ru.soune.nocopy.service.AuthService;

import java.util.Map;
import java.util.Optional;

@RestController
@RequestMapping("v1/api/auth")
@RequiredArgsConstructor
public class AuthController {

    private final AuthService authService;

    private final AuthTokenRepository authTokenRepository;

    @PostMapping("/register")
    public ResponseEntity<AuthResponse> register(@Valid @RequestBody RegisterRequest registerRequest) {
        AuthToken authToken = authService.register(registerRequest);

        return ResponseEntity.ok(new AuthResponse(true, "success.user.register", authToken.getToken(),
                authToken.getExpiresAt()));
    }

    @PostMapping("/login")
    public ResponseEntity<LoginResponse> login(@Valid @RequestBody LoginRequest request) {
        AuthToken login = authService.login(request);

        return ResponseEntity.ok(new LoginResponse(true, login.getUser().getEmail(),
                login.getToken(),login.getExpiresAt().toString()));
    }

    @PostMapping("/logout")
    public ResponseEntity<?> logout(@RequestHeader("Authorization") String tokenHeader) {
        String token = tokenHeader.replace("Bearer ", "");

        Optional<AuthToken> opToken = authTokenRepository.findByToken(token);

        if (opToken.isEmpty()) {
            throw new TokenNotFoundException("Token not found");
        }

        authService.logout(token);

        return ResponseEntity.ok(Map.of("success", true));
    }
}