import { createBrowserRouter, Navigate } from "react-router-dom";
import { lazy, Suspense } from "react";
import TermloansLoaderScreen from "@/components/TermloansLoaderScreen";
import ProtectedRoute from "@/components/ProtectedRoute";
import AnimatedPage from "@/Layout/AnimatedPage";
import AuthOnlyRoute from "@/components/AuthOnlyRoute";

// Lazy loaded components - grouped by feature for better chunking
// Authentication pages
const Login = lazy(() => import("@/pages/NewPages/Login"));
const VerificationCode = lazy(() => import("@/pages/NewPages/VerificationCode"));
const ForgotPassword = lazy(() => import("@/pages/NewPages/ForgotPassword"));
const RecoverPassword = lazy(() => import("@/pages/NewPages/RecoverPassword"));
const EmailVerificationAccount = lazy(() => import("@/pages/NewPages/EmailVerificationAccount"));
const GoogleCallbackPage = lazy(() => import("@/pages/NewPages/GoogleCallback"));

// Registration and onboarding pages
const CreateAccountPage = lazy(() => import("@/pages/NewPages/Forms/CreateAccount"));
const ApplyAccountPage = lazy(() => import("@/pages/NewPages/Forms/ApplyAccount"));
// const QuizAccountPage = lazy(() => import("@/pages/NewPages/Forms/QuizAccount"));
const NexusAccountPage = lazy(() => import("@/pages/NewPages/Forms/NexusAccount"));
const RegistrationPage = lazy(() => import("@/pages/NewPages/Forms/Typeform"));
const ApplyTypeFormPage = lazy(() => import("@/pages/NewPages/Forms/ApplyTypeform"));
// const QuizTypeFormPage = lazy(() => import("@/pages/NewPages/Forms/QuizTypeform"));
const NexusTypeFormPage = lazy(() => import("@/pages/NewPages/Forms/NexusTypeform"));
const MagicLinkTypeform = lazy(() => import("@/pages/NewPages/Forms/MagicLinkTypeform"));

// Banking integration pages
const PlaidPage = lazy(() => import("@/pages/NewPages/Plaid"));
const PlaidProcessingPage = lazy(() => import("@/pages/NewPages/PlaidProcessing"));

// Core application pages
const RequestInProgressPage = lazy(() => import("@/pages/NewPages/RequestInProgress"));

// Error and utility pages
const NotFoundPage = lazy(() => import("@/pages/NewPages/NotFound"));
const ErrorBoundaryPage = lazy(() => import("@/pages/NewPages/ErrorBoundary"));

// Legal pages
const PrivacyPolicy = lazy(() => import("@/pages/Legal/PrivacyPolicy"));
const TermsOfService = lazy(() => import("@/pages/Legal/TermsOfService"));

// Wrapper component for lazy loaded pages with Suspense
const LazyPage = ({ children }: { children: React.ReactNode }) => (
  <Suspense fallback={<TermloansLoaderScreen />}>
    <AnimatedPage>
      {children}
    </AnimatedPage>
  </Suspense>
);

const publicRoutes = [
  {
    path: "",
    element: <Navigate to="/login" replace />,
  },
  {
    path: "/home",
    element: <Navigate to="/request-in-progress" replace />,
  },
  {
    path: "/login",
    element: <LazyPage><Login /></LazyPage>,
  },
  {
    path: "/recover-password",
    element: <LazyPage><ForgotPassword /></LazyPage>,
  },
  {
    path: "/register",
    element: <LazyPage><CreateAccountPage /></LazyPage>,
  },
  {
    path: "/apply",
    element: <LazyPage><ApplyAccountPage /></LazyPage>,
  },
  // {
  //   path: "/quiz",
  //   element: <LazyPage><QuizAccountPage /></LazyPage>,
  // },
  {
    path: "/quiz",
    element: <LazyPage><NexusAccountPage /></LazyPage>,
  },
  {
    path: "/reset-password/verify",
    element: <LazyPage><VerificationCode /></LazyPage>,
  },
  {
    path: "/2fa/verify",
    element: <LazyPage><VerificationCode /></LazyPage>,
  },
  {
    path: "/reset-password/change-password",
    element: <LazyPage><RecoverPassword /></LazyPage>,
  },
  {
    path: "/verification-account",
    element: <LazyPage><EmailVerificationAccount /></LazyPage>,
  },
  {
    path: "/registration",
    element: <AuthOnlyRoute><LazyPage><RegistrationPage /></LazyPage></AuthOnlyRoute>,
  },
  {
    path: "/apply-registration",
    element: <AuthOnlyRoute><LazyPage><ApplyTypeFormPage /></LazyPage></AuthOnlyRoute>,
  },
  // {
  //   path: "/quiz-registration",
  //   element: <AuthOnlyRoute><LazyPage><QuizTypeFormPage /></LazyPage></AuthOnlyRoute>,
  // },
  {
    path: "/quiz-registration",
    element: <AuthOnlyRoute><LazyPage><NexusTypeFormPage /></LazyPage></AuthOnlyRoute>,
  },
  {
    path: "/connect-plaid",
    element: <AuthOnlyRoute><LazyPage><PlaidPage /></LazyPage></AuthOnlyRoute>,
  },
  {
    path: "plaid-processing",
    element: <ProtectedRoute children={<LazyPage><PlaidProcessingPage /></LazyPage>} />,
  },
  {
    path: "request-in-progress",
    element: <ProtectedRoute children={<LazyPage><RequestInProgressPage /></LazyPage>} />,
  },
  {
    path: "/google/callback",
    element: <LazyPage><GoogleCallbackPage /></LazyPage>,
  },
  {
    path: "/partner-registration",
    element: <LazyPage><MagicLinkTypeform /></LazyPage>,
  },
  // Legal pages (accessible to all users)
  {
    path: "/privacy-policy",
    element: <LazyPage><PrivacyPolicy /></LazyPage>,
  },
  {
    path: "/terms-of-service",
    element: <LazyPage><TermsOfService /></LazyPage>,
  },
];


export const router = createBrowserRouter([
  ...publicRoutes,
  // Error routes
  {
    path: "/error",
    element: <LazyPage><ErrorBoundaryPage /></LazyPage>,
  },
  // 404 catch-all route (must be last)
  {
    path: "*",
    element: <LazyPage><NotFoundPage /></LazyPage>,
  },
]);