mirror of
https://github.com/0ceanSlim/grain.git
synced 2024-11-23 17:07:13 +00:00
Compare commits
No commits in common. "d81d78e25fc368ca3d2be413f564390cb90e1c29" and "bda932c06d2b3643eafcda5072f5a2a37b58ecd8" have entirely different histories.
d81d78e25f
...
bda932c06d
76
app/src/app.go
Normal file
76
app/src/app.go
Normal file
@ -0,0 +1,76 @@
|
||||
package app
|
||||
|
||||
import (
|
||||
"grain/server/db"
|
||||
relay "grain/server/types"
|
||||
"html/template"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
type PageData struct {
|
||||
Title string
|
||||
Theme string
|
||||
Events []relay.Event
|
||||
}
|
||||
|
||||
func RootHandler(w http.ResponseWriter, r *http.Request) {
|
||||
// Fetch the top ten most recent events
|
||||
client := db.GetClient()
|
||||
events, err := FetchTopTenRecentEvents(client)
|
||||
if err != nil {
|
||||
http.Error(w, "Unable to fetch events", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
data := PageData{
|
||||
Title: "GRAIN Dashboard",
|
||||
Events: events,
|
||||
}
|
||||
|
||||
RenderTemplate(w, data, "index.html")
|
||||
}
|
||||
|
||||
// Define the base directories for views and templates
|
||||
const (
|
||||
viewsDir = "app/views/"
|
||||
templatesDir = "app/views/templates/"
|
||||
)
|
||||
|
||||
// Define the common layout templates filenames
|
||||
var templateFiles = []string{
|
||||
"layout.html",
|
||||
"header.html",
|
||||
"footer.html",
|
||||
}
|
||||
|
||||
// Initialize the common templates with full paths
|
||||
var layout = PrependDir(templatesDir, templateFiles)
|
||||
|
||||
func RenderTemplate(w http.ResponseWriter, data PageData, view string) {
|
||||
// Append the specific template for the route
|
||||
templates := append(layout, viewsDir+view)
|
||||
|
||||
// Parse all templates
|
||||
tmpl, err := template.ParseFiles(templates...)
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
// Execute the "layout" template
|
||||
err = tmpl.ExecuteTemplate(w, "layout", data)
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||||
}
|
||||
}
|
||||
|
||||
// Helper function to prepend a directory path to a list of filenames
|
||||
func PrependDir(dir string, files []string) []string {
|
||||
var fullPaths []string
|
||||
for _, file := range files {
|
||||
fullPaths = append(fullPaths, dir+file)
|
||||
}
|
||||
return fullPaths
|
||||
}
|
||||
|
||||
|
@ -1,4 +1,4 @@
|
||||
package utils
|
||||
package app
|
||||
|
||||
import (
|
||||
"context"
|
@ -1,9 +1,7 @@
|
||||
package routes
|
||||
|
||||
import (
|
||||
app "grain/app/src/types"
|
||||
"grain/app/src/utils"
|
||||
|
||||
app "grain/app/src"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
@ -13,5 +11,5 @@ func ImportEvents(w http.ResponseWriter, r *http.Request) {
|
||||
}
|
||||
|
||||
// Call RenderTemplate with the specific template for this route
|
||||
utils.RenderTemplate(w, data, "importEvents.html")
|
||||
app.RenderTemplate(w, data, "importEvents.html")
|
||||
}
|
||||
|
@ -1,17 +0,0 @@
|
||||
package routes
|
||||
|
||||
import (
|
||||
app "grain/app/src/types"
|
||||
"grain/app/src/utils"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
func IndexHandler(w http.ResponseWriter, r *http.Request) {
|
||||
|
||||
data := app.PageData{
|
||||
Title: "GRAIN Dashboard",
|
||||
|
||||
}
|
||||
|
||||
utils.RenderTemplate(w, data, "index.html")
|
||||
}
|
@ -1,5 +0,0 @@
|
||||
package types
|
||||
|
||||
type PageData struct {
|
||||
Title string
|
||||
}
|
@ -1,10 +0,0 @@
|
||||
package utils
|
||||
|
||||
// Helper function to prepend a directory path to a list of filenames
|
||||
func PrependDir(dir string, files []string) []string {
|
||||
var fullPaths []string
|
||||
for _, file := range files {
|
||||
fullPaths = append(fullPaths, dir+file)
|
||||
}
|
||||
return fullPaths
|
||||
}
|
@ -1,44 +0,0 @@
|
||||
package utils
|
||||
|
||||
import (
|
||||
app "grain/app/src/types"
|
||||
"html/template"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
// Define the base directories for views and templates
|
||||
const (
|
||||
viewsDir = "app/views/"
|
||||
templatesDir = "app/views/templates/"
|
||||
)
|
||||
|
||||
// Define the common layout templates filenames
|
||||
var templateFiles = []string{
|
||||
"layout.html",
|
||||
"header.html",
|
||||
"footer.html",
|
||||
}
|
||||
|
||||
// Initialize the common templates with full paths
|
||||
var layout = PrependDir(templatesDir, templateFiles)
|
||||
|
||||
func RenderTemplate(w http.ResponseWriter, data app.PageData, view string) {
|
||||
// Append the specific template for the route
|
||||
templates := append(layout, viewsDir+view)
|
||||
|
||||
// Parse all templates
|
||||
tmpl, err := template.ParseFiles(templates...)
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
// Execute the "layout" template
|
||||
err = tmpl.ExecuteTemplate(w, "layout", data)
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), http.StatusInternalServerError)
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
|
@ -7,9 +7,6 @@ server:
|
||||
read_timeout: 10 # Read timeout in seconds
|
||||
write_timeout: 10 # Write timeout in seconds
|
||||
idle_timeout: 120 # Idle timeout in seconds
|
||||
max_connections: 100 # Maximum number of concurrent connections
|
||||
max_subscriptions_per_client: 10 # Maximum number of concurrent subscriptions per client
|
||||
|
||||
pubkey_whitelist:
|
||||
enabled: false
|
||||
pubkeys: #["3fe0ab6cbdb7ee27148202249e3fb3b89423c6f6cda6ef43ea5057c3d93088e4",
|
||||
|
@ -6,12 +6,10 @@ type ServerConfig struct {
|
||||
Database string `yaml:"database"`
|
||||
} `yaml:"mongodb"`
|
||||
Server struct {
|
||||
Port string `yaml:"port"`
|
||||
ReadTimeout int `yaml:"read_timeout"` // Timeout in seconds
|
||||
WriteTimeout int `yaml:"write_timeout"` // Timeout in seconds
|
||||
IdleTimeout int `yaml:"idle_timeout"` // Timeout in seconds
|
||||
MaxConnections int `yaml:"max_connections"` // Maximum number of concurrent connections
|
||||
MaxSubscriptionsPerClient int `yaml:"max_subscriptions_per_client"` // Maximum number of subscriptions per client
|
||||
Port string `yaml:"port"`
|
||||
ReadTimeout int `yaml:"read_timeout"` // Timeout in seconds
|
||||
WriteTimeout int `yaml:"write_timeout"` // Timeout in seconds
|
||||
IdleTimeout int `yaml:"idle_timeout"` // Timeout in seconds
|
||||
} `yaml:"server"`
|
||||
RateLimit RateLimitConfig `yaml:"rate_limit"`
|
||||
PubkeyWhitelist PubkeyWhitelistConfig `yaml:"pubkey_whitelist"`
|
||||
|
8
main.go
8
main.go
@ -2,12 +2,14 @@ package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
app "grain/app/src"
|
||||
"grain/app/src/api"
|
||||
"grain/app/src/routes"
|
||||
"grain/config"
|
||||
configTypes "grain/config/types"
|
||||
relay "grain/server"
|
||||
"grain/server/db"
|
||||
"grain/server/nip"
|
||||
"grain/server/utils"
|
||||
"log"
|
||||
"net/http"
|
||||
@ -34,7 +36,7 @@ func main() {
|
||||
config.SetupRateLimiter(cfg)
|
||||
config.SetupSizeLimiter(cfg)
|
||||
|
||||
err = utils.LoadRelayMetadataJSON()
|
||||
err = nip.LoadRelayMetadataJSON()
|
||||
if err != nil {
|
||||
log.Fatal("Failed to load relay metadata: ", err)
|
||||
}
|
||||
@ -83,8 +85,8 @@ func ListenAndServe(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Header.Get("Upgrade") == "websocket" {
|
||||
wsServer.ServeHTTP(w, r)
|
||||
} else if r.Header.Get("Accept") == "application/nostr+json" {
|
||||
utils.RelayInfoHandler(w, r)
|
||||
nip.RelayInfoHandler(w, r)
|
||||
} else {
|
||||
routes.IndexHandler(w, r)
|
||||
app.RootHandler(w, r)
|
||||
}
|
||||
}
|
||||
|
@ -4,12 +4,10 @@ import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"grain/config"
|
||||
"grain/server/db"
|
||||
"grain/server/handlers/response"
|
||||
relay "grain/server/types"
|
||||
"grain/server/utils"
|
||||
"sync"
|
||||
|
||||
"go.mongodb.org/mongo-driver/bson"
|
||||
"go.mongodb.org/mongo-driver/mongo"
|
||||
@ -18,7 +16,6 @@ import (
|
||||
)
|
||||
|
||||
var subscriptions = make(map[string]relay.Subscription)
|
||||
var mu sync.Mutex // Protect concurrent access to subscriptions map
|
||||
|
||||
func HandleReq(ws *websocket.Conn, message []interface{}, subscriptions map[string][]relay.Filter) {
|
||||
if len(message) < 3 {
|
||||
@ -34,21 +31,6 @@ func HandleReq(ws *websocket.Conn, message []interface{}, subscriptions map[stri
|
||||
return
|
||||
}
|
||||
|
||||
mu.Lock()
|
||||
defer mu.Unlock()
|
||||
|
||||
// Check the current number of subscriptions for the client
|
||||
if len(subscriptions) >= config.GetConfig().Server.MaxSubscriptionsPerClient {
|
||||
// Find and remove the oldest subscription (FIFO)
|
||||
var oldestSubID string
|
||||
for id := range subscriptions {
|
||||
oldestSubID = id
|
||||
break
|
||||
}
|
||||
delete(subscriptions, oldestSubID)
|
||||
fmt.Println("Dropped oldest subscription:", oldestSubID)
|
||||
}
|
||||
|
||||
filters := make([]relay.Filter, len(message)-2)
|
||||
for i, filter := range message[2:] {
|
||||
filterData, ok := filter.(map[string]interface{})
|
||||
@ -70,7 +52,7 @@ func HandleReq(ws *websocket.Conn, message []interface{}, subscriptions map[stri
|
||||
filters[i] = f
|
||||
}
|
||||
|
||||
// Add the new subscription or update the existing one
|
||||
// Update or add the subscription for the given subID
|
||||
subscriptions[subID] = filters
|
||||
fmt.Println("Subscription updated:", subID)
|
||||
|
||||
|
@ -1,4 +1,4 @@
|
||||
package utils
|
||||
package nip
|
||||
|
||||
import (
|
||||
"encoding/json"
|
@ -3,56 +3,22 @@ package relay
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"grain/config"
|
||||
"grain/server/handlers"
|
||||
|
||||
"grain/config"
|
||||
|
||||
relay "grain/server/types"
|
||||
"grain/server/utils"
|
||||
"log"
|
||||
"sync"
|
||||
|
||||
"golang.org/x/net/websocket"
|
||||
)
|
||||
|
||||
// Global connection count
|
||||
var (
|
||||
currentConnections int
|
||||
mu sync.Mutex
|
||||
)
|
||||
|
||||
// Client subscription count
|
||||
var clientSubscriptions = make(map[*websocket.Conn]int)
|
||||
|
||||
func WebSocketHandler(ws *websocket.Conn) {
|
||||
defer func() {
|
||||
mu.Lock()
|
||||
currentConnections--
|
||||
delete(clientSubscriptions, ws)
|
||||
mu.Unlock()
|
||||
ws.Close()
|
||||
}()
|
||||
|
||||
mu.Lock()
|
||||
if currentConnections >= config.GetConfig().Server.MaxConnections {
|
||||
websocket.Message.Send(ws, `{"error": "too many connections"}`)
|
||||
mu.Unlock()
|
||||
return
|
||||
}
|
||||
currentConnections++
|
||||
mu.Unlock()
|
||||
|
||||
clientInfo := utils.ClientInfo{
|
||||
IP: utils.GetClientIP(ws.Request()),
|
||||
UserAgent: ws.Request().Header.Get("User-Agent"),
|
||||
Origin: ws.Request().Header.Get("Origin"),
|
||||
}
|
||||
|
||||
log.Printf("New connection from IP: %s, User-Agent: %s, Origin: %s", clientInfo.IP, clientInfo.UserAgent, clientInfo.Origin)
|
||||
defer ws.Close()
|
||||
|
||||
var msg string
|
||||
rateLimiter := config.GetRateLimiter()
|
||||
|
||||
subscriptions := make(map[string][]relay.Filter) // Subscription map scoped to the connection
|
||||
clientSubscriptions[ws] = 0
|
||||
|
||||
for {
|
||||
err := websocket.Message.Receive(ws, &msg)
|
||||
@ -91,14 +57,6 @@ func WebSocketHandler(ws *websocket.Conn) {
|
||||
case "EVENT":
|
||||
handlers.HandleEvent(ws, message)
|
||||
case "REQ":
|
||||
mu.Lock()
|
||||
if clientSubscriptions[ws] >= config.GetConfig().Server.MaxSubscriptionsPerClient {
|
||||
websocket.Message.Send(ws, `{"error": "too many subscriptions"}`)
|
||||
mu.Unlock()
|
||||
continue
|
||||
}
|
||||
clientSubscriptions[ws]++
|
||||
mu.Unlock()
|
||||
if allowed, msg := rateLimiter.AllowReq(); !allowed {
|
||||
websocket.Message.Send(ws, fmt.Sprintf(`{"error": "%s"}`, msg))
|
||||
ws.Close()
|
||||
|
@ -1,28 +0,0 @@
|
||||
package utils
|
||||
|
||||
import (
|
||||
"net/http"
|
||||
"strings"
|
||||
)
|
||||
|
||||
type ClientInfo struct {
|
||||
IP string
|
||||
UserAgent string
|
||||
Origin string
|
||||
}
|
||||
|
||||
func GetClientIP(r *http.Request) string {
|
||||
xff := r.Header.Get("X-Forwarded-For")
|
||||
if xff != "" {
|
||||
ips := strings.Split(xff, ",")
|
||||
if len(ips) > 0 {
|
||||
return strings.TrimSpace(ips[0])
|
||||
}
|
||||
}
|
||||
|
||||
remoteAddr := r.RemoteAddr
|
||||
if idx := strings.LastIndex(remoteAddr, ":"); idx != -1 {
|
||||
return remoteAddr[:idx]
|
||||
}
|
||||
return remoteAddr
|
||||
}
|
Loading…
Reference in New Issue
Block a user