/*
 * Copyright 2017 ThoughtWorks, Inc.
 *
 * 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 com.thoughtworks.go.plugin.access.notification;

import org.springframework.stereotype.Component;

import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;

@Component
public class NotificationPluginRegistry {
    private final Map<String, Set<String>> notificationNameToPluginsInterestedMap = new ConcurrentHashMap<>();
    private Set<String> notificationPlugins = new HashSet<>();

    public void registerPluginInterests(String pluginId, List<String> notificationNames) {
        if (notificationNames != null && !notificationNames.isEmpty()) {
            for (String notificationName : notificationNames) {
                addToMap(notificationName, pluginId);
            }
        }
    }

    private void addToMap(String notificationName, String pluginId) {
        if (notificationNameToPluginsInterestedMap.get(notificationName) == null) {
            notificationNameToPluginsInterestedMap.put(notificationName, new HashSet<>());
        }
        notificationNameToPluginsInterestedMap.get(notificationName).add(pluginId);
    }

    public void removePluginInterests(String pluginId) {
        for (String key : notificationNameToPluginsInterestedMap.keySet()) {
            if (notificationNameToPluginsInterestedMap.get(key).contains(pluginId)) {
                notificationNameToPluginsInterestedMap.get(key).remove(pluginId);
            }
        }
    }

    public boolean isAnyPluginInterestedIn(String notificationName) {
        return !getPluginsInterestedIn(notificationName).isEmpty();
    }

    public Set<String> getPluginsInterestedIn(String notificationName) {
        Set<String> plugins = notificationNameToPluginsInterestedMap.get(notificationName);
        return plugins == null ? new HashSet<>() : plugins;
    }

    public Set<String> getPluginInterests(String pluginId) {
        Set<String> pluginInterests = new HashSet<>();
        for (String key : notificationNameToPluginsInterestedMap.keySet()) {
            if (notificationNameToPluginsInterestedMap.get(key).contains(pluginId)) {
                pluginInterests.add(key);
            }
        }
        return pluginInterests;
    }

    @Deprecated
    public void clear() {
        notificationNameToPluginsInterestedMap.clear();
    }

    public void registerPlugin(String pluginId) {
        notificationPlugins.add(pluginId);
    }

    public void deregisterPlugin(String pluginId) {
        notificationPlugins.remove(pluginId);
    }

    public Set<String> getNotificationPlugins() {
        return notificationPlugins;
    }
}