Skip to content

Commit

Permalink
fix(react): Make fallback render types more accurate (#7198)
Browse files Browse the repository at this point in the history
  • Loading branch information
AbhiPrasad committed Feb 16, 2023
1 parent 5359ba9 commit 79babe9
Showing 1 changed file with 21 additions and 10 deletions.
31 changes: 21 additions & 10 deletions packages/react/src/errorboundary.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -13,8 +13,8 @@ export const UNKNOWN_COMPONENT = 'unknown';

export type FallbackRender = (errorData: {
error: Error;
componentStack: string | null;
eventId: string | null;
componentStack: string;
eventId: string;
resetError(): void;
}) => React.ReactElement;

Expand Down Expand Up @@ -48,11 +48,17 @@ export type ErrorBoundaryProps = {
beforeCapture?(scope: Scope, error: Error | null, componentStack: string | null): void;
};

type ErrorBoundaryState = {
componentStack: React.ErrorInfo['componentStack'] | null;
error: Error | null;
eventId: string | null;
};
type ErrorBoundaryState =
| {
componentStack: null;
error: null;
eventId: null;
}
| {
componentStack: React.ErrorInfo['componentStack'];
error: Error;
eventId: string;
};

const INITIAL_STATE = {
componentStack: null,
Expand Down Expand Up @@ -133,12 +139,17 @@ class ErrorBoundary extends React.Component<ErrorBoundaryProps, ErrorBoundarySta

public render(): React.ReactNode {
const { fallback, children } = this.props;
const { error, componentStack, eventId } = this.state;
const state = this.state;

if (error) {
if (state.error) {
let element: React.ReactElement | undefined = undefined;
if (typeof fallback === 'function') {
element = fallback({ error, componentStack, resetError: this.resetErrorBoundary, eventId });
element = fallback({
error: state.error,
componentStack: state.componentStack,
resetError: this.resetErrorBoundary,
eventId: state.eventId,
});
} else {
element = fallback;
}
Expand Down

0 comments on commit 79babe9

Please sign in to comment.