zksync-sso
Version:
ZKsync Smart Sign On SDK
100 lines • 5.66 kB
JavaScript
import { bytesToHex, formatTransaction } from "viem";
import { deployContract, getAddresses, getCallsStatus, getCapabilities, getChainId, prepareAuthorization, sendCalls, sendRawTransaction, showCallsStatus, signAuthorization, signMessage, signTypedData, waitForCallsStatus, writeContract, } from "viem/actions";
import { signTransaction } from "viem/zksync";
import { getTransactionWithPaymasterData } from "../../../paymaster/index.js";
import { SessionErrorType, SessionEventType, validateSessionTransaction } from "../../../utils/session.js";
import { sendEip712Transaction } from "../actions/sendEip712Transaction.js";
import { getSessionState, sessionStateNotify } from "../actions/session.js";
const sessionErrorToSessionEventType = {
[SessionErrorType.SessionInactive]: SessionEventType.Inactive,
[SessionErrorType.SessionExpired]: SessionEventType.Expired,
};
/**
* Helper function to check session state and notify via callback
*/
async function getSessionStateAndNotify(client) {
const { sessionState } = await getSessionState(client, {
account: client.account.address,
sessionConfig: client.sessionConfig,
contracts: client.contracts,
});
if (client.onSessionStateChange) {
sessionStateNotify({
sessionConfig: client.sessionConfig,
sessionState,
onSessionStateChange: client.onSessionStateChange,
sessionNotifyTimeout: client._sessionNotifyTimeout,
});
}
return sessionState;
}
export function zksyncSsoWalletActions(client) {
return {
deployContract: (args) => deployContract(client, args),
getAddresses: () => getAddresses(client),
getChainId: () => getChainId(client),
sendRawTransaction: (args) => sendRawTransaction(client, args),
sendTransaction: async (args) => {
// eslint-disable-next-line @typescript-eslint/no-explicit-any
const unformattedTx = Object.assign({}, args);
if ("eip712Meta" in unformattedTx) {
const eip712Meta = unformattedTx.eip712Meta;
unformattedTx.gasPerPubdata = eip712Meta.gasPerPubdata ? BigInt(eip712Meta.gasPerPubdata) : undefined;
unformattedTx.factoryDeps = eip712Meta.factoryDeps;
unformattedTx.customSignature = eip712Meta.customSignature;
unformattedTx.paymaster = eip712Meta.paymasterParams?.paymaster;
unformattedTx.paymasterInput = eip712Meta.paymasterParams?.paymasterInput ? bytesToHex(new Uint8Array(eip712Meta.paymasterParams?.paymasterInput)) : undefined;
delete unformattedTx.eip712Meta;
}
/* eslint-disable @typescript-eslint/no-unused-vars */
const { chainId: _, ...unformattedTxWithPaymaster } = await getTransactionWithPaymasterData(client.chain.id, client.account.address, unformattedTx, client.paymasterHandler);
const formatters = client.chain?.formatters;
const format = formatters?.transaction?.format || formatTransaction;
const tx = {
...format(unformattedTxWithPaymaster),
type: "eip712",
};
if (client.skipPreTransactionStateValidation !== true) {
// Get current session state and trigger callback if needed
const sessionState = await getSessionStateAndNotify(client);
// Validate transaction against session constraints
const validationResult = validateSessionTransaction({
sessionState,
sessionConfig: client.sessionConfig,
transaction: tx,
});
// Throw error if validation fails
if (validationResult.error) {
// If validation fails due to session issues, notify via callback
if (client.onSessionStateChange && Object.keys(sessionErrorToSessionEventType).includes(validationResult.error.type)) {
client.onSessionStateChange({
type: sessionErrorToSessionEventType[validationResult.error.type],
message: validationResult.error.message,
});
}
throw new Error(`Session validation failed: ${validationResult.error.message} (${validationResult.error.type})`);
}
}
return await sendEip712Transaction(client, tx);
},
signMessage: (args) => signMessage(client, args),
signTransaction: async (args) => {
const { chainId: _, ...unformattedTxWithPaymaster } = await getTransactionWithPaymasterData(client.chain.id, client.account.address, args, client.paymasterHandler);
return signTransaction(client, {
...args,
unformattedTxWithPaymaster,
// eslint-disable-next-line @typescript-eslint/no-explicit-any
});
},
signTypedData: (args) => signTypedData(client, args),
writeContract: (args) => writeContract(client, args),
signAuthorization: (args) => signAuthorization(client, args),
getCallsStatus: (args) => getCallsStatus(client, args),
getCapabilities: (args) => getCapabilities(client, args),
prepareAuthorization: (args) => prepareAuthorization(client, args),
sendCalls: (args) => sendCalls(client, args),
showCallsStatus: (args) => showCallsStatus(client, args),
waitForCallsStatus: (args) => waitForCallsStatus(client, args),
};
}
//# sourceMappingURL=wallet.js.map