@puzzlehq/sdk/src/hooks/useRequestSignature.ts

Your portal to privacy

import {
  GenericRequest,
  requestSignature as _requestSignature,
  SignatureRequest,
  SignatureResponse,
  log_sdk,
  SdkError,
} from '@puzzlehq/sdk-core';
import { aleoAddressRegex } from '@puzzlehq/types';
import { useInjectedRequest } from './utils/useRequest.js';
import { useIsConnected } from '../provider/PuzzleWalletProvider.js';

export const useRequestSignature = ({
  message,
  address,
  network,
}: SignatureRequest) => {
  const isConnected = useIsConnected();

  const req: GenericRequest = {
    method: 'requestSignature',
    params: {
      message,
      address: aleoAddressRegex.test(address ?? '') ? address : undefined,
      network,
    } as SignatureRequest,
  };

  const {
    request,
    data,
    error: _error,
    loading,
  } = useInjectedRequest<SignatureResponse | undefined>(
    req,
    async (req) => {
      if (!isConnected) throw new Error(SdkError.NotConnected);
      const response = await _requestSignature(req.params as SignatureRequest);
      return response;
    },
  );

  const error: string | undefined = typeof _error === 'string' ? _error : _error instanceof Error ? _error.message : undefined;
  const response: SignatureResponse | undefined = data;

  const requestSignature = (signatureRequestOverride?: SignatureRequest) => {
    if (signatureRequestOverride && isConnected && !loading) {
      log_sdk(
        'useRequestSignature requesting with override...',
        signatureRequestOverride,
      );
      return request({
        method: 'requestSignature',
        params: {
          ...signatureRequestOverride,
        },
      });
    } else if (isConnected && !loading) {
      log_sdk('useRequestSignature requesting...', [message, address]);
      return request();
    }
  };

  return { requestSignature, response, loading, error };
};