feat: manually mirror opencoze's code from bytedance

Change-Id: I09a73aadda978ad9511264a756b2ce51f5761adf

frontend/packages/common/chat-area/hooks/__tests__/use-event-callback.test.ts

@@ -0,0 +1,52 @@
+/*
+ * Copyright 2025 coze-dev Authors
+ *
+ * Licensed 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 { useRef, useState } from 'react';
+
+import { expect, it, vi } from 'vitest';
+import { act, renderHook } from '@testing-library/react-hooks';
+
+import { useEventCallback } from '../src/hooks/use-event-callback';
+
+it('get a fixed reference', () => {
+  const print = vi.fn();
+  const { result } = renderHook(() => {
+    const [count, setCount] = useState(0);
+    const fn = useEventCallback(print);
+    const fnRef = useRef(fn);
+    const isSame = fnRef.current === fn;
+    const update = setCount;
+    fn(count);
+    return {
+      isSame,
+      update,
+    };
+  });
+
+  act(() => {
+    result.current.update(100);
+  });
+  expect(result.current.isSame).toBe(true);
+  expect(print.mock.calls.length).toBe(2);
+  expect(print.mock.calls[1][0]).toBe(100);
+
+  act(() => {
+    result.current.update(200);
+  });
+  expect(result.current.isSame).toBe(true);
+  expect(print.mock.calls.length).toBe(3);
+  expect(print.mock.calls[2][0]).toBe(200);
+});

frontend/packages/common/chat-area/hooks/__tests__/use-imperative-layout-effect.test.ts

@@ -0,0 +1,32 @@
+/*
+ * Copyright 2025 coze-dev Authors
+ *
+ * Licensed 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 { expect, it, vi } from 'vitest';
+import { act, renderHook } from '@testing-library/react-hooks';
+
+import { useImperativeLayoutEffect } from '../src/hooks/use-imperative-layout-effect';
+
+it('run after layout effect', () => {
+  const fn = vi.fn();
+  const { result } = renderHook(() => useImperativeLayoutEffect(fn));
+  expect(fn.mock.calls.length).toBe(0);
+  act(() => {
+    result.current(22);
+    expect(fn.mock.calls.length).toBe(0);
+  });
+  expect(fn.mock.calls.length).toBe(1);
+  expect(fn.mock.calls[0][0]).toBe(22);
+});

frontend/packages/common/chat-area/hooks/__tests__/use-search.test.ts

@@ -0,0 +1,218 @@
+/*
+ * Copyright 2025 coze-dev Authors
+ *
+ * Licensed 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 { expect, it, vi } from 'vitest';
+import { renderHook, act } from '@testing-library/react-hooks';
+
+import { useSearch } from '../src/hooks/use-search';
+
+const sleep = (t: number) => new Promise(resolve => setTimeout(resolve, t));
+const getSearch = (delay: number) =>
+  vi.fn(async (str: string): Promise<string> => {
+    await sleep(delay);
+    return str;
+  });
+
+const config = {
+  searchWait: 40,
+  debounce: { debounceInterval: 20 },
+};
+
+describe('search', () => {
+  beforeEach(() => {
+    vi.useFakeTimers();
+  });
+
+  it('run once input', async () => {
+    const search = getSearch(config.searchWait);
+    const { result } = renderHook(() => useSearch(search, config.debounce));
+    expect(result.current.searchStage).toBe('empty');
+    act(() => {
+      result.current.setPayload('1');
+    });
+    // 10ms
+    await vi.advanceTimersByTimeAsync(10);
+    expect(result.current.searchStage).toBe('debouncing');
+    // 40ms
+    await vi.advanceTimersByTimeAsync(30);
+    expect(result.current.searchStage).toBe('searching');
+    // 65ms
+    await vi.advanceTimersByTimeAsync(25);
+    expect(result.current.searchStage).toBe('success');
+  });
+
+  it('support adjust debounce time', async () => {
+    const search = getSearch(config.searchWait);
+    const adjustDebounce = (payload: string | null): number => {
+      if (payload === '') {
+        return 0;
+      }
+      return config.debounce.debounceInterval;
+    };
+    const { result } = renderHook(() =>
+      useSearch(search, { ...config.debounce, adjustDebounce }),
+    );
+    expect(result.current.searchStage).toBe('empty');
+    act(() => {
+      result.current.setPayload('');
+    });
+    await vi.advanceTimersByTimeAsync(1);
+    expect(result.current.searchStage).toBe('searching');
+  });
+
+  it('debounce well', async () => {
+    const search = getSearch(config.searchWait);
+    const { result } = renderHook(() => useSearch(search, config.debounce));
+    act(() => {
+      result.current.setPayload('1');
+    });
+    expect(result.current.searchStage).toBe('debouncing');
+    await vi.advanceTimersByTimeAsync(10);
+    act(() => {
+      result.current.setPayload('2');
+    });
+    expect(result.current.searchStage).toBe('debouncing');
+    // 25ms
+    await vi.advanceTimersByTimeAsync(25);
+    expect(result.current.searchStage).toBe('searching');
+    expect(search.mock.calls.length).toBe(1);
+    // 65ms
+    await vi.advanceTimersByTimeAsync(65);
+    expect(result.current.searchStage).toBe('success');
+    expect(search.mock.calls.length).toBe(1);
+  });
+
+  it('use latest result', async () => {
+    const search = getSearch(config.searchWait);
+    const { result } = renderHook(() => useSearch(search, config.debounce));
+    act(() => {
+      result.current.setPayload('1');
+    });
+    await vi.advanceTimersByTimeAsync(25);
+    expect(result.current.searchStage).toBe('searching');
+    expect(search.mock.calls.length).toBe(1);
+    act(() => {
+      result.current.setPayload('2');
+    });
+    expect(result.current.searchStage).toBe('debouncing');
+    // 25ms
+    await vi.advanceTimersByTimeAsync(25);
+    expect(result.current.searchStage).toBe('searching');
+    // 45ms
+    await vi.advanceTimersByTimeAsync(20);
+    expect(result.current.searchStage).toBe('searching');
+    // 65ms
+    await vi.advanceTimersByTimeAsync(20);
+    expect(result.current.searchStage).toBe('success');
+    expect(search.mock.calls.length).toBe(2);
+    expect(result.current.res).toBe('2');
+  });
+
+  it('distinguishes payload between null and other falsy value', () => {
+    const search = getSearch(config.searchWait);
+    const { result } = renderHook(() => useSearch(search, config.debounce));
+    act(() => {
+      result.current.setPayload('');
+    });
+    expect(result.current.searchStage).toBe('debouncing');
+    act(() => {
+      result.current.setPayload(null);
+    });
+    expect(result.current.searchStage).toBe('empty');
+  });
+
+  it('goes empty immediately', async () => {
+    const search = getSearch(config.searchWait);
+    const { result } = renderHook(() => useSearch(search, config.debounce));
+    act(() => {
+      result.current.setPayload('1');
+    });
+    await vi.advanceTimersByTimeAsync(10);
+    expect(result.current.searchStage).toBe('debouncing');
+    // 30ms
+    await vi.advanceTimersByTimeAsync(20);
+    expect(result.current.searchStage).toBe('searching');
+    act(() => {
+      result.current.setPayload(null);
+    });
+    expect(result.current.searchStage).toBe('empty');
+    // 70ms
+    await vi.advanceTimersByTimeAsync(40);
+    expect(result.current.searchStage).toBe('empty');
+    expect(result.current.res).toBe(null);
+  });
+
+  const failSearch = async (str: string) => {
+    await sleep(config.searchWait);
+    throw new Error(str);
+  };
+
+  it('get error', async () => {
+    const { result } = renderHook(() => useSearch(failSearch, config.debounce));
+    act(() => {
+      result.current.setPayload('1');
+    });
+    expect(result.current.searchStage).toBe('debouncing');
+    await vi.advanceTimersByTimeAsync(100);
+    expect(result.current.searchStage).toBe('failed');
+    expect(result.current.res).toBe(null);
+  });
+
+  it('get error but cleared', async () => {
+    const { result } = renderHook(() => useSearch(failSearch, config.debounce));
+    act(() => {
+      result.current.setPayload('1');
+    });
+    await vi.advanceTimersByTimeAsync(25);
+    expect(result.current.searchStage).toBe('searching');
+    act(() => {
+      result.current.setPayload(null);
+    });
+    expect(result.current.searchStage).toBe('empty');
+    // 75ms
+    await vi.advanceTimersByTimeAsync(50);
+    expect(result.current.searchStage).toBe('empty');
+  });
+
+  it('get error but covered', async () => {
+    let err = 0;
+    const failOnceSearch = vi.fn(async (str: string) => {
+      await sleep(config.searchWait);
+      if (!err++) {
+        throw new Error(str);
+      }
+      return str;
+    });
+    const { result } = renderHook(() =>
+      useSearch(failOnceSearch, config.debounce),
+    );
+    act(() => {
+      result.current.setPayload('1');
+    });
+    await vi.advanceTimersByTimeAsync(25);
+    act(() => {
+      result.current.setPayload('2');
+    });
+    // 1: 65ms; 2: 40ms
+    await vi.advanceTimersByTimeAsync(40);
+    expect(failOnceSearch.mock.results[0].value).rejects.toThrowError('1');
+    expect(result.current.searchStage).toBe('searching');
+    // 2: 70ms
+    await vi.advanceTimersByTimeAsync(30);
+    expect(result.current.searchStage).toBe('success');
+    expect(result.current.res).toBe('2');
+  });
+});