[SDK] Refactor: More Viem to Ox migration

.github/workflows/auto-assign.yml

@@ -2,7 +2,7 @@ name: Auto Author Assign
 
 on:
   pull_request:
-    types: [opened, reopened]
+    types: [opened, reopened, ready_for_review, draft]
 
 permissions:
   pull-requests: write

apps/dashboard/src/components/contract-components/contract-publish-form/abi-selector.tsx

@@ -32,14 +32,12 @@ export const AbiSelector: React.FC<AbiSelectorProps> = ({
         options={options}
         defaultValue={options.find((o) => o.value === defaultValue)}
         chakraStyles={{
-          // @ts-expect-error - this works fine
           container: (provided) => ({
             ...provided,
             width: "full",
           }),
         }}
         value={options.find((o) => o.value === value)}
-        // @ts-expect-error - this works fine
         onChange={(selectedFn) => {
           if (selectedFn) {
             onChange((selectedFn as { label: string; value: string }).value);

packages/thirdweb/src/auth/is-erc6492-signature.ts

@@ -1,4 +1,4 @@
-import { sliceHex } from "viem";
+import * as ox__Hex from "ox/Hex";
 import type { Hex } from "../utils/encoding/hex.js";
 import { ERC_6492_MAGIC_VALUE } from "./constants.js";
 
@@ -19,5 +19,5 @@ import { ERC_6492_MAGIC_VALUE } from "./constants.js";
  * @auth
  */
 export function isErc6492Signature(signature: Hex): boolean {
-  return sliceHex(signature, -32) === ERC_6492_MAGIC_VALUE;
+  return ox__Hex.slice(signature, -32) === ERC_6492_MAGIC_VALUE;
 }

packages/thirdweb/src/auth/parse-erc6492-signature.ts

@@ -1,4 +1,6 @@
-import { decodeAbiParameters, isErc6492Signature } from "viem";
+import * as ox__AbiParameters from "ox/AbiParameters";
+import * as ox__Address from "ox/Address";
+import { WrappedSignature as ox__WrappedSignature } from "ox/erc6492";
 import type { Hex } from "../utils/encoding/hex.js";
 import type { OneOf } from "../utils/type-utils.js";
 import type { Erc6492Signature } from "./types.js";
@@ -29,13 +31,17 @@ export type ParseErc6492SignatureReturnType = OneOf<
 export function parseErc6492Signature(
   signature: Hex,
 ): ParseErc6492SignatureReturnType {
-  if (!isErc6492Signature(signature)) {
+  if (!ox__WrappedSignature.validate(signature)) {
     return { signature };
   }
 
-  const [address, data, originalSignature] = decodeAbiParameters(
+  const [address, data, originalSignature] = ox__AbiParameters.decode(
     [{ type: "address" }, { type: "bytes" }, { type: "bytes" }],
     signature,
   );
-  return { address: address, data, signature: originalSignature };
+  return {
+    address: ox__Address.checksum(address),
+    data,
+    signature: originalSignature,
+  };
 }

packages/thirdweb/src/auth/verify-hash.ts

@@ -1,12 +1,8 @@
-import {
-  type Signature,
-  encodeDeployData,
-  encodeFunctionData,
-  isErc6492Signature,
-  serializeSignature,
-  universalSignatureValidatorAbi,
-  universalSignatureValidatorByteCode,
-} from "viem";
+import * as ox__Abi from "ox/Abi";
+import * as ox__AbiConstructor from "ox/AbiConstructor";
+import * as ox__AbiFunction from "ox/AbiFunction";
+import * as ox__Signature from "ox/Signature";
+import { WrappedSignature as ox__WrappedSignature } from "ox/erc6492";
 import type { Chain } from "../chains/types.js";
 import type { ThirdwebClient } from "../client/client.js";
 import { type ThirdwebContract, getContract } from "../contract/contract.js";
@@ -20,7 +16,7 @@
 
 export type VerifyHashParams = {
   hash: Hex;
-  signature: string | Uint8Array | Signature;
+  signature: string | Uint8Array | ox__Signature.Signature;
   address: string;
   client: ThirdwebClient;
   chain: Chain;
@@ -71,7 +67,7 @@
   const signatureHex = (() => {
     if (isHex(signature)) return signature;
     if (typeof signature === "object" && "r" in signature && "s" in signature)
-      return serializeSignature(signature);
+      return ox__Signature.toHex(signature);
     if (signature instanceof Uint8Array) return fromBytes(signature, "hex");
     // We should never hit this but TS doesn't know that
     throw new Error(
@@ -85,7 +81,7 @@
     if (!accountFactory) return signatureHex;
 
     // If this sigature was already wrapped for ERC-6492, carry on
-    if (isErc6492Signature(signatureHex)) return signatureHex;
+    if (ox__WrappedSignature.validate(signatureHex)) return signatureHex;
 
     // Otherwise, serialize the signature for ERC-6492 validation
     return serializeErc6492Signature({
@@ -100,23 +96,23 @@
     data: Hex;
   };
   const zkSyncChain = await isZkSyncChain(chain);
+  const abi = ox__Abi.from(ox__WrappedSignature.universalSignatureValidatorAbi);
   if (zkSyncChain) {
     // zksync chains dont support deploying code with eth_call
     // need to call a deployed contract instead
     verificationData = {
       to: ZKSYNC_VALIDATOR_ADDRESS,
-      data: encodeFunctionData({
-        abi: universalSignatureValidatorAbi,
-        functionName: "isValidSig",
-        args: [address, hash, wrappedSignature],
-      }),
+      data: ox__AbiFunction.encodeData(
+        ox__AbiFunction.fromAbi(abi, "isValidSig"),
+        [address, hash, wrappedSignature],
+      ),
     };
   } else {
+    const validatorConstructor = ox__AbiConstructor.fromAbi(abi);
     verificationData = {
-      data: encodeDeployData({
-        abi: universalSignatureValidatorAbi,
+      data: ox__AbiConstructor.encode(validatorConstructor, {
         args: [address, hash, wrappedSignature],
-        bytecode: universalSignatureValidatorByteCode,
+        bytecode: ox__WrappedSignature.universalSignatureValidatorBytecode,
       }),
     };
   }

packages/thirdweb/src/auth/verify-signature.ts

@@ -1,17 +1,27 @@
-import { type SignableMessage, type Signature, recoverAddress } from "viem";
+import * as ox__Bytes from "ox/Bytes";
+import * as ox__Hex from "ox/Hex";
+import * as ox__Secp256k1 from "ox/Secp256k1";
+import * as ox__Signature from "ox/Signature";
 import type { Chain } from "../chains/types.js";
 import type { ThirdwebClient } from "../client/client.js";
 import { type Hex, isHex } from "../utils/encoding/hex.js";
 import { hashMessage } from "../utils/hashing/hashMessage.js";
 import type { Prettify } from "../utils/type-utils.js";
 import { verifyHash } from "./verify-hash.js";
 
+type Message = Prettify<
+  | string
+  | {
+      raw: Hex | Uint8Array;
+    }
+>;
+
 /**
  * @auth
  */
 export type VerifyEOASignatureParams = {
-  message: string | SignableMessage;
-  signature: string | Uint8Array | Signature;
+  message: string | Message;
+  signature: string | Uint8Array | ox__Signature.Signature;
   address: string;
 };
 
@@ -39,9 +49,9 @@
     return false;
   }
 
-  const recoveredAddress = await recoverAddress({
-    hash: messageHash,
-    signature: options.signature,
+  const recoveredAddress = ox__Secp256k1.recoverAddress({
+    payload: messageHash,
+    signature: ox__Signature.fromHex(options.signature),
   });
 
   if (recoveredAddress.toLowerCase() === options.address.toLowerCase()) {
@@ -103,9 +113,35 @@
   accountFactory,
 }: VerifyContractWalletSignatureParams) {
   const messageHash = hashMessage(message);
+
+  const parsedSignature = (() => {
+    if (ox__Bytes.validate(signature)) {
+      const sig = ox__Signature.fromBytes(signature);
+      return {
+        r: ox__Hex.fromNumber(sig.r),
+        s: ox__Hex.fromNumber(sig.s),
+        yParity: sig.yParity,
+      };
+    } else if (typeof signature === "object") {
+      return {
+        r: ox__Hex.fromNumber(signature.r),
+        s: ox__Hex.fromNumber(signature.s),
+        yParity: signature.yParity,
+      };
+    }
+    return signature;
+  })();
+
   return verifyHash({
     hash: messageHash,
-    signature,
+    signature:
+      typeof parsedSignature === "string"
+        ? parsedSignature
+        : {
+            r: ox__Hex.toBigInt(parsedSignature.r),
+            s: ox__Hex.toBigInt(parsedSignature.s),
+            yParity: parsedSignature.yParity,
+          },
     address,
     client,
     chain,

packages/thirdweb/src/auth/verify-typed-data.ts

@@ -1,18 +1,20 @@
-import type { Signature, TypedData, TypedDataDefinition } from "viem";
-import { hashTypedData } from "viem";
+import type * as ox__Signature from "ox/Signature";
+import * as ox__TypedData from "ox/TypedData";
 import type { Chain } from "../chains/types.js";
 import type { ThirdwebClient } from "../client/client.js";
 import type { Hex } from "../utils/encoding/hex.js";
 import type { HashTypedDataParams } from "../utils/hashing/hashTypedData.js";
 import { type VerifyHashParams, verifyHash } from "./verify-hash.js";
 
 export type VerifyTypedDataParams<
-  typedData extends TypedData | Record<string, unknown> = TypedData,
+  typedData extends
+    | ox__TypedData.TypedData
+    | Record<string, unknown> = ox__TypedData.TypedData,
   primaryType extends keyof typedData | "EIP712Domain" = keyof typedData,
 > = Omit<VerifyHashParams, "hash"> &
-  TypedDataDefinition<typedData, primaryType> & {
+  ox__TypedData.Definition<typedData, primaryType> & {
     address: string;
-    signature: string | Uint8Array | Signature;
+    signature: string | Uint8Array | ox__Signature.Signature;
     client: ThirdwebClient;
     chain: Chain;
     accountFactory?: {
@@ -80,7 +82,7 @@ export type VerifyTypedDataParams<
  * @auth
  */
 export async function verifyTypedData<
-  typedData extends TypedData | Record<string, unknown>,
+  typedData extends ox__TypedData.TypedData | Record<string, unknown>,
   primaryType extends keyof typedData | "EIP712Domain",
 >({
   address,
@@ -93,7 +95,7 @@ export async function verifyTypedData<
   primaryType,
   types,
 }: VerifyTypedDataParams<typedData, primaryType>): Promise<boolean> {
-  const messageHash = hashTypedData({
+  const messageHash = ox__TypedData.getSignPayload({
     message,
     domain,
     primaryType,

packages/thirdweb/src/contract/deployment/utils/create-2-factory.ts

@@ -1,4 +1,4 @@
-import { getContractAddress } from "viem";
+import * as ox__ContractAddress from "ox/ContractAddress";
 import { getGasPrice } from "../../../gas/get-gas-price.js";
 import { eth_getBalance } from "../../../rpc/actions/eth_getBalance.js";
 import { eth_sendRawTransaction } from "../../../rpc/actions/eth_sendRawTransaction.js";
@@ -208,7 +208,7 @@ async function _getCreate2FactoryDeploymentInfo(
     },
     signature: SIGNATURE,
   });
-  const create2FactoryAddress = getContractAddress({
+  const create2FactoryAddress = ox__ContractAddress.from({
     from: deploymentTransaction.signerAddress,
     nonce: 0n,
   });

packages/thirdweb/src/contract/verification/constructor-params.ts

@@ -1,5 +1,6 @@
-import type { Abi } from "abitype";
-import { decodeAbiParameters } from "viem";
+import type * as ox__Abi from "ox/Abi";
+import * as ox__AbiConstructor from "ox/AbiConstructor";
+import * as ox__AbiParameters from "ox/AbiParameters";
 import { eth_getTransactionByHash } from "../../rpc/actions/eth_getTransactionByHash.js";
 import { getRpcClient } from "../../rpc/rpc.js";
 import type { ThirdwebContract } from "../contract.js";
@@ -10,19 +11,9 @@ type FetchConstructorParamsOptions = {
   contract: ThirdwebContract;
   explorerApiUrl: string;
   explorerApiKey: string;
-  abi: Abi;
+  abi: ox__Abi.Abi;
 };
 
-// TODO: move to abi helpers (?)
-function extractConstructorParamsFromAbi(abi: Abi) {
-  for (const input of abi) {
-    if (input.type === "constructor") {
-      return input.inputs || [];
-    }
-  }
-  return [];
-}
-
 const RequestStatus = {
   OK: "1",
   NOTOK: "0",
@@ -37,7 +28,8 @@ const RequestStatus = {
 export async function fetchConstructorParams(
   options: FetchConstructorParamsOptions,
 ): Promise<string> {
-  const constructorParamTypes = extractConstructorParamsFromAbi(options.abi);
+  const abiConstructor = ox__AbiConstructor.fromAbi(options.abi);
+  const constructorParamTypes = ox__AbiParameters.from(abiConstructor.inputs);
   if (constructorParamTypes.length === 0) {
     return "";
   }
@@ -114,7 +106,8 @@ export async function fetchConstructorParams(
   try {
     // sanity check that the constructor params are valid
     // TODO: should we sanity check after each attempt?
-    decodeAbiParameters(constructorParamTypes, `0x${constructorArgs}`);
+
+    ox__AbiParameters.decode(constructorParamTypes, `0x${constructorArgs}`);
   } catch {
     throw new Error(
       "Verifying this contract requires it to be published. Run `npx thirdweb publish` to publish this contract, then try again.",

packages/thirdweb/src/event/types.ts

@@ -1,5 +1,6 @@
 import type { AbiParameter, AbiParameterToPrimitiveType } from "abitype";
-import type { Hex, LogTopic } from "viem";
+import type * as ox__Hex from "ox/Hex";
+import type { Log as ox__Log } from "ox/Log";
 import type { Filter, MaybeRequired, Prettify } from "../utils/type-utils.js";
 
 //////////////////////////////////////////////////////////////////////
@@ -83,11 +84,11 @@ type _HasUnnamedAbiParameter<TAbiParameters extends readonly AbiParameter[]> =
  * @internal
  */
 type LogTopicType<
-  TPrimitiveType = Hex,
-  TTopic extends LogTopic = LogTopic,
-> = TTopic extends Hex
+  TPrimitiveType = ox__Hex.Hex,
+  TTopic extends ox__Log["topics"][0] = ox__Log["topics"][0],
+> = TTopic extends ox__Hex.Hex
   ? TPrimitiveType
-  : TTopic extends Hex[]
+  : TTopic extends ox__Hex.Hex[]
     ? TPrimitiveType[]
     : TTopic extends null
       ? null

packages/thirdweb/src/exports/utils.ts

@@ -167,7 +167,7 @@ export { stringify } from "../utils/json.js";
 // ------------------------------------------------
 // values
 // ------------------------------------------------
-export { maxUint256 } from "viem";
+export { maxUint256 } from "ox/Solidity";
 
 // ------------------------------------------------
 // jwt