package de.effigenix.application.production;

import de.effigenix.application.production.command.CompleteBatchCommand;
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("CompleteBatch Use Case")
class CompleteBatchTest {

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

    private CompleteBatch completeBatch;
    private ActorId performedBy;

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

    private Batch inProductionBatchWithConsumption(String id) {
        var batch = 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()
        );
        batch.recordConsumption(new ConsumptionDraft("input-1", "article-1", "5.0", "KILOGRAM"));
        return batch;
    }

    private Batch inProductionBatchWithoutConsumption(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 CompleteBatchCommand validCommand(String batchId) {
        return new CompleteBatchCommand(batchId, "95", "KILOGRAM", "5", "KILOGRAM", "Test remarks");
    }

    @Test
    @DisplayName("should complete batch with actual quantity and waste")
    void should_CompleteBatch_When_Valid() {
        var batchId = BatchId.of("batch-1");
        var batch = inProductionBatchWithConsumption("batch-1");
        when(authPort.can(performedBy, ProductionAction.BATCH_COMPLETE)).thenReturn(true);
        when(batchRepository.findById(batchId)).thenReturn(Result.success(Optional.of(batch)));
        when(batchRepository.save(any())).thenReturn(Result.success(null));

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

        assertThat(result.isSuccess()).isTrue();
        assertThat(result.unsafeGetValue().status()).isEqualTo(BatchStatus.COMPLETED);
        assertThat(result.unsafeGetValue().actualQuantity().amount()).isEqualByComparingTo(new BigDecimal("95"));
        assertThat(result.unsafeGetValue().waste().amount()).isEqualByComparingTo(new BigDecimal("5"));
        assertThat(result.unsafeGetValue().remarks()).isEqualTo("Test remarks");
        verify(batchRepository).save(batch);
    }

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

        var result = completeBatch.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 no consumptions recorded")
    void should_Fail_When_NoConsumptions() {
        var batchId = BatchId.of("batch-1");
        var batch = inProductionBatchWithoutConsumption("batch-1");
        when(authPort.can(performedBy, ProductionAction.BATCH_COMPLETE)).thenReturn(true);
        when(batchRepository.findById(batchId)).thenReturn(Result.success(Optional.of(batch)));

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

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

    @Test
    @DisplayName("should fail when actualQuantity is zero")
    void should_Fail_When_ActualQuantityZero() {
        var batchId = BatchId.of("batch-1");
        var batch = inProductionBatchWithConsumption("batch-1");
        when(authPort.can(performedBy, ProductionAction.BATCH_COMPLETE)).thenReturn(true);
        when(batchRepository.findById(batchId)).thenReturn(Result.success(Optional.of(batch)));

        var cmd = new CompleteBatchCommand("batch-1", "0", "KILOGRAM", "5", "KILOGRAM", null);
        var result = completeBatch.execute(cmd, performedBy);

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

    @Test
    @DisplayName("should fail when batch is PLANNED (not in production)")
    void should_Fail_When_BatchIsPlanned() {
        var batchId = BatchId.of("batch-1");
        var batch = Batch.reconstitute(
                batchId,
                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()
        );
        when(authPort.can(performedBy, ProductionAction.BATCH_COMPLETE)).thenReturn(true);
        when(batchRepository.findById(batchId)).thenReturn(Result.success(Optional.of(batch)));

        var result = completeBatch.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 waste is negative")
    void should_Fail_When_NegativeWaste() {
        var batchId = BatchId.of("batch-1");
        var batch = inProductionBatchWithConsumption("batch-1");
        when(authPort.can(performedBy, ProductionAction.BATCH_COMPLETE)).thenReturn(true);
        when(batchRepository.findById(batchId)).thenReturn(Result.success(Optional.of(batch)));

        var cmd = new CompleteBatchCommand("batch-1", "95", "KILOGRAM", "-1", "KILOGRAM", null);
        var result = completeBatch.execute(cmd, performedBy);

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

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

        var result = completeBatch.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 with RepositoryFailure on repository error during find")
    void should_FailWithRepositoryFailure_When_FindFails() {
        when(authPort.can(performedBy, ProductionAction.BATCH_COMPLETE)).thenReturn(true);
        when(batchRepository.findById(any()))
                .thenReturn(Result.failure(new RepositoryError.DatabaseError("connection lost")));

        var result = completeBatch.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 = inProductionBatchWithConsumption("batch-1");
        when(authPort.can(performedBy, ProductionAction.BATCH_COMPLETE)).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 = completeBatch.execute(validCommand("batch-1"), performedBy);

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