mirror of
https://github.com/documize/community.git
synced 2025-07-19 13:19:43 +02:00
135 lines
3.3 KiB
Go
135 lines
3.3 KiB
Go
// Copyright 2016 Documize Inc. <legal@documize.com>. All rights reserved.
|
|
//
|
|
// This software (Documize Community Edition) is licensed under
|
|
// GNU AGPL v3 http://www.gnu.org/licenses/agpl-3.0.en.html
|
|
//
|
|
// You can operate outside the AGPL restrictions by purchasing
|
|
// Documize Enterprise Edition and obtaining a commercial license
|
|
// by contacting <sales@documize.com>.
|
|
//
|
|
// https://documize.com
|
|
|
|
package mysql
|
|
|
|
import (
|
|
"bytes"
|
|
"database/sql"
|
|
"fmt"
|
|
|
|
"github.com/documize/community/core/env"
|
|
"github.com/documize/community/core/streamutil"
|
|
"github.com/pkg/errors"
|
|
)
|
|
|
|
// Scope provides data access to MySQL.
|
|
type Scope struct {
|
|
Runtime *env.Runtime
|
|
}
|
|
|
|
// Get fetches a configuration JSON element from the config table.
|
|
func (s Scope) Get(area, path string) (value string) {
|
|
if path != "" {
|
|
path = "." + path
|
|
}
|
|
sql := "SELECT JSON_EXTRACT(`config`,'$" + path + "') FROM `config` WHERE `key` = '" + area + "';"
|
|
|
|
stmt, err := s.Runtime.Db.Preparex(sql)
|
|
defer streamutil.Close(stmt)
|
|
|
|
if err != nil {
|
|
s.Runtime.Log.Error(fmt.Sprintf("setting.Get %s %s", area, path), err)
|
|
return ""
|
|
}
|
|
|
|
var item = make([]uint8, 0)
|
|
|
|
err = stmt.Get(&item)
|
|
if err != nil {
|
|
s.Runtime.Log.Error(fmt.Sprintf("setting.Get %s %s", area, path), err)
|
|
return ""
|
|
}
|
|
|
|
if len(item) > 1 {
|
|
q := []byte(`"`)
|
|
value = string(bytes.TrimPrefix(bytes.TrimSuffix(item, q), q))
|
|
}
|
|
|
|
return value
|
|
}
|
|
|
|
// Set writes a configuration JSON element to the config table.
|
|
func (s Scope) Set(area, json string) error {
|
|
if area == "" {
|
|
return errors.New("no area")
|
|
}
|
|
|
|
sql := "INSERT INTO `config` (`key`,`config`) " +
|
|
"VALUES ('" + area + "','" + json +
|
|
"') ON DUPLICATE KEY UPDATE `config`='" + json + "';"
|
|
|
|
stmt, err := s.Runtime.Db.Preparex(sql)
|
|
defer streamutil.Close(stmt)
|
|
|
|
if err != nil {
|
|
err = errors.Wrap(err, "failed to save global config value")
|
|
return err
|
|
}
|
|
|
|
_, err = stmt.Exec()
|
|
return err
|
|
}
|
|
|
|
// GetUser fetches a configuration JSON element from the userconfig table for a given orgid/userid combination.
|
|
// Errors return the empty string. A blank path returns the whole JSON object, as JSON.
|
|
func (s Scope) GetUser(orgID, userID, area, path string) (value string) {
|
|
if path != "" {
|
|
path = "." + path
|
|
}
|
|
|
|
qry := "SELECT JSON_EXTRACT(`config`,'$" + path + "') FROM `userconfig` WHERE `key` = '" + area +
|
|
"' AND `orgid` = '" + orgID + "' AND `userid` = '" + userID + "';"
|
|
|
|
stmt, err := s.Runtime.Db.Preparex(qry)
|
|
defer streamutil.Close(stmt)
|
|
|
|
if err != nil {
|
|
return ""
|
|
}
|
|
|
|
var item = make([]uint8, 0)
|
|
|
|
err = stmt.Get(&item)
|
|
if err != nil && err != sql.ErrNoRows {
|
|
s.Runtime.Log.Error(fmt.Sprintf("setting.GetUser for user %s %s %s", userID, area, path), err)
|
|
return ""
|
|
}
|
|
|
|
if len(item) > 1 {
|
|
q := []byte(`"`)
|
|
value = string(bytes.TrimPrefix(bytes.TrimSuffix(item, q), q))
|
|
}
|
|
|
|
return value
|
|
}
|
|
|
|
// SetUser writes a configuration JSON element to the userconfig table for the current user.
|
|
func (s Scope) SetUser(orgID, userID, area, json string) error {
|
|
if area == "" {
|
|
return errors.New("no area")
|
|
}
|
|
|
|
sql := "INSERT INTO `userconfig` (`orgid`,`userid`,`key`,`config`) " +
|
|
"VALUES ('" + orgID + "','" + userID + "','" + area + "','" + json +
|
|
"') ON DUPLICATE KEY UPDATE `config`='" + json + "';"
|
|
|
|
stmt, err := s.Runtime.Db.Preparex(sql)
|
|
defer streamutil.Close(stmt)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
_, err = stmt.Exec()
|
|
|
|
return err
|
|
}
|