package server

import (
	"context"

	"tss-rocks-be/ent"
	"tss-rocks-be/internal/config"

	_ "github.com/mattn/go-sqlite3"
	"github.com/rs/zerolog/log"
)

// NewEntClient creates a new ent client
func NewEntClient(cfg *config.Config) *ent.Client {
	// 使用配置文件中的数据库设置
	client, err := ent.Open(cfg.Database.Driver, cfg.Database.DSN)
	if err != nil {
		log.Fatal().Err(err).Msg("Failed to connect to database")
	}

	// Run the auto migration tool
	if err := client.Schema.Create(context.Background()); err != nil {
		log.Fatal().Err(err).Msg("Failed to create schema resources")
	}

	return client
}