aswi2022code-of-duty-gitlab/backend/src/main/java/cz/zcu/kiv/backendapi/catalog/CatalogItemDto.java @ abaa5f46

package cz.zcu.kiv.backendapi.catalog;

import com.fasterxml.jackson.annotation.JsonProperty;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;

import java.util.*;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import java.util.stream.Collectors;

/**
 * Class representing catalog item DTO
 */
@Data
@AllArgsConstructor
@NoArgsConstructor
public class CatalogItemDto {
    private static final String INTEGER_PATTERN = "\\d+";
    private static final String DOUBLE_PATTERN = "(\\d+[.]\\d+)|(\\d+)";
    private static final String EMPTY_ENTRY = "–";
    /**
     * Id
     */
    @JsonProperty(access = JsonProperty.Access.READ_ONLY)
    private UUID id;

    /**
     * Name of catalog item
     */
    private String name;

    /**
     * Certainty
     */
    private int certainty;

    /**
     * Longitude
     */
    private double longitude;

    /**
     * Latitude
     */
    private double latitude;

    /**
     * Bibliography
     */
    private Set<String> bibliography = Collections.emptySet();

    /**
     * Countries
     */
    private Set<String> countries = Collections.emptySet();

    /**
     * Written forms
     */
    private Set<String> writtenForms = Collections.emptySet();

    /**
     * Alternative names
     */
    private Set<String> alternativeNames = Collections.emptySet();

    /**
     * Types
     */
    private Set<String> types = Collections.emptySet();

    public CatalogItemDto(List<String> csvFields) {
        this.name = csvFields.get(1);
        Set<String> stringList = processListField(csvFields.get(2));
        this.alternativeNames = new HashSet<>(stringList);

        this.certainty = processIntField(csvFields.get(3));
        this.latitude = processDoubleField(csvFields.get(4));
        this.longitude = processDoubleField(csvFields.get(5));

        stringList = processListField(csvFields.get(6));
        this.writtenForms = new HashSet<>(stringList);

        stringList = processListField(csvFields.get(7));
        this.types = new HashSet<>(stringList);

        stringList = processListField(csvFields.get(8));
        this.countries = new HashSet<>(stringList);

        stringList = processListField(csvFields.get(9));
        this.bibliography = new HashSet<>(stringList);
    }

    private int processIntField(String field) {
        Matcher matcher = Pattern.compile(INTEGER_PATTERN).matcher(field);
        if (matcher.find()) {
            return Integer.parseInt(matcher.group());
        } else {
            return 0;
        }
    }

    private double processDoubleField(String field) {
        Matcher matcher = Pattern.compile(DOUBLE_PATTERN).matcher(field);
        if (matcher.find()) {
            return Double.parseDouble(matcher.group());
        } else {
            return 0.0;
        }
    }

    private Set<String> processListField(String field) {
        if (field.isEmpty() || field.equals(EMPTY_ENTRY)) {
            return new HashSet<>();
        }
        return Arrays.stream(field.split(",")).map(String::trim).filter(item -> !item.isEmpty()).collect(Collectors.toSet());
    }

}