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 org.apache.archiva.metadata.model.ArtifactMetadata;
  22 import org.apache.archiva.metadata.model.maven2.MavenArtifactFacet;
  23 import org.apache.archiva.model.ArtifactReference;
  24 import org.apache.archiva.repository.ManagedRepositoryContent;
  25 import org.apache.archiva.maven2.model.Artifact;
  26 import org.apache.commons.io.FilenameUtils;
  27
  28 import java.io.File;
  29 import java.text.DecimalFormat;
  30 import java.text.DecimalFormatSymbols;
  31 import java.util.Locale;
  32
  33 /**
  34  * @author Olivier Lamy
  35  * @since 1.4-M3
  36  */
  37 public class ArtifactBuilder
  38 {
  39
  40     private ManagedRepositoryContent managedRepositoryContent;
  41
  42     private ArtifactMetadata artifactMetadata;
  43
  44     public ArtifactBuilder()
  45     {
  46         // no op
  47     }
  48
  49
  50     public ArtifactBuilder withManagedRepositoryContent( ManagedRepositoryContent managedRepositoryContent )
  51     {
  52         this.managedRepositoryContent = managedRepositoryContent;
  53         return this;
  54     }
  55
  56     public ArtifactBuilder forArtifactMetadata( ArtifactMetadata artifactMetadata )
  57     {
  58         this.artifactMetadata = artifactMetadata;
  59         return this;
  60     }
  61
  62     /**
  63      *
  64      * @param filename
  65      * @return
  66      */
  67     private static String specialExtensions( String filename )
  68     {
  69         String[] special = {
  70             "tar.gz"
  71         };
  72         for ( String extension : special )
  73         {
  74             if ( filename.endsWith( extension ) )
  75             {
  76                 return extension;
  77             }
  78         }
  79         return null;
  80     }
  81     public Artifact build()
  82     {
  83         ArtifactReference ref = new ArtifactReference();
  84         ref.setArtifactId( artifactMetadata.getProject() );
  85         ref.setGroupId( artifactMetadata.getNamespace() );
  86         ref.setVersion( artifactMetadata.getVersion() );
  87
  88         String type = null, classifier = null;
  89
  90         MavenArtifactFacet facet = (MavenArtifactFacet) artifactMetadata.getFacet( MavenArtifactFacet.FACET_ID );
  91         if ( facet != null )
  92         {
  93             type = facet.getType();
  94             classifier = facet.getClassifier();
  95         }
  96
  97         ref.setClassifier( classifier );
  98         ref.setType( type );
  99         File file = managedRepositoryContent.toFile( ref );
 100
 101         String extension = FilenameUtils.getExtension( file.getName() );
 102         // handle more than one point extension we know.
 103         if ( specialExtensions( file.getName() ) != null )
 104         {
 105             extension = specialExtensions( file.getName() );
 106         }
 107
 108         Artifact artifact = new Artifact( ref.getGroupId(), ref.getArtifactId(), ref.getVersion() );
 109         artifact.setRepositoryId( artifactMetadata.getRepositoryId() );
 110         artifact.setClassifier( classifier );
 111         artifact.setPackaging( type );
 112         artifact.setType( type );
 113         artifact.setFileExtension( extension );
 114         artifact.setPath( managedRepositoryContent.toPath( ref ) );
 115         // TODO: find a reusable formatter for this
 116         double s = this.artifactMetadata.getSize();
 117         String symbol = "b";
 118         if ( s > 1024 )
 119         {
 120             symbol = "K";
 121             s /= 1024;
 122
 123             if ( s > 1024 )
 124             {
 125                 symbol = "M";
 126                 s /= 1024;
 127
 128                 if ( s > 1024 )
 129                 {
 130                     symbol = "G";
 131                     s /= 1024;
 132                 }
 133             }
 134         }
 135         artifact.setContext( managedRepositoryContent.getId() );
 136         DecimalFormat df = new DecimalFormat( "#,###.##", new DecimalFormatSymbols( Locale.US ) );
 137         artifact.setSize( df.format( s ) + " " + symbol );
 138
 139         artifact.setId( ref.getArtifactId() + "-" + ref.getVersion() + "." + ref.getType() );
 140
 141         return artifact;
 142
 143     }
 144
 145
 146 }