package org.apache.maven.plugins.enforcer;
   
   /*
    * 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 org.apache.maven.enforcer.rule.api.EnforcerRule;
  import org.apache.maven.enforcer.rule.api.EnforcerRuleException;
  import org.apache.maven.enforcer.rule.api.EnforcerRuleHelper;
  import org.apache.maven.plugin.logging.Log;
  import org.apache.maven.project.MavenProject;
  import org.codehaus.plexus.component.configurator.expression.ExpressionEvaluationException;
  
  /**
   * Rule to validate the main artifact is within certain size constraints.
   *
   * @author brianf
   * @author Roman Stumm
   */
  public class RequireFilesSize
      extends AbstractRequireFiles
  {
  
      private static final long MAXSIZE = 10000;
  
      /** the max size allowed. */
      private long maxsize = MAXSIZE;
  
      /** the min size allowed. */
      private long minsize = 0;
  
      /** The error msg. */
      private String errorMsg;
  
      /** The log. */
      private Log log;
  
      @Override
      public void execute( EnforcerRuleHelper helper )
          throws EnforcerRuleException
      {
          this.log = helper.getLog();
  
          // if the file is already defined, use that. Otherwise get the main artifact.
          if ( getFiles().length == 0 )
          {
              try
              {
                  MavenProject project = (MavenProject) helper.evaluate( "${project}" );
                  setFiles( new File[1] );
                  getFiles()[0] = project.getArtifact().getFile();
  
                  super.execute( helper );
              }
              catch ( ExpressionEvaluationException e )
              {
                  throw new EnforcerRuleException( "Unable to retrieve the project.", e );
              }
          }
          else
          {
              super.execute( helper );
          }
  
      }
  
      @Override
      public boolean isCacheable()
      {
          return false;
      }
  
      @Override
      public boolean isResultValid( EnforcerRule cachedRule )
      {
          return false;
      }
  
      @Override
      boolean checkFile( File file )
      {
          if ( file == null )
         {
             // if we get here and it's null, treat it as a success.
             return true;
         }
 
         // check the file now
         if ( file.exists() )
         {
             long length = file.length();
             if ( length < minsize )
             {
                 this.errorMsg = ( file + " size (" + length + ") too small. Min. is " + minsize );
                 return false;
             }
             else if ( length > maxsize )
             {
                 this.errorMsg = ( file + " size (" + length + ") too large. Max. is " + maxsize );
                 return false;
             }
             else
             {
 
                 this.log.debug( file
                     + " size ("
                     + length
                     + ") is OK ("
                     + ( minsize == maxsize || minsize == 0 ? ( "max. " + maxsize )
                                     : ( "between " + minsize + " and " + maxsize ) ) + " byte)." );
 
                 return true;
             }
         }
         else
         {
             this.errorMsg = ( file + " does not exist!" );
             return false;
         }
     }
 
     @Override
     String getErrorMsg()
     {
         return this.errorMsg;
     }
 
     public long getMaxsize()
     {
         return maxsize;
     }
 
     public void setMaxsize( long maxsize )
     {
         this.maxsize = maxsize;
     }
 
     public long getMinsize()
     {
         return minsize;
     }
 
     public void setMinsize( long minsize )
     {
         this.minsize = minsize;
     }
 }