feat(id): brand SessionID through Drizzle and Zod schemas (#16953)

This commit is contained in:
Kit Langton
2026-03-11 19:16:56 -04:00
committed by GitHub
parent 4e73473119
commit cb67465675
44 changed files with 226 additions and 158 deletions

View File

@@ -1,6 +1,7 @@
import type { Argv } from "yargs"
import type { Session as SDKSession, Message, Part } from "@opencode-ai/sdk/v2"
import { Session } from "../../session"
import { SessionID } from "../../session/schema"
import { cmd } from "./cmd"
import { bootstrap } from "../bootstrap"
import { Database } from "../../storage/db"
@@ -152,7 +153,12 @@ export const ImportCommand = cmd({
return
}
const row = Session.toRow({ ...exportData.info, projectID: Instance.project.id })
const row = Session.toRow({
...exportData.info,
id: SessionID.make(exportData.info.id),
parentID: exportData.info.parentID ? SessionID.make(exportData.info.parentID) : undefined,
projectID: Instance.project.id,
})
Database.use((db) =>
db
.insert(SessionTable)
@@ -167,7 +173,7 @@ export const ImportCommand = cmd({
.insert(MessageTable)
.values({
id: msg.info.id,
session_id: exportData.info.id,
session_id: row.id,
time_created: msg.info.time?.created ?? Date.now(),
data: msg.info,
})
@@ -182,7 +188,7 @@ export const ImportCommand = cmd({
.values({
id: part.id,
message_id: msg.info.id,
session_id: exportData.info.id,
session_id: row.id,
data: part,
})
.onConflictDoNothing()