package com.tim2.travelservice.common.utils;

import com.tim2.travelservice.entity.CityLine;
import com.tim2.travelservice.entity.PriceList;
import com.tim2.travelservice.entity.PriceListItem;
import com.tim2.travelservice.entity.Ticket;
import com.tim2.travelservice.web.dto.PriceListItemDto;
import org.modelmapper.ModelMapper;

public class JsonUtil {

    private static final ModelMapper modelMapper = new ModelMapper();

    public static PriceListItemDto map(PriceListItem priceListItem) {
        hideUnimportantFields(priceListItem);
        return modelMapper.map(priceListItem, PriceListItemDto.class);
    }

    private static void hideUnimportantFields(PriceListItem priceListItem) {
        if (priceListItem == null) {
            return;
        }
        hideUnimportantFields(priceListItem.getPriceList());
        hideUnimportantFields(priceListItem.getTicket());
    }

    private static void hideUnimportantFields(Ticket ticket) {
        if (ticket == null) {
            return;
        }
        ticket.setBoughtTickets(null);
        ticket.setPriceListItems(null);
        hideUnimportantFields(ticket.getCityLine());
    }

    private static void hideUnimportantFields(CityLine cityLine) {
        if (cityLine == null) {
            return;
        }
        cityLine.setLineStations(null);
        cityLine.setSchedules(null);
        cityLine.setVehicles(null);
    }

    public static void hideUnimportantFields(PriceList priceList) {
        if (priceList == null) {
            return;
        }
        priceList.setPriceListItems(null);
    }
}