71 lines
1.4 KiB
Go
71 lines
1.4 KiB
Go
|
package beebot
|
||
|
|
|
||
|
|
import (
|
||
|
|
"regexp"
|
||
|
|
"text/template"
|
||
|
|
|
||
|
|
"github.com/jmoiron/sqlx"
|
||
|
|
)
|
||
|
|
|
||
|
|
// Filter represents a comment response for the bot
|
||
|
|
type Filter struct {
|
||
|
|
ID int64
|
||
|
|
Name string
|
||
|
|
RegexStr string `db:"regex"`
|
||
|
|
Template string
|
||
|
|
|
||
|
|
regex *regexp.Regexp
|
||
|
|
db *sqlx.DB
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewFilter creates and saves a Filter
|
||
|
|
func (b *BeeBot) NewFilter(name, regex, tpl string) (*Filter, error) {
|
||
|
|
// Verify the template
|
||
|
|
_, err := template.New("tmp").Parse(tpl)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
_, err = regexp.Compile(regex)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
f := &Filter{
|
||
|
|
Name: name,
|
||
|
|
RegexStr: regex,
|
||
|
|
Template: tpl,
|
||
|
|
}
|
||
|
|
f.populate(b.db)
|
||
|
|
return f, nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (f *Filter) populate(db *sqlx.DB) {
|
||
|
|
f.regex = regexp.MustCompile(f.RegexStr)
|
||
|
|
f.db = db
|
||
|
|
}
|
||
|
|
|
||
|
|
// Save commits a template to the database
|
||
|
|
func (f *Filter) Save() error {
|
||
|
|
q := `insert into filters (name, regexstr, template) values (?, ? ,?)
|
||
|
|
on conflict(name) do update set regexstr=?, template=?;`
|
||
|
|
res, err := f.db.Exec(q, f.Name, f.RegexStr, f.Template,
|
||
|
|
f.RegexStr, f.Template)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
f.ID, err = res.LastInsertId()
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
// AllFilters returns every filter known
|
||
|
|
func (b *BeeBot) AllFilters() ([]Filter, error) {
|
||
|
|
filters := []Filter{}
|
||
|
|
err := b.db.Select(&filters, `select * from Filters`)
|
||
|
|
if handleErr(err, "could not get Filter list") {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
for _, f := range filters {
|
||
|
|
f.populate(b.db)
|
||
|
|
}
|
||
|
|
return filters, nil
|
||
|
|
}
|