Initial commit
This commit is contained in:
commit
0ddeff0c90
|
@ -0,0 +1 @@
|
|||
*.sqlite
|
|
@ -0,0 +1,10 @@
|
|||
# Dockerfile
|
||||
FROM golang:latest
|
||||
|
||||
WORKDIR /app
|
||||
|
||||
COPY . .
|
||||
|
||||
RUN go build -o main .
|
||||
|
||||
CMD ["/app/main"]
|
|
@ -0,0 +1,17 @@
|
|||
<body>
|
||||
<div class="container">
|
||||
<h2>Your Tracking Pixel Link</h2>
|
||||
<input type="text" id="pixelLink" value="Generated Link Here" readonly>
|
||||
<button onclick="copyLink()">Copy Link</button>
|
||||
</div>
|
||||
|
||||
<script>
|
||||
function copyLink() {
|
||||
var copyText = document.getElementById("pixelLink");
|
||||
copyText.select();
|
||||
copyText.setSelectionRange(0, 99999); /* For mobile devices */
|
||||
document.execCommand("copy");
|
||||
alert("Copied the link: " + copyText.value);
|
||||
}
|
||||
</script>
|
||||
</body>
|
|
@ -0,0 +1,28 @@
|
|||
<!DOCTYPE html>
|
||||
<html>
|
||||
|
||||
<head>
|
||||
<title>Dashboard</title>
|
||||
</head>
|
||||
|
||||
<body>
|
||||
<h1>Tracking Pixel Dashboard</h1>
|
||||
<table>
|
||||
<tr>
|
||||
<th>Title</th>
|
||||
<th>Total Views</th>
|
||||
<th>Unique Views</th>
|
||||
<th>Link</th>
|
||||
</tr>
|
||||
{{range .}}
|
||||
<tr>
|
||||
<td>{{.Title}}</td>
|
||||
<td>{{.TotalViews}}</td>
|
||||
<td>{{.UniqueViews}}</td>
|
||||
<td><a href="/p/{{.UUID}}" target="_blank">link</a></td>
|
||||
</tr>
|
||||
{{end}}
|
||||
</table>
|
||||
</body>
|
||||
|
||||
</html>
|
|
@ -0,0 +1,66 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"log"
|
||||
)
|
||||
|
||||
var db *sql.DB
|
||||
|
||||
func initDB(filepath string) *sql.DB {
|
||||
db, err := sql.Open("sqlite3", filepath)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
|
||||
// Create tables if they do not exist
|
||||
_, err = db.Exec(`
|
||||
CREATE TABLE IF NOT EXISTS pixels (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
uuid TEXT UNIQUE,
|
||||
title TEXT
|
||||
);
|
||||
CREATE TABLE IF NOT EXISTS stats (
|
||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||
pixel_id INTEGER,
|
||||
view_time DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||||
ip TEXT,
|
||||
user_agent TEXT,
|
||||
fingerprint TEXT,
|
||||
FOREIGN KEY(pixel_id) REFERENCES pixels(id)
|
||||
);
|
||||
`)
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
|
||||
return db
|
||||
}
|
||||
|
||||
func addStat(pixelID int, ip, userAgent string, fingerprint string) error {
|
||||
statement, err := db.Prepare("INSERT INTO stats (pixel_id, ip, user_agent, fingerprint) VALUES (?, ?, ?, ?)")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
_, err = statement.Exec(pixelID, ip, userAgent, fingerprint)
|
||||
return err
|
||||
}
|
||||
|
||||
func getPixelIDFromUUID(uuid string) (int, error) {
|
||||
var id int
|
||||
row := db.QueryRow("SELECT id FROM pixels WHERE uuid = ?", uuid)
|
||||
err := row.Scan(&id)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
return id, nil
|
||||
}
|
||||
|
||||
func savePixel(title, uuid string) error {
|
||||
statement, err := db.Prepare("INSERT INTO pixels (title, uuid) VALUES (?, ?)")
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
_, err = statement.Exec(title, uuid)
|
||||
return err
|
||||
}
|
|
@ -0,0 +1,8 @@
|
|||
module trakr
|
||||
|
||||
go 1.21.4
|
||||
|
||||
require (
|
||||
github.com/google/uuid v1.5.0
|
||||
github.com/mattn/go-sqlite3 v1.14.19
|
||||
)
|
|
@ -0,0 +1,4 @@
|
|||
github.com/google/uuid v1.5.0 h1:1p67kYwdtXjb0gL0BPiP1Av9wiZPo5A8z2cWkTZ+eyU=
|
||||
github.com/google/uuid v1.5.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
|
||||
github.com/mattn/go-sqlite3 v1.14.19 h1:fhGleo2h1p8tVChob4I9HpmVFIAkKGpiukdrgQbWfGI=
|
||||
github.com/mattn/go-sqlite3 v1.14.19/go.mod h1:2eHXhiwb8IkHr+BDWZGa96P6+rkvnG63S2DGjv9HUNg=
|
|
@ -0,0 +1,134 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"crypto/sha256"
|
||||
"fmt"
|
||||
"html/template"
|
||||
"log"
|
||||
"net/http"
|
||||
"strings"
|
||||
|
||||
"github.com/google/uuid"
|
||||
)
|
||||
|
||||
func newPixelPageHandler(w http.ResponseWriter, r *http.Request) {
|
||||
tmpl, err := template.ParseFiles("new.html")
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
err = tmpl.Execute(w, nil)
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
}
|
||||
|
||||
func createPixelHandler(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodPost {
|
||||
http.Error(w, "Only POST method is allowed", http.StatusMethodNotAllowed)
|
||||
return
|
||||
}
|
||||
|
||||
if err := r.ParseForm(); err != nil {
|
||||
http.Error(w, "Error parsing form", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
title := r.FormValue("title")
|
||||
|
||||
pixelUUID := uuid.New().String()
|
||||
|
||||
err := savePixel(title, pixelUUID)
|
||||
if err != nil {
|
||||
http.Error(w, "Error saving pixel to the database", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
http.Redirect(w, r, "/dashboard", http.StatusSeeOther)
|
||||
}
|
||||
|
||||
func pixelHandler(w http.ResponseWriter, r *http.Request) {
|
||||
pathParts := strings.Split(r.URL.Path, "/")
|
||||
if len(pathParts) < 3 {
|
||||
http.Error(w, "Invalid request", http.StatusBadRequest)
|
||||
return
|
||||
}
|
||||
uuid := pathParts[2]
|
||||
|
||||
// Get the pixel ID from the database
|
||||
pixelID, err := getPixelIDFromUUID(uuid)
|
||||
if err != nil {
|
||||
log.Println("Error getting pixel ID:", err)
|
||||
http.Error(w, "Internal Server Error", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
addr := strings.Split(r.RemoteAddr, ":")
|
||||
|
||||
rawFingerprint := fmt.Sprintf(
|
||||
"%s|%s|%s",
|
||||
addr[0],
|
||||
r.Header.Get("User-Agent"),
|
||||
r.Header.Get("Accept-Language"),
|
||||
)
|
||||
hash := sha256.Sum256([]byte(rawFingerprint))
|
||||
fingerprint := string(hash[:])
|
||||
|
||||
err = addStat(pixelID, addr[0], r.UserAgent(), fingerprint)
|
||||
if err != nil {
|
||||
log.Println("Error updating stats:", err)
|
||||
}
|
||||
|
||||
// Serve the pixel image
|
||||
w.Header().Set("Content-Type", "image/gif")
|
||||
w.Write(onePixelGIF)
|
||||
}
|
||||
|
||||
func dashboardHandler(w http.ResponseWriter, r *http.Request) {
|
||||
query := `
|
||||
SELECT p.title, p.uuid, COUNT(s.id) AS total_views, COUNT(DISTINCT s.ip) AS unique_views
|
||||
FROM pixels p
|
||||
LEFT JOIN stats s ON p.id = s.pixel_id
|
||||
GROUP BY p.id;`
|
||||
|
||||
rows, err := db.Query(query)
|
||||
if err != nil {
|
||||
http.Error(w, "Error fetching data", http.StatusInternalServerError)
|
||||
log.Println("Error fetching data:", err)
|
||||
return
|
||||
}
|
||||
defer rows.Close()
|
||||
|
||||
var stats []PixelStats
|
||||
for rows.Next() {
|
||||
var s PixelStats
|
||||
if err := rows.Scan(&s.Title, &s.UUID, &s.TotalViews, &s.UniqueViews); err != nil {
|
||||
http.Error(w, "Error reading data", http.StatusInternalServerError)
|
||||
log.Println("Error reading data:", err)
|
||||
return
|
||||
}
|
||||
stats = append(stats, s)
|
||||
}
|
||||
|
||||
// Handle any errors encountered during iteration
|
||||
if err = rows.Err(); err != nil {
|
||||
http.Error(w, "Error iterating data", http.StatusInternalServerError)
|
||||
log.Println("Error iterating data:", err)
|
||||
return
|
||||
}
|
||||
|
||||
// Render the template
|
||||
tmpl, err := template.ParseFiles("dashboard.html")
|
||||
if err != nil {
|
||||
http.Error(w, "Error loading template", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
err = tmpl.Execute(w, stats)
|
||||
if err != nil {
|
||||
http.Error(w, "Error executing template", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
}
|
|
@ -0,0 +1,39 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"log"
|
||||
"net/http"
|
||||
|
||||
_ "github.com/mattn/go-sqlite3"
|
||||
)
|
||||
|
||||
var onePixelGIF = []byte{
|
||||
0x47, 0x49, 0x46, 0x38, 0x39, 0x61, 0x01, 0x00,
|
||||
0x01, 0x00, 0x80, 0xff, 0x00, 0xff, 0xff, 0xff,
|
||||
0x00, 0x00, 0x00, 0x2c, 0x00, 0x00, 0x00, 0x00,
|
||||
0x01, 0x00, 0x01, 0x00, 0x00, 0x02, 0x02, 0x44,
|
||||
0x01, 0x00, 0x3b,
|
||||
}
|
||||
|
||||
func main() {
|
||||
db = initDB("data.sqlite")
|
||||
|
||||
http.HandleFunc("/new", newPixelPageHandler)
|
||||
http.HandleFunc("/create-pixel", createPixelHandler)
|
||||
http.HandleFunc("/p/", pixelHandler)
|
||||
http.HandleFunc("/dashboard", dashboardHandler)
|
||||
|
||||
addr := ":8080"
|
||||
|
||||
fmt.Printf("Server is up and running on http://localhost%s\n", addr)
|
||||
|
||||
log.Fatal(http.ListenAndServe(addr, nil))
|
||||
}
|
||||
|
||||
type PixelStats struct {
|
||||
Title string
|
||||
TotalViews int
|
||||
UniqueViews int
|
||||
UUID string
|
||||
}
|
|
@ -0,0 +1,63 @@
|
|||
<!DOCTYPE html>
|
||||
<html lang="en">
|
||||
|
||||
<head>
|
||||
<meta charset="UTF-8">
|
||||
<title>Create New Tracking Pixel</title>
|
||||
<style>
|
||||
body {
|
||||
font-family: Arial, sans-serif;
|
||||
margin: 0;
|
||||
padding: 0;
|
||||
display: flex;
|
||||
justify-content: center;
|
||||
align-items: center;
|
||||
height: 100vh;
|
||||
}
|
||||
|
||||
.container {
|
||||
width: 300px;
|
||||
padding: 20px;
|
||||
border: 1px solid #ddd;
|
||||
border-radius: 5px;
|
||||
box-shadow: 0 2px 4px rgba(0, 0, 0, 0.1);
|
||||
}
|
||||
|
||||
form {
|
||||
display: flex;
|
||||
flex-direction: column;
|
||||
}
|
||||
|
||||
input[type="text"] {
|
||||
padding: 10px;
|
||||
margin-bottom: 10px;
|
||||
border: 1px solid #ddd;
|
||||
border-radius: 3px;
|
||||
}
|
||||
|
||||
button {
|
||||
padding: 10px;
|
||||
background-color: #007bff;
|
||||
color: white;
|
||||
border: none;
|
||||
border-radius: 3px;
|
||||
cursor: pointer;
|
||||
}
|
||||
|
||||
button:hover {
|
||||
background-color: #0056b3;
|
||||
}
|
||||
</style>
|
||||
</head>
|
||||
|
||||
<body>
|
||||
<div class="container">
|
||||
<h2>Create New Tracking Pixel</h2>
|
||||
<form action="/create-pixel" method="post">
|
||||
<input type="text" name="title" placeholder="Enter pixel title" required>
|
||||
<button type="submit">Create Pixel</button>
|
||||
</form>
|
||||
</div>
|
||||
</body>
|
||||
|
||||
</html>
|
Loading…
Reference in New Issue