001 /*
002 * Sonar, open source software quality management tool.
003 * Copyright (C) 2008-2012 SonarSource
004 * mailto:contact AT sonarsource DOT com
005 *
006 * Sonar is free software; you can redistribute it and/or
007 * modify it under the terms of the GNU Lesser General Public
008 * License as published by the Free Software Foundation; either
009 * version 3 of the License, or (at your option) any later version.
010 *
011 * Sonar is distributed in the hope that it will be useful,
012 * but WITHOUT ANY WARRANTY; without even the implied warranty of
013 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
014 * Lesser General Public License for more details.
015 *
016 * You should have received a copy of the GNU Lesser General Public
017 * License along with Sonar; if not, write to the Free Software
018 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02
019 */
020 package org.sonar.server.plugins;
021
022 import com.google.common.collect.ArrayListMultimap;
023 import com.google.common.collect.ListMultimap;
024 import org.sonar.api.Extension;
025 import org.sonar.api.ExtensionProvider;
026 import org.sonar.api.Plugin;
027 import org.sonar.api.ServerExtension;
028 import org.sonar.api.platform.ComponentContainer;
029 import org.sonar.api.platform.PluginMetadata;
030 import org.sonar.api.platform.PluginRepository;
031
032 import java.util.Map;
033
034 public class ServerExtensionInstaller {
035 private PluginRepository pluginRepository;
036
037 public ServerExtensionInstaller(PluginRepository pluginRepository) {
038 this.pluginRepository = pluginRepository;
039 }
040
041 public void registerExtensions(ComponentContainer container) {
042 ListMultimap<PluginMetadata, Object> installedExtensionsByPlugin = ArrayListMultimap.create();
043
044 for (PluginMetadata pluginMetadata : pluginRepository.getMetadata()) {
045 Plugin plugin = pluginRepository.getPlugin(pluginMetadata.getKey());
046 container.addExtension(pluginMetadata, plugin);
047
048 for (Object extension : plugin.getExtensions()) {
049 if (installExtension(container, pluginMetadata, extension, true) != null) {
050 installedExtensionsByPlugin.put(pluginMetadata, extension);
051 } else {
052 container.declareExtension(pluginMetadata, extension);
053 }
054 }
055 }
056 for (Map.Entry<PluginMetadata, Object> entry : installedExtensionsByPlugin.entries()) {
057 PluginMetadata plugin = entry.getKey();
058 Object extension = entry.getValue();
059 if (isExtensionProvider(extension)) {
060 ExtensionProvider provider = (ExtensionProvider) container.getComponentByKey(extension);
061 installProvider(container, plugin, provider);
062 }
063 }
064 }
065
066 private void installProvider(ComponentContainer container, PluginMetadata plugin, ExtensionProvider provider) {
067 Object obj = provider.provide();
068 if (obj != null) {
069 if (obj instanceof Iterable) {
070 for (Object ext : (Iterable) obj) {
071 installExtension(container, plugin, ext, false);
072 }
073 } else {
074 installExtension(container, plugin, obj, false);
075 }
076 }
077 }
078
079 Object installExtension(ComponentContainer container, PluginMetadata pluginMetadata, Object extension, boolean acceptProvider) {
080 if (isType(extension, ServerExtension.class)) {
081 if (!acceptProvider && isExtensionProvider(extension)) {
082 throw new IllegalStateException("ExtensionProvider can not include providers itself: " + extension);
083 }
084 container.addExtension(pluginMetadata, extension);
085 return extension;
086 }
087 return null;
088 }
089
090 static boolean isExtensionProvider(Object extension) {
091 return isType(extension, ExtensionProvider.class) || extension instanceof ExtensionProvider;
092 }
093
094 static boolean isType(Object extension, Class<? extends Extension> extensionClass) {
095 Class clazz = (extension instanceof Class ? (Class) extension : extension.getClass());
096 return extensionClass.isAssignableFrom(clazz);
097 }
098 }