source.dussan.org Git - archiva.git/blob

   1 package org.apache.archiva.rest.services.utils;
   2 /*
   3  * Licensed to the Apache Software Foundation (ASF) under one
   4  * or more contributor license agreements.  See the NOTICE file
   5  * distributed with this work for additional information
   6  * regarding copyright ownership.  The ASF licenses this file
   7  * to you under the Apache License, Version 2.0 (the
   8  * "License"); you may not use this file except in compliance
   9  * with the License.  You may obtain a copy of the License at
  10  *
  11  *   http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing,
  14  * software distributed under the License is distributed on an
  15  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  16  * KIND, either express or implied.  See the License for the
  17  * specific language governing permissions and limitations
  18  * under the License.
  19  */
  20
  21 import net.sf.beanlib.provider.replicator.BeanReplicator;
  22 import org.apache.archiva.rest.api.model.Artifact;
  23 import org.apache.archiva.rest.api.model.TreeEntry;
  24 import org.apache.maven.shared.dependency.tree.DependencyNode;
  25 import org.apache.maven.shared.dependency.tree.traversal.DependencyNodeVisitor;
  26
  27 import java.util.List;
  28
  29 /**
  30  * @author Olivier Lamy
  31  * @since 1.4-M3
  32  */
  33 public class TreeDependencyNodeVisitor
  34     implements DependencyNodeVisitor
  35 {
  36
  37     final List<TreeEntry> treeEntries;
  38
  39     private TreeEntry currentEntry;
  40
  41     private DependencyNode firstNode;
  42
  43     public TreeDependencyNodeVisitor( List<TreeEntry> treeEntries )
  44     {
  45         this.treeEntries = treeEntries;
  46     }
  47
  48     public boolean visit( DependencyNode node )
  49     {
  50         TreeEntry entry = new TreeEntry( new BeanReplicator().replicateBean( node.getArtifact(), Artifact.class ) );
  51         entry.setParent( currentEntry );
  52         currentEntry = entry;
  53
  54         if ( firstNode == null )
  55         {
  56             firstNode = node;
  57             treeEntries.add( currentEntry );
  58         }
  59         else
  60         {
  61             currentEntry.getParent().getChilds().add( currentEntry );
  62         }
  63         return true;
  64     }
  65
  66     public boolean endVisit( DependencyNode node )
  67     {
  68         currentEntry = currentEntry.getParent();
  69         return true;
  70     }
  71
  72 }