/*!
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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 { Box, LocaleProvider } from "@chakra-ui/react";
import { useEffect, type PropsWithChildren } from "react";
import { useTranslation } from "react-i18next";
import { Outlet } from "react-router-dom";

import { usePluginServiceGetPlugins } from "openapi/queries";
import type { ReactAppResponse } from "openapi/requests/types.gen";
import { KeyboardShortcutsModal } from "src/components/KeyboardShortcuts";
import { ReactPlugin } from "src/pages/ReactPlugin";
import { useConfig } from "src/queries/useConfig";
import { DocumentTitleProvider } from "src/utils";

import { Nav } from "./Nav";

export const BaseLayout = ({ children }: PropsWithChildren) => {
  const { i18n } = useTranslation();
  const { data: pluginData } = usePluginServiceGetPlugins();
  const theme = useConfig("theme") as unknown as { icon?: string; icon_dark_mode?: string } | undefined;

  const baseReactPlugins =
    pluginData?.plugins
      .flatMap((plugin) => plugin.react_apps)
      .filter((reactApp: ReactAppResponse) => reactApp.destination === "base") ?? [];

  useEffect(() => {
    const html = document.documentElement;

    const updateHtml = (language: string) => {
      if (language) {
        html.setAttribute("dir", i18n.dir(language));
        html.setAttribute("lang", language);
      }
    };

    i18n.on("languageChanged", updateHtml);

    return () => {
      i18n.off("languageChanged", updateHtml);
    };
  }, [i18n]);

  useEffect(() => {
    const link = document.querySelector<HTMLLinkElement>("link[rel='icon']");

    if (!link) {
      return undefined;
    }

    const defaultFavicon = link.href;
    const darkIcon = theme?.icon_dark_mode;
    const lightIcon = theme?.icon;
    // favicon color theme should follow system color scheme, not the one set in Airflow UI.
    // (tab colors in browsers are based on system color scheme, not the one set in the website)
    const darkModeQuery = globalThis.matchMedia("(prefers-color-scheme: dark)");

    const updateFavicon = () => {
      const customIcon =
        darkModeQuery.matches && typeof darkIcon === "string" && darkIcon.length > 0 ? darkIcon : lightIcon;

      if (typeof customIcon === "string" && customIcon.length > 0) {
        link.href = customIcon;

        const img = new Image();

        img.addEventListener("error", () => {
          link.href = defaultFavicon;
        });
        img.src = customIcon;
      } else {
        link.href = defaultFavicon;
      }
    };

    updateFavicon();
    darkModeQuery.addEventListener("change", updateFavicon);

    return () => {
      darkModeQuery.removeEventListener("change", updateFavicon);
      link.href = defaultFavicon;
    };
  }, [theme?.icon, theme?.icon_dark_mode]);

  return (
    <LocaleProvider locale={i18n.language || "en"}>
      <DocumentTitleProvider>
        <Box display="flex" flexDirection="column" h="100vh">
          <KeyboardShortcutsModal />
          <Nav />
          <Box
            _ltr={{ ml: 16 }}
            _rtl={{ mr: 16 }}
            data-testid="main-content"
            display="flex"
            flex={1}
            flexDirection="column"
            minH={0}
            overflow="auto"
            p={3}
          >
            {baseReactPlugins.map((plugin) => (
              <ReactPlugin key={plugin.name} reactApp={plugin} />
            ))}
            {children ?? <Outlet />}
          </Box>
        </Box>
      </DocumentTitleProvider>
    </LocaleProvider>
  );
};
