[fix] bugfixes from testing

- sqlite wrapper func context bug

database/sqlx.go

@@ -2,7 +2,6 @@ package database
 
 import (
 	"context"
-	"database/sql"
 	"fmt"
 	"os"
 	"time"
@@ -32,9 +31,6 @@ type Database interface {
 	Close() error
 	//internals
 	timeout() (context.Context, context.CancelFunc)
-	query(query string, args ...any) (*sqlx.Rows, error)
-	queryRow(query string, args ...any) *sqlx.Row
-	exec(query string, args ...any) (sql.Result, error)
 	//command methods
 	GetCmd(name string) (string, error)
 	GetAllCmds() ([]model.Command, error)
@@ -84,8 +80,10 @@ func (d *database) Init() error {
 			active INTEGER NOT NULL DEFAULT 0
 		);
 	`
-
-	_, err := d.exec(query)
+	// _, err := d.exec(query)
+	ctx, cl := d.timeout()
+	defer cl()
+	_, err := d.ExecContext(ctx, query)
 	return err
 }
 
@@ -96,7 +94,9 @@ func (d *database) Destroy() error {
 		DROP TABLE servers;
 	`
 
-	_, err := d.exec(query)
+	ctx, cl := d.timeout()
+	defer cl()
+	_, err := d.ExecContext(ctx, query)
 	return err
 }
 
@@ -108,27 +108,6 @@ func (d *database) timeout() (context.Context, context.CancelFunc) {
 	return context.WithTimeout(context.Background(), DB_TIMEOUT)
 }
 
-func (d *database) query(query string, args ...any) (*sqlx.Rows, error) {
-	ctx, cl := d.timeout()
-	defer cl()
-
-	return d.QueryxContext(ctx, query, args...)
-}
-
-func (d *database) queryRow(query string, args ...any) *sqlx.Row {
-	ctx, cl := d.timeout()
-	defer cl()
-
-	return d.QueryRowxContext(ctx, query, args...)
-}
-
-func (d *database) exec(query string, args ...any) (sql.Result, error) {
-	ctx, cl := d.timeout()
-	defer cl()
-
-	return d.ExecContext(ctx, query, args...)
-}
-
 // gets command using name
 func (d *database) GetCmd(name string) (string, error) {
 	query := `
@@ -141,7 +120,9 @@ func (d *database) GetCmd(name string) (string, error) {
 	`
 
 	var cmd string
-	err := d.queryRow(query, name).Scan(&cmd)
+	ctx, cl := d.timeout()
+	defer cl()
+	err := d.QueryRowxContext(ctx, query, name).Scan(&cmd)
 	if err != nil {
 		return "", err
 	}
@@ -159,7 +140,9 @@ func (d *database) GetAllCmds() ([]model.Command, error) {
 			commands
 	`
 
-	rows, err := d.query(query)
+	ctx, cancel := d.timeout()
+	defer cancel()
+	rows, err := d.QueryxContext(ctx, query)
 	if err != nil {
 		return nil, err
 	}
@@ -186,7 +169,9 @@ func (d *database) SaveCmd(name, cmd string) error {
 		VALUES(?, ?)
 	`
 
-	_, err := d.exec(query, name, cmd)
+	ctx, cancel := d.timeout()
+	defer cancel()
+	_, err := d.ExecContext(ctx, query, name, cmd)
 	return err
 }
 
@@ -201,7 +186,9 @@ func (d *database) UpdateCmd(name, cmd string) error {
 			name = ?
 	`
 
-	_, err := d.exec(query, cmd, name)
+	ctx, cancel := d.timeout()
+	defer cancel()
+	_, err := d.ExecContext(ctx, query, cmd, name)
 	return err
 }
 
@@ -211,7 +198,9 @@ func (d *database) DeleteCmd(name string) error {
 		WHERE name = ?
 	`
 
-	_, err := d.exec(query, name)
+	ctx, cancel := d.timeout()
+	defer cancel()
+	_, err := d.ExecContext(ctx, query, name)
 	return err
 }
 
@@ -221,15 +210,18 @@ func (d *database) GetServer(name string) (model.Server, error) {
 			name,
 			server,
 			password,
-			port
+			port,
+			active
 		FROM
 			servers
 		WHERE
 			name = ?
 	`
 
+	ctx, cancel := d.timeout()
+	defer cancel()
 	var s model.Server
-	err := d.queryRow(query, name).StructScan(&s)
+	err := d.QueryRowxContext(ctx, query, name).StructScan(&s)
 	if err != nil {
 		return model.Server{}, err
 	}
@@ -242,15 +234,18 @@ func (d *database) GetActiveServer() (model.Server, error) {
 			name,
 			server,
 			password,
-			port
+			port,
+			active
 		FROM
 			servers
 		WHERE
 			active = 1
 	`
 
+	ctx, cancel := d.timeout()
+	defer cancel()
 	var s model.Server
-	err := d.queryRow(query).StructScan(&s)
+	err := d.QueryRowxContext(ctx, query).StructScan(&s)
 	return s, err
 }
 
@@ -260,12 +255,15 @@ func (d *database) GetAllServers() ([]model.Server, error) {
 			name,
 			server,
 			password,
-			port
+			port,
+			active
 		FROM
 			servers
 	`
 
-	rows, err := d.query(query)
+	ctx, cancel := d.timeout()
+	defer cancel()
+	rows, err := d.QueryxContext(ctx, query)
 	if err != nil {
 		return nil, err
 	}
@@ -291,26 +289,39 @@ func (d *database) SaveServer(srv model.Server) error {
 		VALUES(?, ?, ?, ?)
 	`
 
-	_, err := d.exec(query, srv.Name, srv.Server, srv.Password, srv.Port)
+	ctx, cancel := d.timeout()
+	defer cancel()
+	_, err := d.ExecContext(ctx, query, srv.Name, srv.Server, srv.Password, srv.Port)
 	return err
 }
 
 func (d *database) SetActiveServer(name string) error {
-	query := `
+	clear := `
 		UPDATE
 			servers
 		SET
-			active = 0;
-
-		UPDATE
-			servers
-		SET
-			active = 1
-		WHERE
-			name = ?;
+			active = 0
 	`
 
-	_, err := d.exec(query, name)
+	clrctx, clrcancel := d.timeout()
+	_, err := d.ExecContext(clrctx, clear)
+	if err != nil {
+		return err
+	}
+	defer clrcancel()
+
+	update := `
+		UPDATE 
+			servers
+		SET 
+			active = 1
+		WHERE 
+			name = ?
+	`
+
+	ctx, cancel := d.timeout()
+	defer cancel()
+	_, err = d.ExecContext(ctx, update)
 	return err
 }
 
@@ -324,7 +335,9 @@ func (d *database) UpdateServer(name, password string) error {
 			name = ?
 	`
 
-	_, err := d.exec(query, password, name)
+	ctx, cancel := d.timeout()
+	defer cancel()
+	_, err := d.ExecContext(ctx, query, password, name)
 	return err
 }
 
@@ -335,6 +348,8 @@ func (d *database) DeleteServer(name string) error {
 			name = ?
 	`
 
-	_, err := d.exec(query, name)
+	ctx, cancel := d.timeout()
+	defer cancel()
+	_, err := d.ExecContext(ctx, query, name)
 	return err
 }