From b44a8d914876f0d0f8bc57c226b6e61f42ae0159 Mon Sep 17 00:00:00 2001
From: Torrey Searle <torrey.searle@wavecrest.com>
Date: Mon, 12 Feb 2024 12:34:10 +0100
Subject: [PATCH] add cid to rtcp

---
 README.md         |  4 ++--
 remotelog/loki.go | 23 ++++++++++++++++++++++-
 2 files changed, 24 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index d3a2a78..e571ce3 100644
--- a/README.md
+++ b/README.md
@@ -13,9 +13,9 @@ You have 3 options to get **heplify-server** up and running:
 * Docker [compose](https://github.com/sipcapture/heplify-server/tree/master/docker/hom5-hep-prom-graf)
 * Compile from sources:  
   
-  Install luajit dev libary
+  Install gcc and luajit dev libary
   
-  `apt-get install libluajit-5.1-dev`
+  `apt-get install build-essential libluajit-5.1-dev`
   
   or 
   
diff --git a/remotelog/loki.go b/remotelog/loki.go
index 214b8d3..8ab309f 100644
--- a/remotelog/loki.go
+++ b/remotelog/loki.go
@@ -4,6 +4,7 @@ import (
 	"bufio"
 	"bytes"
 	"context"
+	"encoding/json"
 	"fmt"
 	"io"
 	"net/http"
@@ -106,7 +107,26 @@ func (l *Loki) start(hCh chan *decoder.HEP) {
 			lastPktTime = curPktTime
 
 			pktMeta.Reset()
-			pktMeta.WriteString(pkt.Payload)
+
+			if pkt.ProtoString == "rtcp" {
+				var document map[string]interface{}
+				err := json.Unmarshal([]byte(pkt.Payload), &document)
+				if err != nil {
+					logp.Err("Unable to decode rtcp json: %v", err)
+					pktMeta.WriteString(pkt.Payload)
+				} else {
+					document["cid"] = pkt.CID
+					documentJson, err := json.Marshal(document)
+					if err != nil {
+						logp.Err("Unable to re-generate rtcp json: %v", err)
+						pktMeta.WriteString(pkt.Payload)
+					} else {
+						pktMeta.Write(documentJson)
+					}
+				}
+			} else {
+				pktMeta.WriteString(pkt.Payload)
+			}
 			l.entry = entry{model.LabelSet{}, logproto.Entry{Timestamp: curPktTime}}
 
 			switch {
@@ -121,6 +141,7 @@ func (l *Loki) start(hCh chan *decoder.HEP) {
 					protocol = "sip"
 				}
 				l.entry.labels["protocol"] = model.LabelValue(protocol)
+
 			}
 
 			l.entry.labels["job"] = jobName