package org.apache.velocity.site.news.macro;
   
   /*
    * 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 java.io.File;
  import java.lang.reflect.Method;
  import java.util.ListIterator;
  import java.util.Locale;
  import java.util.ResourceBundle;
  
  import org.apache.maven.doxia.macro.AbstractMacro;
  import org.apache.maven.doxia.macro.MacroExecutionException;
  import org.apache.maven.doxia.macro.MacroRequest;
  import org.apache.maven.doxia.parser.Parser;
  import org.apache.maven.doxia.sink.Sink;
  import org.apache.velocity.site.news.NewsBlockGenerator;
  import org.apache.velocity.site.news.NewsReport;
  import org.apache.velocity.site.news.VelocityNewsException;
  import org.apache.velocity.site.news.VelocityNewsUtils;
  import org.apache.velocity.site.news.model.Item;
  import org.apache.velocity.site.news.model.NewsManager;
  import org.apache.velocity.site.news.model.NewsModel;
  import org.apache.velocity.site.news.plugin.VelocityNewsPlugin;
  import org.apache.velocity.site.plexus.PlexusKludgeException;
  import org.apache.velocity.site.plexus.PlexusKludgeUtils;
  import org.codehaus.plexus.PlexusConstants;
  import org.codehaus.plexus.PlexusContainer;
  import org.codehaus.plexus.component.repository.exception.ComponentLookupException;
  import org.codehaus.plexus.context.Context;
  import org.codehaus.plexus.context.ContextException;
  import org.codehaus.plexus.personality.plexus.lifecycle.phase.Contextualizable;
  
  /**
   * apt usage: %{news}
   *
   * @plexus.component role="org.apache.maven.doxia.macro.Macro" role-hint="news"
   */
  public class VelocityNewsMacro extends AbstractMacro implements Contextualizable
  {
  
      private PlexusContainer container = null;
  
      public void contextualize(final Context context) throws ContextException
      {
          this.container = (PlexusContainer) context.get(PlexusConstants.PLEXUS_KEY);
      }
  
      public void execute(final Sink sink, final MacroRequest request) throws MacroExecutionException
      {
  
          try
          {
              int count = Integer.parseInt(getParameter(request, "count", "5"));
              generateNewsBlock(sink, count);
  
          }
          catch (VelocityNewsException vne)
          {
              throw new MacroExecutionException("While executing news macro: ", vne);
          }
      }
  
      protected String getParameter(final MacroRequest request, final String field, final String defaultValue)
      {
          String value = (String) request.getParameter(field);
          return value != null ? value : defaultValue;
      }
  
      protected void generateNewsBlock(final Sink sink, final int maxCount) throws VelocityNewsException
      {
  
          try
          {
  
              Object newsManagerRef = getNewsManager();
  
              Method newsFileNameMethod = newsManagerRef.getClass().getMethod(NewsManager.NEWS_FILE_NAME_METHOD, new Class[] {});
              String newsFileName = (String) newsFileNameMethod.invoke(newsManagerRef, new Object[] {});
  
              Method newsSiteDirectoryMethod = newsManagerRef.getClass().getMethod(NewsManager.NEWS_SITE_DIRECTORY_METHOD,
                      new Class[] {});
              File siteDirectory = (File) newsSiteDirectoryMethod.invoke(newsManagerRef, new Object[] {});
 
             final NewsManager newsManager = new NewsManager();
             newsManager.load(siteDirectory, newsFileName);
 
             NewsBlockGenerator newsBlockGenerator = new NewsBlockGenerator(new NewsReport()
             {
 
                 private Parser aptParser = null;
 
                 public Sink getSink()
                 {
                     return sink;
                 }
 
                 public Parser getAptParser() throws VelocityNewsException
                 {
                     if (aptParser == null)
                     {
                         try
                         {
                             aptParser = (Parser) container.lookup(Parser.ROLE, "apt");
                         }
                         catch (ComponentLookupException cle)
                         {
                             throw new VelocityNewsException(cle);
                         }
                     }
                     return aptParser;
                 }
 
                 public ResourceBundle getBundle(final Locale locale)
                 {
                     return ResourceBundle.getBundle("velocity-news-report", locale);
                 }
 
                 public String getOutputName()
                 {
                     try
                     {
                         return (String) newsManager.getNewsFileName();
                     }
                     catch (RuntimeException re)
                     {
                         throw re;
                     }
                     catch (Exception e)
                     {
                         throw new IllegalStateException("Could not retrieve page name from model!");
                     }
                 }
 
             }, Locale.ENGLISH);
 
             NewsModel newsModel = newsManager.getNewsModel();
 
             for (ListIterator it = VelocityNewsUtils.sortItemsByReverseDate(newsModel.getItems()).listIterator(); it.hasNext()
                     && it.nextIndex() < maxCount;)
             {
                 Item newsItem = (Item) it.next();
                 newsBlockGenerator.displayItem(newsItem);
             }
 
         }
         catch (RuntimeException re)
         {
             throw re;
         }
         catch (Exception e)
         {
             throw new IllegalStateException("Could not retrieve page name from model!");
         }
     }
 
     /**
      * Returns the News Manager from the Plugin.
      *
      * @return A {@link DoxiaVelocityContextFactory} instance that generates
      *         {@link VelocityContext} objects for the render process.
      */
     private Object getNewsManager() throws VelocityNewsException
     {
         try
         {
             PlexusContainer velocityContainer = PlexusKludgeUtils.lookupContainer(container,
                     VelocityNewsPlugin.CONTAINER_LOOKUP_NAME);
             return velocityContainer.lookup(NewsManager.ROLE);
         }
         catch (PlexusKludgeException pke)
         {
             throw new VelocityNewsException("While retrieving News Manager: ", pke);
         }
         catch (ComponentLookupException cle)
         {
             throw new VelocityNewsException("While retrieving News Manager: ", cle);
         }
     }
 }