simstudioai--sim/apps/sim/lib/workspaces/host-context.ts

import { cache } from 'react'
import type { WorkspaceHostContext } from '@/lib/api/contracts/workspaces'
import { getWorkspaceOwnerSubscriptionAccess } from '@/lib/billing/core/workspace-access'
import { getOrganizationSettingsAccess } from '@/lib/organizations/settings-access'
import { checkWorkspaceAccess } from '@/lib/workspaces/permissions/utils'

/**
 * Resolves all workspace-bound identity and entitlement context from the routed
 * workspace after verifying the viewer's effective workspace permission.
 *
 * Session active-organization state is intentionally absent: it describes the
 * viewer's account, not the workspace host.
 */
async function resolveWorkspaceHostContextForViewer(
  workspaceId: string,
  userId: string
): Promise<WorkspaceHostContext | null> {
  const access = await checkWorkspaceAccess(workspaceId, userId)
  if (!access.exists || !access.hasAccess || !access.workspace || !access.permission) {
    return null
  }

  const hostOrganizationId = access.workspace.organizationId
  const [ownerBilling, hostOrganizationAccess] = await Promise.all([
    getWorkspaceOwnerSubscriptionAccess(workspaceId),
    hostOrganizationId
      ? getOrganizationSettingsAccess(hostOrganizationId, userId)
      : Promise.resolve({ isMember: false, isAdmin: false }),
  ])

  return {
    workspace: {
      id: access.workspace.id,
      name: access.workspace.name,
      workspaceMode: access.workspace.workspaceMode,
      billedAccountUserId: access.workspace.billedAccountUserId,
    },
    hostOrganizationId,
    ownerBilling,
    viewer: {
      permission: access.permission,
      isHostOrganizationMember: hostOrganizationAccess.isMember,
      isHostOrganizationAdmin: hostOrganizationAccess.isAdmin,
    },
  }
}

/**
 * Request-memoized workspace host resolution shared by nested Server
 * Components. Outside a Server Component render, React evaluates the resolver
 * normally without retaining a cross-request cache.
 */
export const getWorkspaceHostContextForViewer = cache(resolveWorkspaceHostContextForViewer)