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

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 && (
        <MessageDialog title={title} subtitle={subtitle} onClose={handleClose}></MessageDialog>
    );

	const getDialogComponent = () => {
		return MessageDialogComponent;
	};

    return { message, MessageDialogComponent };
}

export default useMessage;