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

   1 package org.apache.archiva.rss.processor;
   2
   3 /*
   4  * Licensed to the Apache Software Foundation (ASF) under one
   5  * or more contributor license agreements.  See the NOTICE file
   6  * distributed with this work for additional information
   7  * regarding copyright ownership.  The ASF licenses this file
   8  * to you under the Apache License, Version 2.0 (the
   9  * "License"); you may not use this file except in compliance
  10  * with the License.  You may obtain a copy of the License at
  11  *
  12  *  http://www.apache.org/licenses/LICENSE-2.0
  13  *
  14  * Unless required by applicable law or agreed to in writing,
  15  * software distributed under the License is distributed on an
  16  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
  17  * KIND, either express or implied.  See the License for the
  18  * specific language governing permissions and limitations
  19  * under the License.
  20  */
  21
  22 import java.util.ArrayList;
  23 import java.util.Calendar;
  24 import java.util.Collections;
  25 import java.util.Enumeration;
  26 import java.util.HashMap;
  27 import java.util.List;
  28 import java.util.Map;
  29 import java.util.Properties;
  30
  31 import org.apache.archiva.rss.RssFeedEntry;
  32 import org.apache.archiva.rss.RssFeedGenerator;
  33 import org.apache.commons.lang.StringUtils;
  34 import org.apache.maven.archiva.model.ArchivaArtifact;
  35 import org.slf4j.Logger;
  36 import org.slf4j.LoggerFactory;
  37
  38 /**
  39  * Process new artifacts in the repository and generate RSS feeds.
  40  *
  41  * @author <a href="mailto:oching@apache.org">Maria Odea Ching</a>
  42  * @version
  43  * @plexus.component role="org.apache.archiva.rss.processor.RssFeedProcessor" role-hint="new-artifacts"
  44  */
  45 public class NewArtifactsRssFeedProcessor
  46     implements RssFeedProcessor
  47 {
  48     public static final String NEW_ARTIFACTS_IN_REPO = "New Artifacts in Repository ";
  49
  50     public static final String NEW_VERSIONS_OF_ARTIFACT = "New Versions of Artifact ";
  51
  52     /**
  53      * @plexus.requirement
  54      */
  55     private RssFeedGenerator generator;
  56
  57     private Logger log = LoggerFactory.getLogger( NewArtifactsRssFeedProcessor.class );
  58
  59     /**
  60      * The hostname that will be used in the urls for the feed links.
  61      */
  62     private String host = "localhost";
  63
  64     /**
  65      * The port that will be used in the urls for the feed links.
  66      */
  67     private String port = "8080";
  68
  69     /**
  70      * Process the newly discovered artifacts in the repository. Generate feeds for new artifacts in the repository and
  71      * new versions of artifact.
  72      */
  73     public void process( List<ArchivaArtifact> data )
  74     {
  75         log.debug( "Process new artifacts into rss feeds." );
  76
  77         if ( System.getProperty( "jetty.host" ) != null )
  78         {
  79             host = System.getProperty( "jetty.host" );
  80         }
  81
  82         if ( System.getProperty( "jetty.port" ) != null )
  83         {
  84             port = System.getProperty( "jetty.port" );
  85         }
  86
  87         processNewArtifactsInRepo( data );
  88         processNewVersionsOfArtifact( data );
  89     }
  90
  91     private void processNewArtifactsInRepo( List<ArchivaArtifact> data )
  92     {
  93         List<RssFeedEntry> entries = new ArrayList<RssFeedEntry>();
  94         String repoId = getRepoId( data );
  95
  96         RssFeedEntry entry =
  97             new RssFeedEntry( NEW_ARTIFACTS_IN_REPO + "\'" + repoId + "\'" + " as of " +
  98                 Calendar.getInstance().getTime(), getBaseUrl() + "/archiva/rss/new_artifacts_" + repoId + ".xml" );
  99         String description = "These are the new artifacts found in repository " + "\'" + repoId + "\'" + ": \n";
 100
 101         for ( ArchivaArtifact artifact : data )
 102         {
 103             description = description + artifact.toString() + " | ";
 104         }
 105         entry.setDescription( description );
 106         entries.add( entry );
 107
 108         generateFeed( "new_artifacts_" + repoId + ".xml", NEW_ARTIFACTS_IN_REPO + "\'" + repoId + "\'",
 109                       getBaseUrl() + "/archiva/repository/rss/new_artifacts_" + repoId + ".xml",
 110                       "New artifacts found in repository " + "\'" + repoId + "\'" + " during repository scan.", entries );
 111     }
 112
 113     private void processNewVersionsOfArtifact( List<ArchivaArtifact> data )
 114     {
 115         String repoId = getRepoId( data );
 116
 117         List<String> artifacts = new ArrayList<String>();
 118
 119         for ( ArchivaArtifact artifact : data )
 120         {
 121             artifacts.add( artifact.toString() );
 122         }
 123
 124         Collections.sort( artifacts );
 125
 126         Map<String, String> artifactsMap = toMap( artifacts );
 127
 128         for ( String key : artifactsMap.keySet() )
 129         {
 130             List<RssFeedEntry> entries = new ArrayList<RssFeedEntry>();
 131             RssFeedEntry entry =
 132                 new RssFeedEntry( NEW_VERSIONS_OF_ARTIFACT + "\'" + key + "\'" + " as of " +
 133                     Calendar.getInstance().getTime(), getBaseUrl() + "/archiva/rss/new_versions_" + key + ".xml" );
 134
 135             String description =
 136                 "These are the new versions of artifact " + "\'" + key + "\'" + " in the repository: \n" +
 137                     ( (String) artifactsMap.get( key ) );
 138
 139             entry.setDescription( description );
 140             entries.add( entry );
 141
 142             generateFeed( "new_versions_" + key + ".xml", NEW_VERSIONS_OF_ARTIFACT + "\'" + key + "\'",
 143                           getBaseUrl() + "/archiva/rss/new_versions_" + key + ".xml",
 144                           "New versions of artifact " + "\'" + key + "\' found in repository " + "\'" + repoId + "\'" +
 145                               " during repository scan.", entries );
 146         }
 147     }
 148
 149     private String getRepoId( List<ArchivaArtifact> data )
 150     {
 151         String repoId = "";
 152         if ( !data.isEmpty() )
 153         {
 154             repoId = ( (ArchivaArtifact) data.get( 0 ) ).getModel().getRepositoryId();
 155         }
 156
 157         return repoId;
 158     }
 159
 160     private void generateFeed( String filename, String title, String link, String description,
 161                                List<RssFeedEntry> dataEntries )
 162     {
 163         generator.generateFeed( title, link, description, dataEntries, filename );
 164     }
 165
 166     private Map<String, String> toMap( List<String> artifacts )
 167     {
 168         Map<String, String> artifactsMap = new HashMap<String, String>();
 169         for ( String id : artifacts )
 170         {
 171             String key = StringUtils.substringBefore( id, ":" );
 172             key = key + ":" + StringUtils.substringBefore( StringUtils.substringAfter( id, ":" ), ":" );
 173
 174             String value = (String) artifactsMap.get( key );
 175             if ( value != null )
 176             {
 177                 value = value + " | " + id;
 178             }
 179             else
 180             {
 181                 value = id;
 182             }
 183             artifactsMap.put( key, value );
 184         }
 185
 186         return artifactsMap;
 187     }
 188
 189     public RssFeedGenerator getGenerator()
 190     {
 191         return generator;
 192     }
 193
 194     public void setGenerator( RssFeedGenerator generator )
 195     {
 196         this.generator = generator;
 197     }
 198
 199     private String getBaseUrl()
 200     {
 201         String baseUrl = "http://" + host;
 202
 203         if( port != null && !"".equals( port ) )
 204         {
 205             baseUrl = baseUrl + ":" + port;
 206         }
 207
 208         return baseUrl;
 209     }
 210
 211 }