// Copyright (C) Stichting Deltares 2017. All rights reserved.
//
// This file is part of Ringtoets.
//
// Ringtoets is free software: you can redistribute it and/or modify
// it under the terms of the GNU 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 General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
//
// All names, logos, and references to "Deltares" are registered trademarks of
// Stichting Deltares and remain full property of Stichting Deltares at all times.
// All rights reserved.

using System;
using System.Collections.Generic;
using System.IO;
using Core.Common.Base.IO;
using Core.Common.IO.Exceptions;
using Core.Common.Utils.Builders;
using log4net;
using Ringtoets.Common.Data.AssessmentSection;
using Ringtoets.Common.Data.Hydraulics;
using Ringtoets.Common.IO.HydraRing;
using Ringtoets.Common.IO.Properties;
using Ringtoets.HydraRing.IO.HydraulicBoundaryDatabaseContext;
using Ringtoets.HydraRing.IO.HydraulicLocationConfigurationDatabaseContext;

namespace Ringtoets.Common.IO.FileImporters
{
    /// <summary>
    /// Imports locations read from an Hydraulic boundary .sqlite file (SqlLite database file) to a 
    /// collection of <see cref="HydraulicBoundaryLocation"/> in a <see cref="HydraulicBoundaryDatabase"/>.
    /// </summary>
    public class HydraulicBoundaryDatabaseImporter : IDisposable
    {
        private readonly ILog log = LogManager.GetLogger(typeof(HydraulicBoundaryDatabaseImporter));
        private HydraulicBoundarySqLiteDatabaseReader hydraulicBoundaryDatabaseReader;
        private HydraulicLocationConfigurationSqLiteDatabaseReader hydraulicLocationConfigurationDatabaseReader;

        /// <summary>
        /// Creates a new instance of <see cref="HydraulicBoundaryDatabase"/>, based upon the data read from 
        /// the hydraulic boundary database file, and saved into <paramref name="targetItem"/>.
        /// </summary>
        /// <param name="targetItem"><see cref="IAssessmentSection"/> to set the newly 
        /// created <see cref="HydraulicBoundaryDatabase"/>.</param>
        /// <param name="filePath">The path of the hydraulic boundary database file to import from.</param>
        /// <returns><c>true</c> if the import was successful, <c>false</c> otherwise.</returns>
        /// <exception cref="ArgumentNullException">Thrown when <paramref name="targetItem"/> is <c>null</c>.</exception>
        /// <exception cref="CriticalFileReadException">Thrown when:
        /// <list type="bullet">
        /// <item>The given file at <paramref name="filePath"/> cannot be read.</item>
        /// <item>The file 'HLCD.sqlite' in the same folder as <paramref name="filePath"/> cannot be read.</item>
        /// </list>
        /// </exception>
        public bool Import(IAssessmentSection targetItem, string filePath)
        {
            if (targetItem == null)
            {
                throw new ArgumentNullException(nameof(targetItem));
            }
            ValidateAndConnectTo(filePath);

            HydraulicBoundaryDatabase hydraulicBoundaryDatabase = targetItem.HydraulicBoundaryDatabase;
            if (!IsImportRequired(hydraulicBoundaryDatabase))
            {
                bool isNotificationRequired = hydraulicBoundaryDatabase.FilePath != filePath;
                hydraulicBoundaryDatabase.FilePath = filePath;
                if (isNotificationRequired)
                {
                    targetItem.NotifyObservers();
                }
            }
            else
            {
                HydraulicBoundaryDatabase importResult = GetHydraulicBoundaryDatabase();

                if (importResult == null)
                {
                    return false;
                }

                AddImportedDataToModel(targetItem, importResult);
                log.Info(Resources.HydraulicBoundaryDatabaseImporter_Import_All_hydraulic_locations_read);
            }

            return true;
        }

        public void Dispose()
        {
            if (hydraulicBoundaryDatabaseReader != null)
            {
                hydraulicBoundaryDatabaseReader.Dispose();
                hydraulicBoundaryDatabaseReader = null;
            }
            if (hydraulicLocationConfigurationDatabaseReader != null)
            {
                hydraulicLocationConfigurationDatabaseReader.Dispose();
                hydraulicLocationConfigurationDatabaseReader = null;
            }
        }

        /// <summary>
        /// Validates the file and opens a connection.
        /// </summary>
        /// <param name="filePath">The path to the file to read.</param>
        /// <exception cref="CriticalFileReadException">Thrown when: 
        /// <list type="bullet">
        /// <item>The given file at <paramref name="filePath"/> cannot be read.</item>
        /// <item>The file 'HLCD.sqlite' in the same folder as <paramref name="filePath"/> cannot be read.</item>
        /// </list>
        /// </exception>
        private void ValidateAndConnectTo(string filePath)
        {
            hydraulicBoundaryDatabaseReader = new HydraulicBoundarySqLiteDatabaseReader(filePath);

            string hlcdFilePath = Path.Combine(Path.GetDirectoryName(filePath), "hlcd.sqlite");
            try
            {
                hydraulicLocationConfigurationDatabaseReader = new HydraulicLocationConfigurationSqLiteDatabaseReader(hlcdFilePath);
            }
            catch (CriticalFileReadException)
            {
                string message = new FileReaderErrorMessageBuilder(filePath).Build(Resources.HydraulicBoundaryDatabaseImporter_HLCD_sqlite_Not_Found);
                throw new CriticalFileReadException(message);
            }
            try
            {
                string settingsDatabaseFileName = HydraulicBoundaryDatabaseHelper.GetHydraulicBoundarySettingsDatabase(filePath);
                using (new DesignTablesSettingsProvider(settingsDatabaseFileName)) {}
                using (new TimeIntegrationSettingsProvider(settingsDatabaseFileName)) {}
                using (new NumericsSettingsProvider(settingsDatabaseFileName)) {}
            }
            catch (CriticalFileReadException e)
            {
                string errorMessage = string.Format(Resources.HydraulicBoundaryDatabaseImporter_Cannot_open_hydaulic_calculation_settings_file_0_,
                                                    e.Message);
                string message = new FileReaderErrorMessageBuilder(filePath).Build(errorMessage);
                throw new CriticalFileReadException(message);
            }
        }

        private bool IsImportRequired(HydraulicBoundaryDatabase hydraulicBoundaryDatabase)
        {
            return hydraulicBoundaryDatabase == null || hydraulicBoundaryDatabaseReader.GetVersion() != hydraulicBoundaryDatabase.Version;
        }

        private void HandleException(Exception e)
        {
            log.Error(e.Message, e);
        }

        private HydraulicBoundaryDatabase GetHydraulicBoundaryDatabase()
        {
            long trackId = GetTrackId();
            if (trackId == 0)
            {
                return null;
            }

            try
            {
                string hydraulicBoundaryDatabaseFilePath = hydraulicBoundaryDatabaseReader.Path;
                bool canUsePreprocessor = hydraulicLocationConfigurationDatabaseReader.GetCanUsePreprocessorByTrackId(trackId);

                HydraulicBoundaryDatabase hydraulicBoundaryDatabase = canUsePreprocessor
                                                                          ? new HydraulicBoundaryDatabase(true, Path.GetDirectoryName(hydraulicBoundaryDatabaseFilePath))
                                                                          : new HydraulicBoundaryDatabase();

                hydraulicBoundaryDatabase.FilePath = hydraulicBoundaryDatabaseFilePath;
                hydraulicBoundaryDatabase.Version = hydraulicBoundaryDatabaseReader.GetVersion();

                // Locations directory of HLCD location ids and HRD location ids
                Dictionary<long, long> locationIdsDictionary = hydraulicLocationConfigurationDatabaseReader.GetLocationIdsByTrackId(trackId);

                var filter = new HydraulicBoundaryLocationFilter(
                    HydraulicBoundaryDatabaseHelper.GetHydraulicBoundarySettingsDatabase(hydraulicBoundaryDatabase.FilePath));

                // Prepare query to fetch hrd locations
                hydraulicBoundaryDatabaseReader.PrepareReadLocation();
                while (hydraulicBoundaryDatabaseReader.HasNext)
                {
                    HrdLocation hrdLocation = hydraulicBoundaryDatabaseReader.ReadLocation();

                    long locationId;
                    locationIdsDictionary.TryGetValue(hrdLocation.HrdLocationId, out locationId);

                    if (filter.ShouldInclude(locationId))
                    {
                        var hydraulicBoundaryLocation = new HydraulicBoundaryLocation(
                            locationId,
                            hrdLocation.Name,
                            hrdLocation.LocationX,
                            hrdLocation.LocationY);
                        hydraulicBoundaryDatabase.Locations.Add(hydraulicBoundaryLocation);
                    }
                }
                return hydraulicBoundaryDatabase;
            }
            catch (Exception e) when (e is LineParseException || e is CriticalFileReadException)
            {
                HandleException(e);
                return null;
            }
        }

        private long GetTrackId()
        {
            try
            {
                return hydraulicBoundaryDatabaseReader.GetTrackId();
            }
            catch (Exception e)
            {
                if (e is LineParseException || e is CriticalFileReadException)
                {
                    HandleException(e);
                    return 0;
                }
                throw;
            }
        }

        private static void AddImportedDataToModel(IAssessmentSection assessmentSection, HydraulicBoundaryDatabase importedData)
        {
            assessmentSection.HydraulicBoundaryDatabase = importedData;
            assessmentSection.NotifyObservers();
        }
    }
}