using System.Data.Common;
using Server.Exceptions;
using Server.Helper;
using System.Reflection;

namespace Server.Migrations;

public static class DatabaseMigrations
{
    public static void Migrate(SQLHelper helper)
    {
        int initialDatabaseVersion = DatabaseGetVersion(helper);
        int databaseVersion = initialDatabaseVersion;

        var updateMethods = typeof(DatabaseMigrations)
            .GetMethods(BindingFlags.Public | BindingFlags.Static)
            .Where(m => m.Name.StartsWith("UpdateFrom") && m.ReturnType == typeof(int))
            .OrderBy(m => int.Parse(m.Name["UpdateFrom".Length..]))
            .ToList();

        foreach (var method in updateMethods)
        {
            var version = int.Parse(method.Name["UpdateFrom".Length..]);
            if (version >= databaseVersion)
            {
                databaseVersion = (int)method.Invoke(null, new object[] { helper });
            }
        }

        if (databaseVersion != initialDatabaseVersion)
        {
            helper.ExecuteSQLNonQuery("UPDATE settings SET value = @databaseVersion", new() { ["databaseVersion"] = databaseVersion.ToString() });
        }
    }

    public static int DatabaseGetVersion(SQLHelper helper)
    {
        DbDataReader reader = helper.ExecuteSQLCommand("show tables", []);
        bool hasTables = reader.Read();
        reader.Close();
        if (!hasTables)
        {
            return 0;
        }

        reader = helper.ExecuteSQLCommand("show tables like '%settings%'", []);
        bool hasSystemTable = reader.Read();
        reader.Close();
        if (!hasSystemTable)
        {
            return 1;
        }
        reader = helper.ExecuteSQLCommand("SELECT value FROM settings WHERE name=\"DatabaseVersion\"", []);
        reader.Read();
        string rawVersion = reader.GetString(0);
        reader.Close();
        bool success = int.TryParse(rawVersion, out int version);
        if (!success)
        {
            throw new DatabaseVersionException();
        }
        return version;
    }

    public static int Create(SQLHelper helper)
    {
        helper.ExecuteSQLNonQuery("CREATE TABLE searchdomain (id int PRIMARY KEY auto_increment, name varchar(512), settings JSON);", []);
        helper.ExecuteSQLNonQuery("CREATE TABLE entity (id int PRIMARY KEY auto_increment, name varchar(512), probmethod varchar(128), id_searchdomain int, FOREIGN KEY (id_searchdomain) REFERENCES searchdomain(id));", []);
        helper.ExecuteSQLNonQuery("CREATE TABLE attribute (id int PRIMARY KEY auto_increment, id_entity int, attribute varchar(512), value longtext, FOREIGN KEY (id_entity) REFERENCES entity(id));", []);
        helper.ExecuteSQLNonQuery("CREATE TABLE datapoint (id int PRIMARY KEY auto_increment, name varchar(512), probmethod_embedding varchar(512), id_entity int, FOREIGN KEY (id_entity) REFERENCES entity(id));", []);
        helper.ExecuteSQLNonQuery("CREATE TABLE embedding (id int PRIMARY KEY auto_increment, id_datapoint int, model varchar(512), embedding blob, FOREIGN KEY (id_datapoint) REFERENCES datapoint(id));", []);
        return 1;
    }

    public static int UpdateFrom1(SQLHelper helper)
    {
        helper.ExecuteSQLNonQuery("CREATE TABLE settings (name varchar(512), value varchar(8192));", []);
        helper.ExecuteSQLNonQuery("INSERT INTO settings (name, value) VALUES (\"DatabaseVersion\", \"2\");", []);
        return 2;
    }

    public static int UpdateFrom2(SQLHelper helper)
    {
        helper.ExecuteSQLNonQuery("ALTER TABLE datapoint ADD hash VARCHAR(44);", []);
        helper.ExecuteSQLNonQuery("UPDATE datapoint SET hash='';", []);
        return 3;
    }

    public static int UpdateFrom3(SQLHelper helper)
    {
        helper.ExecuteSQLNonQuery("ALTER TABLE datapoint ADD COLUMN similaritymethod VARCHAR(512) NULL DEFAULT 'Cosine' AFTER probmethod_embedding", []);
        return 4;
    }

    public static int UpdateFrom4(SQLHelper helper)
    {
        helper.ExecuteSQLNonQuery("UPDATE searchdomain SET settings = JSON_SET(settings, '$.QueryCacheSize', 1000000) WHERE JSON_EXTRACT(settings, '$.QueryCacheSize') is NULL;", []); // Set QueryCacheSize to a default of 1000000
        return 5;
    }
}