org.cqframework.cql.cql2elm.ModelInfoLoader Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of cql-to-elm Show documentation
The cql-to-elm library for the Clinical Quality Language Java reference implementation
There is a newer version: 3.18.0
package org.cqframework.cql.cql2elm;

import java.nio.file.Path;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;
import org.hl7.cql.model.ModelIdentifier;
import org.hl7.cql.model.ModelInfoProvider;
import org.hl7.cql.model.NamespaceAware;
import org.hl7.cql.model.NamespaceManager;
import org.hl7.elm_modelinfo.r1.ModelInfo;

public class ModelInfoLoader implements NamespaceAware, PathAware {

    private Path path;

    private NamespaceManager namespaceManager;

    private final List providers = new ArrayList<>();

    private boolean initialized = false;

    public ModelInfoLoader() {}

    private List getProviders() {
        if (!initialized) {
            initialized = true;
            for (Iterator it = ModelInfoProviderFactory.providers(false); it.hasNext(); ) {
                ModelInfoProvider provider = it.next();
                registerModelInfoProvider(provider);
            }
        }
        return providers;
    }

    public ModelInfo getModelInfo(ModelIdentifier modelIdentifier) {
        checkModelIdentifier(modelIdentifier);

        ModelInfo modelInfo = null;

        for (ModelInfoProvider provider : getProviders()) {
            modelInfo = provider.load(modelIdentifier);
            if (modelInfo != null) {
                break;
            }
        }

        if (modelInfo == null) {
            throw new IllegalArgumentException(String.format(
                    "Could not resolve model info provider for model %s, version %s.",
                    modelIdentifier.getSystem() == null
                            ? modelIdentifier.getId()
                            : NamespaceManager.getPath(modelIdentifier.getSystem(), modelIdentifier.getId()),
                    modelIdentifier.getVersion()));
        }

        return modelInfo;
    }

    public void registerModelInfoProvider(ModelInfoProvider provider) {
        registerModelInfoProvider(provider, false);
    }

    public void registerModelInfoProvider(ModelInfoProvider provider, boolean priority) {
        if (provider == null) {
            throw new IllegalArgumentException("Provider is null");
        }

        if (namespaceManager != null) {
            if (provider instanceof NamespaceAware) {
                ((NamespaceAware) provider).setNamespaceManager(namespaceManager);
            }
        }

        if (path != null) {
            if (provider instanceof PathAware) {
                ((PathAware) provider).setPath(path);
            }
        }

        if (priority) {
            providers.add(0, provider);
        } else {
            providers.add(provider);
        }
    }

    public void unregisterModelInfoProvider(ModelInfoProvider provider) {
        providers.remove(provider);
    }

    public void clearModelInfoProviders() {
        providers.clear();
        initialized = false;
    }

    private void checkModelIdentifier(ModelIdentifier modelIdentifier) {
        if (modelIdentifier == null) {
            throw new IllegalArgumentException("modelIdentifier is null.");
        }

        if (modelIdentifier.getId() == null || modelIdentifier.getId().equals("")) {
            throw new IllegalArgumentException("modelIdentifier Id is null or empty.");
        }
    }

    @Override
    public void setNamespaceManager(NamespaceManager namespaceManager) {
        this.namespaceManager = namespaceManager;

        for (ModelInfoProvider provider : getProviders()) {
            if (provider instanceof NamespaceAware) {
                ((NamespaceAware) provider).setNamespaceManager(namespaceManager);
            }
        }
    }

    public void setPath(Path path) {
        if (path == null || !path.toFile().isDirectory()) {
            throw new IllegalArgumentException(String.format("path '%s' is not a valid directory", path));
        }

        this.path = path;

        for (ModelInfoProvider provider : getProviders()) {
            if (provider instanceof PathAware) {
                ((PathAware) provider).setPath(path);
            }
        }
    }
}