Compare commits
12 commits
3f133e152a
...
36c982473c
Author | SHA1 | Date | |
---|---|---|---|
36c982473c | |||
bb9ff7234c | |||
1729caf06a | |||
95193685c2 | |||
e4e819ee90 | |||
2615da9ef8 | |||
2153dc8434 | |||
6ea61bf820 | |||
bdee792d75 | |||
d0c398f6f7 | |||
c423749f26 | |||
3b8a50d3e8 |
20 changed files with 640 additions and 340 deletions
|
@ -33,5 +33,11 @@ alerting:
|
||||||
renotify: 30m
|
renotify: 30m
|
||||||
notify:
|
notify:
|
||||||
- provider: slackwebhook
|
- provider: slackwebhook
|
||||||
|
# subjectTemplate: "Stillbox Alert ({{ highest . }})"
|
||||||
|
# bodyTemplate: |
|
||||||
|
# {{ range . -}}
|
||||||
|
# {{ .TGName }} is active with a score of {{ f .Score.Score 4 }}! ({{ f .Score.RecentCount 0 }}/{{ .Score.Count }} recent calls)
|
||||||
|
#
|
||||||
|
# {{ end -}}
|
||||||
config:
|
config:
|
||||||
webhookURL: "http://somewhere"
|
webhookURL: "http://somewhere"
|
||||||
|
|
|
@ -1,9 +1,6 @@
|
||||||
package common
|
package common
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
|
||||||
"strconv"
|
|
||||||
|
|
||||||
"github.com/spf13/cobra"
|
"github.com/spf13/cobra"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -47,10 +44,3 @@ func PtrOrNull[T comparable](val T) *T {
|
||||||
|
|
||||||
return &val
|
return &val
|
||||||
}
|
}
|
||||||
|
|
||||||
func FmtFloat(v float64, places ...int) string {
|
|
||||||
if len(places) > 0 {
|
|
||||||
return fmt.Sprintf("%."+strconv.Itoa(places[0])+"f", v)
|
|
||||||
}
|
|
||||||
return fmt.Sprintf("%.4f", v)
|
|
||||||
}
|
|
||||||
|
|
48
internal/common/template.go
Normal file
48
internal/common/template.go
Normal file
|
@ -0,0 +1,48 @@
|
||||||
|
package common
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"strconv"
|
||||||
|
"text/template"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"dynatron.me/x/stillbox/internal/jsontime"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
FuncMap = template.FuncMap{
|
||||||
|
"f": fmtFloat,
|
||||||
|
"dict": func(values ...interface{}) (map[string]interface{}, error) {
|
||||||
|
if len(values)%2 != 0 {
|
||||||
|
return nil, errors.New("invalid dict call")
|
||||||
|
}
|
||||||
|
dict := make(map[string]interface{}, len(values)/2)
|
||||||
|
for i := 0; i < len(values); i += 2 {
|
||||||
|
key, ok := values[i].(string)
|
||||||
|
if !ok {
|
||||||
|
return nil, errors.New("dict keys must be strings")
|
||||||
|
}
|
||||||
|
dict[key] = values[i+1]
|
||||||
|
}
|
||||||
|
return dict, nil
|
||||||
|
},
|
||||||
|
"formTime": func(t jsontime.Time) string {
|
||||||
|
return time.Time(t).Format("2006-01-02T15:04")
|
||||||
|
},
|
||||||
|
"ago": func(s string) (string, error) {
|
||||||
|
d, err := time.ParseDuration(s)
|
||||||
|
if err != nil {
|
||||||
|
return "", err
|
||||||
|
}
|
||||||
|
return time.Now().Add(-d).Format("2006-01-02T15:04"), nil
|
||||||
|
},
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
func fmtFloat(v float64, places ...int) string {
|
||||||
|
if len(places) > 0 {
|
||||||
|
return fmt.Sprintf("%."+strconv.Itoa(places[0])+"f", v)
|
||||||
|
}
|
||||||
|
return fmt.Sprintf("%.4f", v)
|
||||||
|
}
|
|
@ -1,6 +1,7 @@
|
||||||
package forms
|
package forms
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"encoding/json"
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
|
@ -16,15 +17,22 @@ import (
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
ErrNotStruct = errors.New("destination is not a struct")
|
ErrNotStruct = errors.New("destination is not a struct")
|
||||||
ErrNotPointer = errors.New("destination is not a pointer")
|
ErrNotPointer = errors.New("destination is not a pointer")
|
||||||
|
ErrContentType = errors.New("bad content type")
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
MaxMultipartMemory int64 = 1024 * 1024 // 1MB
|
||||||
)
|
)
|
||||||
|
|
||||||
type options struct {
|
type options struct {
|
||||||
tagOverride *string
|
tagOverride *string
|
||||||
parseTimeIn *time.Location
|
parseTimeIn *time.Location
|
||||||
parseLocal bool
|
parseLocal bool
|
||||||
acceptBlank bool
|
acceptBlank bool
|
||||||
|
maxMultipartMemory int64
|
||||||
|
defaultOmitEmpty bool
|
||||||
}
|
}
|
||||||
|
|
||||||
type Option func(*options)
|
type Option func(*options)
|
||||||
|
@ -53,6 +61,18 @@ func WithTag(t string) Option {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func WithMaxMultipartSize(s int64) Option {
|
||||||
|
return func(o *options) {
|
||||||
|
o.maxMultipartMemory = s
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func WithOmitEmpty() Option {
|
||||||
|
return func(o *options) {
|
||||||
|
o.defaultOmitEmpty = true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
func (o *options) Tag() string {
|
func (o *options) Tag() string {
|
||||||
if o.tagOverride != nil {
|
if o.tagOverride != nil {
|
||||||
return *o.tagOverride
|
return *o.tagOverride
|
||||||
|
@ -147,17 +167,19 @@ func (o *options) parseDuration(s string) (v time.Duration, set bool, err error)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
func (o *options) iterFields(r *http.Request, rv reflect.Value) error {
|
var typeOfByteSlice = reflect.TypeOf([]byte(nil))
|
||||||
rt := rv.Type()
|
|
||||||
for i := 0; i < rv.NumField(); i++ {
|
func (o *options) iterFields(r *http.Request, destStruct reflect.Value) error {
|
||||||
f := rv.Field(i)
|
structType := destStruct.Type()
|
||||||
tf := rt.Field(i)
|
for i := 0; i < destStruct.NumField(); i++ {
|
||||||
if !tf.IsExported() && !tf.Anonymous {
|
destFieldVal := destStruct.Field(i)
|
||||||
|
fieldType := structType.Field(i)
|
||||||
|
if !fieldType.IsExported() && !fieldType.Anonymous {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
if f.Kind() == reflect.Struct && tf.Anonymous {
|
if destFieldVal.Kind() == reflect.Struct && fieldType.Anonymous {
|
||||||
err := o.iterFields(r, f)
|
err := o.iterFields(r, destFieldVal)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
@ -165,51 +187,38 @@ func (o *options) iterFields(r *http.Request, rv reflect.Value) error {
|
||||||
|
|
||||||
var tAr []string
|
var tAr []string
|
||||||
var formField string
|
var formField string
|
||||||
formTag, has := rt.Field(i).Tag.Lookup(o.Tag())
|
var omitEmpty bool
|
||||||
|
if o.defaultOmitEmpty {
|
||||||
|
omitEmpty = true
|
||||||
|
}
|
||||||
|
|
||||||
|
formTag, has := structType.Field(i).Tag.Lookup(o.Tag())
|
||||||
if has {
|
if has {
|
||||||
tAr = strings.Split(formTag, ",")
|
tAr = strings.Split(formTag, ",")
|
||||||
formField = tAr[0]
|
formField = tAr[0]
|
||||||
|
for _, v := range tAr[1:] {
|
||||||
|
if v == "omitempty" {
|
||||||
|
omitEmpty = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if !has || formField == "-" {
|
if !has || formField == "-" {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
fi := f.Interface()
|
destFieldIntf := destFieldVal.Interface()
|
||||||
|
|
||||||
switch v := fi.(type) {
|
if destFieldVal.Kind() == reflect.Slice && destFieldVal.Type() == typeOfByteSlice {
|
||||||
case string, *string:
|
|
||||||
s := r.Form.Get(formField)
|
|
||||||
setVal(f, s != "" || o.acceptBlank, v, s)
|
|
||||||
case int, uint, *int, *uint:
|
|
||||||
ff := r.Form.Get(formField)
|
|
||||||
val, set, err := o.parseInt(ff)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
setVal(f, set, v, val)
|
|
||||||
case float64:
|
|
||||||
ff := r.Form.Get(formField)
|
|
||||||
val, set, err := o.parseFloat64(ff)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
setVal(f, set, v, val)
|
|
||||||
case bool, *bool:
|
|
||||||
ff := r.Form.Get(formField)
|
|
||||||
val, set, err := o.parseBool(ff)
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
setVal(f, set, v, val)
|
|
||||||
case []byte:
|
|
||||||
file, hdr, err := r.FormFile(formField)
|
file, hdr, err := r.FormFile(formField)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("get form file: %w", err)
|
return fmt.Errorf("get form file: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
nameField, hasFilename := rt.Field(i).Tag.Lookup("filenameField")
|
nameField, hasFilename := structType.Field(i).Tag.Lookup("filenameField")
|
||||||
if hasFilename {
|
if hasFilename {
|
||||||
fnf := rv.FieldByName(nameField)
|
fnf := destStruct.FieldByName(nameField)
|
||||||
if fnf == (reflect.Value{}) {
|
if fnf == (reflect.Value{}) {
|
||||||
panic(fmt.Errorf("filenameField '%s' does not exist", nameField))
|
panic(fmt.Errorf("filenameField '%s' does not exist", nameField))
|
||||||
}
|
}
|
||||||
|
@ -221,23 +230,52 @@ func (o *options) iterFields(r *http.Request, rv reflect.Value) error {
|
||||||
return fmt.Errorf("file read: %w", err)
|
return fmt.Errorf("file read: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
f.SetBytes(audioBytes)
|
destFieldVal.SetBytes(audioBytes)
|
||||||
|
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
if !r.Form.Has(formField) && omitEmpty {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
ff := r.Form.Get(formField)
|
||||||
|
|
||||||
|
switch v := destFieldIntf.(type) {
|
||||||
|
case string, *string:
|
||||||
|
setVal(destFieldVal, ff != "" || o.acceptBlank, ff)
|
||||||
|
case int, uint, *int, *uint:
|
||||||
|
val, set, err := o.parseInt(ff)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
setVal(destFieldVal, set, val)
|
||||||
|
case float64:
|
||||||
|
val, set, err := o.parseFloat64(ff)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
setVal(destFieldVal, set, val)
|
||||||
|
case bool, *bool:
|
||||||
|
val, set, err := o.parseBool(ff)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
setVal(destFieldVal, set, val)
|
||||||
case time.Time, *time.Time, jsontime.Time, *jsontime.Time:
|
case time.Time, *time.Time, jsontime.Time, *jsontime.Time:
|
||||||
tval := r.Form.Get(formField)
|
t, set, err := o.parseTime(ff)
|
||||||
t, set, err := o.parseTime(tval)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
setVal(f, set, v, t)
|
setVal(destFieldVal, set, t)
|
||||||
case time.Duration, *time.Duration, jsontime.Duration, *jsontime.Duration:
|
case time.Duration, *time.Duration, jsontime.Duration, *jsontime.Duration:
|
||||||
dval := r.Form.Get(formField)
|
d, set, err := o.parseDuration(ff)
|
||||||
d, set, err := o.parseDuration(dval)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
setVal(f, set, v, d)
|
setVal(destFieldVal, set, d)
|
||||||
case []int:
|
case []int:
|
||||||
val := strings.Trim(r.Form.Get(formField), "[]")
|
val := strings.Trim(ff, "[]")
|
||||||
if val == "" && o.acceptBlank {
|
if val == "" && o.acceptBlank {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
@ -249,7 +287,7 @@ func (o *options) iterFields(r *http.Request, rv reflect.Value) error {
|
||||||
ar = append(ar, i)
|
ar = append(ar, i)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
f.Set(reflect.ValueOf(ar))
|
destFieldVal.Set(reflect.ValueOf(ar))
|
||||||
default:
|
default:
|
||||||
panic(fmt.Errorf("unsupported type %T", v))
|
panic(fmt.Errorf("unsupported type %T", v))
|
||||||
}
|
}
|
||||||
|
@ -258,48 +296,77 @@ func (o *options) iterFields(r *http.Request, rv reflect.Value) error {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func setVal(setField reflect.Value, set bool, fv any, sv any) {
|
func setVal(destFieldVal reflect.Value, set bool, src any) {
|
||||||
if !set {
|
if !set {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
rv := reflect.TypeOf(fv)
|
destType := destFieldVal.Type()
|
||||||
svo := reflect.ValueOf(sv)
|
srcVal := reflect.ValueOf(src)
|
||||||
|
|
||||||
if svo.CanConvert(rv) {
|
if srcVal.Kind() == reflect.Ptr {
|
||||||
svo = svo.Convert(rv)
|
srcVal = srcVal.Elem()
|
||||||
}
|
}
|
||||||
|
|
||||||
if rv.Kind() == reflect.Ptr {
|
if destType.Kind() == reflect.Ptr {
|
||||||
svo = svo.Addr()
|
if !srcVal.CanAddr() {
|
||||||
|
if srcVal.CanConvert(destType.Elem()) {
|
||||||
|
srcVal = srcVal.Convert(destType.Elem())
|
||||||
|
}
|
||||||
|
copy := reflect.New(srcVal.Type())
|
||||||
|
copy.Elem().Set(srcVal)
|
||||||
|
srcVal = copy
|
||||||
|
}
|
||||||
|
} else if srcVal.CanConvert(destFieldVal.Type()) {
|
||||||
|
srcVal = srcVal.Convert(destFieldVal.Type())
|
||||||
}
|
}
|
||||||
|
|
||||||
setField.Set(svo)
|
destFieldVal.Set(srcVal)
|
||||||
}
|
}
|
||||||
|
|
||||||
func Unmarshal(r *http.Request, dest any, opt ...Option) error {
|
func Unmarshal(r *http.Request, dest any, opt ...Option) error {
|
||||||
o := options{}
|
o := options{
|
||||||
|
maxMultipartMemory: MaxMultipartMemory,
|
||||||
|
}
|
||||||
|
|
||||||
for _, opt := range opt {
|
for _, opt := range opt {
|
||||||
opt(&o)
|
opt(&o)
|
||||||
}
|
}
|
||||||
|
|
||||||
rv := reflect.ValueOf(dest)
|
contentType := strings.Split(r.Header.Get("Content-Type"), ";")[0]
|
||||||
if k := rv.Kind(); k == reflect.Ptr {
|
|
||||||
rv = rv.Elem()
|
|
||||||
} else {
|
|
||||||
return ErrNotPointer
|
|
||||||
}
|
|
||||||
|
|
||||||
if rv.Kind() != reflect.Struct {
|
switch contentType {
|
||||||
return ErrNotStruct
|
case "multipart/form-data":
|
||||||
}
|
err := r.ParseMultipartForm(o.maxMultipartMemory)
|
||||||
|
if err != nil {
|
||||||
|
return fmt.Errorf("ParseForm: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
if strings.HasPrefix(r.Header.Get("Content-Type"), "application/x-www-form-urlencoded") {
|
return o.unmarshalForm(r, dest)
|
||||||
|
case "application/x-www-form-urlencoded":
|
||||||
err := r.ParseForm()
|
err := r.ParseForm()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return fmt.Errorf("ParseForm: %w", err)
|
return fmt.Errorf("ParseForm: %w", err)
|
||||||
}
|
}
|
||||||
|
return o.unmarshalForm(r, dest)
|
||||||
|
case "application/json":
|
||||||
|
return json.NewDecoder(r.Body).Decode(dest)
|
||||||
}
|
}
|
||||||
|
|
||||||
return o.iterFields(r, rv)
|
return ErrContentType
|
||||||
|
}
|
||||||
|
|
||||||
|
func (o *options) unmarshalForm(r *http.Request, dest any) error {
|
||||||
|
destVal := reflect.ValueOf(dest)
|
||||||
|
if k := destVal.Kind(); k == reflect.Ptr {
|
||||||
|
destVal = destVal.Elem()
|
||||||
|
} else {
|
||||||
|
return ErrNotPointer
|
||||||
|
}
|
||||||
|
|
||||||
|
if destVal.Kind() != reflect.Struct {
|
||||||
|
return ErrNotStruct
|
||||||
|
}
|
||||||
|
|
||||||
|
return o.iterFields(r, destVal)
|
||||||
}
|
}
|
||||||
|
|
|
@ -8,6 +8,7 @@ import (
|
||||||
"testing"
|
"testing"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"dynatron.me/x/stillbox/internal/common"
|
||||||
"dynatron.me/x/stillbox/internal/forms"
|
"dynatron.me/x/stillbox/internal/forms"
|
||||||
"dynatron.me/x/stillbox/internal/jsontime"
|
"dynatron.me/x/stillbox/internal/jsontime"
|
||||||
|
|
||||||
|
@ -54,6 +55,14 @@ type urlEncTestJT struct {
|
||||||
ScoreEnd jsontime.Time `json:"scoreEnd"`
|
ScoreEnd jsontime.Time `json:"scoreEnd"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type ptrTestJT struct {
|
||||||
|
LookbackDays uint `form:"lookbackDays"`
|
||||||
|
HalfLife *jsontime.Duration `form:"halfLife"`
|
||||||
|
Recent *string `form:"recent"`
|
||||||
|
ScoreStart *jsontime.Time `form:"scoreStart"`
|
||||||
|
ScoreEnd jsontime.Time `form:"scoreEnd"`
|
||||||
|
}
|
||||||
|
|
||||||
var (
|
var (
|
||||||
UrlEncTest = urlEncTest{
|
UrlEncTest = urlEncTest{
|
||||||
LookbackDays: 7,
|
LookbackDays: 7,
|
||||||
|
@ -69,6 +78,13 @@ var (
|
||||||
ScoreStart: jsontime.Time(time.Date(2024, time.October, 28, 9, 25, 0, 0, time.UTC)),
|
ScoreStart: jsontime.Time(time.Date(2024, time.October, 28, 9, 25, 0, 0, time.UTC)),
|
||||||
}
|
}
|
||||||
|
|
||||||
|
PtrTestJT = ptrTestJT{
|
||||||
|
LookbackDays: 7,
|
||||||
|
HalfLife: common.PtrTo(jsontime.Duration(30 * time.Minute)),
|
||||||
|
Recent: common.PtrTo("2h0m0s"),
|
||||||
|
ScoreStart: common.PtrTo(jsontime.Time(time.Date(2024, time.October, 28, 9, 25, 0, 0, time.UTC))),
|
||||||
|
}
|
||||||
|
|
||||||
UrlEncTestJTLocal = urlEncTestJT{
|
UrlEncTestJTLocal = urlEncTestJT{
|
||||||
LookbackDays: 7,
|
LookbackDays: 7,
|
||||||
HalfLife: jsontime.Duration(30 * time.Minute),
|
HalfLife: jsontime.Duration(30 * time.Minute),
|
||||||
|
@ -122,29 +138,29 @@ func TestUnmarshal(t *testing.T) {
|
||||||
name string
|
name string
|
||||||
r *http.Request
|
r *http.Request
|
||||||
dest any
|
dest any
|
||||||
compare any
|
expect any
|
||||||
expectErr error
|
expectErr error
|
||||||
opts []forms.Option
|
opts []forms.Option
|
||||||
}{
|
}{
|
||||||
{
|
{
|
||||||
name: "base case",
|
name: "base case",
|
||||||
r: makeRequest("call1.http"),
|
r: makeRequest("call1.http"),
|
||||||
dest: &callUploadRequest{},
|
dest: &callUploadRequest{},
|
||||||
compare: &Call1,
|
expect: &Call1,
|
||||||
opts: []forms.Option{forms.WithAcceptBlank()},
|
opts: []forms.Option{forms.WithAcceptBlank()},
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
name: "base case no accept blank",
|
name: "base case no accept blank",
|
||||||
r: makeRequest("call1.http"),
|
r: makeRequest("call1.http"),
|
||||||
dest: &callUploadRequest{},
|
dest: &callUploadRequest{},
|
||||||
compare: &Call1,
|
expect: &Call1,
|
||||||
expectErr: errors.New(`parsebool(''): strconv.ParseBool: parsing "": invalid syntax`),
|
expectErr: errors.New(`parsebool(''): strconv.ParseBool: parsing "": invalid syntax`),
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
name: "not a pointer",
|
name: "not a pointer",
|
||||||
r: makeRequest("call1.http"),
|
r: makeRequest("call1.http"),
|
||||||
dest: callUploadRequest{},
|
dest: callUploadRequest{},
|
||||||
compare: callUploadRequest{},
|
expect: callUploadRequest{},
|
||||||
expectErr: forms.ErrNotPointer,
|
expectErr: forms.ErrNotPointer,
|
||||||
opts: []forms.Option{forms.WithAcceptBlank()},
|
opts: []forms.Option{forms.WithAcceptBlank()},
|
||||||
},
|
},
|
||||||
|
@ -152,7 +168,7 @@ func TestUnmarshal(t *testing.T) {
|
||||||
name: "not a struct",
|
name: "not a struct",
|
||||||
r: makeRequest("call1.http"),
|
r: makeRequest("call1.http"),
|
||||||
dest: &str,
|
dest: &str,
|
||||||
compare: callUploadRequest{},
|
expect: callUploadRequest{},
|
||||||
expectErr: forms.ErrNotStruct,
|
expectErr: forms.ErrNotStruct,
|
||||||
opts: []forms.Option{forms.WithAcceptBlank()},
|
opts: []forms.Option{forms.WithAcceptBlank()},
|
||||||
},
|
},
|
||||||
|
@ -160,44 +176,51 @@ func TestUnmarshal(t *testing.T) {
|
||||||
name: "url encoded",
|
name: "url encoded",
|
||||||
r: makeRequest("urlenc.http"),
|
r: makeRequest("urlenc.http"),
|
||||||
dest: &urlEncTest{},
|
dest: &urlEncTest{},
|
||||||
compare: &UrlEncTest,
|
expect: &UrlEncTest,
|
||||||
expectErr: errors.New(`Could not find format for ""`),
|
expectErr: errors.New(`Could not find format for ""`),
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
name: "url encoded accept blank",
|
name: "url encoded accept blank",
|
||||||
r: makeRequest("urlenc.http"),
|
r: makeRequest("urlenc.http"),
|
||||||
dest: &urlEncTest{},
|
dest: &urlEncTest{},
|
||||||
compare: &UrlEncTest,
|
expect: &UrlEncTest,
|
||||||
opts: []forms.Option{forms.WithAcceptBlank()},
|
opts: []forms.Option{forms.WithAcceptBlank()},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "url encoded accept blank pointer",
|
||||||
|
r: makeRequest("urlenc.http"),
|
||||||
|
dest: &ptrTestJT{},
|
||||||
|
expect: &PtrTestJT,
|
||||||
|
opts: []forms.Option{forms.WithAcceptBlank()},
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
name: "url encoded jsontime",
|
name: "url encoded jsontime",
|
||||||
r: makeRequest("urlenc.http"),
|
r: makeRequest("urlenc.http"),
|
||||||
dest: &urlEncTestJT{},
|
dest: &urlEncTestJT{},
|
||||||
compare: &UrlEncTestJT,
|
expect: &UrlEncTestJT,
|
||||||
expectErr: errors.New(`Could not find format for ""`),
|
expectErr: errors.New(`Could not find format for ""`),
|
||||||
opts: []forms.Option{forms.WithTag("json")},
|
opts: []forms.Option{forms.WithTag("json")},
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
name: "url encoded jsontime with tz",
|
name: "url encoded jsontime with tz",
|
||||||
r: makeRequest("urlenc.http"),
|
r: makeRequest("urlenc.http"),
|
||||||
dest: &urlEncTestJT{},
|
dest: &urlEncTestJT{},
|
||||||
compare: &UrlEncTestJT,
|
expect: &UrlEncTestJT,
|
||||||
opts: []forms.Option{forms.WithAcceptBlank(), forms.WithParseTimeInTZ(time.UTC), forms.WithTag("json")},
|
opts: []forms.Option{forms.WithAcceptBlank(), forms.WithParseTimeInTZ(time.UTC), forms.WithTag("json")},
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
name: "url encoded jsontime with local",
|
name: "url encoded jsontime with local",
|
||||||
r: makeRequest("urlenc.http"),
|
r: makeRequest("urlenc.http"),
|
||||||
dest: &urlEncTestJT{},
|
dest: &urlEncTestJT{},
|
||||||
compare: &UrlEncTestJTLocal,
|
expect: &UrlEncTestJTLocal,
|
||||||
opts: []forms.Option{forms.WithAcceptBlank(), forms.WithParseLocalTime(), forms.WithTag("json")},
|
opts: []forms.Option{forms.WithAcceptBlank(), forms.WithParseLocalTime(), forms.WithTag("json")},
|
||||||
},
|
},
|
||||||
{
|
{
|
||||||
name: "sim real data",
|
name: "sim real data",
|
||||||
r: makeRequest("urlenc2.http"),
|
r: makeRequest("urlenc2.http"),
|
||||||
dest: &alerting.Simulation{},
|
dest: &alerting.Simulation{},
|
||||||
compare: realSim,
|
expect: realSim,
|
||||||
opts: []forms.Option{forms.WithAcceptBlank(), forms.WithParseLocalTime()},
|
opts: []forms.Option{forms.WithAcceptBlank(), forms.WithParseLocalTime()},
|
||||||
},
|
},
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -209,7 +232,7 @@ func TestUnmarshal(t *testing.T) {
|
||||||
assert.Contains(t, tc.expectErr.Error(), err.Error())
|
assert.Contains(t, tc.expectErr.Error(), err.Error())
|
||||||
} else {
|
} else {
|
||||||
require.NoError(t, err)
|
require.NoError(t, err)
|
||||||
assert.Equal(t, tc.compare, tc.dest)
|
assert.Equal(t, tc.expect, tc.dest)
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
80
pkg/alerting/alert/alert.go
Normal file
80
pkg/alerting/alert/alert.go
Normal file
|
@ -0,0 +1,80 @@
|
||||||
|
package alert
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
"strconv"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"dynatron.me/x/stillbox/internal/trending"
|
||||||
|
"dynatron.me/x/stillbox/pkg/database"
|
||||||
|
"dynatron.me/x/stillbox/pkg/talkgroups"
|
||||||
|
|
||||||
|
"github.com/google/uuid"
|
||||||
|
"github.com/jackc/pgx/v5/pgtype"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Alert struct {
|
||||||
|
ID uuid.UUID
|
||||||
|
Timestamp time.Time
|
||||||
|
TGName string
|
||||||
|
Score trending.Score[talkgroups.ID]
|
||||||
|
OrigScore float64
|
||||||
|
Weight float32
|
||||||
|
Suppressed bool
|
||||||
|
}
|
||||||
|
|
||||||
|
func (a *Alert) ToAddAlertParams() database.AddAlertParams {
|
||||||
|
f32score := float32(a.Score.Score)
|
||||||
|
f32origscore := float32(a.OrigScore)
|
||||||
|
|
||||||
|
var origScore *float32
|
||||||
|
if a.Score.Score != a.OrigScore {
|
||||||
|
origScore = &f32origscore
|
||||||
|
}
|
||||||
|
|
||||||
|
return database.AddAlertParams{
|
||||||
|
ID: a.ID,
|
||||||
|
Time: pgtype.Timestamptz{Time: a.Timestamp, Valid: true},
|
||||||
|
PackedTg: a.Score.ID.Pack(),
|
||||||
|
Weight: &a.Weight,
|
||||||
|
Score: &f32score,
|
||||||
|
OrigScore: origScore,
|
||||||
|
Notified: !a.Suppressed,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Make creates an alert for later rendering or storage.
|
||||||
|
func Make(ctx context.Context, store talkgroups.Store, score trending.Score[talkgroups.ID], origScore float64) (Alert, error) {
|
||||||
|
d := Alert{
|
||||||
|
ID: uuid.New(),
|
||||||
|
Score: score,
|
||||||
|
Timestamp: time.Now(),
|
||||||
|
Weight: 1.0,
|
||||||
|
OrigScore: origScore,
|
||||||
|
}
|
||||||
|
|
||||||
|
tgRecord, err := store.TG(ctx, score.ID)
|
||||||
|
switch err {
|
||||||
|
case nil:
|
||||||
|
d.Weight = tgRecord.Talkgroup.Weight
|
||||||
|
if tgRecord.System.Name == "" {
|
||||||
|
tgRecord.System.Name = strconv.Itoa(int(score.ID.System))
|
||||||
|
}
|
||||||
|
|
||||||
|
if tgRecord.Talkgroup.Name != nil {
|
||||||
|
d.TGName = fmt.Sprintf("%s %s [%d]", tgRecord.System.Name, *tgRecord.Talkgroup.Name, score.ID.Talkgroup)
|
||||||
|
} else {
|
||||||
|
d.TGName = fmt.Sprintf("%s:%d", tgRecord.System.Name, int(score.ID.Talkgroup))
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
system, has := store.SystemName(ctx, int(score.ID.System))
|
||||||
|
if has {
|
||||||
|
d.TGName = fmt.Sprintf("%s:%d", system, int(score.ID.Talkgroup))
|
||||||
|
} else {
|
||||||
|
d.TGName = fmt.Sprintf("%d:%d", int(score.ID.System), int(score.ID.Talkgroup))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return d, nil
|
||||||
|
}
|
|
@ -1,16 +1,14 @@
|
||||||
package alerting
|
package alerting
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
|
||||||
"context"
|
"context"
|
||||||
"fmt"
|
"fmt"
|
||||||
"net/http"
|
"net/http"
|
||||||
"sort"
|
"sort"
|
||||||
"strconv"
|
|
||||||
"sync"
|
"sync"
|
||||||
"text/template"
|
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"dynatron.me/x/stillbox/pkg/alerting/alert"
|
||||||
"dynatron.me/x/stillbox/pkg/calls"
|
"dynatron.me/x/stillbox/pkg/calls"
|
||||||
"dynatron.me/x/stillbox/pkg/config"
|
"dynatron.me/x/stillbox/pkg/config"
|
||||||
"dynatron.me/x/stillbox/pkg/database"
|
"dynatron.me/x/stillbox/pkg/database"
|
||||||
|
@ -21,8 +19,6 @@ import (
|
||||||
"dynatron.me/x/stillbox/internal/timeseries"
|
"dynatron.me/x/stillbox/internal/timeseries"
|
||||||
"dynatron.me/x/stillbox/internal/trending"
|
"dynatron.me/x/stillbox/internal/trending"
|
||||||
|
|
||||||
"github.com/google/uuid"
|
|
||||||
"github.com/jackc/pgx/v5/pgtype"
|
|
||||||
"github.com/rs/zerolog/log"
|
"github.com/rs/zerolog/log"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
@ -51,7 +47,7 @@ type alerter struct {
|
||||||
scores trending.Scores[talkgroups.ID]
|
scores trending.Scores[talkgroups.ID]
|
||||||
lastScore time.Time
|
lastScore time.Time
|
||||||
sim *Simulation
|
sim *Simulation
|
||||||
alertCache map[talkgroups.ID]Alert
|
alertCache map[talkgroups.ID]alert.Alert
|
||||||
renotify time.Duration
|
renotify time.Duration
|
||||||
notifier notify.Notifier
|
notifier notify.Notifier
|
||||||
tgCache talkgroups.Store
|
tgCache talkgroups.Store
|
||||||
|
@ -96,7 +92,7 @@ func New(cfg config.Alerting, tgCache talkgroups.Store, opts ...AlertOption) Ale
|
||||||
|
|
||||||
as := &alerter{
|
as := &alerter{
|
||||||
cfg: cfg,
|
cfg: cfg,
|
||||||
alertCache: make(map[talkgroups.ID]Alert),
|
alertCache: make(map[talkgroups.ID]alert.Alert),
|
||||||
clock: timeseries.DefaultClock,
|
clock: timeseries.DefaultClock,
|
||||||
renotify: DefaultRenotify,
|
renotify: DefaultRenotify,
|
||||||
tgCache: tgCache,
|
tgCache: tgCache,
|
||||||
|
@ -150,22 +146,18 @@ func (as *alerter) Go(ctx context.Context) {
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
const notificationTemplStr = `{{ range . -}}
|
func (as *alerter) eval(ctx context.Context, now time.Time, testMode bool) ([]alert.Alert, error) {
|
||||||
{{ .TGName }} is active with a score of {{ f .Score.Score 4 }}! ({{ f .Score.RecentCount 0 }}/{{ .Score.Count }} recent calls)
|
|
||||||
|
|
||||||
{{ end -}}`
|
|
||||||
|
|
||||||
var notificationTemplate = template.Must(template.New("notification").Funcs(funcMap).Parse(notificationTemplStr))
|
|
||||||
|
|
||||||
func (as *alerter) eval(ctx context.Context, now time.Time, testMode bool) ([]Alert, error) {
|
|
||||||
err := as.tgCache.Hint(ctx, as.scoredTGs())
|
err := as.tgCache.Hint(ctx, as.scoredTGs())
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("prime TG cache: %w", err)
|
return nil, fmt.Errorf("prime TG cache: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
as.Lock()
|
||||||
|
defer as.Unlock()
|
||||||
|
|
||||||
db := database.FromCtx(ctx)
|
db := database.FromCtx(ctx)
|
||||||
|
|
||||||
var notifications []Alert
|
var notifications []alert.Alert
|
||||||
for _, s := range as.scores {
|
for _, s := range as.scores {
|
||||||
origScore := s.Score
|
origScore := s.Score
|
||||||
tgr, err := as.tgCache.TG(ctx, s.ID)
|
tgr, err := as.tgCache.TG(ctx, s.ID)
|
||||||
|
@ -176,7 +168,7 @@ func (as *alerter) eval(ctx context.Context, now time.Time, testMode bool) ([]Al
|
||||||
if s.Score > as.cfg.AlertThreshold || testMode {
|
if s.Score > as.cfg.AlertThreshold || testMode {
|
||||||
if old, inCache := as.alertCache[s.ID]; !inCache || now.Sub(old.Timestamp) > as.renotify {
|
if old, inCache := as.alertCache[s.ID]; !inCache || now.Sub(old.Timestamp) > as.renotify {
|
||||||
s.Score *= as.tgCache.Weight(ctx, s.ID, now)
|
s.Score *= as.tgCache.Weight(ctx, s.ID, now)
|
||||||
a, err := as.makeAlert(ctx, s, origScore)
|
a, err := alert.Make(ctx, as.tgCache, s, origScore)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, fmt.Errorf("makeAlert: %w", err)
|
return nil, fmt.Errorf("makeAlert: %w", err)
|
||||||
}
|
}
|
||||||
|
@ -206,7 +198,7 @@ func (as *alerter) eval(ctx context.Context, now time.Time, testMode bool) ([]Al
|
||||||
}
|
}
|
||||||
|
|
||||||
func (as *alerter) testNotifyHandler(w http.ResponseWriter, r *http.Request) {
|
func (as *alerter) testNotifyHandler(w http.ResponseWriter, r *http.Request) {
|
||||||
alerts := make([]Alert, 0, len(as.scores))
|
alerts := make([]alert.Alert, 0, len(as.scores))
|
||||||
ctx := r.Context()
|
ctx := r.Context()
|
||||||
|
|
||||||
alerts, err := as.eval(ctx, time.Now(), true)
|
alerts, err := as.eval(ctx, time.Now(), true)
|
||||||
|
@ -216,7 +208,7 @@ func (as *alerter) testNotifyHandler(w http.ResponseWriter, r *http.Request) {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
err = as.sendNotification(ctx, alerts)
|
err = as.notifier.Send(ctx, alerts)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Error().Err(err).Msg("test notification send")
|
log.Error().Err(err).Msg("test notification send")
|
||||||
http.Error(w, err.Error(), http.StatusInternalServerError)
|
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||||||
|
@ -258,92 +250,12 @@ func (as *alerter) notify(ctx context.Context) error {
|
||||||
}
|
}
|
||||||
|
|
||||||
if len(notifications) > 0 {
|
if len(notifications) > 0 {
|
||||||
return as.sendNotification(ctx, notifications)
|
return as.notifier.Send(ctx, notifications)
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
type Alert struct {
|
|
||||||
ID uuid.UUID
|
|
||||||
Timestamp time.Time
|
|
||||||
TGName string
|
|
||||||
Score trending.Score[talkgroups.ID]
|
|
||||||
OrigScore float64
|
|
||||||
Weight float32
|
|
||||||
Suppressed bool
|
|
||||||
}
|
|
||||||
|
|
||||||
func (a *Alert) ToAddAlertParams() database.AddAlertParams {
|
|
||||||
f32score := float32(a.Score.Score)
|
|
||||||
f32origscore := float32(a.OrigScore)
|
|
||||||
|
|
||||||
var origScore *float32
|
|
||||||
if a.Score.Score != a.OrigScore {
|
|
||||||
origScore = &f32origscore
|
|
||||||
}
|
|
||||||
|
|
||||||
return database.AddAlertParams{
|
|
||||||
ID: a.ID,
|
|
||||||
Time: pgtype.Timestamptz{Time: a.Timestamp, Valid: true},
|
|
||||||
PackedTg: a.Score.ID.Pack(),
|
|
||||||
Weight: &a.Weight,
|
|
||||||
Score: &f32score,
|
|
||||||
OrigScore: origScore,
|
|
||||||
Notified: !a.Suppressed,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// sendNotification renders and sends the notification.
|
|
||||||
func (as *alerter) sendNotification(ctx context.Context, n []Alert) error {
|
|
||||||
msgBuffer := new(bytes.Buffer)
|
|
||||||
|
|
||||||
err := notificationTemplate.Execute(msgBuffer, n)
|
|
||||||
if err != nil {
|
|
||||||
return fmt.Errorf("notification template render: %w", err)
|
|
||||||
}
|
|
||||||
|
|
||||||
log.Debug().Str("msg", msgBuffer.String()).Msg("notifying")
|
|
||||||
|
|
||||||
return as.notifier.Send(ctx, NotificationSubject, msgBuffer.String())
|
|
||||||
}
|
|
||||||
|
|
||||||
// makeAlert creates a notification for later rendering by the template.
|
|
||||||
// It takes a talkgroup Score as input.
|
|
||||||
func (as *alerter) makeAlert(ctx context.Context, score trending.Score[talkgroups.ID], origScore float64) (Alert, error) {
|
|
||||||
d := Alert{
|
|
||||||
ID: uuid.New(),
|
|
||||||
Score: score,
|
|
||||||
Timestamp: time.Now(),
|
|
||||||
Weight: 1.0,
|
|
||||||
OrigScore: origScore,
|
|
||||||
}
|
|
||||||
|
|
||||||
tgRecord, err := as.tgCache.TG(ctx, score.ID)
|
|
||||||
switch err {
|
|
||||||
case nil:
|
|
||||||
d.Weight = tgRecord.Talkgroup.Weight
|
|
||||||
if tgRecord.System.Name == "" {
|
|
||||||
tgRecord.System.Name = strconv.Itoa(int(score.ID.System))
|
|
||||||
}
|
|
||||||
|
|
||||||
if tgRecord.Talkgroup.Name != nil {
|
|
||||||
d.TGName = fmt.Sprintf("%s %s (%d)", tgRecord.System.Name, *tgRecord.Talkgroup.Name, score.ID.Talkgroup)
|
|
||||||
} else {
|
|
||||||
d.TGName = fmt.Sprintf("%s:%d", tgRecord.System.Name, int(score.ID.Talkgroup))
|
|
||||||
}
|
|
||||||
default:
|
|
||||||
system, has := as.tgCache.SystemName(ctx, int(score.ID.System))
|
|
||||||
if has {
|
|
||||||
d.TGName = fmt.Sprintf("%s:%d", system, int(score.ID.Talkgroup))
|
|
||||||
} else {
|
|
||||||
d.TGName = fmt.Sprintf("%d:%d", int(score.ID.System), int(score.ID.Talkgroup))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return d, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// cleanCache clears the cache of aged-out entries
|
// cleanCache clears the cache of aged-out entries
|
||||||
func (as *alerter) cleanCache() {
|
func (as *alerter) cleanCache() {
|
||||||
if as.notifier == nil {
|
if as.notifier == nil {
|
||||||
|
|
|
@ -2,7 +2,6 @@ package alerting
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"encoding/json"
|
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
@ -114,24 +113,15 @@ func (s *Simulation) Simulate(ctx context.Context) (trending.Scores[talkgroups.I
|
||||||
func (as *alerter) simulateHandler(w http.ResponseWriter, r *http.Request) {
|
func (as *alerter) simulateHandler(w http.ResponseWriter, r *http.Request) {
|
||||||
ctx := r.Context()
|
ctx := r.Context()
|
||||||
s := new(Simulation)
|
s := new(Simulation)
|
||||||
switch r.Header.Get("Content-Type") {
|
|
||||||
case "application/json":
|
err := forms.Unmarshal(r, s, forms.WithAcceptBlank(), forms.WithParseLocalTime())
|
||||||
err := json.NewDecoder(r.Body).Decode(s)
|
if err != nil {
|
||||||
if err != nil {
|
err = fmt.Errorf("simulate unmarshal: %w", err)
|
||||||
err = fmt.Errorf("simulate decode: %w", err)
|
http.Error(w, err.Error(), http.StatusBadRequest)
|
||||||
http.Error(w, err.Error(), http.StatusBadRequest)
|
return
|
||||||
return
|
|
||||||
}
|
|
||||||
default:
|
|
||||||
err := forms.Unmarshal(r, s, forms.WithAcceptBlank(), forms.WithParseLocalTime())
|
|
||||||
if err != nil {
|
|
||||||
err = fmt.Errorf("simulate unmarshal: %w", err)
|
|
||||||
http.Error(w, err.Error(), http.StatusBadRequest)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
err := s.verify()
|
err = s.verify()
|
||||||
if err != nil {
|
if err != nil {
|
||||||
err = fmt.Errorf("simulation profile verify: %w", err)
|
err = fmt.Errorf("simulation profile verify: %w", err)
|
||||||
http.Error(w, err.Error(), http.StatusBadRequest)
|
http.Error(w, err.Error(), http.StatusBadRequest)
|
||||||
|
|
|
@ -2,7 +2,6 @@ package alerting
|
||||||
|
|
||||||
import (
|
import (
|
||||||
_ "embed"
|
_ "embed"
|
||||||
"errors"
|
|
||||||
"html/template"
|
"html/template"
|
||||||
"net/http"
|
"net/http"
|
||||||
"time"
|
"time"
|
||||||
|
@ -12,7 +11,6 @@ import (
|
||||||
"dynatron.me/x/stillbox/pkg/talkgroups"
|
"dynatron.me/x/stillbox/pkg/talkgroups"
|
||||||
|
|
||||||
"dynatron.me/x/stillbox/internal/common"
|
"dynatron.me/x/stillbox/internal/common"
|
||||||
"dynatron.me/x/stillbox/internal/jsontime"
|
|
||||||
"dynatron.me/x/stillbox/internal/trending"
|
"dynatron.me/x/stillbox/internal/trending"
|
||||||
|
|
||||||
"github.com/go-chi/chi/v5"
|
"github.com/go-chi/chi/v5"
|
||||||
|
@ -22,41 +20,14 @@ import (
|
||||||
//go:embed stats.html
|
//go:embed stats.html
|
||||||
var statsTemplateFile string
|
var statsTemplateFile string
|
||||||
|
|
||||||
|
var (
|
||||||
|
statTmpl = template.Must(template.New("stats").Funcs(common.FuncMap).Parse(statsTemplateFile))
|
||||||
|
)
|
||||||
|
|
||||||
type stats interface {
|
type stats interface {
|
||||||
PrivateRoutes(chi.Router)
|
PrivateRoutes(chi.Router)
|
||||||
}
|
}
|
||||||
|
|
||||||
var (
|
|
||||||
funcMap = template.FuncMap{
|
|
||||||
"f": common.FmtFloat,
|
|
||||||
"dict": func(values ...interface{}) (map[string]interface{}, error) {
|
|
||||||
if len(values)%2 != 0 {
|
|
||||||
return nil, errors.New("invalid dict call")
|
|
||||||
}
|
|
||||||
dict := make(map[string]interface{}, len(values)/2)
|
|
||||||
for i := 0; i < len(values); i += 2 {
|
|
||||||
key, ok := values[i].(string)
|
|
||||||
if !ok {
|
|
||||||
return nil, errors.New("dict keys must be strings")
|
|
||||||
}
|
|
||||||
dict[key] = values[i+1]
|
|
||||||
}
|
|
||||||
return dict, nil
|
|
||||||
},
|
|
||||||
"formTime": func(t jsontime.Time) string {
|
|
||||||
return time.Time(t).Format("2006-01-02T15:04")
|
|
||||||
},
|
|
||||||
"ago": func(s string) (string, error) {
|
|
||||||
d, err := time.ParseDuration(s)
|
|
||||||
if err != nil {
|
|
||||||
return "", err
|
|
||||||
}
|
|
||||||
return time.Now().Add(-d).Format("2006-01-02T15:04"), nil
|
|
||||||
},
|
|
||||||
}
|
|
||||||
statTmpl = template.Must(template.New("stats").Funcs(funcMap).Parse(statsTemplateFile))
|
|
||||||
)
|
|
||||||
|
|
||||||
func (as *alerter) PrivateRoutes(r chi.Router) {
|
func (as *alerter) PrivateRoutes(r chi.Router) {
|
||||||
r.Get("/tgstats", as.tgStatsHandler)
|
r.Get("/tgstats", as.tgStatsHandler)
|
||||||
r.Post("/tgstats", as.simulateHandler)
|
r.Post("/tgstats", as.simulateHandler)
|
||||||
|
|
|
@ -18,13 +18,10 @@ type API interface {
|
||||||
}
|
}
|
||||||
|
|
||||||
type api struct {
|
type api struct {
|
||||||
tgs talkgroups.Store
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func New(tgs talkgroups.Store) API {
|
func New() API {
|
||||||
s := &api{
|
s := new(api)
|
||||||
tgs: tgs,
|
|
||||||
}
|
|
||||||
|
|
||||||
return s
|
return s
|
||||||
}
|
}
|
||||||
|
@ -32,9 +29,8 @@ func New(tgs talkgroups.Store) API {
|
||||||
func (a *api) Subrouter() http.Handler {
|
func (a *api) Subrouter() http.Handler {
|
||||||
r := chi.NewMux()
|
r := chi.NewMux()
|
||||||
|
|
||||||
r.Get("/talkgroup/{system:\\d+}/{id:\\d+}", a.talkgroup)
|
r.Mount("/talkgroup", new(talkgroupAPI).routes())
|
||||||
r.Get("/talkgroup/{system:\\d+}/", a.talkgroup)
|
|
||||||
r.Get("/talkgroup/", a.talkgroup)
|
|
||||||
return r
|
return r
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -58,7 +54,7 @@ func httpCode(err error) int {
|
||||||
return http.StatusInternalServerError
|
return http.StatusInternalServerError
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *api) writeResponse(w http.ResponseWriter, r *http.Request, data interface{}, err error) {
|
func writeResponse(w http.ResponseWriter, r *http.Request, data interface{}, err error) {
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Error().Str("path", r.URL.Path).Err(err).Msg("request failed")
|
log.Error().Str("path", r.URL.Path).Err(err).Msg("request failed")
|
||||||
http.Error(w, err.Error(), httpCode(err))
|
http.Error(w, err.Error(), httpCode(err))
|
||||||
|
@ -75,6 +71,10 @@ func (a *api) writeResponse(w http.ResponseWriter, r *http.Request, data interfa
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func reqErr(w http.ResponseWriter, err error, code int) {
|
||||||
|
http.Error(w, err.Error(), code)
|
||||||
|
}
|
||||||
|
|
||||||
func decodeParams(d interface{}, r *http.Request) error {
|
func decodeParams(d interface{}, r *http.Request) error {
|
||||||
params := chi.RouteContext(r.Context()).URLParams
|
params := chi.RouteContext(r.Context()).URLParams
|
||||||
m := make(map[string]string, len(params.Keys))
|
m := make(map[string]string, len(params.Keys))
|
||||||
|
@ -96,32 +96,6 @@ func decodeParams(d interface{}, r *http.Request) error {
|
||||||
return dec.Decode(m)
|
return dec.Decode(m)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (a *api) badReq(w http.ResponseWriter, err error) {
|
func badReq(w http.ResponseWriter, err error) {
|
||||||
http.Error(w, err.Error(), http.StatusBadRequest)
|
reqErr(w, err, http.StatusBadRequest)
|
||||||
}
|
|
||||||
|
|
||||||
func (a *api) talkgroup(w http.ResponseWriter, r *http.Request) {
|
|
||||||
ctx := r.Context()
|
|
||||||
p := struct {
|
|
||||||
System *int `param:"system"`
|
|
||||||
ID *int `param:"id"`
|
|
||||||
}{}
|
|
||||||
|
|
||||||
err := decodeParams(&p, r)
|
|
||||||
if err != nil {
|
|
||||||
a.badReq(w, err)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
var res interface{}
|
|
||||||
switch {
|
|
||||||
case p.System != nil && p.ID != nil:
|
|
||||||
res, err = a.tgs.TG(ctx, talkgroups.TG(*p.System, *p.ID))
|
|
||||||
case p.System != nil:
|
|
||||||
res, err = a.tgs.SystemTGs(ctx, int32(*p.System))
|
|
||||||
default:
|
|
||||||
res, err = a.tgs.TGs(ctx, nil)
|
|
||||||
}
|
|
||||||
|
|
||||||
a.writeResponse(w, r, res, err)
|
|
||||||
}
|
}
|
||||||
|
|
116
pkg/api/talkgroups.go
Normal file
116
pkg/api/talkgroups.go
Normal file
|
@ -0,0 +1,116 @@
|
||||||
|
package api
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"dynatron.me/x/stillbox/internal/forms"
|
||||||
|
"dynatron.me/x/stillbox/pkg/talkgroups"
|
||||||
|
|
||||||
|
"github.com/go-chi/chi/v5"
|
||||||
|
)
|
||||||
|
|
||||||
|
type talkgroupAPI struct {
|
||||||
|
}
|
||||||
|
|
||||||
|
func (tga *talkgroupAPI) routes() http.Handler {
|
||||||
|
r := chi.NewMux()
|
||||||
|
|
||||||
|
r.Get("/{system:\\d+}/{id:\\d+}", tga.talkgroup)
|
||||||
|
r.Put("/{system:\\d+}/{id:\\d+}", tga.putTalkgroup)
|
||||||
|
r.Get("/{system:\\d+}/", tga.talkgroup)
|
||||||
|
r.Get("/", tga.talkgroup)
|
||||||
|
|
||||||
|
return r
|
||||||
|
}
|
||||||
|
|
||||||
|
type tgParams struct {
|
||||||
|
System *int `param:"system"`
|
||||||
|
ID *int `param:"id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t tgParams) haveBoth() bool {
|
||||||
|
return t.System != nil && t.ID != nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t tgParams) ToID() talkgroups.ID {
|
||||||
|
nilOr := func(i *int) uint32 {
|
||||||
|
if i == nil {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
return uint32(*i)
|
||||||
|
}
|
||||||
|
|
||||||
|
return talkgroups.ID{
|
||||||
|
System: nilOr(t.System),
|
||||||
|
Talkgroup: nilOr(t.ID),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (tga *talkgroupAPI) talkgroup(w http.ResponseWriter, r *http.Request) {
|
||||||
|
ctx := r.Context()
|
||||||
|
tgs := talkgroups.StoreFrom(ctx)
|
||||||
|
|
||||||
|
var p tgParams
|
||||||
|
|
||||||
|
err := decodeParams(&p, r)
|
||||||
|
if err != nil {
|
||||||
|
badReq(w, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
var res interface{}
|
||||||
|
switch {
|
||||||
|
case p.System != nil && p.ID != nil:
|
||||||
|
res, err = tgs.TG(ctx, talkgroups.TG(*p.System, *p.ID))
|
||||||
|
case p.System != nil:
|
||||||
|
res, err = tgs.SystemTGs(ctx, int32(*p.System))
|
||||||
|
default:
|
||||||
|
res, err = tgs.TGs(ctx, nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
writeResponse(w, r, res, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (tga *talkgroupAPI) putTalkgroup(w http.ResponseWriter, r *http.Request) {
|
||||||
|
var id tgParams
|
||||||
|
err := decodeParams(&id, r)
|
||||||
|
if err != nil {
|
||||||
|
badReq(w, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
/*
|
||||||
|
ctx := r.Context()
|
||||||
|
tgs := talkgroups.StoreFrom(ctx)
|
||||||
|
|
||||||
|
tg, err := tgs.TG(ctx, id.ToID())
|
||||||
|
switch err {
|
||||||
|
case nil:
|
||||||
|
case talkgroups.ErrNotFound:
|
||||||
|
reqErr(w, err, http.StatusNotFound)
|
||||||
|
return
|
||||||
|
default:
|
||||||
|
reqErr(w, err, http.StatusInternalServerError)
|
||||||
|
}
|
||||||
|
*/
|
||||||
|
|
||||||
|
input := struct {
|
||||||
|
Name *string `form:"name"`
|
||||||
|
AlphaTag *string `form:"alpha_tag"`
|
||||||
|
TgGroup *string `form:"tg_group"`
|
||||||
|
Frequency *int32 `form:"frequency"`
|
||||||
|
Metadata []byte `form:"metadata"`
|
||||||
|
Tags []string `form:"tags"`
|
||||||
|
Alert *bool `form:"alert"`
|
||||||
|
AlertConfig []byte `form:"alert_config"`
|
||||||
|
Weight *float32 `form:"weight"`
|
||||||
|
}{}
|
||||||
|
|
||||||
|
err = forms.Unmarshal(r, &input, forms.WithAcceptBlank(), forms.WithOmitEmpty())
|
||||||
|
if err != nil {
|
||||||
|
reqErr(w, err, http.StatusBadRequest)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
fmt.Fprintf(w, "%+v\n", input)
|
||||||
|
}
|
|
@ -19,7 +19,7 @@ func (d CallDuration) Duration() time.Duration {
|
||||||
return time.Duration(d)
|
return time.Duration(d)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (d CallDuration) Int32Ptr() *int32 {
|
func (d CallDuration) MsInt32Ptr() *int32 {
|
||||||
if time.Duration(d) == 0 {
|
if time.Duration(d) == 0 {
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
@ -106,7 +106,7 @@ func (c *Call) ToPB() *pb.Call {
|
||||||
Frequencies: toInt64Slice(c.Frequencies),
|
Frequencies: toInt64Slice(c.Frequencies),
|
||||||
Patches: toInt32Slice(c.Patches),
|
Patches: toInt32Slice(c.Patches),
|
||||||
Sources: toInt32Slice(c.Sources),
|
Sources: toInt32Slice(c.Sources),
|
||||||
Duration: c.Duration.Int32Ptr(),
|
Duration: c.Duration.MsInt32Ptr(),
|
||||||
Audio: c.Audio,
|
Audio: c.Audio,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -65,8 +65,10 @@ type Alerting struct {
|
||||||
type Notify []NotifyService
|
type Notify []NotifyService
|
||||||
|
|
||||||
type NotifyService struct {
|
type NotifyService struct {
|
||||||
Provider string `json:"provider"`
|
Provider string `yaml:"provider" json:"provider"`
|
||||||
Config map[string]interface{} `json:"config"`
|
SubjectTemplate *string `yaml:"subjectTemplate" json:"subjectTemplate"`
|
||||||
|
BodyTemplate *string `yaml:"bodyTemplate" json:"bodyTemplate"`
|
||||||
|
Config map[string]interface{} `yaml:"config" json:"config"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func (n *NotifyService) GetS(k, defaultVal string) string {
|
func (n *NotifyService) GetS(k, defaultVal string) string {
|
||||||
|
|
|
@ -58,11 +58,11 @@ func NewClient(ctx context.Context, conf config.DB) (*DB, error) {
|
||||||
|
|
||||||
type DBCtxKey string
|
type DBCtxKey string
|
||||||
|
|
||||||
const DBCTXKeyValue DBCtxKey = "dbctx"
|
const DBCtxKeyValue DBCtxKey = "dbctx"
|
||||||
|
|
||||||
// FromCtx returns the database handle from the provided Context.
|
// FromCtx returns the database handle from the provided Context.
|
||||||
func FromCtx(ctx context.Context) *DB {
|
func FromCtx(ctx context.Context) *DB {
|
||||||
c, ok := ctx.Value(DBCTXKeyValue).(*DB)
|
c, ok := ctx.Value(DBCtxKeyValue).(*DB)
|
||||||
if !ok {
|
if !ok {
|
||||||
panic("no DB in context")
|
panic("no DB in context")
|
||||||
}
|
}
|
||||||
|
@ -72,7 +72,7 @@ func FromCtx(ctx context.Context) *DB {
|
||||||
|
|
||||||
// CtxWithDB returns a Context with the provided database handle.
|
// CtxWithDB returns a Context with the provided database handle.
|
||||||
func CtxWithDB(ctx context.Context, conn *DB) context.Context {
|
func CtxWithDB(ctx context.Context, conn *DB) context.Context {
|
||||||
return context.WithValue(ctx, DBCTXKeyValue, conn)
|
return context.WithValue(ctx, DBCtxKeyValue, conn)
|
||||||
}
|
}
|
||||||
|
|
||||||
// IsNoRows is a convenience function that returns whether a returned error is a database
|
// IsNoRows is a convenience function that returns whether a returned error is a database
|
||||||
|
|
|
@ -1,10 +1,15 @@
|
||||||
package notify
|
package notify
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
|
"context"
|
||||||
"fmt"
|
"fmt"
|
||||||
stdhttp "net/http"
|
stdhttp "net/http"
|
||||||
|
"text/template"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
|
"dynatron.me/x/stillbox/internal/common"
|
||||||
|
"dynatron.me/x/stillbox/pkg/alerting/alert"
|
||||||
"dynatron.me/x/stillbox/pkg/config"
|
"dynatron.me/x/stillbox/pkg/config"
|
||||||
|
|
||||||
"github.com/go-viper/mapstructure/v2"
|
"github.com/go-viper/mapstructure/v2"
|
||||||
|
@ -13,15 +18,86 @@ import (
|
||||||
)
|
)
|
||||||
|
|
||||||
type Notifier interface {
|
type Notifier interface {
|
||||||
notify.Notifier
|
Send(ctx context.Context, alerts []alert.Alert) error
|
||||||
|
}
|
||||||
|
|
||||||
|
type backend struct {
|
||||||
|
*notify.Notify
|
||||||
|
subject *template.Template
|
||||||
|
body *template.Template
|
||||||
}
|
}
|
||||||
|
|
||||||
type notifier struct {
|
type notifier struct {
|
||||||
*notify.Notify
|
backends []backend
|
||||||
}
|
}
|
||||||
|
|
||||||
func (n *notifier) buildSlackWebhookPayload(cfg *slackWebhookConfig) func(string, string) any {
|
func highest(a []alert.Alert) string {
|
||||||
|
if len(a) < 1 {
|
||||||
|
return "none"
|
||||||
|
}
|
||||||
|
|
||||||
|
top := a[0]
|
||||||
|
for _, a := range a {
|
||||||
|
if a.Score.Score > top.Score.Score {
|
||||||
|
top = a
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return top.TGName
|
||||||
|
}
|
||||||
|
|
||||||
|
var alertFm = template.FuncMap{
|
||||||
|
"highest": highest,
|
||||||
|
}
|
||||||
|
|
||||||
|
const (
|
||||||
|
defaultBodyTemplStr = `{{ range . -}}
|
||||||
|
{{ .TGName }} is active with a score of {{ f .Score.Score 4 }}! ({{ f .Score.RecentCount 0 }}/{{ .Score.Count }} recent calls)
|
||||||
|
|
||||||
|
{{ end -}}`
|
||||||
|
defaultSubjectTemplStr = `Stillbox Alert ({{ highest . }})`
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
defaultTemplate *template.Template
|
||||||
|
)
|
||||||
|
|
||||||
|
func init() {
|
||||||
|
defaultTemplate = template.New("notification")
|
||||||
|
defaultTemplate.Funcs(common.FuncMap).Funcs(alertFm)
|
||||||
|
_, err := defaultTemplate.New("body").Parse(defaultBodyTemplStr)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
|
||||||
|
_, err = defaultTemplate.New("subject").Parse(defaultSubjectTemplStr)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Send renders and sends the Alerts.
|
||||||
|
func (b *backend) Send(ctx context.Context, alerts []alert.Alert) (err error) {
|
||||||
|
var subject, body bytes.Buffer
|
||||||
|
err = b.subject.ExecuteTemplate(&subject, "subject", alerts)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
err = b.body.ExecuteTemplate(&body, "body", alerts)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
err = b.Notify.Send(ctx, subject.String(), body.String())
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func buildSlackWebhookPayload(cfg *slackWebhookConfig) func(string, string) any {
|
||||||
type Attachment struct {
|
type Attachment struct {
|
||||||
Title string `json:"title"`
|
Title string `json:"title"`
|
||||||
Text string `json:"text"`
|
Text string `json:"text"`
|
||||||
|
@ -53,12 +129,44 @@ func (n *notifier) buildSlackWebhookPayload(cfg *slackWebhookConfig) func(string
|
||||||
}
|
}
|
||||||
|
|
||||||
type slackWebhookConfig struct {
|
type slackWebhookConfig struct {
|
||||||
WebhookURL string `mapstructure:"webhookURL"`
|
WebhookURL string `mapstructure:"webhookURL"`
|
||||||
Icon string `mapstructure:"icon"`
|
Icon string `mapstructure:"icon"`
|
||||||
MessageURL string `mapstructure:"messageURL"`
|
MessageURL string `mapstructure:"messageURL"`
|
||||||
|
SubjectTemplate string `mapstructure:"subjectTemplate"`
|
||||||
|
BodyTemplate string `mapstructure:"bodyTemplate"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func (n *notifier) addService(cfg config.NotifyService) error {
|
func (n *notifier) addService(cfg config.NotifyService) (err error) {
|
||||||
|
be := backend{}
|
||||||
|
|
||||||
|
switch cfg.SubjectTemplate {
|
||||||
|
case nil:
|
||||||
|
be.subject = defaultTemplate.Lookup("subject")
|
||||||
|
if be.subject == nil {
|
||||||
|
panic("subject template nil")
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
be.subject, err = template.New("subject").Funcs(common.FuncMap).Funcs(alertFm).Parse(*cfg.SubjectTemplate)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
switch cfg.BodyTemplate {
|
||||||
|
case nil:
|
||||||
|
be.body = defaultTemplate.Lookup("body")
|
||||||
|
if be.body == nil {
|
||||||
|
panic("body template nil")
|
||||||
|
}
|
||||||
|
default:
|
||||||
|
be.body, err = template.New("body").Funcs(common.FuncMap).Funcs(alertFm).Parse(*cfg.BodyTemplate)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
be.Notify = notify.New()
|
||||||
|
|
||||||
switch cfg.Provider {
|
switch cfg.Provider {
|
||||||
case "slackwebhook":
|
case "slackwebhook":
|
||||||
swc := &slackWebhookConfig{
|
swc := &slackWebhookConfig{
|
||||||
|
@ -74,20 +182,31 @@ func (n *notifier) addService(cfg config.NotifyService) error {
|
||||||
Header: make(stdhttp.Header),
|
Header: make(stdhttp.Header),
|
||||||
Method: stdhttp.MethodPost,
|
Method: stdhttp.MethodPost,
|
||||||
URL: swc.WebhookURL,
|
URL: swc.WebhookURL,
|
||||||
BuildPayload: n.buildSlackWebhookPayload(swc),
|
BuildPayload: buildSlackWebhookPayload(swc),
|
||||||
})
|
})
|
||||||
n.UseServices(hs)
|
be.UseServices(hs)
|
||||||
default:
|
default:
|
||||||
return fmt.Errorf("unknown provider '%s'", cfg.Provider)
|
return fmt.Errorf("unknown provider '%s'", cfg.Provider)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
n.backends = append(n.backends, be)
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (n *notifier) Send(ctx context.Context, alerts []alert.Alert) error {
|
||||||
|
for _, be := range n.backends {
|
||||||
|
err := be.Send(ctx, alerts)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func New(cfg config.Notify) (Notifier, error) {
|
func New(cfg config.Notify) (Notifier, error) {
|
||||||
n := ¬ifier{
|
n := new(notifier)
|
||||||
Notify: notify.NewWithServices(),
|
|
||||||
}
|
|
||||||
|
|
||||||
for _, s := range cfg {
|
for _, s := range cfg {
|
||||||
err := n.addService(s)
|
err := n.addService(s)
|
||||||
|
|
|
@ -9,6 +9,7 @@ import (
|
||||||
"dynatron.me/x/stillbox/internal/version"
|
"dynatron.me/x/stillbox/internal/version"
|
||||||
"dynatron.me/x/stillbox/pkg/config"
|
"dynatron.me/x/stillbox/pkg/config"
|
||||||
"dynatron.me/x/stillbox/pkg/database"
|
"dynatron.me/x/stillbox/pkg/database"
|
||||||
|
"dynatron.me/x/stillbox/pkg/talkgroups"
|
||||||
"github.com/go-chi/chi/v5"
|
"github.com/go-chi/chi/v5"
|
||||||
"github.com/go-chi/chi/v5/middleware"
|
"github.com/go-chi/chi/v5/middleware"
|
||||||
"github.com/go-chi/httprate"
|
"github.com/go-chi/httprate"
|
||||||
|
@ -26,7 +27,8 @@ func (s *Server) setupRoutes() {
|
||||||
}
|
}
|
||||||
|
|
||||||
r := s.r
|
r := s.r
|
||||||
r.Use(middleware.WithValue(database.DBCTXKeyValue, s.db))
|
r.Use(middleware.WithValue(database.DBCtxKeyValue, s.db))
|
||||||
|
r.Use(middleware.WithValue(talkgroups.StoreCtxKeyValue, s.tgs))
|
||||||
|
|
||||||
s.installPprof()
|
s.installPprof()
|
||||||
|
|
||||||
|
|
|
@ -61,7 +61,7 @@ func New(ctx context.Context, cfg *config.Config) (*Server, error) {
|
||||||
}
|
}
|
||||||
|
|
||||||
tgCache := talkgroups.NewCache()
|
tgCache := talkgroups.NewCache()
|
||||||
api := api.New(tgCache)
|
api := api.New()
|
||||||
|
|
||||||
srv := &Server{
|
srv := &Server{
|
||||||
auth: authenticator,
|
auth: authenticator,
|
||||||
|
|
|
@ -50,7 +50,7 @@ func (s *DatabaseSink) toAddCallParams(call *calls.Call) database.AddCallParams
|
||||||
AudioName: common.PtrOrNull(call.AudioName),
|
AudioName: common.PtrOrNull(call.AudioName),
|
||||||
AudioBlob: call.Audio,
|
AudioBlob: call.Audio,
|
||||||
AudioType: common.PtrOrNull(call.AudioType),
|
AudioType: common.PtrOrNull(call.AudioType),
|
||||||
Duration: call.Duration.Int32Ptr(),
|
Duration: call.Duration.MsInt32Ptr(),
|
||||||
Frequency: call.Frequency,
|
Frequency: call.Frequency,
|
||||||
Frequencies: call.Frequencies,
|
Frequencies: call.Frequencies,
|
||||||
Patches: call.Patches,
|
Patches: call.Patches,
|
||||||
|
|
|
@ -49,16 +49,16 @@ type Store interface {
|
||||||
HUP(*config.Config)
|
HUP(*config.Config)
|
||||||
}
|
}
|
||||||
|
|
||||||
type CtxStoreKeyT string
|
type CtxStoreKey string
|
||||||
|
|
||||||
const CtxStoreKey CtxStoreKeyT = "store"
|
const StoreCtxKeyValue CtxStoreKey = "store"
|
||||||
|
|
||||||
func CtxWithStore(ctx context.Context, s Store) context.Context {
|
func CtxWithStore(ctx context.Context, s Store) context.Context {
|
||||||
return context.WithValue(ctx, CtxStoreKey, s)
|
return context.WithValue(ctx, StoreCtxKeyValue, s)
|
||||||
}
|
}
|
||||||
|
|
||||||
func StoreFrom(ctx context.Context) Store {
|
func StoreFrom(ctx context.Context) Store {
|
||||||
s, ok := ctx.Value(CtxStoreKey).(Store)
|
s, ok := ctx.Value(StoreCtxKeyValue).(Store)
|
||||||
if !ok {
|
if !ok {
|
||||||
return NewCache()
|
return NewCache()
|
||||||
}
|
}
|
||||||
|
@ -127,7 +127,7 @@ func (t *cache) add(rec *Talkgroup) error {
|
||||||
t.Lock()
|
t.Lock()
|
||||||
defer t.Unlock()
|
defer t.Unlock()
|
||||||
|
|
||||||
tg := TG(rec.System.ID, int(rec.Talkgroup.Tgid))
|
tg := TG(rec.System.ID, rec.Talkgroup.Tgid)
|
||||||
t.tgs[tg] = rec
|
t.tgs[tg] = rec
|
||||||
t.systems[int32(rec.System.ID)] = rec.System.Name
|
t.systems[int32(rec.System.ID)] = rec.System.Name
|
||||||
|
|
||||||
|
|
|
@ -12,6 +12,11 @@ type Talkgroup struct {
|
||||||
Learned bool `json:"learned"`
|
Learned bool `json:"learned"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type Names struct {
|
||||||
|
System string
|
||||||
|
Talkgroup string
|
||||||
|
}
|
||||||
|
|
||||||
type ID struct {
|
type ID struct {
|
||||||
System uint32 `json:"sys"`
|
System uint32 `json:"sys"`
|
||||||
Talkgroup uint32 `json:"tg"`
|
Talkgroup uint32 `json:"tg"`
|
||||||
|
@ -28,7 +33,11 @@ func (ids *IDs) Packed() []int64 {
|
||||||
return r
|
return r
|
||||||
}
|
}
|
||||||
|
|
||||||
func TG[T int | uint | int64 | uint64 | int32 | uint32](sys, tgid T) ID {
|
type intId interface {
|
||||||
|
int | uint | int64 | uint64 | int32 | uint32
|
||||||
|
}
|
||||||
|
|
||||||
|
func TG[T intId, U intId](sys T, tgid U) ID {
|
||||||
return ID{
|
return ID{
|
||||||
System: uint32(sys),
|
System: uint32(sys),
|
||||||
Talkgroup: uint32(tgid),
|
Talkgroup: uint32(tgid),
|
||||||
|
@ -42,14 +51,5 @@ func (t ID) Pack() int64 {
|
||||||
|
|
||||||
func (t ID) String() string {
|
func (t ID) String() string {
|
||||||
return fmt.Sprintf("%d:%d", t.System, t.Talkgroup)
|
return fmt.Sprintf("%d:%d", t.System, t.Talkgroup)
|
||||||
}
|
|
||||||
|
|
||||||
func PackedTGs(tg []ID) []int64 {
|
|
||||||
s := make([]int64, len(tg))
|
|
||||||
|
|
||||||
for i, v := range tg {
|
|
||||||
s[i] = v.Pack()
|
|
||||||
}
|
|
||||||
|
|
||||||
return s
|
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue