feat: support new ui

JianJroh · Nov 23, 2023 · c2d724d · c2d724d
1 parent 9f98e47
commit c2d724d
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 6 deletions.
diff --git a/assets/demo.gif b/assets/demo.gif
diff --git a/src/App.tsx b/src/App.tsx
@@ -15,9 +15,8 @@ function App() {
     if (event.target instanceof HTMLLIElement) {
       const targetIndex = Number(event.target.dataset.index);
       if (!isSharePage) {
-        const isScrollUp = activeQuestionIndex != null && targetIndex < activeQuestionIndex;
         questionEls.forEach((q) => {
-          q.style.scrollMarginTop = isScrollUp ? '60px' : '0';
+          q.style.scrollMarginTop = '56px';
         });
       }
       questionEls?.[targetIndex]?.scrollIntoView({ behavior: 'smooth' });

diff --git a/src/helpers/index.ts b/src/helpers/index.ts
@@ -6,9 +6,8 @@ export function queryChatContainer() {
 }
 
 export function queryQuestionEls() {
-  const CHAT_ITEM_EL_CLASS = 'group';
   return Array.from((queryChatContainer()?.children ?? []) as HTMLDivElement[])
-    .filter((child) => child.classList.contains(CHAT_ITEM_EL_CLASS))
+    .filter((child) => child.hasAttribute('data-testid'))
     .filter((_, index) => index % 2 === 0);
 }
 

diff --git a/src/helpers/sharePage.ts b/src/helpers/sharePage.ts
@@ -1,3 +1,3 @@
 export const isSharePage = location.pathname.startsWith('/share/');
 
-export const scrollMarginTop = 45;
+export const scrollMarginTop = 0;
diff --git a/src/hooks/useQuestionEls.ts b/src/hooks/useQuestionEls.ts
@@ -6,7 +6,13 @@ import { useMountedCallbackValue } from './common/useMountedCallbackValue';
 export function useQuestionEls() {
   const [questionEls, setQuestionEls] = useState(queryQuestionEls());
 
-  const questions = useMemo(() => questionEls.map((q) => q.innerText.trim() ?? ''), [questionEls]);
+  const questions = useMemo(
+    () =>
+      questionEls
+        .map((q) => q.querySelector<HTMLDivElement>('[data-message-author-role]')?.innerText)
+        .filter((s): s is string => !!s),
+    [questionEls]
+  );
 
   const chatContainer = useMountedCallbackValue(queryChatContainer);