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

package ru.soune.nocopy.controller;

import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;
import ru.soune.nocopy.dto.search.GlobalSearchStartRequest;
import ru.soune.nocopy.dto.search.GlobalSearchStartResponse;
import ru.soune.nocopy.dto.search.GlobalSearchStatusResponse;
import ru.soune.nocopy.entity.file.FileEntity;
import ru.soune.nocopy.entity.search.GlobalSearchTask;
import ru.soune.nocopy.entity.search.SearchStatus;
import ru.soune.nocopy.exception.NotFoundAuthToken;
import ru.soune.nocopy.repository.GlobalSearchTaskRepository;
import ru.soune.nocopy.service.register.AuthService;
import ru.soune.nocopy.service.search.GlobalSearchService;

import java.util.*;

@Slf4j
@RestController
@RequestMapping("/api/v1/global-search")
@RequiredArgsConstructor
public class GlobalSearchController {

    private final AuthService authService;

    private final GlobalSearchService globalSearchService;

    private final GlobalSearchTaskRepository globalSearchTaskRepository;

    private final GlobalSearchTaskRepository searchTaskRepository;

    @PostMapping("/start")
    public ResponseEntity<?> startSearch(
            @RequestBody GlobalSearchStartRequest request,
            @RequestHeader(value = "Authorization", required = false) String tokenHeader) {
        if (tokenHeader == null || tokenHeader.isBlank()) {
            Map<String, Object> errorData = new HashMap<>();
            errorData.put("token", tokenHeader);

            return ResponseEntity.ok().body(Map.of("error", errorData));
        }

        GlobalSearchStartResponse response = new GlobalSearchStartResponse();

        try {
            Long userId = authService.useUserAuthToken(tokenHeader);

            List<FileEntity> filesToProcess = globalSearchService.getFilesToProcess(request, userId);

            if (filesToProcess.isEmpty()) return ResponseEntity.ok().body("Files for search not found");

            String taskId = globalSearchService.startSearch(request, userId, filesToProcess);

            Optional<GlobalSearchTask> taskOptional = globalSearchTaskRepository.findById(taskId);

            if (taskOptional.isEmpty()) return ResponseEntity.ok().body("Task not found");

            GlobalSearchTask task = taskOptional.orElseThrow();

            response.setTaskId(taskId);
            response.setStatus(SearchStatus.ACCEPTED.name());
            response.setTotalFiles(task.getTotalFiles());
        } catch (NotFoundAuthToken e) {
            Map<String, Object> errorData = new HashMap<>();
            errorData.put("token", tokenHeader);

            return ResponseEntity.ok().body(Map.of("error", errorData));
        }

        return ResponseEntity.ok(response);
    }

    @GetMapping("/actual-task")
    public ResponseEntity<?> getStatus(@RequestHeader(value = "Authorization", required = false) String tokenHeader) {

        if (tokenHeader == null || tokenHeader.isBlank()) {
            Map<String, Object> errorData = new HashMap<>();
            errorData.put("token", tokenHeader);

            return ResponseEntity.ok().body(Map.of("error", errorData));
        }

        Optional<GlobalSearchTask> taskOptional;

        try {
            Long userId = authService.useUserAuthToken(tokenHeader);

            taskOptional = searchTaskRepository.findByUserIdAndStatus(userId,
                    "PROCESSING");

            if (taskOptional.isEmpty()) return ResponseEntity.ok().body("Task not found");
        } catch (NotFoundAuthToken e) {
            Map<String, Object> errorData = new HashMap<>();
            errorData.put("token", tokenHeader);

            return ResponseEntity.ok().body(Map.of("error", errorData));
        }

        return ResponseEntity.ok(taskOptional.get());
    }

    @GetMapping("/status/{taskId}")
    public ResponseEntity<?> getStatus(@PathVariable String taskId,
            @RequestHeader(value = "Authorization", required = false) String tokenHeader) {

        if (tokenHeader == null || tokenHeader.isBlank()) {
            Map<String, Object> errorData = new HashMap<>();
            errorData.put("token", tokenHeader);

            return ResponseEntity.ok().body(Map.of("error", errorData));
        }
        GlobalSearchStatusResponse response;

        try {
            Long userId = authService.useUserAuthToken(tokenHeader);

            Optional<GlobalSearchTask> taskOptional = searchTaskRepository.findByTaskIdAndUserId(taskId, userId);

            if (taskOptional.isEmpty()) return ResponseEntity.ok().body("Task not found");

            GlobalSearchTask task = taskOptional.orElseThrow();

            int progress = task.getTotalFiles() > 0 ? task.getProcessedFiles() * 100 / task.getTotalFiles() : 0;

            response = new GlobalSearchStatusResponse();
            response.setTaskId(taskId);
            response.setStatus(task.getStatus());
            response.setProgress(progress);
        } catch (NotFoundAuthToken e) {
            Map<String, Object> errorData = new HashMap<>();
            errorData.put("token", tokenHeader);

            return ResponseEntity.ok().body(Map.of("error", errorData));
        }

        return ResponseEntity.ok(response);
    }
}