chore: minor changes to client and server logic (#32)

example/.env.example

@@ -1 +1,2 @@
-SPENDING_KEY=""
+SPENDING_KEY=""
+WALLET_SERVER_HOST="walletserver.ironfish.network:50051"

example/src/Client/Client.ts

@@ -5,7 +5,7 @@ import { AccountsManager } from "./utils/AccountsManager";
 import { BlockCache } from "./utils/BlockCache";
 
 const client = new LightStreamerClient(
-  "localhost:50051",
+  process.env["WALLET_SERVER_HOST"] ?? "localhost:50051",
   credentials.createInsecure(),
 );
 

example/src/index.ts

@@ -1,8 +1,8 @@
 import { config } from "dotenv";
-import { Client } from "./Client/Client";
-
 config();
 
+import { Client } from "./Client/Client";
+
 async function main() {
   const client = new Client();
   await client.start();

src/services/LightStreamer.ts

@@ -95,22 +95,39 @@ class LightStreamer implements LightStreamerServer {
 
   public getBlockRange: handleServerStreamingCall<BlockRange, LightBlock> =
     async (call) => {
-      if (!call.request.start?.sequence || !call.request.end?.sequence) {
+      if (
+        (!call.request.start?.sequence && call.request.start?.sequence !== 0) ||
+        (!call.request.end?.sequence && call.request.end?.sequence !== 0)
+      ) {
         call.emit("error", {
           code: status.INVALID_ARGUMENT,
           details: "Must provide sequence for start and end",
         });
         call.end();
         return;
       }
-      if (call.request.start.sequence >= call.request.end.sequence) {
+
+      if (call.request.start.sequence > call.request.end.sequence) {
         call.emit("error", {
           code: status.INVALID_ARGUMENT,
-          details: "End sequence must be greater than start sequence",
+          details:
+            "End sequence must be greater than or equal to start sequence",
         });
         call.end();
         return;
       }
+
+      const requested_block_amount =
+        call.request.end.sequence - call.request.start.sequence;
+      if (
+        process.env["BLOCK_RANGE_MAX"] &&
+        requested_block_amount > Number(process.env["BLOCK_RANGE_MAX"])
+      ) {
+        call.emit("error", {
+          code: status.INVALID_ARGUMENT,
+          details: `Maximum block range exceeded for server, requested: ${requested_block_amount}, allowed: ${process.env["BLOCK_RANGE_MAX"]}}`,
+        });
+      }
       try {
         for (
           let i = call.request.start.sequence;

test/server.test.slow.ts

@@ -184,7 +184,7 @@ describe("getBlock", () => {
       });
     });
     expect(promise).rejects.toThrow(
-      "INVALID_ARGUMENT: End sequence must be greater than start sequence",
+      "INVALID_ARGUMENT: End sequence must be greater than or equal to start sequence",
     );
   });