package database

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

	"github.com/jinzhu/gorm"
)

// Postgres database driver
import _ "github.com/jinzhu/gorm/dialects/postgres"

// Db is the open Database
var Db *gorm.DB

// URL defines the structure of a shortened URL
type URL struct {
	ID        uint `gorm:"primary_key:auto_increment"`
	CreatedAt time.Time
	UpdatedAt time.Time

	Short sql.NullString `gorm:"unique_index:short_url;not null"`
	Long  sql.NullString `gorm:"unique;not null"`
	Hits  sql.NullInt64
}

// InitDb opens a database connection and runs the auto migration
func InitDb(DBType, DBConnection string) {
	var err error
	Db, err = gorm.Open(DBType, DBConnection)
	if err != nil {
		log.Fatal("Could not open database:", err)
	}
	if os.Getenv("GIN_MODE") != "release" {
		Db.LogMode(true)
	}

	Db.AutoMigrate(&URL{})
}