Update application UI and functionality

server/unified-management/internal/db/audit_store.go

@@ -1,6 +1,7 @@
 package db
 
 import (
+	"context"
 	"fmt"
 	"strings"
 	"time"
@@ -170,10 +171,18 @@ func (s *Store) RecentSourceCalls(limit int) ([]map[string]any, error) {
 }
 
 func (s *Store) InsertAudit(log AuditLog) error {
+	return s.InsertAuditContext(context.Background(), log)
+}
+
+func (s *Store) InsertAuditContext(ctx context.Context, log AuditLog) error {
 	if log.CreatedAt == "" {
 		log.CreatedAt = Now()
 	}
-	_, err := s.exec(`INSERT INTO audit_logs (actor, type, target, message, ip, user_agent, created_at) VALUES (?, ?, ?, ?, ?, ?, ?)`,
+	conn, d := s.active()
+	if conn == nil {
+		return fmt.Errorf("database is not available")
+	}
+	_, err := conn.ExecContext(ctx, d.rebind(`INSERT INTO audit_logs (actor, type, target, message, ip, user_agent, created_at) VALUES (?, ?, ?, ?, ?, ?, ?)`),
 		sanitize(log.Actor), sanitize(log.Type), sanitize(log.Target), sanitize(log.Message), sanitize(log.IP), sanitize(log.UserAgent), log.CreatedAt)
 	return err
 }