/*

  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.

  *

  * Copyright (c) 2012-2013 Oracle and/or its affiliates. All rights reserved.

  *

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

  *

  * 

  * This file incorporates work covered by the following copyright and

  * permission notice:

  *

  * Copyright 2006 Guillaume Nodet

  *

  * 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.codehaus.mojo.jaxws;

 import java.io.File;

 import java.io.IOException;

 import java.net.URL;

 import java.net.URLClassLoader;

 import java.util.ArrayList;

 import java.util.HashSet;

 import java.util.Set;

 import javax.jws.WebService;

 import org.apache.maven.model.Resource;

 import org.apache.maven.plugin.MojoExecutionException;

 import org.apache.maven.plugin.MojoFailureException;

 import org.apache.maven.plugins.annotations.Parameter;

 import org.codehaus.plexus.util.FileUtils;

 /**

  * 

  *

  * @author gnodet (gnodet@apache.org)

  * @author dantran (dantran@apache.org)

  */

 abstract class AbstractWsGenMojo

     extends AbstractJaxwsMojo

 {

     /**

      * Specify that a WSDL file should be generated in <code>${resourceDestDir}</code>.

      */

     @Parameter( defaultValue = "false" )

     protected boolean genWsdl;

     /**

      * Service endpoint implementation class name.

      */

     @Parameter

     private String sei;

     /**

      * Used in conjunction with <code>genWsdl<code> to specify the protocol to use in the

      * <code>wsdl:binding</code>. Valid values are "<code>soap1.1</code>" or "<code>Xsoap1.2</code>",

      * default is "<code>soap1.1</code>". "<code>Xsoap1.2</code>" is not standard

      * and can only be used in conjunction with the <code>extension</code> option.

      */

     @Parameter

     private String protocol;

     /**

      * Specify the Service name to use in the generated WSDL.

      * Used in conjunction with the <code>genWsdl</code> option.

      */

     @Parameter

     private String servicename;

     /**

      * Specify the Port name to use in the generated WSDL.

      * Used in conjunction with the <code>genWsdl</code> option.

      */

     @Parameter

     private String portname;

     /**

      * Inline schemas in the generated WSDL.

      * Used in conjunction with the <code>genWsdl</code> option.

      */

     @Parameter( defaultValue = "false" )

     private boolean inlineSchemas;

     /**

      * Turn off compilation after code generation and let generated sources be

      * compiled by maven during compilation phase; keep is turned on with this option.

      */

     @Parameter( defaultValue = "false" )

     private boolean xnocompile;

     /**

      */

     @Parameter( defaultValue = "false" )

     private boolean xdonotoverwrite;

     /**

      * Metadata file for wsgen. See

      * <a href="https://jax-ws.java.net/2.2.8/docs/ch03.html#users-guide-external-metadata">the JAX-WS Guide</a>

      * for the description of this feature.

      * Unmatched files will be ignored.

      *

      * @since 2.3

      * @see <a href="https://jax-ws.java.net/2.2.8/docs/ch03.html#users-guide-external-metadata">External Web Service Metadata</a>

      */

     @Parameter

     private File metadata;

     protected abstract File getResourceDestDir();

     protected abstract File getClassesDir();

     @Override

     public void executeJaxws()

         throws MojoExecutionException, MojoFailureException

     {

         Set<String> seis = new HashSet<>();

         if ( sei != null )

         {

             seis.add( sei );

         }

         else

         {

             // find all SEIs within current classes

             seis.addAll( getSEIs( getClassesDir() ) );

         }

         if ( seis.isEmpty() )

         {

             throw new MojoFailureException( "No @javax.jws.WebService found." );

         }

         for ( String aSei : seis )

         {

             processSei( aSei );

         }

     }

     protected void processSei( String aSei )

         throws MojoExecutionException

     {

         getLog().info( "Processing: " + aSei );

         ArrayList<String> args = getWsGenArgs( aSei );

         getLog().info( "jaxws:wsgen args: " + args );

         exec( args );

         if ( metadata != null )

         {

             try

             {

                 FileUtils.copyFileToDirectory( metadata, getClassesDir() );

             }

             catch ( IOException ioe )

             {

                 throw new MojoExecutionException( ioe.getMessage(), ioe );

             }

         }

     }

     @Override

     protected String getMain()

     {

         return "com.sun.tools.ws.wscompile.WsgenTool";

     }

     @Override

     protected String getToolName()

     {

         return "wsgen";

     }

     @Override

     protected String getExtraClasspath()

     {

         return String.join( File.pathSeparator,

                 getClassesDir().getAbsolutePath(),

                 getCPasString( project.getArtifacts() ) );

     }

     @Override

     protected boolean isXnocompile()

     {

         return xnocompile;

     }

     /**

      * Returns wsgen's command arguments as a list

      */

     private ArrayList<String> getWsGenArgs( String aSei )

         throws MojoExecutionException

     {

         ArrayList<String> args = new ArrayList<>();

         args.addAll( getCommonArgs() );

         if ( this.genWsdl )

         {

             if ( this.protocol != null )

             {

                 args.add( "-wsdl:" + this.protocol );

             }

             else

             {

                 args.add( "-wsdl" );

             }

             if ( inlineSchemas )

             {

                 maybeUnsupportedOption( "-inlineSchemas", null, args );

             }

             if ( servicename != null )

             {

                 args.add( "-servicename" );

                 args.add( servicename );

             }

             if ( portname != null )

             {

                 args.add( "-portname" );

                 args.add( portname );

             }

             File resourceDir = getResourceDestDir();

             if ( !resourceDir.mkdirs() && !resourceDir.exists() )

             {

                 getLog().warn( "Cannot create directory: " + resourceDir.getAbsolutePath() );

             }

             args.add( "-r" );

             args.add( "'" + resourceDir.getAbsolutePath() + "'" );

             if ( !"war".equals( project.getPackaging() ) )

             {

                 Resource r = new Resource();

                 r.setDirectory( getRelativePath( project.getBasedir(), getResourceDestDir() ) );

                 project.addResource( r );

             }

         }

         if ( xdonotoverwrite )

         {

             args.add( "-Xdonotoverwrite" );

         }

         if ( metadata != null && isArgSupported( "-x" ) )

         {

             maybeUnsupportedOption( "-x", "'" + metadata.getAbsolutePath() + "'", args );

         }

         args.add( aSei );

         return args;

     }

     private String getRelativePath( File root, File f )

     {

         return root.toURI().relativize( f.toURI() ).getPath();

     }

     private Set<String> getSEIs( File directory )

         throws MojoExecutionException

     {

         Set<String> seis = new HashSet<>();

         if ( !directory.exists() || directory.isFile() )

         {

             return seis;

         }

         try ( URLClassLoader cl = new URLClassLoader( new URL[] { directory.toURI().toURL() }, getClass().getClassLoader() ) )

         {

             for ( String s : FileUtils.getFileAndDirectoryNames( directory, "**/*.class", null, false, true, true,

                                                                  false ) )

             {

                 try

                 {

                     String clsName = s.replace( File.separator, "." );

                     Class<?> c = cl.loadClass( clsName.substring( 0, clsName.length() - 6 ) );

                     WebService ann = c.getAnnotation( WebService.class );

                     if ( !c.isInterface() && ann != null )

                     {

                         // more sophisticated checks are done by wsgen itself

                         seis.add( c.getName() );

                     }

                 }

                 catch ( ClassNotFoundException ex )

                 {

                     throw new MojoExecutionException( ex.getMessage(), ex );

                 }

             }

         }

         catch ( IOException ex )

         {

             throw new MojoExecutionException( ex.getMessage(), ex );

         }

         return seis;

     }

 }