package repository

import (
	"context"
	"fmt"

	"github.com/testcontainers/testcontainers-go"
	"github.com/testcontainers/testcontainers-go/wait"
)

const (
	MySQLUser     = "root"
	MySQLPassword = "password"
	MySQLDatabase = "testdb"
	MySQLPort     = "3306"
)

// 啟動 MySQL container
func startMySQLContainer() (host string, port string, dsn string, tearDown func(), err error) {
	ctx := context.Background()

	req := testcontainers.ContainerRequest{
		Image: "mysql:8.0",
		Env: map[string]string{
			"MYSQL_ROOT_PASSWORD": MySQLPassword,
			"MYSQL_DATABASE":      MySQLDatabase,
		},
		ExposedPorts: []string{"3306/tcp"},
		WaitingFor:   wait.ForListeningPort("3306/tcp"),
	}

	mysqlC, err := testcontainers.GenericContainer(ctx, testcontainers.GenericContainerRequest{
		ContainerRequest: req,
		Started:          true,
	})
	if err != nil {
		return "", "", "", nil, err
	}

	mappedPort, err := mysqlC.MappedPort(ctx, "3306")
	if err != nil {
		return "", "", "", nil, err
	}

	containerHost, err := mysqlC.Host(ctx)
	if err != nil {
		return "", "", "", nil, err
	}

	// 組成 DSN（Data Source Name）
	dsn = fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?parseTime=true",
		MySQLUser,
		MySQLPassword,
		containerHost,
		mappedPort.Port(),
		MySQLDatabase,
	)

	tearDown = func() {
		_ = mysqlC.Terminate(ctx)
	}

	fmt.Printf("MySQL ready at: %s\n", dsn)
	return containerHost, mappedPort.Port(), dsn, tearDown, nil
}