From 3742f6cedc0645cb76f4b996012ad13307908288 Mon Sep 17 00:00:00 2001
From: Innei <tukon479@gmail.com>
Date: Fri, 16 Jun 2023 22:52:18 +0800
Subject: [PATCH] feat: to top

Signed-off-by: Innei <tukon479@gmail.com>
---
 src/components/layout/root/Root.tsx           |   5 +
 src/components/ui/fab/BackToTopFAB.tsx        |  28 ++++
 src/components/ui/fab/FABContainer.tsx        | 146 ++++++++++++++++++
 src/components/ui/fab/index.ts                |   2 +
 .../ui/markdown/parsers/mention.tsx           |  21 +--
 .../root/page-scroll-info-provider.tsx        |   3 +-
 6 files changed, 195 insertions(+), 10 deletions(-)
 create mode 100644 src/components/ui/fab/BackToTopFAB.tsx
 create mode 100644 src/components/ui/fab/FABContainer.tsx
 create mode 100644 src/components/ui/fab/index.ts

diff --git a/src/components/layout/root/Root.tsx b/src/components/layout/root/Root.tsx
index ed8b7aefe1..03027e96c8 100644
--- a/src/components/layout/root/Root.tsx
+++ b/src/components/layout/root/Root.tsx
@@ -1,3 +1,5 @@
+import { BackToTopFAB, FABContainer } from '~/components/ui/fab'
+
 import { Content } from '../content/Content'
 import { Footer } from '../footer'
 import { Header } from '../header'
@@ -9,6 +11,9 @@ export const Root: Component = ({ children }) => {
       <Content>{children}</Content>
 
       <Footer />
+      <FABContainer>
+        <BackToTopFAB />
+      </FABContainer>
     </>
   )
 }
diff --git a/src/components/ui/fab/BackToTopFAB.tsx b/src/components/ui/fab/BackToTopFAB.tsx
new file mode 100644
index 0000000000..628e9f8395
--- /dev/null
+++ b/src/components/ui/fab/BackToTopFAB.tsx
@@ -0,0 +1,28 @@
+'use client'
+
+import { useViewport } from '~/atoms'
+import { usePageScrollLocationSelector } from '~/providers/root/page-scroll-info-provider'
+import { springScrollToTop } from '~/utils/spring'
+
+import { FABBase } from './FABContainer'
+
+export const BackToTopFAB = () => {
+  const windowHeight = useViewport((v) => v.h)
+  const shouldShow = usePageScrollLocationSelector(
+    (scrollTop) => {
+      return scrollTop > windowHeight / 5
+    },
+    [windowHeight],
+  )
+
+  return (
+    <FABBase
+      id="to-top"
+      aria-label="Back to top"
+      show={shouldShow}
+      onClick={springScrollToTop}
+    >
+      <i className="icon-[mingcute--arow-to-up-line]" />
+    </FABBase>
+  )
+}
diff --git a/src/components/ui/fab/FABContainer.tsx b/src/components/ui/fab/FABContainer.tsx
new file mode 100644
index 0000000000..91c409b833
--- /dev/null
+++ b/src/components/ui/fab/FABContainer.tsx
@@ -0,0 +1,146 @@
+'use client'
+
+import React, { useEffect, useState } from 'react'
+import type { PropsWithChildren } from 'react'
+
+import { useStateToRef } from '~/hooks/common/use-state-ref'
+import { clsxm } from '~/utils/helper'
+
+export interface FABConfig {
+  id: string
+  icon: JSX.Element
+  onClick: () => void
+}
+
+class FABStatic {
+  private setState: React.Dispatch<React.SetStateAction<FABConfig[]>> | null =
+    null
+  register(setter: any) {
+    this.setState = setter
+  }
+  destroy() {
+    this.setState = null
+  }
+
+  add(fabConfig: FABConfig) {
+    if (!this.setState) return
+
+    const id = fabConfig.id
+
+    this.setState((state) => {
+      if (state.find((config) => config.id === id)) return state
+      return [...state, fabConfig]
+    })
+
+    return () => {
+      this.remove(fabConfig.id)
+    }
+  }
+
+  remove(id: string) {
+    if (!this.setState) return
+    this.setState((state) => {
+      return state.filter((config) => config.id !== id)
+    })
+  }
+}
+
+const fab = new FABStatic()
+
+export const useFAB = (fabConfig: FABConfig) => {
+  useEffect(() => {
+    return fab.add(fabConfig)
+  }, [])
+}
+
+export const FABBase = (
+  props: PropsWithChildren<
+    {
+      id: string
+      show?: boolean
+      children: JSX.Element
+    } & React.DetailedHTMLProps<
+      React.ButtonHTMLAttributes<HTMLButtonElement>,
+      HTMLButtonElement
+    >
+  >,
+) => {
+  const { children, show = true, ...extra } = props
+  const { className, onTransitionEnd, ...rest } = extra
+
+  const [mounted, setMounted] = useState(true)
+  const [appearTransition, setAppearTransition] = useState(false)
+  const getMounted = useStateToRef(mounted)
+  const handleTransitionEnd: React.TransitionEventHandler<HTMLButtonElement> = (
+    e,
+  ) => {
+    onTransitionEnd?.(e)
+
+    !show && setMounted(false)
+  }
+
+  useEffect(() => {
+    if (show && !getMounted.current) {
+      setAppearTransition(true)
+      setMounted(true)
+
+      requestAnimationFrame(() => {
+        setAppearTransition(false)
+      })
+    }
+  }, [show])
+
+  return (
+    <button
+      className={clsxm(
+        'mt-2 inline-flex h-8 w-8 items-center justify-center rounded-md border border-accent bg-white text-accent opacity-50 transition-all duration-300 hover:opacity-100 focus:opacity-100 focus:outline-none',
+        (!show || appearTransition) && 'translate-x-[60px]',
+        !mounted && 'hidden',
+        className,
+      )}
+      onTransitionEnd={handleTransitionEnd}
+      {...rest}
+    >
+      {children}
+    </button>
+  )
+}
+
+export const FABContainer = (props: {
+  children: JSX.Element | JSX.Element[]
+}) => {
+  const [fabConfig, setFabConfig] = useState<FABConfig[]>([])
+  useEffect(() => {
+    fab.register(setFabConfig)
+    return () => {
+      fab.destroy()
+    }
+  }, [])
+
+  const [serverSide, setServerSide] = useState(true)
+
+  useEffect(() => {
+    setServerSide(false)
+  }, [])
+
+  if (serverSide) return null
+
+  return (
+    <div
+      data-testid="fab-container"
+      className={clsxm(
+        'font-lg fixed bottom-4 bottom-[calc(1rem+env(safe-area-inset-bottom))] right-4 z-[9] flex flex-col',
+      )}
+    >
+      {fabConfig.map((config) => {
+        const { id, onClick, icon } = config
+        return (
+          <FABBase id={id} onClick={onClick} key={id}>
+            {icon}
+          </FABBase>
+        )
+      })}
+      {props.children}
+    </div>
+  )
+}
diff --git a/src/components/ui/fab/index.ts b/src/components/ui/fab/index.ts
new file mode 100644
index 0000000000..5a4aa4660b
--- /dev/null
+++ b/src/components/ui/fab/index.ts
@@ -0,0 +1,2 @@
+export * from './BackToTopFAB'
+export * from './FABContainer'
diff --git a/src/components/ui/markdown/parsers/mention.tsx b/src/components/ui/markdown/parsers/mention.tsx
index 42c2399e7c..419659d675 100644
--- a/src/components/ui/markdown/parsers/mention.tsx
+++ b/src/components/ui/markdown/parsers/mention.tsx
@@ -1,9 +1,13 @@
-import type { MarkdownToJSX } from 'markdown-to-jsx'
-import { Priority, simpleInlineRegex } from 'markdown-to-jsx'
 import React from 'react'
-import { CodiconGithubInverted, MdiTwitter, IcBaselineTelegram } from '~/components/icons/menu-collection'
+import { Priority, simpleInlineRegex } from 'markdown-to-jsx'
+import type { MarkdownToJSX } from 'markdown-to-jsx'
+
+import {
+  CodiconGithubInverted,
+  IcBaselineTelegram,
+  MdiTwitter,
+} from '~/components/icons/menu-collection'
 
- 
 const prefixToIconMap = {
   GH: <CodiconGithubInverted />,
   TW: <MdiTwitter />,
@@ -42,17 +46,16 @@ export const MentionRule: MarkdownToJSX.Rule = {
     const { prefix, name } = content
     if (!name) {
       return null as any
-
     }
 
     // @ts-ignore
     const Icon = prefixToIconMap[prefix]
     // @ts-ignore
-    const urlPrefix = prefixToUrlMap[prefix] 
+    const urlPrefix = prefixToUrlMap[prefix]
 
     return (
-      <div
-        className="mr-2 inline-flex items-center space-x-2 align-bottom"
+      <span
+        className="mx-1 inline-flex items-center space-x-1 align-bottom"
         key={state?.key}
       >
         {Icon}
@@ -63,7 +66,7 @@ export const MentionRule: MarkdownToJSX.Rule = {
         >
           {name}
         </a>
-      </div>
+      </span>
     )
   },
 }
diff --git a/src/providers/root/page-scroll-info-provider.tsx b/src/providers/root/page-scroll-info-provider.tsx
index 18b03932d5..c2f4ba76ea 100644
--- a/src/providers/root/page-scroll-info-provider.tsx
+++ b/src/providers/root/page-scroll-info-provider.tsx
@@ -41,12 +41,13 @@ const usePageScrollLocation = () => useAtomValue(pageScrollLocationAtom)
 const usePageScrollDirection = () => useAtomValue(pageScrollDirectionAtom)
 const usePageScrollLocationSelector = <T,>(
   selector: (scrollY: number) => T,
+  deps: any[] = [],
 ): T =>
   useAtomValue(
     // @ts-ignore
     selectAtom(
       pageScrollLocationAtom,
-      useCallback(($) => selector($), []),
+      useCallback(($) => selector($), deps),
     ),
   )
 export {