docs: add llms copy button (#1146)

Author: Timeless0911

website/theme/index.tsx

@@ -1,4 +1,7 @@
-import { Layout as BaseLayout } from '@rspress/core/theme';
+import {
+  getCustomMDXComponent as BaseGetCustomMDXComponent,
+  Layout as BaseLayout,
+} from '@rspress/core/theme';
 import {
   Search as PluginAlgoliaSearch,
   ZH_LOCALES,
@@ -7,6 +10,31 @@ import { NavIcon } from '@rstack-dev/doc-ui/nav-icon';
 import { HomeLayout } from './pages';
 import './index.scss';
 import { useLang } from '@rspress/core/runtime';
+import {
+  LlmsContainer,
+  LlmsCopyButton,
+  LlmsViewOptions,
+} from '@rspress/plugin-llms/runtime';
+
+export function getCustomMDXComponent() {
+  const { h1: H1, ...mdxComponents } = BaseGetCustomMDXComponent();
+
+  const MyH1 = ({ ...props }) => {
+    return (
+      <>
+        <H1 {...props} />
+        <LlmsContainer>
+          <LlmsCopyButton />
+          <LlmsViewOptions />
+        </LlmsContainer>
+      </>
+    );
+  };
+  return {
+    ...mdxComponents,
+    h1: MyH1,
+  };
+}
 
 const Layout = () => {
   return <BaseLayout beforeNavTitle={<NavIcon />} />;