91 lines
1.5 KiB
Go
91 lines
1.5 KiB
Go
package smshook
|
|
|
|
import (
|
|
"encoding/json"
|
|
"net/http"
|
|
"sync"
|
|
|
|
"SmsHook/ringbuffer"
|
|
)
|
|
|
|
|
|
type WebhookServer struct {
|
|
buffer *ringbuffer.RingBuffer
|
|
cond *sync.Cond
|
|
mux *http.ServeMux
|
|
}
|
|
|
|
|
|
type WebhookPayload struct {
|
|
Content string `json:"content"`
|
|
}
|
|
|
|
|
|
func NewWebhookServer(bufferSize int) *WebhookServer {
|
|
mutex := sync.Mutex{}
|
|
server := &WebhookServer{
|
|
buffer: ringbuffer.NewRingBuffer(bufferSize),
|
|
cond: sync.NewCond(&mutex),
|
|
mux: http.NewServeMux(),
|
|
}
|
|
|
|
server.mux.HandleFunc("/webhook", server.webhookHandler)
|
|
return server
|
|
}
|
|
|
|
|
|
func (s *WebhookServer) webhookHandler(w http.ResponseWriter, r *http.Request) {
|
|
var payload WebhookPayload
|
|
|
|
err := json.NewDecoder(r.Body).Decode(&payload)
|
|
if err != nil {
|
|
http.Error(w, err.Error(), http.StatusBadRequest)
|
|
return
|
|
}
|
|
|
|
s.buffer.Add(payload.Content)
|
|
s.cond.Broadcast()
|
|
}
|
|
|
|
|
|
func (s *WebhookServer) GetLastItem() (string, int) {
|
|
return s.buffer.GetLast()
|
|
}
|
|
|
|
|
|
func (s *WebhookServer) WaitForNewItem() string {
|
|
s.cond.L.Lock()
|
|
s.cond.Wait()
|
|
newItem := s.buffer.WaitForNewItem()
|
|
s.cond.L.Unlock()
|
|
return newItem
|
|
}
|
|
|
|
|
|
func (s *WebhookServer) Start(address string) error {
|
|
return http.ListenAndServe(address, s.mux)
|
|
}
|
|
|
|
var server *WebhookServer
|
|
|
|
|
|
func Init(bufferSize int, address string) {
|
|
server = NewWebhookServer(bufferSize)
|
|
go func() {
|
|
if err := server.Start(address); err != nil {
|
|
panic(err)
|
|
}
|
|
}()
|
|
}
|
|
|
|
|
|
func GetLast() (string, int) {
|
|
return server.GetLastItem()
|
|
}
|
|
|
|
|
|
func WaitForNew() string {
|
|
return server.WaitForNewItem()
|
|
}
|
|
|