turn relayer into a server framework and put actual relay code into ./basic

README.md

@@ -1,3 +1 @@
-This is a simple relay implementation for [nostr](https://github.com/fiatjaf/nostr).
-
-There is a public instance at https://nostr-relay.herokuapp.com/.
+Nostr Relay Framework -- use it to implement your own custom relay.

basic/cleanup.go

@@ -1,9 +1,13 @@
 package main
 
-import "time"
+import (
+	"time"
+
+	"github.com/jmoiron/sqlx"
+)
 
 // every hour, delete all very old events
-func cleanupRoutine() {
+func cleanupRoutine(db *sqlx.DB) {
 	for {
 		time.Sleep(60 * time.Minute)
 		db.Exec(`DELETE FROM event WHERE created_at < $1`, time.Now().AddDate(0, -3, 0))

basic/main.go

@@ -0,0 +1,44 @@
+package main
+
+import (
+	"fmt"
+
+	"github.com/fiatjaf/relayer"
+	"github.com/jmoiron/sqlx"
+	"github.com/jmoiron/sqlx/reflectx"
+	"github.com/kelseyhightower/envconfig"
+)
+
+type BasicRelay struct {
+	PostgresDatabase string `envconfig:"POSTGRESQL_DATABASE"`
+
+	DB *sqlx.DB
+}
+
+func (b *BasicRelay) Name() string {
+	return "BasicRelay"
+}
+
+func (b *BasicRelay) Init() error {
+	err := envconfig.Process("", b)
+	if err != nil {
+		return fmt.Errorf("couldn't process envconfig: %w", err)
+	}
+
+	if db, err := initDB(b.PostgresDatabase); err != nil {
+		return fmt.Errorf("failed to open database: %w", err)
+	} else {
+		db.Mapper = reflectx.NewMapperFunc("json", sqlx.NameMapper)
+		b.DB = db
+	}
+
+	go cleanupRoutine(b.DB)
+
+	return nil
+}
+
+func main() {
+	var b BasicRelay
+
+	relayer.Start(&b)
+}

basic/postgresql.go

@@ -3,10 +3,11 @@ package main
 import (
 	"github.com/jmoiron/sqlx"
 	_ "github.com/lib/pq"
+	"github.com/rs/zerolog/log"
 )
 
-func initDB() (*sqlx.DB, error) {
-	db, err := sqlx.Connect("postgres", s.PostgresDatabase)
+func initDB(dburl string) (*sqlx.DB, error) {
+	db, err := sqlx.Connect("postgres", dburl)
 	if err != nil {
 		return nil, err
 	}

basic/query.go

@@ -9,9 +9,12 @@ import (
 
 	"github.com/fiatjaf/go-nostr/event"
 	"github.com/fiatjaf/go-nostr/filter"
+	"github.com/rs/zerolog/log"
 )
 
-func queryEvents(filter *filter.EventFilter) (events []event.Event, err error) {
+func (b *BasicRelay) QueryEvents(
+	filter *filter.EventFilter,
+) (events []event.Event, err error) {
 	var conditions []string
 	var params []interface{}
 
@@ -69,11 +72,11 @@ func queryEvents(filter *filter.EventFilter) (events []event.Event, err error) {
 		conditions = append(conditions, "true")
 	}
 
-	query := db.Rebind("SELECT * FROM event WHERE " +
+	query := b.DB.Rebind("SELECT * FROM event WHERE " +
 		strings.Join(conditions, " AND ") +
 		" ORDER BY created_at LIMIT 100")
 
-	err = db.Select(&events, query, params...)
+	err = b.DB.Select(&events, query, params...)
 	if err != nil && err != sql.ErrNoRows {
 		log.Warn().Err(err).Interface("filter", filter).Msg("failed to fetch events")
 		err = fmt.Errorf("failed to fetch events: %w", err)

basic/save.go

@@ -0,0 +1,55 @@
+package main
+
+import (
+	"encoding/json"
+	"errors"
+	"fmt"
+	"strings"
+
+	"github.com/fiatjaf/go-nostr/event"
+)
+
+func (b *BasicRelay) SaveEvent(evt *event.Event) error {
+	// disallow large contents
+	if len(evt.Content) > 1000 {
+		return errors.New("event content too large")
+	}
+
+	// react to different kinds of events
+	switch evt.Kind {
+	case event.KindSetMetadata:
+		// delete past set_metadata events from this user
+		b.DB.Exec(`DELETE FROM event WHERE pubkey = $1 AND kind = 0`, evt.PubKey)
+	case event.KindRecommendServer:
+		// delete past recommend_server events equal to this one
+		b.DB.Exec(`DELETE FROM event WHERE pubkey = $1 AND kind = 2 AND content = $2`,
+			evt.PubKey, evt.Content)
+	case event.KindContactList:
+		// delete past contact lists from this same pubkey
+		b.DB.Exec(`DELETE FROM event WHERE pubkey = $1 AND kind = 3`, evt.PubKey)
+	default:
+		// delete all but the 10 most recent ones
+		b.DB.Exec(`DELETE FROM event WHERE pubkey = $1 AND kind = $2 AND created_at < (
+          SELECT created_at FROM event WHERE pubkey = $1
+          ORDER BY created_at DESC OFFSET 10 LIMIT 1
+        )`,
+			evt.PubKey, evt.Kind)
+	}
+
+	// insert
+	tagsj, _ := json.Marshal(evt.Tags)
+	_, err := b.DB.Exec(`
+        INSERT INTO event (id, pubkey, created_at, kind, tags, content, sig)
+        VALUES ($1, $2, $3, $4, $5, $6, $7)
+    `, evt.ID, evt.PubKey, evt.CreatedAt, evt.Kind, tagsj, evt.Content, evt.Sig)
+	if err != nil {
+		if strings.Index(err.Error(), "UNIQUE") != -1 {
+			// already exists
+			return nil
+		}
+
+		return fmt.Errorf("failed to save event from %s", evt.PubKey)
+	}
+
+	return nil
+}

handlers.go

@@ -1,12 +1,12 @@
-package main
+package relayer
 
 import (
 	"crypto/sha256"
 	"encoding/hex"
 	"encoding/json"
 	"errors"
+	"fmt"
 	"net/http"
-	"strings"
 	"time"
 
 	"github.com/fiatjaf/go-nostr/event"
@@ -34,7 +34,8 @@ var upgrader = websocket.Upgrader{
 	CheckOrigin:     func(r *http.Request) bool { return true },
 }
 
-func handleWebsocket(w http.ResponseWriter, r *http.Request) {
+func handleWebsocket(relay Relay) func(http.ResponseWriter, *http.Request) {
+	return func(w http.ResponseWriter, r *http.Request) {
 		conn, err := upgrader.Upgrade(w, r, nil)
 		if err != nil {
 			log.Warn().Err(err).Msg("failed to upgrade websocket")
@@ -81,7 +82,7 @@ func handleWebsocket(w http.ResponseWriter, r *http.Request) {
 					var request []json.RawMessage
 					err = json.Unmarshal(message, &request)
 					if err == nil && len(request) < 2 {
-					err = errors.New("request has less than parameters")
+						err = errors.New("request has less than 2 parameters")
 						return
 					}
 					if err != nil {
@@ -95,8 +96,35 @@ func handleWebsocket(w http.ResponseWriter, r *http.Request) {
 					switch typ {
 					case "EVENT":
 						// it's a new event
-					err = saveEvent(request[1])
+						var evt event.Event
+						err := json.Unmarshal(request[1], &evt)
+						if err != nil {
+							err = fmt.Errorf("failed to decode event: %w", err)
+							return
+						}
 
+						// check serialization
+						serialized := evt.Serialize()
+
+						// assign ID
+						hash := sha256.Sum256(serialized)
+						evt.ID = hex.EncodeToString(hash[:])
+
+						// check signature (requires the ID to be set)
+						if ok, err := evt.CheckSignature(); err != nil {
+							err = errors.New("signature verification error")
+							return
+						} else if !ok {
+							err = errors.New("signature invalid")
+							return
+						}
+
+						err = relay.SaveEvent(&evt)
+						if err != nil {
+							return
+						}
+
+						notifyListeners(&evt)
 					case "REQ":
 						var id string
 						json.Unmarshal(request[1], &id)
@@ -112,7 +140,7 @@ func handleWebsocket(w http.ResponseWriter, r *http.Request) {
 								return
 							}
 
-						events, err := queryEvents(&filters[i])
+							events, err := relay.QueryEvents(&filters[i])
 							if err == nil {
 								for _, event := range events {
 									conn.WriteJSON([]interface{}{"EVENT", id, event})
@@ -155,75 +183,5 @@ func handleWebsocket(w http.ResponseWriter, r *http.Request) {
 				}
 			}
 		}()
-}
-
-func saveEvent(body []byte) error {
-	var evt event.Event
-	err := json.Unmarshal(body, &evt)
-	if err != nil {
-		log.Warn().Err(err).Str("body", string(body)).Msg("couldn't decode body")
-		return errors.New("failed to decode event")
-	}
-
-	// disallow large contents
-	if len(evt.Content) > 1000 {
-		log.Warn().Err(err).Msg("event content too large")
-		return errors.New("event content too large")
-	}
-
-	// check serialization
-	serialized := evt.Serialize()
-
-	// assign ID
-	hash := sha256.Sum256(serialized)
-	evt.ID = hex.EncodeToString(hash[:])
-
-	// check signature (requires the ID to be set)
-	if ok, err := evt.CheckSignature(); err != nil {
-		log.Warn().Err(err).Msg("signature verification error")
-		return errors.New("signature verification error")
-	} else if !ok {
-		log.Warn().Err(err).Msg("signature invalid")
-		return errors.New("signature invalid")
-	}
-
-	// react to different kinds of events
-	switch evt.Kind {
-	case event.KindSetMetadata:
-		// delete past set_metadata events from this user
-		db.Exec(`DELETE FROM event WHERE pubkey = $1 AND kind = 0`, evt.PubKey)
-	case event.KindRecommendServer:
-		// delete past recommend_server events equal to this one
-		db.Exec(`DELETE FROM event WHERE pubkey = $1 AND kind = 2 AND content = $2`,
-			evt.PubKey, evt.Content)
-	case event.KindContactList:
-		// delete past contact lists from this same pubkey
-		db.Exec(`DELETE FROM event WHERE pubkey = $1 AND kind = 3`, evt.PubKey)
-	default:
-		// delete all but the 10 most recent ones
-		db.Exec(`DELETE FROM event WHERE pubkey = $1 AND kind = $2 AND created_at < (
-          SELECT created_at FROM event WHERE pubkey = $1
-          ORDER BY created_at DESC OFFSET 10 LIMIT 1
-        )`,
-			evt.PubKey, evt.Kind)
-	}
-
-	// insert
-	tagsj, _ := json.Marshal(evt.Tags)
-	_, err = db.Exec(`
-        INSERT INTO event (id, pubkey, created_at, kind, tags, content, sig)
-        VALUES ($1, $2, $3, $4, $5, $6, $7)
-    `, evt.ID, evt.PubKey, evt.CreatedAt, evt.Kind, tagsj, evt.Content, evt.Sig)
-	if err != nil {
-		if strings.Index(err.Error(), "UNIQUE") != -1 {
-			// already exists
-			return nil
-		}
-
-		log.Warn().Err(err).Str("pubkey", evt.PubKey).Msg("failed to save")
-		return errors.New("failed to save event")
-	}
-
-	notifyListeners(&evt)
-	return nil
+	}
 }

interface.go

@@ -0,0 +1,15 @@
+package relayer
+
+import (
+	"github.com/fiatjaf/go-nostr/event"
+	"github.com/fiatjaf/go-nostr/filter"
+)
+
+var Log = log
+
+type Relay interface {
+	Name() string
+	Init() error
+	SaveEvent(*event.Event) error
+	QueryEvents(*filter.EventFilter) ([]event.Event, error)
+}

listener.go

@@ -1,4 +1,4 @@
-package main
+package relayer
 
 import (
 	"sync"

notice.go

@@ -1,4 +1,4 @@
-package main
+package relayer
 
 type Notice struct {
 	Kind    string `json:"kind"`

relayer.go

@@ -1,4 +1,4 @@
-package main
+package relayer
 
 import (
 	"net/http"
@@ -6,9 +6,6 @@ import (
 	"time"
 
 	"github.com/gorilla/mux"
-	"github.com/jmoiron/sqlx"
-	"github.com/jmoiron/sqlx/reflectx"
-	"github.com/kelseyhightower/envconfig"
 	"github.com/rs/cors"
 	"github.com/rs/zerolog"
 )
@@ -16,32 +13,22 @@ import (
 type Settings struct {
 	Host string `envconfig:"HOST" default:"0.0.0.0"`
 	Port string `envconfig:"PORT" default:"7447"`
-
-	PostgresDatabase string `envconfig:"POSTGRESQL_DATABASE"`
 }
 
 var s Settings
-var err error
-var db *sqlx.DB
 var log = zerolog.New(os.Stderr).Output(zerolog.ConsoleWriter{Out: os.Stderr})
+
 var router = mux.NewRouter()
 
-func main() {
-	err = envconfig.Process("", &s)
-	if err != nil {
-		log.Fatal().Err(err).Msg("couldn't process envconfig")
-	}
+func Start(relay Relay) {
+	Log = log.With().Str("name", relay.Name()).Logger()
 
-	db, err = initDB()
-	if err != nil {
-		log.Fatal().Err(err).Msg("failed to open database")
+	if err := relay.Init(); err != nil {
+		Log.Fatal().Err(err).Msg("failed to start")
 	}
-	db.Mapper = reflectx.NewMapperFunc("json", sqlx.NameMapper)
-
-	go cleanupRoutine()
 
 	// NIP01
-	router.Path("/").Methods("GET").HandlerFunc(handleWebsocket)
+	router.Path("/").Methods("GET").HandlerFunc(handleWebsocket(relay))
 
 	srv := &http.Server{
 		Handler:           cors.Default().Handler(router),