dashboard / erock/tuns / chore(analytics): send content-type to `metric-drain` #38 rss

old

new

 	"github.com/gin-gonic/gin"
 )
 
+func createMetricDrainMdw(state *utils.State) func(*gin.Context) {
+	return func(c *gin.Context) {
+		fmt.Println(c.Keys)
+		originalURI := c.Keys["originalURI"].(string)
+
+		if viper.GetString("admin-console-token") != "" && strings.Contains(originalURI, viper.GetString("admin-console-token")) {
+			originalURI = strings.Replace(originalURI, viper.GetString("admin-console-token"), "[REDACTED]", 1)
+		}
+
+		if viper.GetString("service-console-token") != "" && strings.Contains(originalURI, viper.GetString("service-console-token")) {
+			originalURI = strings.Replace(originalURI, viper.GetString("service-console-token"), "[REDACTED]", 1)
+		}
+
+		if c.Keys["httpHolder"] != nil {
+			currentListener := c.Keys["httpHolder"].(*utils.HTTPHolder)
+
+			if currentListener != nil {
+				currentListener.SSHConnections.Range(func(I string, conn *utils.SSHConnection) bool {
+					userId := conn.UserId()
+
+					visit := &utils.MetricDrainVisit{
+						Namespace: "tuns",
+						UserID:    userId,
+						Host:      c.Request.Host,
+						IpAddress: c.ClientIP(),
+						Status:    c.Writer.Status(),
+						Path:      originalURI,
+						UserAgent: c.Request.UserAgent(),
+						Referer:   c.Request.Referer(),
+					}
+
+					visit.ContentType = c.Writer.Header().Get("content-type")
+
+					select {
+					case state.MetricDrain <- visit:
+					default:
+						slog.Error("unable to send metric drain visit", slog.Any("visit", visit))
+					}
+
+					return true
+				})
+			}
+		}
+
+		c.Next()
+	}
+}
+
 // Start initializes the HTTP service.
 func Start(state *utils.State) {
 	releaseMode := gin.ReleaseMode
 
 					slog.Info(logLine, "user", user, "userId", userId)
 
-					visit := &utils.MetricDrainVisit{
-						Namespace: "tuns",
-						UserID:    userId,
-						Host:      param.Request.Host,
-						IpAddress: param.ClientIP,
-						Status:    param.StatusCode,
-						Path:      originalURI,
-						UserAgent: param.Request.UserAgent(),
-						Referer:   param.Request.Referer(),
-					}
-
-					select {
-					case state.MetricDrain <- visit:
-					default:
-						slog.Error("unable to send metric drain visit", slog.Any("visit", visit))
-					}
-
 					return true
 				})
 			}
 		}
 
 		gin.WrapH(currentListener.Balancer)(c)
-	})
+	}, createMetricDrainMdw(state))
 
 	var acmeIssuer *certmagic.ACMEIssuer = nil
 

old

new

 
 // MetricDrainVisit represents a visit for the metric drain.
 type MetricDrainVisit struct {
-	Namespace string `json:"namespace"`
-	UserID    string `json:"user_id"`
-	Host      string `json:"host"`
-	Path      string `json:"path"`
-	IpAddress string `json:"ip_address"`
-	UserAgent string `json:"user_agent"`
-	Referer   string `json:"referer"`
-	Status    int    `json:"status"`
+	Namespace   string `json:"namespace"`
+	UserID      string `json:"user_id"`
+	Host        string `json:"host"`
+	Path        string `json:"path"`
+	IpAddress   string `json:"ip_address"`
+	UserAgent   string `json:"user_agent"`
+	Referer     string `json:"referer"`
+	Status      int    `json:"status"`
+	ContentType string `json:"content_type"`
 }
 
 // State handles overall state. It retains mutexed maps for various