aboutsummaryrefslogtreecommitdiff
path: root/src/components/ErrorBoundary.tsx
blob: ed565c562076622406b09ac5d57d14b12efd67b1 (plain)
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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
/*
 * Vencord, a modification for Discord's desktop app
 * Copyright (c) 2022 Vendicated and contributors
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
*/

import Logger from "../utils/logger";
import { Margins, React } from "../webpack/common";
import { ErrorCard } from "./ErrorCard";

interface Props {
    fallback?: React.ComponentType<React.PropsWithChildren<{ error: any; message: string; stack: string; }>>;
    onError?(error: Error, errorInfo: React.ErrorInfo): void;
    message?: string;
}

const color = "#e78284";

const logger = new Logger("React ErrorBoundary", color);

const NO_ERROR = {};

export default class ErrorBoundary extends React.Component<React.PropsWithChildren<Props>> {
    static wrap<T = any>(Component: React.ComponentType<T>): (props: T) => React.ReactElement {
        return props => (
            <ErrorBoundary>
                <Component {...props as any/* I hate react typings ??? */} />
            </ErrorBoundary>
        );
    }

    state = {
        error: NO_ERROR as any,
        stack: "",
        message: ""
    };

    static getDerivedStateFromError(error: any) {
        let stack = error?.stack ?? "";
        let message = error?.message || String(error);

        if (error instanceof Error && stack) {
            const eolIdx = stack.indexOf("\n");
            if (eolIdx !== -1) {
                message = stack.slice(0, eolIdx);
                stack = stack.slice(eolIdx + 1).replace(/https:\/\/\S+\/assets\//g, "");
            }
        }

        return { error, stack, message };
    }

    componentDidCatch(error: Error, errorInfo: React.ErrorInfo) {
        this.props.onError?.(error, errorInfo);
        logger.error("A component threw an Error\n", error);
        logger.error("Component Stack", errorInfo.componentStack);
    }

    render() {
        if (this.state.error === NO_ERROR) return this.props.children;

        if (this.props.fallback)
            return <this.props.fallback
                children={this.props.children}
                {...this.state}
            />;

        const msg = this.props.message || "An error occurred while rendering this Component. More info can be found below and in your console.";

        return (
            <ErrorCard style={{
                overflow: "hidden",
            }}>
                <h1>Oh no!</h1>
                <p>{msg}</p>
                <code>
                    {this.state.message}
                    {!!this.state.stack && (
                        <pre className={Margins.marginTop8}>
                            {this.state.stack}
                        </pre>
                    )}
                </code>
            </ErrorCard>
        );
    }
}