@assistant-ui/react/src/context/react/ThreadViewportContext.ts

Open-source TypeScript/React library for building production-grade AI chat experiences

"use client";

import { createContext } from "react";
import type { ReadonlyStore } from "../ReadonlyStore";
import type { UseBoundStore } from "zustand";
import { createContextHook } from "./utils/createContextHook";
import { createContextStoreHook } from "./utils/createContextStoreHook";
import type { ThreadViewportState } from "../stores/ThreadViewport";

export type ThreadViewportContextValue = {
  useThreadViewport: UseBoundStore<ReadonlyStore<ThreadViewportState>>;
};

export const ThreadViewportContext =
  createContext<ThreadViewportContextValue | null>(null);

const useThreadViewportContext = createContextHook(
  ThreadViewportContext,
  "ThreadPrimitive.Viewport",
);

export const { useThreadViewport, useThreadViewportStore } =
  createContextStoreHook(useThreadViewportContext, "useThreadViewport");