import { Button } from "@material-ui/core";
import React from "react";

interface Props {
  setEncPublicKey: (key: Uint8Array) => void;
  providerRequest:
    | ((request: { method: string; params?: Array<any> }) => Promise<any>)
    | undefined;
  address: string | undefined;
}

export default function GetEncryptionPublicKey({
  setEncPublicKey,
  providerRequest,
  address,
}: Props) {
  const requestPublicKey = () => {
    if (!providerRequest) return;
    if (!address) return;

    console.log("Getting Encryption Public Key from Wallet");
    providerRequest({
      method: "eth_getEncryptionPublicKey",
      params: [address],
    })
      .then((key: string | undefined) => {
        console.log("Encryption Public key:", key);

        if (typeof key !== "string") {
          console.error("Could not get encryption key");
          return;
        }

        setEncPublicKey(Buffer.from(key, "base64"));
      })
      .catch((error) => {
        if (error.code === 4001) {
          // EIP-1193 userRejectedRequest error
          console.log("We can't encrypt anything without the key.");
        } else {
          console.error(error);
        }
      });
  };

  return (
    <Button
      variant="contained"
      color="primary"
      onClick={requestPublicKey}
      disabled={!providerRequest || !address}
    >
      Get Encryption Public Key from Wallet
    </Button>
  );
}