package database

import (
	"database/sql"
	"fmt"
	"log"
	"time"

	"gitea.wayfinderak.com/wayfinderak/geeksbot/pkg/database/migrations"
	"github.com/golang-migrate/migrate/v4"
	"github.com/golang-migrate/migrate/v4/database/postgres"
	bindata "github.com/golang-migrate/migrate/v4/source/go_bindata"
)

var (
	db             *sql.DB
	GuildService   guildService
	UserService    userService
	ChannelService channelService
	MessageService messageService
	RequestService requestService
	PatreonService patreonService
	ServerService  serverService
	RaffleService  raffleService
)

func ConnectDatabase(dbConnString string) {
	var err error
	db, err = sql.Open("postgres", dbConnString)
	if err != nil {
		log.Fatal(fmt.Errorf("can't connect to the database: %w", err))
	}

	db.SetMaxOpenConns(5)
	db.SetMaxIdleConns(2)
	db.SetConnMaxLifetime(30 * time.Minute)
	db.SetConnMaxIdleTime(5 * time.Minute)

	if err = db.Ping(); err != nil {
		log.Fatal(fmt.Errorf("can't ping the database: %w", err))
	}

	log.Println("Database Connected.")
	initServices()
	log.Println("Services Initialized")
}

func RunMigrations() {
	s := bindata.Resource(migrations.AssetNames(),
		func(name string) ([]byte, error) {
			return migrations.Asset(name)
		})
	d, err := bindata.WithInstance(s)
	if err != nil {
		log.Fatal(fmt.Errorf("cannot load migrations: %w", err))
	}
	instance, err := postgres.WithInstance(db, &postgres.Config{})
	if err != nil {
		log.Fatal(fmt.Errorf("cannot create db driver: %w", err))
	}
	m, err := migrate.NewWithInstance("go-bindatafoo", d, "postgres", instance)
	if err != nil {
		log.Fatal(fmt.Errorf("cannot create migration instance: %w", err))
	}
	err = m.Up()
	if err != nil {
		if err.Error() == "no change" {
			log.Println(err)
		} else {
			log.Fatal(err)
		}
	}
	log.Println("Migrations Run")

}

func initServices() {
	GuildService = guildService{db: db}
	UserService = userService{db: db}
	ChannelService = channelService{db: db}
	MessageService = messageService{db: db}
	PatreonService = patreonService{db: db}
	RequestService = requestService{db: db}
	ServerService = serverService{db: db}
	RaffleService = raffleService{db: db}
}