diff options
| -rw-r--r-- | .gitignore | 3 | ||||
| -rw-r--r-- | Makefile | 7 | ||||
| -rw-r--r-- | cmd/relay/main.go | 56 | ||||
| -rw-r--r-- | cmd/testclient/main.go | 85 |
4 files changed, 151 insertions, 0 deletions
| @@ -33,6 +33,9 @@ Thumbs.db | |||
| 33 | *.db-shm | 33 | *.db-shm |
| 34 | *.db-wal | 34 | *.db-wal |
| 35 | 35 | ||
| 36 | # Relay data | ||
| 37 | relay.db* | ||
| 38 | |||
| 36 | # Config (may contain secrets) | 39 | # Config (may contain secrets) |
| 37 | config.yaml | 40 | config.yaml |
| 38 | config.yml | 41 | config.yml |
| @@ -20,6 +20,13 @@ test: | |||
| 20 | build: | 20 | build: |
| 21 | go build -o bin/relay ./cmd/relay | 21 | go build -o bin/relay ./cmd/relay |
| 22 | 22 | ||
| 23 | # Build test client | ||
| 24 | build-client: | ||
| 25 | go build -o bin/testclient ./cmd/testclient | ||
| 26 | |||
| 27 | # Build everything | ||
| 28 | build-all: build build-client | ||
| 29 | |||
| 23 | # Clean generated files | 30 | # Clean generated files |
| 24 | clean: | 31 | clean: |
| 25 | rm -rf api/ | 32 | rm -rf api/ |
diff --git a/cmd/relay/main.go b/cmd/relay/main.go new file mode 100644 index 0000000..3db466d --- /dev/null +++ b/cmd/relay/main.go | |||
| @@ -0,0 +1,56 @@ | |||
| 1 | package main | ||
| 2 | |||
| 3 | import ( | ||
| 4 | "flag" | ||
| 5 | "log" | ||
| 6 | "net" | ||
| 7 | "os" | ||
| 8 | "os/signal" | ||
| 9 | "syscall" | ||
| 10 | |||
| 11 | "google.golang.org/grpc" | ||
| 12 | |||
| 13 | pb "northwest.io/nostr-grpc/api/nostr/v1" | ||
| 14 | grpchandler "northwest.io/nostr-grpc/internal/handler/grpc" | ||
| 15 | "northwest.io/nostr-grpc/internal/storage" | ||
| 16 | ) | ||
| 17 | |||
| 18 | func main() { | ||
| 19 | var ( | ||
| 20 | grpcAddr = flag.String("grpc-addr", ":50051", "gRPC server address") | ||
| 21 | dbPath = flag.String("db", "relay.db", "SQLite database path") | ||
| 22 | ) | ||
| 23 | flag.Parse() | ||
| 24 | |||
| 25 | store, err := storage.New(*dbPath) | ||
| 26 | if err != nil { | ||
| 27 | log.Fatalf("failed to create storage: %v", err) | ||
| 28 | } | ||
| 29 | defer store.Close() | ||
| 30 | |||
| 31 | handler := grpchandler.NewServer(store) | ||
| 32 | |||
| 33 | lis, err := net.Listen("tcp", *grpcAddr) | ||
| 34 | if err != nil { | ||
| 35 | log.Fatalf("failed to listen: %v", err) | ||
| 36 | } | ||
| 37 | |||
| 38 | grpcServer := grpc.NewServer() | ||
| 39 | pb.RegisterNostrRelayServer(grpcServer, handler) | ||
| 40 | |||
| 41 | log.Printf("gRPC server listening on %s", *grpcAddr) | ||
| 42 | log.Printf("Database: %s", *dbPath) | ||
| 43 | |||
| 44 | sigChan := make(chan os.Signal, 1) | ||
| 45 | signal.Notify(sigChan, os.Interrupt, syscall.SIGTERM) | ||
| 46 | |||
| 47 | go func() { | ||
| 48 | <-sigChan | ||
| 49 | log.Println("Shutting down...") | ||
| 50 | grpcServer.GracefulStop() | ||
| 51 | }() | ||
| 52 | |||
| 53 | if err := grpcServer.Serve(lis); err != nil { | ||
| 54 | log.Fatalf("failed to serve: %v", err) | ||
| 55 | } | ||
| 56 | } | ||
diff --git a/cmd/testclient/main.go b/cmd/testclient/main.go new file mode 100644 index 0000000..571751e --- /dev/null +++ b/cmd/testclient/main.go | |||
| @@ -0,0 +1,85 @@ | |||
| 1 | package main | ||
| 2 | |||
| 3 | import ( | ||
| 4 | "context" | ||
| 5 | "flag" | ||
| 6 | "log" | ||
| 7 | "time" | ||
| 8 | |||
| 9 | "google.golang.org/grpc" | ||
| 10 | "google.golang.org/grpc/credentials/insecure" | ||
| 11 | |||
| 12 | pb "northwest.io/nostr-grpc/api/nostr/v1" | ||
| 13 | "northwest.io/nostr-grpc/internal/nostr" | ||
| 14 | ) | ||
| 15 | |||
| 16 | func main() { | ||
| 17 | addr := flag.String("addr", "localhost:50051", "relay address") | ||
| 18 | flag.Parse() | ||
| 19 | |||
| 20 | conn, err := grpc.NewClient(*addr, grpc.WithTransportCredentials(insecure.NewCredentials())) | ||
| 21 | if err != nil { | ||
| 22 | log.Fatalf("failed to connect: %v", err) | ||
| 23 | } | ||
| 24 | defer conn.Close() | ||
| 25 | |||
| 26 | client := pb.NewNostrRelayClient(conn) | ||
| 27 | ctx := context.Background() | ||
| 28 | |||
| 29 | key, err := nostr.GenerateKey() | ||
| 30 | if err != nil { | ||
| 31 | log.Fatalf("failed to generate key: %v", err) | ||
| 32 | } | ||
| 33 | |||
| 34 | log.Printf("Generated key: %s", key.Npub()) | ||
| 35 | |||
| 36 | event := &nostr.Event{ | ||
| 37 | PubKey: key.Public(), | ||
| 38 | CreatedAt: time.Now().Unix(), | ||
| 39 | Kind: 1, | ||
| 40 | Tags: nostr.Tags{}, | ||
| 41 | Content: "Hello from gRPC client!", | ||
| 42 | } | ||
| 43 | |||
| 44 | if err := key.Sign(event); err != nil { | ||
| 45 | log.Fatalf("failed to sign event: %v", err) | ||
| 46 | } | ||
| 47 | |||
| 48 | pbEvent := &pb.Event{ | ||
| 49 | Id: event.ID, | ||
| 50 | Pubkey: event.PubKey, | ||
| 51 | CreatedAt: event.CreatedAt, | ||
| 52 | Kind: int32(event.Kind), | ||
| 53 | Tags: []*pb.Tag{}, | ||
| 54 | Content: event.Content, | ||
| 55 | Sig: event.Sig, | ||
| 56 | } | ||
| 57 | |||
| 58 | log.Println("Publishing event...") | ||
| 59 | resp, err := client.PublishEvent(ctx, &pb.PublishEventRequest{Event: pbEvent}) | ||
| 60 | if err != nil { | ||
| 61 | log.Fatalf("failed to publish: %v", err) | ||
| 62 | } | ||
| 63 | |||
| 64 | if resp.Accepted { | ||
| 65 | log.Printf("✓ Event published successfully: %s", event.ID) | ||
| 66 | } else { | ||
| 67 | log.Printf("✗ Event rejected: %s", resp.Message) | ||
| 68 | return | ||
| 69 | } | ||
| 70 | |||
| 71 | log.Println("Querying events...") | ||
| 72 | queryResp, err := client.QueryEvents(ctx, &pb.QueryRequest{ | ||
| 73 | Filters: []*pb.Filter{ | ||
| 74 | {Authors: []string{key.Public()}}, | ||
| 75 | }, | ||
| 76 | }) | ||
| 77 | if err != nil { | ||
| 78 | log.Fatalf("failed to query: %v", err) | ||
| 79 | } | ||
| 80 | |||
| 81 | log.Printf("Found %d events", len(queryResp.Events)) | ||
| 82 | for _, e := range queryResp.Events { | ||
| 83 | log.Printf(" - %s: %s", e.Id[:16], e.Content) | ||
| 84 | } | ||
| 85 | } | ||
