org.finos.tracdap.common.plugin.TracPlugin Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of tracdap-lib-common Show documentation
TRAC D.A.P. common library, interfaces and utilities used across all TRAC components
There is a newer version: 0.6.3
/*
 * Copyright 2022 Accenture Global Solutions Limited
 *
 * 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.
 */

package org.finos.tracdap.common.plugin;

import org.finos.tracdap.common.config.ConfigManager;
import org.finos.tracdap.common.exception.EPluginNotAvailable;
import org.finos.tracdap.config.PluginConfig;

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

public abstract class TracPlugin implements ITracPlugin {

    protected  T createService(String serviceName, Properties properties, ConfigManager configManager) {

        var message = String.format("Plugin [%s] does not support the service [%s]", pluginName(), serviceName);
        throw new EPluginNotAvailable(message);
    }

    protected  T createConfigService(String serviceName, Properties properties) {

        var message = String.format("Plugin [%s] does not support the service [%s]", pluginName(), serviceName);
        throw new EPluginNotAvailable(message);
    }

    @Override
    public final List protocols(Class service) {

        var psi = serviceInfo();

        var matchingPsi = psi.stream()
                .filter(si -> si.serviceClass() == service)
                .collect(Collectors.toList());

        if (matchingPsi.isEmpty())
            throw new IllegalArgumentException();

        return matchingPsi.stream()
                .map(PluginServiceInfo::protocols)
                .reduce(new ArrayList<>(), (ps, p) -> {ps.addAll(p); return ps;});
    }

    @Override
    public final  T createService(Class serviceClass, PluginConfig config, ConfigManager configManager) {

        var serviceInfo = lookupService(serviceClass, config.getProtocol());

        var properties = new Properties();
        properties.putAll(config.getPropertiesMap());

        for (var secret : config.getSecretsMap().entrySet()) {
            var secretKey = secret.getKey();
            var secretValue = configManager.loadPassword(secret.getValue());
            properties.put(secretKey, secretValue);
        }

        return createService(serviceInfo.serviceName(), properties, configManager);
    }

    @Override
    public final  T createConfigService(Class serviceClass, PluginConfig config) {

        var serviceInfo = lookupService(serviceClass, config.getProtocol());

        var properties = new Properties();
        properties.putAll(config.getPropertiesMap());

        return createConfigService(serviceInfo.serviceName(), properties);
    }

    private  PluginServiceInfo lookupService(Class serviceClass, String protocol) {

        var psi = serviceInfo();

        var matchingPsi = psi.stream()
                .filter(psi_ -> psiMatch(psi_, serviceClass, protocol))
                .findFirst();

        if (matchingPsi.isEmpty())
            throw new IllegalArgumentException();

        return matchingPsi.get();
    }

    private static boolean psiMatch(PluginServiceInfo psi, Class service, String protocol) {

        if (psi.serviceClass() != service)
            return false;

        for (var psiProtocol : psi.protocols()) {

            if (psiProtocol.equalsIgnoreCase(protocol))
                return true;
        }

        return false;
    }
}