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.maven2.model.Artifact;
  22 import org.apache.archiva.metadata.model.ArtifactMetadata;
  23 import org.apache.archiva.metadata.maven.model.MavenArtifactFacet;
  24 import org.apache.archiva.model.ArtifactReference;
  25 import org.apache.archiva.repository.BaseRepositoryContentLayout;
  26 import org.apache.archiva.repository.ManagedRepositoryContent;
  27 import org.apache.archiva.repository.LayoutException;
  28 import org.apache.archiva.repository.storage.StorageAsset;
  29 import org.apache.archiva.repository.storage.util.StorageUtil;
  30
  31 import java.text.DecimalFormat;
  32 import java.text.DecimalFormatSymbols;
  33 import java.util.Locale;
  34 import java.util.regex.Matcher;
  35 import java.util.regex.Pattern;
  36
  37 /**
  38  * @author Olivier Lamy
  39  * @since 1.4-M3
  40  */
  41 public class ArtifactBuilder
  42 {
  43
  44     private ManagedRepositoryContent managedRepositoryContent;
  45
  46     private ArtifactMetadata artifactMetadata;
  47
  48     public ArtifactBuilder()
  49     {
  50         // no op
  51     }
  52
  53
  54     public ArtifactBuilder withManagedRepositoryContent( ManagedRepositoryContent managedRepositoryContent )
  55     {
  56         this.managedRepositoryContent = managedRepositoryContent;
  57         return this;
  58     }
  59
  60     public ArtifactBuilder forArtifactMetadata( ArtifactMetadata artifactMetadata )
  61     {
  62         this.artifactMetadata = artifactMetadata;
  63         return this;
  64     }
  65
  66     public Artifact build()
  67     {
  68         ArtifactReference ref = new ArtifactReference();
  69         ref.setArtifactId( artifactMetadata.getProject() );
  70         ref.setGroupId( artifactMetadata.getNamespace() );
  71         ref.setVersion( artifactMetadata.getVersion() );
  72
  73         String type = null, classifier = null;
  74
  75         MavenArtifactFacet facet = (MavenArtifactFacet) artifactMetadata.getFacet( MavenArtifactFacet.FACET_ID );
  76         if ( facet != null )
  77         {
  78             type = facet.getType();
  79             classifier = facet.getClassifier();
  80         }
  81
  82         ref.setClassifier( classifier );
  83         ref.setType( type );
  84         BaseRepositoryContentLayout layout;
  85         try
  86         {
  87              layout = managedRepositoryContent.getLayout( BaseRepositoryContentLayout.class );
  88         }
  89         catch ( LayoutException e )
  90         {
  91             throw new RuntimeException( "Could not convert to layout " + e.getMessage( ) );
  92         }
  93         StorageAsset file = layout.toFile( ref );
  94
  95         String extension = getExtensionFromFile(file);
  96
  97         Artifact artifact = new Artifact( ref.getGroupId(), ref.getArtifactId(), ref.getVersion() );
  98         artifact.setRepositoryId( artifactMetadata.getRepositoryId() );
  99         artifact.setClassifier( classifier );
 100         artifact.setPackaging( type );
 101         artifact.setType( type );
 102         artifact.setFileExtension( extension );
 103         artifact.setPath( layout.toPath( ref ) );
 104         // TODO: find a reusable formatter for this
 105         double s = this.artifactMetadata.getSize();
 106         String symbol = "b";
 107         if ( s > 1024 )
 108         {
 109             symbol = "K";
 110             s /= 1024;
 111
 112             if ( s > 1024 )
 113             {
 114                 symbol = "M";
 115                 s /= 1024;
 116
 117                 if ( s > 1024 )
 118                 {
 119                     symbol = "G";
 120                     s /= 1024;
 121                 }
 122             }
 123         }
 124         artifact.setContext( managedRepositoryContent.getId() );
 125         DecimalFormat df = new DecimalFormat( "#,###.##", new DecimalFormatSymbols( Locale.US ) );
 126         artifact.setSize( df.format( s ) + " " + symbol );
 127
 128         artifact.setId( ref.getArtifactId() + "-" + ref.getVersion() + "." + ref.getType() );
 129
 130         return artifact;
 131
 132     }
 133
 134
 135     /**
 136      * Extract file extension
 137      */
 138     String getExtensionFromFile( StorageAsset file )
 139     {
 140         // we are just interested in the section after the last -
 141         String[] parts = file.getName().split( "-" );
 142         if ( parts.length > 0 )
 143         {
 144             // get anything after a dot followed by a letter a-z, including other dots
 145             Pattern p = Pattern.compile( "\\.([a-z]+[a-z0-9\\.]*)" );
 146             Matcher m = p.matcher( parts[parts.length - 1] );
 147             if ( m.find() )
 148             {
 149                 return m.group( 1 );
 150             }
 151         }
 152         // just in case
 153         return StorageUtil.getExtension( file );
 154     }
 155
 156 }