/**
 *  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.
 */
package org.sonatype.maven.plugins.mercury.compare;

import java.io.BufferedWriter;
import java.io.File;
import java.io.FileInputStream;
import java.io.FileWriter;
import java.net.URL;
import java.util.ArrayList;
import java.util.List;
import java.util.Properties;

import org.apache.maven.mercury.MavenDependencyProcessor;
import org.apache.maven.mercury.artifact.ArtifactBasicMetadata;
import org.apache.maven.mercury.artifact.ArtifactMetadata;
import org.apache.maven.mercury.artifact.ArtifactScopeEnum;
import org.apache.maven.mercury.compare.DataBuilder;
import org.apache.maven.mercury.compare.IDepResolver;
import org.apache.maven.mercury.compare.mdo.Dependencies;
import org.apache.maven.mercury.compare.mdo.Dependency;
import org.apache.maven.mercury.metadata.DependencyBuilder;
import org.apache.maven.mercury.metadata.DependencyBuilderFactory;
import org.apache.maven.mercury.metadata.MetadataTreeNode;
import org.apache.maven.mercury.repository.api.Repository;
import org.apache.maven.mercury.repository.api.RepositoryUpdateIntervalPolicy;
import org.apache.maven.mercury.repository.local.m2.LocalRepositoryM2;
import org.apache.maven.mercury.repository.remote.m2.RemoteRepositoryM2;
import org.apache.maven.mercury.transport.api.Server;
import org.apache.maven.mercury.util.Util;

/**
 * 
 * @author Oleg Gusakov
 *
 */

public class MercuryResolve
implements IDepResolver
{
  static final String SYSTEM_PROPERTY_LOCAL_REPO = "localRepo";
  static final String localRepoDir = System.getProperty( SYSTEM_PROPERTY_LOCAL_REPO, "../localRepoMercury" );
//  static final String localRepoDir = System.getProperty( SYSTEM_PROPERTY_LOCAL_REPO, "../localRepoMaven" );
  static final File   localRepoDirFile = new File( localRepoDir );
  
  static final String SYSTEM_PROPERTY_LIST_FILE = "list";
  static final String list = System.getProperty( SYSTEM_PROPERTY_LIST_FILE, "../list.txt" );
  static final File   listFile = new File( list );
  
  static final String SYSTEM_PROPERTY_RESULTS_FILE = "results";
  static final String results = System.getProperty( SYSTEM_PROPERTY_RESULTS_FILE, "../results.html" );
  static final File   resultsFile = new File( results );
  
  static final String SYSTEM_PROPERTY_REPOS_FILE = "repoList";
  static final String repoList = System.getProperty( SYSTEM_PROPERTY_REPOS_FILE, "../repositories.properties" );
  static final File   repoListFile = new File( repoList );

  static final String SYSTEM_PROPERTY_TARGET_DIR = "targetDir";
  static final String targetDir = System.getProperty( SYSTEM_PROPERTY_TARGET_DIR, "../deps" );
  static final File   targetDirFile = new File( targetDir );
  
  static final List<Repository> repos = new ArrayList<Repository>(8);
  
  static LocalRepositoryM2 lRepo;
  
  static DependencyBuilder depBuilder;

  static BufferedWriter bw;
  
  
	//----------------------------------------------------------------
	public static void main( String [] av )
	throws Exception
	{

    if( !localRepoDirFile.exists() )
      localRepoDirFile.mkdirs();

    if( !listFile.exists() )
      throw new Exception(listFile.getCanonicalPath()+" does nor exist");

    if( !repoListFile.exists() )
      throw new Exception(repoListFile.getCanonicalPath()+" does nor exist");

    if( !targetDirFile.exists() )
      targetDirFile.mkdirs();
    
    lRepo = new LocalRepositoryM2( "localMercury", new File(localRepoDir), new MavenDependencyProcessor() );
    repos.add( lRepo );

    MercuryResolve mc = new MercuryResolve();
    
    if( av == null || av.length < 1 )
    {
      Properties rep = new Properties();
      rep.load( new FileInputStream(repoListFile) );
  
      for( Object key : rep.keySet() )
      {
        Server server = new Server( (String)key, new URL(rep.getProperty( (String)key )) );
        RemoteRepositoryM2 repo = new RemoteRepositoryM2( server, new MavenDependencyProcessor() );
        repo.setUpdatePolicy( RepositoryUpdateIntervalPolicy.UPDATE_POLICY_NEVER );
        
        repos.add( repo );
      }
      
      depBuilder = DependencyBuilderFactory.create( DependencyBuilderFactory.JAVA_DEPENDENCY_MODEL, repos, null, null, null );
  
    
      DataBuilder.visitDeps( listFile, mc );
    }
    else
    {
      try
      {
        bw = new BufferedWriter( new FileWriter(resultsFile) );

        bw.write( "<html><body>" );

        DataBuilder.compareDeps( listFile, mc );

        bw.write( "</body></html>" );
      }
      finally
      {
        if( bw != null )
          try { bw.flush(); bw.close(); } catch( Exception e ){}
      }
    }
	}
	
	//------------------------------------------------------------------------
  public void resolve( String groupId, String artifactId, String version, String type )
  throws Exception
  {
    String ac = groupId+":"+artifactId+":"+version+"::"+type;
    
    System.out.println("\n======>  "+ac);
    
    long start = System.currentTimeMillis();
    
    ArtifactBasicMetadata bmd = new ArtifactBasicMetadata( ac );
    
    MetadataTreeNode root = depBuilder.buildTree( bmd, ArtifactScopeEnum.runtime );
    
    MetadataTreeNode.showNode( root, 0 );
    
    long ll = System.currentTimeMillis();
    System.out.println("BuildTree: " + (ll - start) );
    
    List<ArtifactMetadata> dl = depBuilder.resolveConflicts( root );
    System.out.println("resolveDeps: " + (System.currentTimeMillis() - ll) );
    
    File df = DataBuilder.getFile( targetDirFile, bmd.getGroupId(), bmd.getArtifactId(), bmd.getVersion(), bmd.getType() );
    
    System.out.println( df.getCanonicalPath() );
    
    Dependencies deps = null;
    
    if( df.exists() )
    {
      deps = DataBuilder.read( df );
      List l = deps.getMercury();
      if( !Util.isEmpty( l ) )
        l.clear();
    }
    else
    {
      deps = new Dependencies();
    
      deps.setGroupId( bmd.getGroupId() );
      deps.setArtifactId( bmd.getArtifactId() );
      deps.setVersion( bmd.getVersion() );
      deps.setType( bmd.getType() );
    }
    
    deps.setMercuryMillis( System.currentTimeMillis() - start );
    
    for( ArtifactMetadata am : dl )
    {
      Dependency dep = new Dependency();
      dep.setGroupId( am.getGroupId() );
      dep.setArtifactId( am.getArtifactId() );
      dep.setVersion( am.getVersion() );
      dep.setType( am.getType() );
      dep.setScope( am.getScope() );
      
      deps.addMercury( dep );
    }
    
    DataBuilder.write( deps, df );
    
  }
	
  public void visit( String groupId, String artifactId, String version, String type )
  throws Exception
  {
    String ac = groupId+":"+artifactId+":"+version+"::"+type;
    
    System.out.println("\n------->  "+ac);
    
    ArtifactBasicMetadata bmd = new ArtifactBasicMetadata( ac );
    
    File df = DataBuilder.getFile( targetDirFile, bmd.getGroupId(), bmd.getArtifactId(), bmd.getVersion(), bmd.getType() );

    if( ! df.exists() )
      return;
    
    System.out.println( df.getCanonicalPath() );
    
    Dependencies deps = null;
    
    deps = DataBuilder.read( df );
    
    if( deps.getMercuryMillis() == 0L || deps.getMavenMillis() == 0L )
      return;
    
    List<Dependency> mavenDeps = deps.getMaven();
    List<Dependency> mercuryDeps = deps.getMercury();
    
    if( mavenDeps.size() < 2 || mercuryDeps.size() < 2 )
      return;
    
    bw.write( "<table border='1'><tr><th colspan='2'>"+bmd.toString()+"</th></tr>\n" );
    
    bw.write(  "\n<tr><th>Maven: "+mavenDeps.size()+" deps in "+deps.getMavenMillis()+" millis</th>"
    		      + "<th>Mercury: "+mercuryDeps.size()+" deps in "+deps.getMercuryMillis()+" millis</th></tr><tr>\n"
    		    );
    
    bw.write(  "\n<td>\n" );
    for( Dependency d : mavenDeps )
    {
      if( contains( d, mercuryDeps ) )
        continue;
      
      String bad = d.getGroupId()+":"+d.getArtifactId()+":"+d.getVersion()+"::"+d.getType()+":"+d.getScope();
      
      System.out.println(" Maven, not mercury: "+ bad );
      
      bw.write( bad+"<br/>\n" );
    }
    bw.write(  "\n</td>\n" );
    
    bw.write(  "\n<td>\n" );
    for( Dependency d : mercuryDeps )
    {
      if( contains( d, mavenDeps ) )
        continue;

      String bad = d.getGroupId()+":"+d.getArtifactId()+":"+d.getVersion()+"::"+d.getType()+":"+d.getScope();
      
      System.out.println(" Mercury, not maven: "+ bad );
      
      bw.write( bad+"<br/>\n" );
    }
    bw.write(  "\n</td>\n" );

    bw.write( "\n</tr></table><br/><br/><br/><br/>\n" );
  }
  
  private static boolean contains( Dependency dep, List<Dependency> dl )
  {
    for( Dependency d : dl )
    {
      if( 
          d.getGroupId().equals( dep.getGroupId() )
          && d.getArtifactId().equals( dep.getArtifactId() )
          && d.getVersion().equals( dep.getVersion() )
//          && Util.nvlS( d.getScope(), "compile" ).equals( Util.nvlS( dep.getScope(), "compile" ) )
        )
      {
        return true;
      }
    }
    return false;
  }
  //----------------------------------------------------------------
	//----------------------------------------------------------------
}