Tell people in wait room their country is blocked

Author: jahooma

cli/src/hooks/helpers/send-message.ts

@@ -1,15 +1,18 @@
 import { getErrorObject } from '@codebuff/common/util/error'
 
 import {
+  markFreebuffSessionCountryBlocked,
   markFreebuffSessionEnded,
   markFreebuffSessionSuperseded,
   refreshFreebuffSession,
 } from '../use-freebuff-session'
 import { getProjectRoot } from '../../project-files'
 import { useChatStore } from '../../state/chat-store'
+import { IS_FREEBUFF } from '../../utils/constants'
 import { processBashContext } from '../../utils/bash-context-processor'
 import { markRunningAgentsAsCancelled } from '../../utils/block-operations'
 import {
+  getCountryCodeFromFreeModeError,
   getFreebuffGateErrorKind,
   isOutOfCreditsError,
   isFreeModeUnavailableError,
@@ -389,6 +392,11 @@ export const handleRunCompletion = (params: {
 
     if (isFreeModeUnavailableError(output)) {
       updater.setError(FREE_MODE_UNAVAILABLE_MESSAGE)
+      if (IS_FREEBUFF) {
+        markFreebuffSessionCountryBlocked(
+          getCountryCodeFromFreeModeError(output) ?? 'UNKNOWN',
+        )
+      }
       finalizeAfterError()
       return
     }
@@ -484,6 +492,11 @@ export const handleRunError = (params: {
 
   if (isFreeModeUnavailableError(error)) {
     updater.setError(FREE_MODE_UNAVAILABLE_MESSAGE)
+    if (IS_FREEBUFF) {
+      markFreebuffSessionCountryBlocked(
+        getCountryCodeFromFreeModeError(error) ?? 'UNKNOWN',
+      )
+    }
     return
   }
 

cli/src/hooks/use-freebuff-session.ts

@@ -280,6 +280,21 @@ export function markFreebuffSessionSuperseded(): void {
   controller?.apply({ status: 'superseded' })
 }
 
+/** Flip into the terminal `country_blocked` state from outside the poll loop.
+ *  Used when the chat-completions gate rejects on country even though the
+ *  session-level country check had failed open (null detection → admitted).
+ *  Transitioning the session state here unmounts the Chat surface in favor of
+ *  the waiting-room's country_blocked message, so the user can't keep typing
+ *  and sending doomed requests. */
+export function markFreebuffSessionCountryBlocked(countryCode: string): void {
+  if (!IS_FREEBUFF) return
+  controller?.abort()
+  controller?.apply({ status: 'country_blocked', countryCode })
+  // Best-effort DELETE so we don't hold a waiting-room seat on a session the
+  // server is already refusing to serve at chat time.
+  releaseFreebuffSlot().catch(() => {})
+}
+
 /** Flip into the local `ended` state without an instanceId (server has lost
  *  our row). The chat surface stays mounted with the rejoin banner. */
 export function markFreebuffSessionEnded(): void {

cli/src/utils/error-handling.ts

@@ -57,6 +57,22 @@ export const isFreeModeUnavailableError = (error: unknown): boolean => {
   return false
 }
 
+/**
+ * Extract the detected countryCode off a free_mode_unavailable error, if the
+ * server included one. Used to populate the country_blocked screen after the
+ * chat-completions gate rejects a user whose session-level country check had
+ * previously failed open (null country detection → admitted → now blocked).
+ */
+export const getCountryCodeFromFreeModeError = (
+  error: unknown,
+): string | null => {
+  if (!isFreeModeUnavailableError(error)) return null
+  const candidate = (error as { countryCode?: unknown }).countryCode
+  return typeof candidate === 'string' && candidate.length > 0
+    ? candidate
+    : null
+}
+
 /**
  * Freebuff waiting-room gate errors returned by /api/v1/chat/completions.
  *

web/src/app/api/v1/chat/completions/_post.ts

@@ -286,6 +286,7 @@ export async function postChatCompletions(params: {
           {
             error: 'free_mode_unavailable',
             message: 'Free mode is not available in your country.',
+            countryCode,
           },
           { status: 403 },
         )