From 4389be40094ec4d310201088cdf3ade3fad88e41 Mon Sep 17 00:00:00 2001
From: 0ceanSlim <chriskerr@gmx.com>
Date: Wed, 24 Jul 2024 07:51:27 -0400
Subject: [PATCH] handling requests and subscriptions properly

---
 relay/handlers/req.go | 56 ++++++++++++++++++++++++-------------------
 1 file changed, 31 insertions(+), 25 deletions(-)

diff --git a/relay/handlers/req.go b/relay/handlers/req.go
index 328d88b..75edfa1 100644
--- a/relay/handlers/req.go
+++ b/relay/handlers/req.go
@@ -52,8 +52,7 @@ func HandleReq(ws *websocket.Conn, message []interface{}) {
 	fmt.Println("Subscription added:", subID)
 
 	// Query the database with filters and send back the results
-	// TO DO why is this taking a certain kind as an argument for collection???
-	queriedEvents, err := QueryEvents(filters, db.GetClient(), "grain", "event-kind1")
+	queriedEvents, err := QueryEvents(filters, db.GetClient(), "grain")
 	if err != nil {
 		fmt.Println("Error querying events:", err)
 		return
@@ -78,20 +77,19 @@ func HandleReq(ws *websocket.Conn, message []interface{}) {
 		return
 	}
 }
-// QueryEvents queries events from the MongoDB collection based on filters
-func QueryEvents(filters []relay.Filter, client *mongo.Client, databaseName, collectionName string) ([]relay.Event, error) {
-	collection := client.Database(databaseName).Collection(collectionName)
 
+// QueryEvents queries events from the MongoDB collection based on filters
+func QueryEvents(filters []relay.Filter, client *mongo.Client, databaseName string) ([]relay.Event, error) {
 	var results []relay.Event
 
 	for _, filter := range filters {
 		filterBson := bson.M{}
 
 		if len(filter.IDs) > 0 {
-			filterBson["_id"] = bson.M{"$in": filter.IDs}
+			filterBson["id"] = bson.M{"$in": filter.IDs}
 		}
 		if len(filter.Authors) > 0 {
-			filterBson["author"] = bson.M{"$in": filter.Authors}
+			filterBson["pubkey"] = bson.M{"$in": filter.Authors}
 		}
 		if len(filter.Kinds) > 0 {
 			filterBson["kind"] = bson.M{"$in": filter.Kinds}
@@ -99,37 +97,45 @@ func QueryEvents(filters []relay.Filter, client *mongo.Client, databaseName, col
 		if filter.Tags != nil {
 			for key, values := range filter.Tags {
 				if len(values) > 0 {
-					filterBson[key] = bson.M{"$in": values}
+					filterBson["tags."+key] = bson.M{"$in": values}
 				}
 			}
 		}
 		if filter.Since != nil {
-			filterBson["created_at"] = bson.M{"$gte": *filter.Since}
+			filterBson["createdat"] = bson.M{"$gte": *filter.Since}
 		}
 		if filter.Until != nil {
-			filterBson["created_at"] = bson.M{"$lte": *filter.Until}
+			if filterBson["createdat"] == nil {
+				filterBson["createdat"] = bson.M{"$lte": *filter.Until}
+			} else {
+				filterBson["createdat"].(bson.M)["$lte"] = *filter.Until
+			}
 		}
 
-		opts := options.Find()
+		opts := options.Find().SetSort(bson.D{{Key: "createdat", Value: -1}})
 		if filter.Limit != nil {
 			opts.SetLimit(int64(*filter.Limit))
 		}
 
-		cursor, err := collection.Find(context.TODO(), filterBson, opts)
-		if err != nil {
-			return nil, fmt.Errorf("error querying events: %v", err)
-		}
-		defer cursor.Close(context.TODO())
-
-		for cursor.Next(context.TODO()) {
-			var event relay.Event
-			if err := cursor.Decode(&event); err != nil {
-				return nil, fmt.Errorf("error decoding event: %v", err)
+		for _, kind := range filter.Kinds {
+			collectionName := fmt.Sprintf("event-kind%d", kind)
+			collection := client.Database(databaseName).Collection(collectionName)
+			cursor, err := collection.Find(context.TODO(), filterBson, opts)
+			if err != nil {
+				return nil, fmt.Errorf("error querying events: %v", err)
+			}
+			defer cursor.Close(context.TODO())
+
+			for cursor.Next(context.TODO()) {
+				var event relay.Event
+				if err := cursor.Decode(&event); err != nil {
+					return nil, fmt.Errorf("error decoding event: %v", err)
+				}
+				results = append(results, event)
+			}
+			if err := cursor.Err(); err != nil {
+				return nil, fmt.Errorf("cursor error: %v", err)
 			}
-			results = append(results, event)
-		}
-		if err := cursor.Err(); err != nil {
-			return nil, fmt.Errorf("cursor error: %v", err)
 		}
 	}