Fix depleted account handling and UI

Author: ItsMeSamey

build.zig

@@ -313,7 +313,9 @@ pub fn build(b: *std.Build) void {
 
     const backend_tests = b.addTest(.{
         .root_module = b.createModule(.{
-            .root_source_file = b.path("src/rpc.zig"),
+            // Keep package root at repo root so @embedFile("../frontend/...") used by
+            // imported modules remains inside package boundaries during tests.
+            .root_source_file = b.path("rpc_tests.zig"),
             .target = target,
             .optimize = optimize,
             .strip = strip_symbols,

frontend/src/lib/codexAuth.ts

@@ -345,13 +345,34 @@ const decodeBridgeValue = <T>(op: string, value: unknown): T => {
   return value as T;
 };
 
-const callBridge = async <T>(op: string, payload: Record<string, unknown> = {}): Promise<T> => {
-  const request = JSON.stringify({ op, ...payload });
+const getWebuiBridge = (): WebuiRpcBridge | null => {
   const bridge = (globalThis as { webuiRpc?: WebuiRpcBridge }).webuiRpc;
   if (!bridge || typeof bridge.cm_rpc !== "function") {
-    throw new Error("WebUI bridge is unavailable (webuiRpc.cm_rpc missing).");
+    return null;
   }
+  return bridge;
+};
 
+const waitForWebuiBridge = async (): Promise<WebuiRpcBridge> => {
+  const immediate = getWebuiBridge();
+  if (immediate) {
+    return immediate;
+  }
+
+  for (let attempt = 0; attempt < 40; attempt += 1) {
+    await new Promise((resolve) => window.setTimeout(resolve, 25));
+    const bridge = getWebuiBridge();
+    if (bridge) {
+      return bridge;
+    }
+  }
+
+  throw new Error("WebUI bridge is unavailable (webuiRpc.cm_rpc missing).");
+};
+
+const callBridge = async <T>(op: string, payload: Record<string, unknown> = {}): Promise<T> => {
+  const request = JSON.stringify({ op, ...payload });
+  const bridge = await waitForWebuiBridge();
   const rawResponse = await bridge.cm_rpc(request);
   return decodeBridgeValue<T>(op, rawResponse);
 };

rpc_tests.zig

@@ -0,0 +1,5 @@
+const _ = @import("src/rpc.zig");
+
+test {
+    // Tests are defined in src/rpc.zig and are included via import.
+}

src/main.zig

@@ -1,4 +1,5 @@
 const std = @import("std");
+const builtin = @import("builtin");
 const webui = @import("webui");
 
 const embedded_index = @import("embedded_index.zig");
@@ -31,8 +32,16 @@ var oauth_listener_cancel = std.atomic.Value(bool).init(false);
 
 pub fn main() !void {
     var gpa = std.heap.GeneralPurposeAllocator(.{}){};
-    defer _ = gpa.deinit();
-    const allocator = gpa.allocator();
+    defer {
+        if (builtin.mode == .Debug) {
+            _ = gpa.deinit();
+        }
+    }
+
+    const allocator = if (builtin.mode == .Debug)
+        gpa.allocator()
+    else
+        std.heap.smp_allocator;
 
     const args = try std.process.argsAlloc(allocator);
     defer std.process.argsFree(allocator, args);

src/rpc_webui.zig

@@ -22,9 +22,6 @@ pub const RpcBridgeMethods = struct {
         };
         defer std.heap.page_allocator.free(rpc_response);
 
-        bridge_response_lock.lock();
-        defer bridge_response_lock.unlock();
-
         if (rpc_response.len > bridge_response_storage.len) {
             return .{ .data = "{\"ok\":false,\"error\":\"RPC response exceeds bridge buffer\"}" };
         }
@@ -36,9 +33,8 @@ pub const RpcBridgeMethods = struct {
 };
 
 var bridge_cancel_ptr: ?*std.atomic.Value(bool) = null;
-var bridge_response_lock: std.Thread.Mutex = .{};
-var bridge_response_storage: [8 * 1024 * 1024]u8 = undefined;
-var bridge_response_len: usize = 0;
+threadlocal var bridge_response_storage: [8 * 1024 * 1024]u8 = undefined;
+threadlocal var bridge_response_len: usize = 0;
 
 pub fn setCancelPointer(cancel_ptr: *std.atomic.Value(bool)) void {
     bridge_cancel_ptr = cancel_ptr;