com.regnosys.testing.schemeimport.iso.currency.IsoCurrencySchemeEnumReader Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of rosetta-testing Show documentation
Rune Testing is a java library that is utilised by Rosetta Code Generators and models expressed in the Rosetta DSL.
There is a newer version: 11.27.2
package com.regnosys.testing.schemeimport.iso.currency;

/*-
 * ===============
 * Rune Testing
 * ===============
 * Copyright (C) 2022 - 2024 REGnosys
 * ===============
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 * 
 *      http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 * ===============
 */

import com.regnosys.rosetta.rosetta.RosettaEnumValue;
import com.regnosys.rosetta.rosetta.RosettaFactory;
import com.regnosys.rosetta.rosetta.RosettaNamed;
import com.regnosys.rosetta.rosetta.impl.RosettaFactoryImpl;
import com.regnosys.testing.schemeimport.SchemeEnumReader;
import org.apache.commons.lang3.tuple.ImmutablePair;
import org.iso.currency.ISO4217;

import javax.xml.bind.JAXBContext;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Unmarshaller;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.URISyntaxException;
import java.net.URL;
import java.net.http.HttpClient;
import java.net.http.HttpRequest;
import java.net.http.HttpResponse;
import java.nio.charset.StandardCharsets;
import java.time.Duration;
import java.util.Comparator;
import java.util.List;
import java.util.stream.Collectors;

import static java.time.temporal.ChronoUnit.SECONDS;

public class IsoCurrencySchemeEnumReader implements SchemeEnumReader {
    private final HttpClient httpClient;

    public IsoCurrencySchemeEnumReader() {
        this.httpClient = HttpClient.newHttpClient();
    }

    @Override
    public List generateEnumFromScheme(URL schemaLocationForEnum) {
        try {
            ISO4217 iso4217 = parseSchemaFile(schemaLocationForEnum);
            return transformToEnums(iso4217);
        } catch (Exception e) {
            throw new RuntimeException(e);
        }
    }

    private List transformToEnums(ISO4217 iso4217) {
        return iso4217.getCcyTbl().getCcyNtry().stream()
                .filter(ccyNtry -> ccyNtry.getCcy() != null) //filters out Antarctica which has no ccy code!
                .map(ccyNtry -> new ImmutablePair<>(ccyNtry.getCcy(), ccyNtry.getCcyNm().getValue()))
                .distinct()
                .map(pair -> createEnumValue(pair.getLeft(), pair.getRight()))
                .sorted(Comparator.comparing(RosettaNamed::getName))
                .collect(Collectors.toList());
    }

    private RosettaEnumValue createEnumValue(String currencyCode, String currencyName) {
        RosettaFactory factory = RosettaFactoryImpl.eINSTANCE;
        RosettaEnumValue ev = factory.createRosettaEnumValue();
        ev.setName(currencyCode);
        ev.setDefinition(currencyName);
        return ev;
    }

    private ISO4217 parseSchemaFile(URL schemaLocationForEnum) throws JAXBException, IOException, URISyntaxException, InterruptedException {
        String xmlContents = getXmlContents(schemaLocationForEnum);
        JAXBContext jaxbContext = JAXBContext.newInstance(ISO4217.class);
        Unmarshaller jaxbUnmarshaller = jaxbContext.createUnmarshaller();
        InputStream inputStream = new ByteArrayInputStream(removeLineBreaks(xmlContents).getBytes(StandardCharsets.UTF_8));
        return (ISO4217) jaxbUnmarshaller.unmarshal(inputStream);
    }

    private String removeLineBreaks(String response) {
        return response
                .replace("\n", "")
                .replace("\t", "");
    }

    private String getXmlContents(URL schemaLocationForEnum) throws URISyntaxException, IOException, InterruptedException {
        HttpRequest request = HttpRequest.newBuilder()
                .uri(schemaLocationForEnum.toURI())
                .timeout(Duration.of(10, SECONDS))
                .GET()
                .build();

        HttpResponse response = httpClient.send(request, HttpResponse.BodyHandlers.ofString());
        if (response.statusCode() != 200) {
            throw new RuntimeException("Response from xml get: " + response.statusCode());
        }
        return response.body();
    }
}