com.eugenePetrenko.idea.dependencies.actions.OnModuleAction.java Source code

Introduction

Here is the source code for com.eugenePetrenko.idea.dependencies.actions.OnModuleAction.java

Source

/*
 * Copyright 2013-2013 Eugene Petrenko
 *
 * 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.eugenePetrenko.idea.dependencies.actions;

import com.eugenePetrenko.idea.dependencies.AnalyzeStrategy;
import com.eugenePetrenko.idea.dependencies.data.ModulesDependencies;
import com.eugenePetrenko.idea.dependencies.ui.Comparators;
import com.intellij.openapi.actionSystem.AnAction;
import com.intellij.openapi.actionSystem.AnActionEvent;
import com.intellij.openapi.actionSystem.LangDataKeys;
import com.intellij.openapi.module.Module;
import com.intellij.openapi.progress.ProgressIndicator;
import com.intellij.openapi.progress.ProgressManager;
import com.intellij.openapi.progress.Task;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.ui.Messages;
import com.intellij.openapi.util.text.StringUtil;
import com.intellij.openapi.vcs.changes.BackgroundFromStartOption;
import com.intellij.util.Function;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import java.util.Arrays;

import static com.eugenePetrenko.idea.dependencies.AnalyzeStrategies.SKIP_EXPORT_DEPENDENCIES;
import static com.eugenePetrenko.idea.dependencies.AnalyzeStrategies.WITH_EXPORT_DEPENDENCIES;
import static com.eugenePetrenko.idea.dependencies.ModuleDependenciesAnalyzer.processModulesDependencies;

/**
 * Created 07.04.13 15:57
 *
 * @author Eugene Petrenko (eugene.petrenko@jetbrains.com)
 */
public class OnModuleAction extends AnAction {
    @Override
    public void update(AnActionEvent e) {
        super.update(e);

        if (e == null)
            return;
        final Module[] modules = e.getData(LangDataKeys.MODULE_CONTEXT_ARRAY);
        e.getPresentation().setEnabled(null != modules && modules.length > 0 && e.getProject() != null);
    }

    @Override
    public void actionPerformed(@Nullable final AnActionEvent e) {
        if (e == null)
            return;

        final Module[] modules = e.getData(LangDataKeys.MODULE_CONTEXT_ARRAY);
        if (modules == null)
            return;
        final Project project = e.getProject();
        if (project == null)
            return;

        Arrays.sort(modules, Comparators.MODULE_COMPARATOR);

        final AnalyzeStrategy strategy = decideStrategy(project, modules);
        if (strategy == null)
            return;

        ProgressManager.getInstance()
                .run(new Task.Backgroundable(e.getProject(),
                        "Dependencies of [" + StringUtil.join(modules, TO_STRING, ", ") + "]", true,
                        BackgroundFromStartOption.getInstance()) {

                    public void run(@NotNull final ProgressIndicator indicator) {
                        final ModulesDependencies toRemove = processModulesDependencies(strategy, indicator,
                                modules, myProject);
                        PostAction.completeProcess(project, toRemove);
                    }
                });
    }

    @Nullable
    private AnalyzeStrategy decideStrategy(@NotNull final Project project, @NotNull final Module[] modules) {
        boolean suggestExpand = modules.length != WITH_EXPORT_DEPENDENCIES.collectAllModules(project,
                modules).length;
        if (!suggestExpand)
            return WITH_EXPORT_DEPENDENCIES;

        int result = Messages.showYesNoCancelDialog(project,
                "There are some modules with exported dependencies.\n"
                        + "It's required to include more modules to the list "
                        + "in order to analyze exported dependencies.\n\n" + "Do you like to include more modules?",
                "Unused Dependencies", "Include Modules", "Skip Exports", "Cancel", null);
        switch (result) {
        case Messages.YES:
            return WITH_EXPORT_DEPENDENCIES;
        case Messages.NO:
            return SKIP_EXPORT_DEPENDENCIES;
        default:
            return null;
        }
    }

    private static final Function<Module, String> TO_STRING = new Function<Module, String>() {
        public String fun(Module module) {
            return module.getName();
        }
    };
}