import { Formio as ImportedFormio } from "@formio/js";
import { createContext, type PropsWithChildren, useEffect, useState } from "react";

type BaseConfigurationArgs = {
  baseUrl?: string;
  projectUrl?: string;
  Formio?: typeof ImportedFormio;
};

const useBaseConfiguration = ({ baseUrl, projectUrl, Formio }: BaseConfigurationArgs) => {
  if (!Formio) {
    if (baseUrl) {
      ImportedFormio.setBaseUrl(baseUrl);
    }
    if (projectUrl) {
      ImportedFormio.setProjectUrl(projectUrl);
    }
    return {
      Formio: ImportedFormio,
      baseUrl: ImportedFormio.baseUrl,
      projectUrl: ImportedFormio.projectUrl
    };
  }

  if (baseUrl) {
    Formio.setBaseUrl(baseUrl);
  }
  if (projectUrl) {
    Formio.setProjectUrl(projectUrl);
  }

  return {
    Formio,
    baseUrl: Formio.baseUrl,
    projectUrl: Formio.projectUrl
  };
};

const useAuthentication = ({ Formio }: { Formio: typeof ImportedFormio }) => {
  const [token, setToken] = useState(Formio.getToken() || null);
  const [isAuthenticated, setIsAuthenticated] = useState(!!token);

  useEffect(() => {
    const handleUserEvent = async (user: unknown) => {
      if (user) {
        setToken(Formio.getToken());
        setIsAuthenticated(true);
      } else if (isAuthenticated) {
        await Formio.logout();
        setToken(null);
        setIsAuthenticated(false);
      }
    };

    const handleStaleToken = async () => {
      if (isAuthenticated) {
        const user = await Formio.currentUser();
        if (!user) {
          setToken(null);
          setIsAuthenticated(false);
        }
      }
    };

    Formio.events.on("formio.user", handleUserEvent);
    handleStaleToken();

    return () => {
      Formio.events.off("formio.user", handleUserEvent);
    };
  }, [isAuthenticated, Formio]);

  const logout = async () => {
    await Formio.logout();
    setToken(null);
    setIsAuthenticated(false);
  };

  return { token, setToken, isAuthenticated, logout };
};

export const FormioContext = createContext<(ReturnType<typeof useBaseConfiguration> & ReturnType<typeof useAuthentication>) | null>(null);

export function FormioProvider({ children, baseUrl, projectUrl, Formio }: PropsWithChildren<BaseConfigurationArgs>) {
  const baseConfig = useBaseConfiguration({ baseUrl, projectUrl, Formio });
  const auth = useAuthentication({ Formio: baseConfig.Formio });
  const formio = { ...baseConfig, ...auth };
  return <FormioContext.Provider value={formio}>{children}</FormioContext.Provider>;
}