mirror of
https://github.com/abhinavxd/libredesk.git
synced 2026-09-22 02:23:32 +00:00
Abhinav Raut
108 lines
2.4 KiB
Go
108 lines
2.4 KiB
Go
// Package setting handles the management of application settings.
|
|
package setting
|
|
|
|
import (
|
|
"embed"
|
|
"encoding/json"
|
|
|
|
"github.com/abhinavxd/artemis/internal/dbutil"
|
|
"github.com/abhinavxd/artemis/internal/envelope"
|
|
"github.com/abhinavxd/artemis/internal/setting/models"
|
|
"github.com/jmoiron/sqlx"
|
|
"github.com/jmoiron/sqlx/types"
|
|
"github.com/zerodha/logf"
|
|
)
|
|
|
|
var (
|
|
//go:embed queries.sql
|
|
efs embed.FS
|
|
)
|
|
|
|
// Manager handles setting-related operations.
|
|
type Manager struct {
|
|
q queries
|
|
lo *logf.Logger
|
|
}
|
|
|
|
// Opts contains options for initializing the Manager.
|
|
type Opts struct {
|
|
DB *sqlx.DB
|
|
Lo *logf.Logger
|
|
}
|
|
|
|
// queries contains prepared SQL queries.
|
|
type queries struct {
|
|
GetAll *sqlx.Stmt `query:"get-all"`
|
|
Update *sqlx.Stmt `query:"update"`
|
|
GetByPrefix *sqlx.Stmt `query:"get-by-prefix"`
|
|
}
|
|
|
|
// New creates and returns a new instance of the Manager.
|
|
func New(opts Opts) (*Manager, error) {
|
|
var q queries
|
|
|
|
if err := dbutil.ScanSQLFile("queries.sql", &q, opts.DB, efs); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &Manager{
|
|
q: q,
|
|
lo: opts.Lo,
|
|
}, nil
|
|
}
|
|
|
|
// GetAll retrieves all settings as a models.Settings struct.
|
|
func (m *Manager) GetAll() (models.Settings, error) {
|
|
var (
|
|
b types.JSONText
|
|
out models.Settings
|
|
)
|
|
|
|
if err := m.q.GetAll.Get(&b); err != nil {
|
|
return out, err
|
|
}
|
|
|
|
if err := json.Unmarshal([]byte(b), &out); err != nil {
|
|
return out, err
|
|
}
|
|
|
|
return out, nil
|
|
}
|
|
|
|
// GetAllJSON retrieves all settings as JSON.
|
|
func (m *Manager) GetAllJSON() (types.JSONText, error) {
|
|
var b types.JSONText
|
|
|
|
if err := m.q.GetAll.Get(&b); err != nil {
|
|
return b, err
|
|
}
|
|
|
|
return b, nil
|
|
}
|
|
|
|
// Update updates settings.
|
|
func (m *Manager) Update(s interface{}) error {
|
|
// Marshal settings.
|
|
b, err := json.Marshal(s)
|
|
if err != nil {
|
|
return envelope.NewError(envelope.GeneralError, "Error updating settings", nil)
|
|
}
|
|
|
|
// Update the settings in the DB.
|
|
if _, err := m.q.Update.Exec(b); err != nil {
|
|
return envelope.NewError(envelope.GeneralError, "Error updating settings", nil)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// GetByPrefix retrieves settings by prefix as JSON.
|
|
func (m *Manager) GetByPrefix(prefix string) (types.JSONText, error) {
|
|
var b types.JSONText
|
|
if err := m.q.GetByPrefix.Get(&b, prefix+"%"); err != nil {
|
|
m.lo.Error("error fetching settings", "prefix", prefix, "error", err)
|
|
return b, envelope.NewError(envelope.GeneralError, "Error fetching settings", nil)
|
|
}
|
|
return b, nil
|
|
}
|