Hook impl

smshook/smshook.go

@@ -0,0 +1,90 @@
+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()
+}
+