Update trade and marketData exports

api/index.ts

@@ -1,2 +1,2 @@
 export { default as marketData } from "./marketData/index.ts";
-export { default as trade } from "./trade/index.ts";
+export { default as trade } from "./trade/consumer.ts";

api/marketData/index.ts

@@ -40,7 +40,7 @@ import {
   TradeResponse,
 } from "./types/stocks.ts";
 
-export default ({ request }: ClientContext) => ({
+const methods = ({ request }: ClientContext) => ({
   v1beta1: {
     corporateActions: (queryParams: CorporateActionsQueryParams) =>
       request<CorporateActionsResponse>({
@@ -412,3 +412,9 @@ export default ({ request }: ClientContext) => ({
     },
   },
 });
+
+export default <T extends ReturnType<typeof methods>>(
+  context: ClientContext
+): T => {
+  return methods(context) as T;
+};

api/trade/index.ts → api/trade/consumer.ts

@@ -4,15 +4,9 @@ import {
 } from "./types/portfolio.ts";
 
 import {
-  AddAssetToWatchlistParams,
-  CreateWatchlistParams,
-  DeleteSymbolFromWatchlistParams,
-  DeleteWatchlistByNameParams,
-  GetWatchlistByNameParams,
-  UpdateWatchlistByNameParams,
-  UpdateWatchlistParams,
-  Watchlist,
-} from "./types/watchlist.ts";
+  AnnouncementsQueryParams,
+  CorporateActionAnnouncement,
+} from "./types/corporate.ts";
 
 import {
   Account,
@@ -21,11 +15,6 @@ import {
   UpdatedAccountConfigurations,
 } from "./types/account.ts";
 
-import {
-  AnnouncementsQueryParams,
-  CorporateActionAnnouncement,
-} from "./types/corporate.ts";
-
 import {
   CryptoFundingTransfer,
   CryptoFundingWallet,
@@ -36,6 +25,17 @@ import {
   WithdrawalParams,
 } from "./types/cryptoFunding.ts";
 
+import {
+  AddAssetToWatchlistParams,
+  CreateWatchlistParams,
+  DeleteSymbolFromWatchlistParams,
+  DeleteWatchlistByNameParams,
+  GetWatchlistByNameParams,
+  UpdateWatchlistByNameParams,
+  UpdateWatchlistParams,
+  Watchlist,
+} from "./types/watchlist.ts";
+
 import { ClientContext } from "../../factory/createClient.ts";
 import { Asset } from "./types/asset.ts";
 import { MarketCalendar } from "./types/calendar.ts";
@@ -44,7 +44,7 @@ import { OptionContract, OptionContractsQueryParams } from "./types/options.ts";
 import { CreateOrderOptions, Order, PatchOrderOptions } from "./types/order.ts";
 import { ClosePositionOptions, Position } from "./types/position.ts";
 
-export default ({ request }: ClientContext) => ({
+const methods = ({ request }: ClientContext) => ({
   v2: {
     account: {
       get: () =>
@@ -382,3 +382,9 @@ export default ({ request }: ClientContext) => ({
     },
   },
 });
+
+export default <T extends ReturnType<typeof methods>>(
+  context: ClientContext
+): T => {
+  return methods(context) as T;
+};

api/types/shared.ts

@@ -0,0 +1,3 @@
+import { ClientContext } from "../../factory/createClient.ts";
+
+export type ClientContextConsumer<T> = (context: ClientContext) => T;

factory/createClient.ts

@@ -4,6 +4,7 @@ import { StockDataWebSocket } from "../api/trade/types/websocket_2.ts";
 import { CryptoWebSocket } from "../api/trade/types/websocket_3.ts";
 import { NewsWebSocket } from "../api/trade/types/websocket_4.ts";
 import { OptionsWebSocket } from "../api/trade/types/websocket_5.ts";
+import { ClientContextConsumer } from "../api/types/shared.ts";
 import { TokenBucketOptions, createTokenBucket } from "./createTokenBucket.ts";
 
 // Used to share the client options and request function between the different API methods
@@ -32,8 +33,10 @@ type RequestOptions = {
 // The client object that is returned by createClient
 export type Client = {
   rest: {
-    trade: ReturnType<typeof trade>;
-    marketData: ReturnType<typeof marketData>;
+    trade: ReturnType<ClientContextConsumer<ReturnType<typeof trade>>>;
+    marketData: ReturnType<
+      ClientContextConsumer<ReturnType<typeof marketData>>
+    >;
   };
   websocket: {
     trade: TradeWebSocket;