| 1 | import { Outlet, data, type ShouldRevalidateFunctionArgs } from "react-router"; |
| 2 | |
| 3 | import { type ChatSidebar, type WorkspaceAgent, shownUsername } from "@g1t/contracts"; |
| 4 | |
| 5 | import type { Route } from "./+types/layout"; |
| 6 | import type { Mentionable } from "../../../lib/chat"; |
| 7 | import { sidebarOrNull, workspacePeople } from "../../../lib/chat.server"; |
| 8 | import { requireUser, roleIn } from "../../../lib/session.server"; |
| 9 | |
| 10 | /** |
| 11 | * Chat mode: what its sidebar shows (the shell draws it, from this data) |
| 12 | * and who can be mentioned or messaged. Each part is empty rather than an |
| 13 | * error when its service does not answer. |
| 14 | */ |
| 15 | export type ChatLayoutData = { |
| 16 | slug: string; |
| 17 | /** The viewer's role: owners may make agents. */ |
| 18 | role: string; |
| 19 | /** The viewer, named as chat names them (`memberName`). */ |
| 20 | me: { id: string; username: string; display_username: string | null; display_name: string; avatar: string | null }; |
| 21 | sidebar: ChatSidebar | null; |
| 22 | people: Mentionable[]; |
| 23 | agents: WorkspaceAgent[]; |
| 24 | }; |
| 25 | |
| 26 | export async function loader({ params, context, request }: Route.LoaderArgs): Promise<ChatLayoutData> { |
| 27 | const viewer = requireUser(context, request); |
| 28 | const role = roleIn(viewer, params.owner); |
| 29 | if (!role) throw data(null, { status: 404 }); |
| 30 | const slug = params.owner.toLowerCase(); |
| 31 | const [sidebar, { people, agents }] = await Promise.all([sidebarOrNull(slug, viewer), workspacePeople(slug, viewer)]); |
| 32 | const self = people.find((person) => person.name === viewer.username.toLowerCase()); |
| 33 | const display_username = viewer.display_username ?? self?.display_username ?? null; |
| 34 | return { |
| 35 | slug, |
| 36 | role, |
| 37 | me: { |
| 38 | id: viewer.id, |
| 39 | username: viewer.username, |
| 40 | display_username, |
| 41 | display_name: self?.display_name ?? shownUsername({ username: viewer.username, display_username }), |
| 42 | avatar: viewer.avatar ?? null, |
| 43 | }, |
| 44 | sidebar, |
| 45 | people, |
| 46 | agents, |
| 47 | }; |
| 48 | } |
| 49 | |
| 50 | /** |
| 51 | * Moving between conversations keeps the sidebar: it refreshes itself |
| 52 | * (components/chat/sidebar.tsx) and after anything is sent. |
| 53 | */ |
| 54 | export function shouldRevalidate({ currentParams, nextParams, formMethod, defaultShouldRevalidate }: ShouldRevalidateFunctionArgs) { |
| 55 | if (formMethod && formMethod !== "GET") return defaultShouldRevalidate; |
| 56 | return currentParams.owner !== nextParams.owner; |
| 57 | } |
| 58 | |
| 59 | export default function ChatLayout() { |
| 60 | return <Outlet />; |
| 61 | } |