@zoralabs/protocol-deployments 0.1.10 → 0.1.11

package/src/index.ts

@@ -3,5 +3,6 @@
 // The can be generated by running `yarn prepack` in the root
 export * from "./generated/wagmi";
 export * from "./typedData";
+export * from "./types";
 export * as contracts1155 from "./generated/1155";
 export * as mints from "./generated/mints";

package/src/typedData.ts

@@ -4,8 +4,29 @@ import {
   TypedData,
   TypedDataToPrimitiveTypes,
 } from "abitype";
-import { TypedDataDefinition } from "viem";
-import { zoraMints1155Address } from "./generated/wagmi";
+import {
+  Hex,
+  TypedDataDefinition,
+  encodeAbiParameters,
+  getAbiItem,
+  keccak256,
+  toHex,
+} from "viem";
+import {
+  zoraMints1155Address,
+  iPremintDefinitionsABI,
+} from "./generated/wagmi";
+import {
+  PremintConfigEncoded,
+  PremintConfigV1,
+  PremintConfigV2,
+  PremintConfigVersion,
+  PremintConfigWithVersion,
+  TokenConfigWithVersion,
+  TokenCreationConfigV1,
+  TokenCreationConfigV2,
+  TokenCreationConfigV3,
+} from "./types";
 
 const premintTypedDataDomain = ({
   chainId,
@@ -13,7 +34,7 @@ const premintTypedDataDomain = ({
   creator1155Contract: verifyingContract,
 }: {
   chainId: number;
-  version: "1" | "2";
+  version: PremintConfigVersion;
   creator1155Contract: Address;
 }): TypedDataDomain => ({
   chainId,
@@ -47,6 +68,68 @@ const premintV1TypedDataType = {
   ],
 } as const satisfies TypedData;
 
+const encodeTokenConfigV1 = (config: TokenCreationConfigV1) => {
+  const abiItem = getAbiItem({
+    abi: iPremintDefinitionsABI,
+    name: "tokenConfigV1Definition",
+  });
+
+  return encodeAbiParameters(abiItem.inputs, [config]);
+};
+
+const encodeTokenConfigV2 = (config: TokenCreationConfigV2) => {
+  const abiItem = getAbiItem({
+    abi: iPremintDefinitionsABI,
+    name: "tokenConfigV2Definition",
+  });
+
+  return encodeAbiParameters(abiItem.inputs, [config]);
+};
+
+const encodeTokenConfigV3 = (config: TokenCreationConfigV3) => {
+  const abiItem = getAbiItem({
+    abi: iPremintDefinitionsABI,
+    name: "tokenConfigV3Definition",
+  });
+
+  return encodeAbiParameters(abiItem.inputs, [config]);
+};
+
+const encodeTokenConfig = <T extends PremintConfigVersion>({
+  tokenConfig,
+  premintConfigVersion,
+}: TokenConfigWithVersion<T>): Hex => {
+  if (premintConfigVersion === PremintConfigVersion.V1) {
+    return encodeTokenConfigV1(tokenConfig as TokenCreationConfigV1);
+  }
+  if (premintConfigVersion === PremintConfigVersion.V2) {
+    return encodeTokenConfigV2(tokenConfig as TokenCreationConfigV2);
+  }
+  if (premintConfigVersion === PremintConfigVersion.V3) {
+    return encodeTokenConfigV3(tokenConfig as TokenCreationConfigV3);
+  }
+
+  throw new Error("Invalid PremintConfigVersion: " + premintConfigVersion);
+};
+
+export const encodePremintConfig = <T extends PremintConfigVersion>({
+  premintConfig,
+  premintConfigVersion,
+}: PremintConfigWithVersion<T>): PremintConfigEncoded => {
+  const encodedTokenConfig = encodeTokenConfig({
+    premintConfigVersion,
+    tokenConfig: premintConfig.tokenConfig,
+  });
+
+  return {
+    deleted: premintConfig.deleted,
+    uid: premintConfig.uid,
+    version: premintConfig.version,
+    premintConfigVersion: keccak256(toHex(premintConfigVersion)),
+    tokenConfig: encodedTokenConfig,
+  };
+};
+
 /**
  * Builds a typed data definition for a PremintConfigV1 to be signed
  * @returns
@@ -58,9 +141,7 @@ export const premintV1TypedDataDefinition = ({
 }: {
   chainId: number;
   creator1155Contract: Address;
-  message: TypedDataToPrimitiveTypes<
-    typeof premintV1TypedDataType
-  >["CreatorAttribution"];
+  message: PremintConfigV1;
 }): TypedDataDefinition<
   typeof premintV1TypedDataType,
   "CreatorAttribution"
@@ -69,7 +150,7 @@ export const premintV1TypedDataDefinition = ({
   primaryType: "CreatorAttribution",
   domain: premintTypedDataDomain({
     chainId,
-    version: "1",
+    version: PremintConfigVersion.V1,
     creator1155Contract,
   }),
   message,
@@ -110,9 +191,7 @@ export const premintV2TypedDataDefinition = ({
 }: {
   chainId: number;
   creator1155Contract: Address;
-  message: TypedDataToPrimitiveTypes<
-    typeof premintV2TypedDataType
-  >["CreatorAttribution"];
+  message: PremintConfigV2;
 }): TypedDataDefinition<
   typeof premintV2TypedDataType,
   "CreatorAttribution"
@@ -121,12 +200,49 @@ export const premintV2TypedDataDefinition = ({
   primaryType: "CreatorAttribution",
   domain: premintTypedDataDomain({
     chainId,
-    version: "2",
+    version: PremintConfigVersion.V2,
     creator1155Contract,
   }),
   message,
 });
 
+export type PremintTypeDataDefinitionParams<T extends PremintConfigVersion> = {
+  verifyingContract: Address;
+  chainId: number;
+} & PremintConfigWithVersion<T>;
+
+/**
+ * Creates a typed data definition for a premint config.  Works for all versions of the premint config by specifying the premintConfigVersion.
+ *
+ * @param params.verifyingContract the address of the 1155 contract
+ * @param params.chainId the chain id the premint is signed for
+ * @param params.premintConfigVersion the version of the premint config
+ * @param params.premintConfig the premint config
+ * @returns
+ */
+export const premintTypedDataDefinition = <T extends PremintConfigVersion>({
+  verifyingContract,
+  chainId,
+  premintConfigVersion: version,
+  premintConfig,
+}: PremintTypeDataDefinitionParams<T>): TypedDataDefinition => {
+  if (version === PremintConfigVersion.V1)
+    return premintV1TypedDataDefinition({
+      chainId,
+      creator1155Contract: verifyingContract,
+      message: premintConfig as PremintConfigV1,
+    });
+  if (version === PremintConfigVersion.V2) {
+    return premintV2TypedDataDefinition({
+      chainId,
+      creator1155Contract: verifyingContract,
+      message: premintConfig as PremintConfigV2,
+    });
+  }
+
+  throw new Error(`Invalid version ${version}`);
+};
+
 const permitSafeTransferTypedDataType = {
   PermitSafeTransfer: [
     { name: "owner", type: "address" },

package/src/types.ts

@@ -0,0 +1,107 @@
+import { ExtractAbiFunction, AbiParametersToPrimitiveTypes } from "abitype";
+
+import {
+  zoraCreator1155PremintExecutorImplABI,
+  iPremintDefinitionsABI,
+} from "./generated/wagmi";
+
+export enum PremintConfigVersion {
+  V1 = "1",
+  V2 = "2",
+  V3 = "3",
+}
+
+export type ContractCreationConfig = AbiParametersToPrimitiveTypes<
+  ExtractAbiFunction<
+    typeof zoraCreator1155PremintExecutorImplABI,
+    "premint"
+  >["inputs"]
+>[0];
+
+export type TokenCreationConfigV1 = AbiParametersToPrimitiveTypes<
+  ExtractAbiFunction<
+    typeof iPremintDefinitionsABI,
+    "tokenConfigV1Definition"
+  >["inputs"]
+>[0];
+
+export type TokenCreationConfigV2 = AbiParametersToPrimitiveTypes<
+  ExtractAbiFunction<
+    typeof iPremintDefinitionsABI,
+    "tokenConfigV2Definition"
+  >["inputs"]
+>[0];
+
+export type TokenCreationConfigV3 = AbiParametersToPrimitiveTypes<
+  ExtractAbiFunction<
+    typeof iPremintDefinitionsABI,
+    "tokenConfigV3Definition"
+  >["inputs"]
+>[0];
+
+export type PremintConfigEncoded = AbiParametersToPrimitiveTypes<
+  ExtractAbiFunction<
+    typeof zoraCreator1155PremintExecutorImplABI,
+    "premint"
+  >["inputs"]
+>[2];
+
+type PremintConfigCommon = Pick<
+  PremintConfigEncoded,
+  "deleted" | "uid" | "version"
+>;
+
+export type PremintConfigV1 = PremintConfigCommon & {
+  tokenConfig: TokenCreationConfigV1;
+};
+export type PremintConfigV2 = PremintConfigCommon & {
+  tokenConfig: TokenCreationConfigV2;
+};
+export type PremintConfigV3 = PremintConfigCommon & {
+  tokenConfig: TokenCreationConfigV3;
+};
+
+export type PremintMintArguments = AbiParametersToPrimitiveTypes<
+  ExtractAbiFunction<
+    typeof zoraCreator1155PremintExecutorImplABI,
+    "premint"
+  >["inputs"]
+>[5];
+
+export type PremintConfigForVersion<T extends PremintConfigVersion> =
+  T extends PremintConfigVersion.V1
+    ? PremintConfigV1
+    : T extends PremintConfigVersion.V2
+      ? PremintConfigV2
+      : PremintConfigV3;
+
+export type PremintConfigWithVersion<T extends PremintConfigVersion> = {
+  premintConfig: PremintConfigForVersion<T>;
+  premintConfigVersion: T;
+};
+export type PremintConfigAndVersion =
+  | PremintConfigWithVersion<PremintConfigVersion.V1>
+  | PremintConfigWithVersion<PremintConfigVersion.V2>
+  | PremintConfigWithVersion<PremintConfigVersion.V3>;
+
+export type PremintConfig = PremintConfigV1 | PremintConfigV2;
+
+export type TokenCreationConfig =
+  | TokenCreationConfigV1
+  | TokenCreationConfigV2
+  | TokenCreationConfigV3;
+
+export type PremintConfigForTokenCreationConfig<T extends TokenCreationConfig> =
+  T extends TokenCreationConfigV1
+    ? PremintConfigV1
+    : T extends TokenCreationConfigV2
+      ? PremintConfigV2
+      : PremintConfigV3;
+
+export type TokenConfigForVersion<T extends PremintConfigVersion> =
+  PremintConfigForVersion<T>["tokenConfig"];
+
+export type TokenConfigWithVersion<T extends PremintConfigVersion> = {
+  tokenConfig: TokenConfigForVersion<T>;
+  premintConfigVersion: T;
+};