oc-tweaks 0.1.0

package/src/__tests__/index.test.ts

@@ -0,0 +1,180 @@
+// @ts-nocheck
+
+import { afterEach, describe, expect, test } from "bun:test"
+
+import {
+  backgroundSubagentPlugin,
+  compactionPlugin,
+  leaderboardPlugin,
+  notifyPlugin,
+} from "../index"
+
+const originalBunFile = Bun.file
+const originalBunWrite = Bun.write
+const originalFetch = globalThis.fetch
+const originalHome = Bun.env?.HOME
+
+function mockBunFile(mockData: Record<string, any>) {
+  ;(globalThis as any).Bun.file = (path: string) => ({
+    exists: async () => path in mockData,
+    json: async () => {
+      if (!(path in mockData)) throw new Error("ENOENT")
+      const data = mockData[path]
+      if (data instanceof Error) throw data
+      return data
+    },
+    text: async () => JSON.stringify(mockData[path] ?? ""),
+  })
+}
+
+function createShellMock(options?: { availableCommands?: string[] }) {
+  const available = new Set(options?.availableCommands ?? [])
+  const calls: Array<{ command: string }> = []
+
+  const $ = async (strings: TemplateStringsArray, ...values: any[]) => {
+    const segments = Array.from(strings)
+    const command = segments.reduce(
+      (acc, segment, index) =>
+        acc + segment + (index < values.length ? String(values[index]) : ""),
+      "",
+    )
+    calls.push({ command })
+
+    if (command.startsWith("which ")) {
+      const bin = String(values[0] ?? command.slice("which ".length).trim())
+      if (available.has(bin)) return { stdout: `${bin}\n` }
+      throw new Error(`missing ${bin}`)
+    }
+
+    return { stdout: "" }
+  }
+
+  return { $, calls }
+}
+
+afterEach(() => {
+  ;(globalThis as any).Bun.file = originalBunFile
+  ;(globalThis as any).Bun.write = originalBunWrite
+  globalThis.fetch = originalFetch
+  if (originalHome === undefined) {
+    delete (Bun.env as any).HOME
+  } else {
+    ;(Bun.env as any).HOME = originalHome
+  }
+})
+
+describe("index exports", () => {
+  test("all four named exports are functions", () => {
+    expect(typeof backgroundSubagentPlugin).toBe("function")
+    expect(typeof compactionPlugin).toBe("function")
+    expect(typeof leaderboardPlugin).toBe("function")
+    expect(typeof notifyPlugin).toBe("function")
+  })
+
+  test("leaderboardPlugin with default config returns object with event hook", async () => {
+    const home = "/tmp/oc-index-lb-default"
+    ;(Bun.env as any).HOME = home
+    ;(globalThis as any).Bun.write = async () => {}
+    const ocTweaksPath = `${home}/.config/opencode/oc-tweaks.json`
+    const leaderboardPath = `${home}/.claude/leaderboard.json`
+
+    mockBunFile({
+      [ocTweaksPath]: { leaderboard: { enabled: true } },
+      [leaderboardPath]: { twitter_handle: "test", twitter_user_id: "u1" },
+    })
+
+    const hooks = await leaderboardPlugin()
+    expect(typeof hooks).toBe("object")
+    expect(typeof hooks.event === "function" || Object.keys(hooks).length === 0).toBe(true)
+  })
+
+  test("leaderboardPlugin with enabled:false returns {}", async () => {
+    const home = "/tmp/oc-index-lb-disabled"
+    ;(Bun.env as any).HOME = home
+    const ocTweaksPath = `${home}/.config/opencode/oc-tweaks.json`
+    const leaderboardPath = `${home}/.claude/leaderboard.json`
+
+    mockBunFile({
+      [ocTweaksPath]: { leaderboard: { enabled: false } },
+      [leaderboardPath]: { twitter_handle: "test", twitter_user_id: "u1" },
+    })
+
+    const hooks = await leaderboardPlugin()
+    expect(hooks).toEqual({})
+  })
+
+  test("notifyPlugin with default config returns object with event hook", async () => {
+    const home = "/tmp/oc-index-notify-default"
+    ;(Bun.env as any).HOME = home
+    const path = `${home}/.config/opencode/oc-tweaks.json`
+    mockBunFile({ [path]: { notify: { enabled: true } } })
+
+    const { $ } = createShellMock({ availableCommands: ["notify-send"] })
+    const hooks = await notifyPlugin({ $, directory: "/tmp/demo", client: {} })
+    expect(typeof hooks).toBe("object")
+    expect(typeof hooks.event).toBe("function")
+  })
+
+  test("notifyPlugin with enabled:false returns {}", async () => {
+    const home = "/tmp/oc-index-notify-disabled"
+    ;(Bun.env as any).HOME = home
+    const path = `${home}/.config/opencode/oc-tweaks.json`
+    mockBunFile({ [path]: { notify: { enabled: false } } })
+
+    const { $ } = createShellMock({ availableCommands: ["notify-send"] })
+    const hooks = await notifyPlugin({ $, directory: "/tmp/demo", client: {} })
+    expect(hooks).toEqual({})
+  })
+
+  test("leaderboard and notify event handlers coexist without interference", async () => {
+    const home = "/tmp/oc-index-coexist"
+    ;(Bun.env as any).HOME = home
+    ;(globalThis as any).Bun.write = async () => {}
+    const ocTweaksPath = `${home}/.config/opencode/oc-tweaks.json`
+    const leaderboardPath = `${home}/.claude/leaderboard.json`
+
+    mockBunFile({
+      [ocTweaksPath]: { leaderboard: { enabled: true }, notify: { enabled: true } },
+      [leaderboardPath]: { twitter_handle: "coexist", twitter_user_id: "ux" },
+    })
+
+    globalThis.fetch = async () =>
+      ({ ok: true, status: 200, text: async () => "" }) as any
+
+    const lbHooks = await leaderboardPlugin()
+
+    const { $ } = createShellMock({ availableCommands: ["notify-send"] })
+    const notifyHooks = await notifyPlugin({ $, directory: "/tmp/coexist", client: {} })
+
+    expect(typeof lbHooks.event === "function" || Object.keys(lbHooks).length === 0).toBe(true)
+    expect(typeof notifyHooks.event).toBe("function")
+
+    // leaderboard event
+    if (typeof lbHooks.event === "function") {
+      await expect(
+        lbHooks.event({
+          event: {
+            type: "message.updated",
+            properties: {
+              info: {
+                id: "msg-coexist",
+                sessionID: "session-coexist",
+                role: "assistant",
+                time: { created: 1730000000000, completed: 1730000001000 },
+                modelID: "gpt-5.1-codex",
+                providerID: "provider",
+                cost: 0,
+                tokens: { input: 10, output: 5, reasoning: 0, cache: { read: 0, write: 0 } },
+              },
+            },
+          },
+        }),
+      ).resolves.toBeUndefined()
+    }
+
+    // notify event (session.error does not require client.session.messages)
+    await expect(
+      notifyHooks.event({ event: { type: "session.error", properties: {} } }),
+    ).resolves.toBeUndefined()
+  })
+})

package/src/__tests__/leaderboard.test.ts

@@ -0,0 +1,244 @@
+// @ts-nocheck
+
+import { afterEach, beforeEach, describe, expect, test } from "bun:test"
+
+const TEST_HOME = "/tmp/oc-leaderboard"
+
+const originalBunFile = Bun.file
+const originalBunWrite = Bun.write
+const originalFetch = globalThis.fetch
+const originalHome = Bun.env?.HOME
+
+function mockBunFile(mockData: Record<string, any>, existsCalls?: string[]) {
+  ;(globalThis as any).Bun.file = (path: string) => ({
+    exists: async () => {
+      existsCalls?.push(path)
+      return path in mockData
+    },
+    json: async () => {
+      if (!(path in mockData)) throw new Error("ENOENT")
+      const data = mockData[path]
+      if (data instanceof Error) throw data
+      return data
+    },
+    text: async () => {
+      if (!(path in mockData)) return ""
+      const value = mockData[path]
+      return typeof value === "string" ? value : JSON.stringify(value)
+    },
+  })
+}
+
+function buildAssistantInfo(overrides: Record<string, any> = {}) {
+  return {
+    id: "msg-1",
+    sessionID: "session-1",
+    role: "assistant",
+    time: { created: 1730000000000, completed: 1730000001000 },
+    modelID: "gpt-5.1-codex",
+    providerID: "provider",
+    cost: 0,
+    tokens: {
+      input: 120,
+      output: 80,
+      reasoning: 0,
+      cache: { read: 4, write: 2 },
+    },
+    ...overrides,
+  }
+}
+
+function buildMessageUpdatedEvent(infoOverrides: Record<string, any> = {}) {
+  return {
+    event: {
+      type: "message.updated",
+      properties: {
+        info: buildAssistantInfo(infoOverrides),
+      },
+    },
+  }
+}
+
+async function loadPlugin() {
+  const mod = await import("../plugins/leaderboard")
+  return mod.leaderboardPlugin
+}
+
+beforeEach(() => {
+  ;(Bun.env as any).HOME = TEST_HOME
+  ;(globalThis as any).Bun.write = async () => {}
+})
+
+afterEach(() => {
+  ;(globalThis as any).Bun.file = originalBunFile
+  ;(globalThis as any).Bun.write = originalBunWrite
+  globalThis.fetch = originalFetch
+
+  if (originalHome === undefined) {
+    delete (Bun.env as any).HOME
+  } else {
+    ;(Bun.env as any).HOME = originalHome
+  }
+})
+
+describe("leaderboardPlugin", () => {
+  test("mapModel covers direct, regex, and fallback branches", async () => {
+    const ocTweaksPath = `${TEST_HOME}/.config/opencode/oc-tweaks.json`
+    const leaderboardPath = `${TEST_HOME}/.claude/leaderboard.json`
+    const postedPayloads: any[] = []
+
+    mockBunFile({
+      [ocTweaksPath]: { leaderboard: { enabled: true } },
+      [leaderboardPath]: {
+        twitter_handle: "alice",
+        twitter_user_id: "u1",
+      },
+    })
+
+    globalThis.fetch = (async (_url: string, init: any) => {
+      postedPayloads.push(JSON.parse(init.body))
+      return {
+        ok: true,
+        status: 200,
+        text: async () => "",
+      } as any
+    }) as any
+
+    const leaderboardPlugin = await loadPlugin()
+    const hooks = await leaderboardPlugin()
+
+    await hooks.event(
+      buildMessageUpdatedEvent({
+        id: "msg-direct",
+        modelID: "gpt-5.1-codex",
+      }),
+    )
+    await hooks.event(
+      buildMessageUpdatedEvent({
+        id: "msg-regex",
+        modelID: "claude-opus-4-20260101",
+      }),
+    )
+    await hooks.event(
+      buildMessageUpdatedEvent({
+        id: "msg-fallback",
+        modelID: "totally-unknown-model",
+      }),
+    )
+
+    expect(postedPayloads.length).toBe(3)
+    expect(postedPayloads[0].model).toBe("claude-sonnet-4-20250514")
+    expect(postedPayloads[1].model).toBe("claude-opus-4-20260101")
+    expect(postedPayloads[2].model).toBe("claude-sonnet-4-20250514")
+  })
+
+  test("returns empty hooks when leaderboard is disabled", async () => {
+    const ocTweaksPath = `${TEST_HOME}/.config/opencode/oc-tweaks.json`
+    const leaderboardPath = `${TEST_HOME}/.claude/leaderboard.json`
+
+    mockBunFile({
+      [ocTweaksPath]: {
+        leaderboard: { enabled: false },
+      },
+      [leaderboardPath]: {
+        twitter_handle: "alice",
+        twitter_user_id: "u1",
+      },
+    })
+
+    const leaderboardPlugin = await loadPlugin()
+    const hooks = await leaderboardPlugin()
+
+    expect(hooks).toEqual({})
+  })
+
+  test("returns empty hooks when leaderboard config file is missing", async () => {
+    const ocTweaksPath = `${TEST_HOME}/.config/opencode/oc-tweaks.json`
+    mockBunFile({
+      [ocTweaksPath]: { leaderboard: { enabled: true } },
+    })
+
+    const leaderboardPlugin = await loadPlugin()
+    const hooks = await leaderboardPlugin()
+
+    expect(hooks).toEqual({})
+  })
+
+  test("loads valid leaderboard config from default search paths", async () => {
+    const ocTweaksPath = `${TEST_HOME}/.config/opencode/oc-tweaks.json`
+    const secondPath = `${TEST_HOME}/.config/claude/leaderboard.json`
+
+    mockBunFile({
+      [ocTweaksPath]: { leaderboard: { enabled: true } },
+      [secondPath]: {
+        twitter_handle: "bob",
+        twitter_user_id: "u2",
+      },
+    })
+
+    const leaderboardPlugin = await loadPlugin()
+    const hooks = await leaderboardPlugin()
+
+    expect(typeof hooks.event).toBe("function")
+  })
+
+  test("configPath override short-circuits default search", async () => {
+    const existsCalls: string[] = []
+    const ocTweaksPath = `${TEST_HOME}/.config/opencode/oc-tweaks.json`
+    const overridePath = "/tmp/custom/leaderboard.json"
+    const defaultPath1 = `${TEST_HOME}/.claude/leaderboard.json`
+    const defaultPath2 = `${TEST_HOME}/.config/claude/leaderboard.json`
+
+    mockBunFile(
+      {
+        [ocTweaksPath]: {
+          leaderboard: { enabled: true, configPath: overridePath },
+        },
+        [overridePath]: {
+          twitter_handle: "override",
+          twitter_user_id: "u3",
+        },
+      },
+      existsCalls,
+    )
+
+    const leaderboardPlugin = await loadPlugin()
+    const hooks = await leaderboardPlugin()
+
+    expect(typeof hooks.event).toBe("function")
+    expect(existsCalls).toContain(overridePath)
+    expect(existsCalls).not.toContain(defaultPath1)
+    expect(existsCalls).not.toContain(defaultPath2)
+  })
+
+  test("submit flow remains non-blocking and submitted set dedupes", async () => {
+    const ocTweaksPath = `${TEST_HOME}/.config/opencode/oc-tweaks.json`
+    const leaderboardPath = `${TEST_HOME}/.claude/leaderboard.json`
+    const fetchCalls: any[] = []
+
+    mockBunFile({
+      [ocTweaksPath]: { leaderboard: { enabled: true } },
+      [leaderboardPath]: {
+        twitter_handle: "alice",
+        twitter_user_id: "u1",
+      },
+    })
+
+    globalThis.fetch = (async () => {
+      fetchCalls.push(1)
+      return {
+        ok: false,
+        status: 500,
+        text: async () => "server error",
+      } as any
+    }) as any
+
+    const leaderboardPlugin = await loadPlugin()
+    const hooks = await leaderboardPlugin()
+    const eventInput = buildMessageUpdatedEvent({ id: "msg-dedupe" })
+
+    await expect(hooks.event(eventInput)).resolves.toBeUndefined()
+    await expect(hooks.event(eventInput)).resolves.toBeUndefined()
+    expect(fetchCalls.length).toBe(1)
+  })
+})

package/src/__tests__/logger.test.ts

@@ -0,0 +1,84 @@
+// @ts-nocheck
+
+import { describe, test, expect, afterEach } from "bun:test"
+import { log } from "../utils/logger"
+
+const originalBunFile = Bun.file
+const originalBunWrite = Bun.write
+const originalHome = Bun.env?.HOME
+
+afterEach(() => {
+  ;(globalThis as any).Bun.file = originalBunFile
+  ;(globalThis as any).Bun.write = originalBunWrite
+  if (originalHome === undefined) {
+    delete (Bun.env as any).HOME
+  } else {
+    ;(Bun.env as any).HOME = originalHome
+  }
+})
+
+describe("logger", () => {
+  test("does not write when logging is disabled", async () => {
+    let writeCount = 0
+    ;(globalThis as any).Bun.write = async () => {
+      writeCount++
+      return 0
+    }
+    ;(globalThis as any).Bun.file = () => ({
+      exists: async () => false,
+      text: async () => ""
+    })
+
+    await log({ enabled: false }, "INFO", "test message")
+    await log(undefined, "INFO", "test message")
+
+    expect(writeCount).toBe(0)
+  })
+
+  test("writes log when enabled", async () => {
+    const written: Record<string, string> = {}
+    ;(globalThis as any).Bun.write = async (path: string, content: string) => {
+      written[path] = content
+      return 0
+    }
+    ;(globalThis as any).Bun.file = (path: string) => ({
+      exists: async () => path in written,
+      text: async () => written[path] ?? ""
+    })
+    ;(Bun.env as any).HOME = "/tmp/oc-logger-test"
+
+    await log({ enabled: true }, "INFO", "hello world")
+
+    const logPath = "/tmp/oc-logger-test/.config/opencode/plugins/oc-tweaks.log"
+    expect(written[logPath]).toBeDefined()
+    expect(written[logPath]).toContain("[INFO] hello world")
+  })
+
+  test("truncates to keep lines when max exceeded", async () => {
+    const maxLines = 5
+    const keepLines = Math.floor(maxLines / 2) // 2
+
+    // Pre-fill with maxLines lines
+    const existingLines = Array.from({ length: maxLines }, (_, i) => `line${i}`)
+      .join("\n")
+      .concat("\n")
+    const written: Record<string, string> = {}
+    ;(globalThis as any).Bun.write = async (path: string, content: string) => {
+      written[path] = content
+      return 0
+    }
+    ;(globalThis as any).Bun.file = (path: string) => ({
+      exists: async () => true,
+      text: async () => existingLines
+    })
+    ;(Bun.env as any).HOME = "/tmp/oc-logger-truncate"
+
+    await log({ enabled: true, maxLines }, "INFO", "new line")
+
+    const logPath = "/tmp/oc-logger-truncate/.config/opencode/plugins/oc-tweaks.log"
+    const lines = (written[logPath] ?? "")
+      .split("\n")
+      .filter((l: string) => l.length > 0)
+    expect(lines.length).toBeLessThanOrEqual(keepLines)
+  })
+})