@cuongtran001/kanna 1.76.2 → 1.76.4

package/src/shared/beacon-alias.test.ts

@@ -0,0 +1,64 @@
+import { describe, expect, test } from "bun:test"
+import {
+  beaconDisplayName,
+  buildBeaconMentionIndex,
+  resolveBeaconMentions,
+  validateBeaconAliasFormat,
+} from "./beacon-alias"
+import type { BeaconConfig } from "./beacon-config"
+import { DEFAULT_BEACON_SCOPE } from "./beacon-scope"
+
+function config(id: string, label: string, alias?: string): BeaconConfig {
+  return {
+    id,
+    label,
+    ...(alias === undefined ? {} : { alias }),
+    publicKey: "pk",
+    os: "linux",
+    scope: DEFAULT_BEACON_SCOPE,
+    enabled: true,
+    createdAt: "2026-10-13T00:00:00.000Z",
+    updatedAt: "2026-10-13T00:00:00.000Z",
+  }
+}
+
+describe("beacon alias", () => {
+  test("the display name prefers the alias over the hostname label", () => {
+    expect(beaconDisplayName({ alias: "office-mac", label: "MacBook-Pro.local" })).toBe("office-mac")
+    expect(beaconDisplayName({ alias: null, label: "MacBook-Pro.local" })).toBe("MacBook-Pro.local")
+  })
+
+  test("format accepts a slug handle and rejects uppercase, spaces and an overlong name", () => {
+    expect(validateBeaconAliasFormat("office-mac_2")).toBeNull()
+    expect(validateBeaconAliasFormat("Office")).not.toBeNull()
+    expect(validateBeaconAliasFormat("my mac")).not.toBeNull()
+    expect(validateBeaconAliasFormat("a".repeat(41))).not.toBeNull()
+  })
+})
+
+describe("beacon mention index", () => {
+  test("an alias claims its handle before another beacon's hostname slug", () => {
+    const index = buildBeaconMentionIndex([config("a", "office-mac"), config("z", "other", "office-mac")])
+    expect(index.beaconIdByHandle.get("office-mac")).toBe("z")
+    expect(index.handleByBeaconId.get("a")).toBe("office-mac-2")
+  })
+
+  test("a beacon with no alias is addressed by a slug of its hostname", () => {
+    const index = buildBeaconMentionIndex([config("a", "MacBook Pro.local")])
+    expect(index.handleByBeaconId.get("a")).toBe("macbook-pro-local")
+  })
+
+  test("colliding hostname slugs are suffixed in id order regardless of input order", () => {
+    const forward = buildBeaconMentionIndex([config("a", "Build Box"), config("b", "build-box")])
+    const backward = buildBeaconMentionIndex([config("b", "build-box"), config("a", "Build Box")])
+    expect([...forward.handleByBeaconId]).toEqual([["a", "build-box"], ["b", "build-box-2"]])
+    expect([...backward.handleByBeaconId]).toEqual([...forward.handleByBeaconId])
+  })
+
+  test("a message mention resolves to the beacon and an unknown handle is reported unresolved", () => {
+    const beacons = [config("a", "laptop", "office-mac")]
+    const resolved = resolveBeaconMentions("run uname on @beacon/office-mac and @beacon/ghost", beacons)
+    expect(resolved.beacons.map((beacon) => beacon.id)).toEqual(["a"])
+    expect(resolved.unresolvedHandles).toEqual(["ghost"])
+  })
+})

package/src/shared/beacon-alias.ts

@@ -0,0 +1,120 @@
+import type { BeaconConfig } from "./beacon-config"
+
+export const BEACON_ALIAS_PATTERN = /^[a-z0-9][a-z0-9_-]{0,39}$/
+
+export const BEACON_ALIAS_MAX_LENGTH = 40
+
+export const BEACON_MENTION_PREFIX = "beacon/"
+
+export const BEACON_MENTION_PATTERN = "(^|[\\s\\n\\t])@beacon/([A-Za-z0-9][A-Za-z0-9_-]*)"
+
+export function createBeaconMentionRegex(): RegExp {
+  return new RegExp(BEACON_MENTION_PATTERN, "g")
+}
+
+export function validateBeaconAliasFormat(alias: string): string | null {
+  if (alias.length === 0) return "Short name is required"
+  if (alias.length > BEACON_ALIAS_MAX_LENGTH) {
+    return `Short name must be at most ${BEACON_ALIAS_MAX_LENGTH} characters`
+  }
+  if (!BEACON_ALIAS_PATTERN.test(alias)) {
+    return "Use lowercase letters, digits, - and _, starting with a letter or digit"
+  }
+  return null
+}
+
+export function beaconDisplayName(beacon: { alias?: string | null; label: string }): string {
+  return beacon.alias ?? beacon.label
+}
+
+export interface MentionableBeacon {
+  id: string
+  label: string
+  alias?: string | null
+}
+
+export interface BeaconMentionIndex {
+  handleByBeaconId: ReadonlyMap<string, string>
+  beaconIdByHandle: ReadonlyMap<string, string>
+}
+
+const INVALID_HANDLE_CHARS = /[^a-z0-9_-]+/g
+const EDGE_SEPARATORS = /^[-_]+|[-_]+$/g
+
+function slugifyLabel(label: string): string {
+  const slug = label
+    .toLowerCase()
+    .replace(INVALID_HANDLE_CHARS, "-")
+    .replace(EDGE_SEPARATORS, "")
+    .slice(0, BEACON_ALIAS_MAX_LENGTH)
+    .replace(EDGE_SEPARATORS, "")
+  return slug.length > 0 ? slug : "beacon"
+}
+
+function freeHandle(base: string, taken: ReadonlyMap<string, string>): string {
+  if (!taken.has(base)) return base
+  let suffix = 2
+  while (taken.has(`${base}-${suffix}`)) suffix += 1
+  return `${base}-${suffix}`
+}
+
+export function buildBeaconMentionIndex(beacons: readonly MentionableBeacon[]): BeaconMentionIndex {
+  const ordered = [...beacons].sort((left, right) => left.id.localeCompare(right.id))
+  const handleByBeaconId = new Map<string, string>()
+  const beaconIdByHandle = new Map<string, string>()
+
+  for (const beacon of ordered) {
+    if (beacon.alias === undefined || beacon.alias === null) continue
+    if (beaconIdByHandle.has(beacon.alias)) continue
+    handleByBeaconId.set(beacon.id, beacon.alias)
+    beaconIdByHandle.set(beacon.alias, beacon.id)
+  }
+
+  for (const beacon of ordered) {
+    if (handleByBeaconId.has(beacon.id)) continue
+    const handle = freeHandle(slugifyLabel(beacon.label), beaconIdByHandle)
+    handleByBeaconId.set(beacon.id, handle)
+    beaconIdByHandle.set(handle, beacon.id)
+  }
+
+  return { handleByBeaconId, beaconIdByHandle }
+}
+
+export function parseBeaconMentionHandles(text: string): string[] {
+  const handles: string[] = []
+  for (const match of text.matchAll(createBeaconMentionRegex())) {
+    const handle = match[2]
+    if (handle !== undefined) handles.push(handle.toLowerCase())
+  }
+  return handles
+}
+
+export interface BeaconMentionResolution {
+  beacons: BeaconConfig[]
+  unresolvedHandles: string[]
+}
+
+export function resolveBeaconMentions(
+  text: string,
+  beacons: readonly BeaconConfig[],
+): BeaconMentionResolution {
+  const handles = parseBeaconMentionHandles(text)
+  if (handles.length === 0) return { beacons: [], unresolvedHandles: [] }
+  const { beaconIdByHandle } = buildBeaconMentionIndex(beacons)
+  const byId = new Map(beacons.map((beacon) => [beacon.id, beacon]))
+  const resolved: BeaconConfig[] = []
+  const unresolvedHandles: string[] = []
+  const seen = new Set<string>()
+  for (const handle of handles) {
+    const id = beaconIdByHandle.get(handle)
+    const beacon = id === undefined ? undefined : byId.get(id)
+    if (beacon === undefined) {
+      if (!unresolvedHandles.includes(handle)) unresolvedHandles.push(handle)
+      continue
+    }
+    if (seen.has(beacon.id)) continue
+    seen.add(beacon.id)
+    resolved.push(beacon)
+  }
+  return { beacons: resolved, unresolvedHandles }
+}

package/src/shared/beacon-config.ts

@@ -4,6 +4,7 @@ import type { BeaconScope } from "./beacon-scope"
 export interface BeaconConfig {
   id: string
   label: string
+  alias?: string
   publicKey: string
   os: BeaconOs
   scope: BeaconScope
@@ -20,6 +21,7 @@ export interface BeaconInput {
 
 export interface BeaconPatch {
   label?: string
+  alias?: string | null
   scope?: BeaconScope
   enabled?: boolean
 }

package/src/shared/beacon-status.test.ts

@@ -28,6 +28,7 @@ describe("buildBeaconStatusRows", () => {
       {
         id: "a",
         label: "label-a",
+        alias: null,
         os: "linux",
         enabled: true,
         online: true,
@@ -39,6 +40,7 @@ describe("buildBeaconStatusRows", () => {
       {
         id: "b",
         label: "label-b",
+        alias: null,
         os: "windows",
         enabled: false,
         online: false,

package/src/shared/beacon-status.ts

@@ -4,6 +4,7 @@ import type { BeaconConfig } from "./beacon-config"
 export interface BeaconStatusRow {
   id: string
   label: string
+  alias: string | null
   os: BeaconOs
   enabled: boolean
   online: boolean
@@ -37,6 +38,7 @@ export function buildBeaconStatusRows(
     return {
       id: config.id,
       label: config.label,
+      alias: config.alias ?? null,
       os: config.os,
       enabled: config.enabled,
       online: state?.online ?? false,

package/src/shared/chat-activity.ts

@@ -0,0 +1,8 @@
+import type { ChatActivity } from "./types"
+
+export function chatHasPendingWork(activity: ChatActivity): boolean {
+  return activity.backgroundTasks > 0
+    || activity.agents > 0
+    || activity.workflow !== null
+    || activity.loop !== null
+}