package tubsql

import (
	"context"
	"database/sql"
	"log"
	"strings"

	_ "github.com/mattn/go-sqlite3"
)

func Connect_database() *sql.DB {
	db, err := sql.Open("sqlite3", "./test.db")
	if err != nil {
		log.Fatal(err)
	}

	return db
}

func Init_table(db *sql.DB, table_name string, table_data []string) {
	tx, err := db.Begin()
	if err != nil {
		log.Fatal(err)
	}
	ctx := context.Background()

	_, err = tx.ExecContext(
		ctx,
		`CREATE TABLE IF NOT EXISTS ` + table_name + ` ( 
			` + strings.Join(table_data, ", ") + `
		)`,
	)
	if err != nil {
		tx.Rollback()
		log.Fatal(err)
	}

	err = tx.Commit()
	if err != nil {
		tx.Rollback()
		log.Fatal(err)
	}
}