frontend/src/hooks/UseMessage.tsx


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43

import React, { useState } from "react";
import MessageDialog from "@components/MessageDialog";

const useMessage = () => {
  const [isOpen, setIsOpen] = useState(false);

  const [title, setTitle] = useState<string>("");
  const [subtitle, setSubtitle] = useState<string>("");
  const [resolvePromise, setResolvePromise] = useState<(() => void) | null>(
    null
  );

  const message = (title: string, subtitle: string) => {
    setIsOpen(true);
    setTitle(title);
    setSubtitle(subtitle);
    return new Promise(resolve => {
      setResolvePromise(() => resolve);
    });
  };

  const handleClose = () => {
    setIsOpen(false);
    if (resolvePromise) {
      resolvePromise();
      setResolvePromise(null);
    }
  };

  const MessageDialogComponent = isOpen && (
    <div className="dialog-container">
      <MessageDialog
        title={title}
        subtitle={subtitle}
        onClose={handleClose}
      ></MessageDialog>
    </div>
  );

  return { message, MessageDialogComponent };
};

export default useMessage;