package main

import (
	"gorm.io/driver/postgres"
	"gorm.io/gorm"
)

type Task struct {
	gorm.Model
	Title string `json:"title"`
}

func InitDB() *gorm.DB {
	dsn := "host=gopostgres-vlcjiq user=postgres password=postgres dbname=go_postgres port=5432 sslmode=disable"
	db, err := gorm.Open(postgres.Open(dsn), &gorm.Config{})
	if err != nil {
		panic("Failed to connect to database")
	}
	// This automatically creates the 'tasks' table in Postgres
	db.AutoMigrate(&Task{})
	return db
}