mirror of
https://github.com/chuot/rdio-scanner.git
synced 2026-08-13 16:33:01 -06:00
539 lines
12 KiB
Go
539 lines
12 KiB
Go
// Copyright (C) 2019-2022 Chrystian Huot <chrystian.huot@saubeo.solutions>
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program. If not, see <http://www.gnu.org/licenses/>
|
|
|
|
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"database/sql"
|
|
"encoding/json"
|
|
"fmt"
|
|
"mime/multipart"
|
|
"net/http"
|
|
"net/url"
|
|
"path"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/google/uuid"
|
|
)
|
|
|
|
type Downstream struct {
|
|
Id interface{} `json:"_id"`
|
|
Apikey string `json:"apiKey"`
|
|
Disabled bool `json:"disabled"`
|
|
Order interface{} `json:"order"`
|
|
Systems interface{} `json:"systems"`
|
|
Url string `json:"url"`
|
|
}
|
|
|
|
func (downstream *Downstream) FromMap(m map[string]interface{}) {
|
|
switch v := m["_id"].(type) {
|
|
case float64:
|
|
downstream.Id = uint(v)
|
|
}
|
|
|
|
switch v := m["apiKey"].(type) {
|
|
case string:
|
|
downstream.Apikey = v
|
|
}
|
|
|
|
switch v := m["disabled"].(type) {
|
|
case bool:
|
|
downstream.Disabled = v
|
|
}
|
|
|
|
switch v := m["order"].(type) {
|
|
case float64:
|
|
downstream.Order = uint(v)
|
|
}
|
|
|
|
switch v := m["systems"].(type) {
|
|
case []interface{}:
|
|
if b, err := json.Marshal(v); err == nil {
|
|
downstream.Systems = string(b)
|
|
}
|
|
case string:
|
|
downstream.Systems = v
|
|
}
|
|
|
|
switch v := m["url"].(type) {
|
|
case string:
|
|
downstream.Url = v
|
|
}
|
|
}
|
|
|
|
func (downstream *Downstream) HasAccess(call *Call) bool {
|
|
if downstream.Disabled {
|
|
return false
|
|
}
|
|
|
|
switch v := downstream.Systems.(type) {
|
|
case []interface{}:
|
|
for _, f := range v {
|
|
switch v := f.(type) {
|
|
case map[string]interface{}:
|
|
switch id := v["id"].(type) {
|
|
case float64:
|
|
if id == float64(call.System) {
|
|
switch tg := v["talkgroups"].(type) {
|
|
case string:
|
|
if tg == "*" {
|
|
return true
|
|
}
|
|
case []interface{}:
|
|
for _, f := range tg {
|
|
switch tg := f.(type) {
|
|
case float64:
|
|
if tg == float64(call.Talkgroup) {
|
|
return true
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
case string:
|
|
if v == "*" {
|
|
return true
|
|
}
|
|
}
|
|
|
|
return false
|
|
}
|
|
|
|
func (downstream *Downstream) Send(call *Call) error {
|
|
var (
|
|
audioName string
|
|
buf = bytes.Buffer{}
|
|
)
|
|
|
|
if downstream.Disabled {
|
|
return nil
|
|
}
|
|
|
|
formatError := func(err error) error {
|
|
return fmt.Errorf("downstream.send: %s", err.Error())
|
|
}
|
|
|
|
mw := multipart.NewWriter(&buf)
|
|
|
|
switch v := call.AudioName.(type) {
|
|
case string:
|
|
audioName = v
|
|
}
|
|
|
|
if w, err := mw.CreateFormFile("audio", audioName); err == nil {
|
|
if _, err = w.Write(call.Audio); err != nil {
|
|
return formatError(err)
|
|
}
|
|
} else {
|
|
return formatError(err)
|
|
}
|
|
|
|
switch v := call.AudioName.(type) {
|
|
case string:
|
|
if w, err := mw.CreateFormField("audioName"); err == nil {
|
|
if _, err = w.Write([]byte(v)); err != nil {
|
|
return formatError(err)
|
|
}
|
|
} else {
|
|
return formatError(err)
|
|
}
|
|
}
|
|
|
|
switch v := call.AudioType.(type) {
|
|
case string:
|
|
if w, err := mw.CreateFormField("audioType"); err == nil {
|
|
if _, err = w.Write([]byte(v)); err != nil {
|
|
return formatError(err)
|
|
}
|
|
} else {
|
|
return formatError(err)
|
|
}
|
|
}
|
|
|
|
if w, err := mw.CreateFormField("dateTime"); err == nil {
|
|
if _, err = w.Write([]byte(call.DateTime.Format(time.RFC3339))); err != nil {
|
|
return formatError(err)
|
|
}
|
|
} else {
|
|
return formatError(err)
|
|
}
|
|
|
|
switch v := call.Frequencies.(type) {
|
|
case []map[string]interface{}:
|
|
if w, err := mw.CreateFormField("frequencies"); err == nil {
|
|
if b, err := json.Marshal(v); err == nil {
|
|
if _, err = w.Write(b); err != nil {
|
|
return formatError(err)
|
|
}
|
|
} else {
|
|
return formatError(err)
|
|
}
|
|
} else {
|
|
return formatError(err)
|
|
}
|
|
}
|
|
|
|
switch v := call.Frequency.(type) {
|
|
case uint:
|
|
if w, err := mw.CreateFormField("frequency"); err == nil {
|
|
if _, err = w.Write([]byte(fmt.Sprintf("%v", v))); err != nil {
|
|
return formatError(err)
|
|
}
|
|
} else {
|
|
return formatError(err)
|
|
}
|
|
}
|
|
|
|
if w, err := mw.CreateFormField("key"); err == nil {
|
|
if _, err = w.Write([]byte(downstream.Apikey)); err != nil {
|
|
return formatError(err)
|
|
}
|
|
} else {
|
|
return formatError(err)
|
|
}
|
|
|
|
switch v := call.Patches.(type) {
|
|
case []uint:
|
|
if w, err := mw.CreateFormField("patches"); err == nil {
|
|
if b, err := json.Marshal(v); err == nil {
|
|
if _, err = w.Write(b); err != nil {
|
|
return formatError(err)
|
|
}
|
|
} else {
|
|
return formatError(err)
|
|
}
|
|
} else {
|
|
return formatError(err)
|
|
}
|
|
}
|
|
|
|
switch v := call.Source.(type) {
|
|
case uint:
|
|
if w, err := mw.CreateFormField("source"); err == nil {
|
|
if _, err = w.Write([]byte(fmt.Sprintf("%v", v))); err != nil {
|
|
return formatError(err)
|
|
}
|
|
} else {
|
|
return formatError(err)
|
|
}
|
|
}
|
|
|
|
switch v := call.Sources.(type) {
|
|
case []map[string]interface{}:
|
|
if w, err := mw.CreateFormField("sources"); err == nil {
|
|
if b, err := json.Marshal(v); err == nil {
|
|
if _, err = w.Write(b); err != nil {
|
|
return formatError(err)
|
|
}
|
|
} else {
|
|
return formatError(err)
|
|
}
|
|
} else {
|
|
return formatError(err)
|
|
}
|
|
}
|
|
|
|
if w, err := mw.CreateFormField("system"); err == nil {
|
|
if _, err = w.Write([]byte(fmt.Sprintf("%v", call.System))); err != nil {
|
|
return formatError(err)
|
|
}
|
|
} else {
|
|
return formatError(err)
|
|
}
|
|
|
|
switch v := call.systemLabel.(type) {
|
|
case string:
|
|
if w, err := mw.CreateFormField("systemLabel"); err == nil {
|
|
if _, err = w.Write([]byte(v)); err != nil {
|
|
return formatError(err)
|
|
}
|
|
} else {
|
|
return formatError(err)
|
|
}
|
|
}
|
|
|
|
if w, err := mw.CreateFormField("talkgroup"); err == nil {
|
|
if _, err = w.Write([]byte(fmt.Sprintf("%v", call.Talkgroup))); err != nil {
|
|
return formatError(err)
|
|
}
|
|
} else {
|
|
return formatError(err)
|
|
}
|
|
|
|
switch v := call.talkgroupGroup.(type) {
|
|
case string:
|
|
if w, err := mw.CreateFormField("talkgroupGroup"); err == nil {
|
|
if _, err = w.Write([]byte(v)); err != nil {
|
|
return formatError(err)
|
|
}
|
|
} else {
|
|
return formatError(err)
|
|
}
|
|
}
|
|
|
|
switch v := call.talkgroupLabel.(type) {
|
|
case string:
|
|
if w, err := mw.CreateFormField("talkgroupLabel"); err == nil {
|
|
if _, err = w.Write([]byte(v)); err != nil {
|
|
return formatError(err)
|
|
}
|
|
} else {
|
|
return formatError(err)
|
|
}
|
|
}
|
|
|
|
switch v := call.talkgroupName.(type) {
|
|
case string:
|
|
if w, err := mw.CreateFormField("talkgroupName"); err == nil {
|
|
if _, err = w.Write([]byte(v)); err != nil {
|
|
return formatError(err)
|
|
}
|
|
} else {
|
|
return formatError(err)
|
|
}
|
|
}
|
|
|
|
switch v := call.talkgroupTag.(type) {
|
|
case string:
|
|
if w, err := mw.CreateFormField("talkgroupTag"); err == nil {
|
|
if _, err = w.Write([]byte(v)); err != nil {
|
|
return formatError(err)
|
|
}
|
|
} else {
|
|
return formatError(err)
|
|
}
|
|
}
|
|
|
|
if err := mw.Close(); err != nil {
|
|
return formatError(err)
|
|
}
|
|
|
|
if u, err := url.Parse(downstream.Url); err == nil {
|
|
u.Path = path.Join(u.Path, "/api/call-upload")
|
|
|
|
c := http.Client{Timeout: 10 * time.Second}
|
|
|
|
if res, err := c.Post(u.String(), mw.FormDataContentType(), &buf); err == nil {
|
|
if res.StatusCode != http.StatusOK {
|
|
return formatError(fmt.Errorf("bad status: %s", res.Status))
|
|
}
|
|
|
|
} else {
|
|
return formatError(err)
|
|
}
|
|
|
|
} else {
|
|
return formatError(err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
type Downstreams struct {
|
|
List []*Downstream
|
|
mutex sync.Mutex
|
|
}
|
|
|
|
func NewDownstreams() *Downstreams {
|
|
return &Downstreams{
|
|
List: []*Downstream{},
|
|
mutex: sync.Mutex{},
|
|
}
|
|
}
|
|
|
|
func (downstreams *Downstreams) FromMap(f []interface{}) {
|
|
downstreams.mutex.Lock()
|
|
defer downstreams.mutex.Unlock()
|
|
|
|
downstreams.List = []*Downstream{}
|
|
|
|
for _, r := range f {
|
|
switch m := r.(type) {
|
|
case map[string]interface{}:
|
|
downstream := &Downstream{}
|
|
downstream.FromMap(m)
|
|
downstreams.List = append(downstreams.List, downstream)
|
|
}
|
|
}
|
|
}
|
|
|
|
func (downstreams *Downstreams) Read(db *Database) error {
|
|
var (
|
|
err error
|
|
id sql.NullFloat64
|
|
order sql.NullFloat64
|
|
rows *sql.Rows
|
|
systems string
|
|
)
|
|
|
|
downstreams.mutex.Lock()
|
|
defer downstreams.mutex.Unlock()
|
|
|
|
downstreams.List = []*Downstream{}
|
|
|
|
formatError := func(err error) error {
|
|
return fmt.Errorf("downstreams.read: %v", err)
|
|
}
|
|
|
|
if rows, err = db.Sql.Query("select `_id`, `apiKey`, `disabled`, `order`, `systems`, `url` from `rdioScannerDownstreams`"); err != nil {
|
|
return formatError(err)
|
|
}
|
|
|
|
for rows.Next() {
|
|
downstream := &Downstream{}
|
|
|
|
if err = rows.Scan(&id, &downstream.Apikey, &downstream.Disabled, &order, &systems, &downstream.Url); err != nil {
|
|
break
|
|
}
|
|
|
|
if id.Valid && id.Float64 > 0 {
|
|
downstream.Id = uint(id.Float64)
|
|
}
|
|
|
|
if len(downstream.Apikey) == 0 {
|
|
downstream.Apikey = uuid.New().String()
|
|
}
|
|
|
|
if order.Valid && order.Float64 > 0 {
|
|
downstream.Order = uint(order.Float64)
|
|
}
|
|
|
|
if err = json.Unmarshal([]byte(systems), &downstream.Systems); err != nil {
|
|
downstream.Systems = []interface{}{}
|
|
}
|
|
|
|
if len(downstream.Url) == 0 {
|
|
continue
|
|
}
|
|
|
|
downstreams.List = append(downstreams.List, downstream)
|
|
}
|
|
|
|
rows.Close()
|
|
|
|
if err != nil {
|
|
return formatError(err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (downstreams *Downstreams) Send(controller *Controller, call *Call) {
|
|
for _, downstream := range downstreams.List {
|
|
logEvent := func(logLevel string, message string) {
|
|
controller.Logs.LogEvent(logLevel, fmt.Sprintf("downstream: system=%v talkgroup=%v file=%v to %v %v", call.System, call.Talkgroup, call.AudioName, downstream.Url, message))
|
|
}
|
|
|
|
if downstream.HasAccess(call) {
|
|
if err := downstream.Send(call); err == nil {
|
|
logEvent(LogLevelInfo, "success")
|
|
} else {
|
|
logEvent(LogLevelError, err.Error())
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
func (downstreams *Downstreams) Write(db *Database) error {
|
|
var (
|
|
count uint
|
|
err error
|
|
rows *sql.Rows
|
|
rowIds = []uint{}
|
|
systems interface{}
|
|
)
|
|
|
|
downstreams.mutex.Lock()
|
|
defer downstreams.mutex.Unlock()
|
|
|
|
formatError := func(err error) error {
|
|
return fmt.Errorf("downstreams.write: %v", err)
|
|
}
|
|
|
|
for _, downstream := range downstreams.List {
|
|
switch downstream.Systems {
|
|
case "*":
|
|
systems = `"*"`
|
|
default:
|
|
systems = downstream.Systems
|
|
}
|
|
|
|
if err = db.Sql.QueryRow("select count(*) from `rdioScannerDownstreams` where `_id` = ?", downstream.Id).Scan(&count); err != nil {
|
|
break
|
|
}
|
|
|
|
if count == 0 {
|
|
if _, err = db.Sql.Exec("insert into `rdioScannerDownstreams` (`_id`, `apiKey`, `disabled`, `order`, `systems`, `url`) values (?, ?, ?, ?, ?, ?)", downstream.Id, downstream.Apikey, downstream.Disabled, downstream.Order, systems, downstream.Url); err != nil {
|
|
break
|
|
}
|
|
|
|
} else if _, err = db.Sql.Exec("update `rdioScannerDownstreams` set `_id` = ?, `apiKey` = ?, `disabled` = ?, `order` = ?, `systems` = ?, `url` = ? where `_id` = ?", downstream.Id, downstream.Apikey, downstream.Disabled, downstream.Order, systems, downstream.Url, downstream.Id); err != nil {
|
|
break
|
|
}
|
|
}
|
|
|
|
if err != nil {
|
|
return formatError(err)
|
|
}
|
|
|
|
if rows, err = db.Sql.Query("select `_id` from `rdioScannerDownstreams`"); err != nil {
|
|
return formatError(err)
|
|
}
|
|
|
|
for rows.Next() {
|
|
var rowId uint
|
|
if err = rows.Scan(&rowId); err != nil {
|
|
break
|
|
}
|
|
remove := true
|
|
for _, downstream := range downstreams.List {
|
|
if downstream.Id == nil || downstream.Id == rowId {
|
|
remove = false
|
|
break
|
|
}
|
|
}
|
|
if remove {
|
|
rowIds = append(rowIds, rowId)
|
|
}
|
|
}
|
|
|
|
rows.Close()
|
|
|
|
if err != nil {
|
|
return formatError(err)
|
|
}
|
|
|
|
if len(rowIds) > 0 {
|
|
if b, err := json.Marshal(rowIds); err == nil {
|
|
s := string(b)
|
|
s = strings.ReplaceAll(s, "[", "(")
|
|
s = strings.ReplaceAll(s, "]", ")")
|
|
q := fmt.Sprintf("delete from `rdioScannerDownstreams` where `_id` in %v", s)
|
|
if _, err = db.Sql.Exec(q); err != nil {
|
|
return formatError(err)
|
|
}
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|