fix up hooks

Author: jnsdls

src/Provider.tsx

@@ -1,5 +1,5 @@
 import React, { createContext, useEffect, useMemo, useState } from "react";
-import { ThirdwebSDK, SDKOptions } from "@thirdweb-dev/sdk";
+import { ThirdwebSDK, SDKOptions, IStorage } from "@thirdweb-dev/sdk";
 import {
   WagmiProvider,
   ProviderProps as WagmiproviderProps,
@@ -87,6 +87,11 @@ export interface ThirdwebProviderProps<
   desiredChainId: TSupportedChain extends Chain
     ? TSupportedChain["id"]
     : TSupportedChain | undefined;
+
+  /**
+   * The storage interface to use with the sdk.
+   */
+  storageInterface?: IStorage;
 }
 
 const defaultChainRpc: ThirdwebProviderProps["chainRpc"] = {
@@ -246,15 +251,18 @@ interface SDKContext {
 const ThirdwebSDKContext = createContext<SDKContext>({});
 
 const ThirdwebSDKProvider: React.FC<
-  Pick<ThirdwebProviderProps, "desiredChainId" | "sdkOptions">
-> = ({ sdkOptions, desiredChainId, children }) => {
+  Pick<
+    ThirdwebProviderProps,
+    "desiredChainId" | "sdkOptions" | "storageInterface"
+  >
+> = ({ sdkOptions, desiredChainId, storageInterface, children }) => {
   const [sdk, setSDK] = useState<ThirdwebSDK | undefined>(undefined);
   const provider = useProvider();
   const signer = useSigner();
 
   useEffect(() => {
     if (desiredChainId) {
-      const _sdk = new ThirdwebSDK(provider, sdkOptions);
+      const _sdk = new ThirdwebSDK(provider, sdkOptions, storageInterface);
       (_sdk as any)._chainId = desiredChainId;
       setSDK(_sdk);
     }

src/hooks/modules/useContract.ts

@@ -9,10 +9,10 @@ import { useSDK } from "../../Provider";
  */
 export function useContract<TContractType extends ContractType>(
   contractType: TContractType,
-  contractAddress: string,
+  contractAddress?: string,
 ): ContractForContractType<TContractType> {
   const sdk = useSDK();
-  if (!sdk) {
+  if (!sdk || !contractAddress) {
     return undefined;
   }
   return sdk.getContract(contractAddress, contractType);

src/hooks/modules/useEdition.ts

@@ -1,6 +1,6 @@
 import { Edition } from "@thirdweb-dev/sdk";
 import { useContract } from "./useContract";
 
-export function useEdition(moduleAddress: string) {
-  return useContract("edition", moduleAddress) as Edition;
+export function useEdition(moduleAddress?: string): Edition | undefined {
+  return useContract("edition", moduleAddress);
 }

src/hooks/modules/useEditionDrop.ts

@@ -1,6 +1,8 @@
 import { EditionDrop } from "@thirdweb-dev/sdk";
 import { useContract } from "./useContract";
 
-export function useEditionDrop(moduleAddress: string) {
-  return useContract("edition-drop", moduleAddress) as EditionDrop;
+export function useEditionDrop(
+  moduleAddress?: string,
+): EditionDrop | undefined {
+  return useContract("edition-drop", moduleAddress);
 }

src/hooks/modules/useMarketplace.ts

@@ -1,6 +1,8 @@
 import { Marketplace } from "@thirdweb-dev/sdk";
 import { useContract } from "./useContract";
 
-export function useMarketplace(moduleAddress: string) {
-  return useContract("marketplace", moduleAddress) as Marketplace;
+export function useMarketplace(
+  moduleAddress?: string,
+): Marketplace | undefined {
+  return useContract("marketplace", moduleAddress);
 }

src/hooks/modules/useNFTCollection.ts

@@ -1,6 +1,8 @@
 import { NFTCollection } from "@thirdweb-dev/sdk";
 import { useContract } from "./useContract";
 
-export function useNFTCollection(moduleAddress: string) {
-  return useContract("nft-collection", moduleAddress) as NFTCollection;
+export function useNFTCollection(
+  moduleAddress?: string,
+): NFTCollection | undefined {
+  return useContract("nft-collection", moduleAddress);
 }

src/hooks/modules/useNFTDrop.ts

@@ -1,6 +1,6 @@
 import { NFTDrop } from "@thirdweb-dev/sdk";
 import { useContract } from "./useContract";
 
-export function useNFTDrop(moduleAddress: string) {
-  return useContract("nft-drop", moduleAddress) as NFTDrop;
+export function useNFTDrop(moduleAddress?: string): NFTDrop | undefined {
+  return useContract("nft-drop", moduleAddress);
 }

src/hooks/modules/usePack.ts

@@ -1,6 +1,6 @@
 import { Pack } from "@thirdweb-dev/sdk";
 import { useContract } from "./useContract";
 
-export function usePackModule(moduleAddress: string) {
-  return useContract("pack", moduleAddress) as Pack;
+export function usePackModule(moduleAddress?: string): Pack | undefined {
+  return useContract("pack", moduleAddress);
 }

src/hooks/modules/useSplit.ts

@@ -1,6 +1,6 @@
 import { Split } from "@thirdweb-dev/sdk";
 import { useContract } from "./useContract";
 
-export function useSplitModule(moduleAddress: string) {
-  return useContract("split", moduleAddress) as Split;
+export function useSplitModule(moduleAddress?: string): Split | undefined {
+  return useContract("split", moduleAddress);
 }

src/hooks/modules/useTokenModule.ts

@@ -1,6 +1,6 @@
 import { Token } from "@thirdweb-dev/sdk";
 import { useContract } from "./useContract";
 
-export function useToken(moduleAddress: string) {
-  return useContract("token", moduleAddress) as Token;
+export function useToken(moduleAddress?: string): Token | undefined {
+  return useContract("token", moduleAddress);
 }