package server

import (
    "context"
    "tss-rocks-be/ent"

    "github.com/rs/zerolog/log"
)

func InitDatabase(ctx context.Context, driver, dsn string) (*ent.Client, error) {
    client, err := ent.Open(driver, dsn)
    if err != nil {
        log.Error().Err(err).Msg("failed opening database connection")
        return nil, err
    }

    // Run the auto migration tool
    if err := client.Schema.Create(ctx); err != nil {
        log.Error().Err(err).Msg("failed creating schema resources")
        return nil, err
    }

    return client, nil
}