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() {
	// Print ASCII art banner
	fmt.Print(`
╔╗ ╦═╗╔═╗╦  ╦╔═╗  ╦═╗╔═╗╦  ╔═╗╦ ╦
╠╩╗╠╦╝║╣ ╚╗╔╝║╣   ╠╦╝║╣ ║  ╠═╣╚╦╝
╚═╝╩╚═╚═╝ ╚╝ ╚═╝  ╩╚═╚═╝╩═╝╩ ╩ ╩ 
`)

	// create the relay instance
	relay := khatru.NewRelay()

	// set up relay properties with environment variable configuration
	relay.Info.Name = getEnv("RELAY_NAME", "Breve Relay")
	relay.Info.PubKey = getEnv("RELAY_PUBKEY", "79be667ef9dcbbac55a06295ce870b07029bfcdb2dce28d959f2815b16f81798")
	relay.Info.Description = getEnv("RELAY_DESCRIPTION", "Breve 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),
	)

	fmt.Println("running on :3334")
	http.ListenAndServe(":3334", relay)
}