package dockertest

import (
	"database/sql"
	"errors"
	"fmt"
	"log"
	"net/url"
	"time"

	_ "github.com/lib/pq"
)

// SetupPostgreSQLContainer sets up a real PostgreSQL instance for testing purposes,
// using a Docker container. It returns the container ID and its IP address,
// or makes the test fail on error.
func SetupPostgreSQLContainer() (c ContainerID, ip string, port int, err error) {
	port = RandomPort()
	forward := fmt.Sprintf("%d:%d", port, 5432)
	if BindDockerToLocalhost != "" {
		forward = "127.0.0.1:" + forward
	}
	c, ip, err = SetupContainer(PostgresImageName, port, 15*time.Second, func() (string, error) {
		return run("--name", GenerateContainerID(), "-d", "-p", forward, "-e", fmt.Sprintf("POSTGRES_PASSWORD=%s", PostgresPassword), PostgresImageName)
	})
	return
}

// ConnectToPostgreSQL starts a PostgreSQL image and passes the database url to the connector callback.
func ConnectToPostgreSQL(tries int, delay time.Duration, connector func(url string) bool) (c ContainerID, err error) {
	c, ip, port, err := SetupPostgreSQLContainer()
	if err != nil {
		return c, fmt.Errorf("Could not set up PostgreSQL container: %v", err)
	}

	for try := 0; try <= tries; try++ {
		time.Sleep(delay)
		url := fmt.Sprintf("postgres://%s:%s@%s:%d/postgres?sslmode=disable", PostgresUsername, PostgresPassword, ip, port)
		if connector(url) {
			return c, nil
		}
		log.Printf("Try %d failed. Retrying.", try)
	}
	return c, errors.New("Could not set up PostgreSQL container.")
}

// SetUpPostgreDatabase connects postgre container with given $connectURL and also creates a new database named $databaseName
// A modified url used to connect the created database will be returned
func SetUpPostgreDatabase(databaseName, connectURL string) (modifiedURL string, err error) {
	db, err := sql.Open("postgres", connectURL)
	if err != nil {
		return "", err
	}
	defer db.Close()

	count := 0
	err = db.QueryRow(
		fmt.Sprintf("SELECT COUNT(*) FROM pg_catalog.pg_database WHERE datname = '%s' ;", databaseName)).
		Scan(&count)
	if err != nil {
		return "", err
	}
	if count == 0 {
		// not found for $databaseName, create it
		_, err = db.Exec(fmt.Sprintf("CREATE DATABASE %s", databaseName))
		if err != nil {
			return "", err
		}
	}

	// replace dbname in url
	// from: postgres://postgres:docker@192.168.99.100:9071/postgres?sslmode=disable
	// to: postgres://postgres:docker@192.168.99.100:9071/$databaseName?sslmode=disable
	u, err := url.Parse(connectURL)
	if err != nil {
		return "", err
	}
	u.Path = fmt.Sprintf("/%s", databaseName)
	return u.String(), nil
}