Add Rybbit analytics tracking

Author: auravoid

README.md

@@ -9,6 +9,7 @@ A minimal, self-hosted URL shortener written in Go using bbolt (BoltDB). This pr
 - List stored URLs with cursor-based pagination
 - Simple token-based authorization
 - Single-file embedded database (bbolt)
+- Tracking analytics with Rybbit (other providers soon to come)
 
 ## Quick Start
 
@@ -44,6 +45,9 @@ Environment variables:
 - `APP_TOKEN` (required): token used for authorization on protected endpoints
 - `PORT` (optional): HTTP port (default `8080`)
 - `DB_PATH` (optional): path to BoltDB file (default `short-it.db`)
+- `RYBBIT_SITE_ID` (optional): Site ID provided by Rybbit
+- `RYBBIT_SITE_KEY` (optional): API key found in account settings for Rybbit
+- `RYBBIT_SITE_URL` (optional): Base URL for your Rybbit instance
 
 ## HTTP API
 
@@ -111,4 +115,4 @@ Key implementation files:
 
 - The app uses a BoltDB bucket named `urls` to store key → URL mappings.
 - The autogenerated keys are 6 characters drawn from `a-zA-Z0-9`.
-
+- Sends pageview events to Rybbit if configured with the hostname, language, pathname, user-agent, referrer, and IP address gathered from `X-Forwarded-For`.

cmd/short-it/main.go

@@ -1,21 +1,30 @@
 package main
 
 import (
+	"bytes"
+	"context"
 	"crypto/rand"
 	"encoding/json"
 	"fmt"
+	"io"
 	"log"
 	"net/http"
 	"os"
 	"strconv"
 	"strings"
+	"time"
 
 	"go.etcd.io/bbolt"
 )
 
 var db *bbolt.DB
 var authToken string
 
+var rybbitSiteID = os.Getenv("RYBBIT_SITE_ID")
+var rybbitSiteKey = os.Getenv("RYBBIT_SITE_KEY")
+var rybbitSiteURL = os.Getenv("RYBBIT_SITE_URL")
+var rybbitClient = &http.Client{Timeout: 3 * time.Second}
+
 const bucketName = "urls"
 const maxPageSize = 100
 
@@ -216,6 +225,7 @@ func handleGetURL(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 
+	handlePageView(r)
 	http.Redirect(w, r, url, http.StatusFound)
 }
 
@@ -269,6 +279,65 @@ func handleDeleteURL(w http.ResponseWriter, r *http.Request, path string) {
 	w.WriteHeader(http.StatusNoContent)
 }
 
+func handlePageView(r *http.Request) {
+    if rybbitSiteID == "" || rybbitSiteKey == "" || rybbitSiteURL == "" {
+        return
+    }
+
+    ip := r.Header.Get("X-Forwarded-For")
+    if ip == "" {
+        ip = strings.Split(r.RemoteAddr, ":")[0]
+    } else {
+        ip = strings.TrimSpace(strings.Split(ip, ",")[0])
+    }
+
+    hostname := r.Host
+    if hostname == "" {
+        if h := r.URL.Hostname(); h != "" {
+            hostname = h
+        }
+    }
+
+    data := map[string]string{
+        "site_id":    rybbitSiteID,
+        "type":       "pageview",
+        "pathname":   r.URL.Path,
+        "hostname":   hostname,
+        "referrer":   r.Referer(),
+        "language":   r.Header.Get("Accept-Language"),
+        "user_agent": r.UserAgent(),
+        "ip_address": ip,
+    }
+
+    jsonData, err := json.Marshal(data)
+    if err != nil {
+        return
+    }
+
+    go func(body []byte) {
+        ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
+        defer cancel()
+
+        req, err := http.NewRequestWithContext(ctx, "POST", strings.TrimRight(rybbitSiteURL, "/")+"/api/track", bytes.NewReader(body))
+        if err != nil {
+            return
+        }
+        req.Header.Set("Content-Type", "application/json")
+        req.Header.Set("Authorization", "Bearer "+rybbitSiteKey)
+
+        resp, err := rybbitClient.Do(req)
+        if err != nil {
+            return
+        }
+        defer resp.Body.Close()
+
+        if resp.StatusCode < 200 || resp.StatusCode >= 300 {
+            b, _ := io.ReadAll(resp.Body)
+            log.Printf("Rybbit tracking error: %s", string(b))
+        }
+    }(jsonData)
+}
+
 func main() {
 	authToken = os.Getenv("APP_TOKEN")
 	if authToken == "" {

docker-compose.yml

@@ -6,6 +6,9 @@ services:
     ports:
       - "8080:8080"
     environment:
+      #- RYBBIT_SITE_ID=YOUR_SITE_ID_HERE
+      #- RYBBIT_SITE_KEY=YOUR_API_KEY_HERE
+      #- RYBBIT_SITE_URL=https://your-rybbit-instance.example.com
       - APP_TOKEN=YOUR_TOKEN_HERE
       - PORT=8080
       - DB_PATH=/data/short-it.db