org.dylanfoundry.deft.library.LibraryAttachHandler.java Source code

Introduction

Here is the source code for org.dylanfoundry.deft.library.LibraryAttachHandler.java

Source

/*
 * Copyright 2013, Bruce Mitchener, Jr.
 *
 * 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.dylanfoundry.deft.library;

import com.intellij.CommonBundle;
import com.intellij.openapi.application.AccessToken;
import com.intellij.openapi.application.WriteAction;
import com.intellij.openapi.project.Project;
import com.intellij.openapi.roots.OrderRootType;
import com.intellij.openapi.roots.libraries.NewLibraryConfiguration;
import com.intellij.openapi.roots.libraries.ui.OrderRoot;
import com.intellij.openapi.roots.ui.configuration.libraryEditor.LibraryEditor;
import com.intellij.openapi.ui.DialogWrapper;
import com.intellij.openapi.ui.Messages;
import com.intellij.openapi.util.Ref;
import com.intellij.openapi.vfs.VfsUtil;
import com.intellij.openapi.vfs.VirtualFile;
import com.intellij.openapi.vfs.VirtualFileManager;
import org.dylanfoundry.deft.registry.DeftRegistryEntryInfo;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import javax.swing.*;
import java.io.File;
import java.util.ArrayList;
import java.util.List;

public class LibraryAttachHandler {
    @Nullable
    public static NewLibraryConfiguration chooseLibrary(final @NotNull Project project,
            JComponent parentComponent) {
        LibraryAttachDialog dialog = new LibraryAttachDialog(project);
        dialog.setTitle("Attach Library");
        dialog.show();
        if (dialog.getExitCode() != DialogWrapper.OK_EXIT_CODE) {
            return null;
        }

        NewLibraryConfiguration configuration = attachLibrary(project, dialog.getRegistryEntries());
        if (configuration == null) {
            Messages.showErrorDialog(parentComponent, "No libraries attached", CommonBundle.getErrorTitle());
        }
        return configuration;
    }

    @Nullable
    public static NewLibraryConfiguration attachLibrary(final @NotNull Project project,
            List<DeftRegistryEntryInfo> registryEntries) {
        final Ref<NewLibraryConfiguration> result = Ref.create(null);

        for (DeftRegistryEntryInfo registryEntry : registryEntries) {
            AccessToken accessToken = WriteAction.start();
            try {
                final List<OrderRoot> roots = createRoots(registryEntry);

                result.set(new NewLibraryConfiguration(registryEntry.getLibraryName(),
                        DeftLibraryType.getInstance(), new DeftLibraryProperties(registryEntry.getLibraryName())) {
                    @Override
                    public void addRoots(@NotNull LibraryEditor editor) {
                        editor.addRoots(roots);
                    }
                });
            } finally {
                accessToken.finish();
            }
        }

        return result.get();
    }

    public static List<OrderRoot> createRoots(@NotNull DeftRegistryEntryInfo registryInfo) {
        final List<OrderRoot> result = new ArrayList<OrderRoot>();
        final VirtualFileManager manager = VirtualFileManager.getInstance();

        File repoFolder = new File(registryInfo.getLocation());
        final String url = VfsUtil.getUrlForLibraryRoot(repoFolder);
        final VirtualFile file = manager.refreshAndFindFileByUrl(url);
        if (file != null) {
            OrderRootType rootType = OrderRootType.SOURCES;
            result.add(new OrderRoot(file, rootType));
        }

        return result;
    }

}