package com.grash.mapper;

import com.grash.dto.FileShowDTO;
import com.grash.dto.MeterMiniDTO;
import com.grash.dto.MeterPatchDTO;
import com.grash.dto.MeterShowDTO;
import com.grash.model.Meter;
import com.grash.model.Reading;
import com.grash.service.ReadingService;
import com.grash.utils.AuditComparator;
import com.grash.utils.Helper;
import org.mapstruct.AfterMapping;
import org.mapstruct.Context;
import org.mapstruct.Mapper;
import org.mapstruct.MappingTarget;

import java.util.Collection;
import java.util.Collections;
import java.util.Date;

@Mapper(componentModel = "spring", uses = {LocationMapper.class, AssetMapper.class, UserMapper.class, FileMapper.class})
public interface MeterMapper {
    Meter updateMeter(@MappingTarget Meter entity, MeterPatchDTO dto);

    MeterPatchDTO toPatchDto(Meter model);

    MeterShowDTO toShowDto(Meter model, @Context ReadingService readingService);

    @AfterMapping
    default MeterShowDTO toShowDto(Meter model, @MappingTarget MeterShowDTO target,
                                   @Context ReadingService readingService) {
        Collection<Reading> readings = readingService.findByMeter(target.getId());
        if (!readings.isEmpty()) {
            Reading lastReading = Collections.max(readings, new AuditComparator());
            target.setLastReading(lastReading.getCreatedAt());
            Date nextReading = Helper.incrementDays(lastReading.getCreatedAt(),
                    target.getUpdateFrequency());
            target.setNextReading(nextReading);
        }
        return target;
    }

    MeterMiniDTO toMiniDto(Meter model);
}