org.sonar.server.plugins.ws.InstallPluginsWsAction.java Source code

Introduction

Here is the source code for org.sonar.server.plugins.ws.InstallPluginsWsAction.java

Source

/*
 * SonarQube, open source software quality management tool.
 * Copyright (C) 2008-2014 SonarSource
 * mailto:contact AT sonarsource DOT com
 *
 * SonarQube is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 3 of the License, or (at your option) any later version.
 *
 * SonarQube is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
 */
package org.sonar.server.plugins.ws;

import com.google.common.base.Predicate;
import com.google.common.collect.Iterables;
import org.sonar.api.server.ws.Request;
import org.sonar.api.server.ws.Response;
import org.sonar.api.server.ws.WebService;
import org.sonar.core.permission.GlobalPermissions;
import org.sonar.server.plugins.PluginDownloader;
import org.sonar.server.plugins.UpdateCenterMatrixFactory;
import org.sonar.server.user.UserSession;
import org.sonar.updatecenter.common.PluginUpdate;

import javax.annotation.Nullable;

import static java.lang.String.format;

/**
 * Implementation of the {@code install} action for the Plugins WebService.
 */
public class InstallPluginsWsAction implements PluginsWsAction {

    private static final String PARAM_KEY = "key";
    private static final PluginUpdate MISSING_PLUGIN = null;

    private final UpdateCenterMatrixFactory updateCenterFactory;
    private final PluginDownloader pluginDownloader;

    public InstallPluginsWsAction(UpdateCenterMatrixFactory updateCenterFactory,
            PluginDownloader pluginDownloader) {
        this.updateCenterFactory = updateCenterFactory;
        this.pluginDownloader = pluginDownloader;
    }

    @Override
    public void define(WebService.NewController controller) {
        WebService.NewAction action = controller.createAction("install").setPost(true)
                .setDescription("Installs the latest version of a plugin specified by its key." + "<br/>"
                        + "Requires user to be authenticated with Administer System permissions")
                .setHandler(this);

        action.createParam(PARAM_KEY).setRequired(true).setDescription("The key identifying the plugin to install");
    }

    @Override
    public void handle(Request request, Response response) throws Exception {
        UserSession.get().checkGlobalPermission(GlobalPermissions.SYSTEM_ADMIN);
        String key = request.mandatoryParam(PARAM_KEY);
        PluginUpdate pluginUpdate = findAvailablePluginByKey(key);
        pluginDownloader.download(key, pluginUpdate.getRelease().getVersion());
        response.noContent();
    }

    private PluginUpdate findAvailablePluginByKey(String key) {
        PluginUpdate pluginUpdate = Iterables.find(
                updateCenterFactory.getUpdateCenter(false).findAvailablePlugins(), hasKey(key), MISSING_PLUGIN);
        if (pluginUpdate == MISSING_PLUGIN) {
            throw new IllegalArgumentException(format(
                    "No plugin with key '%s' or plugin '%s' is already installed in latest version", key, key));
        }
        return pluginUpdate;
    }

    private static PluginKeyPredicate hasKey(String key) {
        return new PluginKeyPredicate(key);
    }

    private static class PluginKeyPredicate implements Predicate<PluginUpdate> {
        private final String key;

        public PluginKeyPredicate(String key) {
            this.key = key;
        }

        @Override
        public boolean apply(@Nullable PluginUpdate input) {
            return input != null && key.equals(input.getPlugin().getKey());
        }
    }
}