// Copyright (c) 2020-2021 Drew Lemmy
// This file is part of TenebraWeb 2 under AGPL-3.0.
// Full details: https://github.com/tmpim/TenebraWeb2/blob/master/LICENSE.txt
import React, { FC, useContext } from "react";
import { message } from "antd";

import i18n from "@utils/i18n";

import { AuthContext, PromptAuthFn } from "./AuthContext";

interface Props {
  encrypt?: boolean;
  onAuthed?: () => void;
  children: React.ReactNode;
}

export const AuthorisedAction: FC<Props> = ({ encrypt, onAuthed, children }) => {
  const promptAuth = useContext(AuthContext);

  // This is used to pass the 'onClick' prop down to the child. The child MUST
  // support the onClick prop.
  // NOTE: If the child is a custom component, make sure it passes `...props`
  //       down to its child.
  const child = React.Children.only(children) as React.ReactElement;

  // Wrap the single child element and override onClick
  return React.cloneElement(child, { onClick: (e: MouseEvent) => {
    e.preventDefault();
    promptAuth?.(encrypt, onAuthed);
  }});
};

export const useAuth = (): PromptAuthFn =>
  useContext(AuthContext) || (() =>
    message.error(i18n.t("masterPassword.earlyAuthError")));
