zeppelin-web-angular/projects/zeppelin-react/src/components/paragraph/ReactErrorBoundary.tsx

/*
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *     http://www.apache.org/licenses/LICENSE-2.0
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import { Component, ErrorInfo, ReactNode } from 'react';

interface Props {
  onError?: (error: unknown) => void;
  children: ReactNode;
}

interface State {
  hasError: boolean;
}

export class ReactErrorBoundary extends Component<Props, State> {
  state: State = { hasError: false };

  static getDerivedStateFromError(): State {
    return { hasError: true };
  }

  componentDidCatch(error: Error, _info: ErrorInfo): void {
    if (typeof this.props.onError === 'function') {
      try {
        this.props.onError(error);
      } catch {
        /* swallow */
      }
    }
  }

  render(): ReactNode {
    if (this.state.hasError) {
      return null;
    }
    return this.props.children;
  }
}