implement dbhandler package and new test file

This commit is contained in:
Adrian Zürcher
2025-10-31 08:11:07 +01:00
parent 1568ee2482
commit 80675ed328
17 changed files with 674 additions and 572 deletions

View File

@@ -3,52 +3,36 @@ package handlers
import (
"crypto/hmac"
"crypto/sha256"
"database/sql"
"encoding/hex"
"fmt"
"os"
"path/filepath"
_ "modernc.org/sqlite"
"gitea.tecamino.com/paadi/dbHandler"
"gitea.tecamino.com/paadi/tecamino-logger/logging"
)
type DatabaseHandler struct {
database *sql.DB
database *dbHandler.DBHandler
token []byte
}
func NewDatabaseHandler(file string, create bool) (*DatabaseHandler, error) {
if create {
// createOrOpenDB creates or opens a SQLite database at the given path.
// Create the directory if it doesn't exist
dir := filepath.Dir(file)
func NewDatabaseHandler(name, path string, logger *logging.Logger) (*DatabaseHandler, error) {
err := os.MkdirAll(dir, os.ModePerm)
if err != nil {
return nil, fmt.Errorf("failed to create directory: %w", err)
}
} else {
if _, err := os.Stat(file); err != nil {
return nil, fmt.Errorf("%s not found", file)
}
}
// Open the database (creates it if it doesn't exist)
db, err := sql.Open("sqlite", file)
database, err := dbHandler.NewDBHandler(name, path, logger)
if err != nil {
return nil, fmt.Errorf("failed to open database: %w", err)
return nil, err
}
// Test the connection
if err = db.Ping(); err != nil {
return nil, fmt.Errorf("failed to connect to database: %w", err)
}
return &DatabaseHandler{database: db}, nil
return &DatabaseHandler{database: database}, nil
}
func (dh *DatabaseHandler) SetToken(token string) {
dh.token = []byte(token)
func (dH *DatabaseHandler) DatabaseOpened() bool {
return dH.database != nil
}
func (dH *DatabaseHandler) AddNewTable(model any) error {
return dH.database.AddNewTable(model)
}
func (dH *DatabaseHandler) AddNewColum(model any) error {
return dH.database.AddNewColum(model)
}
func (dh *DatabaseHandler) hashField(field string) string {
@@ -56,3 +40,7 @@ func (dh *DatabaseHandler) hashField(field string) string {
h.Write([]byte(field))
return hex.EncodeToString(h.Sum(nil))
}
func (dh *DatabaseHandler) SetToken(token string) {
dh.token = []byte(token)
}