package com.ydtech.modules.ins.model.zm;

import com.ydtech.constants.enums.JingyouEnergyTypes;
import com.ydtech.modules.ins.model.ya.CarModelDTO;
import lombok.Data;

import java.util.List;
import java.util.stream.Collectors;

@Data
public class ZmCarModel {

    private String analogyPrice;
    private String analogyPriceTax;
    private String approvedLoad;
    private String approvedPassengersCapacity;
    private String brand;
    private String brandCode;
    private String carAlias;
    private String carKind;
    private String carType;
    private String carYear;
    private String createDate;
    private String displacement;
    private String energyTypes;
    private String factory;
    private String fuelType;
    private String fuelTypeCode;
    private String grossMass;
    private String isrelief;
    private String jqxClassId;
    private String jqxClassName;
    private String localModelCode;
    private String modelName;
    private String motorType;
    private String newVehicleClass;
    private String newVehicleClassCode;
    private String originalPurchasePrice;
    private String power;
    private String purchasePrice;
    private String purchasePriceTax;
    private String qualityMax;
    private String qualityMin;
    private String remark;
    private String searchCodeAlias;
    private String series;
    private String seriesCode;
    private String shortHandCode;
    private String status;
    private String syxClassName;
    private String syxClassid;
    private String taxReliefFlag;
    private String tractionMass;
    private String transmissionType;
    private String unladenMass;
    private String updateDate;
    private String vehicleSeatMax;
    private String vehicleSeatMin;

    public static List<CarModelDTO> toCarModelDTOs(List<ZmCarModel> data) {
        return data.stream().map(ZmCarModel::toCarModelDTO).collect(Collectors.toList());
    }

    public static CarModelDTO toCarModelDTO(ZmCarModel zmCarModel) {
        CarModelDTO carModelDTO = new CarModelDTO();
        carModelDTO.setModelCode(zmCarModel.getLocalModelCode());
        carModelDTO.setModelName(zmCarModel.getModelName());
        carModelDTO.setBrandCN(zmCarModel.getBrand());
        carModelDTO.setSeries(zmCarModel.getSeries());
        carModelDTO.setRatedPassengerCapacity(zmCarModel.getApprovedPassengersCapacity());
        carModelDTO.setDisplacement(zmCarModel.getDisplacement());
        carModelDTO.setMarketYear(zmCarModel.getCarYear());
        carModelDTO.setReplacementValue(zmCarModel.getOriginalPurchasePrice());
        carModelDTO.setRemark(zmCarModel.getRemark());
        carModelDTO.setFuelType(zmCarModel.getFuelType());

        String energyTypeByCode = JingyouEnergyTypes.getEnergyTypeByCode(zmCarModel.getFuelTypeCode());
        carModelDTO.setFuelTypeCode(energyTypeByCode);
        carModelDTO.setEnergyTypes(energyTypeByCode);
        carModelDTO.setPowerScale(zmCarModel.getPower());
        carModelDTO.setVehicleSeatMax(zmCarModel.getVehicleSeatMax());
        carModelDTO.setVehicleSeatMin(zmCarModel.getVehicleSeatMin());
        return carModelDTO;
    }

}