commit c8da88fabfe011d4e020d52fab602af2dc346971
author Aaron Digulla <digulla@hepe.com> Thu Apr 26 15:55:13 2012 +0200
committer Aaron Digulla <digulla@hepe.com> Thu Apr 26 15:55:13 2012 +0200
tree a730815ff44bc0d25fc3fcd4afdaac548ba3c85f
parent 13e9ae751c2812319a67fd260b98a35e926c38e1

Keep reference to source bundle instead of flag

src/main/groovy/m4e/p2/ui/RepoTreeModel.groovy

diff --git a/src/main/groovy/m4e/p2/ui/RepoTreeModel.groovy b/src/main/groovy/m4e/p2/ui/RepoTreeModel.groovy
index 5b7e626..3406e75 100644
--- a/src/main/groovy/m4e/p2/ui/RepoTreeModel.groovy
+++ b/src/main/groovy/m4e/p2/ui/RepoTreeModel.groovy
@@ -248,7 +248,7 @@
 class SwingBundle extends LazyNode {
     RepoTreeModel model
     P2Bundle bundle
-    boolean hasSource
+    P2Bundle source
     
     SwingBundle( RepoTreeModel model, P2Bundle bundle ) {
         super( bundle.dependencies.findAll { it.id != bundle.id && model.matches( "${it.id}" ) } )
@@ -264,7 +264,7 @@
     
     @Override
     String toString() {
-        String source = hasSource ? ' (+Source)' : ''
+        String source = source ? ' (+Source)' : ''
         
         if( bundle.name ) {
             if( bundle.name == bundle.id ) {
@@ -347,11 +347,16 @@
     
     @Override
     public List createSwingChildren () {
-        Set<String> sourceBundles = new HashSet( bundles.findAll { it.isSourceBundle() }.collect { it.id.removeEnd( '.source' ) } )
+        Map<String, P2Bundle> sourceBundles = [:]
+        bundles.findAll {
+            it.isSourceBundle()
+        }.each {
+            sourceBundles[ it.id.removeEnd( '.source' ) ] = it
+        }
         
         return children.collect {
             def swing = new SwingBundle( model, it )
-            swing.hasSource = sourceBundles.contains( it.id )
+            swing.source = sourceBundles[ it.id ]
             return swing
         }.sort { it.toString() }
     }