/*
    * This file is part of dependency-check-ant.
    *
    * 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.
   *
   * Copyright (c) 2015 Jeremy Long. All Rights Reserved.
   */
  package org.owasp.dependencycheck.taskdefs;
  
  import java.io.File;
  import java.io.IOException;
  import java.io.InputStream;
  import org.apache.tools.ant.BuildException;
  import org.apache.tools.ant.Project;
  import org.apache.tools.ant.Task;
  import org.owasp.dependencycheck.utils.Settings;
  import org.slf4j.impl.StaticLoggerBinder;
  
  /**
   * An Ant task definition to execute dependency-check during an Ant build.
   *
   * @author Jeremy Long
   */
  public class Purge extends Task {
  
      /**
       * The properties file location.
       */
      private static final String PROPERTIES_FILE = "task.properties";
  
      /**
       * Construct a new DependencyCheckTask.
       */
      public Purge() {
          super();
          // Call this before Dependency Check Core starts logging anything - this way, all SLF4J messages from
          // core end up coming through this tasks logger
          StaticLoggerBinder.getSingleton().setTask(this);
      }
  
      /**
       * The location of the data directory that contains
       */
      private String dataDirectory = null;
  
      /**
       * Get the value of dataDirectory.
       *
       * @return the value of dataDirectory
       */
      public String getDataDirectory() {
          return dataDirectory;
      }
  
      /**
       * Set the value of dataDirectory.
       *
       * @param dataDirectory new value of dataDirectory
       */
      public void setDataDirectory(String dataDirectory) {
          this.dataDirectory = dataDirectory;
      }
  
      /**
       * Indicates if dependency-check should fail the build if an exception
       * occurs.
       */
      private boolean failOnError = true;
  
      /**
       * Get the value of failOnError.
       *
       * @return the value of failOnError
       */
      public boolean isFailOnError() {
          return failOnError;
      }
  
      /**
       * Set the value of failOnError.
       *
       * @param failOnError new value of failOnError
       */
      public void setFailOnError(boolean failOnError) {
          this.failOnError = failOnError;
      }
  
      @Override
      public void execute() throws BuildException {
         populateSettings();
         File db;
         try {
             db = new File(Settings.getDataDirectory(), "dc.h2.db");
             if (db.exists()) {
                 if (db.delete()) {
                     log("Database file purged; local copy of the NVD has been removed", Project.MSG_INFO);
                 } else {
                     final String msg = String.format("Unable to delete '%s'; please delete the file manually", db.getAbsolutePath());
                     if (this.failOnError) {
                         throw new BuildException(msg);
                     }
                     log(msg, Project.MSG_ERR);
                 }
             } else {
                 final String msg = String.format("Unable to purge database; the database file does not exists: %s", db.getAbsolutePath());
                 if (this.failOnError) {
                     throw new BuildException(msg);
                 }
                 log(msg, Project.MSG_ERR);
             }
         } catch (IOException ex) {
             final String msg = "Unable to delete the database";
             if (this.failOnError) {
                 throw new BuildException(msg);
             }
             log(msg, Project.MSG_ERR);
         } finally {
             Settings.cleanup(true);
         }
     }
 
     /**
      * Takes the properties supplied and updates the dependency-check settings.
      * Additionally, this sets the system properties required to change the
      * proxy server, port, and connection timeout.
      *
      * @throws BuildException thrown if the properties file cannot be read.
      */
     protected void populateSettings() throws BuildException {
         Settings.initialize();
         InputStream taskProperties = null;
         try {
             taskProperties = this.getClass().getClassLoader().getResourceAsStream(PROPERTIES_FILE);
             Settings.mergeProperties(taskProperties);
         } catch (IOException ex) {
             final String msg = "Unable to load the dependency-check ant task.properties file.";
             if (this.failOnError) {
                 throw new BuildException(msg, ex);
             }
             log(msg, ex, Project.MSG_WARN);
         } finally {
             if (taskProperties != null) {
                 try {
                     taskProperties.close();
                 } catch (IOException ex) {
                     log("", ex, Project.MSG_DEBUG);
                 }
             }
         }
         if (dataDirectory != null) {
             Settings.setString(Settings.KEYS.DATA_DIRECTORY, dataDirectory);
         } else {
             final File jarPath = new File(Purge.class.getProtectionDomain().getCodeSource().getLocation().getPath());
             final File base = jarPath.getParentFile();
             final String sub = Settings.getString(Settings.KEYS.DATA_DIRECTORY);
             final File dataDir = new File(base, sub);
             Settings.setString(Settings.KEYS.DATA_DIRECTORY, dataDir.getAbsolutePath());
         }
     }
 }