storytellurx-v1/app/game-engine/turn-provider.tsx

"use client";

import type { NarrativeMessage } from "./narrative-provider.tsx";
import { useGetAssistantTurn } from "./game-state.ts";
import { createContext, useContext } from "react";

interface TurnContextValue {
  message: NarrativeMessage;
  previousAssistantMessage?: NarrativeMessage;
  nextAssistantMessage?: NarrativeMessage | null;
  previousMessages?: NarrativeMessage[];
  turnData: ReturnType<typeof useGetAssistantTurn>;
  previousTurnData: ReturnType<typeof useGetAssistantTurn>;
  nextTurnData: ReturnType<typeof useGetAssistantTurn> | null;
}

const TurnContext = createContext<TurnContextValue | null>(null);

export const useTurn = () => {
  const context = useContext(TurnContext);
  if (!context) {
    throw new Error("useTurn must be used within a TurnProvider");
  }
  return context;
};

export default function TurnProvider({
  children,
  messages,
  nextAssistantMessage,
}: {
  messages: NarrativeMessage[];
  children: React.ReactNode;
  nextAssistantMessage: NarrativeMessage | null;
}) {
  const previousAssistantMessages = messages
    .filter((t) => t.role === "assistant");

  const message = messages.at(-1);

  if (!message) return null;

  const contextValue: TurnContextValue = {
    message,
    previousAssistantMessage: previousAssistantMessages?.at(-1),
    nextAssistantMessage,
    previousMessages: messages,
    turnData: useGetAssistantTurn(message),
    previousTurnData: useGetAssistantTurn(previousAssistantMessages?.at(-1)),
    nextTurnData: nextAssistantMessage
      ? useGetAssistantTurn(nextAssistantMessage)
      : null,
  };

  return (
    <TurnContext.Provider value={contextValue}>
      {children}
    </TurnContext.Provider>
  );
}