118 lines
3.1 KiB
Go
118 lines
3.1 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/fiatjaf/eventstore/postgresql"
|
|
"github.com/fiatjaf/khatru"
|
|
"github.com/fiatjaf/khatru/policies"
|
|
)
|
|
|
|
func getEnv(key, fallback string) string {
|
|
if value, ok := os.LookupEnv(key); ok {
|
|
return value
|
|
}
|
|
return fallback
|
|
}
|
|
|
|
func main() {
|
|
fmt.Print(`
|
|
SCRAPESTR RELAY
|
|
`)
|
|
|
|
// create the relay instance
|
|
relay := khatru.NewRelay()
|
|
|
|
// set up relay properties with environment variable configuration
|
|
relay.Info.Name = getEnv("RELAY_NAME", "Scrapestr Relay")
|
|
relay.Info.PubKey = getEnv("RELAY_PUBKEY", "79be667ef9dcbbac55a06295ce870b07029bfcdb2dce28d959f2815b16f81798")
|
|
relay.Info.Description = getEnv("RELAY_DESCRIPTION", "Scrapestr Relay")
|
|
relay.Info.Icon = getEnv("RELAY_ICON", "https://external-content.duckduckgo.com/iu/?u=https%3A%2F%2Fliquipedia.net%2Fcommons%2Fimages%2F3%2F35%2FSCProbe.jpg&f=1&nofb=1&ipt=0cbbfef25bce41da63d910e86c3c343e6c3b9d63194ca9755351bb7c2efa3359&ipo=images")
|
|
|
|
// Print relay information
|
|
fmt.Printf("Name: %s\n", relay.Info.Name)
|
|
fmt.Printf("Public Key: %s\n", relay.Info.PubKey)
|
|
fmt.Printf("Description: %s\n\n", relay.Info.Description)
|
|
|
|
// Configure PostgreSQL connection with environment variable
|
|
postgresURL := getEnv("POSTGRES_URL", "postgres://postgres:postgres@postgres/postgres?sslmode=disable")
|
|
db := postgresql.PostgresBackend{DatabaseURL: postgresURL}
|
|
if err := db.Init(); err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
relay.StoreEvent = append(relay.StoreEvent, db.SaveEvent)
|
|
relay.QueryEvents = append(relay.QueryEvents, db.QueryEvents)
|
|
relay.DeleteEvent = append(relay.DeleteEvent, db.DeleteEvent)
|
|
relay.ReplaceEvent = append(relay.ReplaceEvent, db.ReplaceEvent)
|
|
|
|
relay.RejectEvent = append(
|
|
relay.RejectEvent,
|
|
policies.PreventLargeTags(120),
|
|
policies.PreventTimestampsInThePast(time.Hour*2),
|
|
policies.PreventTimestampsInTheFuture(time.Minute*30),
|
|
)
|
|
|
|
mux := relay.Router()
|
|
// set up other http handlers
|
|
mux.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
|
|
w.Header().Set("content-type", "text/html")
|
|
|
|
// Query the total number of events
|
|
count := 0
|
|
row := db.DB.QueryRow("SELECT COUNT(*) FROM event")
|
|
if err := row.Scan(&count); err != nil {
|
|
fmt.Printf("Error counting events: %v\n", err)
|
|
}
|
|
|
|
// Improved HTML content
|
|
fmt.Fprintf(w, `
|
|
<!DOCTYPE html>
|
|
<html lang="en">
|
|
<head>
|
|
<meta charset="UTF-8">
|
|
<meta name="viewport" content="width=device-width, initial-scale=1.0">
|
|
<title>Scrapestr Relay</title>
|
|
<style>
|
|
body {
|
|
font-family: Arial, sans-serif;
|
|
background-color: #f4f4f4;
|
|
margin: 0;
|
|
padding: 0;
|
|
display: flex;
|
|
justify-content: center;
|
|
align-items: center;
|
|
height: 100vh;
|
|
}
|
|
.container {
|
|
background-color: #fff;
|
|
padding: 20px;
|
|
border-radius: 8px;
|
|
box-shadow: 0 0 10px rgba(0, 0, 0, 0.1);
|
|
text-align: center;
|
|
}
|
|
h1 {
|
|
color: #333;
|
|
}
|
|
p {
|
|
color: #666;
|
|
}
|
|
</style>
|
|
</head>
|
|
<body>
|
|
<div class="container">
|
|
<h1>Welcome to Scrapestr!</h1>
|
|
<p>Number of events stored: %d</p>
|
|
</div>
|
|
</body>
|
|
</html>
|
|
`, count)
|
|
})
|
|
|
|
fmt.Println("running on :3334")
|
|
http.ListenAndServe(":3334", relay)
|
|
}
|