84 lines
2.4 KiB
TypeScript
84 lines
2.4 KiB
TypeScript
import { existsSync, readdirSync, readFileSync } from "node:fs"
|
|
import { join } from "node:path"
|
|
import type { PluginInput } from "@opencode-ai/plugin"
|
|
import type { StoredMessageMeta } from "../types"
|
|
import { getMessageDir } from "./message-dir"
|
|
import { isSqliteBackend, normalizeSDKResponse } from "../../../shared"
|
|
import { isRecord } from "../../../shared/record-type-guard"
|
|
|
|
type OpencodeClient = PluginInput["client"]
|
|
|
|
function normalizeSDKMessage(
|
|
sessionID: string,
|
|
value: unknown
|
|
): StoredMessageMeta | null {
|
|
if (!isRecord(value)) return null
|
|
if (typeof value.id !== "string") return null
|
|
|
|
const roleValue = value.role
|
|
const role: StoredMessageMeta["role"] = roleValue === "assistant" ? "assistant" : "user"
|
|
|
|
const created =
|
|
isRecord(value.time) && typeof value.time.created === "number"
|
|
? value.time.created
|
|
: 0
|
|
|
|
return {
|
|
id: value.id,
|
|
sessionID,
|
|
role,
|
|
time: { created },
|
|
}
|
|
}
|
|
|
|
export function readMessages(sessionID: string): StoredMessageMeta[] {
|
|
if (isSqliteBackend()) return []
|
|
|
|
const messageDir = getMessageDir(sessionID)
|
|
if (!messageDir || !existsSync(messageDir)) return []
|
|
|
|
const messages: StoredMessageMeta[] = []
|
|
for (const file of readdirSync(messageDir)) {
|
|
if (!file.endsWith(".json")) continue
|
|
try {
|
|
const content = readFileSync(join(messageDir, file), "utf-8")
|
|
messages.push(JSON.parse(content))
|
|
} catch {
|
|
continue
|
|
}
|
|
}
|
|
|
|
return messages.sort((a, b) => {
|
|
const aTime = a.time?.created ?? 0
|
|
const bTime = b.time?.created ?? 0
|
|
if (aTime !== bTime) return aTime - bTime
|
|
return a.id.localeCompare(b.id)
|
|
})
|
|
}
|
|
|
|
export async function readMessagesFromSDK(
|
|
client: OpencodeClient,
|
|
sessionID: string
|
|
): Promise<StoredMessageMeta[]> {
|
|
try {
|
|
const response = await client.session.messages({ path: { id: sessionID } })
|
|
const data = normalizeSDKResponse(response, [] as unknown[], {
|
|
preferResponseOnMissingData: true,
|
|
})
|
|
if (!Array.isArray(data)) return []
|
|
|
|
const messages = data
|
|
.map((msg): StoredMessageMeta | null => normalizeSDKMessage(sessionID, msg))
|
|
.filter((msg): msg is StoredMessageMeta => msg !== null)
|
|
|
|
return messages.sort((a, b) => {
|
|
const aTime = a.time?.created ?? 0
|
|
const bTime = b.time?.created ?? 0
|
|
if (aTime !== bTime) return aTime - bTime
|
|
return a.id.localeCompare(b.id)
|
|
})
|
|
} catch {
|
|
return []
|
|
}
|
|
}
|