package org.activityinfo.legacy.shared.adapter;

import com.google.common.base.Function;
import com.google.common.collect.Sets;
import org.activityinfo.core.shared.application.ApplicationProperties;
import org.activityinfo.i18n.shared.I18N;
import org.activityinfo.legacy.shared.model.AdminLevelDTO;
import org.activityinfo.legacy.shared.model.CountryDTO;
import org.activityinfo.legacy.shared.model.LocationTypeDTO;
import org.activityinfo.legacy.shared.model.SchemaDTO;
import org.activityinfo.model.form.FormClass;
import org.activityinfo.model.form.FormField;
import org.activityinfo.model.legacy.CuidAdapter;
import org.activityinfo.model.resource.ResourceId;
import org.activityinfo.model.type.ReferenceType;
import org.activityinfo.model.type.geo.GeoPointType;
import org.activityinfo.model.type.primitive.TextType;

import javax.annotation.Nullable;
import java.util.Set;

import static org.activityinfo.model.legacy.CuidAdapter.adminLevelFormClass;

/**
 * Creates a {@code FormClass} for a LocationType given a legacy SchemaDTO.
 */
public class LocationClassAdapter implements Function<SchemaDTO, FormClass> {

    private final int locationTypeId;
    private ResourceId classId;

    public LocationClassAdapter(int locationTypeId) {
        this.locationTypeId = locationTypeId;
        classId = CuidAdapter.locationFormClass(this.locationTypeId);
    }

    public static ResourceId getPointFieldId(ResourceId classId) {
        return CuidAdapter.field(classId, CuidAdapter.GEOMETRY_FIELD);
    }

    public static ResourceId getAxeFieldId(ResourceId classId) {
        return CuidAdapter.field(classId, CuidAdapter.AXE_FIELD);
    }

    public static ResourceId getNameFieldId(ResourceId classId) {
        return CuidAdapter.field(classId, CuidAdapter.NAME_FIELD);
    }

    public static ResourceId getAdminFieldId(ResourceId classId) {
        return CuidAdapter.field(classId, CuidAdapter.ADMIN_FIELD);
    }

    @Nullable @Override
    public FormClass apply(@Nullable SchemaDTO schema) {
        CountryDTO country = findCountry(schema, locationTypeId);
        LocationTypeDTO locationType = country.getLocationTypeById(locationTypeId);

        FormClass formClass = new FormClass(classId);
        formClass.setLabel(locationType.getName());

        FormField nameField = new FormField(getNameFieldId(classId));
        nameField.setLabel(I18N.CONSTANTS.name());
        nameField.setType(TextType.INSTANCE);
        nameField.setRequired(true);
        nameField.setSuperProperty(ApplicationProperties.LABEL_PROPERTY);
        formClass.addElement(nameField);

        FormField axeField = new FormField(getAxeFieldId(classId));
        axeField.setLabel(I18N.CONSTANTS.alternateName());
        axeField.setType(TextType.INSTANCE);
        formClass.addElement(axeField);

        // the range for the location object is any AdminLevel in this country
        Set<ResourceId> adminRange = Sets.newHashSet();
        for (AdminLevelDTO level : country.getAdminLevels()) {
            adminRange.add(adminLevelFormClass(level.getId()));
        }

        FormField adminField = new FormField(getAdminFieldId(classId));
        adminField.setLabel(I18N.CONSTANTS.adminEntities());
        adminField.setType(ReferenceType.single(adminRange));
        adminField.addSuperProperty(ApplicationProperties.HIERARCHIAL);
        formClass.addElement(adminField);

        FormField pointField = new FormField(getPointFieldId(classId));
        pointField.setLabel(I18N.CONSTANTS.geographicCoordinatesFieldLabel());
        pointField.setType(GeoPointType.INSTANCE);
        formClass.addElement(pointField);

        return formClass;
    }

    private CountryDTO findCountry(SchemaDTO schema, int locationTypeId) {
        for (CountryDTO country : schema.getCountries()) {
            for (LocationTypeDTO locationType : country.getLocationTypes()) {
                if (locationType.getId() == locationTypeId) {
                    return country;
                }
            }
        }
        throw new IllegalArgumentException("LocationType with id " + locationTypeId + " not found");
    }

}