/*
 * Data Hub Service (DHuS) - For Space data distribution.
 * Copyright (C) 2013,2014,2015 GAEL Systems
 *
 * This file is part of DHuS software sources.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */
package fr.gael.dhus.database.liquibase;

import liquibase.change.custom.CustomTaskChange;
import liquibase.database.Database;
import liquibase.database.jvm.JdbcConnection;
import liquibase.exception.CustomChangeException;
import liquibase.exception.SetupException;
import liquibase.exception.ValidationErrors;
import liquibase.resource.ResourceAccessor;

import org.apache.logging.log4j.Logger;
import org.apache.logging.log4j.LogManager;

import java.io.File;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.text.SimpleDateFormat;
import java.util.Date;

public class ExtractProductDatesAndDownloadSize implements CustomTaskChange
{
   private static final Logger LOGGER = LogManager.getLogger(ExtractProductDatesAndDownloadSize.class);

   @Override
   public String getConfirmationMessage ()
   {
      return null;
   }

   @Override
   public void setFileOpener (ResourceAccessor resource_accessor)
   {
   }   

   @Override
   public void setUp () throws SetupException
   {
   }

   @Override
   public ValidationErrors validate (Database arg0)
   {
      return null;
   }

   @Override
   public void execute (Database database) throws CustomChangeException
   {
      // contentStart contentEnd ingestionDate download.size
      JdbcConnection databaseConnection =
         (JdbcConnection) database.getConnection ();
      try
      {
         // SELECT PRODUCT_ID, QUERYABLE, VALUE FROM METADA_INDEXES
         PreparedStatement getQueryables =
               databaseConnection
                     .prepareStatement ("SELECT PRODUCT_ID, QUERYABLE, VALUE " +
                  "FROM METADATA_INDEXES");
         ResultSet res = getQueryables.executeQuery ();
         SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss.SSS");
         while (res.next ())
         {
            Long productIdentifier = (Long) res.getObject ("PRODUCT_ID");   
            String queryable = (String) res.getObject ("QUERYABLE");  
            String value = (String) res.getObject ("VALUE");  
            String query = "";
            if ("beginPosition".equals (queryable))
            {
               Date d = new SimpleDateFormat ("yyyy-MM-dd'T'HH:mm:ss.SSS")
                     .parse(value);
               query = "UPDATE PRODUCTS SET CONTENTSTART = '"+sdf.format (d)+
                     "' WHERE ID = "+productIdentifier;
            }
            else if ("endPosition".equals (queryable))
            {
               Date d = new SimpleDateFormat ("yyyy-MM-dd'T'HH:mm:ss.SSS")
                     .parse(value);
               query = "UPDATE PRODUCTS SET CONTENTEND = '"+sdf.format (d)+
                     "' WHERE ID = "+productIdentifier;
            }
            else if ("ingestionDate".equals (queryable))
            {
               Date d = new SimpleDateFormat ("yyyy-MM-dd'T'HH:mm:ss.SSS")
                     .parse(value);
               query = "UPDATE PRODUCTS SET INGESTIONDATE = '"+sdf.format (d)+
                     "' WHERE ID = "+productIdentifier;
            }
            else
            {
               continue;
            }
            PreparedStatement updateProduct = databaseConnection
                  .prepareStatement (query);
            updateProduct.execute ();
            updateProduct.close();
         }
         getQueryables.close ();
      }
      catch (Exception e)
      {
         LOGGER.error ("Error during liquibase update " +
               "'ExtractProductDatesAndDownloadSize'", e);
      }
      
      try
      {
         PreparedStatement getQueryables =
               databaseConnection.prepareStatement (
                           "SELECT p.ID, p.DOWNLOAD_PATH FROM PRODUCTS p ");
         ResultSet res = getQueryables.executeQuery ();

         while (res.next ())
         {
            Long productIdentifier = (Long) res.getObject ("ID");   
            String path = (String) res.getObject ("DOWNLOAD_PATH");
            File dl = new File(path);
            if (dl.exists())
            {
               PreparedStatement updateProduct = databaseConnection
                     .prepareStatement ("UPDATE PRODUCTS SET DOWNLOAD_SIZE = " +
                           dl.length ()+" WHERE ID = "+productIdentifier);
               updateProduct.execute ();
               updateProduct.close();
            }
         }
         getQueryables.close ();
      }
      catch (Exception e)
      {
         LOGGER.error ("Error during liquibase update " +
               "'ExtractProductDatesAndDownloadSize'", e);
      }
   }
}