package de.effigenix.application.production;

import de.effigenix.application.production.command.CancelBatchCommand;
import de.effigenix.domain.production.*;
import de.effigenix.shared.common.Quantity;
import de.effigenix.shared.common.RepositoryError;
import de.effigenix.shared.common.Result;
import de.effigenix.shared.common.UnitOfMeasure;
import de.effigenix.shared.security.ActorId;
import de.effigenix.shared.security.AuthorizationPort;
import de.effigenix.shared.persistence.UnitOfWork;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.DisplayName;
import org.junit.jupiter.api.Test;
import org.junit.jupiter.api.extension.ExtendWith;
import org.mockito.Mock;
import org.mockito.junit.jupiter.MockitoExtension;

import java.math.BigDecimal;
import java.time.LocalDate;
import java.time.OffsetDateTime;
import java.time.ZoneOffset;
import java.util.List;
import java.util.Optional;
import java.util.function.Supplier;

import static org.assertj.core.api.Assertions.assertThat;
import static org.mockito.Mockito.*;

@ExtendWith(MockitoExtension.class)
@DisplayName("CancelBatch Use Case")
class CancelBatchTest {

    @Mock private BatchRepository batchRepository;
    @Mock private AuthorizationPort authPort;
    @Mock private UnitOfWork unitOfWork;

    private CancelBatch cancelBatch;
    private ActorId performedBy;

    @BeforeEach
    void setUp() {
        cancelBatch = new CancelBatch(batchRepository, authPort, unitOfWork);
        performedBy = ActorId.of("admin-user");
        lenient().when(unitOfWork.executeAtomically(any())).thenAnswer(inv -> ((Supplier<?>) inv.getArgument(0)).get());
    }

    private Batch plannedBatch(String id) {
        return Batch.reconstitute(
                BatchId.of(id),
                BatchNumber.generate(LocalDate.of(2026, 3, 1), 1),
                RecipeId.of("recipe-1"),
                BatchStatus.PLANNED,
                Quantity.of(new BigDecimal("100"), UnitOfMeasure.KILOGRAM).unsafeGetValue(),
                null, null, null,
                LocalDate.of(2026, 3, 1),
                LocalDate.of(2026, 6, 1),
                OffsetDateTime.now(ZoneOffset.UTC),
                OffsetDateTime.now(ZoneOffset.UTC),
                null, null, null,
                0L, List.of()
        );
    }

    private Batch inProductionBatch(String id) {
        return Batch.reconstitute(
                BatchId.of(id),
                BatchNumber.generate(LocalDate.of(2026, 3, 1), 1),
                RecipeId.of("recipe-1"),
                BatchStatus.IN_PRODUCTION,
                Quantity.of(new BigDecimal("100"), UnitOfMeasure.KILOGRAM).unsafeGetValue(),
                null, null, null,
                LocalDate.of(2026, 3, 1),
                LocalDate.of(2026, 6, 1),
                OffsetDateTime.now(ZoneOffset.UTC),
                OffsetDateTime.now(ZoneOffset.UTC),
                null, null, null,
                0L, List.of()
        );
    }

    private CancelBatchCommand validCommand(String batchId) {
        return new CancelBatchCommand(batchId, "Qualitätsproblem erkannt");
    }

    @Test
    @DisplayName("should cancel PLANNED batch")
    void should_CancelBatch_When_Planned() {
        var batchId = BatchId.of("batch-1");
        var batch = plannedBatch("batch-1");
        when(authPort.can(performedBy, ProductionAction.BATCH_CANCEL)).thenReturn(true);
        when(batchRepository.findById(batchId)).thenReturn(Result.success(Optional.of(batch)));
        when(batchRepository.save(any())).thenReturn(Result.success(null));

        var result = cancelBatch.execute(validCommand("batch-1"), performedBy);

        assertThat(result.isSuccess()).isTrue();
        assertThat(result.unsafeGetValue().status()).isEqualTo(BatchStatus.CANCELLED);
        assertThat(result.unsafeGetValue().cancellationReason()).isEqualTo("Qualitätsproblem erkannt");
        assertThat(result.unsafeGetValue().cancelledAt()).isNotNull();
        verify(batchRepository).save(batch);
    }

    @Test
    @DisplayName("should cancel IN_PRODUCTION batch")
    void should_CancelBatch_When_InProduction() {
        var batchId = BatchId.of("batch-1");
        var batch = inProductionBatch("batch-1");
        when(authPort.can(performedBy, ProductionAction.BATCH_CANCEL)).thenReturn(true);
        when(batchRepository.findById(batchId)).thenReturn(Result.success(Optional.of(batch)));
        when(batchRepository.save(any())).thenReturn(Result.success(null));

        var result = cancelBatch.execute(validCommand("batch-1"), performedBy);

        assertThat(result.isSuccess()).isTrue();
        assertThat(result.unsafeGetValue().status()).isEqualTo(BatchStatus.CANCELLED);
        verify(batchRepository).save(batch);
    }

    @Test
    @DisplayName("should fail with Unauthorized when actor lacks permission")
    void should_FailWithUnauthorized_When_ActorLacksPermission() {
        when(authPort.can(performedBy, ProductionAction.BATCH_CANCEL)).thenReturn(false);

        var result = cancelBatch.execute(validCommand("batch-1"), performedBy);

        assertThat(result.isFailure()).isTrue();
        assertThat(result.unsafeGetError()).isInstanceOf(BatchError.Unauthorized.class);
        verify(batchRepository, never()).findById(any());
    }

    @Test
    @DisplayName("should fail when batch not found")
    void should_Fail_When_BatchNotFound() {
        when(authPort.can(performedBy, ProductionAction.BATCH_CANCEL)).thenReturn(true);
        when(batchRepository.findById(any())).thenReturn(Result.success(Optional.empty()));

        var result = cancelBatch.execute(validCommand("nonexistent"), performedBy);

        assertThat(result.isFailure()).isTrue();
        assertThat(result.unsafeGetError()).isInstanceOf(BatchError.BatchNotFound.class);
        verify(batchRepository, never()).save(any());
    }

    @Test
    @DisplayName("should fail when batch is COMPLETED")
    void should_Fail_When_BatchIsCompleted() {
        var batchId = BatchId.of("batch-1");
        var batch = Batch.reconstitute(
                batchId,
                BatchNumber.generate(LocalDate.of(2026, 3, 1), 1),
                RecipeId.of("recipe-1"),
                BatchStatus.COMPLETED,
                Quantity.of(new BigDecimal("100"), UnitOfMeasure.KILOGRAM).unsafeGetValue(),
                null, null, null,
                LocalDate.of(2026, 3, 1),
                LocalDate.of(2026, 6, 1),
                OffsetDateTime.now(ZoneOffset.UTC),
                OffsetDateTime.now(ZoneOffset.UTC),
                null, null, null,
                0L, List.of()
        );
        when(authPort.can(performedBy, ProductionAction.BATCH_CANCEL)).thenReturn(true);
        when(batchRepository.findById(batchId)).thenReturn(Result.success(Optional.of(batch)));

        var result = cancelBatch.execute(validCommand("batch-1"), performedBy);

        assertThat(result.isFailure()).isTrue();
        assertThat(result.unsafeGetError()).isInstanceOf(BatchError.InvalidStatusTransition.class);
        verify(batchRepository, never()).save(any());
    }

    @Test
    @DisplayName("should fail when reason is blank")
    void should_Fail_When_ReasonBlank() {
        var batchId = BatchId.of("batch-1");
        var batch = plannedBatch("batch-1");
        when(authPort.can(performedBy, ProductionAction.BATCH_CANCEL)).thenReturn(true);
        when(batchRepository.findById(batchId)).thenReturn(Result.success(Optional.of(batch)));

        var cmd = new CancelBatchCommand("batch-1", "   ");
        var result = cancelBatch.execute(cmd, performedBy);

        assertThat(result.isFailure()).isTrue();
        assertThat(result.unsafeGetError()).isInstanceOf(BatchError.CancellationReasonRequired.class);
        verify(batchRepository, never()).save(any());
    }

    @Test
    @DisplayName("should fail with RepositoryFailure on repository error during find")
    void should_FailWithRepositoryFailure_When_FindFails() {
        when(authPort.can(performedBy, ProductionAction.BATCH_CANCEL)).thenReturn(true);
        when(batchRepository.findById(any()))
                .thenReturn(Result.failure(new RepositoryError.DatabaseError("connection lost")));

        var result = cancelBatch.execute(validCommand("batch-1"), performedBy);

        assertThat(result.isFailure()).isTrue();
        assertThat(result.unsafeGetError()).isInstanceOf(BatchError.RepositoryFailure.class);
    }

    @Test
    @DisplayName("should fail with RepositoryFailure on repository error during save")
    void should_FailWithRepositoryFailure_When_SaveFails() {
        var batchId = BatchId.of("batch-1");
        var batch = plannedBatch("batch-1");
        when(authPort.can(performedBy, ProductionAction.BATCH_CANCEL)).thenReturn(true);
        when(batchRepository.findById(batchId)).thenReturn(Result.success(Optional.of(batch)));
        when(batchRepository.save(any()))
                .thenReturn(Result.failure(new RepositoryError.DatabaseError("write error")));

        var result = cancelBatch.execute(validCommand("batch-1"), performedBy);

        assertThat(result.isFailure()).isTrue();
        assertThat(result.unsafeGetError()).isInstanceOf(BatchError.RepositoryFailure.class);
    }
}
