first commit

database/database.go

@@ -0,0 +1,49 @@
+package database
+
+import (
+	_ "embed"
+	"log"
+
+	_ "github.com/go-sql-driver/mysql"
+	"github.com/jmoiron/sqlx"
+	_ "github.com/lib/pq"
+	"github.com/spf13/viper"
+	_ "modernc.org/sqlite"
+)
+
+var (
+	//go:embed init/users.sql
+	userTable string
+	//go:embed init/otp.sql
+	otpTable string
+)
+
+func InitDB() (*sqlx.DB, error) {
+	driver := viper.GetString("database.driver")
+	dsn := viper.GetString("database.dsn")
+
+	db, err := sqlx.Open(driver, dsn)
+	if err != nil {
+		return nil, err
+	}
+
+	if err := db.Ping(); err != nil {
+		return nil, err
+	}
+
+	log.Println("Connected to database!")
+	return db, nil
+}
+
+func MigrateDB(db *sqlx.DB) error {
+	_, err := db.Exec(userTable)
+	if err != nil {
+		return err
+	}
+
+	_, err = db.Exec(otpTable)
+	if err != nil {
+		return err
+	}
+	return nil
+}