SessionRevert

export namespace SessionRevert {
  export const RevertInput = z.object({
    sessionID: string,
    messageID: string,
    partID?: string,
  })

  export async function revert(input: RevertInput): Promise<Session.Info>

  export async function unrevert(input: { sessionID: string }): Promise<Session.Info>

  export async function cleanup(session: Session.Info): Promise<void>
}

export type Session = {
  // ... 其他字段
  revert?: {
    messageID: string // 要回滚到的消息 ID
    partID?: string // 要回滚到的部分 ID（可选）
    snapshot?: string // 快照 ID（可选）
    diff?: string // 差异字符串（可选）
  }
}

const hash = await Snapshot.track()
// 返回 Git tree hash

await Snapshot.restore(hash)
// 恢复工作目录到指定的快照状态

await Snapshot.revert(patches)
// 只回滚指定的文件更改

const diff = await Snapshot.diff(hash)
// 返回差异信息

// 撤销按钮
await sdk.client.session.revert({
  sessionID: session.id,
  messageID: msg.id,
  partID: part.id,
})

// 重做按钮
await sdk.client.session.unrevert({
  sessionID: session.id,
})

// 消息重做快捷键
messages_redo: {
  keys: ["ctrl+r"],
  action: () => {
    sdk.client.session.unrevert({ sessionID })
  }
}

// 用户想要撤销从某个消息开始的所有更改
await SessionRevert.revert({
  sessionID: "sess_abc123",
  messageID: "msg_xyz789",
})

// 会话现在处于回滚状态
// 文件系统已恢复到该消息之前的状态

// 用户想要撤销部分消息内的更改
await SessionRevert.revert({
  sessionID: "sess_abc123",
  messageID: "msg_xyz789",
  partID: "part_def456",
})

// 只回滚该部分之后的更改部分

// 用户改变主意，想要恢复所有更改
await SessionRevert.unrevert({
  sessionID: "sess_abc123",
})

// 文件系统恢复到最新状态
// 所有消息再次可见

// 当创建新提示词时自动调用
const session = await Session.get("sess_abc123")
await SessionRevert.cleanup(session)

// 回滚点之后的消息被永久删除

const session = await Session.get("sess_abc123")

if (session.revert) {
  console.log("会话处于回滚状态")
  console.log(`回滚到消息: ${session.revert.messageID}`)
  if (session.revert.partID) {
    console.log(`  部分: ${session.revert.partID}`)
  }
  if (session.revert.snapshot) {
    console.log(`  快照: ${session.revert.snapshot}`)
  }
  if (session.revert.diff) {
    console.log(`  差异: ${session.revert.diff}`)
  }
}

// 回滚时自动计算的差异
const session = await Session.get("sess_abc123")

if (session.revert && session.summary) {
  console.log(`回滚统计:`)
  console.log(`  新增行数: ${session.summary.additions}`)
  console.log(`  删除行数: ${session.summary.deletions}`)
  console.log(`  修改文件数: ${session.summary.files}`)
}

// 从存储读取详细差异
const diffs = await Storage.read(["session_diff", "sess_abc123"])
console.log(diffs)

;["session_diff", sessionID] = FileDiff[]