stable-diffusion-webui/extensions/sd-webui-lobe-theme/src/pages/Header/Actions.tsx

import { ActionIcon } from '@lobehub/ui';
import { Space } from 'antd';
import { useResponsive } from 'antd-style';
import { Github, LayoutGrid, LucideIcon, Moon, Settings, Sun } from 'lucide-react';
import qs from 'query-string';
import { memo, useCallback, useState } from 'react';
import { useTranslation } from 'react-i18next';
import { shallow } from 'zustand/shallow';

import { Giscus, Setting } from '@/components';
import { selectors, useAppStore } from '@/store';

const CivitaiLogo: LucideIcon | any = ({ size }: any) => (
  <svg fill="currentColor" height={size} viewBox="0 0 16 16" width={size}>
    <path d="M2 4.5L8 1l6 3.5v7L8 15l-6-3.5v-7zm6-1.194L3.976 5.653v4.694L8 12.694l4.024-2.347V5.653L8 3.306zm0 1.589l2.662 1.552v.824H9.25L8 6.54l-1.25.73v1.458L8 9.46l1.25-.73h1.412v.824L8 11.105 5.338 9.553V6.447L8 4.895z" />
  </svg>
);

interface ActionsProps {
  themeMode: 'dark' | 'light';
}

const Actions = memo<ActionsProps>(() => {
  const [isSettingOpen, setIsSettingOpen] = useState(false);
  const [isModalOpen, setIsModalOpen] = useState(false);
  const themeMode = useAppStore(selectors.themeMode, shallow);
  const { mobile } = useResponsive();
  const { t } = useTranslation();

  const handleSetTheme = useCallback(() => {
    const theme = themeMode === 'light' ? 'dark' : 'light';
    const gradioURL = qs.parseUrl(window.location.href);
    gradioURL.query.__theme = theme;
    window.location.replace(qs.stringifyUrl(gradioURL));
  }, [themeMode]);

  return (
    <>
      <Space.Compact>
        {!mobile && (
          <>
            <a href="https://civitai.com/" rel="noreferrer" target="_blank">
              <ActionIcon icon={CivitaiLogo} title="Civitai" />
            </a>
            <a
              href="https://supagruen.github.io/StableDiffusion-CheatSheet/"
              rel="noreferrer"
              target="_blank"
            >
              <ActionIcon icon={LayoutGrid} title="Cheat Sheet" />
            </a>
            <ActionIcon icon={Github} onClick={() => setIsModalOpen(true)} title={t('feedback')} />
          </>
        )}
        <ActionIcon
          icon={themeMode === 'light' ? Sun : Moon}
          onClick={handleSetTheme}
          title={t('switchTheme')}
        />
        <ActionIcon icon={Settings} onClick={() => setIsSettingOpen(true)} title={t('setting')} />
      </Space.Compact>
      <Setting onCancel={() => setIsSettingOpen(false)} open={isSettingOpen} />
      <Giscus onCancel={() => setIsModalOpen(false)} open={isModalOpen} />
    </>
  );
});

export default Actions;