37 lines
950 B
TypeScript
37 lines
950 B
TypeScript
|
import { existsSync, readdirSync } from "node:fs"
|
||
|
|
import { join } from "node:path"
|
||
|
|
|
||
|
|
import { MESSAGE_STORAGE_DIR } from "./storage-paths"
|
||
|
|
|
||
|
|
export function getMessageDir(sessionID: string): string {
|
||
|
|
if (!existsSync(MESSAGE_STORAGE_DIR)) return ""
|
||
|
|
|
||
|
|
const directPath = join(MESSAGE_STORAGE_DIR, sessionID)
|
||
|
|
if (existsSync(directPath)) {
|
||
|
|
return directPath
|
||
|
|
}
|
||
|
|
|
||
|
|
for (const directory of readdirSync(MESSAGE_STORAGE_DIR)) {
|
||
|
|
const sessionPath = join(MESSAGE_STORAGE_DIR, directory, sessionID)
|
||
|
|
if (existsSync(sessionPath)) {
|
||
|
|
return sessionPath
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return ""
|
||
|
|
}
|
||
|
|
|
||
|
|
export function getMessageIds(sessionID: string): string[] {
|
||
|
|
const messageDir = getMessageDir(sessionID)
|
||
|
|
if (!messageDir || !existsSync(messageDir)) return []
|
||
|
|
|
||
|
|
const messageIds: string[] = []
|
||
|
|
for (const file of readdirSync(messageDir)) {
|
||
|
|
if (!file.endsWith(".json")) continue
|
||
|
|
const messageId = file.replace(".json", "")
|
||
|
|
messageIds.push(messageId)
|
||
|
|
}
|
||
|
|
|
||
|
|
return messageIds
|
||
|
|
}
|