Semaphore/db/sql/option.go

package sql

import (
	"database/sql"
	"errors"
	"fmt"
	"github.com/Masterminds/squirrel"
	"github.com/ansible-semaphore/semaphore/db"
	"regexp"
)

func (d *SqlDb) SetOption(key string, value string) error {
	_, err := d.getOption(key)

	if errors.Is(err, db.ErrNotFound) {
		_, err = d.insert(
			"key",
			"insert into `option` (`key`, `value`) values (?, ?)",
			key, value)
	} else if err == nil {
		_, err = d.exec("update `option` set `value`=? where `key`=?", value, key)
	}

	return err
}

func (d *SqlDb) GetOptions(params db.RetrieveQueryParams) (res map[string]string, err error) {
	var options []db.Option

	if params.Filter != "" {
		var m bool
		m, err = regexp.Match(`^(?:\w.)+$`, []byte(params.Filter))
		if err != nil {
			return
		}

		if !m {
			err = fmt.Errorf("invalid filter format")
			return
		}
	}

	err = d.getObjects(0, db.OptionProps, params, func(q squirrel.SelectBuilder) squirrel.SelectBuilder {
		if params.Filter == "" {
			return q
		}
		return q.Where("`key` = ? OR `key` LIKE ?", params.Filter, params.Filter+".%")
	}, &options)

	if err != nil {
		return
	}

	for _, opt := range options {
		res[opt.Key] = opt.Value
	}

	return
}

func (d *SqlDb) getOption(key string) (value string, err error) {
	q := squirrel.Select("*").
		From("`"+db.OptionProps.TableName+"`").
		Where("`key`=?", key)

	query, args, err := q.ToSql()

	if err != nil {
		return
	}

	var opt db.Option

	err = d.selectOne(&opt, query, args...)

	if errors.Is(err, sql.ErrNoRows) {
		err = db.ErrNotFound
	}

	value = opt.Value

	return
}

func (d *SqlDb) GetOption(key string) (value string, err error) {

	value, err = d.getOption(key)

	if errors.Is(err, db.ErrNotFound) {
		err = nil
	}

	return
}