/*

  * Copyright 2001,2004 The Apache Software Foundation.

  * 

  * Licensed 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.

  */

 package org.apache.commons.jjar;

 import java.net.URL;

 import java.net.URLConnection;

 import java.util.List;

 import java.util.Iterator;

 import org.apache.tools.ant.BuildException;

 import org.apache.tools.ant.Task;

 import org.apache.tools.ant.types.Path;

 /**

  * An ant task for doing JJAR stuff

  *

  * @version $Id: JJARTask.java 155454 2005-02-26 13:23:34Z dirkv $

  */

 public class JJARTask extends Task

 {

     private static String defaultRepository = ClasspathUtil.getDefaultRepository();

     private String pkg = null;

     private String ver = null;

     private String localrep = ".";

     private boolean verifyignore = true;

     private boolean onlydependencies = false;

     private String classpath = null;

     private String pathrefid = null;

     /**

      *  [REQUIRED] sets the package and version.  ver can be 

      *  null to get the default

      */

     public void setPackage( String pkg)

     {

         this.pkg = pkg;

     }

     public void setVersion( String version )

     {

         this.ver = version;

     }

     public void setLocalrepository( String localrep )

     {

         this.localrep = localrep;

     }

     public void setVerifyignore( boolean b )

     {

         this.verifyignore = b;

     }

     public void setOnlyDependencies( boolean b )

     {

         this.onlydependencies = b;

     }

     public void setRepository( String rep )

     {

         this.defaultRepository = rep;

     }

     public void setClasspath( String classpath )

     {

         this.classpath = classpath;

     }

     public void setPathrefid( String pathrefid )

     {

         this.pathrefid = pathrefid;

     }

     /**

      * Execute the input script with Velocity

      *

      * @throws BuildException  

      * BuildExceptions are thrown when required attributes are missing.

      * Exceptions thrown by Velocity are rethrown as BuildExceptions.

      */

     public void execute () 

         throws BuildException

     {

         /*

          *  make sure that we at least have a package name

          */

         if( pkg == null)

             throw new BuildException("Need to specify a package to fetch");

         /*

          *  get the repository

          */

         Repository repository = null;

         try

         {

             repository = new RepositoryXML();

             repository.load(  new URL( defaultRepository + "repository.xml") );

         }

         catch( Exception e )

         {

             throw new BuildException("Problem getting repository info", e );

         }

         log("Fetching package '" + pkg + "' from repository '" + defaultRepository 

             + "' into local repository '" + localrep + "'");

         /*

          *  create an antpath

          */

         Path antpath = new Path( project );

         try

         {

             /*

              *  do it : first, see if there is a dependency list

              */

             if( ver == null)

                 ver = repository.getPackageDefaultVersion( pkg );

             List deplist = repository.getDependencyList( pkg, ver );

             Iterator ii = deplist.iterator();

             while(ii.hasNext())

             {

                 /*

                  *  for now, use the fetchtarget names

                  */

                 String depnameuni = (String) ii.next();

                 JJARPackage jjarp = new JJARPackage( depnameuni );

                 String depname = jjarp.getName();

                 String depver = jjarp.getVersionString();

                 String deptarget = repository.getFetchTarget( depname, depver );

                 /*

                  *  see if the dependent package is already in the classpath

                  */

                 if ( verifyignore && ClasspathUtil.containsVersionedJar( classpath, depname, depver))

                 {

                     log("  Dependency '" + depname + "' already in classpath - skipping");

                     continue;

                 }

                 /*

                  *  now get it

                  */

                 URL url = new URL( defaultRepository + deptarget );

                 URLConnection uconn  = (URLConnection) url.openConnection(); 

                 log("  Fetching dependency : " + deptarget );

                 Transport.fetchJar( uconn, localrep + "/" + deptarget );                

                 /*

                  *  now if they want us to add to a classpath, we will have

                  *  a pathrefid, so for now, pretend it's fresh

                  */

                 log("Adding depedency element " + localrep + "/" + deptarget + " to classpath");

                 antpath.createPathElement().setPath( localrep + "/" + deptarget );

             }

             /*

              *  now get the actual package.  If all we want were the dependencies

              *  for something (example : we are building something...) then

              *  bail, because we are done.

              */

             if( onlydependencies )

             {

                 log("  Dependency-only mode : not fetching target " + pkg );

                 return;

             }

             /*

              *  if we don't have a version, we need to get it from the repository

              */

             if (ver == null)

             {

                 ver = repository.getPackageDefaultVersion( pkg );

             }

             if( verifyignore && ClasspathUtil.containsVersionedJar( classpath, pkg, ver))

             {                

                 log("  Skipping fetch of '" + pkg + "' - already exists in classpath");

                 // ignore and continue

             }

             else

             {

                 String fetchTarget = repository.getFetchTarget( pkg, ver );

                 URL url = new URL( defaultRepository + fetchTarget );

                 URLConnection uconn  = (URLConnection) url.openConnection(); 

                 log("  Fetching package : " + pkg );

                 Transport.fetchJar( uconn, localrep + "/" + fetchTarget );

                 log("Adding package element " + localrep + "/" + fetchTarget + " to classpath");

                 antpath.createPathElement().setPath( localrep + "/" + fetchTarget );

             }

         }

         catch( Exception e )

         {

             throw new BuildException("Exception : ",e);

         }

         finally

         {

             if( pathrefid != null)

             {

                 project.addReference( pathrefid, antpath);

             }

         }

     }

 }