@astralyn/pi 0.82.3 → 0.82.5

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (158) hide show
  1. package/CHANGELOG.md +49 -0
  2. package/dist/cli/config-selector.d.ts.map +1 -1
  3. package/dist/cli/config-selector.js +3 -2
  4. package/dist/cli/config-selector.js.map +1 -1
  5. package/dist/cli/startup-ui.d.ts.map +1 -1
  6. package/dist/cli/startup-ui.js +3 -2
  7. package/dist/cli/startup-ui.js.map +1 -1
  8. package/dist/core/agent-session.d.ts +2 -0
  9. package/dist/core/agent-session.d.ts.map +1 -1
  10. package/dist/core/agent-session.js +33 -14
  11. package/dist/core/agent-session.js.map +1 -1
  12. package/dist/core/extensions/index.d.ts +2 -2
  13. package/dist/core/extensions/index.d.ts.map +1 -1
  14. package/dist/core/extensions/index.js +1 -1
  15. package/dist/core/extensions/index.js.map +1 -1
  16. package/dist/core/extensions/loader.d.ts +1 -1
  17. package/dist/core/extensions/loader.d.ts.map +1 -1
  18. package/dist/core/extensions/loader.js +2 -3
  19. package/dist/core/extensions/loader.js.map +1 -1
  20. package/dist/core/extensions/types.d.ts +18 -0
  21. package/dist/core/extensions/types.d.ts.map +1 -1
  22. package/dist/core/extensions/types.js +11 -0
  23. package/dist/core/extensions/types.js.map +1 -1
  24. package/dist/extensions/index.d.ts.map +1 -1
  25. package/dist/extensions/index.js +2 -0
  26. package/dist/extensions/index.js.map +1 -1
  27. package/dist/extensions/plan/constants.d.ts +31 -0
  28. package/dist/extensions/plan/constants.d.ts.map +1 -0
  29. package/dist/extensions/plan/constants.js +57 -0
  30. package/dist/extensions/plan/constants.js.map +1 -0
  31. package/dist/extensions/plan/index.d.ts +21 -0
  32. package/dist/extensions/plan/index.d.ts.map +1 -0
  33. package/dist/extensions/plan/index.js +291 -0
  34. package/dist/extensions/plan/index.js.map +1 -0
  35. package/dist/extensions/plan/schema.d.ts +15 -0
  36. package/dist/extensions/plan/schema.d.ts.map +1 -0
  37. package/dist/extensions/plan/schema.js +13 -0
  38. package/dist/extensions/plan/schema.js.map +1 -0
  39. package/dist/extensions/plan/state.d.ts +29 -0
  40. package/dist/extensions/plan/state.d.ts.map +1 -0
  41. package/dist/extensions/plan/state.js +74 -0
  42. package/dist/extensions/plan/state.js.map +1 -0
  43. package/dist/extensions/plan/storage.d.ts +18 -0
  44. package/dist/extensions/plan/storage.d.ts.map +1 -0
  45. package/dist/extensions/plan/storage.js +58 -0
  46. package/dist/extensions/plan/storage.js.map +1 -0
  47. package/dist/extensions/question/dialog.d.ts +3 -1
  48. package/dist/extensions/question/dialog.d.ts.map +1 -1
  49. package/dist/extensions/question/dialog.js +106 -52
  50. package/dist/extensions/question/dialog.js.map +1 -1
  51. package/dist/extensions/question/index.d.ts.map +1 -1
  52. package/dist/extensions/question/index.js +2 -2
  53. package/dist/extensions/question/index.js.map +1 -1
  54. package/dist/extensions/question/render-cache.d.ts +9 -0
  55. package/dist/extensions/question/render-cache.d.ts.map +1 -1
  56. package/dist/extensions/question/render-cache.js +22 -0
  57. package/dist/extensions/question/render-cache.js.map +1 -1
  58. package/dist/extensions/question/results.d.ts.map +1 -1
  59. package/dist/extensions/question/results.js +6 -1
  60. package/dist/extensions/question/results.js.map +1 -1
  61. package/dist/extensions/question/schema.d.ts +4 -0
  62. package/dist/extensions/question/schema.d.ts.map +1 -1
  63. package/dist/extensions/question/schema.js +19 -28
  64. package/dist/extensions/question/schema.js.map +1 -1
  65. package/dist/extensions/question/types.d.ts +1 -1
  66. package/dist/extensions/question/types.d.ts.map +1 -1
  67. package/dist/extensions/question/types.js.map +1 -1
  68. package/dist/extensions/rewind/anchor.d.ts +29 -0
  69. package/dist/extensions/rewind/anchor.d.ts.map +1 -0
  70. package/dist/extensions/rewind/anchor.js +35 -0
  71. package/dist/extensions/rewind/anchor.js.map +1 -0
  72. package/dist/extensions/rewind/engine.d.ts +2 -6
  73. package/dist/extensions/rewind/engine.d.ts.map +1 -1
  74. package/dist/extensions/rewind/engine.js +41 -10
  75. package/dist/extensions/rewind/engine.js.map +1 -1
  76. package/dist/extensions/rewind/index.d.ts.map +1 -1
  77. package/dist/extensions/rewind/index.js +55 -21
  78. package/dist/extensions/rewind/index.js.map +1 -1
  79. package/dist/extensions/rewind/menu.d.ts.map +1 -1
  80. package/dist/extensions/rewind/menu.js +7 -2
  81. package/dist/extensions/rewind/menu.js.map +1 -1
  82. package/dist/extensions/rewind/restore.d.ts +26 -9
  83. package/dist/extensions/rewind/restore.d.ts.map +1 -1
  84. package/dist/extensions/rewind/restore.js +58 -16
  85. package/dist/extensions/rewind/restore.js.map +1 -1
  86. package/dist/extensions/subagent/activity.d.ts +1 -0
  87. package/dist/extensions/subagent/activity.d.ts.map +1 -1
  88. package/dist/extensions/subagent/activity.js +20 -4
  89. package/dist/extensions/subagent/activity.js.map +1 -1
  90. package/dist/extensions/subagent/agents.d.ts.map +1 -1
  91. package/dist/extensions/subagent/agents.js +13 -18
  92. package/dist/extensions/subagent/agents.js.map +1 -1
  93. package/dist/extensions/subagent/constants.d.ts +5 -4
  94. package/dist/extensions/subagent/constants.d.ts.map +1 -1
  95. package/dist/extensions/subagent/constants.js +12 -5
  96. package/dist/extensions/subagent/constants.js.map +1 -1
  97. package/dist/extensions/subagent/index.d.ts.map +1 -1
  98. package/dist/extensions/subagent/index.js +12 -5
  99. package/dist/extensions/subagent/index.js.map +1 -1
  100. package/dist/extensions/subagent/render.d.ts.map +1 -1
  101. package/dist/extensions/subagent/render.js +166 -41
  102. package/dist/extensions/subagent/render.js.map +1 -1
  103. package/dist/extensions/subagent/resolve.d.ts +2 -5
  104. package/dist/extensions/subagent/resolve.d.ts.map +1 -1
  105. package/dist/extensions/subagent/resolve.js +8 -13
  106. package/dist/extensions/subagent/resolve.js.map +1 -1
  107. package/dist/extensions/subagent/runner.d.ts +3 -1
  108. package/dist/extensions/subagent/runner.d.ts.map +1 -1
  109. package/dist/extensions/subagent/runner.js +115 -66
  110. package/dist/extensions/subagent/runner.js.map +1 -1
  111. package/dist/extensions/subagent/schema.d.ts +0 -6
  112. package/dist/extensions/subagent/schema.d.ts.map +1 -1
  113. package/dist/extensions/subagent/schema.js +3 -4
  114. package/dist/extensions/subagent/schema.js.map +1 -1
  115. package/dist/extensions/subagent/sdk-runner.d.ts.map +1 -1
  116. package/dist/extensions/subagent/sdk-runner.js +37 -16
  117. package/dist/extensions/subagent/sdk-runner.js.map +1 -1
  118. package/dist/extensions/subagent/types.d.ts +10 -2
  119. package/dist/extensions/subagent/types.d.ts.map +1 -1
  120. package/dist/extensions/subagent/types.js.map +1 -1
  121. package/dist/extensions/todo/constants.d.ts +1 -1
  122. package/dist/extensions/todo/constants.d.ts.map +1 -1
  123. package/dist/extensions/todo/constants.js +29 -4
  124. package/dist/extensions/todo/constants.js.map +1 -1
  125. package/dist/extensions/todo/index.d.ts +1 -1
  126. package/dist/extensions/todo/index.d.ts.map +1 -1
  127. package/dist/extensions/todo/index.js +22 -3
  128. package/dist/extensions/todo/index.js.map +1 -1
  129. package/dist/extensions/todo/schema.d.ts +6 -5
  130. package/dist/extensions/todo/schema.d.ts.map +1 -1
  131. package/dist/extensions/todo/schema.js +20 -9
  132. package/dist/extensions/todo/schema.js.map +1 -1
  133. package/dist/extensions/todo/state.d.ts +5 -1
  134. package/dist/extensions/todo/state.d.ts.map +1 -1
  135. package/dist/extensions/todo/state.js +184 -38
  136. package/dist/extensions/todo/state.js.map +1 -1
  137. package/dist/extensions/todo/view.d.ts.map +1 -1
  138. package/dist/extensions/todo/view.js +10 -6
  139. package/dist/extensions/todo/view.js.map +1 -1
  140. package/dist/modes/interactive/interactive-mode.d.ts +2 -0
  141. package/dist/modes/interactive/interactive-mode.d.ts.map +1 -1
  142. package/dist/modes/interactive/interactive-mode.js +36 -2
  143. package/dist/modes/interactive/interactive-mode.js.map +1 -1
  144. package/dist/utils/coalescing-terminal.d.ts +116 -0
  145. package/dist/utils/coalescing-terminal.d.ts.map +1 -0
  146. package/dist/utils/coalescing-terminal.js +256 -0
  147. package/dist/utils/coalescing-terminal.js.map +1 -0
  148. package/docs/bundled/extensions/README.md +1 -0
  149. package/docs/bundled/extensions/plan.md +31 -0
  150. package/docs/bundled/extensions/question.md +9 -7
  151. package/docs/bundled/extensions/rewind.md +19 -8
  152. package/docs/bundled/extensions/subagent.md +16 -16
  153. package/docs/bundled/extensions/todo.md +5 -3
  154. package/docs/compaction.md +6 -2
  155. package/docs/docs.json +4 -0
  156. package/docs/extensions.md +4 -2
  157. package/npm-shrinkwrap.json +2 -2
  158. package/package.json +1 -1
@@ -1 +1 @@
1
- {"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/extensions/rewind/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA6BG;AACH,OAAO,IAAI,MAAM,WAAW,CAAC;AAK7B,OAAO,EAAE,gBAAgB,EAAqB,kBAAkB,EAAE,MAAM,aAAa,CAAC;AACtF,OAAO,EACN,SAAS,EACT,cAAc,EACd,OAAO,EACP,YAAY,EACZ,yBAAyB,EACzB,eAAe,EACf,yBAAyB,EACzB,SAAS,GACT,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,KAAK,EAAE,iBAAiB,EAAE,MAAM,SAAS,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,MAAM,WAAW,CAAC;AAC1C,OAAO,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAClE,OAAO,EAAE,iBAAiB,EAAE,uBAAuB,EAAE,oBAAoB,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AAClH,OAAO,EAA4B,UAAU,EAAE,mBAAmB,EAAE,MAAM,eAAe,CAAC;AAC1F,OAAO,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AAChD,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,EAAE,mBAAmB,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAEtE,6EAA6E;AAC7E,iFAAiF;AACjF,gBAAgB,CAAC,EAAE,WAAW,EAAE,iBAAiB,EAAE,EAAE,YAAY,EAAE,iBAAiB,EAAE,EAAE,CAAC,CAAC;AAE1F,8EAA8E;AAC9E,6EAA6E;AAC7E,wDAAwD;AACxD,IAAI,MAAM,GAAiB,gBAAgB,EAAE,CAAC;AAQ9C,6DAA6D;AAC7D,MAAM,aAAa,GAAG,IAAI,GAAG,EAAkB,CAAC,CAAC,+CAA+C;AAChG,MAAM,kBAAkB,GAAG,IAAI,GAAG,EAAqC,CAAC,CAAC,oBAAoB;AAE7F,8EAA8E;AAE9E,iFAAiF;AACjF,SAAS,eAAe,CAAC,GAAqB,EAAsB;IACnE,MAAM,MAAM,GAAG,GAAG,CAAC,cAAc,CAAC,SAAS,EAAE,CAAC;IAC9C,KAAK,IAAI,CAAC,GAAG,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;QAC7C,MAAM,CAAC,GAAG,MAAM,CAAC,CAAC,CAAC,CAAC;QACpB,IAAI,CAAC,CAAC,IAAI,KAAK,SAAS,IAAK,CAAC,CAAC,OAAwB,CAAC,IAAI,KAAK,MAAM;YAAE,OAAO,CAAC,CAAC,EAAE,CAAC;IACtF,CAAC;IACD,OAAO,SAAS,CAAC;AAAA,CACjB;AAED,iFAAiF;AACjF,SAAS,gBAAgB,CAAC,GAAqB,EAAyB;IACvE,MAAM,GAAG,GAA0B,EAAE,CAAC;IACtC,KAAK,MAAM,KAAK,IAAI,GAAG,CAAC,cAAc,CAAC,UAAU,EAAE,EAAE,CAAC;QACrD,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,IAAI,KAAK,CAAC,UAAU,KAAK,mBAAmB,IAAI,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;YACnG,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACtB,CAAC;IACF,CAAC;IACD,OAAO,GAAG,CAAC;AAAA,CACX;AAED,2EAA2E;AAC3E,MAAM,qBAAqB,GAAG,CAAC,CAAC;AAChC,wFAAwF;AACxF,MAAM,wBAAwB,GAAG,EAAE,CAAC;AAEpC;;;;;GAKG;AACH,SAAS,oBAAoB,CAAC,YAAsB,EAAE,GAAW,EAAU;IAC1E,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,qBAAqB,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;QACrE,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;QAClC,MAAM,OAAO,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAChF,MAAM,OAAO,GACZ,OAAO,CAAC,MAAM,GAAG,wBAAwB,CAAC,CAAC,CAAC,MAAI,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,wBAAwB,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;QAC5G,OAAO,KAAK,OAAO,EAAE,CAAC;IAAA,CACtB,CAAC,CAAC;IACH,IAAI,YAAY,CAAC,MAAM,GAAG,qBAAqB,EAAE,CAAC;QACjD,KAAK,CAAC,IAAI,CAAC,UAAQ,YAAY,CAAC,MAAM,GAAG,qBAAqB,OAAO,CAAC,CAAC;IACxE,CAAC;IACD,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAAA,CACxB;AAED,8EAA8E;AAE9E,MAAM,CAAC,OAAO,UAAU,MAAM,CAAC,EAAgB,EAAQ;IACtD,gFAAgF;IAChF,EAAE,CAAC,EAAE,CAAC,eAAe,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC;QAC5C,MAAM,GAAG,kBAAkB,EAAE,CAAC;QAC9B,MAAM,GAAG,GAAG,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,CAAC;QAC9C,IAAI,CAAC,GAAG;YAAE,OAAO;QACjB,eAAe,CAAC,GAAG,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC;QAE9B,MAAM,SAAS,GAAG,gBAAgB,CAAC,GAAG,CAAC,CAAC;QACxC,yBAAyB,CAAC,GAAG,EAAE,GAAG,CAAC,GAAG,EAAE,SAAS,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC;QAExE,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,QAAQ,IAAI,KAAK,CAAC,MAAM,KAAK,MAAM,CAAC,IAAI,KAAK,CAAC,mBAAmB,EAAE,CAAC;YACzF,MAAM,OAAO,GAAG,iBAAiB,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;YAC7D,IAAI,OAAO,EAAE,CAAC;gBACb,IAAI,CAAC;oBACJ,MAAM,yBAAyB,CAAC,OAAO,EAAE,GAAG,EAAE,SAAS,CAAC,CAAC;gBAC1D,CAAC;gBAAC,MAAM,CAAC;oBACR,oEAAoE;gBACrE,CAAC;YACF,CAAC;QACF,CAAC;QAED,IAAI,CAAC;YACJ,KAAK,CAAC,MAAM,CAAC,aAAa,EAAE,GAAG,CAAC,CAAC;QAClC,CAAC;QAAC,MAAM,CAAC;YACR,yCAAyC;QAC1C,CAAC;IAAA,CACD,CAAC,CAAC;IAEH,8EAA8E;IAC9E,8EAA8E;IAC9E,EAAE,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC;QAClC,IAAI,KAAK,CAAC,QAAQ,KAAK,MAAM,IAAI,KAAK,CAAC,QAAQ,KAAK,OAAO;YAAE,OAAO;QACpE,IAAI,CAAC,MAAM,CAAC,OAAO;YAAE,OAAO;QAC5B,MAAM,GAAG,GAAG,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,CAAC;QAC9C,IAAI,CAAC,GAAG;YAAE,OAAO;QACjB,MAAM,OAAO,GAAG,mBAAmB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACjD,IAAI,CAAC,OAAO;YAAE,OAAO;QACrB,MAAM,GAAG,GAAG,eAAe,CAAC,OAAO,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC;QAC9C,IAAI,CAAC;YACJ,SAAS,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACrB,CAAC;QAAC,MAAM,CAAC;YACR,2CAA2C;QAC5C,CAAC;IAAA,CACD,CAAC,CAAC;IAEH,sDAAsD;IACtD,EAAE,CAAC,EAAE,CAAC,oBAAoB,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC;QACjD,yEAAyE;QACzE,MAAM,GAAG,gBAAgB,EAAE,CAAC;QAC5B,IAAI,CAAC,MAAM,CAAC,OAAO;YAAE,OAAO;QAC5B,MAAM,GAAG,GAAG,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,CAAC;QAC9C,IAAI,CAAC,GAAG;YAAE,OAAO;QACjB,aAAa,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;QAC3C,IAAI,CAAC;YACJ,MAAM,SAAS,CAAC,GAAG,CAAC,CAAC;QACtB,CAAC;QAAC,MAAM,CAAC;YACR,YAAY;QACb,CAAC;IAAA,CACD,CAAC,CAAC;IAEH,6EAA6E;IAC7E,qEAAqE;IACrE,EAAE,CAAC,EAAE,CAAC,eAAe,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC;QAC7C,IAAI,CAAC,MAAM,CAAC,OAAO;YAAE,OAAO;QAC5B,MAAM,GAAG,GAAG,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,CAAC;QAC9C,IAAI,CAAC,GAAG;YAAE,OAAO;QACjB,MAAM,WAAW,GAAG,eAAe,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC;QAC/C,MAAM,MAAM,GAAG,GAAG,CAAC,cAAc,CAAC,SAAS,EAAE,IAAI,WAAW,CAAC;QAC7D,MAAM,MAAM,GAAG,YAAY,CAAC,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,GAAG,EAAE,EAAE,kBAAkB,EAAE,IAAI,EAAE,CAAC,CAAC;QAC7F,aAAa,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC1B,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,EAAE,WAAW,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC;QACvG,IAAI,KAAK,IAAI,WAAW,EAAE,CAAC;YAC1B,EAAE,CAAC,WAAW,CAAC,mBAAmB,EAAE,KAAK,CAAC,CAAC;QAC5C,CAAC;IAAA,CACD,CAAC,CAAC;IAEH,+EAA+E;IAC/E,EAAE,CAAC,EAAE,CAAC,qBAAqB,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC;QAClD,MAAM,GAAG,GAAG,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,CAAC;QAC9C,IAAI,CAAC,GAAG;YAAE,OAAO;QACjB,kBAAkB,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QAClC,IAAI,CAAC,MAAM,CAAC,OAAO;YAAE,OAAO;QAE5B,MAAM,MAAM,GAAG,gBAAgB,CAAC,YAAY,CAAC,GAAG,CAAC,EAAE,GAAG,CAAC,cAAc,EAAE,KAAK,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QACnG,IAAI,CAAC,MAAM;YAAE,OAAO;QACpB,MAAM,OAAO,GAAG,MAAM,oBAAoB,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACxD,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,CAAC,gCAAgC;QAClE,uEAAuE;QACvE,oEAAoE;QACpE,6BAA6B;QAC7B,IAAI,CAAC,GAAG,CAAC,KAAK;YAAE,OAAO;QAEvB,MAAM,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;QACzB,IAAI,SAAS,GAAG,EAAE,CAAC;QACnB,IAAI,CAAC;YACJ,MAAM,KAAK,GAAG,MAAM,uBAAuB,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;YAClE,IAAI,KAAK,CAAC,UAAU,GAAG,CAAC,IAAI,KAAK,CAAC,SAAS,GAAG,CAAC,EAAE,CAAC;gBACjD,SAAS,GAAG,OAAO,KAAK,CAAC,UAAU,SAAO,KAAK,CAAC,SAAS,GAAG,CAAC;YAC9D,CAAC;QACF,CAAC;QAAC,MAAM,CAAC;YACR,0DAA0D;QAC3D,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,MAAM,CACjC,WAAW,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,kBAAkB,SAAS,KAAK,oBAAoB,CAAC,OAAO,EAAE,GAAG,CAAC,GAAG,CAAC,EAAE,EAC9G,CAAC,oBAAoB,EAAE,uBAAuB,CAAC,CAC/C,CAAC;QACF,IAAI,MAAM,EAAE,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YAC/B,uEAAuE;YACvE,kBAAkB,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,YAAY,EAAE,OAAO,EAAE,CAAC,CAAC;QAC1E,CAAC;IAAA,CACD,CAAC,CAAC;IAEH,wDAAwD;IACxD,EAAE,CAAC,EAAE,CAAC,cAAc,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC;QAC5C,MAAM,GAAG,GAAG,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,CAAC;QAC9C,IAAI,CAAC,GAAG;YAAE,OAAO;QACjB,MAAM,OAAO,GAAG,kBAAkB,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC5C,kBAAkB,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QAClC,IAAI,CAAC,OAAO;YAAE,OAAO;QACrB,IAAI,CAAC;YACJ,MAAM,OAAO,GAAG,MAAM,iBAAiB,CAAC,GAAG,EAAE,OAAO,CAAC,QAAQ,EAAE;gBAC9D,SAAS,EAAE,IAAI,GAAG,CAAC,OAAO,CAAC,YAAY,CAAC;aACxC,CAAC,CAAC;YACH,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACxB,GAAG,CAAC,EAAE,CAAC,MAAM,CACZ,YAAY,OAAO,CAAC,MAAM,QAAQ,OAAO,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,sBAAsB,EACvF,MAAM,CACN,CAAC;YACH,CAAC;QACF,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACZ,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,0BAA0B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;QACjE,CAAC;IAAA,CACD,CAAC,CAAC;IAEH,yDAAyD;IACzD,EAAE,CAAC,EAAE,CAAC,kBAAkB,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC;QAChD,MAAM,GAAG,GAAG,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,CAAC;QAC9C,IAAI,CAAC,GAAG;YAAE,OAAO;QACjB,cAAc,CAAC,GAAG,CAAC,CAAC;QACpB,kBAAkB,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC/B,aAAa,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IAAA,CAC1B,CAAC,CAAC;IAEH,sEAAsE;IACtE,EAAE,CAAC,eAAe,CAAC,QAAQ,EAAE;QAC5B,WAAW,EAAE,2DAA2D;QACxE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC;YAC9B,MAAM,aAAa,CAAC,GAAG,CAAC,CAAC;QAAA,CACzB;KACD,CAAC,CAAC;AAAA,CACH","sourcesContent":["/**\n * rewind — file-backed checkpoint & rewind for Pi.\n *\n * Engine: instead of snapshotting the whole work tree into a per-session shadow\n * git repo (the old design, which froze the UI in large directories like a\n * multi-project parent folder and grew storage without bound), we back up ONLY\n * the files Pi's edit/write tools are about to modify — one copyFile before each\n * edit (see engine.ts, ported from Claude Code's file-history). Cost is\n * proportional to how many files Pi changed, never to project size.\n *\n * Per turn: before_agent_start opens a snapshot frame (re-recording tracked files\n * at their turn-start state); tool_call(edit|write) backs up each newly edited\n * file before it lands; agent_settled persists the frame to the session JSONL\n * (custom \"pi-rewind-snapshot\" entry) when it changed. agent_settled is used\n * instead of agent_end so auto-retry, overflow compaction-retry, and queued\n * follow-ups stay in one logical turn (agent_end can fire mid-continuation).\n * On session_start the index is rebuilt from those entries; resume/fork\n * hard-links the prior session's blobs.\n *\n * Time-travel is fused into /tree: navigating to a node whose turn changed files\n * offers to sync the work tree to that point (session_before_tree/session_tree).\n * /rewind itself is a settings + storage menu (menu.ts), not a restore picker.\n *\n * Restore safety: applySnapshot only rewrites files that differ and never throws\n * out — a broken backup degrades to \"leave the file alone\", so it can never abort\n * the user's session.\n *\n * Architecture informed by oh-my-pi (GPL-3.0) and Claude Code's file-history;\n * independent implementation. No ANSI: all UI is native (ctx.ui.* + theme).\n */\nimport path from \"node:path\";\n\nimport type { AgentMessage } from \"@earendil-works/pi-agent-core\";\nimport type { ExtensionAPI, ExtensionContext } from \"../../core/extensions/types.ts\";\n\nimport { loadRewindConfig, type RewindConfig, reloadRewindConfig } from \"./config.ts\";\nimport {\n\tbeginTurn,\n\tdisposeSession,\n\tendTurn,\n\tgetSnapshots,\n\tmigrateBackupsFromSession,\n\tregisterSession,\n\trestoreStateFromSnapshots,\n\ttrackEdit,\n} from \"./engine.ts\";\nimport { runGc, sessionIdFromFile } from \"./gc.ts\";\nimport { runRewindMenu } from \"./menu.ts\";\nimport { rewindBackupsRoot, sessionsDirectory } from \"./paths.ts\";\nimport { restoreToSnapshot, snapshotChangeDiffStats, snapshotChangedPaths, snapshotForEntry } from \"./restore.ts\";\nimport { type FileHistorySnapshot, isSnapshot, SNAPSHOT_ENTRY_TYPE } from \"./snapshot.ts\";\nimport { configureStorage } from \"./storage.ts\";\nimport { truncateText } from \"./text.ts\";\nimport { editWriteTargetPath, resolveToolPath } from \"./tool-path.ts\";\n\n// Bind the engine/gc storage roots to the real on-disk locations (they avoid\n// importing paths.ts directly so they stay node-testable). Safe to call at load.\nconfigureStorage({ backupsRoot: rewindBackupsRoot(), sessionsRoot: sessionsDirectory() });\n\n// Global (one config.json). Reloaded from disk at session_start; /rewind menu\n// saveRewindConfig updates the in-memory cache so the next turn sees changes\n// without re-reading the file every before_agent_start.\nlet config: RewindConfig = loadRewindConfig();\n\n/** /tree confirm → apply payload (changedPaths skips a second originChanged pass). */\ninterface PendingTreeRestore {\n\tsnapshot: FileHistorySnapshot;\n\tchangedPaths: string[];\n}\n\n// Per-session transient state held by the integration layer.\nconst pendingPrompt = new Map<string, string>(); // turn prompt, captured for the snapshot label\nconst pendingTreeRestore = new Map<string, PendingTreeRestore | null>(); // /tree sync intent\n\n// ---- helpers --------------------------------------------------------------\n\n/** The id of the last user message in the current branch (the turn's anchor). */\nfunction lastUserEntryId(ctx: ExtensionContext): string | undefined {\n\tconst branch = ctx.sessionManager.getBranch();\n\tfor (let i = branch.length - 1; i >= 0; i--) {\n\t\tconst e = branch[i];\n\t\tif (e.type === \"message\" && (e.message as AgentMessage).role === \"user\") return e.id;\n\t}\n\treturn undefined;\n}\n\n/** Rebuild the snapshot list for a session from its persisted custom entries. */\nfunction rebuildSnapshots(ctx: ExtensionContext): FileHistorySnapshot[] {\n\tconst out: FileHistorySnapshot[] = [];\n\tfor (const entry of ctx.sessionManager.getEntries()) {\n\t\tif (entry.type === \"custom\" && entry.customType === SNAPSHOT_ENTRY_TYPE && isSnapshot(entry.data)) {\n\t\t\tout.push(entry.data);\n\t\t}\n\t}\n\treturn out;\n}\n\n/** Max files listed in the /tree restore confirmation before \"+N more\". */\nconst RESTORE_PREVIEW_LIMIT = 8;\n/** Max characters per previewed path (leading-truncated: the filename matters most). */\nconst RESTORE_PREVIEW_PATH_MAX = 64;\n\n/**\n * Bounded cwd-relative file list shown under the /tree restore question, so an\n * irreversible work-tree rewrite is confirmed against WHICH files, not just a\n * count. Multi-line select titles are upstream-sanctioned (ui.confirm joins\n * title + message with \\n through the same selector).\n */\nfunction formatRestorePreview(changedPaths: string[], cwd: string): string {\n\tconst lines = changedPaths.slice(0, RESTORE_PREVIEW_LIMIT).map((p) => {\n\t\tconst rel = path.relative(cwd, p);\n\t\tconst display = rel && !rel.startsWith(\"..\") && !path.isAbsolute(rel) ? rel : p;\n\t\tconst bounded =\n\t\t\tdisplay.length > RESTORE_PREVIEW_PATH_MAX ? `…${display.slice(-(RESTORE_PREVIEW_PATH_MAX - 1))}` : display;\n\t\treturn ` ${bounded}`;\n\t});\n\tif (changedPaths.length > RESTORE_PREVIEW_LIMIT) {\n\t\tlines.push(` … +${changedPaths.length - RESTORE_PREVIEW_LIMIT} more`);\n\t}\n\treturn lines.join(\"\\n\");\n}\n\n// ---- extension entry ------------------------------------------------------\n\nexport default function rewind(pi: ExtensionAPI): void {\n\t// session_start: rebuild the index, migrate fork/resume blobs, reclaim storage.\n\tpi.on(\"session_start\", async (event, ctx) => {\n\t\tconfig = reloadRewindConfig();\n\t\tconst sid = ctx.sessionManager.getSessionId();\n\t\tif (!sid) return;\n\t\tregisterSession(sid, ctx.cwd);\n\n\t\tconst snapshots = rebuildSnapshots(ctx);\n\t\trestoreStateFromSnapshots(sid, ctx.cwd, snapshots, config.maxSnapshots);\n\n\t\tif ((event.reason === \"resume\" || event.reason === \"fork\") && event.previousSessionFile) {\n\t\t\tconst prevSid = sessionIdFromFile(event.previousSessionFile);\n\t\t\tif (prevSid) {\n\t\t\t\ttry {\n\t\t\t\t\tawait migrateBackupsFromSession(prevSid, sid, snapshots);\n\t\t\t\t} catch {\n\t\t\t\t\t// best-effort; a missing blob just means that version can't restore\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\ttry {\n\t\t\trunGc(config.retentionDays, sid);\n\t\t} catch {\n\t\t\t// GC is best-effort; never block startup\n\t\t}\n\t});\n\n\t// tool_call: back up the target file BEFORE the edit/write lands. Synchronous\n\t// so the backup is on disk before the hook returns control to the agent loop.\n\tpi.on(\"tool_call\", (event, ctx) => {\n\t\tif (event.toolName !== \"edit\" && event.toolName !== \"write\") return;\n\t\tif (!config.enabled) return;\n\t\tconst sid = ctx.sessionManager.getSessionId();\n\t\tif (!sid) return;\n\t\tconst rawPath = editWriteTargetPath(event.input);\n\t\tif (!rawPath) return;\n\t\tconst abs = resolveToolPath(rawPath, ctx.cwd);\n\t\ttry {\n\t\t\ttrackEdit(sid, abs);\n\t\t} catch {\n\t\t\t// never block the edit on a backup failure\n\t\t}\n\t});\n\n\t// before_agent_start: open the turn's snapshot frame.\n\tpi.on(\"before_agent_start\", async (event, ctx) => {\n\t\t// Memory-cached; session_start reloads disk, /rewind save updates cache.\n\t\tconfig = loadRewindConfig();\n\t\tif (!config.enabled) return;\n\t\tconst sid = ctx.sessionManager.getSessionId();\n\t\tif (!sid) return;\n\t\tpendingPrompt.set(sid, event.prompt ?? \"\");\n\t\ttry {\n\t\t\tawait beginTurn(sid);\n\t\t} catch {\n\t\t\t// non-fatal\n\t\t}\n\t});\n\n\t// agent_settled: finalize + persist after Pi will not auto-continue (retry /\n\t// compact-retry / follow-up). agent_end alone is too early for that.\n\tpi.on(\"agent_settled\", async (_event, ctx) => {\n\t\tif (!config.enabled) return;\n\t\tconst sid = ctx.sessionManager.getSessionId();\n\t\tif (!sid) return;\n\t\tconst userEntryId = lastUserEntryId(ctx) ?? \"\";\n\t\tconst turnId = ctx.sessionManager.getLeafId() ?? userEntryId;\n\t\tconst prompt = truncateText(pendingPrompt.get(sid) ?? \"\", 120, { collapseWhitespace: true });\n\t\tpendingPrompt.delete(sid);\n\t\tconst frame = endTurn(sid, userEntryId, turnId, prompt, new Date().toISOString(), config.maxSnapshots);\n\t\tif (frame && userEntryId) {\n\t\t\tpi.appendEntry(SNAPSHOT_ENTRY_TYPE, frame);\n\t\t}\n\t});\n\n\t// session_before_tree: offer to sync files when navigating to a changed point.\n\tpi.on(\"session_before_tree\", async (event, ctx) => {\n\t\tconst sid = ctx.sessionManager.getSessionId();\n\t\tif (!sid) return;\n\t\tpendingTreeRestore.set(sid, null);\n\t\tif (!config.enabled) return;\n\n\t\tconst target = snapshotForEntry(getSnapshots(sid), ctx.sessionManager, event.preparation.targetId);\n\t\tif (!target) return;\n\t\tconst changed = await snapshotChangedPaths(sid, target);\n\t\tif (changed.length === 0) return; // silent nav, like native /tree\n\t\t// Lifecycle handler: silent-return without UI. (hasUI is the guard for\n\t\t// ctx.ui.*; checking ctx.mode here would wrongly proceed when a TUI\n\t\t// session has no usable UI.)\n\t\tif (!ctx.hasUI) return;\n\n\t\tconst n = changed.length;\n\t\tlet lineStats = \"\";\n\t\ttry {\n\t\t\tconst stats = await snapshotChangeDiffStats(sid, target, changed);\n\t\t\tif (stats.insertions > 0 || stats.deletions > 0) {\n\t\t\t\tlineStats = ` (+${stats.insertions} / −${stats.deletions})`;\n\t\t\t}\n\t\t} catch {\n\t\t\t// Coarse stats are best-effort; still offer path preview.\n\t\t}\n\t\tconst choice = await ctx.ui.select(\n\t\t\t`Restore ${n} file${n === 1 ? \"\" : \"s\"} to this point?${lineStats}\\n${formatRestorePreview(changed, ctx.cwd)}`,\n\t\t\t[\"Yes, restore files\", \"No, conversation only\"],\n\t\t);\n\t\tif (choice?.startsWith(\"Yes\")) {\n\t\t\t// Cache paths so session_tree can apply without a second full compare.\n\t\t\tpendingTreeRestore.set(sid, { snapshot: target, changedPaths: changed });\n\t\t}\n\t});\n\n\t// session_tree: execute any sync intent recorded above.\n\tpi.on(\"session_tree\", async (_event, ctx) => {\n\t\tconst sid = ctx.sessionManager.getSessionId();\n\t\tif (!sid) return;\n\t\tconst pending = pendingTreeRestore.get(sid);\n\t\tpendingTreeRestore.set(sid, null);\n\t\tif (!pending) return;\n\t\ttry {\n\t\t\tconst changed = await restoreToSnapshot(sid, pending.snapshot, {\n\t\t\t\tonlyPaths: new Set(pending.changedPaths),\n\t\t\t});\n\t\t\tif (changed.length > 0) {\n\t\t\t\tctx.ui.notify(\n\t\t\t\t\t`Restored ${changed.length} file${changed.length === 1 ? \"\" : \"s\"} to this checkpoint.`,\n\t\t\t\t\t\"info\",\n\t\t\t\t);\n\t\t\t}\n\t\t} catch (e) {\n\t\t\tctx.ui.notify(`Rewind restore failed: ${String(e)}`, \"warning\");\n\t\t}\n\t});\n\n\t// session_shutdown: drop this session's in-memory state.\n\tpi.on(\"session_shutdown\", async (_event, ctx) => {\n\t\tconst sid = ctx.sessionManager.getSessionId();\n\t\tif (!sid) return;\n\t\tdisposeSession(sid);\n\t\tpendingTreeRestore.delete(sid);\n\t\tpendingPrompt.delete(sid);\n\t});\n\n\t// /rewind: settings + storage menu (time-travel itself is via /tree).\n\tpi.registerCommand(\"rewind\", {\n\t\tdescription: \"Rewind settings and backup storage (restore is via /tree)\",\n\t\thandler: async (_args, ctx) => {\n\t\t\tawait runRewindMenu(ctx);\n\t\t},\n\t});\n}\n"]}
1
+ {"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/extensions/rewind/index.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GAiCG;AACH,OAAO,IAAI,MAAM,WAAW,CAAC;AAK7B,OAAO,EAAE,qBAAqB,EAAE,MAAM,aAAa,CAAC;AACpD,OAAO,EAAE,gBAAgB,EAAqB,kBAAkB,EAAE,MAAM,aAAa,CAAC;AACtF,OAAO,EACN,SAAS,EACT,YAAY,EACZ,cAAc,EACd,OAAO,EACP,YAAY,EACZ,yBAAyB,EACzB,eAAe,EACf,yBAAyB,EACzB,SAAS,GACT,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,KAAK,EAAE,iBAAiB,EAAE,MAAM,SAAS,CAAC;AACnD,OAAO,EAAE,aAAa,EAAE,MAAM,WAAW,CAAC;AAC1C,OAAO,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,YAAY,CAAC;AAClE,OAAO,EAEN,iBAAiB,EACjB,uBAAuB,EACvB,oBAAoB,EACpB,gBAAgB,GAChB,MAAM,cAAc,CAAC;AACtB,OAAO,EAA4B,UAAU,EAAE,mBAAmB,EAAE,MAAM,eAAe,CAAC;AAC1F,OAAO,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AAChD,OAAO,EAAE,YAAY,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,EAAE,mBAAmB,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAEtE,6EAA6E;AAC7E,iFAAiF;AACjF,gBAAgB,CAAC,EAAE,WAAW,EAAE,iBAAiB,EAAE,EAAE,YAAY,EAAE,iBAAiB,EAAE,EAAE,CAAC,CAAC;AAE1F,8EAA8E;AAC9E,6EAA6E;AAC7E,wDAAwD;AACxD,IAAI,MAAM,GAAiB,gBAAgB,EAAE,CAAC;AAQ9C,6DAA6D;AAC7D,MAAM,aAAa,GAAG,IAAI,GAAG,EAAkB,CAAC,CAAC,+CAA+C;AAChG,MAAM,kBAAkB,GAAG,IAAI,GAAG,EAAqC,CAAC,CAAC,oBAAoB;AAC7F,6EAA6E;AAC7E,iFAA+E;AAC/E,MAAM,eAAe,GAAG,IAAI,GAAG,EAAyB,CAAC;AAEzD,8EAA8E;AAE9E,6EAA6E;AAC7E,SAAS,mBAAmB,CAAC,GAAW,EAAE,GAAqB,EAAQ;IACtE,eAAe,CAAC,GAAG,CAAC,GAAG,EAAE,GAAG,CAAC,cAAc,CAAC,SAAS,EAAE,IAAI,IAAI,CAAC,CAAC;AAAA,CACjE;AAED,kHAAkH;AAClH,SAAS,eAAe,CAAC,GAAqB,EAAiB;IAC9D,OAAO;QACN,QAAQ,CAAC,EAAE,EAAE;YACZ,MAAM,CAAC,GAAG,GAAG,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAC1C,IAAI,CAAC,CAAC;gBAAE,OAAO,SAAS,CAAC;YACzB,MAAM,YAAY,GACjB,CAAC,CAAC,CAAC,IAAI,KAAK,SAAS,IAAK,CAAC,CAAC,OAAwB,CAAC,IAAI,KAAK,MAAM,CAAC,IAAI,CAAC,CAAC,IAAI,KAAK,gBAAgB,CAAC;YACtG,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC,QAAQ,EAAE,YAAY,EAAE,CAAC;QAAA,CACxD;KACD,CAAC;AAAA,CACF;AAED,iFAAiF;AACjF,SAAS,gBAAgB,CAAC,GAAqB,EAAyB;IACvE,MAAM,GAAG,GAA0B,EAAE,CAAC;IACtC,KAAK,MAAM,KAAK,IAAI,GAAG,CAAC,cAAc,CAAC,UAAU,EAAE,EAAE,CAAC;QACrD,IAAI,KAAK,CAAC,IAAI,KAAK,QAAQ,IAAI,KAAK,CAAC,UAAU,KAAK,mBAAmB,IAAI,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC;YACnG,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;QACtB,CAAC;IACF,CAAC;IACD,OAAO,GAAG,CAAC;AAAA,CACX;AAED,2EAA2E;AAC3E,MAAM,qBAAqB,GAAG,CAAC,CAAC;AAChC,wFAAwF;AACxF,MAAM,wBAAwB,GAAG,EAAE,CAAC;AAEpC;;;;;GAKG;AACH,SAAS,oBAAoB,CAAC,YAAsB,EAAE,GAAW,EAAU;IAC1E,MAAM,KAAK,GAAG,YAAY,CAAC,KAAK,CAAC,CAAC,EAAE,qBAAqB,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC;QACrE,MAAM,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;QAClC,MAAM,OAAO,GAAG,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC;QAChF,MAAM,OAAO,GACZ,OAAO,CAAC,MAAM,GAAG,wBAAwB,CAAC,CAAC,CAAC,MAAI,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,wBAAwB,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC;QAC5G,OAAO,KAAK,OAAO,EAAE,CAAC;IAAA,CACtB,CAAC,CAAC;IACH,IAAI,YAAY,CAAC,MAAM,GAAG,qBAAqB,EAAE,CAAC;QACjD,KAAK,CAAC,IAAI,CAAC,UAAQ,YAAY,CAAC,MAAM,GAAG,qBAAqB,OAAO,CAAC,CAAC;IACxE,CAAC;IACD,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAAA,CACxB;AAED,8EAA8E;AAE9E,MAAM,CAAC,OAAO,UAAU,MAAM,CAAC,EAAgB,EAAQ;IACtD,gFAAgF;IAChF,EAAE,CAAC,EAAE,CAAC,eAAe,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC;QAC5C,MAAM,GAAG,kBAAkB,EAAE,CAAC;QAC9B,MAAM,GAAG,GAAG,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,CAAC;QAC9C,IAAI,CAAC,GAAG;YAAE,OAAO;QACjB,eAAe,CAAC,GAAG,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC;QAE9B,MAAM,SAAS,GAAG,gBAAgB,CAAC,GAAG,CAAC,CAAC;QAExC,uEAAuE;QACvE,yEAAyE;QACzE,0EAA0E;QAC1E,kEAAgE;QAChE,IAAI,CAAC,KAAK,CAAC,MAAM,KAAK,QAAQ,IAAI,KAAK,CAAC,MAAM,KAAK,MAAM,CAAC,IAAI,KAAK,CAAC,mBAAmB,EAAE,CAAC;YACzF,MAAM,OAAO,GAAG,iBAAiB,CAAC,KAAK,CAAC,mBAAmB,CAAC,CAAC;YAC7D,IAAI,OAAO,EAAE,CAAC;gBACb,IAAI,CAAC;oBACJ,MAAM,yBAAyB,CAAC,OAAO,EAAE,GAAG,EAAE,YAAY,CAAC,SAAS,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC;gBAC7F,CAAC;gBAAC,MAAM,CAAC;oBACR,oEAAoE;gBACrE,CAAC;YACF,CAAC;QACF,CAAC;QAED,yBAAyB,CAAC,GAAG,EAAE,GAAG,CAAC,GAAG,EAAE,SAAS,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC;QACxE,mBAAmB,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAE9B,IAAI,CAAC;YACJ,KAAK,CAAC,MAAM,CAAC,aAAa,EAAE,GAAG,CAAC,CAAC;QAClC,CAAC;QAAC,MAAM,CAAC;YACR,yCAAyC;QAC1C,CAAC;IAAA,CACD,CAAC,CAAC;IAEH,8EAA8E;IAC9E,8EAA8E;IAC9E,EAAE,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC;QAClC,IAAI,KAAK,CAAC,QAAQ,KAAK,MAAM,IAAI,KAAK,CAAC,QAAQ,KAAK,OAAO;YAAE,OAAO;QACpE,IAAI,CAAC,MAAM,CAAC,OAAO;YAAE,OAAO;QAC5B,MAAM,GAAG,GAAG,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,CAAC;QAC9C,IAAI,CAAC,GAAG;YAAE,OAAO;QACjB,MAAM,OAAO,GAAG,mBAAmB,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACjD,IAAI,CAAC,OAAO;YAAE,OAAO;QACrB,MAAM,GAAG,GAAG,eAAe,CAAC,OAAO,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC;QAC9C,IAAI,CAAC;YACJ,SAAS,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QACrB,CAAC;QAAC,MAAM,CAAC;YACR,2CAA2C;QAC5C,CAAC;IAAA,CACD,CAAC,CAAC;IAEH,sDAAsD;IACtD,EAAE,CAAC,EAAE,CAAC,oBAAoB,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC;QACjD,yEAAyE;QACzE,MAAM,GAAG,gBAAgB,EAAE,CAAC;QAC5B,MAAM,GAAG,GAAG,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,CAAC;QAC9C,IAAI,CAAC,GAAG;YAAE,OAAO;QACjB,0EAA0E;QAC1E,oDAAoD;QACpD,mBAAmB,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAC9B,IAAI,CAAC,MAAM,CAAC,OAAO;YAAE,OAAO;QAC5B,aAAa,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC;QAC3C,IAAI,CAAC;YACJ,MAAM,SAAS,CAAC,GAAG,CAAC,CAAC;QACtB,CAAC;QAAC,MAAM,CAAC;YACR,YAAY;QACb,CAAC;IAAA,CACD,CAAC,CAAC;IAEH,6EAA6E;IAC7E,qEAAqE;IACrE,EAAE,CAAC,EAAE,CAAC,eAAe,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC;QAC7C,IAAI,CAAC,MAAM,CAAC,OAAO;YAAE,OAAO;QAC5B,MAAM,GAAG,GAAG,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,CAAC;QAC9C,IAAI,CAAC,GAAG;YAAE,OAAO;QACjB,sEAAsE;QACtE,yEAAyE;QACzE,uEAAuE;QACvE,4EAA4E;QAC5E,MAAM,SAAS,GAAG,eAAe,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC3C,MAAM,WAAW,GAChB,SAAS,KAAK,SAAS,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,qBAAqB,CAAC,GAAG,CAAC,cAAc,CAAC,SAAS,EAAE,EAAE,SAAS,CAAC,IAAI,EAAE,CAAC,CAAC;QACzG,MAAM,MAAM,GAAG,GAAG,CAAC,cAAc,CAAC,SAAS,EAAE,IAAI,WAAW,CAAC;QAC7D,MAAM,MAAM,GAAG,YAAY,CAAC,aAAa,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,EAAE,EAAE,GAAG,EAAE,EAAE,kBAAkB,EAAE,IAAI,EAAE,CAAC,CAAC;QAC7F,aAAa,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC1B,MAAM,KAAK,GAAG,OAAO,CAAC,GAAG,EAAE,WAAW,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,IAAI,EAAE,CAAC,WAAW,EAAE,EAAE,MAAM,CAAC,YAAY,CAAC,CAAC;QACvG,IAAI,KAAK,EAAE,CAAC;YACX,EAAE,CAAC,WAAW,CAAC,mBAAmB,EAAE,KAAK,CAAC,CAAC;QAC5C,CAAC;QACD,yEAAyE;QACzE,wEAAwE;QACxE,mBAAmB,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;IAAA,CAC9B,CAAC,CAAC;IAEH,+EAA+E;IAC/E,EAAE,CAAC,EAAE,CAAC,qBAAqB,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC;QAClD,MAAM,GAAG,GAAG,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,CAAC;QAC9C,IAAI,CAAC,GAAG;YAAE,OAAO;QACjB,kBAAkB,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QAClC,IAAI,CAAC,MAAM,CAAC,OAAO;YAAE,OAAO;QAE5B,MAAM,MAAM,GAAG,gBAAgB,CAAC,YAAY,CAAC,GAAG,CAAC,EAAE,eAAe,CAAC,GAAG,CAAC,EAAE,KAAK,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QACrG,IAAI,CAAC,MAAM;YAAE,OAAO;QACpB,MAAM,OAAO,GAAG,MAAM,oBAAoB,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACxD,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,CAAC,gCAAgC;QAClE,uEAAuE;QACvE,oEAAoE;QACpE,6BAA6B;QAC7B,IAAI,CAAC,GAAG,CAAC,KAAK;YAAE,OAAO;QAEvB,MAAM,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;QACzB,IAAI,SAAS,GAAG,EAAE,CAAC;QACnB,IAAI,CAAC;YACJ,MAAM,KAAK,GAAG,MAAM,uBAAuB,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;YAClE,IAAI,KAAK,CAAC,UAAU,GAAG,CAAC,IAAI,KAAK,CAAC,SAAS,GAAG,CAAC,EAAE,CAAC;gBACjD,SAAS,GAAG,OAAO,KAAK,CAAC,UAAU,SAAO,KAAK,CAAC,SAAS,GAAG,CAAC;YAC9D,CAAC;QACF,CAAC;QAAC,MAAM,CAAC;YACR,0DAA0D;QAC3D,CAAC;QACD,MAAM,MAAM,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,MAAM,CACjC,WAAW,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,kBAAkB,SAAS,KAAK,oBAAoB,CAAC,OAAO,EAAE,GAAG,CAAC,GAAG,CAAC,8EAA8E,EAC1L,CAAC,oBAAoB,EAAE,uBAAuB,CAAC,CAC/C,CAAC;QACF,IAAI,MAAM,EAAE,UAAU,CAAC,KAAK,CAAC,EAAE,CAAC;YAC/B,uEAAuE;YACvE,kBAAkB,CAAC,GAAG,CAAC,GAAG,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,YAAY,EAAE,OAAO,EAAE,CAAC,CAAC;QAC1E,CAAC;IAAA,CACD,CAAC,CAAC;IAEH,wDAAwD;IACxD,EAAE,CAAC,EAAE,CAAC,cAAc,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC;QAC5C,MAAM,GAAG,GAAG,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,CAAC;QAC9C,IAAI,CAAC,GAAG;YAAE,OAAO;QACjB,yEAAyE;QACzE,iEAAiE;QACjE,mBAAmB,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAC9B,MAAM,OAAO,GAAG,kBAAkB,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC5C,kBAAkB,CAAC,GAAG,CAAC,GAAG,EAAE,IAAI,CAAC,CAAC;QAClC,IAAI,CAAC,OAAO;YAAE,OAAO;QACrB,IAAI,CAAC;YACJ,MAAM,OAAO,GAAG,MAAM,iBAAiB,CAAC,GAAG,EAAE,OAAO,CAAC,QAAQ,EAAE;gBAC9D,SAAS,EAAE,IAAI,GAAG,CAAC,OAAO,CAAC,YAAY,CAAC;aACxC,CAAC,CAAC;YACH,IAAI,OAAO,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACxB,GAAG,CAAC,EAAE,CAAC,MAAM,CACZ,YAAY,OAAO,CAAC,MAAM,QAAQ,OAAO,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,sBAAsB,EACvF,MAAM,CACN,CAAC;YACH,CAAC;QACF,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACZ,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,0BAA0B,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,SAAS,CAAC,CAAC;QACjE,CAAC;IAAA,CACD,CAAC,CAAC;IAEH,yDAAyD;IACzD,EAAE,CAAC,EAAE,CAAC,kBAAkB,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,EAAE,CAAC;QAChD,MAAM,GAAG,GAAG,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,CAAC;QAC9C,IAAI,CAAC,GAAG;YAAE,OAAO;QACjB,cAAc,CAAC,GAAG,CAAC,CAAC;QACpB,kBAAkB,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC/B,aAAa,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC1B,eAAe,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;IAAA,CAC5B,CAAC,CAAC;IAEH,sEAAsE;IACtE,EAAE,CAAC,eAAe,CAAC,QAAQ,EAAE;QAC5B,WAAW,EAAE,2DAA2D;QACxE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC;YAC9B,MAAM,aAAa,CAAC,GAAG,CAAC,CAAC;QAAA,CACzB;KACD,CAAC,CAAC;AAAA,CACH","sourcesContent":["/**\n * rewind — file-backed checkpoint & rewind for Pi.\n *\n * Engine: instead of snapshotting the whole work tree into a per-session shadow\n * git repo (the old design, which froze the UI in large directories like a\n * multi-project parent folder and grew storage without bound), we back up ONLY\n * the files Pi's edit/write tools are about to modify — one copyFile before each\n * edit (see engine.ts, ported from Claude Code's file-history). Cost is\n * proportional to how many files Pi changed, never to project size.\n *\n * Per turn: before_agent_start opens a snapshot frame (re-recording tracked files\n * at their turn-start state); tool_call(edit|write) backs up each newly edited\n * file before it lands; agent_settled persists the frame to the session JSONL\n * (custom \"pi-rewind-snapshot\" entry) when it changed. agent_settled is used\n * instead of agent_end so auto-retry, overflow compaction-retry, and queued\n * follow-ups stay in one logical turn (agent_end can fire mid-continuation).\n * The frame anchors to the FIRST user entry the run appended (tracked via\n * anchorScanStart + anchor.ts) — the entry whose start the frame recorded;\n * steering/follow-up messages append later user entries that must not steal\n * the anchor. On session_start the index is rebuilt from those entries;\n * resume/fork hard-links the prior session's blobs.\n *\n * Time-travel is fused into /tree: navigating to a node whose turn changed files\n * offers to sync the work tree to that point (session_before_tree/session_tree).\n * Snapshot selection mirrors navigateTree's leaf rules (see restore.ts).\n * /rewind itself is a settings + storage menu (menu.ts), not a restore picker.\n *\n * Restore safety: applySnapshot only rewrites files that differ and never throws\n * out — a broken backup degrades to \"leave the file alone\", so it can never abort\n * the user's session.\n *\n * Architecture informed by oh-my-pi (GPL-3.0) and Claude Code's file-history;\n * independent implementation. No ANSI: all UI is native (ctx.ui.* + theme).\n */\nimport path from \"node:path\";\n\nimport type { AgentMessage } from \"@earendil-works/pi-agent-core\";\nimport type { ExtensionAPI, ExtensionContext } from \"../../core/extensions/types.ts\";\n\nimport { firstUserEntryIdAfter } from \"./anchor.ts\";\nimport { loadRewindConfig, type RewindConfig, reloadRewindConfig } from \"./config.ts\";\nimport {\n\tbeginTurn,\n\tcapSnapshots,\n\tdisposeSession,\n\tendTurn,\n\tgetSnapshots,\n\tmigrateBackupsFromSession,\n\tregisterSession,\n\trestoreStateFromSnapshots,\n\ttrackEdit,\n} from \"./engine.ts\";\nimport { runGc, sessionIdFromFile } from \"./gc.ts\";\nimport { runRewindMenu } from \"./menu.ts\";\nimport { rewindBackupsRoot, sessionsDirectory } from \"./paths.ts\";\nimport {\n\ttype EntryTreeView,\n\trestoreToSnapshot,\n\tsnapshotChangeDiffStats,\n\tsnapshotChangedPaths,\n\tsnapshotForEntry,\n} from \"./restore.ts\";\nimport { type FileHistorySnapshot, isSnapshot, SNAPSHOT_ENTRY_TYPE } from \"./snapshot.ts\";\nimport { configureStorage } from \"./storage.ts\";\nimport { truncateText } from \"./text.ts\";\nimport { editWriteTargetPath, resolveToolPath } from \"./tool-path.ts\";\n\n// Bind the engine/gc storage roots to the real on-disk locations (they avoid\n// importing paths.ts directly so they stay node-testable). Safe to call at load.\nconfigureStorage({ backupsRoot: rewindBackupsRoot(), sessionsRoot: sessionsDirectory() });\n\n// Global (one config.json). Reloaded from disk at session_start; /rewind menu\n// saveRewindConfig updates the in-memory cache so the next turn sees changes\n// without re-reading the file every before_agent_start.\nlet config: RewindConfig = loadRewindConfig();\n\n/** /tree confirm → apply payload (changedPaths skips a second originChanged pass). */\ninterface PendingTreeRestore {\n\tsnapshot: FileHistorySnapshot;\n\tchangedPaths: string[];\n}\n\n// Per-session transient state held by the integration layer.\nconst pendingPrompt = new Map<string, string>(); // turn prompt, captured for the snapshot label\nconst pendingTreeRestore = new Map<string, PendingTreeRestore | null>(); // /tree sync intent\n// Leaf id before the current run started (null = branch root). agent_settled\n// scans forward from here for the run's FIRST user entry — the frame's anchor.\nconst anchorScanStart = new Map<string, string | null>();\n\n// ---- helpers --------------------------------------------------------------\n\n/** Record the current leaf as the scan start for the next frame's anchor. */\nfunction markAnchorScanStart(sid: string, ctx: ExtensionContext): void {\n\tanchorScanStart.set(sid, ctx.sessionManager.getLeafId() ?? null);\n}\n\n/** Adapt the session manager to restore.ts's tree view (turn-anchor = navigateTree's \"leaf = parent\" entries). */\nfunction sessionTreeView(ctx: ExtensionContext): EntryTreeView {\n\treturn {\n\t\tgetEntry(id) {\n\t\t\tconst e = ctx.sessionManager.getEntry(id);\n\t\t\tif (!e) return undefined;\n\t\t\tconst isTurnAnchor =\n\t\t\t\t(e.type === \"message\" && (e.message as AgentMessage).role === \"user\") || e.type === \"custom_message\";\n\t\t\treturn { id: e.id, parentId: e.parentId, isTurnAnchor };\n\t\t},\n\t};\n}\n\n/** Rebuild the snapshot list for a session from its persisted custom entries. */\nfunction rebuildSnapshots(ctx: ExtensionContext): FileHistorySnapshot[] {\n\tconst out: FileHistorySnapshot[] = [];\n\tfor (const entry of ctx.sessionManager.getEntries()) {\n\t\tif (entry.type === \"custom\" && entry.customType === SNAPSHOT_ENTRY_TYPE && isSnapshot(entry.data)) {\n\t\t\tout.push(entry.data);\n\t\t}\n\t}\n\treturn out;\n}\n\n/** Max files listed in the /tree restore confirmation before \"+N more\". */\nconst RESTORE_PREVIEW_LIMIT = 8;\n/** Max characters per previewed path (leading-truncated: the filename matters most). */\nconst RESTORE_PREVIEW_PATH_MAX = 64;\n\n/**\n * Bounded cwd-relative file list shown under the /tree restore question, so an\n * irreversible work-tree rewrite is confirmed against WHICH files, not just a\n * count. Multi-line select titles are upstream-sanctioned (ui.confirm joins\n * title + message with \\n through the same selector).\n */\nfunction formatRestorePreview(changedPaths: string[], cwd: string): string {\n\tconst lines = changedPaths.slice(0, RESTORE_PREVIEW_LIMIT).map((p) => {\n\t\tconst rel = path.relative(cwd, p);\n\t\tconst display = rel && !rel.startsWith(\"..\") && !path.isAbsolute(rel) ? rel : p;\n\t\tconst bounded =\n\t\t\tdisplay.length > RESTORE_PREVIEW_PATH_MAX ? `…${display.slice(-(RESTORE_PREVIEW_PATH_MAX - 1))}` : display;\n\t\treturn ` ${bounded}`;\n\t});\n\tif (changedPaths.length > RESTORE_PREVIEW_LIMIT) {\n\t\tlines.push(` … +${changedPaths.length - RESTORE_PREVIEW_LIMIT} more`);\n\t}\n\treturn lines.join(\"\\n\");\n}\n\n// ---- extension entry ------------------------------------------------------\n\nexport default function rewind(pi: ExtensionAPI): void {\n\t// session_start: rebuild the index, migrate fork/resume blobs, reclaim storage.\n\tpi.on(\"session_start\", async (event, ctx) => {\n\t\tconfig = reloadRewindConfig();\n\t\tconst sid = ctx.sessionManager.getSessionId();\n\t\tif (!sid) return;\n\t\tregisterSession(sid, ctx.cwd);\n\n\t\tconst snapshots = rebuildSnapshots(ctx);\n\n\t\t// Migrate BEFORE rebuilding in-memory state: restoreStateFromSnapshots\n\t\t// prunes over-cap blobs from THIS session's directory (fire-and-forget),\n\t\t// and linking only the retained frames first keeps the two steps disjoint\n\t\t// — no race, and dropped frames' blobs are never linked at all.\n\t\tif ((event.reason === \"resume\" || event.reason === \"fork\") && event.previousSessionFile) {\n\t\t\tconst prevSid = sessionIdFromFile(event.previousSessionFile);\n\t\t\tif (prevSid) {\n\t\t\t\ttry {\n\t\t\t\t\tawait migrateBackupsFromSession(prevSid, sid, capSnapshots(snapshots, config.maxSnapshots));\n\t\t\t\t} catch {\n\t\t\t\t\t// best-effort; a missing blob just means that version can't restore\n\t\t\t\t}\n\t\t\t}\n\t\t}\n\n\t\trestoreStateFromSnapshots(sid, ctx.cwd, snapshots, config.maxSnapshots);\n\t\tmarkAnchorScanStart(sid, ctx);\n\n\t\ttry {\n\t\t\trunGc(config.retentionDays, sid);\n\t\t} catch {\n\t\t\t// GC is best-effort; never block startup\n\t\t}\n\t});\n\n\t// tool_call: back up the target file BEFORE the edit/write lands. Synchronous\n\t// so the backup is on disk before the hook returns control to the agent loop.\n\tpi.on(\"tool_call\", (event, ctx) => {\n\t\tif (event.toolName !== \"edit\" && event.toolName !== \"write\") return;\n\t\tif (!config.enabled) return;\n\t\tconst sid = ctx.sessionManager.getSessionId();\n\t\tif (!sid) return;\n\t\tconst rawPath = editWriteTargetPath(event.input);\n\t\tif (!rawPath) return;\n\t\tconst abs = resolveToolPath(rawPath, ctx.cwd);\n\t\ttry {\n\t\t\ttrackEdit(sid, abs);\n\t\t} catch {\n\t\t\t// never block the edit on a backup failure\n\t\t}\n\t});\n\n\t// before_agent_start: open the turn's snapshot frame.\n\tpi.on(\"before_agent_start\", async (event, ctx) => {\n\t\t// Memory-cached; session_start reloads disk, /rewind save updates cache.\n\t\tconfig = loadRewindConfig();\n\t\tconst sid = ctx.sessionManager.getSessionId();\n\t\tif (!sid) return;\n\t\t// The run's user entry is appended after this hook; the frame's anchor is\n\t\t// the first user entry that appears past this leaf.\n\t\tmarkAnchorScanStart(sid, ctx);\n\t\tif (!config.enabled) return;\n\t\tpendingPrompt.set(sid, event.prompt ?? \"\");\n\t\ttry {\n\t\t\tawait beginTurn(sid);\n\t\t} catch {\n\t\t\t// non-fatal\n\t\t}\n\t});\n\n\t// agent_settled: finalize + persist after Pi will not auto-continue (retry /\n\t// compact-retry / follow-up). agent_end alone is too early for that.\n\tpi.on(\"agent_settled\", async (_event, ctx) => {\n\t\tif (!config.enabled) return;\n\t\tconst sid = ctx.sessionManager.getSessionId();\n\t\tif (!sid) return;\n\t\t// Anchor = the run's FIRST appended user entry (whose start the frame\n\t\t// recorded). Steering/follow-up messages consumed in the same run append\n\t\t// later user entries and must not steal the anchor. Unresolvable (scan\n\t\t// start missing / off-branch / no user entry appended) -> endTurn discards.\n\t\tconst scanStart = anchorScanStart.get(sid);\n\t\tconst userEntryId =\n\t\t\tscanStart === undefined ? \"\" : (firstUserEntryIdAfter(ctx.sessionManager.getBranch(), scanStart) ?? \"\");\n\t\tconst turnId = ctx.sessionManager.getLeafId() ?? userEntryId;\n\t\tconst prompt = truncateText(pendingPrompt.get(sid) ?? \"\", 120, { collapseWhitespace: true });\n\t\tpendingPrompt.delete(sid);\n\t\tconst frame = endTurn(sid, userEntryId, turnId, prompt, new Date().toISOString(), config.maxSnapshots);\n\t\tif (frame) {\n\t\t\tpi.appendEntry(SNAPSHOT_ENTRY_TYPE, frame);\n\t\t}\n\t\t// Next run may start without before_agent_start (custom-triggered): scan\n\t\t// from the settled leaf (past this run's entries + the snapshot entry).\n\t\tmarkAnchorScanStart(sid, ctx);\n\t});\n\n\t// session_before_tree: offer to sync files when navigating to a changed point.\n\tpi.on(\"session_before_tree\", async (event, ctx) => {\n\t\tconst sid = ctx.sessionManager.getSessionId();\n\t\tif (!sid) return;\n\t\tpendingTreeRestore.set(sid, null);\n\t\tif (!config.enabled) return;\n\n\t\tconst target = snapshotForEntry(getSnapshots(sid), sessionTreeView(ctx), event.preparation.targetId);\n\t\tif (!target) return;\n\t\tconst changed = await snapshotChangedPaths(sid, target);\n\t\tif (changed.length === 0) return; // silent nav, like native /tree\n\t\t// Lifecycle handler: silent-return without UI. (hasUI is the guard for\n\t\t// ctx.ui.*; checking ctx.mode here would wrongly proceed when a TUI\n\t\t// session has no usable UI.)\n\t\tif (!ctx.hasUI) return;\n\n\t\tconst n = changed.length;\n\t\tlet lineStats = \"\";\n\t\ttry {\n\t\t\tconst stats = await snapshotChangeDiffStats(sid, target, changed);\n\t\t\tif (stats.insertions > 0 || stats.deletions > 0) {\n\t\t\t\tlineStats = ` (+${stats.insertions} / −${stats.deletions})`;\n\t\t\t}\n\t\t} catch {\n\t\t\t// Coarse stats are best-effort; still offer path preview.\n\t\t}\n\t\tconst choice = await ctx.ui.select(\n\t\t\t`Restore ${n} file${n === 1 ? \"\" : \"s\"} to this point?${lineStats}\\n${formatRestorePreview(changed, ctx.cwd)}\\n (covers edit/write changes only; files changed via bash are not tracked)`,\n\t\t\t[\"Yes, restore files\", \"No, conversation only\"],\n\t\t);\n\t\tif (choice?.startsWith(\"Yes\")) {\n\t\t\t// Cache paths so session_tree can apply without a second full compare.\n\t\t\tpendingTreeRestore.set(sid, { snapshot: target, changedPaths: changed });\n\t\t}\n\t});\n\n\t// session_tree: execute any sync intent recorded above.\n\tpi.on(\"session_tree\", async (_event, ctx) => {\n\t\tconst sid = ctx.sessionManager.getSessionId();\n\t\tif (!sid) return;\n\t\t// The leaf moved; the old scan start may now be off-branch. Re-mark so a\n\t\t// custom-triggered run after navigation still anchors correctly.\n\t\tmarkAnchorScanStart(sid, ctx);\n\t\tconst pending = pendingTreeRestore.get(sid);\n\t\tpendingTreeRestore.set(sid, null);\n\t\tif (!pending) return;\n\t\ttry {\n\t\t\tconst changed = await restoreToSnapshot(sid, pending.snapshot, {\n\t\t\t\tonlyPaths: new Set(pending.changedPaths),\n\t\t\t});\n\t\t\tif (changed.length > 0) {\n\t\t\t\tctx.ui.notify(\n\t\t\t\t\t`Restored ${changed.length} file${changed.length === 1 ? \"\" : \"s\"} to this checkpoint.`,\n\t\t\t\t\t\"info\",\n\t\t\t\t);\n\t\t\t}\n\t\t} catch (e) {\n\t\t\tctx.ui.notify(`Rewind restore failed: ${String(e)}`, \"warning\");\n\t\t}\n\t});\n\n\t// session_shutdown: drop this session's in-memory state.\n\tpi.on(\"session_shutdown\", async (_event, ctx) => {\n\t\tconst sid = ctx.sessionManager.getSessionId();\n\t\tif (!sid) return;\n\t\tdisposeSession(sid);\n\t\tpendingTreeRestore.delete(sid);\n\t\tpendingPrompt.delete(sid);\n\t\tanchorScanStart.delete(sid);\n\t});\n\n\t// /rewind: settings + storage menu (time-travel itself is via /tree).\n\tpi.registerCommand(\"rewind\", {\n\t\tdescription: \"Rewind settings and backup storage (restore is via /tree)\",\n\t\thandler: async (_args, ctx) => {\n\t\t\tawait runRewindMenu(ctx);\n\t\t},\n\t});\n}\n"]}
@@ -1 +1 @@
1
- {"version":3,"file":"menu.d.ts","sourceRoot":"","sources":["../../../src/extensions/rewind/menu.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AACH,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,gCAAgC,CAAC;AAQ9E,wBAAsB,aAAa,CAAC,GAAG,EAAE,uBAAuB,GAAG,OAAO,CAAC,IAAI,CAAC,CA4B/E","sourcesContent":["/**\n * menu.ts — the `/rewind` settings menu.\n *\n * Per the design, /rewind no longer performs the time-travel itself — restoring\n * files is fused into /tree navigation (see index.ts session_before_tree/\n * session_tree). /rewind is purely a settings + storage console:\n * - toggle the master switch\n * - set the auto-clean retention window\n * - inspect and prune backup storage (aged / orphaned / all-but-current)\n */\nimport type { ExtensionCommandContext } from \"../../core/extensions/types.ts\";\n\nimport { loadRewindConfig, saveRewindConfig } from \"./config.ts\";\nimport { listSessions, removeSession, runGc } from \"./gc.ts\";\nimport { formatSize } from \"./text.ts\";\n\nconst RETENTION_PRESETS = [7, 14, 30, 60, 90];\n\nexport async function runRewindMenu(ctx: ExtensionCommandContext): Promise<void> {\n\tif (!ctx.hasUI) {\n\t\tctx.ui.notify(\"/rewind requires an interactive UI.\", \"warning\");\n\t\treturn;\n\t}\n\tconst sid = ctx.sessionManager.getSessionId() ?? undefined;\n\n\tfor (;;) {\n\t\tconst cfg = loadRewindConfig();\n\t\tconst sessions = listSessions(sid);\n\t\tconst total = sessions.reduce((n, s) => n + s.bytes, 0);\n\n\t\tconst enabledLabel = `Rewind: ${cfg.enabled ? \"on\" : \"off\"}`;\n\t\tconst retentionLabel = `Auto-clean backups: ${cfg.retentionDays > 0 ? `after ${cfg.retentionDays} days` : \"never (keep forever)\"}`;\n\t\tconst storageLabel = `Storage: ${formatSize(total)} across ${sessions.length} session${sessions.length === 1 ? \"\" : \"s\"}`;\n\n\t\tconst pick = await ctx.ui.select(\"Rewind settings (Esc to close)\", [enabledLabel, retentionLabel, storageLabel]);\n\t\tif (!pick) return;\n\n\t\tif (pick === enabledLabel) {\n\t\t\tsaveRewindConfig({ ...cfg, enabled: !cfg.enabled });\n\t\t\tctx.ui.notify(`Rewind ${cfg.enabled ? \"disabled\" : \"enabled\"}.`, \"info\");\n\t\t} else if (pick === retentionLabel) {\n\t\t\tawait pickRetention(ctx);\n\t\t} else if (pick === storageLabel) {\n\t\t\tawait storageMenu(ctx, sid);\n\t\t}\n\t}\n}\n\nasync function pickRetention(ctx: ExtensionCommandContext): Promise<void> {\n\tconst options = [...RETENTION_PRESETS.map((d) => `${d} days`), \"Keep forever\", \"Custom...\"];\n\tconst pick = await ctx.ui.select(\"Auto-clean backups after\", options);\n\tif (!pick) return;\n\n\tlet days: number;\n\tif (pick === \"Keep forever\") {\n\t\tdays = 0;\n\t} else if (pick === \"Custom...\") {\n\t\tconst value = await ctx.ui.input(\"Days to keep backups (0 = forever)\");\n\t\tif (value === undefined) return;\n\t\tconst n = Number.parseInt(value.trim(), 10);\n\t\tif (!Number.isFinite(n) || n < 0) {\n\t\t\tctx.ui.notify(\"Invalid number of days.\", \"warning\");\n\t\t\treturn;\n\t\t}\n\t\tdays = n;\n\t} else {\n\t\tdays = Number.parseInt(pick, 10);\n\t}\n\n\tconst cfg = loadRewindConfig();\n\tsaveRewindConfig({ ...cfg, retentionDays: days });\n\tctx.ui.notify(days > 0 ? `Backups now kept for ${days} days.` : \"Backups kept forever.\", \"info\");\n}\n\nasync function storageMenu(ctx: ExtensionCommandContext, sid: string | undefined): Promise<void> {\n\tfor (;;) {\n\t\tconst sessions = listSessions(sid);\n\t\tconst total = sessions.reduce((n, s) => n + s.bytes, 0);\n\t\tconst orphans = sessions.filter((s) => s.orphan);\n\t\tconst others = sessions.filter((s) => s.sessionId !== sid);\n\n\t\tconst cleanLabel = \"Clean now (aged + orphaned)\";\n\t\tconst orphanLabel = `Remove orphaned (${orphans.length})`;\n\t\tconst allLabel = `Remove all except current (${others.length})`;\n\n\t\tconst pick = await ctx.ui.select(\n\t\t\t`Rewind storage — ${formatSize(total)}, ${sessions.length} session${sessions.length === 1 ? \"\" : \"s\"} (Esc to close)`,\n\t\t\t[cleanLabel, orphanLabel, allLabel],\n\t\t);\n\t\tif (!pick) return;\n\n\t\tif (pick === cleanLabel) {\n\t\t\tconst r = runGc(loadRewindConfig().retentionDays, sid);\n\t\t\tctx.ui.notify(\n\t\t\t\tr.removed > 0\n\t\t\t\t\t? `Removed ${r.removed} session(s), reclaimed ${formatSize(r.reclaimedBytes)}.`\n\t\t\t\t\t: \"Nothing to clean.\",\n\t\t\t\t\"info\",\n\t\t\t);\n\t\t} else if (pick === orphanLabel) {\n\t\t\tif (orphans.length === 0) {\n\t\t\t\tctx.ui.notify(\"No orphaned backups.\", \"info\");\n\t\t\t\tcontinue;\n\t\t\t}\n\t\t\tconst ok = await ctx.ui.confirm(\n\t\t\t\t\"Remove orphaned backups?\",\n\t\t\t\t`Deletes backups for ${orphans.length} session(s) with no session file.`,\n\t\t\t);\n\t\t\tif (!ok) continue;\n\t\t\tlet bytes = 0;\n\t\t\tlet removed = 0;\n\t\t\tfor (const s of orphans) {\n\t\t\t\tconst b = removeSession(s.sessionId);\n\t\t\t\tif (b !== null) {\n\t\t\t\t\tbytes += b;\n\t\t\t\t\tremoved++;\n\t\t\t\t}\n\t\t\t}\n\t\t\tctx.ui.notify(`Removed ${removed} orphaned session(s), reclaimed ${formatSize(bytes)}.`, \"info\");\n\t\t} else if (pick === allLabel) {\n\t\t\tif (others.length === 0) {\n\t\t\t\tctx.ui.notify(\"No other sessions' backups.\", \"info\");\n\t\t\t\tcontinue;\n\t\t\t}\n\t\t\tconst ok = await ctx.ui.confirm(\n\t\t\t\t\"Remove all other backups?\",\n\t\t\t\t`Deletes backups for ${others.length} session(s), keeping only the current session.`,\n\t\t\t);\n\t\t\tif (!ok) continue;\n\t\t\tlet bytes = 0;\n\t\t\tlet removed = 0;\n\t\t\tfor (const s of others) {\n\t\t\t\tconst b = removeSession(s.sessionId);\n\t\t\t\tif (b !== null) {\n\t\t\t\t\tbytes += b;\n\t\t\t\t\tremoved++;\n\t\t\t\t}\n\t\t\t}\n\t\t\tctx.ui.notify(`Removed ${removed} session(s), reclaimed ${formatSize(bytes)}.`, \"info\");\n\t\t}\n\t}\n}\n"]}
1
+ {"version":3,"file":"menu.d.ts","sourceRoot":"","sources":["../../../src/extensions/rewind/menu.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AACH,OAAO,KAAK,EAAE,uBAAuB,EAAE,MAAM,gCAAgC,CAAC;AAW9E,wBAAsB,aAAa,CAAC,GAAG,EAAE,uBAAuB,GAAG,OAAO,CAAC,IAAI,CAAC,CA4B/E","sourcesContent":["/**\n * menu.ts — the `/rewind` settings menu.\n *\n * Per the design, /rewind no longer performs the time-travel itself — restoring\n * files is fused into /tree navigation (see index.ts session_before_tree/\n * session_tree). /rewind is purely a settings + storage console:\n * - toggle the master switch\n * - set the auto-clean retention window\n * - inspect and prune backup storage (aged / orphaned / all-but-current)\n */\nimport type { ExtensionCommandContext } from \"../../core/extensions/types.ts\";\n\nimport { loadRewindConfig, saveRewindConfig } from \"./config.ts\";\nimport { listSessions, removeSession, runGc } from \"./gc.ts\";\nimport { formatSize } from \"./text.ts\";\n\nconst RETENTION_PRESETS = [7, 14, 30, 60, 90];\n\n/** Backup dirs written to this recently likely belong to a live session in another Pi process. */\nconst RECENT_ACTIVE_MS = 10 * 60 * 1000;\n\nexport async function runRewindMenu(ctx: ExtensionCommandContext): Promise<void> {\n\tif (!ctx.hasUI) {\n\t\tctx.ui.notify(\"/rewind requires an interactive UI.\", \"warning\");\n\t\treturn;\n\t}\n\tconst sid = ctx.sessionManager.getSessionId() ?? undefined;\n\n\tfor (;;) {\n\t\tconst cfg = loadRewindConfig();\n\t\tconst sessions = listSessions(sid);\n\t\tconst total = sessions.reduce((n, s) => n + s.bytes, 0);\n\n\t\tconst enabledLabel = `Rewind: ${cfg.enabled ? \"on\" : \"off\"}`;\n\t\tconst retentionLabel = `Auto-clean backups: ${cfg.retentionDays > 0 ? `after ${cfg.retentionDays} days` : \"never (keep forever)\"}`;\n\t\tconst storageLabel = `Storage: ${formatSize(total)} across ${sessions.length} session${sessions.length === 1 ? \"\" : \"s\"}`;\n\n\t\tconst pick = await ctx.ui.select(\"Rewind settings (Esc to close)\", [enabledLabel, retentionLabel, storageLabel]);\n\t\tif (!pick) return;\n\n\t\tif (pick === enabledLabel) {\n\t\t\tsaveRewindConfig({ ...cfg, enabled: !cfg.enabled });\n\t\t\tctx.ui.notify(`Rewind ${cfg.enabled ? \"disabled\" : \"enabled\"}.`, \"info\");\n\t\t} else if (pick === retentionLabel) {\n\t\t\tawait pickRetention(ctx);\n\t\t} else if (pick === storageLabel) {\n\t\t\tawait storageMenu(ctx, sid);\n\t\t}\n\t}\n}\n\nasync function pickRetention(ctx: ExtensionCommandContext): Promise<void> {\n\tconst options = [...RETENTION_PRESETS.map((d) => `${d} days`), \"Keep forever\", \"Custom...\"];\n\tconst pick = await ctx.ui.select(\"Auto-clean backups after\", options);\n\tif (!pick) return;\n\n\tlet days: number;\n\tif (pick === \"Keep forever\") {\n\t\tdays = 0;\n\t} else if (pick === \"Custom...\") {\n\t\tconst value = await ctx.ui.input(\"Days to keep backups (0 = forever)\");\n\t\tif (value === undefined) return;\n\t\tconst n = Number.parseInt(value.trim(), 10);\n\t\tif (!Number.isFinite(n) || n < 0) {\n\t\t\tctx.ui.notify(\"Invalid number of days.\", \"warning\");\n\t\t\treturn;\n\t\t}\n\t\tdays = n;\n\t} else {\n\t\tdays = Number.parseInt(pick, 10);\n\t}\n\n\tconst cfg = loadRewindConfig();\n\tsaveRewindConfig({ ...cfg, retentionDays: days });\n\tctx.ui.notify(days > 0 ? `Backups now kept for ${days} days.` : \"Backups kept forever.\", \"info\");\n}\n\nasync function storageMenu(ctx: ExtensionCommandContext, sid: string | undefined): Promise<void> {\n\tfor (;;) {\n\t\tconst sessions = listSessions(sid);\n\t\tconst total = sessions.reduce((n, s) => n + s.bytes, 0);\n\t\tconst orphans = sessions.filter((s) => s.orphan);\n\t\t// Skip recently-written dirs: they may belong to another running Pi\n\t\t// process whose in-memory index still references those blobs.\n\t\tconst now = Date.now();\n\t\tconst others = sessions.filter((s) => s.sessionId !== sid && now - s.mtimeMs > RECENT_ACTIVE_MS);\n\n\t\tconst cleanLabel = \"Clean now (aged + orphaned)\";\n\t\tconst orphanLabel = `Remove orphaned (${orphans.length})`;\n\t\tconst allLabel = `Remove all except current (${others.length})`;\n\n\t\tconst pick = await ctx.ui.select(\n\t\t\t`Rewind storage — ${formatSize(total)}, ${sessions.length} session${sessions.length === 1 ? \"\" : \"s\"} (Esc to close)`,\n\t\t\t[cleanLabel, orphanLabel, allLabel],\n\t\t);\n\t\tif (!pick) return;\n\n\t\tif (pick === cleanLabel) {\n\t\t\tconst r = runGc(loadRewindConfig().retentionDays, sid);\n\t\t\tctx.ui.notify(\n\t\t\t\tr.removed > 0\n\t\t\t\t\t? `Removed ${r.removed} session(s), reclaimed ${formatSize(r.reclaimedBytes)}.`\n\t\t\t\t\t: \"Nothing to clean.\",\n\t\t\t\t\"info\",\n\t\t\t);\n\t\t} else if (pick === orphanLabel) {\n\t\t\tif (orphans.length === 0) {\n\t\t\t\tctx.ui.notify(\"No orphaned backups.\", \"info\");\n\t\t\t\tcontinue;\n\t\t\t}\n\t\t\tconst ok = await ctx.ui.confirm(\n\t\t\t\t\"Remove orphaned backups?\",\n\t\t\t\t`Deletes backups for ${orphans.length} session(s) with no session file.`,\n\t\t\t);\n\t\t\tif (!ok) continue;\n\t\t\tlet bytes = 0;\n\t\t\tlet removed = 0;\n\t\t\tfor (const s of orphans) {\n\t\t\t\tconst b = removeSession(s.sessionId);\n\t\t\t\tif (b !== null) {\n\t\t\t\t\tbytes += b;\n\t\t\t\t\tremoved++;\n\t\t\t\t}\n\t\t\t}\n\t\t\tctx.ui.notify(`Removed ${removed} orphaned session(s), reclaimed ${formatSize(bytes)}.`, \"info\");\n\t\t} else if (pick === allLabel) {\n\t\t\tif (others.length === 0) {\n\t\t\t\tctx.ui.notify(\"No other sessions' backups.\", \"info\");\n\t\t\t\tcontinue;\n\t\t\t}\n\t\t\tconst ok = await ctx.ui.confirm(\n\t\t\t\t\"Remove all other backups?\",\n\t\t\t\t`Deletes backups for ${others.length} session(s). Keeps the current session and any written to in the last 10 minutes (possibly running elsewhere).`,\n\t\t\t);\n\t\t\tif (!ok) continue;\n\t\t\tlet bytes = 0;\n\t\t\tlet removed = 0;\n\t\t\tfor (const s of others) {\n\t\t\t\tconst b = removeSession(s.sessionId);\n\t\t\t\tif (b !== null) {\n\t\t\t\t\tbytes += b;\n\t\t\t\t\tremoved++;\n\t\t\t\t}\n\t\t\t}\n\t\t\tctx.ui.notify(`Removed ${removed} session(s), reclaimed ${formatSize(bytes)}.`, \"info\");\n\t\t}\n\t}\n}\n"]}
@@ -2,6 +2,8 @@ import { loadRewindConfig, saveRewindConfig } from "./config.js";
2
2
  import { listSessions, removeSession, runGc } from "./gc.js";
3
3
  import { formatSize } from "./text.js";
4
4
  const RETENTION_PRESETS = [7, 14, 30, 60, 90];
5
+ /** Backup dirs written to this recently likely belong to a live session in another Pi process. */
6
+ const RECENT_ACTIVE_MS = 10 * 60 * 1000;
5
7
  export async function runRewindMenu(ctx) {
6
8
  if (!ctx.hasUI) {
7
9
  ctx.ui.notify("/rewind requires an interactive UI.", "warning");
@@ -62,7 +64,10 @@ async function storageMenu(ctx, sid) {
62
64
  const sessions = listSessions(sid);
63
65
  const total = sessions.reduce((n, s) => n + s.bytes, 0);
64
66
  const orphans = sessions.filter((s) => s.orphan);
65
- const others = sessions.filter((s) => s.sessionId !== sid);
67
+ // Skip recently-written dirs: they may belong to another running Pi
68
+ // process whose in-memory index still references those blobs.
69
+ const now = Date.now();
70
+ const others = sessions.filter((s) => s.sessionId !== sid && now - s.mtimeMs > RECENT_ACTIVE_MS);
66
71
  const cleanLabel = "Clean now (aged + orphaned)";
67
72
  const orphanLabel = `Remove orphaned (${orphans.length})`;
68
73
  const allLabel = `Remove all except current (${others.length})`;
@@ -99,7 +104,7 @@ async function storageMenu(ctx, sid) {
99
104
  ctx.ui.notify("No other sessions' backups.", "info");
100
105
  continue;
101
106
  }
102
- const ok = await ctx.ui.confirm("Remove all other backups?", `Deletes backups for ${others.length} session(s), keeping only the current session.`);
107
+ const ok = await ctx.ui.confirm("Remove all other backups?", `Deletes backups for ${others.length} session(s). Keeps the current session and any written to in the last 10 minutes (possibly running elsewhere).`);
103
108
  if (!ok)
104
109
  continue;
105
110
  let bytes = 0;
@@ -1 +1 @@
1
- {"version":3,"file":"menu.js","sourceRoot":"","sources":["../../../src/extensions/rewind/menu.ts"],"names":[],"mappings":"AAYA,OAAO,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AACjE,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,KAAK,EAAE,MAAM,SAAS,CAAC;AAC7D,OAAO,EAAE,UAAU,EAAE,MAAM,WAAW,CAAC;AAEvC,MAAM,iBAAiB,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;AAE9C,MAAM,CAAC,KAAK,UAAU,aAAa,CAAC,GAA4B,EAAiB;IAChF,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC;QAChB,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,qCAAqC,EAAE,SAAS,CAAC,CAAC;QAChE,OAAO;IACR,CAAC;IACD,MAAM,GAAG,GAAG,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,IAAI,SAAS,CAAC;IAE3D,SAAS,CAAC;QACT,MAAM,GAAG,GAAG,gBAAgB,EAAE,CAAC;QAC/B,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC;QACnC,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QAExD,MAAM,YAAY,GAAG,WAAW,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC;QAC7D,MAAM,cAAc,GAAG,uBAAuB,GAAG,CAAC,aAAa,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,aAAa,OAAO,CAAC,CAAC,CAAC,sBAAsB,EAAE,CAAC;QACnI,MAAM,YAAY,GAAG,YAAY,UAAU,CAAC,KAAK,CAAC,WAAW,QAAQ,CAAC,MAAM,WAAW,QAAQ,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;QAE1H,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,gCAAgC,EAAE,CAAC,YAAY,EAAE,cAAc,EAAE,YAAY,CAAC,CAAC,CAAC;QACjH,IAAI,CAAC,IAAI;YAAE,OAAO;QAElB,IAAI,IAAI,KAAK,YAAY,EAAE,CAAC;YAC3B,gBAAgB,CAAC,EAAE,GAAG,GAAG,EAAE,OAAO,EAAE,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;YACpD,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,UAAU,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,GAAG,EAAE,MAAM,CAAC,CAAC;QAC1E,CAAC;aAAM,IAAI,IAAI,KAAK,cAAc,EAAE,CAAC;YACpC,MAAM,aAAa,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC;aAAM,IAAI,IAAI,KAAK,YAAY,EAAE,CAAC;YAClC,MAAM,WAAW,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAC7B,CAAC;IACF,CAAC;AAAA,CACD;AAED,KAAK,UAAU,aAAa,CAAC,GAA4B,EAAiB;IACzE,MAAM,OAAO,GAAG,CAAC,GAAG,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,cAAc,EAAE,WAAW,CAAC,CAAC;IAC5F,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,0BAA0B,EAAE,OAAO,CAAC,CAAC;IACtE,IAAI,CAAC,IAAI;QAAE,OAAO;IAElB,IAAI,IAAY,CAAC;IACjB,IAAI,IAAI,KAAK,cAAc,EAAE,CAAC;QAC7B,IAAI,GAAG,CAAC,CAAC;IACV,CAAC;SAAM,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;QACjC,MAAM,KAAK,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,oCAAoC,CAAC,CAAC;QACvE,IAAI,KAAK,KAAK,SAAS;YAAE,OAAO;QAChC,MAAM,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;QAC5C,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;YAClC,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,yBAAyB,EAAE,SAAS,CAAC,CAAC;YACpD,OAAO;QACR,CAAC;QACD,IAAI,GAAG,CAAC,CAAC;IACV,CAAC;SAAM,CAAC;QACP,IAAI,GAAG,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IAClC,CAAC;IAED,MAAM,GAAG,GAAG,gBAAgB,EAAE,CAAC;IAC/B,gBAAgB,CAAC,EAAE,GAAG,GAAG,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;IAClD,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,wBAAwB,IAAI,QAAQ,CAAC,CAAC,CAAC,uBAAuB,EAAE,MAAM,CAAC,CAAC;AAAA,CACjG;AAED,KAAK,UAAU,WAAW,CAAC,GAA4B,EAAE,GAAuB,EAAiB;IAChG,SAAS,CAAC;QACT,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC;QACnC,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QACxD,MAAM,OAAO,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;QACjD,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,GAAG,CAAC,CAAC;QAE3D,MAAM,UAAU,GAAG,6BAA6B,CAAC;QACjD,MAAM,WAAW,GAAG,oBAAoB,OAAO,CAAC,MAAM,GAAG,CAAC;QAC1D,MAAM,QAAQ,GAAG,8BAA8B,MAAM,CAAC,MAAM,GAAG,CAAC;QAEhE,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,MAAM,CAC/B,sBAAoB,UAAU,CAAC,KAAK,CAAC,KAAK,QAAQ,CAAC,MAAM,WAAW,QAAQ,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,iBAAiB,EACrH,CAAC,UAAU,EAAE,WAAW,EAAE,QAAQ,CAAC,CACnC,CAAC;QACF,IAAI,CAAC,IAAI;YAAE,OAAO;QAElB,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;YACzB,MAAM,CAAC,GAAG,KAAK,CAAC,gBAAgB,EAAE,CAAC,aAAa,EAAE,GAAG,CAAC,CAAC;YACvD,GAAG,CAAC,EAAE,CAAC,MAAM,CACZ,CAAC,CAAC,OAAO,GAAG,CAAC;gBACZ,CAAC,CAAC,WAAW,CAAC,CAAC,OAAO,0BAA0B,UAAU,CAAC,CAAC,CAAC,cAAc,CAAC,GAAG;gBAC/E,CAAC,CAAC,mBAAmB,EACtB,MAAM,CACN,CAAC;QACH,CAAC;aAAM,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;YACjC,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC1B,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,sBAAsB,EAAE,MAAM,CAAC,CAAC;gBAC9C,SAAS;YACV,CAAC;YACD,MAAM,EAAE,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,OAAO,CAC9B,0BAA0B,EAC1B,uBAAuB,OAAO,CAAC,MAAM,mCAAmC,CACxE,CAAC;YACF,IAAI,CAAC,EAAE;gBAAE,SAAS;YAClB,IAAI,KAAK,GAAG,CAAC,CAAC;YACd,IAAI,OAAO,GAAG,CAAC,CAAC;YAChB,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;gBACzB,MAAM,CAAC,GAAG,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;gBACrC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;oBAChB,KAAK,IAAI,CAAC,CAAC;oBACX,OAAO,EAAE,CAAC;gBACX,CAAC;YACF,CAAC;YACD,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,WAAW,OAAO,mCAAmC,UAAU,CAAC,KAAK,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAClG,CAAC;aAAM,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC9B,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,6BAA6B,EAAE,MAAM,CAAC,CAAC;gBACrD,SAAS;YACV,CAAC;YACD,MAAM,EAAE,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,OAAO,CAC9B,2BAA2B,EAC3B,uBAAuB,MAAM,CAAC,MAAM,gDAAgD,CACpF,CAAC;YACF,IAAI,CAAC,EAAE;gBAAE,SAAS;YAClB,IAAI,KAAK,GAAG,CAAC,CAAC;YACd,IAAI,OAAO,GAAG,CAAC,CAAC;YAChB,KAAK,MAAM,CAAC,IAAI,MAAM,EAAE,CAAC;gBACxB,MAAM,CAAC,GAAG,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;gBACrC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;oBAChB,KAAK,IAAI,CAAC,CAAC;oBACX,OAAO,EAAE,CAAC;gBACX,CAAC;YACF,CAAC;YACD,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,WAAW,OAAO,0BAA0B,UAAU,CAAC,KAAK,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACzF,CAAC;IACF,CAAC;AAAA,CACD","sourcesContent":["/**\n * menu.ts — the `/rewind` settings menu.\n *\n * Per the design, /rewind no longer performs the time-travel itself — restoring\n * files is fused into /tree navigation (see index.ts session_before_tree/\n * session_tree). /rewind is purely a settings + storage console:\n * - toggle the master switch\n * - set the auto-clean retention window\n * - inspect and prune backup storage (aged / orphaned / all-but-current)\n */\nimport type { ExtensionCommandContext } from \"../../core/extensions/types.ts\";\n\nimport { loadRewindConfig, saveRewindConfig } from \"./config.ts\";\nimport { listSessions, removeSession, runGc } from \"./gc.ts\";\nimport { formatSize } from \"./text.ts\";\n\nconst RETENTION_PRESETS = [7, 14, 30, 60, 90];\n\nexport async function runRewindMenu(ctx: ExtensionCommandContext): Promise<void> {\n\tif (!ctx.hasUI) {\n\t\tctx.ui.notify(\"/rewind requires an interactive UI.\", \"warning\");\n\t\treturn;\n\t}\n\tconst sid = ctx.sessionManager.getSessionId() ?? undefined;\n\n\tfor (;;) {\n\t\tconst cfg = loadRewindConfig();\n\t\tconst sessions = listSessions(sid);\n\t\tconst total = sessions.reduce((n, s) => n + s.bytes, 0);\n\n\t\tconst enabledLabel = `Rewind: ${cfg.enabled ? \"on\" : \"off\"}`;\n\t\tconst retentionLabel = `Auto-clean backups: ${cfg.retentionDays > 0 ? `after ${cfg.retentionDays} days` : \"never (keep forever)\"}`;\n\t\tconst storageLabel = `Storage: ${formatSize(total)} across ${sessions.length} session${sessions.length === 1 ? \"\" : \"s\"}`;\n\n\t\tconst pick = await ctx.ui.select(\"Rewind settings (Esc to close)\", [enabledLabel, retentionLabel, storageLabel]);\n\t\tif (!pick) return;\n\n\t\tif (pick === enabledLabel) {\n\t\t\tsaveRewindConfig({ ...cfg, enabled: !cfg.enabled });\n\t\t\tctx.ui.notify(`Rewind ${cfg.enabled ? \"disabled\" : \"enabled\"}.`, \"info\");\n\t\t} else if (pick === retentionLabel) {\n\t\t\tawait pickRetention(ctx);\n\t\t} else if (pick === storageLabel) {\n\t\t\tawait storageMenu(ctx, sid);\n\t\t}\n\t}\n}\n\nasync function pickRetention(ctx: ExtensionCommandContext): Promise<void> {\n\tconst options = [...RETENTION_PRESETS.map((d) => `${d} days`), \"Keep forever\", \"Custom...\"];\n\tconst pick = await ctx.ui.select(\"Auto-clean backups after\", options);\n\tif (!pick) return;\n\n\tlet days: number;\n\tif (pick === \"Keep forever\") {\n\t\tdays = 0;\n\t} else if (pick === \"Custom...\") {\n\t\tconst value = await ctx.ui.input(\"Days to keep backups (0 = forever)\");\n\t\tif (value === undefined) return;\n\t\tconst n = Number.parseInt(value.trim(), 10);\n\t\tif (!Number.isFinite(n) || n < 0) {\n\t\t\tctx.ui.notify(\"Invalid number of days.\", \"warning\");\n\t\t\treturn;\n\t\t}\n\t\tdays = n;\n\t} else {\n\t\tdays = Number.parseInt(pick, 10);\n\t}\n\n\tconst cfg = loadRewindConfig();\n\tsaveRewindConfig({ ...cfg, retentionDays: days });\n\tctx.ui.notify(days > 0 ? `Backups now kept for ${days} days.` : \"Backups kept forever.\", \"info\");\n}\n\nasync function storageMenu(ctx: ExtensionCommandContext, sid: string | undefined): Promise<void> {\n\tfor (;;) {\n\t\tconst sessions = listSessions(sid);\n\t\tconst total = sessions.reduce((n, s) => n + s.bytes, 0);\n\t\tconst orphans = sessions.filter((s) => s.orphan);\n\t\tconst others = sessions.filter((s) => s.sessionId !== sid);\n\n\t\tconst cleanLabel = \"Clean now (aged + orphaned)\";\n\t\tconst orphanLabel = `Remove orphaned (${orphans.length})`;\n\t\tconst allLabel = `Remove all except current (${others.length})`;\n\n\t\tconst pick = await ctx.ui.select(\n\t\t\t`Rewind storage — ${formatSize(total)}, ${sessions.length} session${sessions.length === 1 ? \"\" : \"s\"} (Esc to close)`,\n\t\t\t[cleanLabel, orphanLabel, allLabel],\n\t\t);\n\t\tif (!pick) return;\n\n\t\tif (pick === cleanLabel) {\n\t\t\tconst r = runGc(loadRewindConfig().retentionDays, sid);\n\t\t\tctx.ui.notify(\n\t\t\t\tr.removed > 0\n\t\t\t\t\t? `Removed ${r.removed} session(s), reclaimed ${formatSize(r.reclaimedBytes)}.`\n\t\t\t\t\t: \"Nothing to clean.\",\n\t\t\t\t\"info\",\n\t\t\t);\n\t\t} else if (pick === orphanLabel) {\n\t\t\tif (orphans.length === 0) {\n\t\t\t\tctx.ui.notify(\"No orphaned backups.\", \"info\");\n\t\t\t\tcontinue;\n\t\t\t}\n\t\t\tconst ok = await ctx.ui.confirm(\n\t\t\t\t\"Remove orphaned backups?\",\n\t\t\t\t`Deletes backups for ${orphans.length} session(s) with no session file.`,\n\t\t\t);\n\t\t\tif (!ok) continue;\n\t\t\tlet bytes = 0;\n\t\t\tlet removed = 0;\n\t\t\tfor (const s of orphans) {\n\t\t\t\tconst b = removeSession(s.sessionId);\n\t\t\t\tif (b !== null) {\n\t\t\t\t\tbytes += b;\n\t\t\t\t\tremoved++;\n\t\t\t\t}\n\t\t\t}\n\t\t\tctx.ui.notify(`Removed ${removed} orphaned session(s), reclaimed ${formatSize(bytes)}.`, \"info\");\n\t\t} else if (pick === allLabel) {\n\t\t\tif (others.length === 0) {\n\t\t\t\tctx.ui.notify(\"No other sessions' backups.\", \"info\");\n\t\t\t\tcontinue;\n\t\t\t}\n\t\t\tconst ok = await ctx.ui.confirm(\n\t\t\t\t\"Remove all other backups?\",\n\t\t\t\t`Deletes backups for ${others.length} session(s), keeping only the current session.`,\n\t\t\t);\n\t\t\tif (!ok) continue;\n\t\t\tlet bytes = 0;\n\t\t\tlet removed = 0;\n\t\t\tfor (const s of others) {\n\t\t\t\tconst b = removeSession(s.sessionId);\n\t\t\t\tif (b !== null) {\n\t\t\t\t\tbytes += b;\n\t\t\t\t\tremoved++;\n\t\t\t\t}\n\t\t\t}\n\t\t\tctx.ui.notify(`Removed ${removed} session(s), reclaimed ${formatSize(bytes)}.`, \"info\");\n\t\t}\n\t}\n}\n"]}
1
+ {"version":3,"file":"menu.js","sourceRoot":"","sources":["../../../src/extensions/rewind/menu.ts"],"names":[],"mappings":"AAYA,OAAO,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,MAAM,aAAa,CAAC;AACjE,OAAO,EAAE,YAAY,EAAE,aAAa,EAAE,KAAK,EAAE,MAAM,SAAS,CAAC;AAC7D,OAAO,EAAE,UAAU,EAAE,MAAM,WAAW,CAAC;AAEvC,MAAM,iBAAiB,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;AAE9C,kGAAkG;AAClG,MAAM,gBAAgB,GAAG,EAAE,GAAG,EAAE,GAAG,IAAI,CAAC;AAExC,MAAM,CAAC,KAAK,UAAU,aAAa,CAAC,GAA4B,EAAiB;IAChF,IAAI,CAAC,GAAG,CAAC,KAAK,EAAE,CAAC;QAChB,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,qCAAqC,EAAE,SAAS,CAAC,CAAC;QAChE,OAAO;IACR,CAAC;IACD,MAAM,GAAG,GAAG,GAAG,CAAC,cAAc,CAAC,YAAY,EAAE,IAAI,SAAS,CAAC;IAE3D,SAAS,CAAC;QACT,MAAM,GAAG,GAAG,gBAAgB,EAAE,CAAC;QAC/B,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC;QACnC,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QAExD,MAAM,YAAY,GAAG,WAAW,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,EAAE,CAAC;QAC7D,MAAM,cAAc,GAAG,uBAAuB,GAAG,CAAC,aAAa,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS,GAAG,CAAC,aAAa,OAAO,CAAC,CAAC,CAAC,sBAAsB,EAAE,CAAC;QACnI,MAAM,YAAY,GAAG,YAAY,UAAU,CAAC,KAAK,CAAC,WAAW,QAAQ,CAAC,MAAM,WAAW,QAAQ,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;QAE1H,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,gCAAgC,EAAE,CAAC,YAAY,EAAE,cAAc,EAAE,YAAY,CAAC,CAAC,CAAC;QACjH,IAAI,CAAC,IAAI;YAAE,OAAO;QAElB,IAAI,IAAI,KAAK,YAAY,EAAE,CAAC;YAC3B,gBAAgB,CAAC,EAAE,GAAG,GAAG,EAAE,OAAO,EAAE,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;YACpD,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,UAAU,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,GAAG,EAAE,MAAM,CAAC,CAAC;QAC1E,CAAC;aAAM,IAAI,IAAI,KAAK,cAAc,EAAE,CAAC;YACpC,MAAM,aAAa,CAAC,GAAG,CAAC,CAAC;QAC1B,CAAC;aAAM,IAAI,IAAI,KAAK,YAAY,EAAE,CAAC;YAClC,MAAM,WAAW,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAC7B,CAAC;IACF,CAAC;AAAA,CACD;AAED,KAAK,UAAU,aAAa,CAAC,GAA4B,EAAiB;IACzE,MAAM,OAAO,GAAG,CAAC,GAAG,iBAAiB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,cAAc,EAAE,WAAW,CAAC,CAAC;IAC5F,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,0BAA0B,EAAE,OAAO,CAAC,CAAC;IACtE,IAAI,CAAC,IAAI;QAAE,OAAO;IAElB,IAAI,IAAY,CAAC;IACjB,IAAI,IAAI,KAAK,cAAc,EAAE,CAAC;QAC7B,IAAI,GAAG,CAAC,CAAC;IACV,CAAC;SAAM,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;QACjC,MAAM,KAAK,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,KAAK,CAAC,oCAAoC,CAAC,CAAC;QACvE,IAAI,KAAK,KAAK,SAAS;YAAE,OAAO;QAChC,MAAM,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;QAC5C,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC;YAClC,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,yBAAyB,EAAE,SAAS,CAAC,CAAC;YACpD,OAAO;QACR,CAAC;QACD,IAAI,GAAG,CAAC,CAAC;IACV,CAAC;SAAM,CAAC;QACP,IAAI,GAAG,MAAM,CAAC,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IAClC,CAAC;IAED,MAAM,GAAG,GAAG,gBAAgB,EAAE,CAAC;IAC/B,gBAAgB,CAAC,EAAE,GAAG,GAAG,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,CAAC;IAClD,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,IAAI,GAAG,CAAC,CAAC,CAAC,CAAC,wBAAwB,IAAI,QAAQ,CAAC,CAAC,CAAC,uBAAuB,EAAE,MAAM,CAAC,CAAC;AAAA,CACjG;AAED,KAAK,UAAU,WAAW,CAAC,GAA4B,EAAE,GAAuB,EAAiB;IAChG,SAAS,CAAC;QACT,MAAM,QAAQ,GAAG,YAAY,CAAC,GAAG,CAAC,CAAC;QACnC,MAAM,KAAK,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QACxD,MAAM,OAAO,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;QACjD,oEAAoE;QACpE,8DAA8D;QAC9D,MAAM,GAAG,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QACvB,MAAM,MAAM,GAAG,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,SAAS,KAAK,GAAG,IAAI,GAAG,GAAG,CAAC,CAAC,OAAO,GAAG,gBAAgB,CAAC,CAAC;QAEjG,MAAM,UAAU,GAAG,6BAA6B,CAAC;QACjD,MAAM,WAAW,GAAG,oBAAoB,OAAO,CAAC,MAAM,GAAG,CAAC;QAC1D,MAAM,QAAQ,GAAG,8BAA8B,MAAM,CAAC,MAAM,GAAG,CAAC;QAEhE,MAAM,IAAI,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,MAAM,CAC/B,sBAAoB,UAAU,CAAC,KAAK,CAAC,KAAK,QAAQ,CAAC,MAAM,WAAW,QAAQ,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,iBAAiB,EACrH,CAAC,UAAU,EAAE,WAAW,EAAE,QAAQ,CAAC,CACnC,CAAC;QACF,IAAI,CAAC,IAAI;YAAE,OAAO;QAElB,IAAI,IAAI,KAAK,UAAU,EAAE,CAAC;YACzB,MAAM,CAAC,GAAG,KAAK,CAAC,gBAAgB,EAAE,CAAC,aAAa,EAAE,GAAG,CAAC,CAAC;YACvD,GAAG,CAAC,EAAE,CAAC,MAAM,CACZ,CAAC,CAAC,OAAO,GAAG,CAAC;gBACZ,CAAC,CAAC,WAAW,CAAC,CAAC,OAAO,0BAA0B,UAAU,CAAC,CAAC,CAAC,cAAc,CAAC,GAAG;gBAC/E,CAAC,CAAC,mBAAmB,EACtB,MAAM,CACN,CAAC;QACH,CAAC;aAAM,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;YACjC,IAAI,OAAO,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC1B,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,sBAAsB,EAAE,MAAM,CAAC,CAAC;gBAC9C,SAAS;YACV,CAAC;YACD,MAAM,EAAE,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,OAAO,CAC9B,0BAA0B,EAC1B,uBAAuB,OAAO,CAAC,MAAM,mCAAmC,CACxE,CAAC;YACF,IAAI,CAAC,EAAE;gBAAE,SAAS;YAClB,IAAI,KAAK,GAAG,CAAC,CAAC;YACd,IAAI,OAAO,GAAG,CAAC,CAAC;YAChB,KAAK,MAAM,CAAC,IAAI,OAAO,EAAE,CAAC;gBACzB,MAAM,CAAC,GAAG,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;gBACrC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;oBAChB,KAAK,IAAI,CAAC,CAAC;oBACX,OAAO,EAAE,CAAC;gBACX,CAAC;YACF,CAAC;YACD,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,WAAW,OAAO,mCAAmC,UAAU,CAAC,KAAK,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QAClG,CAAC;aAAM,IAAI,IAAI,KAAK,QAAQ,EAAE,CAAC;YAC9B,IAAI,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACzB,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,6BAA6B,EAAE,MAAM,CAAC,CAAC;gBACrD,SAAS;YACV,CAAC;YACD,MAAM,EAAE,GAAG,MAAM,GAAG,CAAC,EAAE,CAAC,OAAO,CAC9B,2BAA2B,EAC3B,uBAAuB,MAAM,CAAC,MAAM,gHAAgH,CACpJ,CAAC;YACF,IAAI,CAAC,EAAE;gBAAE,SAAS;YAClB,IAAI,KAAK,GAAG,CAAC,CAAC;YACd,IAAI,OAAO,GAAG,CAAC,CAAC;YAChB,KAAK,MAAM,CAAC,IAAI,MAAM,EAAE,CAAC;gBACxB,MAAM,CAAC,GAAG,aAAa,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;gBACrC,IAAI,CAAC,KAAK,IAAI,EAAE,CAAC;oBAChB,KAAK,IAAI,CAAC,CAAC;oBACX,OAAO,EAAE,CAAC;gBACX,CAAC;YACF,CAAC;YACD,GAAG,CAAC,EAAE,CAAC,MAAM,CAAC,WAAW,OAAO,0BAA0B,UAAU,CAAC,KAAK,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;QACzF,CAAC;IACF,CAAC;AAAA,CACD","sourcesContent":["/**\n * menu.ts — the `/rewind` settings menu.\n *\n * Per the design, /rewind no longer performs the time-travel itself — restoring\n * files is fused into /tree navigation (see index.ts session_before_tree/\n * session_tree). /rewind is purely a settings + storage console:\n * - toggle the master switch\n * - set the auto-clean retention window\n * - inspect and prune backup storage (aged / orphaned / all-but-current)\n */\nimport type { ExtensionCommandContext } from \"../../core/extensions/types.ts\";\n\nimport { loadRewindConfig, saveRewindConfig } from \"./config.ts\";\nimport { listSessions, removeSession, runGc } from \"./gc.ts\";\nimport { formatSize } from \"./text.ts\";\n\nconst RETENTION_PRESETS = [7, 14, 30, 60, 90];\n\n/** Backup dirs written to this recently likely belong to a live session in another Pi process. */\nconst RECENT_ACTIVE_MS = 10 * 60 * 1000;\n\nexport async function runRewindMenu(ctx: ExtensionCommandContext): Promise<void> {\n\tif (!ctx.hasUI) {\n\t\tctx.ui.notify(\"/rewind requires an interactive UI.\", \"warning\");\n\t\treturn;\n\t}\n\tconst sid = ctx.sessionManager.getSessionId() ?? undefined;\n\n\tfor (;;) {\n\t\tconst cfg = loadRewindConfig();\n\t\tconst sessions = listSessions(sid);\n\t\tconst total = sessions.reduce((n, s) => n + s.bytes, 0);\n\n\t\tconst enabledLabel = `Rewind: ${cfg.enabled ? \"on\" : \"off\"}`;\n\t\tconst retentionLabel = `Auto-clean backups: ${cfg.retentionDays > 0 ? `after ${cfg.retentionDays} days` : \"never (keep forever)\"}`;\n\t\tconst storageLabel = `Storage: ${formatSize(total)} across ${sessions.length} session${sessions.length === 1 ? \"\" : \"s\"}`;\n\n\t\tconst pick = await ctx.ui.select(\"Rewind settings (Esc to close)\", [enabledLabel, retentionLabel, storageLabel]);\n\t\tif (!pick) return;\n\n\t\tif (pick === enabledLabel) {\n\t\t\tsaveRewindConfig({ ...cfg, enabled: !cfg.enabled });\n\t\t\tctx.ui.notify(`Rewind ${cfg.enabled ? \"disabled\" : \"enabled\"}.`, \"info\");\n\t\t} else if (pick === retentionLabel) {\n\t\t\tawait pickRetention(ctx);\n\t\t} else if (pick === storageLabel) {\n\t\t\tawait storageMenu(ctx, sid);\n\t\t}\n\t}\n}\n\nasync function pickRetention(ctx: ExtensionCommandContext): Promise<void> {\n\tconst options = [...RETENTION_PRESETS.map((d) => `${d} days`), \"Keep forever\", \"Custom...\"];\n\tconst pick = await ctx.ui.select(\"Auto-clean backups after\", options);\n\tif (!pick) return;\n\n\tlet days: number;\n\tif (pick === \"Keep forever\") {\n\t\tdays = 0;\n\t} else if (pick === \"Custom...\") {\n\t\tconst value = await ctx.ui.input(\"Days to keep backups (0 = forever)\");\n\t\tif (value === undefined) return;\n\t\tconst n = Number.parseInt(value.trim(), 10);\n\t\tif (!Number.isFinite(n) || n < 0) {\n\t\t\tctx.ui.notify(\"Invalid number of days.\", \"warning\");\n\t\t\treturn;\n\t\t}\n\t\tdays = n;\n\t} else {\n\t\tdays = Number.parseInt(pick, 10);\n\t}\n\n\tconst cfg = loadRewindConfig();\n\tsaveRewindConfig({ ...cfg, retentionDays: days });\n\tctx.ui.notify(days > 0 ? `Backups now kept for ${days} days.` : \"Backups kept forever.\", \"info\");\n}\n\nasync function storageMenu(ctx: ExtensionCommandContext, sid: string | undefined): Promise<void> {\n\tfor (;;) {\n\t\tconst sessions = listSessions(sid);\n\t\tconst total = sessions.reduce((n, s) => n + s.bytes, 0);\n\t\tconst orphans = sessions.filter((s) => s.orphan);\n\t\t// Skip recently-written dirs: they may belong to another running Pi\n\t\t// process whose in-memory index still references those blobs.\n\t\tconst now = Date.now();\n\t\tconst others = sessions.filter((s) => s.sessionId !== sid && now - s.mtimeMs > RECENT_ACTIVE_MS);\n\n\t\tconst cleanLabel = \"Clean now (aged + orphaned)\";\n\t\tconst orphanLabel = `Remove orphaned (${orphans.length})`;\n\t\tconst allLabel = `Remove all except current (${others.length})`;\n\n\t\tconst pick = await ctx.ui.select(\n\t\t\t`Rewind storage — ${formatSize(total)}, ${sessions.length} session${sessions.length === 1 ? \"\" : \"s\"} (Esc to close)`,\n\t\t\t[cleanLabel, orphanLabel, allLabel],\n\t\t);\n\t\tif (!pick) return;\n\n\t\tif (pick === cleanLabel) {\n\t\t\tconst r = runGc(loadRewindConfig().retentionDays, sid);\n\t\t\tctx.ui.notify(\n\t\t\t\tr.removed > 0\n\t\t\t\t\t? `Removed ${r.removed} session(s), reclaimed ${formatSize(r.reclaimedBytes)}.`\n\t\t\t\t\t: \"Nothing to clean.\",\n\t\t\t\t\"info\",\n\t\t\t);\n\t\t} else if (pick === orphanLabel) {\n\t\t\tif (orphans.length === 0) {\n\t\t\t\tctx.ui.notify(\"No orphaned backups.\", \"info\");\n\t\t\t\tcontinue;\n\t\t\t}\n\t\t\tconst ok = await ctx.ui.confirm(\n\t\t\t\t\"Remove orphaned backups?\",\n\t\t\t\t`Deletes backups for ${orphans.length} session(s) with no session file.`,\n\t\t\t);\n\t\t\tif (!ok) continue;\n\t\t\tlet bytes = 0;\n\t\t\tlet removed = 0;\n\t\t\tfor (const s of orphans) {\n\t\t\t\tconst b = removeSession(s.sessionId);\n\t\t\t\tif (b !== null) {\n\t\t\t\t\tbytes += b;\n\t\t\t\t\tremoved++;\n\t\t\t\t}\n\t\t\t}\n\t\t\tctx.ui.notify(`Removed ${removed} orphaned session(s), reclaimed ${formatSize(bytes)}.`, \"info\");\n\t\t} else if (pick === allLabel) {\n\t\t\tif (others.length === 0) {\n\t\t\t\tctx.ui.notify(\"No other sessions' backups.\", \"info\");\n\t\t\t\tcontinue;\n\t\t\t}\n\t\t\tconst ok = await ctx.ui.confirm(\n\t\t\t\t\"Remove all other backups?\",\n\t\t\t\t`Deletes backups for ${others.length} session(s). Keeps the current session and any written to in the last 10 minutes (possibly running elsewhere).`,\n\t\t\t);\n\t\t\tif (!ok) continue;\n\t\t\tlet bytes = 0;\n\t\t\tlet removed = 0;\n\t\t\tfor (const s of others) {\n\t\t\t\tconst b = removeSession(s.sessionId);\n\t\t\t\tif (b !== null) {\n\t\t\t\t\tbytes += b;\n\t\t\t\t\tremoved++;\n\t\t\t\t}\n\t\t\t}\n\t\t\tctx.ui.notify(`Removed ${removed} session(s), reclaimed ${formatSize(bytes)}.`, \"info\");\n\t\t}\n\t}\n}\n"]}
@@ -1,22 +1,39 @@
1
1
  /**
2
2
  * restore.ts — map a /tree navigation target to the snapshot that should be
3
3
  * restored and apply it via the engine.
4
+ *
5
+ * Snapshot semantics: a frame anchored at user entry U records the work tree
6
+ * at the moment U's run STARTED (see engine.ts beginTurn). The selection here
7
+ * must mirror AgentSession.navigateTree's leaf rules:
8
+ *
9
+ * - user / custom_message target: navigateTree removes the target's turn
10
+ * from the branch (leaf = parent, text returns to the editor) → restore
11
+ * to the state BEFORE that turn = the snapshot anchored at the target.
12
+ * - any other target: the target's turn STAYS in the conversation
13
+ * (leaf = target) → restore to the state AFTER that turn = the first
14
+ * chronological snapshot anchored at a descendant turn (it recorded the
15
+ * work tree when the next run began). No such snapshot → nothing was
16
+ * recorded after the target, so offer no restore at all — falling back to
17
+ * an ancestor frame would roll back the target turn's own edits and leave
18
+ * the work tree older than the conversation.
4
19
  */
5
20
  import { type ApplySnapshotOptions, type CoarseDiffStats } from "./engine.ts";
6
21
  import type { FileHistorySnapshot } from "./snapshot.ts";
22
+ /** Minimal per-entry view needed to walk the tree and classify targets. */
23
+ export interface EntryView {
24
+ id: string;
25
+ parentId: string | null;
26
+ /** True for entries navigateTree treats as turn starts (user / custom_message → leaf = parent). */
27
+ isTurnAnchor: boolean;
28
+ }
7
29
  /** Minimal session view we need to walk the entry tree. */
8
30
  export interface EntryTreeView {
9
- getEntry(id: string): {
10
- id: string;
11
- parentId: string | null;
12
- } | undefined;
31
+ getEntry(id: string): EntryView | undefined;
13
32
  }
14
- /** Collect the ancestor id set of `targetId` (inclusive), walking parentId to root. */
15
- export declare function ancestorIds(view: EntryTreeView, targetId: string): Set<string>;
16
33
  /**
17
- * The snapshot that best matches navigating to `targetId`: the most recent
18
- * snapshot (chronological) whose turn anchor (userEntryId) is an ancestor of, or
19
- * equal to, the target. Undefined when no recorded turn precedes the target.
34
+ * The snapshot that matches navigating to `targetId`, per the rules in the
35
+ * module header. Undefined when no recorded state corresponds to the target
36
+ * (then the work tree should be left alone).
20
37
  */
21
38
  export declare function snapshotForEntry(snapshots: FileHistorySnapshot[], view: EntryTreeView, targetId: string): FileHistorySnapshot | undefined;
22
39
  /**
@@ -1 +1 @@
1
- {"version":3,"file":"restore.d.ts","sourceRoot":"","sources":["../../../src/extensions/rewind/restore.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,OAAO,EACN,KAAK,oBAAoB,EAEzB,KAAK,eAAe,EAGpB,MAAM,aAAa,CAAC;AACrB,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,eAAe,CAAC;AAEzD,2DAA2D;AAC3D,MAAM,WAAW,aAAa;IAC7B,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG;QAAE,EAAE,EAAE,MAAM,CAAC;QAAC,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAA;KAAE,GAAG,SAAS,CAAC;CAC1E;AAED,uFAAuF;AACvF,wBAAgB,WAAW,CAAC,IAAI,EAAE,aAAa,EAAE,QAAQ,EAAE,MAAM,GAAG,GAAG,CAAC,MAAM,CAAC,CAS9E;AAED;;;;GAIG;AACH,wBAAgB,gBAAgB,CAC/B,SAAS,EAAE,mBAAmB,EAAE,EAChC,IAAI,EAAE,aAAa,EACnB,QAAQ,EAAE,MAAM,GACd,mBAAmB,GAAG,SAAS,CAOjC;AAED;;;GAGG;AACH,wBAAgB,oBAAoB,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,mBAAmB,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAExG;AAED;;;GAGG;AACH,wBAAgB,uBAAuB,CACtC,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,mBAAmB,EAC7B,YAAY,EAAE,SAAS,MAAM,EAAE,GAC7B,OAAO,CAAC,eAAe,CAAC,CAE1B;AAED;;;GAGG;AACH,wBAAsB,iBAAiB,CACtC,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,mBAAmB,EAC7B,IAAI,CAAC,EAAE,oBAAoB,GACzB,OAAO,CAAC,MAAM,EAAE,CAAC,CAEnB","sourcesContent":["/**\n * restore.ts — map a /tree navigation target to the snapshot that should be\n * restored and apply it via the engine.\n */\nimport {\n\ttype ApplySnapshotOptions,\n\tapplySnapshot,\n\ttype CoarseDiffStats,\n\tcollectChangeDiffStats,\n\tcollectChanges,\n} from \"./engine.ts\";\nimport type { FileHistorySnapshot } from \"./snapshot.ts\";\n\n/** Minimal session view we need to walk the entry tree. */\nexport interface EntryTreeView {\n\tgetEntry(id: string): { id: string; parentId: string | null } | undefined;\n}\n\n/** Collect the ancestor id set of `targetId` (inclusive), walking parentId to root. */\nexport function ancestorIds(view: EntryTreeView, targetId: string): Set<string> {\n\tconst set = new Set<string>();\n\tlet cur: string | null = targetId;\n\tlet guard = 0;\n\twhile (cur && !set.has(cur) && guard++ < 100_000) {\n\t\tset.add(cur);\n\t\tcur = view.getEntry(cur)?.parentId ?? null;\n\t}\n\treturn set;\n}\n\n/**\n * The snapshot that best matches navigating to `targetId`: the most recent\n * snapshot (chronological) whose turn anchor (userEntryId) is an ancestor of, or\n * equal to, the target. Undefined when no recorded turn precedes the target.\n */\nexport function snapshotForEntry(\n\tsnapshots: FileHistorySnapshot[],\n\tview: EntryTreeView,\n\ttargetId: string,\n): FileHistorySnapshot | undefined {\n\tconst ancestors = ancestorIds(view, targetId);\n\tlet best: FileHistorySnapshot | undefined;\n\tfor (const snap of snapshots) {\n\t\tif (ancestors.has(snap.userEntryId)) best = snap; // snapshots are in chronological order\n\t}\n\treturn best;\n}\n\n/**\n * Absolute file paths restoring to this snapshot would change on disk (empty =\n * none). Callers use the length for the count and the paths for the preview.\n */\nexport function snapshotChangedPaths(sessionId: string, snapshot: FileHistorySnapshot): Promise<string[]> {\n\treturn collectChanges(sessionId, snapshot);\n}\n\n/**\n * Coarse +N/−M line stats for the confirm dialog (paths must already be known\n * changed — typically from snapshotChangedPaths).\n */\nexport function snapshotChangeDiffStats(\n\tsessionId: string,\n\tsnapshot: FileHistorySnapshot,\n\tchangedPaths: readonly string[],\n): Promise<CoarseDiffStats> {\n\treturn collectChangeDiffStats(sessionId, snapshot, changedPaths);\n}\n\n/**\n * Restore the work tree to `snapshot` and return changed absolute paths.\n * Pass `onlyPaths` from a prior snapshotChangedPaths call to skip re-compare.\n */\nexport async function restoreToSnapshot(\n\tsessionId: string,\n\tsnapshot: FileHistorySnapshot,\n\topts?: ApplySnapshotOptions,\n): Promise<string[]> {\n\treturn applySnapshot(sessionId, snapshot, opts);\n}\n"]}
1
+ {"version":3,"file":"restore.d.ts","sourceRoot":"","sources":["../../../src/extensions/rewind/restore.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;GAkBG;AACH,OAAO,EACN,KAAK,oBAAoB,EAEzB,KAAK,eAAe,EAGpB,MAAM,aAAa,CAAC;AACrB,OAAO,KAAK,EAAE,mBAAmB,EAAE,MAAM,eAAe,CAAC;AAEzD,2EAA2E;AAC3E,MAAM,WAAW,SAAS;IACzB,EAAE,EAAE,MAAM,CAAC;IACX,QAAQ,EAAE,MAAM,GAAG,IAAI,CAAC;IACxB,qGAAmG;IACnG,YAAY,EAAE,OAAO,CAAC;CACtB;AAED,2DAA2D;AAC3D,MAAM,WAAW,aAAa;IAC7B,QAAQ,CAAC,EAAE,EAAE,MAAM,GAAG,SAAS,GAAG,SAAS,CAAC;CAC5C;AAiCD;;;;GAIG;AACH,wBAAgB,gBAAgB,CAC/B,SAAS,EAAE,mBAAmB,EAAE,EAChC,IAAI,EAAE,aAAa,EACnB,QAAQ,EAAE,MAAM,GACd,mBAAmB,GAAG,SAAS,CAkBjC;AAED;;;GAGG;AACH,wBAAgB,oBAAoB,CAAC,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,mBAAmB,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAExG;AAED;;;GAGG;AACH,wBAAgB,uBAAuB,CACtC,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,mBAAmB,EAC7B,YAAY,EAAE,SAAS,MAAM,EAAE,GAC7B,OAAO,CAAC,eAAe,CAAC,CAE1B;AAED;;;GAGG;AACH,wBAAsB,iBAAiB,CACtC,SAAS,EAAE,MAAM,EACjB,QAAQ,EAAE,mBAAmB,EAC7B,IAAI,CAAC,EAAE,oBAAoB,GACzB,OAAO,CAAC,MAAM,EAAE,CAAC,CAEnB","sourcesContent":["/**\n * restore.ts — map a /tree navigation target to the snapshot that should be\n * restored and apply it via the engine.\n *\n * Snapshot semantics: a frame anchored at user entry U records the work tree\n * at the moment U's run STARTED (see engine.ts beginTurn). The selection here\n * must mirror AgentSession.navigateTree's leaf rules:\n *\n * - user / custom_message target: navigateTree removes the target's turn\n * from the branch (leaf = parent, text returns to the editor) → restore\n * to the state BEFORE that turn = the snapshot anchored at the target.\n * - any other target: the target's turn STAYS in the conversation\n * (leaf = target) → restore to the state AFTER that turn = the first\n * chronological snapshot anchored at a descendant turn (it recorded the\n * work tree when the next run began). No such snapshot → nothing was\n * recorded after the target, so offer no restore at all — falling back to\n * an ancestor frame would roll back the target turn's own edits and leave\n * the work tree older than the conversation.\n */\nimport {\n\ttype ApplySnapshotOptions,\n\tapplySnapshot,\n\ttype CoarseDiffStats,\n\tcollectChangeDiffStats,\n\tcollectChanges,\n} from \"./engine.ts\";\nimport type { FileHistorySnapshot } from \"./snapshot.ts\";\n\n/** Minimal per-entry view needed to walk the tree and classify targets. */\nexport interface EntryView {\n\tid: string;\n\tparentId: string | null;\n\t/** True for entries navigateTree treats as turn starts (user / custom_message → leaf = parent). */\n\tisTurnAnchor: boolean;\n}\n\n/** Minimal session view we need to walk the entry tree. */\nexport interface EntryTreeView {\n\tgetEntry(id: string): EntryView | undefined;\n}\n\n/** Cycle guard for parent-chain walks (ids are acyclic in practice). */\nconst ANCESTOR_WALK_GUARD = 100_000;\n\n/** True when `id`'s parent chain passes through `ancestorId` (strict descent). */\nfunction isDescendantOf(view: EntryTreeView, id: string, ancestorId: string): boolean {\n\tlet cur = view.getEntry(id)?.parentId ?? null;\n\tlet guard = 0;\n\twhile (cur !== null && guard++ < ANCESTOR_WALK_GUARD) {\n\t\tif (cur === ancestorId) return true;\n\t\tcur = view.getEntry(cur)?.parentId ?? null;\n\t}\n\treturn false;\n}\n\n/**\n * First (chronological) snapshot anchored at a strict descendant of `entryId`\n * — the run that started right after `entryId` existed, so its frame records\n * the work tree state \"just after\" the entry.\n */\nfunction snapshotAfterEntry(\n\tsnapshots: FileHistorySnapshot[],\n\tview: EntryTreeView,\n\tentryId: string,\n): FileHistorySnapshot | undefined {\n\tfor (const snap of snapshots) {\n\t\tif (!snap.userEntryId) continue;\n\t\tif (isDescendantOf(view, snap.userEntryId, entryId)) return snap;\n\t}\n\treturn undefined;\n}\n\n/**\n * The snapshot that matches navigating to `targetId`, per the rules in the\n * module header. Undefined when no recorded state corresponds to the target\n * (then the work tree should be left alone).\n */\nexport function snapshotForEntry(\n\tsnapshots: FileHistorySnapshot[],\n\tview: EntryTreeView,\n\ttargetId: string,\n): FileHistorySnapshot | undefined {\n\tconst target = view.getEntry(targetId);\n\tif (!target) return undefined;\n\n\tif (target.isTurnAnchor) {\n\t\t// Exact anchor match: the frame recording the state right before this\n\t\t// turn ran. Chronologically last wins if duplicate anchors exist.\n\t\tlet exact: FileHistorySnapshot | undefined;\n\t\tfor (const snap of snapshots) {\n\t\t\tif (snap.userEntryId === targetId) exact = snap;\n\t\t}\n\t\tif (exact) return exact;\n\t\t// No frame for this turn (it recorded no changes): the state before it\n\t\t// equals the state after its parent.\n\t\treturn target.parentId === null ? undefined : snapshotAfterEntry(snapshots, view, target.parentId);\n\t}\n\n\treturn snapshotAfterEntry(snapshots, view, targetId);\n}\n\n/**\n * Absolute file paths restoring to this snapshot would change on disk (empty =\n * none). Callers use the length for the count and the paths for the preview.\n */\nexport function snapshotChangedPaths(sessionId: string, snapshot: FileHistorySnapshot): Promise<string[]> {\n\treturn collectChanges(sessionId, snapshot);\n}\n\n/**\n * Coarse +N/−M line stats for the confirm dialog (paths must already be known\n * changed — typically from snapshotChangedPaths).\n */\nexport function snapshotChangeDiffStats(\n\tsessionId: string,\n\tsnapshot: FileHistorySnapshot,\n\tchangedPaths: readonly string[],\n): Promise<CoarseDiffStats> {\n\treturn collectChangeDiffStats(sessionId, snapshot, changedPaths);\n}\n\n/**\n * Restore the work tree to `snapshot` and return changed absolute paths.\n * Pass `onlyPaths` from a prior snapshotChangedPaths call to skip re-compare.\n */\nexport async function restoreToSnapshot(\n\tsessionId: string,\n\tsnapshot: FileHistorySnapshot,\n\topts?: ApplySnapshotOptions,\n): Promise<string[]> {\n\treturn applySnapshot(sessionId, snapshot, opts);\n}\n"]}
@@ -1,32 +1,74 @@
1
1
  /**
2
2
  * restore.ts — map a /tree navigation target to the snapshot that should be
3
3
  * restored and apply it via the engine.
4
+ *
5
+ * Snapshot semantics: a frame anchored at user entry U records the work tree
6
+ * at the moment U's run STARTED (see engine.ts beginTurn). The selection here
7
+ * must mirror AgentSession.navigateTree's leaf rules:
8
+ *
9
+ * - user / custom_message target: navigateTree removes the target's turn
10
+ * from the branch (leaf = parent, text returns to the editor) → restore
11
+ * to the state BEFORE that turn = the snapshot anchored at the target.
12
+ * - any other target: the target's turn STAYS in the conversation
13
+ * (leaf = target) → restore to the state AFTER that turn = the first
14
+ * chronological snapshot anchored at a descendant turn (it recorded the
15
+ * work tree when the next run began). No such snapshot → nothing was
16
+ * recorded after the target, so offer no restore at all — falling back to
17
+ * an ancestor frame would roll back the target turn's own edits and leave
18
+ * the work tree older than the conversation.
4
19
  */
5
20
  import { applySnapshot, collectChangeDiffStats, collectChanges, } from "./engine.js";
6
- /** Collect the ancestor id set of `targetId` (inclusive), walking parentId to root. */
7
- export function ancestorIds(view, targetId) {
8
- const set = new Set();
9
- let cur = targetId;
21
+ /** Cycle guard for parent-chain walks (ids are acyclic in practice). */
22
+ const ANCESTOR_WALK_GUARD = 100_000;
23
+ /** True when `id`'s parent chain passes through `ancestorId` (strict descent). */
24
+ function isDescendantOf(view, id, ancestorId) {
25
+ let cur = view.getEntry(id)?.parentId ?? null;
10
26
  let guard = 0;
11
- while (cur && !set.has(cur) && guard++ < 100_000) {
12
- set.add(cur);
27
+ while (cur !== null && guard++ < ANCESTOR_WALK_GUARD) {
28
+ if (cur === ancestorId)
29
+ return true;
13
30
  cur = view.getEntry(cur)?.parentId ?? null;
14
31
  }
15
- return set;
32
+ return false;
16
33
  }
17
34
  /**
18
- * The snapshot that best matches navigating to `targetId`: the most recent
19
- * snapshot (chronological) whose turn anchor (userEntryId) is an ancestor of, or
20
- * equal to, the target. Undefined when no recorded turn precedes the target.
35
+ * First (chronological) snapshot anchored at a strict descendant of `entryId`
36
+ * — the run that started right after `entryId` existed, so its frame records
37
+ * the work tree state "just after" the entry.
21
38
  */
22
- export function snapshotForEntry(snapshots, view, targetId) {
23
- const ancestors = ancestorIds(view, targetId);
24
- let best;
39
+ function snapshotAfterEntry(snapshots, view, entryId) {
25
40
  for (const snap of snapshots) {
26
- if (ancestors.has(snap.userEntryId))
27
- best = snap; // snapshots are in chronological order
41
+ if (!snap.userEntryId)
42
+ continue;
43
+ if (isDescendantOf(view, snap.userEntryId, entryId))
44
+ return snap;
45
+ }
46
+ return undefined;
47
+ }
48
+ /**
49
+ * The snapshot that matches navigating to `targetId`, per the rules in the
50
+ * module header. Undefined when no recorded state corresponds to the target
51
+ * (then the work tree should be left alone).
52
+ */
53
+ export function snapshotForEntry(snapshots, view, targetId) {
54
+ const target = view.getEntry(targetId);
55
+ if (!target)
56
+ return undefined;
57
+ if (target.isTurnAnchor) {
58
+ // Exact anchor match: the frame recording the state right before this
59
+ // turn ran. Chronologically last wins if duplicate anchors exist.
60
+ let exact;
61
+ for (const snap of snapshots) {
62
+ if (snap.userEntryId === targetId)
63
+ exact = snap;
64
+ }
65
+ if (exact)
66
+ return exact;
67
+ // No frame for this turn (it recorded no changes): the state before it
68
+ // equals the state after its parent.
69
+ return target.parentId === null ? undefined : snapshotAfterEntry(snapshots, view, target.parentId);
28
70
  }
29
- return best;
71
+ return snapshotAfterEntry(snapshots, view, targetId);
30
72
  }
31
73
  /**
32
74
  * Absolute file paths restoring to this snapshot would change on disk (empty =
@@ -1 +1 @@
1
- {"version":3,"file":"restore.js","sourceRoot":"","sources":["../../../src/extensions/rewind/restore.ts"],"names":[],"mappings":"AAAA;;;GAGG;AACH,OAAO,EAEN,aAAa,EAEb,sBAAsB,EACtB,cAAc,GACd,MAAM,aAAa,CAAC;AAQrB,uFAAuF;AACvF,MAAM,UAAU,WAAW,CAAC,IAAmB,EAAE,QAAgB,EAAe;IAC/E,MAAM,GAAG,GAAG,IAAI,GAAG,EAAU,CAAC;IAC9B,IAAI,GAAG,GAAkB,QAAQ,CAAC;IAClC,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,OAAO,GAAG,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,IAAI,KAAK,EAAE,GAAG,OAAO,EAAE,CAAC;QAClD,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QACb,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,QAAQ,IAAI,IAAI,CAAC;IAC5C,CAAC;IACD,OAAO,GAAG,CAAC;AAAA,CACX;AAED;;;;GAIG;AACH,MAAM,UAAU,gBAAgB,CAC/B,SAAgC,EAChC,IAAmB,EACnB,QAAgB,EACkB;IAClC,MAAM,SAAS,GAAG,WAAW,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IAC9C,IAAI,IAAqC,CAAC;IAC1C,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,CAAC;QAC9B,IAAI,SAAS,CAAC,GAAG,CAAC,IAAI,CAAC,WAAW,CAAC;YAAE,IAAI,GAAG,IAAI,CAAC,CAAC,uCAAuC;IAC1F,CAAC;IACD,OAAO,IAAI,CAAC;AAAA,CACZ;AAED;;;GAGG;AACH,MAAM,UAAU,oBAAoB,CAAC,SAAiB,EAAE,QAA6B,EAAqB;IACzG,OAAO,cAAc,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;AAAA,CAC3C;AAED;;;GAGG;AACH,MAAM,UAAU,uBAAuB,CACtC,SAAiB,EACjB,QAA6B,EAC7B,YAA+B,EACJ;IAC3B,OAAO,sBAAsB,CAAC,SAAS,EAAE,QAAQ,EAAE,YAAY,CAAC,CAAC;AAAA,CACjE;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACtC,SAAiB,EACjB,QAA6B,EAC7B,IAA2B,EACP;IACpB,OAAO,aAAa,CAAC,SAAS,EAAE,QAAQ,EAAE,IAAI,CAAC,CAAC;AAAA,CAChD","sourcesContent":["/**\n * restore.ts — map a /tree navigation target to the snapshot that should be\n * restored and apply it via the engine.\n */\nimport {\n\ttype ApplySnapshotOptions,\n\tapplySnapshot,\n\ttype CoarseDiffStats,\n\tcollectChangeDiffStats,\n\tcollectChanges,\n} from \"./engine.ts\";\nimport type { FileHistorySnapshot } from \"./snapshot.ts\";\n\n/** Minimal session view we need to walk the entry tree. */\nexport interface EntryTreeView {\n\tgetEntry(id: string): { id: string; parentId: string | null } | undefined;\n}\n\n/** Collect the ancestor id set of `targetId` (inclusive), walking parentId to root. */\nexport function ancestorIds(view: EntryTreeView, targetId: string): Set<string> {\n\tconst set = new Set<string>();\n\tlet cur: string | null = targetId;\n\tlet guard = 0;\n\twhile (cur && !set.has(cur) && guard++ < 100_000) {\n\t\tset.add(cur);\n\t\tcur = view.getEntry(cur)?.parentId ?? null;\n\t}\n\treturn set;\n}\n\n/**\n * The snapshot that best matches navigating to `targetId`: the most recent\n * snapshot (chronological) whose turn anchor (userEntryId) is an ancestor of, or\n * equal to, the target. Undefined when no recorded turn precedes the target.\n */\nexport function snapshotForEntry(\n\tsnapshots: FileHistorySnapshot[],\n\tview: EntryTreeView,\n\ttargetId: string,\n): FileHistorySnapshot | undefined {\n\tconst ancestors = ancestorIds(view, targetId);\n\tlet best: FileHistorySnapshot | undefined;\n\tfor (const snap of snapshots) {\n\t\tif (ancestors.has(snap.userEntryId)) best = snap; // snapshots are in chronological order\n\t}\n\treturn best;\n}\n\n/**\n * Absolute file paths restoring to this snapshot would change on disk (empty =\n * none). Callers use the length for the count and the paths for the preview.\n */\nexport function snapshotChangedPaths(sessionId: string, snapshot: FileHistorySnapshot): Promise<string[]> {\n\treturn collectChanges(sessionId, snapshot);\n}\n\n/**\n * Coarse +N/−M line stats for the confirm dialog (paths must already be known\n * changed — typically from snapshotChangedPaths).\n */\nexport function snapshotChangeDiffStats(\n\tsessionId: string,\n\tsnapshot: FileHistorySnapshot,\n\tchangedPaths: readonly string[],\n): Promise<CoarseDiffStats> {\n\treturn collectChangeDiffStats(sessionId, snapshot, changedPaths);\n}\n\n/**\n * Restore the work tree to `snapshot` and return changed absolute paths.\n * Pass `onlyPaths` from a prior snapshotChangedPaths call to skip re-compare.\n */\nexport async function restoreToSnapshot(\n\tsessionId: string,\n\tsnapshot: FileHistorySnapshot,\n\topts?: ApplySnapshotOptions,\n): Promise<string[]> {\n\treturn applySnapshot(sessionId, snapshot, opts);\n}\n"]}
1
+ {"version":3,"file":"restore.js","sourceRoot":"","sources":["../../../src/extensions/rewind/restore.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;;;GAkBG;AACH,OAAO,EAEN,aAAa,EAEb,sBAAsB,EACtB,cAAc,GACd,MAAM,aAAa,CAAC;AAgBrB,wEAAwE;AACxE,MAAM,mBAAmB,GAAG,OAAO,CAAC;AAEpC,kFAAkF;AAClF,SAAS,cAAc,CAAC,IAAmB,EAAE,EAAU,EAAE,UAAkB,EAAW;IACrF,IAAI,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE,QAAQ,IAAI,IAAI,CAAC;IAC9C,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,OAAO,GAAG,KAAK,IAAI,IAAI,KAAK,EAAE,GAAG,mBAAmB,EAAE,CAAC;QACtD,IAAI,GAAG,KAAK,UAAU;YAAE,OAAO,IAAI,CAAC;QACpC,GAAG,GAAG,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,QAAQ,IAAI,IAAI,CAAC;IAC5C,CAAC;IACD,OAAO,KAAK,CAAC;AAAA,CACb;AAED;;;;GAIG;AACH,SAAS,kBAAkB,CAC1B,SAAgC,EAChC,IAAmB,EACnB,OAAe,EACmB;IAClC,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,CAAC;QAC9B,IAAI,CAAC,IAAI,CAAC,WAAW;YAAE,SAAS;QAChC,IAAI,cAAc,CAAC,IAAI,EAAE,IAAI,CAAC,WAAW,EAAE,OAAO,CAAC;YAAE,OAAO,IAAI,CAAC;IAClE,CAAC;IACD,OAAO,SAAS,CAAC;AAAA,CACjB;AAED;;;;GAIG;AACH,MAAM,UAAU,gBAAgB,CAC/B,SAAgC,EAChC,IAAmB,EACnB,QAAgB,EACkB;IAClC,MAAM,MAAM,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACvC,IAAI,CAAC,MAAM;QAAE,OAAO,SAAS,CAAC;IAE9B,IAAI,MAAM,CAAC,YAAY,EAAE,CAAC;QACzB,sEAAsE;QACtE,kEAAkE;QAClE,IAAI,KAAsC,CAAC;QAC3C,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,CAAC;YAC9B,IAAI,IAAI,CAAC,WAAW,KAAK,QAAQ;gBAAE,KAAK,GAAG,IAAI,CAAC;QACjD,CAAC;QACD,IAAI,KAAK;YAAE,OAAO,KAAK,CAAC;QACxB,uEAAuE;QACvE,qCAAqC;QACrC,OAAO,MAAM,CAAC,QAAQ,KAAK,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,kBAAkB,CAAC,SAAS,EAAE,IAAI,EAAE,MAAM,CAAC,QAAQ,CAAC,CAAC;IACpG,CAAC;IAED,OAAO,kBAAkB,CAAC,SAAS,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;AAAA,CACrD;AAED;;;GAGG;AACH,MAAM,UAAU,oBAAoB,CAAC,SAAiB,EAAE,QAA6B,EAAqB;IACzG,OAAO,cAAc,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;AAAA,CAC3C;AAED;;;GAGG;AACH,MAAM,UAAU,uBAAuB,CACtC,SAAiB,EACjB,QAA6B,EAC7B,YAA+B,EACJ;IAC3B,OAAO,sBAAsB,CAAC,SAAS,EAAE,QAAQ,EAAE,YAAY,CAAC,CAAC;AAAA,CACjE;AAED;;;GAGG;AACH,MAAM,CAAC,KAAK,UAAU,iBAAiB,CACtC,SAAiB,EACjB,QAA6B,EAC7B,IAA2B,EACP;IACpB,OAAO,aAAa,CAAC,SAAS,EAAE,QAAQ,EAAE,IAAI,CAAC,CAAC;AAAA,CAChD","sourcesContent":["/**\n * restore.ts — map a /tree navigation target to the snapshot that should be\n * restored and apply it via the engine.\n *\n * Snapshot semantics: a frame anchored at user entry U records the work tree\n * at the moment U's run STARTED (see engine.ts beginTurn). The selection here\n * must mirror AgentSession.navigateTree's leaf rules:\n *\n * - user / custom_message target: navigateTree removes the target's turn\n * from the branch (leaf = parent, text returns to the editor) → restore\n * to the state BEFORE that turn = the snapshot anchored at the target.\n * - any other target: the target's turn STAYS in the conversation\n * (leaf = target) → restore to the state AFTER that turn = the first\n * chronological snapshot anchored at a descendant turn (it recorded the\n * work tree when the next run began). No such snapshot → nothing was\n * recorded after the target, so offer no restore at all — falling back to\n * an ancestor frame would roll back the target turn's own edits and leave\n * the work tree older than the conversation.\n */\nimport {\n\ttype ApplySnapshotOptions,\n\tapplySnapshot,\n\ttype CoarseDiffStats,\n\tcollectChangeDiffStats,\n\tcollectChanges,\n} from \"./engine.ts\";\nimport type { FileHistorySnapshot } from \"./snapshot.ts\";\n\n/** Minimal per-entry view needed to walk the tree and classify targets. */\nexport interface EntryView {\n\tid: string;\n\tparentId: string | null;\n\t/** True for entries navigateTree treats as turn starts (user / custom_message → leaf = parent). */\n\tisTurnAnchor: boolean;\n}\n\n/** Minimal session view we need to walk the entry tree. */\nexport interface EntryTreeView {\n\tgetEntry(id: string): EntryView | undefined;\n}\n\n/** Cycle guard for parent-chain walks (ids are acyclic in practice). */\nconst ANCESTOR_WALK_GUARD = 100_000;\n\n/** True when `id`'s parent chain passes through `ancestorId` (strict descent). */\nfunction isDescendantOf(view: EntryTreeView, id: string, ancestorId: string): boolean {\n\tlet cur = view.getEntry(id)?.parentId ?? null;\n\tlet guard = 0;\n\twhile (cur !== null && guard++ < ANCESTOR_WALK_GUARD) {\n\t\tif (cur === ancestorId) return true;\n\t\tcur = view.getEntry(cur)?.parentId ?? null;\n\t}\n\treturn false;\n}\n\n/**\n * First (chronological) snapshot anchored at a strict descendant of `entryId`\n * — the run that started right after `entryId` existed, so its frame records\n * the work tree state \"just after\" the entry.\n */\nfunction snapshotAfterEntry(\n\tsnapshots: FileHistorySnapshot[],\n\tview: EntryTreeView,\n\tentryId: string,\n): FileHistorySnapshot | undefined {\n\tfor (const snap of snapshots) {\n\t\tif (!snap.userEntryId) continue;\n\t\tif (isDescendantOf(view, snap.userEntryId, entryId)) return snap;\n\t}\n\treturn undefined;\n}\n\n/**\n * The snapshot that matches navigating to `targetId`, per the rules in the\n * module header. Undefined when no recorded state corresponds to the target\n * (then the work tree should be left alone).\n */\nexport function snapshotForEntry(\n\tsnapshots: FileHistorySnapshot[],\n\tview: EntryTreeView,\n\ttargetId: string,\n): FileHistorySnapshot | undefined {\n\tconst target = view.getEntry(targetId);\n\tif (!target) return undefined;\n\n\tif (target.isTurnAnchor) {\n\t\t// Exact anchor match: the frame recording the state right before this\n\t\t// turn ran. Chronologically last wins if duplicate anchors exist.\n\t\tlet exact: FileHistorySnapshot | undefined;\n\t\tfor (const snap of snapshots) {\n\t\t\tif (snap.userEntryId === targetId) exact = snap;\n\t\t}\n\t\tif (exact) return exact;\n\t\t// No frame for this turn (it recorded no changes): the state before it\n\t\t// equals the state after its parent.\n\t\treturn target.parentId === null ? undefined : snapshotAfterEntry(snapshots, view, target.parentId);\n\t}\n\n\treturn snapshotAfterEntry(snapshots, view, targetId);\n}\n\n/**\n * Absolute file paths restoring to this snapshot would change on disk (empty =\n * none). Callers use the length for the count and the paths for the preview.\n */\nexport function snapshotChangedPaths(sessionId: string, snapshot: FileHistorySnapshot): Promise<string[]> {\n\treturn collectChanges(sessionId, snapshot);\n}\n\n/**\n * Coarse +N/−M line stats for the confirm dialog (paths must already be known\n * changed — typically from snapshotChangedPaths).\n */\nexport function snapshotChangeDiffStats(\n\tsessionId: string,\n\tsnapshot: FileHistorySnapshot,\n\tchangedPaths: readonly string[],\n): Promise<CoarseDiffStats> {\n\treturn collectChangeDiffStats(sessionId, snapshot, changedPaths);\n}\n\n/**\n * Restore the work tree to `snapshot` and return changed absolute paths.\n * Pass `onlyPaths` from a prior snapshotChangedPaths call to skip re-compare.\n */\nexport async function restoreToSnapshot(\n\tsessionId: string,\n\tsnapshot: FileHistorySnapshot,\n\topts?: ApplySnapshotOptions,\n): Promise<string[]> {\n\treturn applySnapshot(sessionId, snapshot, opts);\n}\n"]}
@@ -5,6 +5,7 @@ export declare function emptyUsage(): SubagentUsage;
5
5
  export declare function addUsage(target: SubagentUsage, usage: Usage | undefined): void;
6
6
  export declare function mergeUsage(target: SubagentUsage, source: SubagentUsage): void;
7
7
  export declare function toNestedUsage(usage: SubagentUsage): Usage;
8
+ export declare function utf8Prefix(text: string, maxBytes: number): string;
8
9
  export declare function boundText(text: string, maxBytes: number): string;
9
10
  export declare function tailText(text: string, maxBytes: number): string;
10
11
  export declare function assistantText(message: AgentMessage | undefined): string;
@@ -1 +1 @@
1
- {"version":3,"file":"activity.d.ts","sourceRoot":"","sources":["../../../src/extensions/subagent/activity.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAClE,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,uBAAuB,CAAC;AAEnD,OAAO,KAAK,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAE9D,wBAAgB,UAAU,IAAI,aAAa,CAW1C;AAED,wBAAgB,QAAQ,CAAC,MAAM,EAAE,aAAa,EAAE,KAAK,EAAE,KAAK,GAAG,SAAS,GAAG,IAAI,CAQ9E;AAED,wBAAgB,UAAU,CAAC,MAAM,EAAE,aAAa,EAAE,MAAM,EAAE,aAAa,GAAG,IAAI,CAS7E;AAED,wBAAgB,aAAa,CAAC,KAAK,EAAE,aAAa,GAAG,KAAK,CAezD;AASD,wBAAgB,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,MAAM,CAgBhE;AAED,wBAAgB,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,MAAM,CAK/D;AAED,wBAAgB,aAAa,CAAC,OAAO,EAAE,YAAY,GAAG,SAAS,GAAG,MAAM,CAMvE;AAED,wBAAgB,kBAAkB,CAAC,QAAQ,EAAE,SAAS,YAAY,EAAE,GAAG,MAAM,CAM5E;AAED,wBAAgB,cAAc,CAAC,UAAU,EAAE,YAAY,EAAE,EAAE,QAAQ,EAAE,YAAY,GAAG,IAAI,CAUvF;AAED,wBAAgB,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAEhD;AAED,wBAAgB,eAAe,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,GAAG,MAAM,CAWvE;AAED,wBAAgB,aAAa,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,CAoBrD","sourcesContent":["import type { AgentMessage } from \"@earendil-works/pi-agent-core\";\nimport type { Usage } from \"@earendil-works/pi-ai\";\nimport { ACTIVITY_LIMIT, ACTIVITY_TEXT_LIMIT, LIVE_TEXT_LIMIT, SINGLE_OUTPUT_LIMIT } from \"./constants.ts\";\nimport type { SubagentUsage, ToolActivity } from \"./types.ts\";\n\nexport function emptyUsage(): SubagentUsage {\n\treturn {\n\t\tturns: 0,\n\t\ttoolUses: 0,\n\t\tinput: 0,\n\t\toutput: 0,\n\t\tcacheRead: 0,\n\t\tcacheWrite: 0,\n\t\ttotalTokens: 0,\n\t\tcost: 0,\n\t};\n}\n\nexport function addUsage(target: SubagentUsage, usage: Usage | undefined): void {\n\tif (!usage) return;\n\ttarget.input += usage.input ?? 0;\n\ttarget.output += usage.output ?? 0;\n\ttarget.cacheRead += usage.cacheRead ?? 0;\n\ttarget.cacheWrite += usage.cacheWrite ?? 0;\n\ttarget.totalTokens += usage.totalTokens ?? 0;\n\ttarget.cost += usage.cost?.total ?? 0;\n}\n\nexport function mergeUsage(target: SubagentUsage, source: SubagentUsage): void {\n\ttarget.turns += source.turns;\n\ttarget.toolUses += source.toolUses;\n\ttarget.input += source.input;\n\ttarget.output += source.output;\n\ttarget.cacheRead += source.cacheRead;\n\ttarget.cacheWrite += source.cacheWrite;\n\ttarget.totalTokens += source.totalTokens;\n\ttarget.cost += source.cost;\n}\n\nexport function toNestedUsage(usage: SubagentUsage): Usage {\n\treturn {\n\t\tinput: usage.input,\n\t\toutput: usage.output,\n\t\tcacheRead: usage.cacheRead,\n\t\tcacheWrite: usage.cacheWrite,\n\t\ttotalTokens: usage.totalTokens,\n\t\tcost: {\n\t\t\tinput: 0,\n\t\t\toutput: 0,\n\t\t\tcacheRead: 0,\n\t\t\tcacheWrite: 0,\n\t\t\ttotal: usage.cost,\n\t\t},\n\t};\n}\n\nfunction utf8Prefix(text: string, maxBytes: number): string {\n\tif (maxBytes <= 0) return \"\";\n\tlet output = text.slice(0, maxBytes);\n\twhile (Buffer.byteLength(output, \"utf8\") > maxBytes) output = output.slice(0, -1);\n\treturn output;\n}\n\nexport function boundText(text: string, maxBytes: number): string {\n\tif (maxBytes <= 0) return \"\";\n\tif (Buffer.byteLength(text, \"utf8\") <= maxBytes) return text;\n\tlet output = utf8Prefix(text, maxBytes);\n\tfor (let attempt = 0; attempt < 8; attempt++) {\n\t\tconst omitted = Buffer.byteLength(text, \"utf8\") - Buffer.byteLength(output, \"utf8\");\n\t\tconst notice = `\\n\\n[Output truncated: ${omitted} bytes omitted.]`;\n\t\tconst available = maxBytes - Buffer.byteLength(notice, \"utf8\");\n\t\tif (available <= 0) return utf8Prefix(\"[Output truncated.]\", maxBytes);\n\t\tconst next = utf8Prefix(text, available);\n\t\tif (next === output) return `${output}${notice}`;\n\t\toutput = next;\n\t}\n\tconst omitted = Buffer.byteLength(text, \"utf8\") - Buffer.byteLength(output, \"utf8\");\n\tconst notice = `\\n\\n[Output truncated: ${omitted} bytes omitted.]`;\n\treturn `${utf8Prefix(output, Math.max(0, maxBytes - Buffer.byteLength(notice, \"utf8\")))}${notice}`;\n}\n\nexport function tailText(text: string, maxBytes: number): string {\n\tif (Buffer.byteLength(text, \"utf8\") <= maxBytes) return text;\n\tlet output = text.slice(-maxBytes);\n\twhile (Buffer.byteLength(output, \"utf8\") > maxBytes) output = output.slice(1);\n\treturn `[Earlier output omitted.]\\n${output}`;\n}\n\nexport function assistantText(message: AgentMessage | undefined): string {\n\tif (!message || message.role !== \"assistant\") return \"\";\n\treturn message.content\n\t\t.filter((part): part is { type: \"text\"; text: string } => part.type === \"text\")\n\t\t.map((part) => part.text)\n\t\t.join(\"\");\n}\n\nexport function finalAssistantText(messages: readonly AgentMessage[]): string {\n\tfor (let index = messages.length - 1; index >= 0; index--) {\n\t\tconst text = assistantText(messages[index]);\n\t\tif (text) return boundText(text, SINGLE_OUTPUT_LIMIT);\n\t}\n\treturn \"\";\n}\n\nexport function appendActivity(activities: ToolActivity[], activity: ToolActivity): void {\n\tactivities.push(activity);\n\twhile (activities.length > ACTIVITY_LIMIT) activities.shift();\n\twhile (\n\t\tactivities.reduce((total, item) => total + item.summary.length + (item.resultSummary?.length ?? 0), 0) >\n\t\tACTIVITY_TEXT_LIMIT\n\t) {\n\t\tif (activities.length <= 1) break;\n\t\tactivities.shift();\n\t}\n}\n\nexport function setLiveText(text: string): string {\n\treturn tailText(text, LIVE_TEXT_LIMIT);\n}\n\nexport function activitySummary(toolName: string, args: unknown): string {\n\tif (!args || typeof args !== \"object\" || Array.isArray(args)) return toolName;\n\tconst input = args as Record<string, unknown>;\n\tconst path =\n\t\ttypeof input.path === \"string\" ? input.path : typeof input.file_path === \"string\" ? input.file_path : undefined;\n\tif (toolName === \"bash\" && typeof input.command === \"string\")\n\t\treturn `Run ${boundText(input.command.replace(/\\s+/gu, \" \"), 160)}`;\n\tif (path) return `${toolName} ${boundText(path, 180)}`;\n\tif (toolName === \"grep\" && typeof input.pattern === \"string\") return `Search ${boundText(input.pattern, 120)}`;\n\tif (toolName === \"find\" && typeof input.pattern === \"string\") return `Find ${boundText(input.pattern, 120)}`;\n\treturn toolName;\n}\n\nexport function resultSummary(result: unknown): string {\n\tif (!result || typeof result !== \"object\") return \"\";\n\tconst content = (result as { content?: unknown }).content;\n\tif (!Array.isArray(content)) return \"\";\n\treturn boundText(\n\t\tcontent\n\t\t\t.filter((part): part is { type: \"text\"; text: string } => {\n\t\t\t\treturn Boolean(\n\t\t\t\t\tpart &&\n\t\t\t\t\t\ttypeof part === \"object\" &&\n\t\t\t\t\t\t(part as { type?: unknown }).type === \"text\" &&\n\t\t\t\t\t\ttypeof (part as { text?: unknown }).text === \"string\",\n\t\t\t\t);\n\t\t\t})\n\t\t\t.map((part) => part.text)\n\t\t\t.join(\"\\n\")\n\t\t\t.replace(/\\s+/gu, \" \")\n\t\t\t.trim(),\n\t\t240,\n\t);\n}\n"]}
1
+ {"version":3,"file":"activity.d.ts","sourceRoot":"","sources":["../../../src/extensions/subagent/activity.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,+BAA+B,CAAC;AAClE,OAAO,KAAK,EAAE,KAAK,EAAE,MAAM,uBAAuB,CAAC;AAEnD,OAAO,KAAK,EAAE,aAAa,EAAE,YAAY,EAAE,MAAM,YAAY,CAAC;AAE9D,wBAAgB,UAAU,IAAI,aAAa,CAW1C;AAED,wBAAgB,QAAQ,CAAC,MAAM,EAAE,aAAa,EAAE,KAAK,EAAE,KAAK,GAAG,SAAS,GAAG,IAAI,CAW9E;AAED,wBAAgB,UAAU,CAAC,MAAM,EAAE,aAAa,EAAE,MAAM,EAAE,aAAa,GAAG,IAAI,CAY7E;AAED,wBAAgB,aAAa,CAAC,KAAK,EAAE,aAAa,GAAG,KAAK,CAezD;AAGD,wBAAgB,UAAU,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,MAAM,CAYjE;AAED,wBAAgB,SAAS,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,MAAM,CAgBhE;AAED,wBAAgB,QAAQ,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,GAAG,MAAM,CAK/D;AAED,wBAAgB,aAAa,CAAC,OAAO,EAAE,YAAY,GAAG,SAAS,GAAG,MAAM,CAMvE;AAED,wBAAgB,kBAAkB,CAAC,QAAQ,EAAE,SAAS,YAAY,EAAE,GAAG,MAAM,CAM5E;AAED,wBAAgB,cAAc,CAAC,UAAU,EAAE,YAAY,EAAE,EAAE,QAAQ,EAAE,YAAY,GAAG,IAAI,CAUvF;AAED,wBAAgB,WAAW,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM,CAEhD;AAED,wBAAgB,eAAe,CAAC,QAAQ,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,GAAG,MAAM,CAWvE;AAED,wBAAgB,aAAa,CAAC,MAAM,EAAE,OAAO,GAAG,MAAM,CAoBrD","sourcesContent":["import type { AgentMessage } from \"@earendil-works/pi-agent-core\";\nimport type { Usage } from \"@earendil-works/pi-ai\";\nimport { ACTIVITY_LIMIT, ACTIVITY_TEXT_LIMIT, LIVE_TEXT_LIMIT, SINGLE_OUTPUT_LIMIT } from \"./constants.ts\";\nimport type { SubagentUsage, ToolActivity } from \"./types.ts\";\n\nexport function emptyUsage(): SubagentUsage {\n\treturn {\n\t\tturns: 0,\n\t\ttoolUses: 0,\n\t\tinput: 0,\n\t\toutput: 0,\n\t\tcacheRead: 0,\n\t\tcacheWrite: 0,\n\t\ttotalTokens: 0,\n\t\tcost: 0,\n\t};\n}\n\nexport function addUsage(target: SubagentUsage, usage: Usage | undefined): void {\n\tif (!usage) return;\n\ttarget.input += usage.input ?? 0;\n\ttarget.output += usage.output ?? 0;\n\ttarget.cacheRead += usage.cacheRead ?? 0;\n\ttarget.cacheWrite += usage.cacheWrite ?? 0;\n\ttarget.totalTokens += usage.totalTokens ?? 0;\n\ttarget.cost += usage.cost?.total ?? 0;\n\t// Watermark, not a sum: the latest request's total is the context size,\n\t// and it can shrink again after the worker auto-compacts.\n\tif (usage.totalTokens) target.contextTokens = usage.totalTokens;\n}\n\nexport function mergeUsage(target: SubagentUsage, source: SubagentUsage): void {\n\ttarget.turns += source.turns;\n\ttarget.toolUses += source.toolUses;\n\ttarget.input += source.input;\n\ttarget.output += source.output;\n\ttarget.cacheRead += source.cacheRead;\n\ttarget.cacheWrite += source.cacheWrite;\n\ttarget.totalTokens += source.totalTokens;\n\ttarget.cost += source.cost;\n\tif (source.contextTokens) {\n\t\ttarget.contextTokens = Math.max(target.contextTokens ?? 0, source.contextTokens);\n\t}\n}\n\nexport function toNestedUsage(usage: SubagentUsage): Usage {\n\treturn {\n\t\tinput: usage.input,\n\t\toutput: usage.output,\n\t\tcacheRead: usage.cacheRead,\n\t\tcacheWrite: usage.cacheWrite,\n\t\ttotalTokens: usage.totalTokens,\n\t\tcost: {\n\t\t\tinput: 0,\n\t\t\toutput: 0,\n\t\t\tcacheRead: 0,\n\t\t\tcacheWrite: 0,\n\t\t\ttotal: usage.cost,\n\t\t},\n\t};\n}\n\n// Iterates code points so a surrogate pair is never split in half.\nexport function utf8Prefix(text: string, maxBytes: number): string {\n\tif (maxBytes <= 0) return \"\";\n\tif (Buffer.byteLength(text, \"utf8\") <= maxBytes) return text;\n\tlet output = \"\";\n\tlet bytes = 0;\n\tfor (const character of text) {\n\t\tconst characterBytes = Buffer.byteLength(character, \"utf8\");\n\t\tif (bytes + characterBytes > maxBytes) break;\n\t\toutput += character;\n\t\tbytes += characterBytes;\n\t}\n\treturn output;\n}\n\nexport function boundText(text: string, maxBytes: number): string {\n\tif (maxBytes <= 0) return \"\";\n\tif (Buffer.byteLength(text, \"utf8\") <= maxBytes) return text;\n\tlet output = utf8Prefix(text, maxBytes);\n\tfor (let attempt = 0; attempt < 8; attempt++) {\n\t\tconst omitted = Buffer.byteLength(text, \"utf8\") - Buffer.byteLength(output, \"utf8\");\n\t\tconst notice = `\\n\\n[Output truncated: ${omitted} bytes omitted.]`;\n\t\tconst available = maxBytes - Buffer.byteLength(notice, \"utf8\");\n\t\tif (available <= 0) return utf8Prefix(\"[Output truncated.]\", maxBytes);\n\t\tconst next = utf8Prefix(text, available);\n\t\tif (next === output) return `${output}${notice}`;\n\t\toutput = next;\n\t}\n\tconst omitted = Buffer.byteLength(text, \"utf8\") - Buffer.byteLength(output, \"utf8\");\n\tconst notice = `\\n\\n[Output truncated: ${omitted} bytes omitted.]`;\n\treturn `${utf8Prefix(output, Math.max(0, maxBytes - Buffer.byteLength(notice, \"utf8\")))}${notice}`;\n}\n\nexport function tailText(text: string, maxBytes: number): string {\n\tif (Buffer.byteLength(text, \"utf8\") <= maxBytes) return text;\n\tlet output = text.slice(-maxBytes);\n\twhile (Buffer.byteLength(output, \"utf8\") > maxBytes) output = output.slice(1);\n\treturn `[Earlier output omitted.]\\n${output}`;\n}\n\nexport function assistantText(message: AgentMessage | undefined): string {\n\tif (!message || message.role !== \"assistant\") return \"\";\n\treturn message.content\n\t\t.filter((part): part is { type: \"text\"; text: string } => part.type === \"text\")\n\t\t.map((part) => part.text)\n\t\t.join(\"\");\n}\n\nexport function finalAssistantText(messages: readonly AgentMessage[]): string {\n\tfor (let index = messages.length - 1; index >= 0; index--) {\n\t\tconst text = assistantText(messages[index]);\n\t\tif (text) return boundText(text, SINGLE_OUTPUT_LIMIT);\n\t}\n\treturn \"\";\n}\n\nexport function appendActivity(activities: ToolActivity[], activity: ToolActivity): void {\n\tactivities.push(activity);\n\twhile (activities.length > ACTIVITY_LIMIT) activities.shift();\n\twhile (\n\t\tactivities.reduce((total, item) => total + item.summary.length + (item.resultSummary?.length ?? 0), 0) >\n\t\tACTIVITY_TEXT_LIMIT\n\t) {\n\t\tif (activities.length <= 1) break;\n\t\tactivities.shift();\n\t}\n}\n\nexport function setLiveText(text: string): string {\n\treturn tailText(text, LIVE_TEXT_LIMIT);\n}\n\nexport function activitySummary(toolName: string, args: unknown): string {\n\tif (!args || typeof args !== \"object\" || Array.isArray(args)) return toolName;\n\tconst input = args as Record<string, unknown>;\n\tconst path =\n\t\ttypeof input.path === \"string\" ? input.path : typeof input.file_path === \"string\" ? input.file_path : undefined;\n\tif (toolName === \"bash\" && typeof input.command === \"string\")\n\t\treturn `Run ${boundText(input.command.replace(/\\s+/gu, \" \"), 160)}`;\n\tif (path) return `${toolName} ${boundText(path, 180)}`;\n\tif (toolName === \"grep\" && typeof input.pattern === \"string\") return `Search ${boundText(input.pattern, 120)}`;\n\tif (toolName === \"find\" && typeof input.pattern === \"string\") return `Find ${boundText(input.pattern, 120)}`;\n\treturn toolName;\n}\n\nexport function resultSummary(result: unknown): string {\n\tif (!result || typeof result !== \"object\") return \"\";\n\tconst content = (result as { content?: unknown }).content;\n\tif (!Array.isArray(content)) return \"\";\n\treturn boundText(\n\t\tcontent\n\t\t\t.filter((part): part is { type: \"text\"; text: string } => {\n\t\t\t\treturn Boolean(\n\t\t\t\t\tpart &&\n\t\t\t\t\t\ttypeof part === \"object\" &&\n\t\t\t\t\t\t(part as { type?: unknown }).type === \"text\" &&\n\t\t\t\t\t\ttypeof (part as { text?: unknown }).text === \"string\",\n\t\t\t\t);\n\t\t\t})\n\t\t\t.map((part) => part.text)\n\t\t\t.join(\"\\n\")\n\t\t\t.replace(/\\s+/gu, \" \")\n\t\t\t.trim(),\n\t\t240,\n\t);\n}\n"]}
@@ -20,6 +20,10 @@ export function addUsage(target, usage) {
20
20
  target.cacheWrite += usage.cacheWrite ?? 0;
21
21
  target.totalTokens += usage.totalTokens ?? 0;
22
22
  target.cost += usage.cost?.total ?? 0;
23
+ // Watermark, not a sum: the latest request's total is the context size,
24
+ // and it can shrink again after the worker auto-compacts.
25
+ if (usage.totalTokens)
26
+ target.contextTokens = usage.totalTokens;
23
27
  }
24
28
  export function mergeUsage(target, source) {
25
29
  target.turns += source.turns;
@@ -30,6 +34,9 @@ export function mergeUsage(target, source) {
30
34
  target.cacheWrite += source.cacheWrite;
31
35
  target.totalTokens += source.totalTokens;
32
36
  target.cost += source.cost;
37
+ if (source.contextTokens) {
38
+ target.contextTokens = Math.max(target.contextTokens ?? 0, source.contextTokens);
39
+ }
33
40
  }
34
41
  export function toNestedUsage(usage) {
35
42
  return {
@@ -47,12 +54,21 @@ export function toNestedUsage(usage) {
47
54
  },
48
55
  };
49
56
  }
50
- function utf8Prefix(text, maxBytes) {
57
+ // Iterates code points so a surrogate pair is never split in half.
58
+ export function utf8Prefix(text, maxBytes) {
51
59
  if (maxBytes <= 0)
52
60
  return "";
53
- let output = text.slice(0, maxBytes);
54
- while (Buffer.byteLength(output, "utf8") > maxBytes)
55
- output = output.slice(0, -1);
61
+ if (Buffer.byteLength(text, "utf8") <= maxBytes)
62
+ return text;
63
+ let output = "";
64
+ let bytes = 0;
65
+ for (const character of text) {
66
+ const characterBytes = Buffer.byteLength(character, "utf8");
67
+ if (bytes + characterBytes > maxBytes)
68
+ break;
69
+ output += character;
70
+ bytes += characterBytes;
71
+ }
56
72
  return output;
57
73
  }
58
74
  export function boundText(text, maxBytes) {
@@ -1 +1 @@
1
- {"version":3,"file":"activity.js","sourceRoot":"","sources":["../../../src/extensions/subagent/activity.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,cAAc,EAAE,mBAAmB,EAAE,eAAe,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAC;AAG3G,MAAM,UAAU,UAAU,GAAkB;IAC3C,OAAO;QACN,KAAK,EAAE,CAAC;QACR,QAAQ,EAAE,CAAC;QACX,KAAK,EAAE,CAAC;QACR,MAAM,EAAE,CAAC;QACT,SAAS,EAAE,CAAC;QACZ,UAAU,EAAE,CAAC;QACb,WAAW,EAAE,CAAC;QACd,IAAI,EAAE,CAAC;KACP,CAAC;AAAA,CACF;AAED,MAAM,UAAU,QAAQ,CAAC,MAAqB,EAAE,KAAwB,EAAQ;IAC/E,IAAI,CAAC,KAAK;QAAE,OAAO;IACnB,MAAM,CAAC,KAAK,IAAI,KAAK,CAAC,KAAK,IAAI,CAAC,CAAC;IACjC,MAAM,CAAC,MAAM,IAAI,KAAK,CAAC,MAAM,IAAI,CAAC,CAAC;IACnC,MAAM,CAAC,SAAS,IAAI,KAAK,CAAC,SAAS,IAAI,CAAC,CAAC;IACzC,MAAM,CAAC,UAAU,IAAI,KAAK,CAAC,UAAU,IAAI,CAAC,CAAC;IAC3C,MAAM,CAAC,WAAW,IAAI,KAAK,CAAC,WAAW,IAAI,CAAC,CAAC;IAC7C,MAAM,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,EAAE,KAAK,IAAI,CAAC,CAAC;AAAA,CACtC;AAED,MAAM,UAAU,UAAU,CAAC,MAAqB,EAAE,MAAqB,EAAQ;IAC9E,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC;IAC7B,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC;IACnC,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC;IAC7B,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,CAAC;IAC/B,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,SAAS,CAAC;IACrC,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,UAAU,CAAC;IACvC,MAAM,CAAC,WAAW,IAAI,MAAM,CAAC,WAAW,CAAC;IACzC,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC;AAAA,CAC3B;AAED,MAAM,UAAU,aAAa,CAAC,KAAoB,EAAS;IAC1D,OAAO;QACN,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,MAAM,EAAE,KAAK,CAAC,MAAM;QACpB,SAAS,EAAE,KAAK,CAAC,SAAS;QAC1B,UAAU,EAAE,KAAK,CAAC,UAAU;QAC5B,WAAW,EAAE,KAAK,CAAC,WAAW;QAC9B,IAAI,EAAE;YACL,KAAK,EAAE,CAAC;YACR,MAAM,EAAE,CAAC;YACT,SAAS,EAAE,CAAC;YACZ,UAAU,EAAE,CAAC;YACb,KAAK,EAAE,KAAK,CAAC,IAAI;SACjB;KACD,CAAC;AAAA,CACF;AAED,SAAS,UAAU,CAAC,IAAY,EAAE,QAAgB,EAAU;IAC3D,IAAI,QAAQ,IAAI,CAAC;QAAE,OAAO,EAAE,CAAC;IAC7B,IAAI,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC;IACrC,OAAO,MAAM,CAAC,UAAU,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,QAAQ;QAAE,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;IAClF,OAAO,MAAM,CAAC;AAAA,CACd;AAED,MAAM,UAAU,SAAS,CAAC,IAAY,EAAE,QAAgB,EAAU;IACjE,IAAI,QAAQ,IAAI,CAAC;QAAE,OAAO,EAAE,CAAC;IAC7B,IAAI,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,QAAQ;QAAE,OAAO,IAAI,CAAC;IAC7D,IAAI,MAAM,GAAG,UAAU,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IACxC,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,CAAC,EAAE,OAAO,EAAE,EAAE,CAAC;QAC9C,MAAM,OAAO,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE,MAAM,CAAC,GAAG,MAAM,CAAC,UAAU,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACpF,MAAM,MAAM,GAAG,0BAA0B,OAAO,kBAAkB,CAAC;QACnE,MAAM,SAAS,GAAG,QAAQ,GAAG,MAAM,CAAC,UAAU,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC/D,IAAI,SAAS,IAAI,CAAC;YAAE,OAAO,UAAU,CAAC,qBAAqB,EAAE,QAAQ,CAAC,CAAC;QACvE,MAAM,IAAI,GAAG,UAAU,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QACzC,IAAI,IAAI,KAAK,MAAM;YAAE,OAAO,GAAG,MAAM,GAAG,MAAM,EAAE,CAAC;QACjD,MAAM,GAAG,IAAI,CAAC;IACf,CAAC;IACD,MAAM,OAAO,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE,MAAM,CAAC,GAAG,MAAM,CAAC,UAAU,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACpF,MAAM,MAAM,GAAG,0BAA0B,OAAO,kBAAkB,CAAC;IACnE,OAAO,GAAG,UAAU,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,GAAG,MAAM,CAAC,UAAU,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,GAAG,MAAM,EAAE,CAAC;AAAA,CACnG;AAED,MAAM,UAAU,QAAQ,CAAC,IAAY,EAAE,QAAgB,EAAU;IAChE,IAAI,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,QAAQ;QAAE,OAAO,IAAI,CAAC;IAC7D,IAAI,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,CAAC;IACnC,OAAO,MAAM,CAAC,UAAU,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,QAAQ;QAAE,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IAC9E,OAAO,8BAA8B,MAAM,EAAE,CAAC;AAAA,CAC9C;AAED,MAAM,UAAU,aAAa,CAAC,OAAiC,EAAU;IACxE,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW;QAAE,OAAO,EAAE,CAAC;IACxD,OAAO,OAAO,CAAC,OAAO;SACpB,MAAM,CAAC,CAAC,IAAI,EAA0C,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;SAC9E,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;SACxB,IAAI,CAAC,EAAE,CAAC,CAAC;AAAA,CACX;AAED,MAAM,UAAU,kBAAkB,CAAC,QAAiC,EAAU;IAC7E,KAAK,IAAI,KAAK,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC;QAC3D,MAAM,IAAI,GAAG,aAAa,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;QAC5C,IAAI,IAAI;YAAE,OAAO,SAAS,CAAC,IAAI,EAAE,mBAAmB,CAAC,CAAC;IACvD,CAAC;IACD,OAAO,EAAE,CAAC;AAAA,CACV;AAED,MAAM,UAAU,cAAc,CAAC,UAA0B,EAAE,QAAsB,EAAQ;IACxF,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1B,OAAO,UAAU,CAAC,MAAM,GAAG,cAAc;QAAE,UAAU,CAAC,KAAK,EAAE,CAAC;IAC9D,OACC,UAAU,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,aAAa,EAAE,MAAM,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC;QACtG,mBAAmB,EAClB,CAAC;QACF,IAAI,UAAU,CAAC,MAAM,IAAI,CAAC;YAAE,MAAM;QAClC,UAAU,CAAC,KAAK,EAAE,CAAC;IACpB,CAAC;AAAA,CACD;AAED,MAAM,UAAU,WAAW,CAAC,IAAY,EAAU;IACjD,OAAO,QAAQ,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;AAAA,CACvC;AAED,MAAM,UAAU,eAAe,CAAC,QAAgB,EAAE,IAAa,EAAU;IACxE,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC;QAAE,OAAO,QAAQ,CAAC;IAC9E,MAAM,KAAK,GAAG,IAA+B,CAAC;IAC9C,MAAM,IAAI,GACT,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,KAAK,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;IACjH,IAAI,QAAQ,KAAK,MAAM,IAAI,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ;QAC3D,OAAO,OAAO,SAAS,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC;IACrE,IAAI,IAAI;QAAE,OAAO,GAAG,QAAQ,IAAI,SAAS,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,CAAC;IACvD,IAAI,QAAQ,KAAK,MAAM,IAAI,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ;QAAE,OAAO,UAAU,SAAS,CAAC,KAAK,CAAC,OAAO,EAAE,GAAG,CAAC,EAAE,CAAC;IAC/G,IAAI,QAAQ,KAAK,MAAM,IAAI,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ;QAAE,OAAO,QAAQ,SAAS,CAAC,KAAK,CAAC,OAAO,EAAE,GAAG,CAAC,EAAE,CAAC;IAC7G,OAAO,QAAQ,CAAC;AAAA,CAChB;AAED,MAAM,UAAU,aAAa,CAAC,MAAe,EAAU;IACtD,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ;QAAE,OAAO,EAAE,CAAC;IACrD,MAAM,OAAO,GAAI,MAAgC,CAAC,OAAO,CAAC;IAC1D,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC;QAAE,OAAO,EAAE,CAAC;IACvC,OAAO,SAAS,CACf,OAAO;SACL,MAAM,CAAC,CAAC,IAAI,EAA0C,EAAE,CAAC;QACzD,OAAO,OAAO,CACb,IAAI;YACH,OAAO,IAAI,KAAK,QAAQ;YACvB,IAA2B,CAAC,IAAI,KAAK,MAAM;YAC5C,OAAQ,IAA2B,CAAC,IAAI,KAAK,QAAQ,CACtD,CAAC;IAAA,CACF,CAAC;SACD,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;SACxB,IAAI,CAAC,IAAI,CAAC;SACV,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC;SACrB,IAAI,EAAE,EACR,GAAG,CACH,CAAC;AAAA,CACF","sourcesContent":["import type { AgentMessage } from \"@earendil-works/pi-agent-core\";\nimport type { Usage } from \"@earendil-works/pi-ai\";\nimport { ACTIVITY_LIMIT, ACTIVITY_TEXT_LIMIT, LIVE_TEXT_LIMIT, SINGLE_OUTPUT_LIMIT } from \"./constants.ts\";\nimport type { SubagentUsage, ToolActivity } from \"./types.ts\";\n\nexport function emptyUsage(): SubagentUsage {\n\treturn {\n\t\tturns: 0,\n\t\ttoolUses: 0,\n\t\tinput: 0,\n\t\toutput: 0,\n\t\tcacheRead: 0,\n\t\tcacheWrite: 0,\n\t\ttotalTokens: 0,\n\t\tcost: 0,\n\t};\n}\n\nexport function addUsage(target: SubagentUsage, usage: Usage | undefined): void {\n\tif (!usage) return;\n\ttarget.input += usage.input ?? 0;\n\ttarget.output += usage.output ?? 0;\n\ttarget.cacheRead += usage.cacheRead ?? 0;\n\ttarget.cacheWrite += usage.cacheWrite ?? 0;\n\ttarget.totalTokens += usage.totalTokens ?? 0;\n\ttarget.cost += usage.cost?.total ?? 0;\n}\n\nexport function mergeUsage(target: SubagentUsage, source: SubagentUsage): void {\n\ttarget.turns += source.turns;\n\ttarget.toolUses += source.toolUses;\n\ttarget.input += source.input;\n\ttarget.output += source.output;\n\ttarget.cacheRead += source.cacheRead;\n\ttarget.cacheWrite += source.cacheWrite;\n\ttarget.totalTokens += source.totalTokens;\n\ttarget.cost += source.cost;\n}\n\nexport function toNestedUsage(usage: SubagentUsage): Usage {\n\treturn {\n\t\tinput: usage.input,\n\t\toutput: usage.output,\n\t\tcacheRead: usage.cacheRead,\n\t\tcacheWrite: usage.cacheWrite,\n\t\ttotalTokens: usage.totalTokens,\n\t\tcost: {\n\t\t\tinput: 0,\n\t\t\toutput: 0,\n\t\t\tcacheRead: 0,\n\t\t\tcacheWrite: 0,\n\t\t\ttotal: usage.cost,\n\t\t},\n\t};\n}\n\nfunction utf8Prefix(text: string, maxBytes: number): string {\n\tif (maxBytes <= 0) return \"\";\n\tlet output = text.slice(0, maxBytes);\n\twhile (Buffer.byteLength(output, \"utf8\") > maxBytes) output = output.slice(0, -1);\n\treturn output;\n}\n\nexport function boundText(text: string, maxBytes: number): string {\n\tif (maxBytes <= 0) return \"\";\n\tif (Buffer.byteLength(text, \"utf8\") <= maxBytes) return text;\n\tlet output = utf8Prefix(text, maxBytes);\n\tfor (let attempt = 0; attempt < 8; attempt++) {\n\t\tconst omitted = Buffer.byteLength(text, \"utf8\") - Buffer.byteLength(output, \"utf8\");\n\t\tconst notice = `\\n\\n[Output truncated: ${omitted} bytes omitted.]`;\n\t\tconst available = maxBytes - Buffer.byteLength(notice, \"utf8\");\n\t\tif (available <= 0) return utf8Prefix(\"[Output truncated.]\", maxBytes);\n\t\tconst next = utf8Prefix(text, available);\n\t\tif (next === output) return `${output}${notice}`;\n\t\toutput = next;\n\t}\n\tconst omitted = Buffer.byteLength(text, \"utf8\") - Buffer.byteLength(output, \"utf8\");\n\tconst notice = `\\n\\n[Output truncated: ${omitted} bytes omitted.]`;\n\treturn `${utf8Prefix(output, Math.max(0, maxBytes - Buffer.byteLength(notice, \"utf8\")))}${notice}`;\n}\n\nexport function tailText(text: string, maxBytes: number): string {\n\tif (Buffer.byteLength(text, \"utf8\") <= maxBytes) return text;\n\tlet output = text.slice(-maxBytes);\n\twhile (Buffer.byteLength(output, \"utf8\") > maxBytes) output = output.slice(1);\n\treturn `[Earlier output omitted.]\\n${output}`;\n}\n\nexport function assistantText(message: AgentMessage | undefined): string {\n\tif (!message || message.role !== \"assistant\") return \"\";\n\treturn message.content\n\t\t.filter((part): part is { type: \"text\"; text: string } => part.type === \"text\")\n\t\t.map((part) => part.text)\n\t\t.join(\"\");\n}\n\nexport function finalAssistantText(messages: readonly AgentMessage[]): string {\n\tfor (let index = messages.length - 1; index >= 0; index--) {\n\t\tconst text = assistantText(messages[index]);\n\t\tif (text) return boundText(text, SINGLE_OUTPUT_LIMIT);\n\t}\n\treturn \"\";\n}\n\nexport function appendActivity(activities: ToolActivity[], activity: ToolActivity): void {\n\tactivities.push(activity);\n\twhile (activities.length > ACTIVITY_LIMIT) activities.shift();\n\twhile (\n\t\tactivities.reduce((total, item) => total + item.summary.length + (item.resultSummary?.length ?? 0), 0) >\n\t\tACTIVITY_TEXT_LIMIT\n\t) {\n\t\tif (activities.length <= 1) break;\n\t\tactivities.shift();\n\t}\n}\n\nexport function setLiveText(text: string): string {\n\treturn tailText(text, LIVE_TEXT_LIMIT);\n}\n\nexport function activitySummary(toolName: string, args: unknown): string {\n\tif (!args || typeof args !== \"object\" || Array.isArray(args)) return toolName;\n\tconst input = args as Record<string, unknown>;\n\tconst path =\n\t\ttypeof input.path === \"string\" ? input.path : typeof input.file_path === \"string\" ? input.file_path : undefined;\n\tif (toolName === \"bash\" && typeof input.command === \"string\")\n\t\treturn `Run ${boundText(input.command.replace(/\\s+/gu, \" \"), 160)}`;\n\tif (path) return `${toolName} ${boundText(path, 180)}`;\n\tif (toolName === \"grep\" && typeof input.pattern === \"string\") return `Search ${boundText(input.pattern, 120)}`;\n\tif (toolName === \"find\" && typeof input.pattern === \"string\") return `Find ${boundText(input.pattern, 120)}`;\n\treturn toolName;\n}\n\nexport function resultSummary(result: unknown): string {\n\tif (!result || typeof result !== \"object\") return \"\";\n\tconst content = (result as { content?: unknown }).content;\n\tif (!Array.isArray(content)) return \"\";\n\treturn boundText(\n\t\tcontent\n\t\t\t.filter((part): part is { type: \"text\"; text: string } => {\n\t\t\t\treturn Boolean(\n\t\t\t\t\tpart &&\n\t\t\t\t\t\ttypeof part === \"object\" &&\n\t\t\t\t\t\t(part as { type?: unknown }).type === \"text\" &&\n\t\t\t\t\t\ttypeof (part as { text?: unknown }).text === \"string\",\n\t\t\t\t);\n\t\t\t})\n\t\t\t.map((part) => part.text)\n\t\t\t.join(\"\\n\")\n\t\t\t.replace(/\\s+/gu, \" \")\n\t\t\t.trim(),\n\t\t240,\n\t);\n}\n"]}
1
+ {"version":3,"file":"activity.js","sourceRoot":"","sources":["../../../src/extensions/subagent/activity.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,cAAc,EAAE,mBAAmB,EAAE,eAAe,EAAE,mBAAmB,EAAE,MAAM,gBAAgB,CAAC;AAG3G,MAAM,UAAU,UAAU,GAAkB;IAC3C,OAAO;QACN,KAAK,EAAE,CAAC;QACR,QAAQ,EAAE,CAAC;QACX,KAAK,EAAE,CAAC;QACR,MAAM,EAAE,CAAC;QACT,SAAS,EAAE,CAAC;QACZ,UAAU,EAAE,CAAC;QACb,WAAW,EAAE,CAAC;QACd,IAAI,EAAE,CAAC;KACP,CAAC;AAAA,CACF;AAED,MAAM,UAAU,QAAQ,CAAC,MAAqB,EAAE,KAAwB,EAAQ;IAC/E,IAAI,CAAC,KAAK;QAAE,OAAO;IACnB,MAAM,CAAC,KAAK,IAAI,KAAK,CAAC,KAAK,IAAI,CAAC,CAAC;IACjC,MAAM,CAAC,MAAM,IAAI,KAAK,CAAC,MAAM,IAAI,CAAC,CAAC;IACnC,MAAM,CAAC,SAAS,IAAI,KAAK,CAAC,SAAS,IAAI,CAAC,CAAC;IACzC,MAAM,CAAC,UAAU,IAAI,KAAK,CAAC,UAAU,IAAI,CAAC,CAAC;IAC3C,MAAM,CAAC,WAAW,IAAI,KAAK,CAAC,WAAW,IAAI,CAAC,CAAC;IAC7C,MAAM,CAAC,IAAI,IAAI,KAAK,CAAC,IAAI,EAAE,KAAK,IAAI,CAAC,CAAC;IACtC,wEAAwE;IACxE,0DAA0D;IAC1D,IAAI,KAAK,CAAC,WAAW;QAAE,MAAM,CAAC,aAAa,GAAG,KAAK,CAAC,WAAW,CAAC;AAAA,CAChE;AAED,MAAM,UAAU,UAAU,CAAC,MAAqB,EAAE,MAAqB,EAAQ;IAC9E,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC;IAC7B,MAAM,CAAC,QAAQ,IAAI,MAAM,CAAC,QAAQ,CAAC;IACnC,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,KAAK,CAAC;IAC7B,MAAM,CAAC,MAAM,IAAI,MAAM,CAAC,MAAM,CAAC;IAC/B,MAAM,CAAC,SAAS,IAAI,MAAM,CAAC,SAAS,CAAC;IACrC,MAAM,CAAC,UAAU,IAAI,MAAM,CAAC,UAAU,CAAC;IACvC,MAAM,CAAC,WAAW,IAAI,MAAM,CAAC,WAAW,CAAC;IACzC,MAAM,CAAC,IAAI,IAAI,MAAM,CAAC,IAAI,CAAC;IAC3B,IAAI,MAAM,CAAC,aAAa,EAAE,CAAC;QAC1B,MAAM,CAAC,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,aAAa,IAAI,CAAC,EAAE,MAAM,CAAC,aAAa,CAAC,CAAC;IAClF,CAAC;AAAA,CACD;AAED,MAAM,UAAU,aAAa,CAAC,KAAoB,EAAS;IAC1D,OAAO;QACN,KAAK,EAAE,KAAK,CAAC,KAAK;QAClB,MAAM,EAAE,KAAK,CAAC,MAAM;QACpB,SAAS,EAAE,KAAK,CAAC,SAAS;QAC1B,UAAU,EAAE,KAAK,CAAC,UAAU;QAC5B,WAAW,EAAE,KAAK,CAAC,WAAW;QAC9B,IAAI,EAAE;YACL,KAAK,EAAE,CAAC;YACR,MAAM,EAAE,CAAC;YACT,SAAS,EAAE,CAAC;YACZ,UAAU,EAAE,CAAC;YACb,KAAK,EAAE,KAAK,CAAC,IAAI;SACjB;KACD,CAAC;AAAA,CACF;AAED,mEAAmE;AACnE,MAAM,UAAU,UAAU,CAAC,IAAY,EAAE,QAAgB,EAAU;IAClE,IAAI,QAAQ,IAAI,CAAC;QAAE,OAAO,EAAE,CAAC;IAC7B,IAAI,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,QAAQ;QAAE,OAAO,IAAI,CAAC;IAC7D,IAAI,MAAM,GAAG,EAAE,CAAC;IAChB,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,KAAK,MAAM,SAAS,IAAI,IAAI,EAAE,CAAC;QAC9B,MAAM,cAAc,GAAG,MAAM,CAAC,UAAU,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;QAC5D,IAAI,KAAK,GAAG,cAAc,GAAG,QAAQ;YAAE,MAAM;QAC7C,MAAM,IAAI,SAAS,CAAC;QACpB,KAAK,IAAI,cAAc,CAAC;IACzB,CAAC;IACD,OAAO,MAAM,CAAC;AAAA,CACd;AAED,MAAM,UAAU,SAAS,CAAC,IAAY,EAAE,QAAgB,EAAU;IACjE,IAAI,QAAQ,IAAI,CAAC;QAAE,OAAO,EAAE,CAAC;IAC7B,IAAI,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,QAAQ;QAAE,OAAO,IAAI,CAAC;IAC7D,IAAI,MAAM,GAAG,UAAU,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;IACxC,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,GAAG,CAAC,EAAE,OAAO,EAAE,EAAE,CAAC;QAC9C,MAAM,OAAO,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE,MAAM,CAAC,GAAG,MAAM,CAAC,UAAU,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACpF,MAAM,MAAM,GAAG,0BAA0B,OAAO,kBAAkB,CAAC;QACnE,MAAM,SAAS,GAAG,QAAQ,GAAG,MAAM,CAAC,UAAU,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QAC/D,IAAI,SAAS,IAAI,CAAC;YAAE,OAAO,UAAU,CAAC,qBAAqB,EAAE,QAAQ,CAAC,CAAC;QACvE,MAAM,IAAI,GAAG,UAAU,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QACzC,IAAI,IAAI,KAAK,MAAM;YAAE,OAAO,GAAG,MAAM,GAAG,MAAM,EAAE,CAAC;QACjD,MAAM,GAAG,IAAI,CAAC;IACf,CAAC;IACD,MAAM,OAAO,GAAG,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE,MAAM,CAAC,GAAG,MAAM,CAAC,UAAU,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACpF,MAAM,MAAM,GAAG,0BAA0B,OAAO,kBAAkB,CAAC;IACnE,OAAO,GAAG,UAAU,CAAC,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,GAAG,MAAM,CAAC,UAAU,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC,GAAG,MAAM,EAAE,CAAC;AAAA,CACnG;AAED,MAAM,UAAU,QAAQ,CAAC,IAAY,EAAE,QAAgB,EAAU;IAChE,IAAI,MAAM,CAAC,UAAU,CAAC,IAAI,EAAE,MAAM,CAAC,IAAI,QAAQ;QAAE,OAAO,IAAI,CAAC;IAC7D,IAAI,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,QAAQ,CAAC,CAAC;IACnC,OAAO,MAAM,CAAC,UAAU,CAAC,MAAM,EAAE,MAAM,CAAC,GAAG,QAAQ;QAAE,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;IAC9E,OAAO,8BAA8B,MAAM,EAAE,CAAC;AAAA,CAC9C;AAED,MAAM,UAAU,aAAa,CAAC,OAAiC,EAAU;IACxE,IAAI,CAAC,OAAO,IAAI,OAAO,CAAC,IAAI,KAAK,WAAW;QAAE,OAAO,EAAE,CAAC;IACxD,OAAO,OAAO,CAAC,OAAO;SACpB,MAAM,CAAC,CAAC,IAAI,EAA0C,EAAE,CAAC,IAAI,CAAC,IAAI,KAAK,MAAM,CAAC;SAC9E,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;SACxB,IAAI,CAAC,EAAE,CAAC,CAAC;AAAA,CACX;AAED,MAAM,UAAU,kBAAkB,CAAC,QAAiC,EAAU;IAC7E,KAAK,IAAI,KAAK,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,EAAE,KAAK,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC;QAC3D,MAAM,IAAI,GAAG,aAAa,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,CAAC;QAC5C,IAAI,IAAI;YAAE,OAAO,SAAS,CAAC,IAAI,EAAE,mBAAmB,CAAC,CAAC;IACvD,CAAC;IACD,OAAO,EAAE,CAAC;AAAA,CACV;AAED,MAAM,UAAU,cAAc,CAAC,UAA0B,EAAE,QAAsB,EAAQ;IACxF,UAAU,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IAC1B,OAAO,UAAU,CAAC,MAAM,GAAG,cAAc;QAAE,UAAU,CAAC,KAAK,EAAE,CAAC;IAC9D,OACC,UAAU,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,aAAa,EAAE,MAAM,IAAI,CAAC,CAAC,EAAE,CAAC,CAAC;QACtG,mBAAmB,EAClB,CAAC;QACF,IAAI,UAAU,CAAC,MAAM,IAAI,CAAC;YAAE,MAAM;QAClC,UAAU,CAAC,KAAK,EAAE,CAAC;IACpB,CAAC;AAAA,CACD;AAED,MAAM,UAAU,WAAW,CAAC,IAAY,EAAU;IACjD,OAAO,QAAQ,CAAC,IAAI,EAAE,eAAe,CAAC,CAAC;AAAA,CACvC;AAED,MAAM,UAAU,eAAe,CAAC,QAAgB,EAAE,IAAa,EAAU;IACxE,IAAI,CAAC,IAAI,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC;QAAE,OAAO,QAAQ,CAAC;IAC9E,MAAM,KAAK,GAAG,IAA+B,CAAC;IAC9C,MAAM,IAAI,GACT,OAAO,KAAK,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,OAAO,KAAK,CAAC,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC;IACjH,IAAI,QAAQ,KAAK,MAAM,IAAI,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ;QAC3D,OAAO,OAAO,SAAS,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC,EAAE,GAAG,CAAC,EAAE,CAAC;IACrE,IAAI,IAAI;QAAE,OAAO,GAAG,QAAQ,IAAI,SAAS,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,CAAC;IACvD,IAAI,QAAQ,KAAK,MAAM,IAAI,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ;QAAE,OAAO,UAAU,SAAS,CAAC,KAAK,CAAC,OAAO,EAAE,GAAG,CAAC,EAAE,CAAC;IAC/G,IAAI,QAAQ,KAAK,MAAM,IAAI,OAAO,KAAK,CAAC,OAAO,KAAK,QAAQ;QAAE,OAAO,QAAQ,SAAS,CAAC,KAAK,CAAC,OAAO,EAAE,GAAG,CAAC,EAAE,CAAC;IAC7G,OAAO,QAAQ,CAAC;AAAA,CAChB;AAED,MAAM,UAAU,aAAa,CAAC,MAAe,EAAU;IACtD,IAAI,CAAC,MAAM,IAAI,OAAO,MAAM,KAAK,QAAQ;QAAE,OAAO,EAAE,CAAC;IACrD,MAAM,OAAO,GAAI,MAAgC,CAAC,OAAO,CAAC;IAC1D,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC;QAAE,OAAO,EAAE,CAAC;IACvC,OAAO,SAAS,CACf,OAAO;SACL,MAAM,CAAC,CAAC,IAAI,EAA0C,EAAE,CAAC;QACzD,OAAO,OAAO,CACb,IAAI;YACH,OAAO,IAAI,KAAK,QAAQ;YACvB,IAA2B,CAAC,IAAI,KAAK,MAAM;YAC5C,OAAQ,IAA2B,CAAC,IAAI,KAAK,QAAQ,CACtD,CAAC;IAAA,CACF,CAAC;SACD,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;SACxB,IAAI,CAAC,IAAI,CAAC;SACV,OAAO,CAAC,OAAO,EAAE,GAAG,CAAC;SACrB,IAAI,EAAE,EACR,GAAG,CACH,CAAC;AAAA,CACF","sourcesContent":["import type { AgentMessage } from \"@earendil-works/pi-agent-core\";\nimport type { Usage } from \"@earendil-works/pi-ai\";\nimport { ACTIVITY_LIMIT, ACTIVITY_TEXT_LIMIT, LIVE_TEXT_LIMIT, SINGLE_OUTPUT_LIMIT } from \"./constants.ts\";\nimport type { SubagentUsage, ToolActivity } from \"./types.ts\";\n\nexport function emptyUsage(): SubagentUsage {\n\treturn {\n\t\tturns: 0,\n\t\ttoolUses: 0,\n\t\tinput: 0,\n\t\toutput: 0,\n\t\tcacheRead: 0,\n\t\tcacheWrite: 0,\n\t\ttotalTokens: 0,\n\t\tcost: 0,\n\t};\n}\n\nexport function addUsage(target: SubagentUsage, usage: Usage | undefined): void {\n\tif (!usage) return;\n\ttarget.input += usage.input ?? 0;\n\ttarget.output += usage.output ?? 0;\n\ttarget.cacheRead += usage.cacheRead ?? 0;\n\ttarget.cacheWrite += usage.cacheWrite ?? 0;\n\ttarget.totalTokens += usage.totalTokens ?? 0;\n\ttarget.cost += usage.cost?.total ?? 0;\n\t// Watermark, not a sum: the latest request's total is the context size,\n\t// and it can shrink again after the worker auto-compacts.\n\tif (usage.totalTokens) target.contextTokens = usage.totalTokens;\n}\n\nexport function mergeUsage(target: SubagentUsage, source: SubagentUsage): void {\n\ttarget.turns += source.turns;\n\ttarget.toolUses += source.toolUses;\n\ttarget.input += source.input;\n\ttarget.output += source.output;\n\ttarget.cacheRead += source.cacheRead;\n\ttarget.cacheWrite += source.cacheWrite;\n\ttarget.totalTokens += source.totalTokens;\n\ttarget.cost += source.cost;\n\tif (source.contextTokens) {\n\t\ttarget.contextTokens = Math.max(target.contextTokens ?? 0, source.contextTokens);\n\t}\n}\n\nexport function toNestedUsage(usage: SubagentUsage): Usage {\n\treturn {\n\t\tinput: usage.input,\n\t\toutput: usage.output,\n\t\tcacheRead: usage.cacheRead,\n\t\tcacheWrite: usage.cacheWrite,\n\t\ttotalTokens: usage.totalTokens,\n\t\tcost: {\n\t\t\tinput: 0,\n\t\t\toutput: 0,\n\t\t\tcacheRead: 0,\n\t\t\tcacheWrite: 0,\n\t\t\ttotal: usage.cost,\n\t\t},\n\t};\n}\n\n// Iterates code points so a surrogate pair is never split in half.\nexport function utf8Prefix(text: string, maxBytes: number): string {\n\tif (maxBytes <= 0) return \"\";\n\tif (Buffer.byteLength(text, \"utf8\") <= maxBytes) return text;\n\tlet output = \"\";\n\tlet bytes = 0;\n\tfor (const character of text) {\n\t\tconst characterBytes = Buffer.byteLength(character, \"utf8\");\n\t\tif (bytes + characterBytes > maxBytes) break;\n\t\toutput += character;\n\t\tbytes += characterBytes;\n\t}\n\treturn output;\n}\n\nexport function boundText(text: string, maxBytes: number): string {\n\tif (maxBytes <= 0) return \"\";\n\tif (Buffer.byteLength(text, \"utf8\") <= maxBytes) return text;\n\tlet output = utf8Prefix(text, maxBytes);\n\tfor (let attempt = 0; attempt < 8; attempt++) {\n\t\tconst omitted = Buffer.byteLength(text, \"utf8\") - Buffer.byteLength(output, \"utf8\");\n\t\tconst notice = `\\n\\n[Output truncated: ${omitted} bytes omitted.]`;\n\t\tconst available = maxBytes - Buffer.byteLength(notice, \"utf8\");\n\t\tif (available <= 0) return utf8Prefix(\"[Output truncated.]\", maxBytes);\n\t\tconst next = utf8Prefix(text, available);\n\t\tif (next === output) return `${output}${notice}`;\n\t\toutput = next;\n\t}\n\tconst omitted = Buffer.byteLength(text, \"utf8\") - Buffer.byteLength(output, \"utf8\");\n\tconst notice = `\\n\\n[Output truncated: ${omitted} bytes omitted.]`;\n\treturn `${utf8Prefix(output, Math.max(0, maxBytes - Buffer.byteLength(notice, \"utf8\")))}${notice}`;\n}\n\nexport function tailText(text: string, maxBytes: number): string {\n\tif (Buffer.byteLength(text, \"utf8\") <= maxBytes) return text;\n\tlet output = text.slice(-maxBytes);\n\twhile (Buffer.byteLength(output, \"utf8\") > maxBytes) output = output.slice(1);\n\treturn `[Earlier output omitted.]\\n${output}`;\n}\n\nexport function assistantText(message: AgentMessage | undefined): string {\n\tif (!message || message.role !== \"assistant\") return \"\";\n\treturn message.content\n\t\t.filter((part): part is { type: \"text\"; text: string } => part.type === \"text\")\n\t\t.map((part) => part.text)\n\t\t.join(\"\");\n}\n\nexport function finalAssistantText(messages: readonly AgentMessage[]): string {\n\tfor (let index = messages.length - 1; index >= 0; index--) {\n\t\tconst text = assistantText(messages[index]);\n\t\tif (text) return boundText(text, SINGLE_OUTPUT_LIMIT);\n\t}\n\treturn \"\";\n}\n\nexport function appendActivity(activities: ToolActivity[], activity: ToolActivity): void {\n\tactivities.push(activity);\n\twhile (activities.length > ACTIVITY_LIMIT) activities.shift();\n\twhile (\n\t\tactivities.reduce((total, item) => total + item.summary.length + (item.resultSummary?.length ?? 0), 0) >\n\t\tACTIVITY_TEXT_LIMIT\n\t) {\n\t\tif (activities.length <= 1) break;\n\t\tactivities.shift();\n\t}\n}\n\nexport function setLiveText(text: string): string {\n\treturn tailText(text, LIVE_TEXT_LIMIT);\n}\n\nexport function activitySummary(toolName: string, args: unknown): string {\n\tif (!args || typeof args !== \"object\" || Array.isArray(args)) return toolName;\n\tconst input = args as Record<string, unknown>;\n\tconst path =\n\t\ttypeof input.path === \"string\" ? input.path : typeof input.file_path === \"string\" ? input.file_path : undefined;\n\tif (toolName === \"bash\" && typeof input.command === \"string\")\n\t\treturn `Run ${boundText(input.command.replace(/\\s+/gu, \" \"), 160)}`;\n\tif (path) return `${toolName} ${boundText(path, 180)}`;\n\tif (toolName === \"grep\" && typeof input.pattern === \"string\") return `Search ${boundText(input.pattern, 120)}`;\n\tif (toolName === \"find\" && typeof input.pattern === \"string\") return `Find ${boundText(input.pattern, 120)}`;\n\treturn toolName;\n}\n\nexport function resultSummary(result: unknown): string {\n\tif (!result || typeof result !== \"object\") return \"\";\n\tconst content = (result as { content?: unknown }).content;\n\tif (!Array.isArray(content)) return \"\";\n\treturn boundText(\n\t\tcontent\n\t\t\t.filter((part): part is { type: \"text\"; text: string } => {\n\t\t\t\treturn Boolean(\n\t\t\t\t\tpart &&\n\t\t\t\t\t\ttypeof part === \"object\" &&\n\t\t\t\t\t\t(part as { type?: unknown }).type === \"text\" &&\n\t\t\t\t\t\ttypeof (part as { text?: unknown }).text === \"string\",\n\t\t\t\t);\n\t\t\t})\n\t\t\t.map((part) => part.text)\n\t\t\t.join(\"\\n\")\n\t\t\t.replace(/\\s+/gu, \" \")\n\t\t\t.trim(),\n\t\t240,\n\t);\n}\n"]}