39948-vm/frontend/src/pages/_app.tsx

import React from 'react';
import type { AppProps } from 'next/app';
import type { ReactElement, ReactNode } from 'react';
import type { NextPage } from 'next';
import Head from 'next/head';
import { store } from '../stores/store';
import { Provider } from 'react-redux';
import '../css/main.css';
import axios from 'axios';
import { baseURLApi } from '../config';
import { useRouter } from 'next/router';
import ErrorBoundary from '../components/ErrorBoundary';
import DevModeBadge from '../components/DevModeBadge';
import 'intro.js/introjs.css';
import { appWithTranslation } from 'next-i18next';
import '../i18n';
import IntroGuide from '../components/IntroGuide';
import {
  appSteps,
  loginSteps,
  usersSteps,
  rolesSteps,
} from '../stores/introSteps';

// Initialize axios
axios.defaults.baseURL = process.env.NEXT_PUBLIC_BACK_API
  ? process.env.NEXT_PUBLIC_BACK_API
  : baseURLApi;

axios.defaults.headers.common['Content-Type'] = 'application/json';

// Set up axios request interceptor synchronously to ensure token is always attached
axios.interceptors.request.use(
  (config) => {
    if (typeof window !== 'undefined') {
      const token =
        sessionStorage.getItem('token') || localStorage.getItem('token');
      if (token) {
        config.headers.Authorization = `Bearer ${token}`;
      } else {
        delete config.headers.Authorization;
      }
    }
    return config;
  },
  (error) => Promise.reject(error),
);

// Set up axios response interceptor to handle 401 errors and redirect to login
axios.interceptors.response.use(
  (response) => response,
  (error) => {
    if (typeof window !== 'undefined') {
      const status = error?.response?.status;
      const requestUrl = `${error?.config?.url || ''}`;
      const isLoginRequest =
        requestUrl.includes('/auth/signin/local') ||
        requestUrl.includes('auth/signin/local');

      if (status === 401 && !isLoginRequest) {
        // Clear stored tokens
        sessionStorage.removeItem('token');
        sessionStorage.removeItem('user');
        localStorage.removeItem('token');
        localStorage.removeItem('user');
        delete axios.defaults.headers.common['Authorization'];

        // Redirect to login if not already there
        if (!window.location.pathname.includes('/login')) {
          window.location.href = '/login';
        }
      }
    }
    return Promise.reject(error);
  },
);

export type NextPageWithLayout<P = Record<string, unknown>, IP = P> = NextPage<
  P,
  IP
> & {
  getLayout?: (page: ReactElement) => ReactNode;
};

type AppPropsWithLayout = AppProps & {
  Component: NextPageWithLayout;
};

function MyApp({ Component, pageProps }: AppPropsWithLayout) {
  // Use the layout defined at the page level, if available
  const getLayout = Component.getLayout || ((page) => page);
  const router = useRouter();
  const [stepsEnabled, setStepsEnabled] = React.useState(false);
  const [stepName, setStepName] = React.useState('');
  const [steps, setSteps] = React.useState([]);

  // TODO: Remove this code in future releases
  React.useEffect(() => {
    const trustedOrigins = new Set<string>([window.location.origin]);
    if (document.referrer) {
      try {
        trustedOrigins.add(new URL(document.referrer).origin);
      } catch (error) {
        console.warn(
          '[postMessage] Failed to parse parent origin from referrer',
          error,
        );
      }
    }

    const isTrustedOrigin = (origin: string) => trustedOrigins.has(origin);

    const handleMessage = async (event: MessageEvent) => {
      if (!isTrustedOrigin(event.origin)) {
        console.warn(
          '[postMessage] Blocked message from untrusted origin',
          event.origin,
        );
        return;
      }

      if (event.data === 'getLocation') {
        event.source?.postMessage(
          { iframeLocation: window.location.pathname },
          event.origin,
        );
        return;
      }

      if (event.data === 'getAuthToken') {
        const token =
          sessionStorage.getItem('token') || localStorage.getItem('token');
        const user =
          sessionStorage.getItem('user') || localStorage.getItem('user');
        event.source?.postMessage(
          { iframeAuthToken: token, iframeAuthUser: user },
          event.origin,
        );
        return;
      }

      if (event.data === 'getScreenshot') {
        try {
          const html2canvas = (await import('html2canvas')).default;
          const canvas = await html2canvas(document.body, { useCORS: true });
          const url = canvas.toDataURL('image/jpeg', 0.8);
          event.source?.postMessage({ iframeScreenshot: url }, event.origin);
        } catch (e) {
          console.error('html2canvas failed', e);
          event.source?.postMessage({ iframeScreenshot: null }, event.origin);
        }
      }
    };

    window.addEventListener('message', handleMessage);
    return () => window.removeEventListener('message', handleMessage);
  }, []);

  React.useEffect(() => {
    // Tour is disabled by default in generated projects.
    return;
    const isCompleted = (stepKey: string) => {
      return localStorage.getItem(`completed_${stepKey}`) === 'true';
    };
    if (router.pathname === '/login' && !isCompleted('loginSteps')) {
      setSteps(loginSteps);
      setStepName('loginSteps');
      setStepsEnabled(true);
    } else if (router.pathname === '/dashboard' && !isCompleted('appSteps')) {
      setTimeout(() => {
        setSteps(appSteps);
        setStepName('appSteps');
        setStepsEnabled(true);
      }, 1000);
    } else if (
      router.pathname === '/users/users-list' &&
      !isCompleted('usersSteps')
    ) {
      setTimeout(() => {
        setSteps(usersSteps);
        setStepName('usersSteps');
        setStepsEnabled(true);
      }, 1000);
    } else if (
      router.pathname === '/roles/roles-list' &&
      !isCompleted('rolesSteps')
    ) {
      setTimeout(() => {
        setSteps(rolesSteps);
        setStepName('rolesSteps');
        setStepsEnabled(true);
      }, 1000);
    } else {
      setSteps([]);
      setStepsEnabled(false);
    }
  }, [router.pathname]);

  const handleExit = () => {
    setStepsEnabled(false);
  };

  const title = 'Shimahara Visual';
  const description =
    'Platform to build, preview (stage), and publish offline-ready interactive tours on project subdomains.';
  const url = 'https://flatlogic.com/';
  const image =
    'https://project-screens.s3.amazonaws.com/screenshots/39215/app-hero-20260316-121031.png';
  const imageWidth = '1920';
  const imageHeight = '960';

  return (
    <Provider store={store}>
      {getLayout(
        <>
          <Head>
            <meta name='description' content={description} />

            <meta property='og:url' content={url} />
            <meta property='og:site_name' content='https://flatlogic.com/' />
            <meta property='og:title' content={title} />
            <meta property='og:description' content={description} />
            <meta property='og:image' content={image} />
            <meta property='og:image:type' content='image/png' />
            <meta property='og:image:width' content={imageWidth} />
            <meta property='og:image:height' content={imageHeight} />

            <meta property='twitter:card' content='summary_large_image' />
            <meta property='twitter:title' content={title} />
            <meta property='twitter:description' content={description} />
            <meta property='twitter:image:src' content={image} />
            <meta property='twitter:image:width' content={imageWidth} />
            <meta property='twitter:image:height' content={imageHeight} />

            <link rel='icon' href='/favicon.svg' />
          </Head>

          <ErrorBoundary>
            <Component {...pageProps} />
          </ErrorBoundary>
          <IntroGuide
            steps={steps}
            stepsName={stepName}
            stepsEnabled={stepsEnabled}
            onExit={handleExit}
          />
          {(process.env.NODE_ENV === 'development' ||
            process.env.NODE_ENV === 'dev_stage') && <DevModeBadge />}
        </>,
      )}
    </Provider>
  );
}

export default appWithTranslation(MyApp);