Skip to content

feat(mobile): PR 5a — kilo-chat foundations (provider + hooks)#2930

Merged
iscekic merged 45 commits intofeat/kilo-chat-migration-pr1from
feat/kilo-chat-migration-pr5a
Apr 30, 2026
Merged

feat(mobile): PR 5a — kilo-chat foundations (provider + hooks)#2930
iscekic merged 45 commits intofeat/kilo-chat-migration-pr1from
feat/kilo-chat-migration-pr5a

Conversation

@iscekic
Copy link
Copy Markdown
Contributor

@iscekic iscekic commented Apr 29, 2026

Summary

PR 5a of the mobile kilo-chat migration plan — Phase 8, Tasks 23–29.

Lays the foundations for the new kilo-chat on mobile. Provider + hooks land here; no routes consume them yet, and Stream chat continues to work in parallel until PR 6 (cutover).

What's in this PR

Config

  • EXPO_PUBLIC_KILO_CHAT_URL / EXPO_PUBLIC_EVENT_SERVICE_URL env keys + KILO_CHAT_URL / EVENT_SERVICE_URL exports in lib/config.ts.
  • ⚠️ EAS env update required (mention in deploy notes): set both env vars.

Workspace deps

  • Added @kilocode/event-service, @kilocode/kilo-chat, @kilocode/notifications (workspace:*) to apps/mobile.

Hooks + provider (apps/mobile/src/components/kilo-chat/)

  • useKiloChatTokenGetter() — module-scoped cache (60s buffer) + in-flight de-dup; one fetch shared by all consumers.
  • useCurrentUserId() — decodes JWT sub.
  • useKiloChatClient() / useEventServiceClient() — read from KiloChatContext, throw if provider missing.
  • KiloChatProvider — constructs EventServiceClient + KiloChatClient once, connects on mount, disconnects on unmount.
  • Mounted in (app)/_layout.tsx (inside the auth boundary).

Drive-by

  • packages/kilo-chat/src/utils.ts — single ! non-null assertion in base64urlEncode loop. The package is now consumed by mobile (whose tsconfig is stricter), so this latent issue surfaced. One-character fix.

Stack

Test plan

  • CI green (typecheck, lint)
  • Verify locally: app boots; provider mounts without crashing; `getToken` succeeds and is cached
  • No regression in existing Stream chat flow (provider is mounted but unused by routes)

Notes for reviewer

  • Mobile codebase already had pre-existing typecheck errors in `chat.tsx` / `use-unread-counts.ts` (`channelId` shape mismatch from PR 3's notifications schema rewrite); those are not regressions from this PR and will be addressed in a later PR.
  • The plan's literal code for Task 23 used CommonJS `module.exports`/`process.env.*`, but `apps/mobile` uses an ESM `ENV_KEYS` map consumed by `app.config.ts` and read from `expoConfig.extra` — implementation follows the existing convention.
  • The plan's literal code for Task 25 used `useTRPC()` for an imperative one-off fetch; mobile convention is to use the vanilla `trpcClient` for callback fetches (matches `mobile-session-manager.ts` etc.).

iscekic added 30 commits April 29, 2026 17:28
The badge_counts.badge_bucket column is a free-form string. To prevent
namespace collisions as more surfaces start emitting badge updates
(per-instance today, per-conversation later), centralize bucket-key
derivation in @kilocode/notifications and route NotificationChannelDO
through it. Mirrors the presence-context builders in @kilocode/event-service.

Safe to introduce now without a data migration because PR 2's migration
already wipes badge_counts.
…-chat producer

Adds kiloclawInstanceContext and kiloclawConversationContext path
builders to @kilocode/event-service, replacing hardcoded template
literals in kilo-chat's event-push.ts and its test so all callers
share a single source of truth.
When a chat message is persisted, fire-and-forget a call to
NOTIFICATIONS.sendPushForConversation so non-sender human members of the
conversation receive a push. Runs after realtime/event-service delivery
inside postCommitFanOut, with errors swallowed so push failures cannot
fail the send.

- Skip when there are no other human recipients or no sandboxId.
- senderUserId = callerId for human senders, null for bot senders.
- title is "<sandboxLabel> · <conversationTitle>"; bodyPreview is the
  first 200 chars of the concatenated text blocks.
- Add @kilocode/notifications workspace dep and layer the RPC method
  shape into Env via bindings.d.ts.
- Add a notifications-stub worker to the vitest config so tests can
  spy on env.NOTIFICATIONS.sendPushForConversation, and globally mock
  sandbox-lookup in setup.ts (it imports pg via @kilocode/db).
…es, fix test mock

- Remove `stream-chat` from `services/notifications/package.json`; the Stream
  webhook (its only consumer) was deleted earlier in the stack.
- Regenerate `worker-configuration.d.ts` so the workerd runtime types match the
  current toolchain (sibling services were on `1.20260312.1`; this one had
  drifted to `1.20251217.0` from a stale local cache).
- Fix the global test mock to reference the renamed `badge_counts` table; the
  setup file was authored against the pre-rename name and never matched.
- Tidy two pre-existing lint nits in the new test files (`import type` for
  type-only import, drop unused `cols` parameter).
…leak

- Switch `NotificationsService` from default-only to a named class export
  with a separate default. `services/kilo-chat/wrangler.jsonc` binds via
  `entrypoint: "NotificationsService"`, which resolves named module
  exports. The default-only form (`export default class NotificationsService`)
  exports under the `default` key — kilo-chat's RPC binding would not have
  resolved at deploy. Mirrors the existing pattern in
  `services/kilo-chat/src/index.ts` (`KiloChatService`).

- `dispatchPush` now uses a two-stage idempotency record (`pending` →
  `delivered`). The badge increment was previously non-idempotent: an
  Expo failure returned `failed` without writing the idempotency key, so
  upstream retries (which the design explicitly invites) re-ran the
  increment before the next send and inflated the badge by one per
  retry. The `pending` marker is written before the increment and
  short-circuits the increment on retry; the `delivered` marker is only
  written on success.

- `setAlarm` is now gated on `getAlarm() === null`. Calling `setAlarm`
  unconditionally on each successful push — as the previous code did —
  replaces the pending alarm and pushes the cleanup forward indefinitely
  on a conversation receiving more than one push per `IDEM_TTL_MS`,
  leaking expired idempotency entries.

Adds two test cases covering the badge-retry and alarm-reset paths.
- Schedule the cleanup alarm when writing the `pending` marker, not only
  on `delivered`. Without this, an Expo failure followed by no further
  push activity for the conversation leaves the `pending` record in DO
  storage forever (no alarm was ever set to prune it).

- After the alarm fires, reschedule for the earliest remaining record's
  expiry instead of leaving the alarm slot empty. Otherwise a quiet
  conversation strands its younger entries until some unrelated future
  dispatch wakes the DO up.

Both paths go through a small `ensureCleanupAlarm` helper that gates on
`getAlarm() === null` so a busy conversation still doesn't push the
alarm forward on every call.
The kiloclaw-scoped presence paths are literally `/presence` prefixed
onto the kiloclaw event-context paths. Build them by composition so the
`/kiloclaw/{sandboxId}[/{conversationId}]` segment shape is defined in
exactly one place — `kiloclaw-contexts.ts`.

Pure refactor; same string output, template-literal types still narrow
to the same shape.
Introduces a single app-shell EventServiceProvider that owns the
EventServiceClient and KiloChatClient for all authenticated routes.
Mounted in (app)/layout.tsx so platform/instance/conversation presence
subscriptions and the kilo-chat UI share one WebSocket.

KiloChatLayout now consumes the global clients via useEventServiceClient()
instead of spinning up its own pair, and the getToken prop is removed from
KiloChatLayoutProps (along with both call sites). The local
useEventService(getToken) factory is dead code and has been deleted;
useInstanceContext / useConversationContext stay since they take
EventServiceClient as a parameter.
Thin hook that subscribes the global EventServiceClient to a single
context for the lifetime of the calling component, gated by an `active`
flag. Will back upcoming platform- and instance-level presence
indicators.
…eSubscription

- Drop dead getToken field from KiloChatContextValue (no consumers).
- Remove useInstanceContext / useConversationContext hooks; both call
  sites now use the shared usePresenceSubscription primitive directly.
- Harden usePresenceSubscription against empty-string contexts.
- usePresenceSubscription: accept 'string | null' instead of empty-string
  sentinel; update call sites (KiloChatLayout, MessageArea, useInstancePresence)
- kilo-chat router: validate expiresAt with z.iso.datetime()
- kilo-chat-router test: verify the JWT payload (kiloUserId, tokenSource,
  version) and that expiresAt lands in the expected ~1h window
- MessageArea: comment distinguishing the always-on chat-event subscription
  from the visibility-gated presence subscription
Multiple consumers can now independently hold the same context without
trampling each other. The wire context.subscribe/context.unsubscribe
messages are only sent on the 0->1 and 1->0 refcount transitions; the
intermediate churn stays client-side.

Resubscribe-on-reconnect dedupes by context key.

Tests cover: double-subscribe collapses to a single wire send, partial
unsubscribe keeps the context alive, last-consumer-out releases it,
mixed batches only send newly-active contexts, unknown-context
unsubscribes are no-ops, and reconnect resubscribes each context once.
@iscekic iscekic self-assigned this Apr 29, 2026
Comment thread apps/mobile/src/components/kilo-chat/hooks/use-kilo-chat-token.ts Outdated
Comment thread apps/mobile/src/components/kilo-chat/hooks/use-kilo-chat-token.ts Outdated
Comment thread apps/mobile/src/components/kilo-chat/hooks/use-current-user-id.ts Outdated
@kilo-code-bot
Copy link
Copy Markdown
Contributor

kilo-code-bot Bot commented Apr 29, 2026

Code Review Summary

Status: No Issues Found | Recommendation: Merge

Resolved Previous Findings
File Line Issue
apps/mobile/src/components/kilo-chat/hooks/use-kilo-chat-token.ts 24 Captured auth token can leave the provider with a stale getter
apps/mobile/src/components/kilo-chat/hooks/use-kilo-chat-token.ts 23 Module-level token cache survives account changes
apps/mobile/src/components/kilo-chat/hooks/use-kilo-chat-token.ts 48 Rejected dedup promise can become unhandled
apps/mobile/src/components/kilo-chat/hooks/use-current-user-id.ts 29 Current user ID reads sub, but generated token uses kiloUserId
Files Reviewed (1 file)
  • apps/mobile/src/components/kilo-chat/hooks/use-kilo-chat-token.ts

Reviewed by gpt-5.5-2026-04-23 · 1,077,352 tokens

- Key the module-level kilo-chat JWT cache and in-flight ref on the
  current auth token, so signing out and back in as a different user
  within the 1h token window no longer returns the previous user's
  cached JWT.
- Restructure dedup so the first caller awaits the same shared promise
  via a slot reference, eliminating the unhandled rejection that the
  prior resolve/reject-pair pattern produced when the only caller's
  fetch failed.
- Decode kiloUserId from the JWT payload instead of the standard `sub`
  claim — generateApiToken writes the user id as kiloUserId, so the
  sub-based version always returned null.
Comment thread apps/mobile/src/components/kilo-chat/hooks/use-kilo-chat-token.ts Outdated
KiloChatProvider builds its EventService and KiloChat clients exactly
once via useState initializer, so it captures whatever getter exists at
first mount. Closing the previous getter over a render-time `authToken`
meant a cold start where the (app) layout mounted before SecureStore
finished loading would freeze the clients with an undefined token,
trapping them in a permanent reconnect loop.

Read the auth token from SecureStore inside the getter, the same pattern
trpcClient uses. The hook returns a stable callback with no React deps,
and the cache stays keyed on the auth token so user-switch safety is
preserved.
Base automatically changed from feat/kilo-chat-migration-pr4-5 to feat/kilo-chat-migration-pr1 April 30, 2026 13:10
…to feat/kilo-chat-migration-pr5a

# Conflicts:
#	packages/notifications/src/badge-buckets.ts
#	services/kilo-chat/wrangler.jsonc
@iscekic iscekic merged commit 6101b20 into feat/kilo-chat-migration-pr1 Apr 30, 2026
1 check passed
@iscekic iscekic deleted the feat/kilo-chat-migration-pr5a branch April 30, 2026 13:20
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant