no-copy/src/main/java/ru/soune/nocopy/handler/ComplaintEntityHandler.java

package ru.soune.nocopy.handler;

import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.data.domain.PageRequest;
import org.springframework.data.domain.Pageable;
import org.springframework.data.domain.Sort;
import org.springframework.stereotype.Component;
import ru.soune.nocopy.dto.BaseRequest;
import ru.soune.nocopy.dto.BaseResponse;
import ru.soune.nocopy.dto.MessageCode;
import ru.soune.nocopy.dto.complaint.ComplaintRequest;
import ru.soune.nocopy.dto.complaint.ComplaintResponse;
import ru.soune.nocopy.entity.complaint.ComplaintStatus;
import ru.soune.nocopy.entity.user.User;
import ru.soune.nocopy.repository.UserRepository;
import ru.soune.nocopy.service.complaint.ComplaintEntityService;
import ru.soune.nocopy.service.register.AuthService;
import ru.soune.nocopy.service.user.UserService;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

@Slf4j
@Component
@RequiredArgsConstructor
public class ComplaintEntityHandler implements RequestHandler {
    private final ComplaintEntityService complaintService;

    private final ObjectMapper objectMapper;

    private final UserRepository userRepository;

    private final UserService userService;

    private final AuthService authService;

    @Override
    public BaseResponse handle(BaseRequest request) throws Exception {
        Integer msgId = request.getMsgId();
        ComplaintRequest complaintRequest = objectMapper.convertValue(request.getMessageBody(), ComplaintRequest.class);
        String action = complaintRequest.getAction();
        String token = complaintRequest.getToken();
        Long userId = authService.useUserAuthToken(token);
        User user = userRepository.findById(userId).orElse(null);

        if (user == null) {
            return new BaseResponse(
                    request.getMsgId(),
                    MessageCode.USER_NOT_FOUND.getCode(),
                    MessageCode.USER_NOT_FOUND.getDescription(),
                    Map.of("token: ", token));
        }

        List<String> emails = user.getCompany() == null ? List.of(user.getEmail()):
                userService.getUsersCompanyEmails(user.getCompany().getId());

        try {
            switch (action.toLowerCase()) {
                case "create":
                    return handleCreate(msgId, complaintRequest);
                case "get":
                    return handleGet(msgId, complaintRequest);
                case "get_all":
                    return handleGetAll(msgId, complaintRequest, emails);
                case "get_by_violation":
                    return handleGetByViolation(msgId, complaintRequest);
                case "update_status":
                    return handleUpdateStatus(msgId, complaintRequest);
                case "update":
                    return handleUpdate(msgId, complaintRequest);
                case "delete":
                    return handleDelete(msgId, complaintRequest);
                default:
                    return errorResponse(msgId, MessageCode.MSG_ID_NOT_FOUND.getCode(), "Unknown action: " + action);
            }
        } catch (Exception e) {
            log.error("Error handling complaint request", e);
            return errorResponse(msgId, MessageCode.INVALID_JSON_BODY.getCode(), e.getMessage());
        }
    }

    private BaseResponse handleCreate(Integer msgId, ComplaintRequest req) {
        if (req.getViolationId() == null) return errorResponse(msgId, MessageCode.NOT_FOUND.getCode(),
                "violation_id is required");
        if (req.getComplaintText() == null) return errorResponse(msgId, MessageCode.NOT_FOUND.getCode(),
                "text is required");

        ComplaintResponse response = complaintService.createComplaint(req);

        return successResponse(msgId, "Complaint created successfully", response);
    }

    private BaseResponse handleGet(Integer msgId, ComplaintRequest req) {
        if (req.getId() == null) return errorResponse(msgId, MessageCode.NOT_FOUND.getCode(), "id is required");

        ComplaintResponse response = complaintService.getComplaintById(req.getId());
        return successResponse(msgId, "Complaint retrieved successfully", response);
    }

    private BaseResponse handleGetAll(Integer msgId, ComplaintRequest req, List<String> emails) {
        Pageable pageable = PageRequest.of(
                req.getPage() != null ? req.getPage() : 0,
                req.getSize() != null ? req.getSize() : 5,
                Sort.by(Sort.Direction.fromString(req.getSortDirection() != null ? req.getSortDirection() : "desc"),
                        req.getSortBy() != null ? req.getSortBy() : "updatedAt"));

        var page = complaintService.getAllComplaintsByEmail(pageable, emails);

        Map<String, Object> data = new HashMap<>();
        data.put("content", page.getContent());
        data.put("page", page.getNumber());
        data.put("size", page.getSize());
        data.put("totalElements", page.getTotalElements());
        data.put("totalPages", page.getTotalPages());

        return successResponse(msgId, "Complaints retrieved successfully", data);
    }

    private BaseResponse handleGetByViolation(Integer msgId, ComplaintRequest req) {
        if (req.getViolationId() == null) return errorResponse(msgId, MessageCode.NOT_FOUND.getCode(),
                "violation_id is required");

        var complaints = complaintService.getComplaintsByViolationId(req.getViolationId());
        Map<String, Object> data = new HashMap<>();
        data.put("content", complaints);
        data.put("totalElements", complaints.size());

        return successResponse(msgId, "Complaints retrieved successfully", data);
    }

    private BaseResponse handleUpdateStatus(Integer msgId, ComplaintRequest req) {
        if (req.getId() == null) return errorResponse(msgId, MessageCode.NOT_FOUND.getCode(), "id is required");
        if (req.getStatus() == null) return errorResponse(msgId, MessageCode.NOT_FOUND.getCode(), "status is required");

        ComplaintStatus status = ComplaintStatus.valueOf(req.getStatus().toUpperCase());
        ComplaintResponse response = complaintService.updateComplaintStatus(req.getId(), status);
        return successResponse(msgId, "Status updated successfully", response);
    }

    private BaseResponse handleUpdate(Integer msgId, ComplaintRequest req) {
        if (req.getId() == null) return errorResponse(msgId, MessageCode.NOT_FOUND.getCode(), "id is required");

        ComplaintResponse response = complaintService.updateComplaint(req.getId(), req);

        return successResponse(msgId, "Complaint updated successfully", response);
    }

    private BaseResponse handleDelete(Integer msgId, ComplaintRequest req) {
        if (req.getId() == null) return errorResponse(msgId, MessageCode.NOT_FOUND.getCode(), "id is required");

        ComplaintResponse response = complaintService.deleteComplaint(req.getId());
        return successResponse(msgId, "Complaint deleted successfully", response);
    }

    private BaseResponse successResponse(Integer msgId, String message, Object data) {
        return new BaseResponse(msgId, MessageCode.SUCCESS.getCode(), message, data);
    }

    private BaseResponse errorResponse(Integer msgId, Integer code, String message) {
        return new BaseResponse(msgId, code, message, new HashMap<>());
    }
}