package org.apache.archiva.rss.processor;
   
   /*
    * Licensed to the Apache Software Foundation (ASF) under one
    * or more contributor license agreements.  See the NOTICE file
    * distributed with this work for additional information
    * regarding copyright ownership.  The ASF licenses this file
    * to you under the Apache License, Version 2.0 (the
    * "License"); you may not use this file except in compliance
   * with the License.  You may obtain a copy of the License at
   *
   *  http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing,
   * software distributed under the License is distributed on an
   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
   * KIND, either express or implied.  See the License for the
   * specific language governing permissions and limitations
   * under the License.
   */
  
  import com.rometools.rome.feed.synd.SyndFeed;
  import com.rometools.rome.io.FeedException;
  import org.apache.archiva.metadata.model.ArtifactMetadata;
  import org.apache.archiva.metadata.repository.MetadataRepositoryException;
  import org.apache.archiva.metadata.repository.RepositorySession;
  import org.apache.archiva.metadata.repository.RepositorySessionFactory;
  import org.apache.archiva.rss.RssFeedEntry;
  import org.apache.archiva.rss.RssFeedGenerator;
  import org.slf4j.Logger;
  import org.slf4j.LoggerFactory;
  import org.springframework.stereotype.Service;
  
  import javax.inject.Inject;
  import java.time.ZonedDateTime;
  import java.time.temporal.ChronoUnit;
  import java.util.ArrayList;
  import java.util.Date;
  import java.util.List;
  import java.util.Map;
  import java.util.TimeZone;
  
  /**
   * Retrieve and process all artifacts of a repository from the database and generate a rss feed.
   * The artifacts will be grouped by the date when the artifacts were gathered.
   * Each group will appear as one entry in the feed.
   *
   */
  @Service("rssFeedProcessor#new-artifacts")
  public class NewArtifactsRssFeedProcessor
      extends AbstractArtifactsRssFeedProcessor
  {
      private int numberOfDaysBeforeNow = 30;
  
      private static final String title = "New Artifacts in Repository ";
  
      private static final String desc = "These are the new artifacts found in the repository ";
  
      /**
       *
       */
      @Inject
      private RssFeedGenerator generator;
  
      @Inject
      private RepositorySessionFactory repositorySessionFactory;
  
  
      private Logger log = LoggerFactory.getLogger( NewArtifactsRssFeedProcessor.class );
  
  
      /**
       * Process the newly discovered artifacts in the repository. Generate feeds for new artifacts in the repository and
       * new versions of artifact.
       */
      @Override
      public SyndFeed process( Map<String, String> reqParams )
          throws FeedException
      {
          log.debug( "Process new artifacts into rss feeds." );
  
          String repoId = reqParams.get( RssFeedProcessor.KEY_REPO_ID );
          if ( repoId != null )
          {
              return processNewArtifactsInRepo( repoId );
          }
  
          return null;
      }
  
      private SyndFeed processNewArtifactsInRepo( String repoId )
          throws FeedException
      {
  
          ZonedDateTime greaterThanThisDate = ZonedDateTime.now().minusDays(
                  getNumberOfDaysBeforeNow()
          ).truncatedTo(ChronoUnit.SECONDS);
          List<ArtifactMetadata> artifacts;
          try(RepositorySession session = repositorySessionFactory.createSession())
         {
             artifacts = session.getRepository().getArtifactsByDateRange(session , repoId, greaterThanThisDate, null );
         }
         catch ( MetadataRepositoryException e )
         {
             throw new FeedException( "Unable to construct feed, metadata could not be retrieved: " + e.getMessage(),
                                      e );
         }
 
         long tmp = 0;
         RssFeedEntry entry = null;
         List<RssFeedEntry> entries = new ArrayList<>();
         String description = "";
         int idx = 0;
         for ( ArtifactMetadata artifact : artifacts )
         {
             long whenGathered = artifact.getWhenGathered().toInstant().toEpochMilli();
 
             String id = artifact.getNamespace() + "/" + artifact.getProject() + "/" + artifact.getId();
             if ( tmp != whenGathered )
             {
                 if ( entry != null )
                 {
                     entry.setDescription( description );
                     entries.add( entry );
                     entry = null;
                 }
 
                 String repoId1 = artifact.getRepositoryId();
                 entry = new RssFeedEntry( this.getTitle() + "\'" + repoId1 + "\'" + " as of " + new Date(
                     whenGathered ) );
                 entry.setPublishedDate( Date.from(artifact.getWhenGathered().toInstant()) );
                 description = this.getDescription() + "\'" + repoId1 + "\'" + ": \n" + id + " | ";
             }
             else
             {
                 description = description + id + " | ";
             }
 
             if ( idx == ( artifacts.size() - 1 ) )
             {
                 entry.setDescription( description );
                 entries.add( entry );
             }
 
             tmp = whenGathered;
             idx++;
         }
 
         return generator.generateFeed( getTitle() + "\'" + repoId + "\'",
                                        "New artifacts found in repository " + "\'" + repoId + "\'" +
                                            " during repository scan.", entries );
     }
 
     @Override
     public String getTitle()
     {
         return title;
     }
 
     @Override
     public String getDescription()
     {
         return desc;
     }
 
     public RssFeedGenerator getGenerator()
     {
         return generator;
     }
 
     public void setGenerator( RssFeedGenerator generator )
     {
         this.generator = generator;
     }
 
     public int getNumberOfDaysBeforeNow()
     {
         return numberOfDaysBeforeNow;
     }
 
     public void setNumberOfDaysBeforeNow( int numberOfDaysBeforeNow )
     {
         this.numberOfDaysBeforeNow = numberOfDaysBeforeNow;
     }
 
     public RepositorySessionFactory getRepositorySessionFactory( )
     {
         return repositorySessionFactory;
     }
 
     public void setRepositorySessionFactory( RepositorySessionFactory repositorySessionFactory )
     {
         this.repositorySessionFactory = repositorySessionFactory;
     }
 }