using MycroForge.CLI.CodeGen;

namespace MycroForge.CLI.Features;

public sealed class Orm : IFeature
{
    #region Defaults

    private static readonly string[] Settings =
    [
        "connectionstring = \"mysql+asyncmy://root:root@localhost:3306/example\"",
        "",
        "class OrmSettings:",
        "\tdef get_connectionstring() -> str:",
        "\t\treturn connectionstring"
    ];

    private static readonly string[] AsyncSession =
    [
        "from sqlalchemy.ext.asyncio import create_async_engine, AsyncEngine, AsyncSession",
        "from orm.settings import OrmSettings",
        "",
        "async_engine: AsyncEngine = create_async_engine(OrmSettings.get_connectionstring())",
        "",
        "def async_session():",
        "\treturn AsyncSession(async_engine, expire_on_commit=False)"
    ];

    private static readonly string[] EntityBase =
    [
        "from sqlalchemy.orm import DeclarativeBase",
        "",
        "class EntityBase(DeclarativeBase):",
        "\tpass"
    ];

    private static readonly string[] User =
    [
        "from sqlalchemy import String",
        "from sqlalchemy.orm import Mapped, mapped_column",
        "from orm.entities.entity_base import EntityBase",
        "",
        "class User(EntityBase):",
        "\t__tablename__ = \"users\"",
        "\tid: Mapped[int] = mapped_column(primary_key=True)",
        "\tfirstname: Mapped[str] = mapped_column(String(255))",
        "\tlastname: Mapped[str] = mapped_column(String(255))",
        "",
        "\tdef __repr__(self) -> str:",
        "\t\treturn f\"User(id={self.id!r}, firstname={self.firstname!r}, lastname={self.lastname!r})\""
    ];

    #endregion

    public const string FeatureName = "orm";

    public string Name => FeatureName;

    public async Task ExecuteAsync(ProjectContext context)
    {
        var config = await context.LoadConfig();

        await context.Bash(
            "source .venv/bin/activate",
            "python3 -m pip install asyncmy sqlalchemy alembic",
            "python3 -m pip freeze > requirements.txt",
            "alembic init -t async orm"
        );

        var env = await context.ReadFile("orm/env.py");
        env = new OrmEnvInitializer(env).Rewrite();
        env = new OrmEnvUpdater(env, "user", "User").Rewrite();
        await context.WriteFile("orm/env.py", env);

        await context.CreateFile("orm/settings.py", Settings);

        await context.CreateFile("orm/engine/async_session.py", AsyncSession);

        await context.CreateFile("orm/entities/entity_base.py", EntityBase);

        await context.CreateFile("orm/entities/user.py", User);

        await context.SaveConfig(config);
    }
}