use sea_orm_migration::{prelude::*, schema::*};

#[derive(DeriveMigrationName)]
pub struct Migration;

#[async_trait::async_trait]
impl MigrationTrait for Migration {
    async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
        manager
            .create_table(
                Table::create()
                    .table(Alias::new("ble_gateways"))
                    .col(uuid("id").primary_key())
                    .col(uuid("tenant_id").not_null())
                    .col(string("gateway_id").unique_key().not_null())
                    .col(string("name").not_null())
                    .col(string("api_key_hash").not_null())
                    .col(string("api_key_prefix").not_null())
                    .col(string("status").default("active").not_null())
                    .col(string("firmware_version").null())
                    .col(string("ip_address").null())
                    .col(timestamp_with_time_zone("last_heartbeat_at").null())
                    .col(json("metadata").null())
                    .col(timestamp_with_time_zone("created_at").default(Expr::current_timestamp()).not_null())
                    .col(timestamp_with_time_zone("updated_at").default(Expr::current_timestamp()).not_null())
                    .col(uuid("created_by").null())
                    .col(uuid("updated_by").null())
                    .col(timestamp_with_time_zone("deleted_at").null())
                    .col(integer("version").default(1).not_null())
                    .to_owned(),
            )
            .await?;

        manager
            .create_table(
                Table::create()
                    .table(Alias::new("gateway_patient_bindings"))
                    .col(uuid("id").primary_key())
                    .col(uuid("tenant_id").not_null())
                    .col(uuid("gateway_id_fk").not_null())
                    .col(uuid("patient_id").not_null())
                    .col(string("peripheral_mac").null())
                    .col(string("device_type").null())
                    .col(string("status").default("active").not_null())
                    .col(timestamp_with_time_zone("created_at").default(Expr::current_timestamp()).not_null())
                    .col(timestamp_with_time_zone("updated_at").default(Expr::current_timestamp()).not_null())
                    .col(uuid("created_by").null())
                    .col(uuid("updated_by").null())
                    .col(timestamp_with_time_zone("deleted_at").null())
                    .col(integer("version").default(1).not_null())
                    .to_owned(),
            )
            .await?;

        // 索引
        manager
            .create_index(
                Index::create()
                    .name("idx_ble_gateways_tenant_id")
                    .table(Alias::new("ble_gateways"))
                    .col(Alias::new("tenant_id"))
                    .to_owned(),
            )
            .await?;

        manager
            .create_index(
                Index::create()
                    .name("idx_ble_gateways_api_key_prefix")
                    .table(Alias::new("ble_gateways"))
                    .col(Alias::new("api_key_prefix"))
                    .to_owned(),
            )
            .await?;

        manager
            .create_index(
                Index::create()
                    .name("idx_gateway_patient_bindings_gateway")
                    .table(Alias::new("gateway_patient_bindings"))
                    .col(Alias::new("gateway_id_fk"))
                    .to_owned(),
            )
            .await?;

        manager
            .create_index(
                Index::create()
                    .name("idx_gateway_patient_bindings_patient")
                    .table(Alias::new("gateway_patient_bindings"))
                    .col(Alias::new("patient_id"))
                    .to_owned(),
            )
            .await?;

        // 外键约束
        manager
            .create_foreign_key(
                ForeignKey::create()
                    .name("fk_gpb_gateway")
                    .from(Alias::new("gateway_patient_bindings"), Alias::new("gateway_id_fk"))
                    .to(Alias::new("ble_gateways"), Alias::new("id"))
                    .on_delete(ForeignKeyAction::Cascade)
                    .to_owned(),
            )
            .await?;

        manager
            .create_foreign_key(
                ForeignKey::create()
                    .name("fk_gpb_patient")
                    .from(Alias::new("gateway_patient_bindings"), Alias::new("patient_id"))
                    .to(Alias::new("patients"), Alias::new("id"))
                    .on_delete(ForeignKeyAction::Cascade)
                    .to_owned(),
            )
            .await?;

        Ok(())
    }

    async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
        manager
            .drop_table(Table::drop().table(Alias::new("gateway_patient_bindings")).to_owned())
            .await?;
        manager
            .drop_table(Table::drop().table(Alias::new("ble_gateways")).to_owned())
            .await?;
        Ok(())
    }
}