Colosseum/Assets/External/Models/SidekickCharacters/Scripts/Runtime/Database/DatabaseManager.cs

// Copyright (c) 2024 Synty Studios Limited. All rights reserved.
//
// Use of this software is subject to the terms and conditions of the Synty Studios End User Licence Agreement (EULA)
// available at: https://syntystore.com/pages/end-user-licence-agreement
//
// For additional details, see the LICENSE.MD file bundled with this software.

// using SqlCipher4Unity3D;

// using SqlCipher4Unity3D;

using SQLite;
using Synty.SidekickCharacters.Database.DTO;
using System;
using System.Linq;

namespace Synty.SidekickCharacters.Database
{
    /// <summary>
    ///     Manages connecting to, initializing and validating a local Sidekick SQLite database
    /// </summary>
    public class DatabaseManager
    {
        private static readonly string _DATABASE_PATH = "Assets/External/Models/SidekickCharacters/Database/Side_Kick_Data.db";
        private readonly string _CURRENT_VERSION = "1.0.2";

        private static SQLiteConnection _connection;
        private static int _connectionHash;

        /// <summary>
        ///     Gets the DB connection with the given connection details.
        ///     If the current connection is not for the given details, the connection is replaced with the correct connection.
        /// </summary>
        /// <param name="databasePath">The path to the DB.</param>
        /// <param name="connectionKey">The connection key for the DB.</param>
        /// <param name="checkDbOnLoad">Whether or not to valiate the structure of the DB after connecting to it.</param>
        /// <returns>A connection to a DB with the given connection details.</returns>
        public SQLiteConnection GetDbConnection(bool checkDbOnLoad = false)
        {
            if (_connection == null)
            {
                _connection = new SQLiteConnection(_DATABASE_PATH, true);
            }
            else
            {
                return _connection;
            }

            if (checkDbOnLoad)
            {
                bool createTables = !IsDatabaseConfigured();

                InitialiseDatabase(createTables);
            }

            return _connection;
        }

        /// <summary>
        ///     Returns whatever the current DB connection is, regardless of state.
        /// </summary>
        /// <returns>The current DB connection.</returns>
        public SQLiteConnection GetCurrentDbConnection()
        {
            return _connection;
        }

        /// <summary>
        ///     Closes the current DB connection.
        /// </summary>
        public void CloseConnection()
        {
            if (_connection == null)
            {
                return;
            }

            // NOTE: Previously, if we didn't clear the pool in the connection, it resulted in a file lock on the database,
            //       which in some cases could cause problems when trying to update the DB to a new version.
            //       Our SQLCipher library doesn't use pool clearing unless using SQLiteAsyncConnection.
            _connection.Dispose();
            // Our SQLCipher library doesn't surface checking connection state; disposed connections need their reference removed
            _connection = null;
        }

        /// <summary>
        ///     Initialises the Sidekicks database with required data, if they don't already exist.
        /// </summary>
        /// <param name="createTables">Whether to update the database and create the needed tables or not.</param>
        private void InitialiseDatabase(bool createTables = false)
        {
            // ensure we have a default color set (a bunch of other code relies on this, not safe to remove yet)
            try
            {
                SidekickColorSet.GetDefault(this);
            }
            catch (Exception)
            {
                SidekickColorSet newSet = new SidekickColorSet
                {
                    Species = new SidekickSpecies { ID = -1, Name = "None" },
                    Name = "Default",
                    SourceColorPath = "Assets/External/Models/SidekickCharacters/Resources/Textures/T_ColorMap.png",
                    SourceMetallicPath = "Assets/External/Models/SidekickCharacters/Resources/Textures/T_MetallicMap.png",
                    SourceSmoothnessPath = "Assets/External/Models/SidekickCharacters/Resources/Textures/T_SmoothnessMap.png",
                    SourceReflectionPath = "Assets/External/Models/SidekickCharacters/Resources/Textures/T_ReflectionMap.png",
                    SourceEmissionPath = "Assets/External/Models/SidekickCharacters/Resources/Textures/T_EmissionMap.png",
                    SourceOpacityPath = "Assets/External/Models/SidekickCharacters/Resources/Textures/T_OpacityMap.png",
                };

                newSet.Save(this);
            }

            if (createTables)
            {
                GetCurrentDbConnection().CreateTable<SidekickDBVersion>();

                SidekickDBVersion version;

                if (GetCurrentDbConnection().Table<SidekickDBVersion>().Any())
                {
                    version = GetCurrentDbConnection().Table<SidekickDBVersion>().FirstOrDefault();
                    version.SemanticVersion = _CURRENT_VERSION;
                    version.LastUpdated = DateTime.Now;
                }
                else
                {
                    version = new SidekickDBVersion()
                    {
                        ID = -1,
                        SemanticVersion = _CURRENT_VERSION,
                        LastUpdated = DateTime.Now
                    };
                }

                version.Save(this);

                GetCurrentDbConnection().CreateTable<SidekickPart>();
                GetCurrentDbConnection().CreateTable<SidekickPartImage>();
                GetCurrentDbConnection().CreateTable<SidekickPartFilter>();
                GetCurrentDbConnection().CreateTable<SidekickPartFilterRow>();
                GetCurrentDbConnection().CreateTable<SidekickPartPresetRow>();
                GetCurrentDbConnection().CreateTable<SidekickPartSpeciesLink>();
                GetCurrentDbConnection().CreateTable<SidekickPresetFilter>();
                GetCurrentDbConnection().CreateTable<SidekickPresetFilterRow>();
            }
        }

        /// <summary>
        ///     Checks to see if the current database has the required tables.
        ///     TODO: Check DB version, not just if table is present.
        /// </summary>
        /// <returns>True if the tables are present; otherwise false.</returns>
        private bool IsDatabaseConfigured()
        {
            bool configured = !(GetCurrentDbConnection().GetTableInfo("sk_vdata").Count < 1);

            if (GetCurrentDbConnection().GetTableInfo("sk_part").Count < 9)
            {
                configured = false;
            }

            if (GetCurrentDbConnection().GetTableInfo("sk_part_image").Count < 1)
            {
                configured = false;
            }

            if (GetCurrentDbConnection().GetTableInfo("sk_part_filter").Count < 1)
            {
                configured = false;
            }

            if (GetCurrentDbConnection().GetTableInfo("sk_part_filter_row").Count < 1)
            {
                configured = false;
            }

            if (GetCurrentDbConnection().GetTableInfo("sk_part_preset_row").Count < 5)
            {
                configured = false;
            }

            if (GetCurrentDbConnection().GetTableInfo("sk_part_species_link").Count < 3)
            {
                configured = false;
            }

            if (GetCurrentDbConnection().GetTableInfo("sk_preset_filter").Count < 1)
            {
                configured = false;
            }

            if (GetCurrentDbConnection().GetTableInfo("sk_preset_filter_row").Count < 1)
            {
                configured = false;
            }

            return configured;
        }

        /// <summary>
        ///     Retrieves the current database version.
        /// </summary>
        /// <returns>Semantic version (major.minor.patch).</returns>
        public Version GetDatabaseVersion()
        {
            return new Version(GetCurrentDbConnection()?.Table<SidekickDBVersion>().FirstOrDefault().SemanticVersion ?? "0.0.1ea");
        }
    }
}