Added missing beam extension and fixed paths

Author: Segfaultd

src/client/LumClient.ts

@@ -14,10 +14,11 @@ import {
 
 import { BaseAccount } from '../codec/cosmos/auth/v1beta1/auth';
 import { LumWallet, LumUtils, LumTypes, LumRegistry } from '..';
+import { BeamExtension, setupBeamExtension as BeamSetupBeamExtension } from '../extensions';
 
 export class LumClient {
     readonly tmClient: Tendermint34Client;
-    readonly queryClient: StargateQueryClient & AuthExtension & BankExtension & DistributionExtension & StakingExtension;
+    readonly queryClient: StargateQueryClient & AuthExtension & BankExtension & DistributionExtension & StakingExtension & BeamExtension;
     private chainId?: string;
 
     /**
@@ -27,7 +28,14 @@ export class LumClient {
      */
     constructor(tmClient: Tendermint34Client) {
         this.tmClient = tmClient;
-        this.queryClient = StargateQueryClient.withExtensions(tmClient, StargateSetupAuthExtension, StargateSetupBankExtension, StargateDistributionExtension, StargateStakingExtension);
+        this.queryClient = StargateQueryClient.withExtensions(
+            tmClient,
+            StargateSetupAuthExtension,
+            StargateSetupBankExtension,
+            StargateDistributionExtension,
+            StargateStakingExtension,
+            BeamSetupBeamExtension,
+        );
 
         // Used for debugging while gasWanted, gasUsed and codespace are still waiting to be included in the code lib
         // // @ts-ignore

src/codec/chain/beam/query.ts

@@ -11,15 +11,15 @@ export interface QueryGetBeamRequest {
 }
 
 export interface QueryGetBeamResponse {
-    Beam?: Beam;
+    beam?: Beam;
 }
 
 export interface QueryFetchBeamsRequest {
     pagination?: PageRequest;
 }
 
 export interface QueryFetchBeamsResponse {
-    Beam: Beam[];
+    beams: Beam[];
     pagination?: PageResponse;
 }
 
@@ -82,8 +82,8 @@ const baseQueryGetBeamResponse: object = {};
 
 export const QueryGetBeamResponse = {
     encode(message: QueryGetBeamResponse, writer: _m0.Writer = _m0.Writer.create()): _m0.Writer {
-        if (message.Beam !== undefined) {
-            Beam.encode(message.Beam, writer.uint32(10).fork()).ldelim();
+        if (message.beam !== undefined) {
+            Beam.encode(message.beam, writer.uint32(10).fork()).ldelim();
         }
         return writer;
     },
@@ -96,7 +96,7 @@ export const QueryGetBeamResponse = {
             const tag = reader.uint32();
             switch (tag >>> 3) {
                 case 1:
-                    message.Beam = Beam.decode(reader, reader.uint32());
+                    message.beam = Beam.decode(reader, reader.uint32());
                     break;
                 default:
                     reader.skipType(tag & 7);
@@ -108,26 +108,26 @@ export const QueryGetBeamResponse = {
 
     fromJSON(object: any): QueryGetBeamResponse {
         const message = { ...baseQueryGetBeamResponse } as QueryGetBeamResponse;
-        if (object.Beam !== undefined && object.Beam !== null) {
-            message.Beam = Beam.fromJSON(object.Beam);
+        if (object.beam !== undefined && object.beam !== null) {
+            message.beam = Beam.fromJSON(object.beam);
         } else {
-            message.Beam = undefined;
+            message.beam = undefined;
         }
         return message;
     },
 
     toJSON(message: QueryGetBeamResponse): unknown {
         const obj: any = {};
-        message.Beam !== undefined && (obj.Beam = message.Beam ? Beam.toJSON(message.Beam) : undefined);
+        message.beam !== undefined && (obj.beam = message.beam ? Beam.toJSON(message.beam) : undefined);
         return obj;
     },
 
     fromPartial(object: DeepPartial<QueryGetBeamResponse>): QueryGetBeamResponse {
         const message = { ...baseQueryGetBeamResponse } as QueryGetBeamResponse;
-        if (object.Beam !== undefined && object.Beam !== null) {
-            message.Beam = Beam.fromPartial(object.Beam);
+        if (object.beam !== undefined && object.beam !== null) {
+            message.beam = Beam.fromPartial(object.beam);
         } else {
-            message.Beam = undefined;
+            message.beam = undefined;
         }
         return message;
     },
@@ -192,7 +192,7 @@ const baseQueryFetchBeamsResponse: object = {};
 
 export const QueryFetchBeamsResponse = {
     encode(message: QueryFetchBeamsResponse, writer: _m0.Writer = _m0.Writer.create()): _m0.Writer {
-        for (const v of message.Beam) {
+        for (const v of message.beams) {
             Beam.encode(v!, writer.uint32(10).fork()).ldelim();
         }
         if (message.pagination !== undefined) {
@@ -205,12 +205,12 @@ export const QueryFetchBeamsResponse = {
         const reader = input instanceof _m0.Reader ? input : new _m0.Reader(input);
         let end = length === undefined ? reader.len : reader.pos + length;
         const message = { ...baseQueryFetchBeamsResponse } as QueryFetchBeamsResponse;
-        message.Beam = [];
+        message.beams = [];
         while (reader.pos < end) {
             const tag = reader.uint32();
             switch (tag >>> 3) {
                 case 1:
-                    message.Beam.push(Beam.decode(reader, reader.uint32()));
+                    message.beams.push(Beam.decode(reader, reader.uint32()));
                     break;
                 case 2:
                     message.pagination = PageResponse.decode(reader, reader.uint32());
@@ -225,10 +225,10 @@ export const QueryFetchBeamsResponse = {
 
     fromJSON(object: any): QueryFetchBeamsResponse {
         const message = { ...baseQueryFetchBeamsResponse } as QueryFetchBeamsResponse;
-        message.Beam = [];
-        if (object.Beam !== undefined && object.Beam !== null) {
-            for (const e of object.Beam) {
-                message.Beam.push(Beam.fromJSON(e));
+        message.beams = [];
+        if (object.beams !== undefined && object.beams !== null) {
+            for (const e of object.beams) {
+                message.beams.push(Beam.fromJSON(e));
             }
         }
         if (object.pagination !== undefined && object.pagination !== null) {
@@ -241,21 +241,21 @@ export const QueryFetchBeamsResponse = {
 
     toJSON(message: QueryFetchBeamsResponse): unknown {
         const obj: any = {};
-        if (message.Beam) {
-            obj.Beam = message.Beam.map((e) => (e ? Beam.toJSON(e) : undefined));
+        if (message.beams) {
+            obj.beams = message.beams.map((e) => (e ? Beam.toJSON(e) : undefined));
         } else {
-            obj.Beam = [];
+            obj.beams = [];
         }
         message.pagination !== undefined && (obj.pagination = message.pagination ? PageResponse.toJSON(message.pagination) : undefined);
         return obj;
     },
 
     fromPartial(object: DeepPartial<QueryFetchBeamsResponse>): QueryFetchBeamsResponse {
         const message = { ...baseQueryFetchBeamsResponse } as QueryFetchBeamsResponse;
-        message.Beam = [];
-        if (object.Beam !== undefined && object.Beam !== null) {
-            for (const e of object.Beam) {
-                message.Beam.push(Beam.fromPartial(e));
+        message.beams = [];
+        if (object.beams !== undefined && object.beams !== null) {
+            for (const e of object.beams) {
+                message.beams.push(Beam.fromPartial(e));
             }
         }
         if (object.pagination !== undefined && object.pagination !== null) {

src/extensions/beam.ts

@@ -0,0 +1,33 @@
+import { createProtobufRpcClient, QueryClient } from '@cosmjs/stargate';
+
+import { assert } from '@cosmjs/utils';
+
+import { Beam } from '../codec/chain/beam/beam';
+import { QueryClientImpl } from '../codec/chain/beam/query';
+
+export interface BeamExtension {
+    readonly beam: {
+        readonly get: (id: string) => Promise<Beam>;
+        readonly fetch: () => Promise<Beam[]>;
+    };
+}
+
+export const setupBeamExtension = (base: QueryClient): BeamExtension => {
+    const rpc = createProtobufRpcClient(base);
+    const queryService = new QueryClientImpl(rpc);
+
+    return {
+        beam: {
+            get: async (id: string) => {
+                const { beam } = await queryService.Beam({ id });
+                assert(beam);
+                return beam;
+            },
+            fetch: async () => {
+                const { beams } = await queryService.Beams({});
+                assert(beams);
+                return beams;
+            },
+        },
+    };
+};

src/extensions/index.ts

@@ -0,0 +1 @@
+export * from './beam';