LibraryManager: better type filtering

Author: Federico Fissore

app/src/cc/arduino/contributions/libraries/filters/InstalledLibraryPredicate.java

@@ -0,0 +1,39 @@
+package cc.arduino.contributions.libraries.filters;
+
+import cc.arduino.contributions.libraries.ContributedLibrary;
+import cc.arduino.contributions.libraries.LibrariesIndex;
+import com.google.common.base.Predicate;
+import com.google.common.collect.Collections2;
+
+import java.util.Collection;
+
+public class InstalledLibraryPredicate implements Predicate<ContributedLibrary> {
+
+  private final LibrariesIndex index;
+
+  public InstalledLibraryPredicate(LibrariesIndex index) {
+    this.index = index;
+  }
+
+  @Override
+  public boolean apply(ContributedLibrary input) {
+    if (input.isInstalled()) {
+      return true;
+    }
+
+    Collection<ContributedLibrary> installed = Collections2.filter(index.find(input.getName()), new Predicate<ContributedLibrary>() {
+      @Override
+      public boolean apply(ContributedLibrary input) {
+        return input.isInstalled();
+      }
+    });
+
+    return !installed.isEmpty();
+  }
+
+  @Override
+  public boolean equals(Object obj) {
+    return obj instanceof InstalledLibraryPredicate;
+  }
+
+}

app/src/cc/arduino/contributions/libraries/ui/ContributedLibraryTableCell.java

@@ -33,6 +33,7 @@
 import cc.arduino.contributions.libraries.ContributedLibrary;
 import cc.arduino.contributions.libraries.ContributedLibraryComparator;
 import cc.arduino.contributions.libraries.filters.BuiltInPredicate;
+import cc.arduino.contributions.libraries.filters.InstalledLibraryPredicate;
 import cc.arduino.contributions.libraries.filters.OnlyUpstreamReleasePredicate;
 import cc.arduino.contributions.ui.InstallerTableCell;
 import cc.arduino.contributions.ui.listeners.DelegatingKeyListener;
@@ -65,6 +66,7 @@
 @SuppressWarnings("serial")
 public class ContributedLibraryTableCell extends InstallerTableCell {
 
+  private final LibraryManagerUI indexer;
   private JPanel panel;
   private JButton installButton;
   private Component installButtonPlaceholder;
@@ -75,7 +77,9 @@ public class ContributedLibraryTableCell extends InstallerTableCell {
   private JPanel inactiveButtonsPanel;
   private JLabel statusLabel;
 
-  public ContributedLibraryTableCell() {
+  public ContributedLibraryTableCell(LibraryManagerUI indexer) {
+    this.indexer = indexer;
+
     {
       installButton = new JButton(_("Install"));
       installButton.addActionListener(new ActionListener() {

app/src/cc/arduino/contributions/libraries/ui/DropdownInstalledLibraryItem.java

@@ -0,0 +1,27 @@
+package cc.arduino.contributions.libraries.ui;
+
+import cc.arduino.contributions.libraries.ContributedLibrary;
+import cc.arduino.contributions.libraries.LibrariesIndex;
+import cc.arduino.contributions.libraries.filters.InstalledLibraryPredicate;
+import cc.arduino.contributions.ui.DropdownItem;
+import com.google.common.base.Predicate;
+
+import static processing.app.I18n._;
+
+public class DropdownInstalledLibraryItem implements DropdownItem<ContributedLibrary> {
+
+  private final LibrariesIndex index;
+
+  public DropdownInstalledLibraryItem(LibrariesIndex index) {
+    this.index = index;
+  }
+
+  public String toString() {
+    return _("Installed");
+  }
+
+  @Override
+  public Predicate<ContributedLibrary> getFilterPredicate() {
+    return new InstalledLibraryPredicate(index);
+  }
+}

app/src/cc/arduino/contributions/libraries/ui/LibraryManagerUI.java

@@ -39,7 +39,6 @@
 import java.awt.*;
 import java.awt.event.ActionEvent;
 import java.awt.event.ActionListener;
-import java.util.Arrays;
 import java.util.Collection;
 
 import static processing.app.I18n._;
@@ -62,12 +61,12 @@ private LibrariesIndexTableModel getContribModel() {
 
   @Override
   protected InstallerTableCell createCellRenderer() {
-    return new ContributedLibraryTableCell();
+    return new ContributedLibraryTableCell(this);
   }
 
   @Override
   protected InstallerTableCell createCellEditor() {
-    return new ContributedLibraryTableCell() {
+    return new ContributedLibraryTableCell(this) {
       @Override
       protected void onInstall(ContributedLibrary selectedLibrary, ContributedLibrary installedLibrary) {
         if (selectedLibrary.isReadOnly()) {
@@ -150,7 +149,7 @@ public void setIndexer(LibrariesIndexer indexer) {
     typeFilter = null;
     typeChooser.removeAllItems();
     typeChooser.addItem(new DropdownAllItem());
-    typeChooser.addItem(new DropdownInstalledContributionItem());
+    typeChooser.addItem(new DropdownInstalledLibraryItem(indexer.getIndex()));
     Collection<String> types = indexer.getIndex().getTypes();
     for (String type : types) {
       typeChooser.addItem(new DropdownLibraryOfTypeItem(type));
@@ -169,6 +168,10 @@ public void onProgress(Progress progress) {
     };
   }
 
+  public LibrariesIndexer getIndexer() {
+    return indexer;
+  }
+
   public void setProgress(Progress progress) {
     progressBar.setValue(progress);
   }

app/src/cc/arduino/contributions/ui/DropdownInstalledContributionItem.java

@@ -26,18 +26,32 @@
  * invalidate any other reasons why the executable file might be covered by
  * the GNU General Public License.
  */
+
 package cc.arduino.contributions.libraries;
 
+import com.google.common.base.Predicate;
+import com.google.common.collect.Collections2;
+
 import java.util.*;
 
 public abstract class LibrariesIndex {
 
   public abstract List<ContributedLibrary> getLibraries();
 
+  public List<ContributedLibrary> find(final String name) {
+    return new LinkedList<ContributedLibrary>(Collections2.filter(getLibraries(), new Predicate<ContributedLibrary>() {
+      @Override
+      public boolean apply(ContributedLibrary contributedLibrary) {
+        return name.equals(contributedLibrary.getName());
+      }
+    }));
+  }
+
   public ContributedLibrary find(String name, String version) {
-    for (ContributedLibrary lib : getLibraries()) {
-      if (lib.getName().equals(name) && lib.getVersion().equals(version))
+    for (ContributedLibrary lib : find(name)) {
+      if (lib.getVersion().equals(version)) {
         return lib;
+      }
     }
     return null;
   }

arduino-core/src/cc/arduino/contributions/libraries/LibrariesIndex.java

@@ -28,14 +28,9 @@
  */
 package cc.arduino.contributions.libraries;
 
-import static processing.app.I18n._;
-
-import java.io.File;
-import java.io.FileInputStream;
-import java.io.IOException;
-import java.io.InputStream;
-import java.util.List;
-
+import com.fasterxml.jackson.databind.DeserializationFeature;
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.fasterxml.jackson.module.mrbean.MrBeanModule;
 import processing.app.BaseNoGui;
 import processing.app.I18n;
 import processing.app.helpers.FileUtils;
@@ -44,9 +39,14 @@
 import processing.app.packages.LibraryList;
 import processing.app.packages.UserLibrary;
 
-import com.fasterxml.jackson.databind.DeserializationFeature;
-import com.fasterxml.jackson.databind.ObjectMapper;
-import com.fasterxml.jackson.module.mrbean.MrBeanModule;
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.IOException;
+import java.io.InputStream;
+import java.util.Arrays;
+import java.util.List;
+
+import static processing.app.I18n._;
 
 public class LibrariesIndexer {
 
@@ -60,7 +60,7 @@ public class LibrariesIndexer {
   public LibrariesIndexer(File preferencesFolder) {
     indexFile = new File(preferencesFolder, "library_index.json");
     stagingFolder = new File(preferencesFolder, "staging" + File.separator +
-        "libraries");
+            "libraries");
   }
 
   public void parseIndex() throws IOException {
@@ -109,31 +109,28 @@ private void scanInstalledLibraries(File folder) {
     for (File subfolder : list) {
       if (!BaseNoGui.isSanitaryName(subfolder.getName())) {
         String mess = I18n.format(_("The library \"{0}\" cannot be used.\n"
-            + "Library names must contain only basic letters and numbers.\n"
-            + "(ASCII only and no spaces, and it cannot start with a number)"),
-                                  subfolder.getName());
+                        + "Library names must contain only basic letters and numbers.\n"
+                        + "(ASCII only and no spaces, and it cannot start with a number)"),
+                subfolder.getName());
         BaseNoGui.showMessage(_("Ignoring bad library name"), mess);
         continue;
       }
 
       try {
         scanLibrary(subfolder);
       } catch (IOException e) {
-        System.out.println(I18n.format(_("Invalid library found in {0}: {1}"),
-                                       subfolder, e.getMessage()));
+        System.out.println(I18n.format(_("Invalid library found in {0}: {1}"), subfolder, e.getMessage()));
       }
     }
   }
 
   private void scanLibrary(File folder) throws IOException {
-    boolean readOnly = !FileUtils
-        .isSubDirectory(sketchbookLibrariesFolder, folder);
+    boolean readOnly = !FileUtils.isSubDirectory(sketchbookLibrariesFolder, folder);
 
     // A library is considered "legacy" if it doesn't contains
     // a file called "library.properties"
     File check = new File(folder, "library.properties");
     if (!check.exists() || !check.isFile()) {
-
       // Create a legacy library and exit
       LegacyUserLibrary lib = LegacyUserLibrary.create(folder);
       lib.setReadOnly(readOnly);
@@ -148,12 +145,20 @@ private void scanLibrary(File folder) throws IOException {
 
     // Check if we can find the same library in the index
     // and mark it as installed
-    String libName = folder.getName(); // XXX: lib.getName()?
-    ContributedLibrary foundLib = index.find(libName, lib.getVersion());
+    ContributedLibrary foundLib = index.find(lib.getName(), lib.getVersion());
     if (foundLib != null) {
       foundLib.setInstalled(true);
       foundLib.setInstalledFolder(folder);
       foundLib.setReadOnly(readOnly);
+      lib.setTypes(foundLib.getTypes());
+    }
+
+    if (lib.isReadOnly() && lib.getTypes() == null && !lib.getDeclaredTypes().isEmpty()) {
+      lib.setTypes(lib.getDeclaredTypes());
+    }
+
+    if (lib.getTypes() == null) {
+      lib.setTypes(Arrays.asList("Contributed"));
     }
   }