package main


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

	_ "github.com/lib/pq" // PostgreSQL driver
	"code_runner/config"
)


func main() {
	// Connection string
	connStr := config.GetDSN()

	// Open database connection
	db, err := sql.Open("postgres", connStr)
	if err != nil {
		log.Fatal(err)
	}
	defer db.Close()

	// Test the connection
	err = db.Ping()
	if err != nil {
		log.Fatal(err)
	}

	fmt.Println("Connected to the database!")
	// Read DDL SQL statements to create DB
	stmt, err := os.ReadFile("db.sql")
	if err != nil {
		log.Fatal(err)
	}

	fmt.Printf("Executed statement: %s\n", string(stmt))
	_, err = db.Exec(string(stmt))
	if err != nil {
		log.Printf("Error executing statement: %s\nError: %v", stmt, err)
		log.Fatal(err)
		return // Exit if any statement fails
	}	

	fmt.Println("- Database schema created successfully!")
	fmt.Println("- Migration completed successfully")
	fmt.Println("- Added 10 languages")
	fmt.Println("- Added 5 Python tasks")

}