SurfSense/surfsense_web/components/tool-ui/plan/index.tsx

"use client";

import { Component, type ReactNode } from "react";
import { Card, CardContent } from "@/components/ui/card";

export * from "./plan";
export * from "./schema";

// ============================================================================
// Error Boundary
// ============================================================================

interface PlanErrorBoundaryProps {
  children: ReactNode;
  fallback?: ReactNode;
}

interface PlanErrorBoundaryState {
  hasError: boolean;
  error?: Error;
}

export class PlanErrorBoundary extends Component<PlanErrorBoundaryProps, PlanErrorBoundaryState> {
  constructor(props: PlanErrorBoundaryProps) {
    super(props);
    this.state = { hasError: false };
  }

  static getDerivedStateFromError(error: Error): PlanErrorBoundaryState {
    return { hasError: true, error };
  }

  render() {
    if (this.state.hasError) {
      if (this.props.fallback) {
        return this.props.fallback;
      }

      return (
        <Card className="w-full max-w-xl border-destructive/50">
          <CardContent className="pt-6">
            <div className="flex items-center gap-2 text-destructive">
              <span className="text-sm">Failed to render plan</span>
            </div>
          </CardContent>
        </Card>
      );
    }

    return this.props.children;
  }
}