java/idea-ui/src/com/intellij/facet/impl/ui/libraries/LibraryCompositionSettings.java

   1 /*
   2  * Copyright 2000-2009 JetBrains s.r.o.
   3  *
   4  * Licensed under the Apache License, Version 2.0 (the "License");
   5  * you may not use this file except in compliance with the License.
   6  * You may obtain a copy of the License at
   7  *
   8  * http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  * Unless required by applicable law or agreed to in writing, software
  11  * distributed under the License is distributed on an "AS IS" BASIS,
  12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  * See the License for the specific language governing permissions and
  14  * limitations under the License.
  15  */
  16 package com.intellij.facet.impl.ui.libraries;
  17
  18 import com.intellij.facet.ui.libraries.LibraryDownloadInfo;
  19 import com.intellij.facet.ui.libraries.LibraryInfo;
  20 import com.intellij.openapi.roots.ModifiableRootModel;
  21 import com.intellij.openapi.roots.OrderRootType;
  22 import com.intellij.openapi.roots.libraries.Library;
  23 import com.intellij.openapi.roots.ui.configuration.projectRoot.LibrariesContainer;
  24 import com.intellij.openapi.roots.ui.configuration.projectRoot.LibrariesContainerFactory;
  25 import com.intellij.openapi.vfs.VirtualFile;
  26 import org.jetbrains.annotations.NonNls;
  27 import org.jetbrains.annotations.NotNull;
  28 import org.jetbrains.annotations.Nullable;
  29
  30 import javax.swing.*;
  31 import java.util.*;
  32
  33 /**
  34  * @author nik
  35 */
  36 public class LibraryCompositionSettings {
  37   @NonNls private static final String DEFAULT_LIB_FOLDER = "lib";
  38   private final LibraryInfo[] myLibraryInfos;
  39   private final String myBaseDirectoryForDownloadedFiles;
  40   private final String myTitle;
  41   private String myDirectoryForDownloadedLibrariesPath;
  42   private final Set<VirtualFile> myAddedJars = new LinkedHashSet<VirtualFile>();
  43   private boolean myDownloadLibraries = true;
  44   private final Set<Library> myUsedLibraries = new LinkedHashSet<Library>();
  45   private LibrariesContainer.LibraryLevel myLibraryLevel = LibrariesContainer.LibraryLevel.PROJECT;
  46   private String myLibraryName;
  47   private final Icon myIcon;
  48
  49   public LibraryCompositionSettings(final @NotNull LibraryInfo[] libraryInfos,
  50                                     final @NotNull String defaultLibraryName,
  51                                     final @NotNull String baseDirectoryForDownloadedFiles,
  52                                     final String title, @Nullable Icon icon) {
  53     myLibraryInfos = libraryInfos;
  54     myBaseDirectoryForDownloadedFiles = baseDirectoryForDownloadedFiles;
  55     myTitle = title;
  56     myLibraryName = defaultLibraryName;
  57     myIcon = icon;
  58   }
  59
  60   @NotNull
  61   public LibraryInfo[] getLibraryInfos() {
  62     return myLibraryInfos;
  63   }
  64
  65   @NotNull
  66   public String getBaseDirectoryForDownloadedFiles() {
  67     return myBaseDirectoryForDownloadedFiles;
  68   }
  69
  70   public void setDirectoryForDownloadedLibrariesPath(final String directoryForDownloadedLibrariesPath) {
  71     myDirectoryForDownloadedLibrariesPath = directoryForDownloadedLibrariesPath;
  72   }
  73
  74   public Set<VirtualFile> getAddedJars() {
  75     return myAddedJars;
  76   }
  77
  78   public void setAddedJars(final Collection<VirtualFile> addedJars) {
  79     myAddedJars.clear();
  80     myAddedJars.addAll(addedJars);
  81   }
  82
  83   public boolean isDownloadLibraries() {
  84     return myDownloadLibraries;
  85   }
  86
  87   public void setDownloadLibraries(final boolean downloadLibraries) {
  88     myDownloadLibraries = downloadLibraries;
  89   }
  90
  91   public void setUsedLibraries(Collection<Library> addedLibraries) {
  92     myUsedLibraries.clear();
  93     myUsedLibraries.addAll(addedLibraries);
  94   }
  95
  96   public void setLibraryLevel(final LibrariesContainer.LibraryLevel libraryLevel) {
  97     myLibraryLevel = libraryLevel;
  98   }
  99
 100   public void setLibraryName(final String libraryName) {
 101     myLibraryName = libraryName;
 102   }
 103
 104   public String getDirectoryForDownloadedLibrariesPath() {
 105     if (myDirectoryForDownloadedLibrariesPath == null) {
 106       myDirectoryForDownloadedLibrariesPath = myBaseDirectoryForDownloadedFiles + "/" + DEFAULT_LIB_FOLDER;
 107     }
 108     return myDirectoryForDownloadedLibrariesPath;
 109   }
 110
 111   public String getTitle() {
 112     return myTitle;
 113   }
 114
 115   public boolean downloadFiles(final @NotNull LibraryDownloadingMirrorsMap mirrorsMap, @NotNull LibrariesContainer librariesContainer, final @NotNull JComponent parent) {
 116     if (myDownloadLibraries) {
 117       RequiredLibrariesInfo requiredLibraries = new RequiredLibrariesInfo(getLibraryInfos());
 118
 119       List<VirtualFile> roots = new ArrayList<VirtualFile>();
 120       roots.addAll(myAddedJars);
 121       for (Library library : myUsedLibraries) {
 122         roots.addAll(Arrays.asList(librariesContainer.getLibraryFiles(library, OrderRootType.CLASSES)));
 123       }
 124       VirtualFile[] jars = roots.toArray(new VirtualFile[roots.size()]);
 125       RequiredLibrariesInfo.RequiredClassesNotFoundInfo info = requiredLibraries.checkLibraries(jars);
 126       if (info != null) {
 127         LibraryDownloadInfo[] downloadingInfos = LibraryDownloader.getDownloadingInfos(info.getLibraryInfos());
 128         if (downloadingInfos.length > 0) {
 129           LibraryDownloader downloader = new LibraryDownloader(downloadingInfos, null, parent,
 130                                                                getDirectoryForDownloadedLibrariesPath(), myLibraryName,
 131                                                                mirrorsMap);
 132           VirtualFile[] files = downloader.download();
 133           if (files.length != downloadingInfos.length) {
 134             return false;
 135           }
 136           myAddedJars.addAll(Arrays.asList(files));
 137         }
 138       }
 139     }
 140     return true;
 141   }
 142
 143   @Nullable
 144   private Library createLibrary(final ModifiableRootModel rootModel, @Nullable LibrariesContainer additionalContainer) {
 145     if (!myAddedJars.isEmpty()) {
 146       VirtualFile[] roots = myAddedJars.toArray(new VirtualFile[myAddedJars.size()]);
 147       return LibrariesContainerFactory.createLibrary(additionalContainer, LibrariesContainerFactory.createContainer(rootModel),
 148                                                      myLibraryName, myLibraryLevel, roots, VirtualFile.EMPTY_ARRAY);
 149     }
 150     return null;
 151   }
 152
 153   public LibrariesContainer.LibraryLevel getLibraryLevel() {
 154     return myLibraryLevel;
 155   }
 156
 157   public String getLibraryName() {
 158     return myLibraryName;
 159   }
 160
 161   public Collection<Library> getUsedLibraries() {
 162     return Collections.unmodifiableCollection(myUsedLibraries);
 163   }
 164
 165   public Icon getIcon() {
 166     return myIcon;
 167   }
 168
 169   @Nullable
 170   public Library addLibraries(final ModifiableRootModel rootModel, final List<Library> addedLibraries) {
 171     return addLibraries(rootModel, addedLibraries, null);
 172   }
 173
 174   @Nullable
 175   public Library addLibraries(final @NotNull ModifiableRootModel rootModel, final @NotNull List<Library> addedLibraries,
 176                               final @Nullable LibrariesContainer librariesContainer) {
 177     Library library = createLibrary(rootModel, librariesContainer);
 178
 179     if (library != null) {
 180       addedLibraries.add(library);
 181       if (getLibraryLevel() != LibrariesContainer.LibraryLevel.MODULE) {
 182         rootModel.addLibraryEntry(library);
 183       }
 184     }
 185     for (Library usedLibrary : getUsedLibraries()) {
 186       addedLibraries.add(usedLibrary);
 187       rootModel.addLibraryEntry(usedLibrary);
 188     }
 189     return library;
 190   }
 191 }