pi-oxidized 0.1.1

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 (1517) hide show
  1. package/.agent-tasks/pi-rust-rewrite/Cargo.lock +5022 -0
  2. package/.agent-tasks/pi-rust-rewrite/Cargo.toml +22 -0
  3. package/.agent-tasks/pi-rust-rewrite/GOALS.md +1 -0
  4. package/.agent-tasks/pi-rust-rewrite/fixtures/phase7/goldens/branch_summary.json +8 -0
  5. package/.agent-tasks/pi-rust-rewrite/fixtures/phase7/goldens/changelog.md +16 -0
  6. package/.agent-tasks/pi-rust-rewrite/fixtures/phase7/goldens/debug_dump.txt +11 -0
  7. package/.agent-tasks/pi-rust-rewrite/fixtures/phase7/goldens/events.jsonl +6 -0
  8. package/.agent-tasks/pi-rust-rewrite/fixtures/phase7/goldens/legacy_auth_migration.json +16 -0
  9. package/.agent-tasks/pi-rust-rewrite/fixtures/phase7/goldens/selector_snapshot.txt +9 -0
  10. package/.agent-tasks/pi-rust-rewrite/fixtures/phase7/goldens/session_stats.json +16 -0
  11. package/.agent-tasks/pi-rust-rewrite/fixtures/phase7/goldens/share_result.json +7 -0
  12. package/.agent-tasks/pi-rust-rewrite/fixtures/phase7/sessions/export-roundtrip.jsonl +6 -0
  13. package/.agent-tasks/pi-rust-rewrite/fixtures/provider-drain/isolation-expectations.json +21 -0
  14. package/.agent-tasks/pi-rust-rewrite/fixtures/rpc/commands.json +238 -0
  15. package/.agent-tasks/pi-rust-rewrite/fixtures/rpc/host-state.json +128 -0
  16. package/.agent-tasks/pi-rust-rewrite/fixtures/rpc/ts-goldens.json +1 -0
  17. package/.agent-tasks/pi-rust-rewrite/fixtures/rpc/ts-replay.ts +317 -0
  18. package/.agent-tasks/pi-rust-rewrite/fixtures/sessions/v1/linear-with-compaction.expected.json +576 -0
  19. package/.agent-tasks/pi-rust-rewrite/fixtures/sessions/v1/linear-with-compaction.jsonl +12 -0
  20. package/.agent-tasks/pi-rust-rewrite/fixtures/sessions/v2/branched.expected.json +387 -0
  21. package/.agent-tasks/pi-rust-rewrite/fixtures/sessions/v2/branched.jsonl +8 -0
  22. package/.agent-tasks/pi-rust-rewrite/fixtures/sessions/v3/basic.expected.json +256 -0
  23. package/.agent-tasks/pi-rust-rewrite/fixtures/sessions/v3/basic.jsonl +5 -0
  24. package/.agent-tasks/pi-rust-rewrite/fixtures/sessions/v3/branched-labels.expected.json +474 -0
  25. package/.agent-tasks/pi-rust-rewrite/fixtures/sessions/v3/branched-labels.jsonl +13 -0
  26. package/.agent-tasks/pi-rust-rewrite/fixtures/sessions/v3/branched-session.expected.json +352 -0
  27. package/.agent-tasks/pi-rust-rewrite/fixtures/sessions/v3/branched-session.jsonl +9 -0
  28. package/.agent-tasks/pi-rust-rewrite/fixtures/sessions/v3/compacted-twice.expected.json +590 -0
  29. package/.agent-tasks/pi-rust-rewrite/fixtures/sessions/v3/compacted-twice.jsonl +13 -0
  30. package/.agent-tasks/pi-rust-rewrite/fixtures/sessions/v3/custom-messages.expected.json +400 -0
  31. package/.agent-tasks/pi-rust-rewrite/fixtures/sessions/v3/custom-messages.jsonl +8 -0
  32. package/.agent-tasks/pi-rust-rewrite/fixtures/sessions/v3/forked-header.expected.json +257 -0
  33. package/.agent-tasks/pi-rust-rewrite/fixtures/sessions/v3/forked-header.jsonl +5 -0
  34. package/.agent-tasks/pi-rust-rewrite/fixtures/sessions/v3/unknown-entries.expected.json +297 -0
  35. package/.agent-tasks/pi-rust-rewrite/fixtures/sessions/v3/unknown-entries.jsonl +6 -0
  36. package/.agent-tasks/pi-rust-rewrite/fixtures/tool-schemas/bash.json +16 -0
  37. package/.agent-tasks/pi-rust-rewrite/fixtures/tool-schemas/edit.json +34 -0
  38. package/.agent-tasks/pi-rust-rewrite/fixtures/tool-schemas/find.json +20 -0
  39. package/.agent-tasks/pi-rust-rewrite/fixtures/tool-schemas/grep.json +36 -0
  40. package/.agent-tasks/pi-rust-rewrite/fixtures/tool-schemas/ls.json +13 -0
  41. package/.agent-tasks/pi-rust-rewrite/fixtures/tool-schemas/read.json +20 -0
  42. package/.agent-tasks/pi-rust-rewrite/fixtures/tool-schemas/write.json +17 -0
  43. package/.agent-tasks/pi-rust-rewrite/fixtures/tools/edit-cases.json +93 -0
  44. package/.agent-tasks/pi-rust-rewrite/helpers/reopen_session.ts +168 -0
  45. package/.agent-tasks/pi-rust-rewrite/src/lib.rs +13 -0
  46. package/.agent-tasks/pi-rust-rewrite/tests/phase7_fixtures.rs +742 -0
  47. package/.agent-tasks/pi-rust-rewrite/tests/pi_ai_contract.rs +101 -0
  48. package/.agent-tasks/pi-rust-rewrite/tests/provider_drain_isolation.rs +760 -0
  49. package/.agent-tasks/pi-rust-rewrite/tests/rpc_parity.rs +679 -0
  50. package/.agent-tasks/pi-rust-rewrite/tests/session_compat.rs +839 -0
  51. package/.agent-tasks/pi-rust-rewrite/tests/tool_parity.rs +761 -0
  52. package/.github/workflows/release-verification.yml +204 -0
  53. package/.gitmodules +5 -0
  54. package/.outline/audit/coverage.json +3262 -0
  55. package/.outline/audit/queue.json +4841 -0
  56. package/.outline/review-fix-grill/iterations/1.json +767 -0
  57. package/.outline/review-fix-grill/iterations/2.json +69 -0
  58. package/.outline/review-fix-grill/iterations/3.json +81 -0
  59. package/.outline/review-fix-grill/iterations/4.json +35 -0
  60. package/.outline/review-fix-grill/iterations/5.json +21 -0
  61. package/.outline/review-fix-grill/queue.json +6454 -0
  62. package/.references/pi/.gitattributes +23 -0
  63. package/.references/pi/.github/APPROVED_CONTRIBUTORS +281 -0
  64. package/.references/pi/.github/ISSUE_TEMPLATE/bug.yml +45 -0
  65. package/.references/pi/.github/ISSUE_TEMPLATE/config.yml +5 -0
  66. package/.references/pi/.github/ISSUE_TEMPLATE/contribution.yml +36 -0
  67. package/.references/pi/.github/ISSUE_TEMPLATE/package-report.yml +49 -0
  68. package/.references/pi/.github/workflows/approve-contributor.yml +179 -0
  69. package/.references/pi/.github/workflows/build-binaries.yml +294 -0
  70. package/.references/pi/.github/workflows/ci.yml +42 -0
  71. package/.references/pi/.github/workflows/issue-analysis.yml +634 -0
  72. package/.references/pi/.github/workflows/issue-gate.yml +129 -0
  73. package/.references/pi/.github/workflows/issue-triage-labels.yml +142 -0
  74. package/.references/pi/.github/workflows/npm-audit.yml +31 -0
  75. package/.references/pi/.github/workflows/pr-gate.yml +128 -0
  76. package/.references/pi/.github/workflows/publish-model-catalog.yml +112 -0
  77. package/.references/pi/.github/workflows/remove-inprogress-on-close.yml +31 -0
  78. package/.references/pi/.husky/pre-commit +45 -0
  79. package/.references/pi/.pi/extensions/import-repro.ts +351 -0
  80. package/.references/pi/.pi/extensions/prompt-url-widget.ts +270 -0
  81. package/.references/pi/.pi/extensions/redraws.ts +24 -0
  82. package/.references/pi/.pi/extensions/tps.ts +47 -0
  83. package/.references/pi/.pi/git/.gitignore +2 -0
  84. package/.references/pi/.pi/npm/.gitignore +2 -0
  85. package/.references/pi/.pi/prompts/cl.md +54 -0
  86. package/.references/pi/.pi/prompts/is.md +28 -0
  87. package/.references/pi/.pi/prompts/pr.md +37 -0
  88. package/.references/pi/.pi/prompts/sa.md +163 -0
  89. package/.references/pi/.pi/prompts/wr.md +40 -0
  90. package/.references/pi/.pi/skills/add-llm-provider.md +57 -0
  91. package/.references/pi/AGENTS.md +162 -0
  92. package/.references/pi/CONTRIBUTING.md +102 -0
  93. package/.references/pi/LICENSE +21 -0
  94. package/.references/pi/README.md +99 -0
  95. package/.references/pi/SECURITY.md +87 -0
  96. package/.references/pi/biome.json +39 -0
  97. package/.references/pi/bun.lock +961 -0
  98. package/.references/pi/package-lock.json +6172 -0
  99. package/.references/pi/package.json +66 -0
  100. package/.references/pi/packages/agent/CHANGELOG.md +594 -0
  101. package/.references/pi/packages/agent/README.md +488 -0
  102. package/.references/pi/packages/agent/docs/agent-harness.md +488 -0
  103. package/.references/pi/packages/agent/docs/durable-harness.md +212 -0
  104. package/.references/pi/packages/agent/docs/hooks.md +445 -0
  105. package/.references/pi/packages/agent/docs/models.md +964 -0
  106. package/.references/pi/packages/agent/docs/observability.md +376 -0
  107. package/.references/pi/packages/agent/package.json +60 -0
  108. package/.references/pi/packages/agent/src/agent-loop.ts +792 -0
  109. package/.references/pi/packages/agent/src/agent.ts +575 -0
  110. package/.references/pi/packages/agent/src/harness/agent-harness.ts +1029 -0
  111. package/.references/pi/packages/agent/src/harness/compaction/branch-summarization.ts +261 -0
  112. package/.references/pi/packages/agent/src/harness/compaction/compaction.ts +753 -0
  113. package/.references/pi/packages/agent/src/harness/compaction/utils.ts +144 -0
  114. package/.references/pi/packages/agent/src/harness/env/nodejs.ts +569 -0
  115. package/.references/pi/packages/agent/src/harness/messages.ts +164 -0
  116. package/.references/pi/packages/agent/src/harness/prompt-templates.ts +267 -0
  117. package/.references/pi/packages/agent/src/harness/session/jsonl-repo.ts +179 -0
  118. package/.references/pi/packages/agent/src/harness/session/jsonl-storage.ts +314 -0
  119. package/.references/pi/packages/agent/src/harness/session/memory-repo.ts +50 -0
  120. package/.references/pi/packages/agent/src/harness/session/memory-storage.ts +133 -0
  121. package/.references/pi/packages/agent/src/harness/session/repo-utils.ts +51 -0
  122. package/.references/pi/packages/agent/src/harness/session/session.ts +338 -0
  123. package/.references/pi/packages/agent/src/harness/session/uuid.ts +54 -0
  124. package/.references/pi/packages/agent/src/harness/skills.ts +375 -0
  125. package/.references/pi/packages/agent/src/harness/system-prompt.ts +34 -0
  126. package/.references/pi/packages/agent/src/harness/types.ts +838 -0
  127. package/.references/pi/packages/agent/src/harness/utils/shell-output.ts +135 -0
  128. package/.references/pi/packages/agent/src/harness/utils/truncate.ts +344 -0
  129. package/.references/pi/packages/agent/src/index.ts +46 -0
  130. package/.references/pi/packages/agent/src/node.ts +2 -0
  131. package/.references/pi/packages/agent/src/proxy.ts +367 -0
  132. package/.references/pi/packages/agent/src/types.ts +430 -0
  133. package/.references/pi/packages/agent/test/agent-loop.test.ts +1424 -0
  134. package/.references/pi/packages/agent/test/agent.test.ts +699 -0
  135. package/.references/pi/packages/agent/test/e2e.test.ts +404 -0
  136. package/.references/pi/packages/agent/test/harness/agent-harness-stream.test.ts +213 -0
  137. package/.references/pi/packages/agent/test/harness/agent-harness.test.ts +608 -0
  138. package/.references/pi/packages/agent/test/harness/compaction.test.ts +655 -0
  139. package/.references/pi/packages/agent/test/harness/nodejs-env.test.ts +321 -0
  140. package/.references/pi/packages/agent/test/harness/prompt-templates.test.ts +90 -0
  141. package/.references/pi/packages/agent/test/harness/repo.test.ts +92 -0
  142. package/.references/pi/packages/agent/test/harness/resource-formatting.test.ts +24 -0
  143. package/.references/pi/packages/agent/test/harness/session-test-utils.ts +55 -0
  144. package/.references/pi/packages/agent/test/harness/session-uuid.test.ts +50 -0
  145. package/.references/pi/packages/agent/test/harness/session.test.ts +203 -0
  146. package/.references/pi/packages/agent/test/harness/skills.test.ts +116 -0
  147. package/.references/pi/packages/agent/test/harness/storage.test.ts +341 -0
  148. package/.references/pi/packages/agent/test/harness/system-prompt.test.ts +66 -0
  149. package/.references/pi/packages/agent/test/harness/truncate.test.ts +169 -0
  150. package/.references/pi/packages/agent/test/scratch/simple.ts +72 -0
  151. package/.references/pi/packages/agent/test/utils/calculate.ts +32 -0
  152. package/.references/pi/packages/agent/test/utils/get-current-time.ts +46 -0
  153. package/.references/pi/packages/agent/tsconfig.build.json +13 -0
  154. package/.references/pi/packages/agent/vitest.config.ts +21 -0
  155. package/.references/pi/packages/agent/vitest.harness.config.ts +28 -0
  156. package/.references/pi/packages/ai/CHANGELOG.md +1739 -0
  157. package/.references/pi/packages/ai/README.md +1621 -0
  158. package/.references/pi/packages/ai/bedrock-provider.d.ts +1 -0
  159. package/.references/pi/packages/ai/bedrock-provider.js +1 -0
  160. package/.references/pi/packages/ai/package.json +97 -0
  161. package/.references/pi/packages/ai/scripts/generate-image-models.ts +131 -0
  162. package/.references/pi/packages/ai/scripts/generate-models.ts +2401 -0
  163. package/.references/pi/packages/ai/scripts/generate-test-image.ts +33 -0
  164. package/.references/pi/packages/ai/src/api/anthropic-messages.lazy.ts +4 -0
  165. package/.references/pi/packages/ai/src/api/anthropic-messages.ts +1313 -0
  166. package/.references/pi/packages/ai/src/api/azure-openai-responses.lazy.ts +4 -0
  167. package/.references/pi/packages/ai/src/api/azure-openai-responses.ts +298 -0
  168. package/.references/pi/packages/ai/src/api/bedrock-converse-stream.lazy.ts +30 -0
  169. package/.references/pi/packages/ai/src/api/bedrock-converse-stream.ts +1091 -0
  170. package/.references/pi/packages/ai/src/api/cloudflare.ts +15 -0
  171. package/.references/pi/packages/ai/src/api/github-copilot-headers.ts +37 -0
  172. package/.references/pi/packages/ai/src/api/google-generative-ai.lazy.ts +4 -0
  173. package/.references/pi/packages/ai/src/api/google-generative-ai.ts +509 -0
  174. package/.references/pi/packages/ai/src/api/google-shared.ts +350 -0
  175. package/.references/pi/packages/ai/src/api/google-vertex.lazy.ts +4 -0
  176. package/.references/pi/packages/ai/src/api/google-vertex.ts +584 -0
  177. package/.references/pi/packages/ai/src/api/lazy.ts +75 -0
  178. package/.references/pi/packages/ai/src/api/mistral-conversations.lazy.ts +4 -0
  179. package/.references/pi/packages/ai/src/api/mistral-conversations.ts +664 -0
  180. package/.references/pi/packages/ai/src/api/openai-codex-responses.lazy.ts +4 -0
  181. package/.references/pi/packages/ai/src/api/openai-codex-responses.ts +1573 -0
  182. package/.references/pi/packages/ai/src/api/openai-completions.lazy.ts +4 -0
  183. package/.references/pi/packages/ai/src/api/openai-completions.ts +1355 -0
  184. package/.references/pi/packages/ai/src/api/openai-prompt-cache.ts +8 -0
  185. package/.references/pi/packages/ai/src/api/openai-responses-shared.ts +613 -0
  186. package/.references/pi/packages/ai/src/api/openai-responses.lazy.ts +4 -0
  187. package/.references/pi/packages/ai/src/api/openai-responses.ts +318 -0
  188. package/.references/pi/packages/ai/src/api/openrouter-images.lazy.ts +10 -0
  189. package/.references/pi/packages/ai/src/api/openrouter-images.ts +185 -0
  190. package/.references/pi/packages/ai/src/api/pi-messages.lazy.ts +4 -0
  191. package/.references/pi/packages/ai/src/api/pi-messages.ts +436 -0
  192. package/.references/pi/packages/ai/src/api/simple-options.ts +77 -0
  193. package/.references/pi/packages/ai/src/api/transform-messages.ts +223 -0
  194. package/.references/pi/packages/ai/src/auth/context.ts +45 -0
  195. package/.references/pi/packages/ai/src/auth/credential-store.ts +51 -0
  196. package/.references/pi/packages/ai/src/auth/helpers.ts +47 -0
  197. package/.references/pi/packages/ai/src/auth/oauth/anthropic.ts +350 -0
  198. package/.references/pi/packages/ai/src/auth/oauth/device-code.ts +98 -0
  199. package/.references/pi/packages/ai/src/auth/oauth/github-copilot.ts +379 -0
  200. package/.references/pi/packages/ai/src/auth/oauth/load.ts +56 -0
  201. package/.references/pi/packages/ai/src/auth/oauth/oauth-page.ts +109 -0
  202. package/.references/pi/packages/ai/src/auth/oauth/openai-codex.ts +538 -0
  203. package/.references/pi/packages/ai/src/auth/oauth/pkce.ts +34 -0
  204. package/.references/pi/packages/ai/src/auth/oauth/radius.ts +410 -0
  205. package/.references/pi/packages/ai/src/auth/oauth/xai.ts +238 -0
  206. package/.references/pi/packages/ai/src/auth/resolve.ts +139 -0
  207. package/.references/pi/packages/ai/src/auth/types.ts +220 -0
  208. package/.references/pi/packages/ai/src/bedrock-provider.ts +6 -0
  209. package/.references/pi/packages/ai/src/bun-oauth.ts +17 -0
  210. package/.references/pi/packages/ai/src/cli.ts +118 -0
  211. package/.references/pi/packages/ai/src/compat/extension-oauth-types.ts +45 -0
  212. package/.references/pi/packages/ai/src/compat.ts +298 -0
  213. package/.references/pi/packages/ai/src/env-api-keys.ts +178 -0
  214. package/.references/pi/packages/ai/src/image-models.generated.ts +534 -0
  215. package/.references/pi/packages/ai/src/image-models.ts +42 -0
  216. package/.references/pi/packages/ai/src/images-api-registry.ts +53 -0
  217. package/.references/pi/packages/ai/src/images-models.ts +274 -0
  218. package/.references/pi/packages/ai/src/images.ts +21 -0
  219. package/.references/pi/packages/ai/src/index.ts +45 -0
  220. package/.references/pi/packages/ai/src/legacy-api-aliases.ts +108 -0
  221. package/.references/pi/packages/ai/src/models-store.ts +38 -0
  222. package/.references/pi/packages/ai/src/models.generated.ts +76 -0
  223. package/.references/pi/packages/ai/src/models.ts +705 -0
  224. package/.references/pi/packages/ai/src/oauth.ts +10 -0
  225. package/.references/pi/packages/ai/src/providers/all.ts +140 -0
  226. package/.references/pi/packages/ai/src/providers/amazon-bedrock.models.ts +1897 -0
  227. package/.references/pi/packages/ai/src/providers/amazon-bedrock.ts +80 -0
  228. package/.references/pi/packages/ai/src/providers/ant-ling.models.ts +62 -0
  229. package/.references/pi/packages/ai/src/providers/ant-ling.ts +15 -0
  230. package/.references/pi/packages/ai/src/providers/anthropic.models.ts +257 -0
  231. package/.references/pi/packages/ai/src/providers/anthropic.ts +20 -0
  232. package/.references/pi/packages/ai/src/providers/azure-openai-responses.models.ts +816 -0
  233. package/.references/pi/packages/ai/src/providers/azure-openai-responses.ts +14 -0
  234. package/.references/pi/packages/ai/src/providers/cerebras.models.ts +61 -0
  235. package/.references/pi/packages/ai/src/providers/cerebras.ts +15 -0
  236. package/.references/pi/packages/ai/src/providers/cloudflare-ai-gateway.models.ts +760 -0
  237. package/.references/pi/packages/ai/src/providers/cloudflare-ai-gateway.ts +23 -0
  238. package/.references/pi/packages/ai/src/providers/cloudflare-auth.ts +96 -0
  239. package/.references/pi/packages/ai/src/providers/cloudflare-stream.ts +28 -0
  240. package/.references/pi/packages/ai/src/providers/cloudflare-workers-ai.models.ts +241 -0
  241. package/.references/pi/packages/ai/src/providers/cloudflare-workers-ai.ts +15 -0
  242. package/.references/pi/packages/ai/src/providers/deepseek.models.ts +45 -0
  243. package/.references/pi/packages/ai/src/providers/deepseek.ts +15 -0
  244. package/.references/pi/packages/ai/src/providers/faux.ts +538 -0
  245. package/.references/pi/packages/ai/src/providers/fireworks.models.ts +297 -0
  246. package/.references/pi/packages/ai/src/providers/fireworks.ts +19 -0
  247. package/.references/pi/packages/ai/src/providers/github-copilot.models.ts +543 -0
  248. package/.references/pi/packages/ai/src/providers/github-copilot.ts +34 -0
  249. package/.references/pi/packages/ai/src/providers/google-vertex.models.ts +184 -0
  250. package/.references/pi/packages/ai/src/providers/google-vertex.ts +93 -0
  251. package/.references/pi/packages/ai/src/providers/google.models.ts +290 -0
  252. package/.references/pi/packages/ai/src/providers/google.ts +15 -0
  253. package/.references/pi/packages/ai/src/providers/groq.models.ts +127 -0
  254. package/.references/pi/packages/ai/src/providers/groq.ts +15 -0
  255. package/.references/pi/packages/ai/src/providers/huggingface.models.ts +889 -0
  256. package/.references/pi/packages/ai/src/providers/huggingface.ts +15 -0
  257. package/.references/pi/packages/ai/src/providers/images/register-builtins.ts +50 -0
  258. package/.references/pi/packages/ai/src/providers/kimi-coding.models.ts +103 -0
  259. package/.references/pi/packages/ai/src/providers/kimi-coding.ts +15 -0
  260. package/.references/pi/packages/ai/src/providers/minimax-cn.models.ts +58 -0
  261. package/.references/pi/packages/ai/src/providers/minimax-cn.ts +15 -0
  262. package/.references/pi/packages/ai/src/providers/minimax.models.ts +58 -0
  263. package/.references/pi/packages/ai/src/providers/minimax.ts +15 -0
  264. package/.references/pi/packages/ai/src/providers/mistral.models.ts +517 -0
  265. package/.references/pi/packages/ai/src/providers/mistral.ts +15 -0
  266. package/.references/pi/packages/ai/src/providers/moonshotai-cn.models.ts +190 -0
  267. package/.references/pi/packages/ai/src/providers/moonshotai-cn.ts +15 -0
  268. package/.references/pi/packages/ai/src/providers/moonshotai.models.ts +190 -0
  269. package/.references/pi/packages/ai/src/providers/moonshotai.ts +15 -0
  270. package/.references/pi/packages/ai/src/providers/nvidia.models.ts +387 -0
  271. package/.references/pi/packages/ai/src/providers/nvidia.ts +15 -0
  272. package/.references/pi/packages/ai/src/providers/openai-codex.models.ts +144 -0
  273. package/.references/pi/packages/ai/src/providers/openai-codex.ts +18 -0
  274. package/.references/pi/packages/ai/src/providers/openai.models.ts +830 -0
  275. package/.references/pi/packages/ai/src/providers/openai.ts +15 -0
  276. package/.references/pi/packages/ai/src/providers/opencode-go.models.ts +278 -0
  277. package/.references/pi/packages/ai/src/providers/opencode-go.ts +18 -0
  278. package/.references/pi/packages/ai/src/providers/opencode.models.ts +1001 -0
  279. package/.references/pi/packages/ai/src/providers/opencode.ts +24 -0
  280. package/.references/pi/packages/ai/src/providers/openrouter-images.ts +14 -0
  281. package/.references/pi/packages/ai/src/providers/openrouter.models.ts +4915 -0
  282. package/.references/pi/packages/ai/src/providers/openrouter.ts +15 -0
  283. package/.references/pi/packages/ai/src/providers/radius-config.ts +95 -0
  284. package/.references/pi/packages/ai/src/providers/radius.ts +67 -0
  285. package/.references/pi/packages/ai/src/providers/together.models.ts +382 -0
  286. package/.references/pi/packages/ai/src/providers/together.ts +15 -0
  287. package/.references/pi/packages/ai/src/providers/vercel-ai-gateway.models.ts +3268 -0
  288. package/.references/pi/packages/ai/src/providers/vercel-ai-gateway.ts +15 -0
  289. package/.references/pi/packages/ai/src/providers/xai.models.ts +62 -0
  290. package/.references/pi/packages/ai/src/providers/xai.ts +27 -0
  291. package/.references/pi/packages/ai/src/providers/xiaomi-token-plan-ams.models.ts +61 -0
  292. package/.references/pi/packages/ai/src/providers/xiaomi-token-plan-ams.ts +15 -0
  293. package/.references/pi/packages/ai/src/providers/xiaomi-token-plan-cn.models.ts +61 -0
  294. package/.references/pi/packages/ai/src/providers/xiaomi-token-plan-cn.ts +15 -0
  295. package/.references/pi/packages/ai/src/providers/xiaomi-token-plan-sgp.models.ts +61 -0
  296. package/.references/pi/packages/ai/src/providers/xiaomi-token-plan-sgp.ts +15 -0
  297. package/.references/pi/packages/ai/src/providers/xiaomi.models.ts +115 -0
  298. package/.references/pi/packages/ai/src/providers/xiaomi.ts +15 -0
  299. package/.references/pi/packages/ai/src/providers/zai-coding-cn.models.ts +116 -0
  300. package/.references/pi/packages/ai/src/providers/zai-coding-cn.ts +15 -0
  301. package/.references/pi/packages/ai/src/providers/zai.models.ts +116 -0
  302. package/.references/pi/packages/ai/src/providers/zai.ts +15 -0
  303. package/.references/pi/packages/ai/src/session-resources.ts +24 -0
  304. package/.references/pi/packages/ai/src/types.ts +738 -0
  305. package/.references/pi/packages/ai/src/utils/abort-signals.ts +41 -0
  306. package/.references/pi/packages/ai/src/utils/deferred-tools.ts +39 -0
  307. package/.references/pi/packages/ai/src/utils/diagnostics.ts +45 -0
  308. package/.references/pi/packages/ai/src/utils/error-body.ts +127 -0
  309. package/.references/pi/packages/ai/src/utils/estimate.ts +143 -0
  310. package/.references/pi/packages/ai/src/utils/event-stream.ts +88 -0
  311. package/.references/pi/packages/ai/src/utils/hash.ts +13 -0
  312. package/.references/pi/packages/ai/src/utils/headers.ts +18 -0
  313. package/.references/pi/packages/ai/src/utils/json-parse.ts +124 -0
  314. package/.references/pi/packages/ai/src/utils/node-http-proxy.ts +112 -0
  315. package/.references/pi/packages/ai/src/utils/overflow.ts +165 -0
  316. package/.references/pi/packages/ai/src/utils/provider-env.ts +52 -0
  317. package/.references/pi/packages/ai/src/utils/retry.ts +101 -0
  318. package/.references/pi/packages/ai/src/utils/sanitize-unicode.ts +25 -0
  319. package/.references/pi/packages/ai/src/utils/typebox-helpers.ts +24 -0
  320. package/.references/pi/packages/ai/src/utils/validation.ts +310 -0
  321. package/.references/pi/packages/ai/test/abort.test.ts +303 -0
  322. package/.references/pi/packages/ai/test/anthropic-adaptive-thinking-models.test.ts +39 -0
  323. package/.references/pi/packages/ai/test/anthropic-cache-write-1h-cost.test.ts +86 -0
  324. package/.references/pi/packages/ai/test/anthropic-eager-tool-input-compat.test.ts +122 -0
  325. package/.references/pi/packages/ai/test/anthropic-eager-tool-input-e2e.test.ts +155 -0
  326. package/.references/pi/packages/ai/test/anthropic-empty-thinking-signature-compat.test.ts +108 -0
  327. package/.references/pi/packages/ai/test/anthropic-force-adaptive-thinking.test.ts +125 -0
  328. package/.references/pi/packages/ai/test/anthropic-long-cache-retention-e2e.test.ts +127 -0
  329. package/.references/pi/packages/ai/test/anthropic-oauth.test.ts +137 -0
  330. package/.references/pi/packages/ai/test/anthropic-opus-4-8-smoke.test.ts +71 -0
  331. package/.references/pi/packages/ai/test/anthropic-sse-parsing.test.ts +275 -0
  332. package/.references/pi/packages/ai/test/anthropic-temperature-compat.test.ts +103 -0
  333. package/.references/pi/packages/ai/test/anthropic-thinking-disable.test.ts +172 -0
  334. package/.references/pi/packages/ai/test/anthropic-tool-name-normalization.test.ts +204 -0
  335. package/.references/pi/packages/ai/test/azure-openai-base-url.test.ts +175 -0
  336. package/.references/pi/packages/ai/test/azure-openai-responses-reasoning-replay.test.ts +136 -0
  337. package/.references/pi/packages/ai/test/azure-utils.ts +28 -0
  338. package/.references/pi/packages/ai/test/bedrock-convert-messages.test.ts +245 -0
  339. package/.references/pi/packages/ai/test/bedrock-custom-headers.test.ts +202 -0
  340. package/.references/pi/packages/ai/test/bedrock-endpoint-resolution.test.ts +208 -0
  341. package/.references/pi/packages/ai/test/bedrock-models.test.ts +65 -0
  342. package/.references/pi/packages/ai/test/bedrock-thinking-payload.test.ts +246 -0
  343. package/.references/pi/packages/ai/test/bedrock-utils.ts +18 -0
  344. package/.references/pi/packages/ai/test/cache-retention.test.ts +498 -0
  345. package/.references/pi/packages/ai/test/cloudflare-stream.test.ts +65 -0
  346. package/.references/pi/packages/ai/test/cloudflare-utils.ts +9 -0
  347. package/.references/pi/packages/ai/test/codex-websocket-cached-probe.ts +299 -0
  348. package/.references/pi/packages/ai/test/compat-env.test.ts +74 -0
  349. package/.references/pi/packages/ai/test/context-estimate.test.ts +81 -0
  350. package/.references/pi/packages/ai/test/context-overflow.test.ts +734 -0
  351. package/.references/pi/packages/ai/test/cross-provider-handoff.test.ts +502 -0
  352. package/.references/pi/packages/ai/test/data/red-circle.png +0 -0
  353. package/.references/pi/packages/ai/test/deferred-tools.test.ts +481 -0
  354. package/.references/pi/packages/ai/test/empty.test.ts +741 -0
  355. package/.references/pi/packages/ai/test/env-api-keys.test.ts +60 -0
  356. package/.references/pi/packages/ai/test/error-body.test.ts +154 -0
  357. package/.references/pi/packages/ai/test/faux-provider.test.ts +597 -0
  358. package/.references/pi/packages/ai/test/fireworks-models.test.ts +269 -0
  359. package/.references/pi/packages/ai/test/github-copilot-anthropic.test.ts +120 -0
  360. package/.references/pi/packages/ai/test/github-copilot-oauth.test.ts +444 -0
  361. package/.references/pi/packages/ai/test/google-shared-convert-tools.test.ts +187 -0
  362. package/.references/pi/packages/ai/test/google-shared-gemini3-unsigned-tool-call.test.ts +116 -0
  363. package/.references/pi/packages/ai/test/google-shared-image-tool-result-routing.test.ts +102 -0
  364. package/.references/pi/packages/ai/test/google-thinking-disable.test.ts +160 -0
  365. package/.references/pi/packages/ai/test/google-thinking-signature.test.ts +38 -0
  366. package/.references/pi/packages/ai/test/google-vertex-api-key-resolution.test.ts +223 -0
  367. package/.references/pi/packages/ai/test/image-tool-result.test.ts +501 -0
  368. package/.references/pi/packages/ai/test/images-models.test.ts +209 -0
  369. package/.references/pi/packages/ai/test/images.test.ts +90 -0
  370. package/.references/pi/packages/ai/test/interleaved-thinking.test.ts +144 -0
  371. package/.references/pi/packages/ai/test/lax-message-content.test.ts +67 -0
  372. package/.references/pi/packages/ai/test/lazy-module-load.test.ts +119 -0
  373. package/.references/pi/packages/ai/test/max-thinking.test.ts +89 -0
  374. package/.references/pi/packages/ai/test/mistral-reasoning-mode.test.ts +97 -0
  375. package/.references/pi/packages/ai/test/mistral-tool-schema.test.ts +60 -0
  376. package/.references/pi/packages/ai/test/models-runtime.test.ts +689 -0
  377. package/.references/pi/packages/ai/test/node-http-proxy.test.ts +76 -0
  378. package/.references/pi/packages/ai/test/oauth-auth.test.ts +141 -0
  379. package/.references/pi/packages/ai/test/oauth-device-code.test.ts +139 -0
  380. package/.references/pi/packages/ai/test/oauth.ts +83 -0
  381. package/.references/pi/packages/ai/test/openai-codex-cache-affinity-e2e.test.ts +35 -0
  382. package/.references/pi/packages/ai/test/openai-codex-oauth.test.ts +479 -0
  383. package/.references/pi/packages/ai/test/openai-codex-stream.test.ts +2028 -0
  384. package/.references/pi/packages/ai/test/openai-completions-cache-control-format.test.ts +189 -0
  385. package/.references/pi/packages/ai/test/openai-completions-empty-tools.test.ts +304 -0
  386. package/.references/pi/packages/ai/test/openai-completions-prompt-cache.test.ts +264 -0
  387. package/.references/pi/packages/ai/test/openai-completions-reasoning-details.test.ts +118 -0
  388. package/.references/pi/packages/ai/test/openai-completions-response-model.test.ts +140 -0
  389. package/.references/pi/packages/ai/test/openai-completions-retry.test.ts +86 -0
  390. package/.references/pi/packages/ai/test/openai-completions-thinking-as-text.test.ts +218 -0
  391. package/.references/pi/packages/ai/test/openai-completions-tool-choice.test.ts +1716 -0
  392. package/.references/pi/packages/ai/test/openai-completions-tool-result-images.test.ts +152 -0
  393. package/.references/pi/packages/ai/test/openai-responses-cache-affinity-e2e.test.ts +31 -0
  394. package/.references/pi/packages/ai/test/openai-responses-compat.test.ts +472 -0
  395. package/.references/pi/packages/ai/test/openai-responses-empty-tool-result.test.ts +58 -0
  396. package/.references/pi/packages/ai/test/openai-responses-foreign-toolcall-id.test.ts +66 -0
  397. package/.references/pi/packages/ai/test/openai-responses-message-id.test.ts +48 -0
  398. package/.references/pi/packages/ai/test/openai-responses-partial-json-cleanup.test.ts +106 -0
  399. package/.references/pi/packages/ai/test/openai-responses-reasoning-replay-e2e.test.ts +291 -0
  400. package/.references/pi/packages/ai/test/openai-responses-terminal-event.test.ts +233 -0
  401. package/.references/pi/packages/ai/test/openai-responses-tool-result-images.test.ts +194 -0
  402. package/.references/pi/packages/ai/test/openrouter-cache-write-repro.test.ts +77 -0
  403. package/.references/pi/packages/ai/test/openrouter-images.test.ts +140 -0
  404. package/.references/pi/packages/ai/test/overflow.test.ts +139 -0
  405. package/.references/pi/packages/ai/test/pi-messages.test.ts +243 -0
  406. package/.references/pi/packages/ai/test/provider-error-body-passthrough.test.ts +78 -0
  407. package/.references/pi/packages/ai/test/provider-error-body-regression.test.ts +189 -0
  408. package/.references/pi/packages/ai/test/providers.test.ts +404 -0
  409. package/.references/pi/packages/ai/test/responseid.test.ts +120 -0
  410. package/.references/pi/packages/ai/test/retry.test.ts +59 -0
  411. package/.references/pi/packages/ai/test/scratch.ts +57 -0
  412. package/.references/pi/packages/ai/test/stream.test.ts +1632 -0
  413. package/.references/pi/packages/ai/test/supports-xhigh.test.ts +145 -0
  414. package/.references/pi/packages/ai/test/together-models.test.ts +78 -0
  415. package/.references/pi/packages/ai/test/tokens.test.ts +333 -0
  416. package/.references/pi/packages/ai/test/tool-call-id-normalization.test.ts +290 -0
  417. package/.references/pi/packages/ai/test/tool-call-without-result.test.ts +327 -0
  418. package/.references/pi/packages/ai/test/total-tokens.test.ts +789 -0
  419. package/.references/pi/packages/ai/test/transform-messages-copilot-openai-to-anthropic.test.ts +191 -0
  420. package/.references/pi/packages/ai/test/unicode-surrogate.test.ts +774 -0
  421. package/.references/pi/packages/ai/test/validation.test.ts +116 -0
  422. package/.references/pi/packages/ai/test/xai-oauth.test.ts +330 -0
  423. package/.references/pi/packages/ai/test/xai-responses.test.ts +117 -0
  424. package/.references/pi/packages/ai/test/xhigh.test.ts +70 -0
  425. package/.references/pi/packages/ai/test/xiaomi-models.test.ts +17 -0
  426. package/.references/pi/packages/ai/test/xiaomi-token-plan-ams-anthropic-empty-signature-smoke.test.ts +114 -0
  427. package/.references/pi/packages/ai/test/zen.test.ts +25 -0
  428. package/.references/pi/packages/ai/tsconfig.build.json +9 -0
  429. package/.references/pi/packages/ai/vitest.config.ts +11 -0
  430. package/.references/pi/packages/coding-agent/CHANGELOG.md +5038 -0
  431. package/.references/pi/packages/coding-agent/README.md +691 -0
  432. package/.references/pi/packages/coding-agent/docs/compaction.md +396 -0
  433. package/.references/pi/packages/coding-agent/docs/containerization.md +111 -0
  434. package/.references/pi/packages/coding-agent/docs/custom-provider.md +735 -0
  435. package/.references/pi/packages/coding-agent/docs/development.md +71 -0
  436. package/.references/pi/packages/coding-agent/docs/docs.json +156 -0
  437. package/.references/pi/packages/coding-agent/docs/extensions.md +2911 -0
  438. package/.references/pi/packages/coding-agent/docs/images/doom-extension.png +0 -0
  439. package/.references/pi/packages/coding-agent/docs/images/exy.png +0 -0
  440. package/.references/pi/packages/coding-agent/docs/images/interactive-mode.png +0 -0
  441. package/.references/pi/packages/coding-agent/docs/images/tree-view.png +0 -0
  442. package/.references/pi/packages/coding-agent/docs/index.md +82 -0
  443. package/.references/pi/packages/coding-agent/docs/json.md +82 -0
  444. package/.references/pi/packages/coding-agent/docs/keybindings.md +198 -0
  445. package/.references/pi/packages/coding-agent/docs/models.md +540 -0
  446. package/.references/pi/packages/coding-agent/docs/packages.md +228 -0
  447. package/.references/pi/packages/coding-agent/docs/prompt-templates.md +95 -0
  448. package/.references/pi/packages/coding-agent/docs/providers.md +290 -0
  449. package/.references/pi/packages/coding-agent/docs/quickstart.md +165 -0
  450. package/.references/pi/packages/coding-agent/docs/rpc.md +1480 -0
  451. package/.references/pi/packages/coding-agent/docs/sdk.md +1183 -0
  452. package/.references/pi/packages/coding-agent/docs/security.md +59 -0
  453. package/.references/pi/packages/coding-agent/docs/session-format.md +422 -0
  454. package/.references/pi/packages/coding-agent/docs/sessions.md +145 -0
  455. package/.references/pi/packages/coding-agent/docs/settings.md +319 -0
  456. package/.references/pi/packages/coding-agent/docs/shell-aliases.md +13 -0
  457. package/.references/pi/packages/coding-agent/docs/skills.md +231 -0
  458. package/.references/pi/packages/coding-agent/docs/terminal-setup.md +142 -0
  459. package/.references/pi/packages/coding-agent/docs/termux.md +127 -0
  460. package/.references/pi/packages/coding-agent/docs/themes.md +297 -0
  461. package/.references/pi/packages/coding-agent/docs/tmux.md +63 -0
  462. package/.references/pi/packages/coding-agent/docs/tui.md +927 -0
  463. package/.references/pi/packages/coding-agent/docs/usage.md +310 -0
  464. package/.references/pi/packages/coding-agent/docs/windows.md +17 -0
  465. package/.references/pi/packages/coding-agent/examples/README.md +25 -0
  466. package/.references/pi/packages/coding-agent/examples/extensions/README.md +213 -0
  467. package/.references/pi/packages/coding-agent/examples/extensions/auto-commit-on-exit.ts +49 -0
  468. package/.references/pi/packages/coding-agent/examples/extensions/bash-spawn-hook.ts +30 -0
  469. package/.references/pi/packages/coding-agent/examples/extensions/bookmark.ts +50 -0
  470. package/.references/pi/packages/coding-agent/examples/extensions/border-status-editor.ts +150 -0
  471. package/.references/pi/packages/coding-agent/examples/extensions/built-in-tool-renderer.ts +249 -0
  472. package/.references/pi/packages/coding-agent/examples/extensions/claude-rules.ts +86 -0
  473. package/.references/pi/packages/coding-agent/examples/extensions/commands.ts +72 -0
  474. package/.references/pi/packages/coding-agent/examples/extensions/confirm-destructive.ts +59 -0
  475. package/.references/pi/packages/coding-agent/examples/extensions/custom-compaction.ts +128 -0
  476. package/.references/pi/packages/coding-agent/examples/extensions/custom-footer.ts +64 -0
  477. package/.references/pi/packages/coding-agent/examples/extensions/custom-header.ts +73 -0
  478. package/.references/pi/packages/coding-agent/examples/extensions/custom-provider-anthropic/index.ts +604 -0
  479. package/.references/pi/packages/coding-agent/examples/extensions/custom-provider-anthropic/package-lock.json +24 -0
  480. package/.references/pi/packages/coding-agent/examples/extensions/custom-provider-anthropic/package.json +19 -0
  481. package/.references/pi/packages/coding-agent/examples/extensions/custom-provider-gitlab-duo/index.ts +404 -0
  482. package/.references/pi/packages/coding-agent/examples/extensions/custom-provider-gitlab-duo/package.json +16 -0
  483. package/.references/pi/packages/coding-agent/examples/extensions/custom-provider-gitlab-duo/test.ts +82 -0
  484. package/.references/pi/packages/coding-agent/examples/extensions/dirty-repo-guard.ts +56 -0
  485. package/.references/pi/packages/coding-agent/examples/extensions/doom-overlay/README.md +46 -0
  486. package/.references/pi/packages/coding-agent/examples/extensions/doom-overlay/doom/build/doom.js +21 -0
  487. package/.references/pi/packages/coding-agent/examples/extensions/doom-overlay/doom/build/doom.wasm +0 -0
  488. package/.references/pi/packages/coding-agent/examples/extensions/doom-overlay/doom/build.sh +152 -0
  489. package/.references/pi/packages/coding-agent/examples/extensions/doom-overlay/doom/doomgeneric_pi.c +72 -0
  490. package/.references/pi/packages/coding-agent/examples/extensions/doom-overlay/doom-component.ts +132 -0
  491. package/.references/pi/packages/coding-agent/examples/extensions/doom-overlay/doom-engine.ts +173 -0
  492. package/.references/pi/packages/coding-agent/examples/extensions/doom-overlay/doom-keys.ts +104 -0
  493. package/.references/pi/packages/coding-agent/examples/extensions/doom-overlay/index.ts +74 -0
  494. package/.references/pi/packages/coding-agent/examples/extensions/doom-overlay/wad-finder.ts +51 -0
  495. package/.references/pi/packages/coding-agent/examples/extensions/dynamic-resources/SKILL.md +8 -0
  496. package/.references/pi/packages/coding-agent/examples/extensions/dynamic-resources/dynamic.json +79 -0
  497. package/.references/pi/packages/coding-agent/examples/extensions/dynamic-resources/dynamic.md +5 -0
  498. package/.references/pi/packages/coding-agent/examples/extensions/dynamic-resources/index.ts +15 -0
  499. package/.references/pi/packages/coding-agent/examples/extensions/dynamic-tools.ts +74 -0
  500. package/.references/pi/packages/coding-agent/examples/extensions/entry-renderer.ts +41 -0
  501. package/.references/pi/packages/coding-agent/examples/extensions/event-bus.ts +43 -0
  502. package/.references/pi/packages/coding-agent/examples/extensions/file-trigger.ts +41 -0
  503. package/.references/pi/packages/coding-agent/examples/extensions/git-checkpoint.ts +53 -0
  504. package/.references/pi/packages/coding-agent/examples/extensions/git-merge-and-resolve.ts +115 -0
  505. package/.references/pi/packages/coding-agent/examples/extensions/github-issue-autocomplete.ts +185 -0
  506. package/.references/pi/packages/coding-agent/examples/extensions/gondolin/index.ts +531 -0
  507. package/.references/pi/packages/coding-agent/examples/extensions/gondolin/package-lock.json +185 -0
  508. package/.references/pi/packages/coding-agent/examples/extensions/gondolin/package.json +19 -0
  509. package/.references/pi/packages/coding-agent/examples/extensions/handoff.ts +191 -0
  510. package/.references/pi/packages/coding-agent/examples/extensions/hello.ts +26 -0
  511. package/.references/pi/packages/coding-agent/examples/extensions/hidden-thinking-label.ts +53 -0
  512. package/.references/pi/packages/coding-agent/examples/extensions/inline-bash.ts +94 -0
  513. package/.references/pi/packages/coding-agent/examples/extensions/input-transform-streaming.ts +39 -0
  514. package/.references/pi/packages/coding-agent/examples/extensions/input-transform.ts +43 -0
  515. package/.references/pi/packages/coding-agent/examples/extensions/interactive-shell.ts +196 -0
  516. package/.references/pi/packages/coding-agent/examples/extensions/kimi-deferred-tools.ts +61 -0
  517. package/.references/pi/packages/coding-agent/examples/extensions/mac-system-theme.ts +47 -0
  518. package/.references/pi/packages/coding-agent/examples/extensions/message-renderer.ts +59 -0
  519. package/.references/pi/packages/coding-agent/examples/extensions/minimal-mode.ts +426 -0
  520. package/.references/pi/packages/coding-agent/examples/extensions/modal-editor.ts +85 -0
  521. package/.references/pi/packages/coding-agent/examples/extensions/model-status.ts +31 -0
  522. package/.references/pi/packages/coding-agent/examples/extensions/notify.ts +55 -0
  523. package/.references/pi/packages/coding-agent/examples/extensions/overlay-qa-tests.ts +1450 -0
  524. package/.references/pi/packages/coding-agent/examples/extensions/overlay-test.ts +153 -0
  525. package/.references/pi/packages/coding-agent/examples/extensions/permission-gate.ts +34 -0
  526. package/.references/pi/packages/coding-agent/examples/extensions/pirate.ts +47 -0
  527. package/.references/pi/packages/coding-agent/examples/extensions/plan-mode/README.md +66 -0
  528. package/.references/pi/packages/coding-agent/examples/extensions/plan-mode/index.ts +390 -0
  529. package/.references/pi/packages/coding-agent/examples/extensions/plan-mode/utils.ts +168 -0
  530. package/.references/pi/packages/coding-agent/examples/extensions/preset.ts +436 -0
  531. package/.references/pi/packages/coding-agent/examples/extensions/project-trust.ts +64 -0
  532. package/.references/pi/packages/coding-agent/examples/extensions/prompt-customizer.ts +97 -0
  533. package/.references/pi/packages/coding-agent/examples/extensions/protected-paths.ts +30 -0
  534. package/.references/pi/packages/coding-agent/examples/extensions/provider-payload.ts +18 -0
  535. package/.references/pi/packages/coding-agent/examples/extensions/qna.ts +122 -0
  536. package/.references/pi/packages/coding-agent/examples/extensions/question.ts +286 -0
  537. package/.references/pi/packages/coding-agent/examples/extensions/questionnaire.ts +448 -0
  538. package/.references/pi/packages/coding-agent/examples/extensions/rainbow-editor.ts +88 -0
  539. package/.references/pi/packages/coding-agent/examples/extensions/reload-runtime.ts +37 -0
  540. package/.references/pi/packages/coding-agent/examples/extensions/rpc-demo.ts +118 -0
  541. package/.references/pi/packages/coding-agent/examples/extensions/sandbox/index.ts +321 -0
  542. package/.references/pi/packages/coding-agent/examples/extensions/sandbox/package-lock.json +92 -0
  543. package/.references/pi/packages/coding-agent/examples/extensions/sandbox/package.json +19 -0
  544. package/.references/pi/packages/coding-agent/examples/extensions/send-user-message.ts +97 -0
  545. package/.references/pi/packages/coding-agent/examples/extensions/session-name.ts +27 -0
  546. package/.references/pi/packages/coding-agent/examples/extensions/shutdown-command.ts +63 -0
  547. package/.references/pi/packages/coding-agent/examples/extensions/snake.ts +343 -0
  548. package/.references/pi/packages/coding-agent/examples/extensions/space-invaders.ts +560 -0
  549. package/.references/pi/packages/coding-agent/examples/extensions/ssh.ts +220 -0
  550. package/.references/pi/packages/coding-agent/examples/extensions/status-line.ts +32 -0
  551. package/.references/pi/packages/coding-agent/examples/extensions/structured-output.ts +65 -0
  552. package/.references/pi/packages/coding-agent/examples/extensions/subagent/README.md +175 -0
  553. package/.references/pi/packages/coding-agent/examples/extensions/subagent/agents/planner.md +37 -0
  554. package/.references/pi/packages/coding-agent/examples/extensions/subagent/agents/reviewer.md +35 -0
  555. package/.references/pi/packages/coding-agent/examples/extensions/subagent/agents/scout.md +50 -0
  556. package/.references/pi/packages/coding-agent/examples/extensions/subagent/agents/worker.md +24 -0
  557. package/.references/pi/packages/coding-agent/examples/extensions/subagent/agents.ts +126 -0
  558. package/.references/pi/packages/coding-agent/examples/extensions/subagent/index.ts +1015 -0
  559. package/.references/pi/packages/coding-agent/examples/extensions/subagent/prompts/implement-and-review.md +10 -0
  560. package/.references/pi/packages/coding-agent/examples/extensions/subagent/prompts/implement.md +10 -0
  561. package/.references/pi/packages/coding-agent/examples/extensions/subagent/prompts/scout-and-plan.md +9 -0
  562. package/.references/pi/packages/coding-agent/examples/extensions/summarize.ts +207 -0
  563. package/.references/pi/packages/coding-agent/examples/extensions/system-prompt-header.ts +17 -0
  564. package/.references/pi/packages/coding-agent/examples/extensions/tic-tac-toe.ts +1008 -0
  565. package/.references/pi/packages/coding-agent/examples/extensions/timed-confirm.ts +70 -0
  566. package/.references/pi/packages/coding-agent/examples/extensions/titlebar-spinner.ts +58 -0
  567. package/.references/pi/packages/coding-agent/examples/extensions/todo.ts +297 -0
  568. package/.references/pi/packages/coding-agent/examples/extensions/tool-override.ts +144 -0
  569. package/.references/pi/packages/coding-agent/examples/extensions/tools.ts +146 -0
  570. package/.references/pi/packages/coding-agent/examples/extensions/trigger-compact.ts +50 -0
  571. package/.references/pi/packages/coding-agent/examples/extensions/truncated-tool.ts +195 -0
  572. package/.references/pi/packages/coding-agent/examples/extensions/widget-placement.ts +9 -0
  573. package/.references/pi/packages/coding-agent/examples/extensions/with-deps/index.ts +32 -0
  574. package/.references/pi/packages/coding-agent/examples/extensions/with-deps/package-lock.json +31 -0
  575. package/.references/pi/packages/coding-agent/examples/extensions/with-deps/package.json +22 -0
  576. package/.references/pi/packages/coding-agent/examples/extensions/working-indicator.ts +123 -0
  577. package/.references/pi/packages/coding-agent/examples/extensions/working-message-test.ts +25 -0
  578. package/.references/pi/packages/coding-agent/examples/rpc-extension-ui.ts +632 -0
  579. package/.references/pi/packages/coding-agent/examples/sdk/01-minimal.ts +26 -0
  580. package/.references/pi/packages/coding-agent/examples/sdk/02-custom-model.ts +50 -0
  581. package/.references/pi/packages/coding-agent/examples/sdk/03-custom-prompt.ts +75 -0
  582. package/.references/pi/packages/coding-agent/examples/sdk/04-skills.ts +55 -0
  583. package/.references/pi/packages/coding-agent/examples/sdk/05-tools.ts +48 -0
  584. package/.references/pi/packages/coding-agent/examples/sdk/06-extensions.ts +99 -0
  585. package/.references/pi/packages/coding-agent/examples/sdk/07-context-files.ts +47 -0
  586. package/.references/pi/packages/coding-agent/examples/sdk/08-prompt-templates.ts +51 -0
  587. package/.references/pi/packages/coding-agent/examples/sdk/09-api-keys-and-oauth.ts +34 -0
  588. package/.references/pi/packages/coding-agent/examples/sdk/10-settings.ts +53 -0
  589. package/.references/pi/packages/coding-agent/examples/sdk/11-sessions.ts +52 -0
  590. package/.references/pi/packages/coding-agent/examples/sdk/12-full-control.ts +72 -0
  591. package/.references/pi/packages/coding-agent/examples/sdk/13-session-runtime.ts +67 -0
  592. package/.references/pi/packages/coding-agent/examples/sdk/README.md +140 -0
  593. package/.references/pi/packages/coding-agent/install-lock/package-lock.json +1849 -0
  594. package/.references/pi/packages/coding-agent/install-lock/package.json +18 -0
  595. package/.references/pi/packages/coding-agent/npm-shrinkwrap.json +1839 -0
  596. package/.references/pi/packages/coding-agent/package.json +100 -0
  597. package/.references/pi/packages/coding-agent/scripts/migrate-sessions.sh +93 -0
  598. package/.references/pi/packages/coding-agent/src/bun/cli.ts +15 -0
  599. package/.references/pi/packages/coding-agent/src/bun/register-bedrock.ts +4 -0
  600. package/.references/pi/packages/coding-agent/src/bun/restore-sandbox-env.ts +36 -0
  601. package/.references/pi/packages/coding-agent/src/cli/args.ts +390 -0
  602. package/.references/pi/packages/coding-agent/src/cli/config-selector.ts +56 -0
  603. package/.references/pi/packages/coding-agent/src/cli/file-processor.ts +87 -0
  604. package/.references/pi/packages/coding-agent/src/cli/initial-message.ts +43 -0
  605. package/.references/pi/packages/coding-agent/src/cli/list-models.ts +111 -0
  606. package/.references/pi/packages/coding-agent/src/cli/project-trust.ts +62 -0
  607. package/.references/pi/packages/coding-agent/src/cli/session-picker.ts +55 -0
  608. package/.references/pi/packages/coding-agent/src/cli/startup-ui.ts +239 -0
  609. package/.references/pi/packages/coding-agent/src/cli.ts +20 -0
  610. package/.references/pi/packages/coding-agent/src/config.ts +566 -0
  611. package/.references/pi/packages/coding-agent/src/core/agent-session-runtime.ts +438 -0
  612. package/.references/pi/packages/coding-agent/src/core/agent-session-services.ts +207 -0
  613. package/.references/pi/packages/coding-agent/src/core/agent-session.ts +3279 -0
  614. package/.references/pi/packages/coding-agent/src/core/auth-guidance.ts +25 -0
  615. package/.references/pi/packages/coding-agent/src/core/auth-storage.ts +271 -0
  616. package/.references/pi/packages/coding-agent/src/core/bash-executor.ts +156 -0
  617. package/.references/pi/packages/coding-agent/src/core/cache-stats.ts +164 -0
  618. package/.references/pi/packages/coding-agent/src/core/compaction/branch-summarization.ts +371 -0
  619. package/.references/pi/packages/coding-agent/src/core/compaction/compaction.ts +872 -0
  620. package/.references/pi/packages/coding-agent/src/core/compaction/index.ts +7 -0
  621. package/.references/pi/packages/coding-agent/src/core/compaction/utils.ts +170 -0
  622. package/.references/pi/packages/coding-agent/src/core/defaults.ts +3 -0
  623. package/.references/pi/packages/coding-agent/src/core/diagnostics.ts +15 -0
  624. package/.references/pi/packages/coding-agent/src/core/event-bus.ts +33 -0
  625. package/.references/pi/packages/coding-agent/src/core/exec.ts +107 -0
  626. package/.references/pi/packages/coding-agent/src/core/experimental.ts +3 -0
  627. package/.references/pi/packages/coding-agent/src/core/export-html/ansi-to-html.ts +258 -0
  628. package/.references/pi/packages/coding-agent/src/core/export-html/index.ts +316 -0
  629. package/.references/pi/packages/coding-agent/src/core/export-html/template.css +1066 -0
  630. package/.references/pi/packages/coding-agent/src/core/export-html/template.html +55 -0
  631. package/.references/pi/packages/coding-agent/src/core/export-html/template.js +1864 -0
  632. package/.references/pi/packages/coding-agent/src/core/export-html/tool-renderer.ts +172 -0
  633. package/.references/pi/packages/coding-agent/src/core/export-html/vendor/highlight.min.js +1213 -0
  634. package/.references/pi/packages/coding-agent/src/core/export-html/vendor/marked.min.js +78 -0
  635. package/.references/pi/packages/coding-agent/src/core/extensions/index.ts +184 -0
  636. package/.references/pi/packages/coding-agent/src/core/extensions/loader.ts +708 -0
  637. package/.references/pi/packages/coding-agent/src/core/extensions/runner.ts +1189 -0
  638. package/.references/pi/packages/coding-agent/src/core/extensions/types.ts +1674 -0
  639. package/.references/pi/packages/coding-agent/src/core/extensions/wrapper.ts +45 -0
  640. package/.references/pi/packages/coding-agent/src/core/footer-data-provider.ts +388 -0
  641. package/.references/pi/packages/coding-agent/src/core/http-dispatcher.ts +106 -0
  642. package/.references/pi/packages/coding-agent/src/core/index.ts +80 -0
  643. package/.references/pi/packages/coding-agent/src/core/keybindings.ts +370 -0
  644. package/.references/pi/packages/coding-agent/src/core/messages.ts +195 -0
  645. package/.references/pi/packages/coding-agent/src/core/model-config.ts +287 -0
  646. package/.references/pi/packages/coding-agent/src/core/model-registry.ts +126 -0
  647. package/.references/pi/packages/coding-agent/src/core/model-resolver.ts +705 -0
  648. package/.references/pi/packages/coding-agent/src/core/model-runtime.ts +566 -0
  649. package/.references/pi/packages/coding-agent/src/core/models-store.ts +57 -0
  650. package/.references/pi/packages/coding-agent/src/core/output-guard.ts +108 -0
  651. package/.references/pi/packages/coding-agent/src/core/package-manager.ts +2650 -0
  652. package/.references/pi/packages/coding-agent/src/core/project-trust.ts +96 -0
  653. package/.references/pi/packages/coding-agent/src/core/prompt-templates.ts +284 -0
  654. package/.references/pi/packages/coding-agent/src/core/provider-attribution.ts +97 -0
  655. package/.references/pi/packages/coding-agent/src/core/provider-composer.ts +548 -0
  656. package/.references/pi/packages/coding-agent/src/core/radius.ts +1 -0
  657. package/.references/pi/packages/coding-agent/src/core/remote-catalog-provider.ts +83 -0
  658. package/.references/pi/packages/coding-agent/src/core/resolve-config-value.ts +287 -0
  659. package/.references/pi/packages/coding-agent/src/core/resource-loader.ts +1039 -0
  660. package/.references/pi/packages/coding-agent/src/core/runtime-credentials.ts +48 -0
  661. package/.references/pi/packages/coding-agent/src/core/sdk.ts +393 -0
  662. package/.references/pi/packages/coding-agent/src/core/session-cwd.ts +59 -0
  663. package/.references/pi/packages/coding-agent/src/core/session-manager.ts +1623 -0
  664. package/.references/pi/packages/coding-agent/src/core/settings-manager.ts +1234 -0
  665. package/.references/pi/packages/coding-agent/src/core/skills.ts +487 -0
  666. package/.references/pi/packages/coding-agent/src/core/slash-commands.ts +42 -0
  667. package/.references/pi/packages/coding-agent/src/core/source-info.ts +40 -0
  668. package/.references/pi/packages/coding-agent/src/core/system-prompt.ts +162 -0
  669. package/.references/pi/packages/coding-agent/src/core/telemetry.ts +13 -0
  670. package/.references/pi/packages/coding-agent/src/core/timings.ts +50 -0
  671. package/.references/pi/packages/coding-agent/src/core/tools/bash.ts +470 -0
  672. package/.references/pi/packages/coding-agent/src/core/tools/edit-diff.ts +560 -0
  673. package/.references/pi/packages/coding-agent/src/core/tools/edit.ts +437 -0
  674. package/.references/pi/packages/coding-agent/src/core/tools/file-mutation-queue.ts +61 -0
  675. package/.references/pi/packages/coding-agent/src/core/tools/find.ts +374 -0
  676. package/.references/pi/packages/coding-agent/src/core/tools/grep.ts +385 -0
  677. package/.references/pi/packages/coding-agent/src/core/tools/index.ts +196 -0
  678. package/.references/pi/packages/coding-agent/src/core/tools/ls.ts +225 -0
  679. package/.references/pi/packages/coding-agent/src/core/tools/output-accumulator.ts +222 -0
  680. package/.references/pi/packages/coding-agent/src/core/tools/path-utils.ts +118 -0
  681. package/.references/pi/packages/coding-agent/src/core/tools/read.ts +351 -0
  682. package/.references/pi/packages/coding-agent/src/core/tools/render-utils.ts +85 -0
  683. package/.references/pi/packages/coding-agent/src/core/tools/tool-definition-wrapper.ts +45 -0
  684. package/.references/pi/packages/coding-agent/src/core/tools/truncate.ts +276 -0
  685. package/.references/pi/packages/coding-agent/src/core/tools/write.ts +267 -0
  686. package/.references/pi/packages/coding-agent/src/core/trust-manager.ts +244 -0
  687. package/.references/pi/packages/coding-agent/src/index.ts +394 -0
  688. package/.references/pi/packages/coding-agent/src/main.ts +857 -0
  689. package/.references/pi/packages/coding-agent/src/migrations.ts +315 -0
  690. package/.references/pi/packages/coding-agent/src/modes/index.ts +15 -0
  691. package/.references/pi/packages/coding-agent/src/modes/interactive/assets/clankolas.png +0 -0
  692. package/.references/pi/packages/coding-agent/src/modes/interactive/components/armin.ts +382 -0
  693. package/.references/pi/packages/coding-agent/src/modes/interactive/components/assistant-message.ts +180 -0
  694. package/.references/pi/packages/coding-agent/src/modes/interactive/components/bash-execution.ts +220 -0
  695. package/.references/pi/packages/coding-agent/src/modes/interactive/components/bordered-loader.ts +68 -0
  696. package/.references/pi/packages/coding-agent/src/modes/interactive/components/branch-summary-message.ts +58 -0
  697. package/.references/pi/packages/coding-agent/src/modes/interactive/components/compaction-summary-message.ts +59 -0
  698. package/.references/pi/packages/coding-agent/src/modes/interactive/components/config-selector.ts +942 -0
  699. package/.references/pi/packages/coding-agent/src/modes/interactive/components/countdown-timer.ts +39 -0
  700. package/.references/pi/packages/coding-agent/src/modes/interactive/components/custom-editor.ts +80 -0
  701. package/.references/pi/packages/coding-agent/src/modes/interactive/components/custom-entry.ts +62 -0
  702. package/.references/pi/packages/coding-agent/src/modes/interactive/components/custom-message.ts +99 -0
  703. package/.references/pi/packages/coding-agent/src/modes/interactive/components/daxnuts.ts +164 -0
  704. package/.references/pi/packages/coding-agent/src/modes/interactive/components/diff.ts +147 -0
  705. package/.references/pi/packages/coding-agent/src/modes/interactive/components/dynamic-border.ts +25 -0
  706. package/.references/pi/packages/coding-agent/src/modes/interactive/components/earendil-announcement.ts +53 -0
  707. package/.references/pi/packages/coding-agent/src/modes/interactive/components/extension-editor.ts +167 -0
  708. package/.references/pi/packages/coding-agent/src/modes/interactive/components/extension-input.ts +87 -0
  709. package/.references/pi/packages/coding-agent/src/modes/interactive/components/extension-selector.ts +112 -0
  710. package/.references/pi/packages/coding-agent/src/modes/interactive/components/first-time-setup.ts +145 -0
  711. package/.references/pi/packages/coding-agent/src/modes/interactive/components/footer.ts +245 -0
  712. package/.references/pi/packages/coding-agent/src/modes/interactive/components/index.ts +38 -0
  713. package/.references/pi/packages/coding-agent/src/modes/interactive/components/keybinding-hints.ts +48 -0
  714. package/.references/pi/packages/coding-agent/src/modes/interactive/components/login-dialog.ts +233 -0
  715. package/.references/pi/packages/coding-agent/src/modes/interactive/components/model-selector.ts +361 -0
  716. package/.references/pi/packages/coding-agent/src/modes/interactive/components/oauth-selector.ts +214 -0
  717. package/.references/pi/packages/coding-agent/src/modes/interactive/components/scoped-models-selector.ts +360 -0
  718. package/.references/pi/packages/coding-agent/src/modes/interactive/components/session-selector-search.ts +194 -0
  719. package/.references/pi/packages/coding-agent/src/modes/interactive/components/session-selector.ts +1031 -0
  720. package/.references/pi/packages/coding-agent/src/modes/interactive/components/settings-selector.ts +838 -0
  721. package/.references/pi/packages/coding-agent/src/modes/interactive/components/show-images-selector.ts +50 -0
  722. package/.references/pi/packages/coding-agent/src/modes/interactive/components/skill-invocation-message.ts +55 -0
  723. package/.references/pi/packages/coding-agent/src/modes/interactive/components/status-indicator.ts +114 -0
  724. package/.references/pi/packages/coding-agent/src/modes/interactive/components/theme-selector.ts +67 -0
  725. package/.references/pi/packages/coding-agent/src/modes/interactive/components/thinking-selector.ts +75 -0
  726. package/.references/pi/packages/coding-agent/src/modes/interactive/components/tool-execution.ts +377 -0
  727. package/.references/pi/packages/coding-agent/src/modes/interactive/components/tree-selector.ts +1427 -0
  728. package/.references/pi/packages/coding-agent/src/modes/interactive/components/trust-selector.ts +134 -0
  729. package/.references/pi/packages/coding-agent/src/modes/interactive/components/user-message-selector.ts +155 -0
  730. package/.references/pi/packages/coding-agent/src/modes/interactive/components/user-message.ts +57 -0
  731. package/.references/pi/packages/coding-agent/src/modes/interactive/components/visual-truncate.ts +50 -0
  732. package/.references/pi/packages/coding-agent/src/modes/interactive/interactive-mode.ts +6005 -0
  733. package/.references/pi/packages/coding-agent/src/modes/interactive/model-search.ts +21 -0
  734. package/.references/pi/packages/coding-agent/src/modes/interactive/theme/dark.json +87 -0
  735. package/.references/pi/packages/coding-agent/src/modes/interactive/theme/light.json +86 -0
  736. package/.references/pi/packages/coding-agent/src/modes/interactive/theme/theme-controller.ts +126 -0
  737. package/.references/pi/packages/coding-agent/src/modes/interactive/theme/theme-schema.json +340 -0
  738. package/.references/pi/packages/coding-agent/src/modes/interactive/theme/theme.ts +1294 -0
  739. package/.references/pi/packages/coding-agent/src/modes/print-mode.ts +159 -0
  740. package/.references/pi/packages/coding-agent/src/modes/rpc/jsonl.ts +58 -0
  741. package/.references/pi/packages/coding-agent/src/modes/rpc/rpc-client.ts +592 -0
  742. package/.references/pi/packages/coding-agent/src/modes/rpc/rpc-mode.ts +795 -0
  743. package/.references/pi/packages/coding-agent/src/modes/rpc/rpc-types.ts +281 -0
  744. package/.references/pi/packages/coding-agent/src/package-manager-cli.ts +887 -0
  745. package/.references/pi/packages/coding-agent/src/rpc-entry.ts +12 -0
  746. package/.references/pi/packages/coding-agent/src/utils/ansi.ts +60 -0
  747. package/.references/pi/packages/coding-agent/src/utils/changelog.ts +196 -0
  748. package/.references/pi/packages/coding-agent/src/utils/child-process.ts +137 -0
  749. package/.references/pi/packages/coding-agent/src/utils/clipboard-image.ts +300 -0
  750. package/.references/pi/packages/coding-agent/src/utils/clipboard-native.ts +33 -0
  751. package/.references/pi/packages/coding-agent/src/utils/clipboard.ts +141 -0
  752. package/.references/pi/packages/coding-agent/src/utils/deprecation.ts +14 -0
  753. package/.references/pi/packages/coding-agent/src/utils/exif-orientation.ts +183 -0
  754. package/.references/pi/packages/coding-agent/src/utils/frontmatter.ts +39 -0
  755. package/.references/pi/packages/coding-agent/src/utils/fs-watch.ts +30 -0
  756. package/.references/pi/packages/coding-agent/src/utils/git.ts +226 -0
  757. package/.references/pi/packages/coding-agent/src/utils/highlight-js-lib-index.d.ts +19 -0
  758. package/.references/pi/packages/coding-agent/src/utils/html.ts +51 -0
  759. package/.references/pi/packages/coding-agent/src/utils/image-convert.ts +49 -0
  760. package/.references/pi/packages/coding-agent/src/utils/image-process.ts +119 -0
  761. package/.references/pi/packages/coding-agent/src/utils/image-resize-core.ts +164 -0
  762. package/.references/pi/packages/coding-agent/src/utils/image-resize-worker.ts +42 -0
  763. package/.references/pi/packages/coding-agent/src/utils/image-resize.ts +123 -0
  764. package/.references/pi/packages/coding-agent/src/utils/json.ts +6 -0
  765. package/.references/pi/packages/coding-agent/src/utils/mime.ts +116 -0
  766. package/.references/pi/packages/coding-agent/src/utils/open-browser.ts +24 -0
  767. package/.references/pi/packages/coding-agent/src/utils/paths.ts +118 -0
  768. package/.references/pi/packages/coding-agent/src/utils/photon.ts +139 -0
  769. package/.references/pi/packages/coding-agent/src/utils/pi-user-agent.ts +4 -0
  770. package/.references/pi/packages/coding-agent/src/utils/shell.ts +225 -0
  771. package/.references/pi/packages/coding-agent/src/utils/sleep.ts +18 -0
  772. package/.references/pi/packages/coding-agent/src/utils/syntax-highlight.ts +146 -0
  773. package/.references/pi/packages/coding-agent/src/utils/tools-manager.ts +369 -0
  774. package/.references/pi/packages/coding-agent/src/utils/version-check.ts +80 -0
  775. package/.references/pi/packages/coding-agent/src/utils/windows-self-update.ts +84 -0
  776. package/.references/pi/packages/coding-agent/test/agent-session-auto-compaction-queue.test.ts +450 -0
  777. package/.references/pi/packages/coding-agent/test/agent-session-branching.test.ts +156 -0
  778. package/.references/pi/packages/coding-agent/test/agent-session-compaction.test.ts +208 -0
  779. package/.references/pi/packages/coding-agent/test/agent-session-concurrent.test.ts +629 -0
  780. package/.references/pi/packages/coding-agent/test/agent-session-dynamic-provider.test.ts +122 -0
  781. package/.references/pi/packages/coding-agent/test/agent-session-dynamic-tools.test.ts +185 -0
  782. package/.references/pi/packages/coding-agent/test/agent-session-retry.test.ts +322 -0
  783. package/.references/pi/packages/coding-agent/test/agent-session-runtime-events.test.ts +257 -0
  784. package/.references/pi/packages/coding-agent/test/agent-session-stats.test.ts +169 -0
  785. package/.references/pi/packages/coding-agent/test/agent-session-tree-navigation.test.ts +323 -0
  786. package/.references/pi/packages/coding-agent/test/ansi-utils.test.ts +110 -0
  787. package/.references/pi/packages/coding-agent/test/args.test.ts +441 -0
  788. package/.references/pi/packages/coding-agent/test/assistant-message.test.ts +130 -0
  789. package/.references/pi/packages/coding-agent/test/auth-storage.test.ts +217 -0
  790. package/.references/pi/packages/coding-agent/test/bash-close-hang-windows.test.ts +126 -0
  791. package/.references/pi/packages/coding-agent/test/bash-execution-width.test.ts +80 -0
  792. package/.references/pi/packages/coding-agent/test/block-images.test.ts +148 -0
  793. package/.references/pi/packages/coding-agent/test/cache-stats.test.ts +143 -0
  794. package/.references/pi/packages/coding-agent/test/changelog.test.ts +49 -0
  795. package/.references/pi/packages/coding-agent/test/clipboard-image-bmp-conversion.test.ts +88 -0
  796. package/.references/pi/packages/coding-agent/test/clipboard-image.test.ts +184 -0
  797. package/.references/pi/packages/coding-agent/test/clipboard-native.test.ts +32 -0
  798. package/.references/pi/packages/coding-agent/test/clipboard.test.ts +166 -0
  799. package/.references/pi/packages/coding-agent/test/compaction-extensions-example.test.ts +66 -0
  800. package/.references/pi/packages/coding-agent/test/compaction-extensions.test.ts +415 -0
  801. package/.references/pi/packages/coding-agent/test/compaction-serialization.test.ts +79 -0
  802. package/.references/pi/packages/coding-agent/test/compaction-summary-reasoning.test.ts +134 -0
  803. package/.references/pi/packages/coding-agent/test/compaction.test.ts +596 -0
  804. package/.references/pi/packages/coding-agent/test/config-value-migration.test.ts +178 -0
  805. package/.references/pi/packages/coding-agent/test/config.test.ts +437 -0
  806. package/.references/pi/packages/coding-agent/test/edit-tool-legacy-input.test.ts +116 -0
  807. package/.references/pi/packages/coding-agent/test/edit-tool-no-full-redraw.test.ts +235 -0
  808. package/.references/pi/packages/coding-agent/test/experimental.test.ts +44 -0
  809. package/.references/pi/packages/coding-agent/test/export-html-skill-block.test.ts +40 -0
  810. package/.references/pi/packages/coding-agent/test/export-html-whitespace.test.ts +42 -0
  811. package/.references/pi/packages/coding-agent/test/export-html-xss.test.ts +67 -0
  812. package/.references/pi/packages/coding-agent/test/extensions-discovery.test.ts +528 -0
  813. package/.references/pi/packages/coding-agent/test/extensions-input-event.test.ts +125 -0
  814. package/.references/pi/packages/coding-agent/test/extensions-runner.test.ts +986 -0
  815. package/.references/pi/packages/coding-agent/test/file-mutation-queue.test.ts +274 -0
  816. package/.references/pi/packages/coding-agent/test/first-time-setup-fork.test.ts +39 -0
  817. package/.references/pi/packages/coding-agent/test/first-time-setup.test.ts +95 -0
  818. package/.references/pi/packages/coding-agent/test/fixtures/assistant-message-with-thinking-code.json +33 -0
  819. package/.references/pi/packages/coding-agent/test/fixtures/before-compaction.jsonl +1003 -0
  820. package/.references/pi/packages/coding-agent/test/fixtures/empty-agent/.gitkeep +0 -0
  821. package/.references/pi/packages/coding-agent/test/fixtures/empty-cwd/.gitkeep +0 -0
  822. package/.references/pi/packages/coding-agent/test/fixtures/large-session.jsonl +1019 -0
  823. package/.references/pi/packages/coding-agent/test/fixtures/skills/consecutive-hyphens/SKILL.md +8 -0
  824. package/.references/pi/packages/coding-agent/test/fixtures/skills/disable-model-invocation/SKILL.md +9 -0
  825. package/.references/pi/packages/coding-agent/test/fixtures/skills/invalid-name-chars/SKILL.md +8 -0
  826. package/.references/pi/packages/coding-agent/test/fixtures/skills/invalid-yaml/SKILL.md +8 -0
  827. package/.references/pi/packages/coding-agent/test/fixtures/skills/long-name/SKILL.md +8 -0
  828. package/.references/pi/packages/coding-agent/test/fixtures/skills/missing-description/SKILL.md +7 -0
  829. package/.references/pi/packages/coding-agent/test/fixtures/skills/multiline-description/SKILL.md +11 -0
  830. package/.references/pi/packages/coding-agent/test/fixtures/skills/name-mismatch/SKILL.md +8 -0
  831. package/.references/pi/packages/coding-agent/test/fixtures/skills/nested/child-skill/SKILL.md +8 -0
  832. package/.references/pi/packages/coding-agent/test/fixtures/skills/no-frontmatter/SKILL.md +3 -0
  833. package/.references/pi/packages/coding-agent/test/fixtures/skills/root-skill-preferred/SKILL.md +3 -0
  834. package/.references/pi/packages/coding-agent/test/fixtures/skills/root-skill-preferred/nested-child/SKILL.md +3 -0
  835. package/.references/pi/packages/coding-agent/test/fixtures/skills/unknown-field/SKILL.md +10 -0
  836. package/.references/pi/packages/coding-agent/test/fixtures/skills/valid-skill/SKILL.md +8 -0
  837. package/.references/pi/packages/coding-agent/test/fixtures/skills-collision/first/calendar/SKILL.md +8 -0
  838. package/.references/pi/packages/coding-agent/test/fixtures/skills-collision/second/calendar/SKILL.md +8 -0
  839. package/.references/pi/packages/coding-agent/test/footer-data-provider.test.ts +265 -0
  840. package/.references/pi/packages/coding-agent/test/footer-width.test.ts +144 -0
  841. package/.references/pi/packages/coding-agent/test/format-resume-command.test.ts +135 -0
  842. package/.references/pi/packages/coding-agent/test/frontmatter.test.ts +60 -0
  843. package/.references/pi/packages/coding-agent/test/git-merge-and-resolve-extension.test.ts +206 -0
  844. package/.references/pi/packages/coding-agent/test/git-ssh-url.test.ts +91 -0
  845. package/.references/pi/packages/coding-agent/test/git-update.test.ts +484 -0
  846. package/.references/pi/packages/coding-agent/test/http-dispatcher.test.ts +53 -0
  847. package/.references/pi/packages/coding-agent/test/image-process.test.ts +53 -0
  848. package/.references/pi/packages/coding-agent/test/image-processing.test.ts +170 -0
  849. package/.references/pi/packages/coding-agent/test/image-resize-callers.test.ts +53 -0
  850. package/.references/pi/packages/coding-agent/test/initial-message.test.ts +48 -0
  851. package/.references/pi/packages/coding-agent/test/input-transform-streaming-example.test.ts +84 -0
  852. package/.references/pi/packages/coding-agent/test/interactive-mode-anthropic-warning.test.ts +89 -0
  853. package/.references/pi/packages/coding-agent/test/interactive-mode-clone-command.test.ts +72 -0
  854. package/.references/pi/packages/coding-agent/test/interactive-mode-compaction.test.ts +85 -0
  855. package/.references/pi/packages/coding-agent/test/interactive-mode-import-command.test.ts +143 -0
  856. package/.references/pi/packages/coding-agent/test/interactive-mode-startup-input.test.ts +72 -0
  857. package/.references/pi/packages/coding-agent/test/interactive-mode-status.test.ts +1123 -0
  858. package/.references/pi/packages/coding-agent/test/interactive-mode-suspend.test.ts +149 -0
  859. package/.references/pi/packages/coding-agent/test/keybindings-migration.test.ts +88 -0
  860. package/.references/pi/packages/coding-agent/test/max-thinking.test.ts +45 -0
  861. package/.references/pi/packages/coding-agent/test/model-registry.test.ts +1934 -0
  862. package/.references/pi/packages/coding-agent/test/model-resolver.test.ts +690 -0
  863. package/.references/pi/packages/coding-agent/test/model-runtime-auth-options.test.ts +256 -0
  864. package/.references/pi/packages/coding-agent/test/model-runtime-cloudflare-compat.test.ts +95 -0
  865. package/.references/pi/packages/coding-agent/test/model-runtime-modify-models-compat.test.ts +85 -0
  866. package/.references/pi/packages/coding-agent/test/model-runtime-test-utils.ts +25 -0
  867. package/.references/pi/packages/coding-agent/test/models-store.test.ts +51 -0
  868. package/.references/pi/packages/coding-agent/test/oauth-selector.test.ts +147 -0
  869. package/.references/pi/packages/coding-agent/test/package-command-paths.test.ts +746 -0
  870. package/.references/pi/packages/coding-agent/test/package-manager-ssh.test.ts +97 -0
  871. package/.references/pi/packages/coding-agent/test/package-manager.test.ts +2512 -0
  872. package/.references/pi/packages/coding-agent/test/path-utils.test.ts +174 -0
  873. package/.references/pi/packages/coding-agent/test/paths.test.ts +150 -0
  874. package/.references/pi/packages/coding-agent/test/pi-user-agent.test.ts +12 -0
  875. package/.references/pi/packages/coding-agent/test/plan-mode-extension.test.ts +167 -0
  876. package/.references/pi/packages/coding-agent/test/plan-mode-utils.test.ts +261 -0
  877. package/.references/pi/packages/coding-agent/test/print-mode.test.ts +142 -0
  878. package/.references/pi/packages/coding-agent/test/prompt-templates.test.ts +615 -0
  879. package/.references/pi/packages/coding-agent/test/radius.test.ts +151 -0
  880. package/.references/pi/packages/coding-agent/test/remote-catalog-provider.test.ts +118 -0
  881. package/.references/pi/packages/coding-agent/test/resolve-config-value.test.ts +121 -0
  882. package/.references/pi/packages/coding-agent/test/resource-loader.test.ts +739 -0
  883. package/.references/pi/packages/coding-agent/test/restore-sandbox-env.test.ts +77 -0
  884. package/.references/pi/packages/coding-agent/test/rpc-client-clone.test.ts +29 -0
  885. package/.references/pi/packages/coding-agent/test/rpc-client-process-exit.test.ts +38 -0
  886. package/.references/pi/packages/coding-agent/test/rpc-example.ts +86 -0
  887. package/.references/pi/packages/coding-agent/test/rpc-jsonl.test.ts +65 -0
  888. package/.references/pi/packages/coding-agent/test/rpc-prompt-response-semantics.test.ts +288 -0
  889. package/.references/pi/packages/coding-agent/test/rpc.test.ts +377 -0
  890. package/.references/pi/packages/coding-agent/test/runtime-credentials.test.ts +42 -0
  891. package/.references/pi/packages/coding-agent/test/sdk-codex-cache-probe-tool-loop.ts +497 -0
  892. package/.references/pi/packages/coding-agent/test/sdk-openrouter-attribution.test.ts +266 -0
  893. package/.references/pi/packages/coding-agent/test/sdk-session-manager.test.ts +95 -0
  894. package/.references/pi/packages/coding-agent/test/sdk-skills.test.ts +109 -0
  895. package/.references/pi/packages/coding-agent/test/sdk-stream-options.test.ts +188 -0
  896. package/.references/pi/packages/coding-agent/test/session-cwd.test.ts +91 -0
  897. package/.references/pi/packages/coding-agent/test/session-file-invalid.test.ts +65 -0
  898. package/.references/pi/packages/coding-agent/test/session-id-readonly.test.ts +190 -0
  899. package/.references/pi/packages/coding-agent/test/session-info-modified-timestamp.test.ts +83 -0
  900. package/.references/pi/packages/coding-agent/test/session-manager/build-context.test.ts +305 -0
  901. package/.references/pi/packages/coding-agent/test/session-manager/custom-session-id.test.ts +169 -0
  902. package/.references/pi/packages/coding-agent/test/session-manager/file-operations.test.ts +315 -0
  903. package/.references/pi/packages/coding-agent/test/session-manager/labels.test.ts +211 -0
  904. package/.references/pi/packages/coding-agent/test/session-manager/migration.test.ts +78 -0
  905. package/.references/pi/packages/coding-agent/test/session-manager/save-entry.test.ts +55 -0
  906. package/.references/pi/packages/coding-agent/test/session-manager/tree-traversal.test.ts +533 -0
  907. package/.references/pi/packages/coding-agent/test/session-selector-path-delete.test.ts +354 -0
  908. package/.references/pi/packages/coding-agent/test/session-selector-rename.test.ts +111 -0
  909. package/.references/pi/packages/coding-agent/test/session-selector-search.test.ts +195 -0
  910. package/.references/pi/packages/coding-agent/test/settings-manager-bug.test.ts +147 -0
  911. package/.references/pi/packages/coding-agent/test/settings-manager.test.ts +511 -0
  912. package/.references/pi/packages/coding-agent/test/skills.test.ts +432 -0
  913. package/.references/pi/packages/coding-agent/test/startup-session-name.test.ts +135 -0
  914. package/.references/pi/packages/coding-agent/test/status-indicator.test.ts +32 -0
  915. package/.references/pi/packages/coding-agent/test/stdout-cleanliness.test.ts +128 -0
  916. package/.references/pi/packages/coding-agent/test/streaming-render-debug.ts +97 -0
  917. package/.references/pi/packages/coding-agent/test/suite/README.md +16 -0
  918. package/.references/pi/packages/coding-agent/test/suite/agent-session-bash-persistence.test.ts +242 -0
  919. package/.references/pi/packages/coding-agent/test/suite/agent-session-compaction.test.ts +446 -0
  920. package/.references/pi/packages/coding-agent/test/suite/agent-session-model-extension.test.ts +373 -0
  921. package/.references/pi/packages/coding-agent/test/suite/agent-session-prompt.test.ts +398 -0
  922. package/.references/pi/packages/coding-agent/test/suite/agent-session-queue.test.ts +445 -0
  923. package/.references/pi/packages/coding-agent/test/suite/agent-session-retry-events.test.ts +364 -0
  924. package/.references/pi/packages/coding-agent/test/suite/agent-session-runtime.test.ts +609 -0
  925. package/.references/pi/packages/coding-agent/test/suite/harness.ts +219 -0
  926. package/.references/pi/packages/coding-agent/test/suite/lax-message-content.test.ts +162 -0
  927. package/.references/pi/packages/coding-agent/test/suite/regressions/1717-2113-agent-session-event-settlement.test.ts +95 -0
  928. package/.references/pi/packages/coding-agent/test/suite/regressions/2023-queued-slash-command-followup.test.ts +80 -0
  929. package/.references/pi/packages/coding-agent/test/suite/regressions/2753-reload-stale-resource-settings.test.ts +105 -0
  930. package/.references/pi/packages/coding-agent/test/suite/regressions/2781-skill-collision-precedence.test.ts +120 -0
  931. package/.references/pi/packages/coding-agent/test/suite/regressions/2791-fswatch-error-crash.test.ts +106 -0
  932. package/.references/pi/packages/coding-agent/test/suite/regressions/2835-tools-allowlist-filters-extension-tools.test.ts +94 -0
  933. package/.references/pi/packages/coding-agent/test/suite/regressions/2860-replaced-session-context.test.ts +279 -0
  934. package/.references/pi/packages/coding-agent/test/suite/regressions/3217-scoped-model-order.test.ts +104 -0
  935. package/.references/pi/packages/coding-agent/test/suite/regressions/3302-find-path-glob.test.ts +72 -0
  936. package/.references/pi/packages/coding-agent/test/suite/regressions/3303-find-nested-gitignore.test.ts +83 -0
  937. package/.references/pi/packages/coding-agent/test/suite/regressions/3317-network-connection-lost-retry.test.ts +33 -0
  938. package/.references/pi/packages/coding-agent/test/suite/regressions/3592-no-builtin-tools-keeps-extension-tools.test.ts +119 -0
  939. package/.references/pi/packages/coding-agent/test/suite/regressions/3616-settings-inmemory-reload.test.ts +86 -0
  940. package/.references/pi/packages/coding-agent/test/suite/regressions/3686-session-name-event.test.ts +61 -0
  941. package/.references/pi/packages/coding-agent/test/suite/regressions/3688-tree-cancel-compacting.test.ts +36 -0
  942. package/.references/pi/packages/coding-agent/test/suite/regressions/3982-message-end-cost-override.test.ts +56 -0
  943. package/.references/pi/packages/coding-agent/test/suite/regressions/4167-thinking-toggle-pending-tool-render.test.ts +182 -0
  944. package/.references/pi/packages/coding-agent/test/suite/regressions/5080-signal-shutdown-extension-cleanup.test.ts +185 -0
  945. package/.references/pi/packages/coding-agent/test/suite/regressions/5109-exclude-tools.test.ts +81 -0
  946. package/.references/pi/packages/coding-agent/test/suite/regressions/5208-late-bash-output.test.ts +29 -0
  947. package/.references/pi/packages/coding-agent/test/suite/regressions/5217-compaction-reason.test.ts +95 -0
  948. package/.references/pi/packages/coding-agent/test/suite/regressions/5303-bash-output-truncation.test.ts +79 -0
  949. package/.references/pi/packages/coding-agent/test/suite/regressions/5433-extension-oauth-prompt-input.test.ts +119 -0
  950. package/.references/pi/packages/coding-agent/test/suite/regressions/5596-missing-theme-export.test.ts +89 -0
  951. package/.references/pi/packages/coding-agent/test/suite/regressions/5661-uppercase-header-values.test.ts +91 -0
  952. package/.references/pi/packages/coding-agent/test/suite/regressions/5724-sigterm-signal-exit.test.ts +94 -0
  953. package/.references/pi/packages/coding-agent/test/suite/regressions/5868-rpc-unknown-command-id.test.ts +113 -0
  954. package/.references/pi/packages/coding-agent/test/suite/regressions/5943-session-start-notify.test.ts +515 -0
  955. package/.references/pi/packages/coding-agent/test/suite/regressions/5996-session-name-newlines.test.ts +40 -0
  956. package/.references/pi/packages/coding-agent/test/suite/regressions/6019-explicit-provider-retry-message.test.ts +31 -0
  957. package/.references/pi/packages/coding-agent/test/suite/regressions/6162-extension-active-tools-next-turn.test.ts +192 -0
  958. package/.references/pi/packages/coding-agent/test/suite/regressions/6260-inline-extension-naming.test.ts +99 -0
  959. package/.references/pi/packages/coding-agent/test/suite/regressions/6324-branch-summary-ambient-auth.test.ts +61 -0
  960. package/.references/pi/packages/coding-agent/test/suite/regressions/6363-agent-settled-event.test.ts +158 -0
  961. package/.references/pi/packages/coding-agent/test/suite/regressions/extension-factory-cache.test.ts +131 -0
  962. package/.references/pi/packages/coding-agent/test/suite/regressions/pre-prompt-compaction-no-continue.test.ts +75 -0
  963. package/.references/pi/packages/coding-agent/test/syntax-highlight.test.ts +76 -0
  964. package/.references/pi/packages/coding-agent/test/system-prompt.test.ts +114 -0
  965. package/.references/pi/packages/coding-agent/test/test-harness.test.ts +321 -0
  966. package/.references/pi/packages/coding-agent/test/test-harness.ts +463 -0
  967. package/.references/pi/packages/coding-agent/test/test-theme-colors.ts +246 -0
  968. package/.references/pi/packages/coding-agent/test/theme-detection.test.ts +133 -0
  969. package/.references/pi/packages/coding-agent/test/theme-export.test.ts +104 -0
  970. package/.references/pi/packages/coding-agent/test/theme-picker.test.ts +51 -0
  971. package/.references/pi/packages/coding-agent/test/tool-execution-component.test.ts +529 -0
  972. package/.references/pi/packages/coding-agent/test/tools.test.ts +1213 -0
  973. package/.references/pi/packages/coding-agent/test/tree-selector.test.ts +702 -0
  974. package/.references/pi/packages/coding-agent/test/trigger-compact-extension.test.ts +59 -0
  975. package/.references/pi/packages/coding-agent/test/truncate-to-width.test.ts +81 -0
  976. package/.references/pi/packages/coding-agent/test/trust-manager.test.ts +67 -0
  977. package/.references/pi/packages/coding-agent/test/trust-selector.test.ts +87 -0
  978. package/.references/pi/packages/coding-agent/test/user-message.test.ts +25 -0
  979. package/.references/pi/packages/coding-agent/test/utilities.ts +318 -0
  980. package/.references/pi/packages/coding-agent/test/version-check.test.ts +91 -0
  981. package/.references/pi/packages/coding-agent/tsconfig.build.json +17 -0
  982. package/.references/pi/packages/coding-agent/tsconfig.examples.json +16 -0
  983. package/.references/pi/packages/coding-agent/vitest.config.ts +38 -0
  984. package/.references/pi/packages/orchestrator/CHANGELOG.md +19 -0
  985. package/.references/pi/packages/orchestrator/README.md +11 -0
  986. package/.references/pi/packages/orchestrator/package.json +45 -0
  987. package/.references/pi/packages/orchestrator/src/cli.ts +161 -0
  988. package/.references/pi/packages/orchestrator/src/config.ts +69 -0
  989. package/.references/pi/packages/orchestrator/src/handler.ts +161 -0
  990. package/.references/pi/packages/orchestrator/src/index.ts +10 -0
  991. package/.references/pi/packages/orchestrator/src/ipc/client.ts +63 -0
  992. package/.references/pi/packages/orchestrator/src/ipc/protocol.ts +142 -0
  993. package/.references/pi/packages/orchestrator/src/ipc/server.ts +209 -0
  994. package/.references/pi/packages/orchestrator/src/radius.ts +440 -0
  995. package/.references/pi/packages/orchestrator/src/rpc-process.ts +201 -0
  996. package/.references/pi/packages/orchestrator/src/serve.ts +77 -0
  997. package/.references/pi/packages/orchestrator/src/storage.ts +70 -0
  998. package/.references/pi/packages/orchestrator/src/supervisor.ts +354 -0
  999. package/.references/pi/packages/orchestrator/src/types.ts +25 -0
  1000. package/.references/pi/packages/orchestrator/tsconfig.build.json +9 -0
  1001. package/.references/pi/packages/tui/CHANGELOG.md +1005 -0
  1002. package/.references/pi/packages/tui/README.md +791 -0
  1003. package/.references/pi/packages/tui/native/darwin/prebuilds/darwin-arm64/darwin-modifiers.node +0 -0
  1004. package/.references/pi/packages/tui/native/darwin/prebuilds/darwin-x64/darwin-modifiers.node +0 -0
  1005. package/.references/pi/packages/tui/native/darwin/src/darwin-modifiers.c +70 -0
  1006. package/.references/pi/packages/tui/native/win32/prebuilds/win32-arm64/win32-console-mode.node +0 -0
  1007. package/.references/pi/packages/tui/native/win32/prebuilds/win32-x64/win32-console-mode.node +0 -0
  1008. package/.references/pi/packages/tui/native/win32/src/win32-console-mode.c +53 -0
  1009. package/.references/pi/packages/tui/package.json +47 -0
  1010. package/.references/pi/packages/tui/src/autocomplete.ts +786 -0
  1011. package/.references/pi/packages/tui/src/components/box.ts +137 -0
  1012. package/.references/pi/packages/tui/src/components/cancellable-loader.ts +40 -0
  1013. package/.references/pi/packages/tui/src/components/editor.ts +2333 -0
  1014. package/.references/pi/packages/tui/src/components/image.ts +126 -0
  1015. package/.references/pi/packages/tui/src/components/input.ts +447 -0
  1016. package/.references/pi/packages/tui/src/components/loader.ts +92 -0
  1017. package/.references/pi/packages/tui/src/components/markdown.ts +858 -0
  1018. package/.references/pi/packages/tui/src/components/select-list.ts +229 -0
  1019. package/.references/pi/packages/tui/src/components/settings-list.ts +250 -0
  1020. package/.references/pi/packages/tui/src/components/spacer.ts +28 -0
  1021. package/.references/pi/packages/tui/src/components/text.ts +106 -0
  1022. package/.references/pi/packages/tui/src/components/truncated-text.ts +65 -0
  1023. package/.references/pi/packages/tui/src/editor-component.ts +74 -0
  1024. package/.references/pi/packages/tui/src/fuzzy.ts +137 -0
  1025. package/.references/pi/packages/tui/src/index.ts +114 -0
  1026. package/.references/pi/packages/tui/src/keybindings.ts +244 -0
  1027. package/.references/pi/packages/tui/src/keys.ts +1401 -0
  1028. package/.references/pi/packages/tui/src/kill-ring.ts +46 -0
  1029. package/.references/pi/packages/tui/src/native-modifiers.ts +59 -0
  1030. package/.references/pi/packages/tui/src/stdin-buffer.ts +434 -0
  1031. package/.references/pi/packages/tui/src/terminal-colors.ts +73 -0
  1032. package/.references/pi/packages/tui/src/terminal-image.ts +488 -0
  1033. package/.references/pi/packages/tui/src/terminal.ts +531 -0
  1034. package/.references/pi/packages/tui/src/tui.ts +1714 -0
  1035. package/.references/pi/packages/tui/src/undo-stack.ts +28 -0
  1036. package/.references/pi/packages/tui/src/utils.ts +1209 -0
  1037. package/.references/pi/packages/tui/src/word-navigation.ts +117 -0
  1038. package/.references/pi/packages/tui/test/autocomplete.test.ts +542 -0
  1039. package/.references/pi/packages/tui/test/bug-regression-isimageline-startswith-bug.test.ts +237 -0
  1040. package/.references/pi/packages/tui/test/chat-simple.ts +129 -0
  1041. package/.references/pi/packages/tui/test/editor.test.ts +4051 -0
  1042. package/.references/pi/packages/tui/test/fuzzy.test.ts +112 -0
  1043. package/.references/pi/packages/tui/test/image-test.ts +56 -0
  1044. package/.references/pi/packages/tui/test/input.test.ts +647 -0
  1045. package/.references/pi/packages/tui/test/key-tester.ts +123 -0
  1046. package/.references/pi/packages/tui/test/keybindings.test.ts +46 -0
  1047. package/.references/pi/packages/tui/test/keys.test.ts +622 -0
  1048. package/.references/pi/packages/tui/test/markdown.test.ts +1442 -0
  1049. package/.references/pi/packages/tui/test/overlay-non-capturing.test.ts +1202 -0
  1050. package/.references/pi/packages/tui/test/overlay-options.test.ts +541 -0
  1051. package/.references/pi/packages/tui/test/overlay-short-content.test.ts +61 -0
  1052. package/.references/pi/packages/tui/test/regression-overlay-cjk-boundary.test.ts +68 -0
  1053. package/.references/pi/packages/tui/test/regression-regional-indicator-width.test.ts +52 -0
  1054. package/.references/pi/packages/tui/test/select-list.test.ts +116 -0
  1055. package/.references/pi/packages/tui/test/stdin-buffer.test.ts +458 -0
  1056. package/.references/pi/packages/tui/test/tab-width.test.ts +87 -0
  1057. package/.references/pi/packages/tui/test/terminal-colors.test.ts +249 -0
  1058. package/.references/pi/packages/tui/test/terminal-image.test.ts +492 -0
  1059. package/.references/pi/packages/tui/test/terminal.test.ts +233 -0
  1060. package/.references/pi/packages/tui/test/test-themes.ts +38 -0
  1061. package/.references/pi/packages/tui/test/truncate-to-width.test.ts +76 -0
  1062. package/.references/pi/packages/tui/test/truncated-text.test.ts +129 -0
  1063. package/.references/pi/packages/tui/test/tui-cell-size-input.test.ts +81 -0
  1064. package/.references/pi/packages/tui/test/tui-overlay-style-leak.test.ts +80 -0
  1065. package/.references/pi/packages/tui/test/tui-render.test.ts +767 -0
  1066. package/.references/pi/packages/tui/test/tui-shrink.test.ts +44 -0
  1067. package/.references/pi/packages/tui/test/viewport-overwrite-repro.ts +106 -0
  1068. package/.references/pi/packages/tui/test/virtual-terminal.ts +218 -0
  1069. package/.references/pi/packages/tui/test/word-navigation.test.ts +191 -0
  1070. package/.references/pi/packages/tui/test/wrap-ansi.test.ts +266 -0
  1071. package/.references/pi/packages/tui/tsconfig.build.json +9 -0
  1072. package/.references/pi/packages/tui/vitest.config.ts +7 -0
  1073. package/.references/pi/pi-test.bat +14 -0
  1074. package/.references/pi/pi-test.ps1 +71 -0
  1075. package/.references/pi/pi-test.sh +57 -0
  1076. package/.references/pi/scripts/browser-smoke-entry.ts +61 -0
  1077. package/.references/pi/scripts/build-binaries.sh +245 -0
  1078. package/.references/pi/scripts/check-browser-smoke.mjs +36 -0
  1079. package/.references/pi/scripts/check-lockfile-commit.mjs +120 -0
  1080. package/.references/pi/scripts/check-pinned-deps.mjs +63 -0
  1081. package/.references/pi/scripts/check-ts-relative-imports.mjs +74 -0
  1082. package/.references/pi/scripts/cost.ts +183 -0
  1083. package/.references/pi/scripts/edit-tool-stats.mjs +834 -0
  1084. package/.references/pi/scripts/generate-coding-agent-install-lock.mjs +439 -0
  1085. package/.references/pi/scripts/generate-coding-agent-shrinkwrap.mjs +365 -0
  1086. package/.references/pi/scripts/local-release.mjs +284 -0
  1087. package/.references/pi/scripts/profile-coding-agent-node.mjs +639 -0
  1088. package/.references/pi/scripts/publish-model-catalog.mjs +295 -0
  1089. package/.references/pi/scripts/publish.mjs +125 -0
  1090. package/.references/pi/scripts/read-tool-stats.mjs +505 -0
  1091. package/.references/pi/scripts/release-notes.mjs +364 -0
  1092. package/.references/pi/scripts/release.mjs +207 -0
  1093. package/.references/pi/scripts/repro-5893-wsl-bash.mjs +55 -0
  1094. package/.references/pi/scripts/session-context-stats.mjs +405 -0
  1095. package/.references/pi/scripts/session-transcripts.ts +406 -0
  1096. package/.references/pi/scripts/stats.ts +233 -0
  1097. package/.references/pi/scripts/sync-versions.js +96 -0
  1098. package/.references/pi/scripts/tool-stats.ts +232 -0
  1099. package/.references/pi/scripts/update-source-imports-to-ts.sh +10 -0
  1100. package/.references/pi/test.sh +81 -0
  1101. package/.references/pi/tsconfig.base.json +25 -0
  1102. package/.references/pi/tsconfig.json +27 -0
  1103. package/AGENTS.md +8 -0
  1104. package/Cargo.lock +5169 -0
  1105. package/Cargo.toml +39 -0
  1106. package/LICENSE +21 -0
  1107. package/README.md +43 -0
  1108. package/aliases/pi-rusty/Cargo.toml +19 -0
  1109. package/aliases/pi-rusty/LICENSE +21 -0
  1110. package/aliases/pi-rusty/README.md +10 -0
  1111. package/aliases/pi-rusty/src/main.rs +8 -0
  1112. package/bun.lock +687 -0
  1113. package/crates/pi/Cargo.toml +66 -0
  1114. package/crates/pi/LICENSE +21 -0
  1115. package/crates/pi/assets/export-html/template.css +1066 -0
  1116. package/crates/pi/assets/export-html/template.html +55 -0
  1117. package/crates/pi/assets/export-html/template.js +1864 -0
  1118. package/crates/pi/assets/export-html/vendor/dark.json +29 -0
  1119. package/crates/pi/assets/export-html/vendor/highlight.min.js +1213 -0
  1120. package/crates/pi/assets/export-html/vendor/light.json +6 -0
  1121. package/crates/pi/assets/export-html/vendor/marked.min.js +78 -0
  1122. package/crates/pi/src/cli/args.rs +889 -0
  1123. package/crates/pi/src/cli/bootstrap.rs +2134 -0
  1124. package/crates/pi/src/cli/entry.rs +1332 -0
  1125. package/crates/pi/src/cli/help.rs +326 -0
  1126. package/crates/pi/src/cli/mod.rs +22 -0
  1127. package/crates/pi/src/cli/package_manager_cli.rs +1515 -0
  1128. package/crates/pi/src/core/agent_session/bash.rs +453 -0
  1129. package/crates/pi/src/core/agent_session/compaction.rs +1804 -0
  1130. package/crates/pi/src/core/agent_session/events.rs +621 -0
  1131. package/crates/pi/src/core/agent_session/extension.rs +2146 -0
  1132. package/crates/pi/src/core/agent_session/extension_runner.rs +471 -0
  1133. package/crates/pi/src/core/agent_session/mod.rs +1768 -0
  1134. package/crates/pi/src/core/agent_session/model.rs +775 -0
  1135. package/crates/pi/src/core/agent_session/persistence.rs +265 -0
  1136. package/crates/pi/src/core/agent_session/prompt.rs +2289 -0
  1137. package/crates/pi/src/core/agent_session/retry.rs +448 -0
  1138. package/crates/pi/src/core/agent_session/stats.rs +302 -0
  1139. package/crates/pi/src/core/agent_session/subscribe.rs +450 -0
  1140. package/crates/pi/src/core/agent_session/tools.rs +515 -0
  1141. package/crates/pi/src/core/agent_session/tree.rs +1219 -0
  1142. package/crates/pi/src/core/agent_session_runtime.rs +1612 -0
  1143. package/crates/pi/src/core/agent_session_services.rs +1613 -0
  1144. package/crates/pi/src/core/compaction/branch.rs +767 -0
  1145. package/crates/pi/src/core/compaction/mod.rs +1955 -0
  1146. package/crates/pi/src/core/compaction/utils.rs +357 -0
  1147. package/crates/pi/src/core/config.rs +1102 -0
  1148. package/crates/pi/src/core/config_value.rs +179 -0
  1149. package/crates/pi/src/core/experimental.rs +29 -0
  1150. package/crates/pi/src/core/export_html/ansi_to_html.rs +279 -0
  1151. package/crates/pi/src/core/export_html/mod.rs +787 -0
  1152. package/crates/pi/src/core/export_html/session_data.rs +149 -0
  1153. package/crates/pi/src/core/extension_host/acquire/tests.rs +579 -0
  1154. package/crates/pi/src/core/extension_host/acquire.rs +431 -0
  1155. package/crates/pi/src/core/extension_host/tests.rs +2353 -0
  1156. package/crates/pi/src/core/extension_host.rs +2541 -0
  1157. package/crates/pi/src/core/lockfile.rs +913 -0
  1158. package/crates/pi/src/core/messages.rs +706 -0
  1159. package/crates/pi/src/core/migrations.rs +828 -0
  1160. package/crates/pi/src/core/mod.rs +28 -0
  1161. package/crates/pi/src/core/model_resolver.rs +1473 -0
  1162. package/crates/pi/src/core/model_runtime.rs +2720 -0
  1163. package/crates/pi/src/core/output_guard.rs +977 -0
  1164. package/crates/pi/src/core/package_manager/tests.rs +1304 -0
  1165. package/crates/pi/src/core/package_manager.rs +2015 -0
  1166. package/crates/pi/src/core/platform/clipboard.rs +702 -0
  1167. package/crates/pi/src/core/platform/command.rs +223 -0
  1168. package/crates/pi/src/core/platform/debug_dump.rs +274 -0
  1169. package/crates/pi/src/core/platform/external_editor.rs +294 -0
  1170. package/crates/pi/src/core/platform/first_run.rs +120 -0
  1171. package/crates/pi/src/core/platform/image.rs +237 -0
  1172. package/crates/pi/src/core/platform/mod.rs +27 -0
  1173. package/crates/pi/src/core/platform/open_browser.rs +187 -0
  1174. package/crates/pi/src/core/platform/release_packaging.rs +785 -0
  1175. package/crates/pi/src/core/resources/diagnostics.rs +148 -0
  1176. package/crates/pi/src/core/resources/discovery.rs +2570 -0
  1177. package/crates/pi/src/core/resources/frontmatter.rs +205 -0
  1178. package/crates/pi/src/core/resources/mod.rs +1402 -0
  1179. package/crates/pi/src/core/resources/prompts.rs +510 -0
  1180. package/crates/pi/src/core/resources/skills.rs +981 -0
  1181. package/crates/pi/src/core/resources/slash.rs +199 -0
  1182. package/crates/pi/src/core/resources/source_info.rs +141 -0
  1183. package/crates/pi/src/core/resources/themes.rs +297 -0
  1184. package/crates/pi/src/core/session_transfer.rs +390 -0
  1185. package/crates/pi/src/core/sessions/context.rs +657 -0
  1186. package/crates/pi/src/core/sessions/cwd.rs +176 -0
  1187. package/crates/pi/src/core/sessions/entries.rs +1364 -0
  1188. package/crates/pi/src/core/sessions/list.rs +610 -0
  1189. package/crates/pi/src/core/sessions/mod.rs +2117 -0
  1190. package/crates/pi/src/core/settings.rs +3794 -0
  1191. package/crates/pi/src/core/share.rs +577 -0
  1192. package/crates/pi/src/core/system_prompt.rs +654 -0
  1193. package/crates/pi/src/core/tools/bash.rs +1574 -0
  1194. package/crates/pi/src/core/tools/edit.rs +919 -0
  1195. package/crates/pi/src/core/tools/edit_diff.rs +1150 -0
  1196. package/crates/pi/src/core/tools/find.rs +652 -0
  1197. package/crates/pi/src/core/tools/grep.rs +965 -0
  1198. package/crates/pi/src/core/tools/ls.rs +545 -0
  1199. package/crates/pi/src/core/tools/mod.rs +312 -0
  1200. package/crates/pi/src/core/tools/mutation_queue.rs +392 -0
  1201. package/crates/pi/src/core/tools/output_accumulator.rs +689 -0
  1202. package/crates/pi/src/core/tools/path_utils.rs +517 -0
  1203. package/crates/pi/src/core/tools/read.rs +1413 -0
  1204. package/crates/pi/src/core/tools/truncate.rs +644 -0
  1205. package/crates/pi/src/core/tools/write.rs +524 -0
  1206. package/crates/pi/src/core/trust.rs +1414 -0
  1207. package/crates/pi/src/core/update/changelog.rs +583 -0
  1208. package/crates/pi/src/core/update/deprecation.rs +148 -0
  1209. package/crates/pi/src/core/update/mod.rs +8 -0
  1210. package/crates/pi/src/core/update/self_update.rs +1327 -0
  1211. package/crates/pi/src/core/update/user_agent.rs +59 -0
  1212. package/crates/pi/src/core/update/version_check.rs +511 -0
  1213. package/crates/pi/src/core/update/windows_quarantine.rs +262 -0
  1214. package/crates/pi/src/lib.rs +23 -0
  1215. package/crates/pi/src/main.rs +13 -0
  1216. package/crates/pi/src/modes/interactive/footer.rs +260 -0
  1217. package/crates/pi/src/modes/interactive/header.rs +83 -0
  1218. package/crates/pi/src/modes/interactive/input.rs +923 -0
  1219. package/crates/pi/src/modes/interactive/messages.rs +640 -0
  1220. package/crates/pi/src/modes/interactive/mod.rs +51 -0
  1221. package/crates/pi/src/modes/interactive/progress.rs +181 -0
  1222. package/crates/pi/src/modes/interactive/runtime.rs +7227 -0
  1223. package/crates/pi/src/modes/interactive/selectors.rs +146 -0
  1224. package/crates/pi/src/modes/interactive/snapshots/pi__modes__interactive__tests__bash_state_widths.snap +138 -0
  1225. package/crates/pi/src/modes/interactive/snapshots/pi__modes__interactive__tests__compact_state_widths.snap +137 -0
  1226. package/crates/pi/src/modes/interactive/snapshots/pi__modes__interactive__tests__custom_branch_skill_messages.snap +145 -0
  1227. package/crates/pi/src/modes/interactive/snapshots/pi__modes__interactive__tests__empty_state_widths.snap +136 -0
  1228. package/crates/pi/src/modes/interactive/snapshots/pi__modes__interactive__tests__error_state_widths.snap +139 -0
  1229. package/crates/pi/src/modes/interactive/snapshots/pi__modes__interactive__tests__loading_state_widths.snap +136 -0
  1230. package/crates/pi/src/modes/interactive/snapshots/pi__modes__interactive__tests__queue_state_widths.snap +139 -0
  1231. package/crates/pi/src/modes/interactive/snapshots/pi__modes__interactive__tests__retry_state_widths.snap +136 -0
  1232. package/crates/pi/src/modes/interactive/snapshots/pi__modes__interactive__tests__streaming_state_widths.snap +137 -0
  1233. package/crates/pi/src/modes/interactive/snapshots/pi__modes__interactive__tests__user_assistant_messages.snap +145 -0
  1234. package/crates/pi/src/modes/interactive/startup.rs +194 -0
  1235. package/crates/pi/src/modes/interactive/state.rs +759 -0
  1236. package/crates/pi/src/modes/interactive/status.rs +70 -0
  1237. package/crates/pi/src/modes/interactive/tests.rs +945 -0
  1238. package/crates/pi/src/modes/interactive/theme.rs +1402 -0
  1239. package/crates/pi/src/modes/interactive/tool_renderer.rs +152 -0
  1240. package/crates/pi/src/modes/interactive/view.rs +511 -0
  1241. package/crates/pi/src/modes/mod.rs +6 -0
  1242. package/crates/pi/src/modes/print/input.rs +415 -0
  1243. package/crates/pi/src/modes/print/json.rs +195 -0
  1244. package/crates/pi/src/modes/print/mod.rs +403 -0
  1245. package/crates/pi/src/modes/print/text.rs +412 -0
  1246. package/crates/pi/src/modes/rpc/extension_ui.rs +361 -0
  1247. package/crates/pi/src/modes/rpc/jsonl.rs +429 -0
  1248. package/crates/pi/src/modes/rpc/mod.rs +19 -0
  1249. package/crates/pi/src/modes/rpc/mode_host.rs +466 -0
  1250. package/crates/pi/src/modes/rpc/server.rs +3037 -0
  1251. package/crates/pi/src/modes/rpc/types.rs +2999 -0
  1252. package/crates/pi/src/modes/run.rs +636 -0
  1253. package/crates/pi/tests/fixtures/fake_extension_host.rs +104 -0
  1254. package/crates/pi/tests/fixtures/tool-schemas/bash.json +16 -0
  1255. package/crates/pi/tests/fixtures/tool-schemas/edit.json +34 -0
  1256. package/crates/pi/tests/fixtures/tool-schemas/find.json +20 -0
  1257. package/crates/pi/tests/fixtures/tool-schemas/grep.json +36 -0
  1258. package/crates/pi/tests/fixtures/tool-schemas/ls.json +13 -0
  1259. package/crates/pi/tests/fixtures/tool-schemas/read.json +20 -0
  1260. package/crates/pi/tests/fixtures/tool-schemas/write.json +17 -0
  1261. package/crates/pi-agent/Cargo.toml +30 -0
  1262. package/crates/pi-agent/LICENSE +21 -0
  1263. package/crates/pi-agent/src/agent.rs +1780 -0
  1264. package/crates/pi-agent/src/bus.rs +770 -0
  1265. package/crates/pi-agent/src/config.rs +445 -0
  1266. package/crates/pi-agent/src/drain.rs +941 -0
  1267. package/crates/pi-agent/src/error.rs +61 -0
  1268. package/crates/pi-agent/src/event.rs +270 -0
  1269. package/crates/pi-agent/src/lib.rs +49 -0
  1270. package/crates/pi-agent/src/message.rs +296 -0
  1271. package/crates/pi-agent/src/queue.rs +222 -0
  1272. package/crates/pi-agent/src/run.rs +1870 -0
  1273. package/crates/pi-agent/src/schedule.rs +2153 -0
  1274. package/crates/pi-agent/src/state.rs +377 -0
  1275. package/crates/pi-agent/src/tool.rs +496 -0
  1276. package/crates/pi-ai/Cargo.toml +45 -0
  1277. package/crates/pi-ai/LICENSE +21 -0
  1278. package/crates/pi-ai/data/builtin-models.json +25193 -0
  1279. package/crates/pi-ai/src/auth/config_value.rs +598 -0
  1280. package/crates/pi-ai/src/auth/context.rs +215 -0
  1281. package/crates/pi-ai/src/auth/credential_store.rs +318 -0
  1282. package/crates/pi-ai/src/auth/env_keys.rs +647 -0
  1283. package/crates/pi-ai/src/auth/error.rs +194 -0
  1284. package/crates/pi-ai/src/auth/file_store.rs +1179 -0
  1285. package/crates/pi-ai/src/auth/http.rs +526 -0
  1286. package/crates/pi-ai/src/auth/mod.rs +29 -0
  1287. package/crates/pi-ai/src/auth/oauth/anthropic.rs +1482 -0
  1288. package/crates/pi-ai/src/auth/oauth/callback_server.rs +584 -0
  1289. package/crates/pi-ai/src/auth/oauth/device_code.rs +573 -0
  1290. package/crates/pi-ai/src/auth/oauth/github_copilot.rs +1527 -0
  1291. package/crates/pi-ai/src/auth/oauth/mod.rs +26 -0
  1292. package/crates/pi-ai/src/auth/oauth/openai_codex.rs +2005 -0
  1293. package/crates/pi-ai/src/auth/oauth/page.rs +159 -0
  1294. package/crates/pi-ai/src/auth/oauth/pkce.rs +145 -0
  1295. package/crates/pi-ai/src/auth/oauth/radius.rs +1685 -0
  1296. package/crates/pi-ai/src/auth/oauth/xai.rs +989 -0
  1297. package/crates/pi-ai/src/auth/resolve.rs +701 -0
  1298. package/crates/pi-ai/src/auth/runtime_credentials.rs +226 -0
  1299. package/crates/pi-ai/src/auth/types.rs +534 -0
  1300. package/crates/pi-ai/src/catalog/merge.rs +316 -0
  1301. package/crates/pi-ai/src/catalog/mod.rs +485 -0
  1302. package/crates/pi-ai/src/lib.rs +13 -0
  1303. package/crates/pi-ai/src/models_store.rs +974 -0
  1304. package/crates/pi-ai/src/provider.rs +285 -0
  1305. package/crates/pi-ai/src/providers/anthropic_messages.rs +1473 -0
  1306. package/crates/pi-ai/src/providers/azure_openai_responses.rs +559 -0
  1307. package/crates/pi-ai/src/providers/bedrock_converse_stream.rs +2430 -0
  1308. package/crates/pi-ai/src/providers/google_generative_ai.rs +622 -0
  1309. package/crates/pi-ai/src/providers/google_vertex.rs +1565 -0
  1310. package/crates/pi-ai/src/providers/mistral_conversations.rs +1245 -0
  1311. package/crates/pi-ai/src/providers/mod.rs +35 -0
  1312. package/crates/pi-ai/src/providers/openai_codex_responses.rs +2003 -0
  1313. package/crates/pi-ai/src/providers/openai_completions.rs +1873 -0
  1314. package/crates/pi-ai/src/providers/openai_responses.rs +757 -0
  1315. package/crates/pi-ai/src/providers/pi_messages.rs +1235 -0
  1316. package/crates/pi-ai/src/providers/registry.rs +923 -0
  1317. package/crates/pi-ai/src/providers/shared/google.rs +1193 -0
  1318. package/crates/pi-ai/src/providers/shared/responses.rs +1873 -0
  1319. package/crates/pi-ai/src/providers/shared.rs +671 -0
  1320. package/crates/pi-ai/src/providers/stream_state.rs +683 -0
  1321. package/crates/pi-ai/src/providers/transport.rs +488 -0
  1322. package/crates/pi-ai/src/text.rs +322 -0
  1323. package/crates/pi-ai/src/types.rs +1050 -0
  1324. package/crates/pi-ai/tests/conformance_anthropic_mistral.rs +664 -0
  1325. package/crates/pi-ai/tests/conformance_bedrock_pi_messages.rs +908 -0
  1326. package/crates/pi-ai/tests/conformance_codex.rs +815 -0
  1327. package/crates/pi-ai/tests/conformance_google.rs +743 -0
  1328. package/crates/pi-ai/tests/conformance_openai.rs +607 -0
  1329. package/crates/pi-ai/tests/fixtures/provenance/anthropic-sse-parsing.test.ts +275 -0
  1330. package/crates/pi-ai/tests/fixtures/provenance/mistral-reasoning-mode.test.ts +97 -0
  1331. package/crates/pi-ai/tests/fixtures/provenance/upstream/src/api/anthropic-messages.ts +1313 -0
  1332. package/crates/pi-ai/tests/fixtures/provenance/upstream/src/api/bedrock-converse-stream.ts +1091 -0
  1333. package/crates/pi-ai/tests/fixtures/provenance/upstream/src/api/google-generative-ai.ts +509 -0
  1334. package/crates/pi-ai/tests/fixtures/provenance/upstream/src/api/google-shared.ts +350 -0
  1335. package/crates/pi-ai/tests/fixtures/provenance/upstream/src/api/mistral-conversations.ts +664 -0
  1336. package/crates/pi-ai/tests/fixtures/provenance/upstream/src/api/openai-completions.ts +1355 -0
  1337. package/crates/pi-ai/tests/fixtures/provenance/upstream/src/api/openai-responses-shared.ts +613 -0
  1338. package/crates/pi-ai/tests/fixtures/provenance/upstream/src/api/openai-responses.ts +318 -0
  1339. package/crates/pi-ai/tests/fixtures/provenance/upstream/test/abort.test.ts +303 -0
  1340. package/crates/pi-ai/tests/fixtures/provenance/upstream/test/azure-openai-base-url.test.ts +175 -0
  1341. package/crates/pi-ai/tests/fixtures/provenance/upstream/test/bedrock-endpoint-resolution.test.ts +208 -0
  1342. package/crates/pi-ai/tests/fixtures/provenance/upstream/test/bedrock-thinking-payload.test.ts +246 -0
  1343. package/crates/pi-ai/tests/fixtures/provenance/upstream/test/google-vertex-api-key-resolution.test.ts +223 -0
  1344. package/crates/pi-ai/tests/fixtures/provenance/upstream/test/openai-codex-stream.test.ts +2028 -0
  1345. package/crates/pi-ai/tests/fixtures/provenance/upstream/test/openai-completions-prompt-cache.test.ts +264 -0
  1346. package/crates/pi-ai/tests/fixtures/provenance/upstream/test/openai-completions-retry.test.ts +86 -0
  1347. package/crates/pi-ai/tests/fixtures/provenance/upstream/test/openai-completions-thinking-as-text.test.ts +218 -0
  1348. package/crates/pi-ai/tests/fixtures/provenance/upstream/test/openai-completions-tool-choice.test.ts +1716 -0
  1349. package/crates/pi-ai/tests/fixtures/provenance/upstream/test/openai-responses-partial-json-cleanup.test.ts +106 -0
  1350. package/crates/pi-ai/tests/fixtures/provenance/upstream/test/openai-responses-terminal-event.test.ts +233 -0
  1351. package/crates/pi-ai/tests/fixtures/provenance/upstream/test/pi-messages.test.ts +243 -0
  1352. package/crates/pi-ai/tests/fixtures/provenance/upstream/test/provider-error-body-passthrough.test.ts +78 -0
  1353. package/crates/pi-ai/tests/fixtures/providers/anthropic-messages/cases.jsonl +4 -0
  1354. package/crates/pi-ai/tests/fixtures/providers/azure-openai-responses/cases.jsonl +2 -0
  1355. package/crates/pi-ai/tests/fixtures/providers/bedrock-converse-stream/cases.jsonl +8 -0
  1356. package/crates/pi-ai/tests/fixtures/providers/google-generative-ai/cases.jsonl +6 -0
  1357. package/crates/pi-ai/tests/fixtures/providers/google-vertex/cases.jsonl +6 -0
  1358. package/crates/pi-ai/tests/fixtures/providers/mistral-conversations/cases.jsonl +4 -0
  1359. package/crates/pi-ai/tests/fixtures/providers/openai-codex-responses/cases.jsonl +7 -0
  1360. package/crates/pi-ai/tests/fixtures/providers/openai-completions/cases.jsonl +9 -0
  1361. package/crates/pi-ai/tests/fixtures/providers/openai-responses/cases.jsonl +5 -0
  1362. package/crates/pi-ai/tests/fixtures/providers/pi-messages/cases.jsonl +8 -0
  1363. package/crates/pi-ai/tests/provider_registry_contract.rs +85 -0
  1364. package/crates/pi-ai/tests/support/golden.rs +191 -0
  1365. package/crates/pi-ai/tests/support/http.rs +525 -0
  1366. package/crates/pi-ai/tests/support/mod.rs +4 -0
  1367. package/crates/pi-ext/Cargo.toml +34 -0
  1368. package/crates/pi-ext/LICENSE +21 -0
  1369. package/crates/pi-ext/src/adapters.rs +1915 -0
  1370. package/crates/pi-ext/src/client.rs +1883 -0
  1371. package/crates/pi-ext/src/host.rs +318 -0
  1372. package/crates/pi-ext/src/lib.rs +10 -0
  1373. package/crates/pi-ext/src/protocol.rs +1880 -0
  1374. package/crates/pi-ext/src/sanitize.rs +524 -0
  1375. package/crates/pi-ext/src/test_support.rs +93 -0
  1376. package/crates/pi-ext/tests/fixtures/fake_jsonl_host.rs +35 -0
  1377. package/crates/pi-ext/tests/fixtures/protocol/frames.jsonl +30 -0
  1378. package/crates/pi-ext/tests/scaling.rs +530 -0
  1379. package/crates/pi-tui/Cargo.toml +56 -0
  1380. package/crates/pi-tui/LICENSE +21 -0
  1381. package/crates/pi-tui/src/bin/pi_tui_pty_fixture.rs +722 -0
  1382. package/crates/pi-tui/src/component.rs +101 -0
  1383. package/crates/pi-tui/src/components/editor/mod.rs +2626 -0
  1384. package/crates/pi-tui/src/components/editor/paste.rs +406 -0
  1385. package/crates/pi-tui/src/components/editor/state.rs +240 -0
  1386. package/crates/pi-tui/src/components/image.rs +352 -0
  1387. package/crates/pi-tui/src/components/input.rs +742 -0
  1388. package/crates/pi-tui/src/components/loader.rs +364 -0
  1389. package/crates/pi-tui/src/components/markdown.rs +1154 -0
  1390. package/crates/pi-tui/src/components/mod.rs +30 -0
  1391. package/crates/pi-tui/src/components/padded.rs +212 -0
  1392. package/crates/pi-tui/src/components/select_list.rs +464 -0
  1393. package/crates/pi-tui/src/components/settings_list.rs +597 -0
  1394. package/crates/pi-tui/src/components/spacer.rs +67 -0
  1395. package/crates/pi-tui/src/components/text.rs +225 -0
  1396. package/crates/pi-tui/src/components/truncated_text.rs +137 -0
  1397. package/crates/pi-tui/src/components/util.rs +324 -0
  1398. package/crates/pi-tui/src/editor_support/autocomplete.rs +1039 -0
  1399. package/crates/pi-tui/src/editor_support/history.rs +248 -0
  1400. package/crates/pi-tui/src/editor_support/kill_ring.rs +139 -0
  1401. package/crates/pi-tui/src/editor_support/mod.rs +29 -0
  1402. package/crates/pi-tui/src/editor_support/undo.rs +73 -0
  1403. package/crates/pi-tui/src/editor_support/word_nav.rs +412 -0
  1404. package/crates/pi-tui/src/editor_support/wrap.rs +317 -0
  1405. package/crates/pi-tui/src/focus.rs +564 -0
  1406. package/crates/pi-tui/src/frame.rs +163 -0
  1407. package/crates/pi-tui/src/fuzzy.rs +221 -0
  1408. package/crates/pi-tui/src/image.rs +749 -0
  1409. package/crates/pi-tui/src/keybindings.rs +602 -0
  1410. package/crates/pi-tui/src/keys.rs +1242 -0
  1411. package/crates/pi-tui/src/layout.rs +563 -0
  1412. package/crates/pi-tui/src/lib.rs +16 -0
  1413. package/crates/pi-tui/src/link.rs +408 -0
  1414. package/crates/pi-tui/src/overlay.rs +895 -0
  1415. package/crates/pi-tui/src/terminal/backend.rs +348 -0
  1416. package/crates/pi-tui/src/terminal/caps.rs +237 -0
  1417. package/crates/pi-tui/src/terminal/guard.rs +476 -0
  1418. package/crates/pi-tui/src/terminal/input.rs +266 -0
  1419. package/crates/pi-tui/src/terminal/mod.rs +31 -0
  1420. package/crates/pi-tui/src/terminal/probe.rs +498 -0
  1421. package/crates/pi-tui/src/terminal/sink.rs +128 -0
  1422. package/crates/pi-tui/src/terminal/writer.rs +1541 -0
  1423. package/crates/pi-tui/src/text/ansi.rs +415 -0
  1424. package/crates/pi-tui/src/text/mod.rs +29 -0
  1425. package/crates/pi-tui/src/text/slice.rs +645 -0
  1426. package/crates/pi-tui/src/text/tests.rs +359 -0
  1427. package/crates/pi-tui/src/text/width.rs +311 -0
  1428. package/crates/pi-tui/src/text/wrap.rs +344 -0
  1429. package/crates/pi-tui/tests/pty_no_flicker.rs +929 -0
  1430. package/deny.toml +38 -0
  1431. package/npm-package/bin/pi.js +31 -0
  1432. package/npm-package/package.json +35 -0
  1433. package/npm-package/tests/archive.test.ts +139 -0
  1434. package/npm-package/tests/wrapper.test.ts +70 -0
  1435. package/package.json +34 -0
  1436. package/packages/extension-host/bun.lock +664 -0
  1437. package/packages/extension-host/fixtures/extensions/all-events.ts +55 -0
  1438. package/packages/extension-host/fixtures/extensions/control-compat-shadow.ts +14 -0
  1439. package/packages/extension-host/fixtures/extensions/control-compat.ts +35 -0
  1440. package/packages/extension-host/fixtures/extensions/crash.ts +36 -0
  1441. package/packages/extension-host/fixtures/extensions/hooks.ts +45 -0
  1442. package/packages/extension-host/fixtures/extensions/hostile.ts +39 -0
  1443. package/packages/extension-host/fixtures/extensions/idle.ts +9 -0
  1444. package/packages/extension-host/fixtures/extensions/provider-stream.ts +121 -0
  1445. package/packages/extension-host/fixtures/extensions/terminal-input-fast.ts +20 -0
  1446. package/packages/extension-host/fixtures/extensions/terminal-input-slow.ts +23 -0
  1447. package/packages/extension-host/fixtures/extensions/tool-progress.ts +72 -0
  1448. package/packages/extension-host/fixtures/extensions/tool.ts +71 -0
  1449. package/packages/extension-host/fixtures/extensions/virtual-imports.ts +87 -0
  1450. package/packages/extension-host/fixtures/extensions/widget-active.ts +16 -0
  1451. package/packages/extension-host/fixtures/photon-probe.ts +45 -0
  1452. package/packages/extension-host/fixtures/runtime-import.ts +54 -0
  1453. package/packages/extension-host/package.json +54 -0
  1454. package/packages/extension-host/src/host.ts +1651 -0
  1455. package/packages/extension-host/src/index.ts +21 -0
  1456. package/packages/extension-host/src/main.ts +51 -0
  1457. package/packages/extension-host/src/protocol.ts +10 -0
  1458. package/packages/extension-host/src/refs.d.ts +364 -0
  1459. package/packages/extension-host/src/sanitize.ts +503 -0
  1460. package/packages/extension-host/src/virtual-modules.ts +90 -0
  1461. package/packages/extension-host/tests/acceptance.test.ts +1498 -0
  1462. package/packages/extension-host/tests/host.test.ts +244 -0
  1463. package/packages/extension-host/tests/sanitize.test.ts +359 -0
  1464. package/packages/extension-host/tests/scaling.test.ts +661 -0
  1465. package/packages/extension-host/tests/virtual-modules.test.ts +122 -0
  1466. package/packages/extension-host/tsconfig.build.json +44 -0
  1467. package/packages/extension-host/tsconfig.check.json +29 -0
  1468. package/packages/extension-host/tsconfig.json +8 -0
  1469. package/packages/pi-tui-protocol/package.json +41 -0
  1470. package/packages/pi-tui-protocol/src/client.ts +341 -0
  1471. package/packages/pi-tui-protocol/src/codec.ts +439 -0
  1472. package/packages/pi-tui-protocol/src/index.ts +77 -0
  1473. package/packages/pi-tui-protocol/src/types.ts +358 -0
  1474. package/packages/pi-tui-protocol/tests/client.test.ts +175 -0
  1475. package/packages/pi-tui-protocol/tests/codec.test.ts +196 -0
  1476. package/packages/pi-tui-protocol/tests/fixtures/frames.jsonl +30 -0
  1477. package/packages/pi-tui-protocol/tsconfig.build.json +23 -0
  1478. package/packages/pi-tui-protocol/tsconfig.json +9 -0
  1479. package/rust-toolchain.toml +3 -0
  1480. package/rustfmt.toml +5 -0
  1481. package/scripts/bench-extension-scaling.ts +466 -0
  1482. package/scripts/build-extension-host.ts +51 -0
  1483. package/scripts/generate-builtin-models.ts +344 -0
  1484. package/scripts/generate-session-fixtures.ts +1758 -0
  1485. package/scripts/generate-tool-schemas.ts +444 -0
  1486. package/scripts/package-npm.ts +263 -0
  1487. package/scripts/package-release.ts +422 -0
  1488. package/scripts/release/archive.ts +594 -0
  1489. package/scripts/release/args.ts +187 -0
  1490. package/scripts/release/extract-host-asset.ts +224 -0
  1491. package/scripts/release/host.ts +825 -0
  1492. package/scripts/release/runner.ts +435 -0
  1493. package/scripts/release/runtime.ts +130 -0
  1494. package/scripts/release/stage.ts +437 -0
  1495. package/scripts/release/targets.ts +150 -0
  1496. package/scripts/tests/archive.test.ts +210 -0
  1497. package/scripts/tests/args.test.ts +180 -0
  1498. package/scripts/tests/cli-dry-run.test.ts +61 -0
  1499. package/scripts/tests/compat-authority.test.ts +79 -0
  1500. package/scripts/tests/compat-matrix.test.ts +389 -0
  1501. package/scripts/tests/extract-host-asset.test.ts +176 -0
  1502. package/scripts/tests/host.test.ts +443 -0
  1503. package/scripts/tests/npm-installed-smoke.test.ts +104 -0
  1504. package/scripts/tests/package-release.test.ts +516 -0
  1505. package/scripts/tests/repro-guards.test.ts +241 -0
  1506. package/scripts/tests/runner.test.ts +188 -0
  1507. package/scripts/tests/stage.test.ts +232 -0
  1508. package/scripts/tests/targets.test.ts +93 -0
  1509. package/scripts/tsconfig.json +47 -0
  1510. package/scripts/verification/compat-matrix.json +455 -0
  1511. package/scripts/verification/compat-matrix.ts +632 -0
  1512. package/scripts/verification/e2e-smoke.ts +1093 -0
  1513. package/scripts/verification/extension.ts +339 -0
  1514. package/scripts/verification/foundation.test.ts +339 -0
  1515. package/scripts/verification/npm-installed-smoke.ts +117 -0
  1516. package/scripts/verification/performance.ts +1425 -0
  1517. package/scripts/verification/pty.ts +332 -0
@@ -0,0 +1,1685 @@
1
+ //! Radius gateway OAuth flow.
2
+ //!
3
+ //! Radius is a pi-messages gateway. OAuth endpoints are discovered from
4
+ //! `GET <gateway>/v1/oauth`. Login offers browser (PKCE + loopback callback) or
5
+ //! device-code paths; tokens are exchanged as form-urlencoded posts.
6
+
7
+ use std::collections::BTreeMap;
8
+ use std::net::{IpAddr, Ipv4Addr};
9
+ use std::time::{SystemTime, UNIX_EPOCH};
10
+
11
+ use futures::future::BoxFuture;
12
+ use serde::Deserialize;
13
+ use serde_json::Value;
14
+ use tokio_util::sync::CancellationToken;
15
+
16
+ use super::super::error::AuthError;
17
+ use super::super::http::AuthHttpClient;
18
+ use super::super::types::{
19
+ AuthEvent, AuthInteraction, AuthPrompt, AuthSelectOption, ModelAuth, OAuthAuth, OAuthCredential,
20
+ };
21
+ use super::callback_server::{OAuthCallbackConfig, OAuthCallbackServer};
22
+ use super::device_code::{
23
+ OAuthDeviceCodePollOptions, OAuthDeviceCodePollResult, poll_oauth_device_code_flow,
24
+ };
25
+ use super::pkce::generate_pkce;
26
+
27
+ /// Fixed loopback host used by Radius callback and redirect URI.
28
+ pub const CALLBACK_HOST: IpAddr = IpAddr::V4(Ipv4Addr::LOCALHOST);
29
+
30
+ /// Fixed callback port used by Radius browser login.
31
+ pub const CALLBACK_PORT: u16 = 1456;
32
+
33
+ /// Callback path used by Radius browser login.
34
+ pub const CALLBACK_PATH: &str = "/oauth/callback";
35
+
36
+ /// Redirect URI registered for the browser authorization-code flow.
37
+ pub const REDIRECT_URI: &str = "http://127.0.0.1:1456/oauth/callback";
38
+
39
+ /// Access-token expiry skew applied after every successful token response.
40
+ pub const TOKEN_EXPIRY_SKEW_MS: i64 = 60_000;
41
+
42
+ const LOGIN_METHOD_BROWSER: &str = "browser";
43
+ const LOGIN_METHOD_DEVICE_CODE: &str = "device-code";
44
+
45
+ /// Factory options for [`create_radius_oauth`].
46
+ #[derive(Clone, Debug, Eq, PartialEq)]
47
+ pub struct RadiusOAuthOptions {
48
+ /// Display name (for example `"Radius"`).
49
+ pub name: String,
50
+ /// Gateway base URL or host; normalized before use.
51
+ pub gateway: String,
52
+ }
53
+
54
+ /// Radius OAuth implementation discovered from a gateway.
55
+ #[derive(Clone, Debug)]
56
+ pub struct RadiusOAuth {
57
+ name: String,
58
+ gateway: String,
59
+ http: AuthHttpClient,
60
+ /// Production uses [`CALLBACK_PORT`]; tests may override for free ports.
61
+ callback_port: u16,
62
+ }
63
+
64
+ /// OAuth endpoints advertised by `GET <gateway>/v1/oauth`.
65
+ #[derive(Clone, Debug, Deserialize, Eq, PartialEq)]
66
+ #[serde(rename_all = "camelCase")]
67
+ pub struct RadiusOAuthConfig {
68
+ /// OAuth issuer identifier.
69
+ pub issuer: String,
70
+ /// Browser authorization endpoint.
71
+ pub authorization_endpoint: String,
72
+ /// Token endpoint used for code, device, and refresh exchanges.
73
+ pub token_endpoint: String,
74
+ /// Device authorization endpoint.
75
+ pub device_authorization_endpoint: String,
76
+ /// Optional device-authorization events endpoint (preserved from config).
77
+ #[serde(default)]
78
+ pub device_authorization_events_endpoint: String,
79
+ /// Fallback verification URI when the device response omits one.
80
+ pub verification_endpoint: String,
81
+ /// OAuth client id for this gateway.
82
+ pub client_id: String,
83
+ /// Space-delimited scope string requested at authorize/device time.
84
+ pub scope: String,
85
+ /// `grant_type` value used while polling the device token endpoint.
86
+ pub device_code_grant_type: String,
87
+ }
88
+
89
+ #[derive(Clone, Debug)]
90
+ struct DeviceAuthorizationResponse {
91
+ device_code: String,
92
+ user_code: String,
93
+ verification_uri: Option<String>,
94
+ expires_in: u64,
95
+ interval: Option<u64>,
96
+ }
97
+
98
+ #[derive(Clone, Debug)]
99
+ struct OAuthResponseError {
100
+ oauth_error: Option<String>,
101
+ message: String,
102
+ }
103
+
104
+ impl std::fmt::Display for OAuthResponseError {
105
+ fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
106
+ f.write_str(&self.message)
107
+ }
108
+ }
109
+
110
+ /// Normalize a Radius gateway URL: ensure `http(s)://` and strip trailing slashes.
111
+ #[must_use]
112
+ pub fn normalize_radius_gateway_url(value: &str) -> String {
113
+ let trimmed = value.trim();
114
+ let with_scheme = if trimmed
115
+ .get(..8)
116
+ .is_some_and(|prefix| prefix.eq_ignore_ascii_case("https://"))
117
+ || trimmed
118
+ .get(..7)
119
+ .is_some_and(|prefix| prefix.eq_ignore_ascii_case("http://"))
120
+ {
121
+ trimmed.to_owned()
122
+ } else {
123
+ format!("https://{trimmed}")
124
+ };
125
+ with_scheme.trim_end_matches('/').to_owned()
126
+ }
127
+
128
+ fn is_loopback_destination(url: &reqwest::Url) -> bool {
129
+ let Some(host) = url.host_str() else {
130
+ return false;
131
+ };
132
+ host.eq_ignore_ascii_case("localhost")
133
+ || host
134
+ .parse::<IpAddr>()
135
+ .is_ok_and(|address| address.is_loopback())
136
+ }
137
+
138
+ fn validate_radius_verification_uri(value: &str, gateway: &str) -> Result<String, AuthError> {
139
+ let untrusted = || AuthError::message("Untrusted verification URI in Radius OAuth response");
140
+ let uri = reqwest::Url::parse(value).map_err(|_| untrusted())?;
141
+ if !uri.username().is_empty() || uri.password().is_some() || uri.host_str().is_none() {
142
+ return Err(untrusted());
143
+ }
144
+ if uri.scheme() == "https" {
145
+ return Ok(uri.to_string());
146
+ }
147
+ if uri.scheme() == "http" && is_loopback_destination(&uri) {
148
+ let configured_gateway = reqwest::Url::parse(gateway).map_err(|_| untrusted())?;
149
+ if configured_gateway.scheme() == "http" && is_loopback_destination(&configured_gateway) {
150
+ return Ok(uri.to_string());
151
+ }
152
+ }
153
+ Err(untrusted())
154
+ }
155
+
156
+ /// Build the fixed (or test-overridden) redirect URI for a callback port.
157
+ #[must_use]
158
+ pub fn radius_redirect_uri(port: u16) -> String {
159
+ if port == CALLBACK_PORT {
160
+ REDIRECT_URI.to_owned()
161
+ } else {
162
+ format!("http://127.0.0.1:{port}{CALLBACK_PATH}")
163
+ }
164
+ }
165
+
166
+ /// Create a Radius OAuth flow for `options.gateway`.
167
+ ///
168
+ /// # Errors
169
+ ///
170
+ /// Returns an error when the shared HTTP client cannot be constructed.
171
+ pub fn create_radius_oauth(options: RadiusOAuthOptions) -> Result<RadiusOAuth, AuthError> {
172
+ RadiusOAuth::new(options)
173
+ }
174
+
175
+ impl RadiusOAuth {
176
+ /// Create a Radius OAuth flow with the shared HTTP client and fixed port.
177
+ ///
178
+ /// # Errors
179
+ ///
180
+ /// Returns an error when the shared HTTP client cannot be constructed.
181
+ pub fn new(options: RadiusOAuthOptions) -> Result<Self, AuthError> {
182
+ let http =
183
+ AuthHttpClient::new().map_err(super::super::http::AuthHttpError::into_auth_error)?;
184
+ Ok(Self::with_client(options, http, CALLBACK_PORT))
185
+ }
186
+
187
+ /// Create a Radius OAuth flow with an injected HTTP client (tests/mocks).
188
+ #[must_use]
189
+ pub fn with_client(
190
+ options: RadiusOAuthOptions,
191
+ http: AuthHttpClient,
192
+ callback_port: u16,
193
+ ) -> Self {
194
+ Self {
195
+ name: options.name,
196
+ gateway: normalize_radius_gateway_url(&options.gateway),
197
+ http,
198
+ callback_port,
199
+ }
200
+ }
201
+
202
+ /// Normalized gateway base URL used for config discovery.
203
+ #[must_use]
204
+ pub fn gateway(&self) -> &str {
205
+ &self.gateway
206
+ }
207
+
208
+ /// Callback port used by browser login.
209
+ #[must_use]
210
+ pub fn callback_port(&self) -> u16 {
211
+ self.callback_port
212
+ }
213
+
214
+ /// Discover OAuth endpoints from the gateway.
215
+ ///
216
+ /// # Errors
217
+ ///
218
+ /// Returns a message error when the gateway is unreachable or returns a
219
+ /// non-success / invalid payload.
220
+ pub async fn load_oauth_config(
221
+ &self,
222
+ signal: Option<&CancellationToken>,
223
+ ) -> Result<RadiusOAuthConfig, AuthError> {
224
+ load_radius_oauth_config(&self.http, &self.gateway, signal).await
225
+ }
226
+
227
+ async fn login_with_browser(
228
+ &self,
229
+ oauth: &RadiusOAuthConfig,
230
+ interaction: &dyn AuthInteraction,
231
+ ) -> Result<OAuthCredential, AuthError> {
232
+ let pkce = generate_pkce()?;
233
+ let state = random_uuid_v4()?;
234
+ let redirect_uri = radius_redirect_uri(self.callback_port);
235
+ let authorize_url = build_authorize_url(oauth, &pkce.challenge, &state, &redirect_uri);
236
+
237
+ let callback_server = OAuthCallbackServer::start_soft(OAuthCallbackConfig {
238
+ port: self.callback_port,
239
+ path: CALLBACK_PATH.to_owned(),
240
+ expected_state: state.clone(),
241
+ success_message: "Signed in to Radius. You may now close this page.".to_owned(),
242
+ host: Some(CALLBACK_HOST),
243
+ })
244
+ .await;
245
+
246
+ interaction.notify(AuthEvent::Progress {
247
+ message: format!("Listening for OAuth callback on {redirect_uri}"),
248
+ });
249
+ interaction.notify(AuthEvent::AuthUrl {
250
+ url: authorize_url,
251
+ instructions: Some("Continue in your browser.".to_owned()),
252
+ });
253
+
254
+ let wait = callback_server.wait_for_code();
255
+ let code = if let Some(signal) = interaction.signal() {
256
+ tokio::select! {
257
+ () = signal.cancelled() => {
258
+ callback_server.close().await;
259
+ return Err(AuthError::Cancelled);
260
+ }
261
+ code = wait => code,
262
+ }
263
+ } else {
264
+ wait.await
265
+ };
266
+
267
+ callback_server.close().await;
268
+
269
+ let Some(code) = code else {
270
+ if interaction
271
+ .signal()
272
+ .as_ref()
273
+ .is_some_and(CancellationToken::is_cancelled)
274
+ {
275
+ return Err(AuthError::Cancelled);
276
+ }
277
+ return Err(AuthError::message("OAuth callback did not complete."));
278
+ };
279
+
280
+ // State is validated by the callback server before settling.
281
+ if code.state != state {
282
+ return Err(AuthError::message("OAuth state mismatch"));
283
+ }
284
+
285
+ request_oauth_token(
286
+ &self.http,
287
+ oauth,
288
+ form_fields([
289
+ ("grant_type", "authorization_code"),
290
+ ("client_id", oauth.client_id.as_str()),
291
+ ("redirect_uri", redirect_uri.as_str()),
292
+ ("code", code.code.as_str()),
293
+ ("code_verifier", pkce.verifier.as_str()),
294
+ ]),
295
+ interaction.signal().as_ref(),
296
+ )
297
+ .await
298
+ .map_err(map_token_error)
299
+ }
300
+
301
+ async fn login_with_device_code(
302
+ &self,
303
+ oauth: &RadiusOAuthConfig,
304
+ interaction: &dyn AuthInteraction,
305
+ ) -> Result<OAuthCredential, AuthError> {
306
+ let device =
307
+ request_device_authorization(&self.http, oauth, interaction.signal().as_ref()).await?;
308
+
309
+ let verification_uri = device
310
+ .verification_uri
311
+ .as_deref()
312
+ .unwrap_or(oauth.verification_endpoint.as_str());
313
+ let verification_uri = validate_radius_verification_uri(verification_uri, &self.gateway)?;
314
+
315
+ interaction.notify(AuthEvent::DeviceCode {
316
+ user_code: device.user_code.clone(),
317
+ verification_uri,
318
+ interval_seconds: device.interval,
319
+ expires_in_seconds: Some(device.expires_in),
320
+ });
321
+
322
+ let http = self.http.clone();
323
+ let oauth = oauth.clone();
324
+ let device_code = device.device_code.clone();
325
+ let signal = interaction.signal();
326
+
327
+ let mut options = OAuthDeviceCodePollOptions::new(move || {
328
+ let http = http.clone();
329
+ let oauth = oauth.clone();
330
+ let device_code = device_code.clone();
331
+ let signal = signal.clone();
332
+ async move {
333
+ match request_oauth_token(
334
+ &http,
335
+ &oauth,
336
+ form_fields([
337
+ ("grant_type", oauth.device_code_grant_type.as_str()),
338
+ ("client_id", oauth.client_id.as_str()),
339
+ ("device_code", device_code.as_str()),
340
+ ]),
341
+ signal.as_ref(),
342
+ )
343
+ .await
344
+ {
345
+ Ok(credentials) => {
346
+ Ok(OAuthDeviceCodePollResult::Complete { value: credentials })
347
+ }
348
+ Err(error) => match error.oauth_error.as_deref() {
349
+ Some("authorization_pending") => Ok(OAuthDeviceCodePollResult::Pending),
350
+ Some("slow_down") => Ok(OAuthDeviceCodePollResult::SlowDown {
351
+ interval_seconds: None,
352
+ }),
353
+ Some("expired_token") => Ok(OAuthDeviceCodePollResult::Failed {
354
+ message: "Device authorization expired.".to_owned(),
355
+ }),
356
+ Some("access_denied") => Ok(OAuthDeviceCodePollResult::Failed {
357
+ message: "Device authorization was denied.".to_owned(),
358
+ }),
359
+ _ => Err(AuthError::message(error.message)),
360
+ },
361
+ }
362
+ }
363
+ });
364
+ options.interval_seconds = device.interval;
365
+ options.expires_in_seconds = Some(device.expires_in);
366
+ options.wait_before_first_poll = false;
367
+ options.signal = interaction.signal();
368
+
369
+ poll_oauth_device_code_flow(options).await
370
+ }
371
+ }
372
+
373
+ impl OAuthAuth for RadiusOAuth {
374
+ fn name(&self) -> &str {
375
+ &self.name
376
+ }
377
+
378
+ fn login_label(&self) -> Option<&str> {
379
+ None
380
+ }
381
+
382
+ fn login<'a>(
383
+ &'a self,
384
+ interaction: &'a dyn AuthInteraction,
385
+ ) -> BoxFuture<'a, Result<OAuthCredential, AuthError>> {
386
+ Box::pin(async move {
387
+ let oauth = self
388
+ .load_oauth_config(interaction.signal().as_ref())
389
+ .await?;
390
+ let login_method = interaction
391
+ .prompt(AuthPrompt::Select {
392
+ message: format!("Sign in to {}:", self.name),
393
+ options: vec![
394
+ AuthSelectOption {
395
+ id: LOGIN_METHOD_BROWSER.to_owned(),
396
+ label: "Sign in with browser (recommended)".to_owned(),
397
+ description: None,
398
+ },
399
+ AuthSelectOption {
400
+ id: LOGIN_METHOD_DEVICE_CODE.to_owned(),
401
+ label: "Sign in with device code (when signing in from another device)"
402
+ .to_owned(),
403
+ description: None,
404
+ },
405
+ ],
406
+ signal: None,
407
+ })
408
+ .await?;
409
+
410
+ if login_method == LOGIN_METHOD_DEVICE_CODE {
411
+ self.login_with_device_code(&oauth, interaction).await
412
+ } else if login_method == LOGIN_METHOD_BROWSER {
413
+ self.login_with_browser(&oauth, interaction).await
414
+ } else {
415
+ Err(AuthError::message(format!(
416
+ "Unknown {} sign-in method: {login_method}",
417
+ self.name
418
+ )))
419
+ }
420
+ })
421
+ }
422
+
423
+ fn refresh<'a>(
424
+ &'a self,
425
+ credential: &'a OAuthCredential,
426
+ signal: Option<CancellationToken>,
427
+ ) -> BoxFuture<'a, Result<OAuthCredential, AuthError>> {
428
+ Box::pin(async move {
429
+ let oauth = self.load_oauth_config(signal.as_ref()).await?;
430
+ request_oauth_token(
431
+ &self.http,
432
+ &oauth,
433
+ form_fields([
434
+ ("grant_type", "refresh_token"),
435
+ ("client_id", oauth.client_id.as_str()),
436
+ ("refresh_token", credential.refresh.as_str()),
437
+ ]),
438
+ signal.as_ref(),
439
+ )
440
+ .await
441
+ .map_err(map_token_error)
442
+ })
443
+ }
444
+
445
+ fn to_auth<'a>(
446
+ &'a self,
447
+ credential: &'a OAuthCredential,
448
+ ) -> BoxFuture<'a, Result<ModelAuth, AuthError>> {
449
+ // pi-messages uses the access token as a Bearer API key. Gateway base
450
+ // URL lives on the provider/model config, not ModelAuth; open extras
451
+ // (for example scope) stay on the stored credential.
452
+ Box::pin(async move {
453
+ Ok(ModelAuth {
454
+ api_key: Some(credential.access.clone()),
455
+ headers: None,
456
+ base_url: None,
457
+ })
458
+ })
459
+ }
460
+ }
461
+
462
+ /// Load and parse gateway OAuth discovery config.
463
+ ///
464
+ /// # Errors
465
+ ///
466
+ /// Returns a message error for transport failures, non-success responses, or
467
+ /// JSON that is not a Radius OAuth config object.
468
+ pub async fn load_radius_oauth_config(
469
+ http: &AuthHttpClient,
470
+ gateway: &str,
471
+ signal: Option<&CancellationToken>,
472
+ ) -> Result<RadiusOAuthConfig, AuthError> {
473
+ let url = format!("{}/v1/oauth", gateway.trim_end_matches('/'));
474
+ let response = http
475
+ .get_json(&url, None, signal)
476
+ .await
477
+ .map_err(super::super::http::AuthHttpError::into_auth_error)?;
478
+
479
+ if !response.ok {
480
+ return Err(AuthError::message(format!(
481
+ "Could not load Radius OAuth config from {gateway}: {} {}",
482
+ response.status, response.raw_body
483
+ )));
484
+ }
485
+
486
+ serde_json::from_value(response.body).map_err(|error| {
487
+ AuthError::message(format!(
488
+ "Could not load Radius OAuth config from {gateway}: invalid JSON ({error})"
489
+ ))
490
+ })
491
+ }
492
+
493
+ fn build_authorize_url(
494
+ oauth: &RadiusOAuthConfig,
495
+ challenge: &str,
496
+ state: &str,
497
+ redirect_uri: &str,
498
+ ) -> String {
499
+ let mut params = BTreeMap::new();
500
+ params.insert("response_type".to_owned(), "code".to_owned());
501
+ params.insert("client_id".to_owned(), oauth.client_id.clone());
502
+ params.insert("redirect_uri".to_owned(), redirect_uri.to_owned());
503
+ params.insert("scope".to_owned(), oauth.scope.clone());
504
+ params.insert("code_challenge".to_owned(), challenge.to_owned());
505
+ params.insert("code_challenge_method".to_owned(), "S256".to_owned());
506
+ params.insert("handoff".to_owned(), "url".to_owned());
507
+ params.insert("state".to_owned(), state.to_owned());
508
+
509
+ let query = encode_query(&params);
510
+ if oauth.authorization_endpoint.contains('?') {
511
+ format!("{}&{query}", oauth.authorization_endpoint)
512
+ } else {
513
+ format!("{}?{query}", oauth.authorization_endpoint)
514
+ }
515
+ }
516
+
517
+ async fn request_device_authorization(
518
+ http: &AuthHttpClient,
519
+ oauth: &RadiusOAuthConfig,
520
+ signal: Option<&CancellationToken>,
521
+ ) -> Result<DeviceAuthorizationResponse, AuthError> {
522
+ let response = http
523
+ .post_form(
524
+ &oauth.device_authorization_endpoint,
525
+ &form_fields([
526
+ ("client_id", oauth.client_id.as_str()),
527
+ ("scope", oauth.scope.as_str()),
528
+ ]),
529
+ None,
530
+ signal,
531
+ )
532
+ .await
533
+ .map_err(super::super::http::AuthHttpError::into_auth_error)?;
534
+
535
+ if !response.ok {
536
+ return Err(AuthError::message(
537
+ read_oauth_response_error(
538
+ response.status,
539
+ &response.body,
540
+ &response.raw_body,
541
+ "Radius OAuth device authorization failed",
542
+ )
543
+ .message,
544
+ ));
545
+ }
546
+
547
+ let device_code = response
548
+ .body
549
+ .get("device_code")
550
+ .and_then(Value::as_str)
551
+ .filter(|value| !value.is_empty())
552
+ .map(str::to_owned);
553
+ let user_code = response
554
+ .body
555
+ .get("user_code")
556
+ .and_then(Value::as_str)
557
+ .filter(|value| !value.is_empty())
558
+ .map(str::to_owned);
559
+ let expires_in = response
560
+ .body
561
+ .get("expires_in")
562
+ .and_then(json_u64)
563
+ .filter(|value| *value > 0);
564
+
565
+ let (Some(device_code), Some(user_code), Some(expires_in)) =
566
+ (device_code, user_code, expires_in)
567
+ else {
568
+ return Err(AuthError::message(
569
+ "Radius OAuth device authorization response is missing required fields",
570
+ ));
571
+ };
572
+
573
+ Ok(DeviceAuthorizationResponse {
574
+ device_code,
575
+ user_code,
576
+ verification_uri: response
577
+ .body
578
+ .get("verification_uri")
579
+ .and_then(Value::as_str)
580
+ .map(str::to_owned),
581
+ expires_in,
582
+ interval: response.body.get("interval").and_then(json_u64),
583
+ })
584
+ }
585
+
586
+ async fn request_oauth_token(
587
+ http: &AuthHttpClient,
588
+ oauth: &RadiusOAuthConfig,
589
+ body: BTreeMap<String, String>,
590
+ signal: Option<&CancellationToken>,
591
+ ) -> Result<OAuthCredential, OAuthResponseError> {
592
+ let response = http
593
+ .post_form(&oauth.token_endpoint, &body, None, signal)
594
+ .await
595
+ .map_err(|error| match error {
596
+ super::super::http::AuthHttpError::Cancelled => OAuthResponseError {
597
+ oauth_error: None,
598
+ message: "Login cancelled".to_owned(),
599
+ },
600
+ other => OAuthResponseError {
601
+ oauth_error: None,
602
+ message: other.to_string(),
603
+ },
604
+ })?;
605
+
606
+ if !response.ok {
607
+ return Err(read_oauth_response_error(
608
+ response.status,
609
+ &response.body,
610
+ &response.raw_body,
611
+ "Radius OAuth token request failed",
612
+ ));
613
+ }
614
+
615
+ credential_from_token_body(&response.body).map_err(|message| OAuthResponseError {
616
+ oauth_error: None,
617
+ message,
618
+ })
619
+ }
620
+
621
+ fn credential_from_token_body(body: &Value) -> Result<OAuthCredential, String> {
622
+ let access = body
623
+ .get("access_token")
624
+ .and_then(Value::as_str)
625
+ .filter(|value| !value.is_empty())
626
+ .ok_or_else(|| "Radius OAuth token response is missing access_token".to_owned())?;
627
+ let refresh = body
628
+ .get("refresh_token")
629
+ .and_then(Value::as_str)
630
+ .filter(|value| !value.is_empty())
631
+ .ok_or_else(|| "Radius OAuth token response is missing refresh_token".to_owned())?;
632
+ let expires_in = body
633
+ .get("expires_in")
634
+ .and_then(json_u64)
635
+ .ok_or_else(|| "Radius OAuth token response is missing expires_in".to_owned())?;
636
+
637
+ let mut extra = BTreeMap::new();
638
+ if let Some(scope) = body.get("scope").and_then(Value::as_str) {
639
+ extra.insert("scope".to_owned(), Value::String(scope.to_owned()));
640
+ }
641
+
642
+ Ok(OAuthCredential {
643
+ refresh: refresh.to_owned(),
644
+ access: access.to_owned(),
645
+ expires: now_ms()
646
+ .saturating_add(i64::try_from(expires_in.saturating_mul(1000)).unwrap_or(i64::MAX))
647
+ .saturating_sub(TOKEN_EXPIRY_SKEW_MS),
648
+ extra,
649
+ })
650
+ }
651
+
652
+ fn read_oauth_response_error(
653
+ status: u16,
654
+ body: &Value,
655
+ raw_body: &str,
656
+ message: &str,
657
+ ) -> OAuthResponseError {
658
+ let oauth_error = body.get("error").and_then(Value::as_str).map(str::to_owned);
659
+ let description = body
660
+ .get("error_description")
661
+ .and_then(Value::as_str)
662
+ .map(str::to_owned)
663
+ .or_else(|| {
664
+ if oauth_error.is_none() && !raw_body.trim().is_empty() {
665
+ Some(raw_body.to_owned())
666
+ } else {
667
+ None
668
+ }
669
+ });
670
+
671
+ let detail = match (&oauth_error, &description) {
672
+ (Some(error), Some(description)) => format!("{error}: {description}"),
673
+ (Some(error), None) => error.clone(),
674
+ (None, Some(description)) => description.clone(),
675
+ (None, None) => status.to_string(),
676
+ };
677
+
678
+ OAuthResponseError {
679
+ oauth_error,
680
+ message: format!("{message}: {detail}"),
681
+ }
682
+ }
683
+
684
+ fn map_token_error(error: OAuthResponseError) -> AuthError {
685
+ if error.message == "Login cancelled" {
686
+ AuthError::Cancelled
687
+ } else {
688
+ AuthError::message(error.message)
689
+ }
690
+ }
691
+
692
+ fn form_fields<'a, I>(pairs: I) -> BTreeMap<String, String>
693
+ where
694
+ I: IntoIterator<Item = (&'a str, &'a str)>,
695
+ {
696
+ pairs
697
+ .into_iter()
698
+ .map(|(key, value)| (key.to_owned(), value.to_owned()))
699
+ .collect()
700
+ }
701
+
702
+ fn encode_query(fields: &BTreeMap<String, String>) -> String {
703
+ let mut out = String::new();
704
+ for (index, (key, value)) in fields.iter().enumerate() {
705
+ if index > 0 {
706
+ out.push('&');
707
+ }
708
+ out.push_str(&urlencoding_encode(key));
709
+ out.push('=');
710
+ out.push_str(&urlencoding_encode(value));
711
+ }
712
+ out
713
+ }
714
+
715
+ fn urlencoding_encode(input: &str) -> String {
716
+ let mut out = String::with_capacity(input.len());
717
+ for byte in input.bytes() {
718
+ match byte {
719
+ b'A'..=b'Z' | b'a'..=b'z' | b'0'..=b'9' | b'-' | b'_' | b'.' | b'~' => {
720
+ out.push(char::from(byte));
721
+ }
722
+ _ => {
723
+ use std::fmt::Write as _;
724
+ let _ = write!(out, "%{byte:02X}");
725
+ }
726
+ }
727
+ }
728
+ out
729
+ }
730
+
731
+ fn json_u64(value: &Value) -> Option<u64> {
732
+ match value {
733
+ Value::Number(number) => number
734
+ .as_u64()
735
+ .or_else(|| number.as_i64().and_then(|n| u64::try_from(n).ok())),
736
+ Value::String(text) => text.trim().parse().ok(),
737
+ _ => None,
738
+ }
739
+ }
740
+
741
+ fn now_ms() -> i64 {
742
+ match SystemTime::now().duration_since(UNIX_EPOCH) {
743
+ Ok(duration) => i64::try_from(duration.as_millis()).unwrap_or(i64::MAX),
744
+ Err(_) => 0,
745
+ }
746
+ }
747
+
748
+ /// RFC 4122 UUID v4 string (same shape as `crypto.randomUUID()`).
749
+ fn random_uuid_v4() -> Result<String, AuthError> {
750
+ let mut bytes = [0_u8; 16];
751
+ getrandom::fill(&mut bytes)
752
+ .map_err(|error| AuthError::message(format!("failed to generate OAuth state: {error}")))?;
753
+ bytes[6] = (bytes[6] & 0x0f) | 0x40;
754
+ bytes[8] = (bytes[8] & 0x3f) | 0x80;
755
+ Ok(format!(
756
+ "{:02x}{:02x}{:02x}{:02x}-{:02x}{:02x}-{:02x}{:02x}-{:02x}{:02x}-{:02x}{:02x}{:02x}{:02x}{:02x}{:02x}",
757
+ bytes[0],
758
+ bytes[1],
759
+ bytes[2],
760
+ bytes[3],
761
+ bytes[4],
762
+ bytes[5],
763
+ bytes[6],
764
+ bytes[7],
765
+ bytes[8],
766
+ bytes[9],
767
+ bytes[10],
768
+ bytes[11],
769
+ bytes[12],
770
+ bytes[13],
771
+ bytes[14],
772
+ bytes[15],
773
+ ))
774
+ }
775
+
776
+ #[cfg(test)]
777
+ fn looks_like_uuid_v4(value: &str) -> bool {
778
+ let bytes = value.as_bytes();
779
+ bytes.len() == 36
780
+ && bytes[8] == b'-'
781
+ && bytes[13] == b'-'
782
+ && bytes[14] == b'4'
783
+ && bytes[18] == b'-'
784
+ && matches!(bytes[19], b'8' | b'9' | b'a' | b'b')
785
+ && bytes[23] == b'-'
786
+ && value.chars().enumerate().all(|(index, ch)| match index {
787
+ 8 | 13 | 18 | 23 => ch == '-',
788
+ _ => ch.is_ascii_hexdigit(),
789
+ })
790
+ }
791
+
792
+ #[cfg(test)]
793
+ mod tests {
794
+ use std::io::{Read, Write};
795
+ use std::net::TcpListener;
796
+ use std::pin::Pin;
797
+ use std::sync::{Arc, Mutex};
798
+ use std::thread;
799
+ use std::time::Duration;
800
+
801
+ use super::*;
802
+
803
+ type TestResult = Result<(), String>;
804
+
805
+ fn err(msg: impl Into<String>) -> String {
806
+ msg.into()
807
+ }
808
+
809
+ fn expect_err<T, E>(result: Result<T, E>, label: &str) -> Result<E, String> {
810
+ match result {
811
+ Ok(_) => Err(err(label)),
812
+ Err(error) => Ok(error),
813
+ }
814
+ }
815
+
816
+ fn lock_mutex<'a, T>(
817
+ mutex: &'a Mutex<T>,
818
+ label: &'static str,
819
+ ) -> Result<std::sync::MutexGuard<'a, T>, String> {
820
+ mutex
821
+ .lock()
822
+ .map_err(|_| err(format!("{label} lock poisoned")))
823
+ }
824
+
825
+ #[derive(Default)]
826
+ struct MockInteraction {
827
+ select: Mutex<Option<String>>,
828
+ events: Mutex<Vec<AuthEvent>>,
829
+ signal: Option<CancellationToken>,
830
+ }
831
+
832
+ impl MockInteraction {
833
+ fn browser() -> Self {
834
+ Self {
835
+ select: Mutex::new(Some(LOGIN_METHOD_BROWSER.to_owned())),
836
+ events: Mutex::new(Vec::new()),
837
+ signal: None,
838
+ }
839
+ }
840
+
841
+ fn device() -> Self {
842
+ Self {
843
+ select: Mutex::new(Some(LOGIN_METHOD_DEVICE_CODE.to_owned())),
844
+ events: Mutex::new(Vec::new()),
845
+ signal: None,
846
+ }
847
+ }
848
+
849
+ fn with_signal(mut self, signal: CancellationToken) -> Self {
850
+ self.signal = Some(signal);
851
+ self
852
+ }
853
+
854
+ fn events(&self) -> Result<Vec<AuthEvent>, String> {
855
+ lock_mutex(&self.events, "events").map(|guard| guard.clone())
856
+ }
857
+ }
858
+
859
+ impl AuthInteraction for MockInteraction {
860
+ fn prompt(&self, prompt: AuthPrompt) -> BoxFuture<'_, Result<String, AuthError>> {
861
+ Box::pin(async move {
862
+ match prompt {
863
+ AuthPrompt::Select { .. } => self
864
+ .select
865
+ .lock()
866
+ .map_err(|_| AuthError::message("select lock poisoned"))?
867
+ .clone()
868
+ .ok_or_else(|| AuthError::message("missing select")),
869
+ AuthPrompt::ManualCode { .. } => {
870
+ Err(AuthError::message("manual code not expected for radius"))
871
+ }
872
+ other => Err(AuthError::message(format!("unexpected prompt: {other:?}"))),
873
+ }
874
+ })
875
+ }
876
+
877
+ fn notify(&self, event: AuthEvent) {
878
+ if let Ok(mut events) = self.events.lock() {
879
+ events.push(event);
880
+ }
881
+ }
882
+
883
+ fn signal(&self) -> Option<CancellationToken> {
884
+ self.signal.clone()
885
+ }
886
+ }
887
+
888
+ type MockRoute = (String, u16, String, Option<String>);
889
+
890
+ fn sample_config(token_url: &str, device_url: &str, authorize_url: &str) -> Value {
891
+ serde_json::json!({
892
+ "issuer": "https://issuer.example",
893
+ "authorizationEndpoint": authorize_url,
894
+ "tokenEndpoint": token_url,
895
+ "deviceAuthorizationEndpoint": device_url,
896
+ "deviceAuthorizationEventsEndpoint": "https://issuer.example/events",
897
+ "verificationEndpoint": "https://issuer.example/device",
898
+ "clientId": "radius-client",
899
+ "scope": "openid profile offline_access",
900
+ "deviceCodeGrantType": "urn:ietf:params:oauth:grant-type:device_code"
901
+ })
902
+ }
903
+
904
+ fn spawn_json_server(routes: Arc<Mutex<Vec<MockRoute>>>) -> Result<String, String> {
905
+ let listener = TcpListener::bind("127.0.0.1:0").map_err(|e| err(e.to_string()))?;
906
+ let address = listener.local_addr().map_err(|e| err(e.to_string()))?;
907
+ thread::spawn(move || {
908
+ while let Ok((mut stream, _)) = listener.accept() {
909
+ let mut buf = [0_u8; 8192];
910
+ let n = stream.read(&mut buf).unwrap_or(0);
911
+ let request = String::from_utf8_lossy(&buf[..n]);
912
+ let first = request.lines().next().unwrap_or_default().to_owned();
913
+ let body_start = request.find("\r\n\r\n").map_or(n, |idx| idx + 4);
914
+ let req_body = request.get(body_start..).unwrap_or("").to_owned();
915
+
916
+ let Ok(mut routes) = routes.lock() else {
917
+ continue;
918
+ };
919
+ let Some(index) = routes
920
+ .iter()
921
+ .position(|(method_path, _, _, _)| first.starts_with(method_path))
922
+ else {
923
+ let response =
924
+ "HTTP/1.1 404 Not Found\r\nContent-Length: 0\r\nConnection: close\r\n\r\n";
925
+ let _ = stream.write_all(response.as_bytes());
926
+ continue;
927
+ };
928
+ let (_method_path, status, body, expected_body_substr) = routes.remove(index);
929
+ if let Some(expected) = expected_body_substr
930
+ && !req_body.contains(&expected)
931
+ {
932
+ continue;
933
+ }
934
+ let response = format!(
935
+ "HTTP/1.1 {status} OK\r\nContent-Length: {}\r\nContent-Type: application/json\r\nConnection: close\r\n\r\n{body}",
936
+ body.len()
937
+ );
938
+ let _ = stream.write_all(response.as_bytes());
939
+ }
940
+ });
941
+ Ok(format!("http://{address}"))
942
+ }
943
+
944
+ fn free_port() -> Result<u16, String> {
945
+ let listener = TcpListener::bind("127.0.0.1:0").map_err(|e| err(e.to_string()))?;
946
+ Ok(listener
947
+ .local_addr()
948
+ .map_err(|e| err(e.to_string()))?
949
+ .port())
950
+ }
951
+
952
+ fn radius_with(gateway: String, port: u16) -> Result<RadiusOAuth, String> {
953
+ Ok(RadiusOAuth::with_client(
954
+ RadiusOAuthOptions {
955
+ name: "Radius".into(),
956
+ gateway,
957
+ },
958
+ AuthHttpClient::new().map_err(|e| err(e.to_string()))?,
959
+ port,
960
+ ))
961
+ }
962
+
963
+ fn push_oauth_config(
964
+ routes: &Mutex<Vec<MockRoute>>,
965
+ token_url: &str,
966
+ device_url: &str,
967
+ authorize_url: &str,
968
+ ) -> TestResult {
969
+ let mut routes = lock_mutex(routes, "routes")?;
970
+ routes.push((
971
+ "GET /v1/oauth".to_owned(),
972
+ 200,
973
+ sample_config(token_url, device_url, authorize_url).to_string(),
974
+ None,
975
+ ));
976
+ Ok(())
977
+ }
978
+
979
+ fn browser_token_server() -> Result<String, String> {
980
+ let token_routes = Arc::new(Mutex::new(vec![(
981
+ "POST /token".to_owned(),
982
+ 200,
983
+ serde_json::json!({
984
+ "access_token": "browser-access",
985
+ "refresh_token": "browser-refresh",
986
+ "expires_in": 120,
987
+ "scope": "openid"
988
+ })
989
+ .to_string(),
990
+ Some("grant_type=authorization_code".to_owned()),
991
+ )]));
992
+ Ok(format!("{}/token", spawn_json_server(token_routes)?))
993
+ }
994
+
995
+ async fn wait_for_browser_auth_url<F>(
996
+ interaction: &MockInteraction,
997
+ login: &mut Pin<&mut F>,
998
+ ) -> Result<String, String>
999
+ where
1000
+ F: Future<Output = Result<OAuthCredential, AuthError>>,
1001
+ {
1002
+ // The login future must be polled so config discovery, callback bind, and
1003
+ // AuthUrl notification can run. Sleep-only waiting never advances it.
1004
+ loop {
1005
+ if let Some(url) = interaction
1006
+ .events()?
1007
+ .into_iter()
1008
+ .find_map(|event| match event {
1009
+ AuthEvent::AuthUrl { url, .. } => Some(url),
1010
+ _ => None,
1011
+ })
1012
+ {
1013
+ return Ok(url);
1014
+ }
1015
+ tokio::task::yield_now().await;
1016
+ match tokio::time::timeout(Duration::from_millis(10), login.as_mut()).await {
1017
+ Ok(Ok(_)) => return Err(err("login finished before callback")),
1018
+ Ok(Err(e)) => return Err(err(e.to_string())),
1019
+ Err(_) => {}
1020
+ }
1021
+ }
1022
+ }
1023
+
1024
+ #[test]
1025
+ fn normalize_gateway_adds_https_and_strips_slash() {
1026
+ assert_eq!(
1027
+ normalize_radius_gateway_url("radius.pi.dev/"),
1028
+ "https://radius.pi.dev"
1029
+ );
1030
+ assert_eq!(
1031
+ normalize_radius_gateway_url("http://localhost:8080///"),
1032
+ "http://localhost:8080"
1033
+ );
1034
+ assert_eq!(
1035
+ normalize_radius_gateway_url("https://gw.example"),
1036
+ "https://gw.example"
1037
+ );
1038
+ }
1039
+
1040
+ #[test]
1041
+ fn verification_destination_requires_https_except_loopback_development() -> TestResult {
1042
+ assert_eq!(
1043
+ validate_radius_verification_uri(
1044
+ "https://identity.example/device",
1045
+ "https://gateway.example",
1046
+ )
1047
+ .map_err(|error| error.to_string())?,
1048
+ "https://identity.example/device"
1049
+ );
1050
+ assert_eq!(
1051
+ validate_radius_verification_uri(
1052
+ "http://127.0.0.1:9000/device",
1053
+ "http://localhost:8000",
1054
+ )
1055
+ .map_err(|error| error.to_string())?,
1056
+ "http://127.0.0.1:9000/device"
1057
+ );
1058
+ for value in [
1059
+ "http://identity.example/device",
1060
+ "file:///etc/passwd",
1061
+ "javascript:alert(1)",
1062
+ "https://user@identity.example/device",
1063
+ ] {
1064
+ let error = expect_err(
1065
+ validate_radius_verification_uri(value, "https://gateway.example"),
1066
+ "untrusted verification destination",
1067
+ )?;
1068
+ assert_eq!(
1069
+ error.to_string(),
1070
+ "Untrusted verification URI in Radius OAuth response"
1071
+ );
1072
+ }
1073
+ assert!(
1074
+ validate_radius_verification_uri(
1075
+ "http://localhost:9000/device",
1076
+ "https://gateway.example",
1077
+ )
1078
+ .is_err()
1079
+ );
1080
+ Ok(())
1081
+ }
1082
+
1083
+ #[test]
1084
+ fn redirect_uri_defaults_to_fixed_radius_callback() {
1085
+ assert_eq!(radius_redirect_uri(CALLBACK_PORT), REDIRECT_URI);
1086
+ assert_eq!(
1087
+ radius_redirect_uri(3456),
1088
+ "http://127.0.0.1:3456/oauth/callback"
1089
+ );
1090
+ assert_eq!(CALLBACK_HOST, IpAddr::V4(Ipv4Addr::LOCALHOST));
1091
+ assert!(!CALLBACK_HOST.is_unspecified());
1092
+ }
1093
+
1094
+ #[test]
1095
+ fn authorize_url_includes_pkce_uuid_state_and_handoff() -> TestResult {
1096
+ let config = RadiusOAuthConfig {
1097
+ issuer: "https://issuer".into(),
1098
+ authorization_endpoint: "https://issuer/authorize".into(),
1099
+ token_endpoint: "https://issuer/token".into(),
1100
+ device_authorization_endpoint: "https://issuer/device".into(),
1101
+ device_authorization_events_endpoint: String::new(),
1102
+ verification_endpoint: "https://issuer/verify".into(),
1103
+ client_id: "client".into(),
1104
+ scope: "openid".into(),
1105
+ device_code_grant_type: "urn:ietf:params:oauth:grant-type:device_code".into(),
1106
+ };
1107
+ let state = random_uuid_v4().map_err(|e| err(e.to_string()))?;
1108
+ let url = build_authorize_url(&config, "challenge", &state, REDIRECT_URI);
1109
+ assert!(url.starts_with("https://issuer/authorize?"));
1110
+ assert!(url.contains("response_type=code"));
1111
+ assert!(url.contains("client_id=client"));
1112
+ assert!(url.contains("code_challenge=challenge"));
1113
+ assert!(url.contains("code_challenge_method=S256"));
1114
+ assert!(url.contains("handoff=url"));
1115
+ assert!(url.contains(&format!("state={state}")));
1116
+ assert!(url.contains("redirect_uri=http%3A%2F%2F127.0.0.1%3A1456%2Foauth%2Fcallback"));
1117
+ // UUID state is distinct from PKCE challenge/verifier.
1118
+ assert_ne!(state, "challenge");
1119
+ assert!(looks_like_uuid_v4(&state));
1120
+ Ok(())
1121
+ }
1122
+
1123
+ #[test]
1124
+ fn credential_from_token_preserves_scope_extra_and_skew() -> TestResult {
1125
+ let before = now_ms();
1126
+ let credential = credential_from_token_body(&serde_json::json!({
1127
+ "access_token": "access-1",
1128
+ "refresh_token": "refresh-1",
1129
+ "expires_in": 3600,
1130
+ "scope": "openid profile"
1131
+ }))
1132
+ .map_err(err)?;
1133
+ let after = now_ms();
1134
+ assert_eq!(credential.access, "access-1");
1135
+ assert_eq!(credential.refresh, "refresh-1");
1136
+ assert_eq!(
1137
+ credential.extra.get("scope"),
1138
+ Some(&Value::String("openid profile".into()))
1139
+ );
1140
+ let expected_low = before + 3600 * 1000 - TOKEN_EXPIRY_SKEW_MS;
1141
+ let expected_high = after + 3600 * 1000 - TOKEN_EXPIRY_SKEW_MS;
1142
+ assert!(credential.expires >= expected_low);
1143
+ assert!(credential.expires <= expected_high);
1144
+
1145
+ let encoded =
1146
+ serde_json::to_value(crate::auth::types::Credential::Oauth(credential.clone()))
1147
+ .map_err(|e| err(e.to_string()))?;
1148
+ assert_eq!(encoded["type"], "oauth");
1149
+ assert_eq!(encoded["scope"], "openid profile");
1150
+ assert!(encoded.get("gatewayConfig").is_none());
1151
+ Ok(())
1152
+ }
1153
+
1154
+ #[tokio::test]
1155
+ async fn config_discovery_loads_camel_case_endpoints() -> TestResult {
1156
+ let routes = Arc::new(Mutex::new(vec![(
1157
+ "GET /v1/oauth".to_owned(),
1158
+ 200,
1159
+ sample_config(
1160
+ "https://issuer/token",
1161
+ "https://issuer/device",
1162
+ "https://issuer/authorize",
1163
+ )
1164
+ .to_string(),
1165
+ None,
1166
+ )]));
1167
+ let gateway = spawn_json_server(routes)?;
1168
+ let http = AuthHttpClient::new().map_err(|e| err(e.to_string()))?;
1169
+ let config = load_radius_oauth_config(&http, &gateway, None)
1170
+ .await
1171
+ .map_err(|e| err(e.to_string()))?;
1172
+ assert_eq!(config.client_id, "radius-client");
1173
+ assert_eq!(config.token_endpoint, "https://issuer/token");
1174
+ assert_eq!(
1175
+ config.device_code_grant_type,
1176
+ "urn:ietf:params:oauth:grant-type:device_code"
1177
+ );
1178
+ assert_eq!(config.scope, "openid profile offline_access");
1179
+ Ok(())
1180
+ }
1181
+
1182
+ #[tokio::test]
1183
+ async fn config_discovery_error_includes_status_and_body() -> TestResult {
1184
+ let routes = Arc::new(Mutex::new(vec![(
1185
+ "GET /v1/oauth".to_owned(),
1186
+ 503,
1187
+ "gateway down".to_owned(),
1188
+ None,
1189
+ )]));
1190
+ let gateway = spawn_json_server(routes)?;
1191
+ let http = AuthHttpClient::new().map_err(|e| err(e.to_string()))?;
1192
+ let err_value = expect_err(
1193
+ load_radius_oauth_config(&http, &gateway, None).await,
1194
+ "should fail",
1195
+ )?;
1196
+ let message = err_value.to_string();
1197
+ assert!(message.contains("Could not load Radius OAuth config from"));
1198
+ assert!(message.contains("503"));
1199
+ assert!(message.contains("gateway down"));
1200
+ Ok(())
1201
+ }
1202
+
1203
+ #[tokio::test]
1204
+ async fn browser_callback_rejects_wrong_state() -> TestResult {
1205
+ let port = free_port()?;
1206
+ let routes = Arc::new(Mutex::new(Vec::new()));
1207
+ let gateway = spawn_json_server(routes.clone())?;
1208
+ let token_url = browser_token_server()?;
1209
+ push_oauth_config(
1210
+ &routes,
1211
+ &token_url,
1212
+ "https://issuer/device",
1213
+ "https://issuer/authorize",
1214
+ )?;
1215
+ let oauth = radius_with(gateway, port)?;
1216
+ let interaction = MockInteraction::browser();
1217
+ let login = oauth.login(&interaction);
1218
+ tokio::pin!(login);
1219
+ let auth_url = wait_for_browser_auth_url(&interaction, &mut login).await?;
1220
+ let state = auth_url
1221
+ .split('&')
1222
+ .find_map(|part| part.strip_prefix("state="))
1223
+ .ok_or_else(|| err("state"))?
1224
+ .to_owned();
1225
+ assert!(looks_like_uuid_v4(&state));
1226
+ assert!(auth_url.contains("handoff=url"));
1227
+
1228
+ let client = reqwest::Client::new();
1229
+ let bad = client
1230
+ .get(format!(
1231
+ "http://127.0.0.1:{port}{CALLBACK_PATH}?code=bad&state=not-the-state"
1232
+ ))
1233
+ .send()
1234
+ .await
1235
+ .map_err(|e| err(e.to_string()))?;
1236
+ assert_eq!(bad.status(), reqwest::StatusCode::BAD_REQUEST);
1237
+ let bad_html = bad.text().await.map_err(|e| err(e.to_string()))?;
1238
+ assert!(
1239
+ bad_html.contains("State mismatch") || bad_html.contains("state"),
1240
+ "{bad_html}"
1241
+ );
1242
+ Ok(())
1243
+ }
1244
+
1245
+ #[tokio::test]
1246
+ async fn browser_callback_exchanges_code_and_to_auth() -> TestResult {
1247
+ let port = free_port()?;
1248
+ let routes = Arc::new(Mutex::new(Vec::new()));
1249
+ let gateway = spawn_json_server(routes.clone())?;
1250
+ let token_url = browser_token_server()?;
1251
+ push_oauth_config(
1252
+ &routes,
1253
+ &token_url,
1254
+ "https://issuer/device",
1255
+ "https://issuer/authorize",
1256
+ )?;
1257
+ let oauth = radius_with(gateway, port)?;
1258
+ let interaction = MockInteraction::browser();
1259
+ let login = oauth.login(&interaction);
1260
+ tokio::pin!(login);
1261
+ let auth_url = wait_for_browser_auth_url(&interaction, &mut login).await?;
1262
+ let state = auth_url
1263
+ .split('&')
1264
+ .find_map(|part| part.strip_prefix("state="))
1265
+ .ok_or_else(|| err("state"))?
1266
+ .to_owned();
1267
+
1268
+ let client = reqwest::Client::new();
1269
+ let good = client
1270
+ .get(format!(
1271
+ "http://127.0.0.1:{port}{CALLBACK_PATH}?code=auth-code&state={state}"
1272
+ ))
1273
+ .send()
1274
+ .await
1275
+ .map_err(|e| err(e.to_string()))?;
1276
+ assert_eq!(good.status(), reqwest::StatusCode::OK);
1277
+
1278
+ let credential = match tokio::time::timeout(Duration::from_secs(3), login).await {
1279
+ Ok(Ok(value)) => value,
1280
+ Ok(Err(e)) => return Err(err(e.to_string())),
1281
+ Err(_) => return Err(err("login timeout")),
1282
+ };
1283
+ assert_eq!(credential.access, "browser-access");
1284
+ assert_eq!(credential.refresh, "browser-refresh");
1285
+ assert_eq!(
1286
+ credential.extra.get("scope"),
1287
+ Some(&Value::String("openid".into()))
1288
+ );
1289
+
1290
+ let model_auth = oauth
1291
+ .to_auth(&credential)
1292
+ .await
1293
+ .map_err(|e| err(e.to_string()))?;
1294
+ assert_eq!(model_auth.api_key.as_deref(), Some("browser-access"));
1295
+ assert!(model_auth.base_url.is_none());
1296
+ assert!(model_auth.headers.is_none());
1297
+
1298
+ let events = interaction.events()?;
1299
+ assert!(events.iter().any(|event| matches!(
1300
+ event,
1301
+ AuthEvent::Progress { message } if message.contains("Listening for OAuth callback")
1302
+ )));
1303
+ assert!(events.iter().any(|event| matches!(
1304
+ event,
1305
+ AuthEvent::AuthUrl {
1306
+ instructions: Some(text),
1307
+ ..
1308
+ } if text == "Continue in your browser."
1309
+ )));
1310
+ Ok(())
1311
+ }
1312
+
1313
+ #[tokio::test]
1314
+ async fn browser_state_mismatch_does_not_settle_callback() -> TestResult {
1315
+ let port = free_port()?;
1316
+ let server = OAuthCallbackServer::start(OAuthCallbackConfig {
1317
+ port,
1318
+ path: CALLBACK_PATH.into(),
1319
+ expected_state: "expected-uuid-state".into(),
1320
+ success_message: "Signed in to Radius. You may now close this page.".into(),
1321
+ host: Some(CALLBACK_HOST),
1322
+ })
1323
+ .await
1324
+ .map_err(|e| err(e.to_string()))?;
1325
+
1326
+ let client = reqwest::Client::new();
1327
+ let bad = client
1328
+ .get(format!(
1329
+ "http://127.0.0.1:{port}{CALLBACK_PATH}?code=x&state=other"
1330
+ ))
1331
+ .send()
1332
+ .await
1333
+ .map_err(|e| err(e.to_string()))?;
1334
+ assert_eq!(bad.status(), reqwest::StatusCode::BAD_REQUEST);
1335
+
1336
+ let wait = server.wait_for_code();
1337
+ let timed_out = tokio::time::timeout(Duration::from_millis(100), wait).await;
1338
+ assert!(timed_out.is_err(), "bad state must not settle wait");
1339
+ server.close().await;
1340
+ Ok(())
1341
+ }
1342
+
1343
+ fn device_issuer_routes(
1344
+ token_body_initial: String,
1345
+ token_body_refresh: String,
1346
+ ) -> Vec<MockRoute> {
1347
+ vec![
1348
+ (
1349
+ "POST /device".to_owned(),
1350
+ 200,
1351
+ serde_json::json!({
1352
+ "device_code": "dev-code",
1353
+ "user_code": "ABCD-EFGH",
1354
+ "verification_uri": "https://issuer.example/device",
1355
+ "expires_in": 600,
1356
+ "interval": 1
1357
+ })
1358
+ .to_string(),
1359
+ Some("client_id=radius-client".to_owned()),
1360
+ ),
1361
+ (
1362
+ "POST /token".to_owned(),
1363
+ 400,
1364
+ serde_json::json!({
1365
+ "error": "authorization_pending"
1366
+ })
1367
+ .to_string(),
1368
+ Some(
1369
+ "grant_type=urn%3Aietf%3Aparams%3Aoauth%3Agrant-type%3Adevice_code".to_owned(),
1370
+ ),
1371
+ ),
1372
+ (
1373
+ "POST /token".to_owned(),
1374
+ 200,
1375
+ token_body_initial,
1376
+ Some("device_code=dev-code".to_owned()),
1377
+ ),
1378
+ (
1379
+ "POST /token".to_owned(),
1380
+ 200,
1381
+ token_body_refresh,
1382
+ Some("grant_type=refresh_token".to_owned()),
1383
+ ),
1384
+ ]
1385
+ }
1386
+
1387
+ #[tokio::test]
1388
+ async fn device_login_exchange_completes_with_scope() -> TestResult {
1389
+ let token_body_initial = serde_json::json!({
1390
+ "access_token": "device-access",
1391
+ "refresh_token": "device-refresh",
1392
+ "expires_in": 90,
1393
+ "scope": "openid profile"
1394
+ })
1395
+ .to_string();
1396
+ let token_body_refresh = serde_json::json!({
1397
+ "access_token": "refreshed-access",
1398
+ "refresh_token": "refreshed-refresh",
1399
+ "expires_in": 90,
1400
+ "scope": "openid profile offline"
1401
+ })
1402
+ .to_string();
1403
+
1404
+ let routes = Arc::new(Mutex::new(Vec::new()));
1405
+ let gateway = spawn_json_server(routes.clone())?;
1406
+ let issuer = spawn_json_server(Arc::new(Mutex::new(device_issuer_routes(
1407
+ token_body_initial,
1408
+ token_body_refresh,
1409
+ ))))?;
1410
+ let device_url = format!("{issuer}/device");
1411
+ let token_url = format!("{issuer}/token");
1412
+ {
1413
+ let mut routes = lock_mutex(&routes, "routes")?;
1414
+ for _ in 0..2 {
1415
+ routes.push((
1416
+ "GET /v1/oauth".to_owned(),
1417
+ 200,
1418
+ sample_config(&token_url, &device_url, "https://issuer/authorize").to_string(),
1419
+ None,
1420
+ ));
1421
+ }
1422
+ }
1423
+
1424
+ let oauth = radius_with(gateway, free_port()?)?;
1425
+ let interaction = MockInteraction::device();
1426
+ let credential = oauth
1427
+ .login(&interaction)
1428
+ .await
1429
+ .map_err(|e| err(e.to_string()))?;
1430
+ assert_eq!(credential.access, "device-access");
1431
+ assert_eq!(credential.refresh, "device-refresh");
1432
+ assert_eq!(
1433
+ credential.extra.get("scope"),
1434
+ Some(&Value::String("openid profile".into()))
1435
+ );
1436
+ let events = interaction.events()?;
1437
+ assert!(events.iter().any(|event| matches!(
1438
+ event,
1439
+ AuthEvent::DeviceCode {
1440
+ user_code,
1441
+ verification_uri,
1442
+ interval_seconds: Some(1),
1443
+ expires_in_seconds: Some(600),
1444
+ } if user_code == "ABCD-EFGH" && verification_uri == "https://issuer.example/device"
1445
+ )));
1446
+ Ok(())
1447
+ }
1448
+
1449
+ #[tokio::test]
1450
+ async fn device_login_rejects_untrusted_verification_destination() -> TestResult {
1451
+ let routes = Arc::new(Mutex::new(Vec::new()));
1452
+ let gateway = spawn_json_server(routes.clone())?;
1453
+ let issuer_routes = Arc::new(Mutex::new(vec![(
1454
+ "POST /device".to_owned(),
1455
+ 200,
1456
+ serde_json::json!({
1457
+ "device_code": "dev-code",
1458
+ "user_code": "ABCD-EFGH",
1459
+ "verification_uri": "http://attacker.example/phish",
1460
+ "expires_in": 600,
1461
+ "interval": 1
1462
+ })
1463
+ .to_string(),
1464
+ Some("client_id=radius-client".to_owned()),
1465
+ )]));
1466
+ let issuer = spawn_json_server(issuer_routes)?;
1467
+ let device_url = format!("{issuer}/device");
1468
+ {
1469
+ let mut routes = lock_mutex(&routes, "routes")?;
1470
+ routes.push((
1471
+ "GET /v1/oauth".to_owned(),
1472
+ 200,
1473
+ sample_config(
1474
+ "https://issuer.example/token",
1475
+ &device_url,
1476
+ "https://issuer.example/authorize",
1477
+ )
1478
+ .to_string(),
1479
+ None,
1480
+ ));
1481
+ }
1482
+
1483
+ let oauth = radius_with(gateway, free_port()?)?;
1484
+ let interaction = MockInteraction::device();
1485
+ let error = expect_err(oauth.login(&interaction).await, "untrusted destination")?;
1486
+ assert_eq!(
1487
+ error.to_string(),
1488
+ "Untrusted verification URI in Radius OAuth response"
1489
+ );
1490
+ assert!(
1491
+ !interaction
1492
+ .events()?
1493
+ .iter()
1494
+ .any(|event| matches!(event, AuthEvent::DeviceCode { .. }))
1495
+ );
1496
+ Ok(())
1497
+ }
1498
+
1499
+ #[tokio::test]
1500
+ async fn device_refresh_rotates_tokens_and_scope() -> TestResult {
1501
+ let token_body_initial = serde_json::json!({
1502
+ "access_token": "device-access",
1503
+ "refresh_token": "device-refresh",
1504
+ "expires_in": 90,
1505
+ "scope": "openid profile"
1506
+ })
1507
+ .to_string();
1508
+ let token_body_refresh = serde_json::json!({
1509
+ "access_token": "refreshed-access",
1510
+ "refresh_token": "refreshed-refresh",
1511
+ "expires_in": 90,
1512
+ "scope": "openid profile offline"
1513
+ })
1514
+ .to_string();
1515
+
1516
+ let routes = Arc::new(Mutex::new(Vec::new()));
1517
+ let gateway = spawn_json_server(routes.clone())?;
1518
+ let issuer = spawn_json_server(Arc::new(Mutex::new(device_issuer_routes(
1519
+ token_body_initial,
1520
+ token_body_refresh,
1521
+ ))))?;
1522
+ let device_url = format!("{issuer}/device");
1523
+ let token_url = format!("{issuer}/token");
1524
+ {
1525
+ let mut routes = lock_mutex(&routes, "routes")?;
1526
+ for _ in 0..2 {
1527
+ routes.push((
1528
+ "GET /v1/oauth".to_owned(),
1529
+ 200,
1530
+ sample_config(&token_url, &device_url, "https://issuer/authorize").to_string(),
1531
+ None,
1532
+ ));
1533
+ }
1534
+ }
1535
+
1536
+ let oauth = radius_with(gateway, free_port()?)?;
1537
+ let interaction = MockInteraction::device();
1538
+ let credential = oauth
1539
+ .login(&interaction)
1540
+ .await
1541
+ .map_err(|e| err(e.to_string()))?;
1542
+ let refreshed = oauth
1543
+ .refresh(&credential, None)
1544
+ .await
1545
+ .map_err(|e| err(e.to_string()))?;
1546
+ assert_eq!(refreshed.access, "refreshed-access");
1547
+ assert_eq!(refreshed.refresh, "refreshed-refresh");
1548
+ assert_eq!(
1549
+ refreshed.extra.get("scope"),
1550
+ Some(&Value::String("openid profile offline".into()))
1551
+ );
1552
+ Ok(())
1553
+ }
1554
+
1555
+ #[tokio::test]
1556
+ async fn device_access_denied_error() -> TestResult {
1557
+ let routes = Arc::new(Mutex::new(Vec::new()));
1558
+ let gateway = spawn_json_server(routes.clone())?;
1559
+ let device_routes = Arc::new(Mutex::new(vec![
1560
+ (
1561
+ "POST /device".to_owned(),
1562
+ 200,
1563
+ serde_json::json!({
1564
+ "device_code": "dev-code",
1565
+ "user_code": "ZZZZ",
1566
+ "expires_in": 30,
1567
+ "interval": 1
1568
+ })
1569
+ .to_string(),
1570
+ None,
1571
+ ),
1572
+ (
1573
+ "POST /token".to_owned(),
1574
+ 400,
1575
+ serde_json::json!({
1576
+ "error": "access_denied",
1577
+ "error_description": "user said no"
1578
+ })
1579
+ .to_string(),
1580
+ None,
1581
+ ),
1582
+ ]));
1583
+ let issuer = spawn_json_server(device_routes)?;
1584
+ {
1585
+ let mut routes = lock_mutex(&routes, "routes")?;
1586
+ routes.push((
1587
+ "GET /v1/oauth".to_owned(),
1588
+ 200,
1589
+ sample_config(
1590
+ &format!("{issuer}/token"),
1591
+ &format!("{issuer}/device"),
1592
+ "https://issuer/authorize",
1593
+ )
1594
+ .to_string(),
1595
+ None,
1596
+ ));
1597
+ }
1598
+
1599
+ let oauth = radius_with(gateway, free_port()?)?;
1600
+ let err_value = expect_err(oauth.login(&MockInteraction::device()).await, "denied")?;
1601
+ assert_eq!(err_value.to_string(), "Device authorization was denied.");
1602
+ Ok(())
1603
+ }
1604
+
1605
+ #[tokio::test]
1606
+ async fn device_cancel_before_config_load() -> TestResult {
1607
+ let routes = Arc::new(Mutex::new(Vec::new()));
1608
+ let gateway = spawn_json_server(routes.clone())?;
1609
+ let issuer = spawn_json_server(Arc::new(Mutex::new(Vec::new())))?;
1610
+ {
1611
+ let mut routes = lock_mutex(&routes, "routes")?;
1612
+ routes.push((
1613
+ "GET /v1/oauth".to_owned(),
1614
+ 200,
1615
+ sample_config(
1616
+ &format!("{issuer}/token"),
1617
+ &format!("{issuer}/device"),
1618
+ "https://issuer/authorize",
1619
+ )
1620
+ .to_string(),
1621
+ None,
1622
+ ));
1623
+ }
1624
+ let oauth = radius_with(gateway, free_port()?)?;
1625
+ let token = CancellationToken::new();
1626
+ token.cancel();
1627
+ let cancelled = expect_err(
1628
+ oauth
1629
+ .login(&MockInteraction::device().with_signal(token))
1630
+ .await,
1631
+ "cancelled",
1632
+ )?;
1633
+ assert!(matches!(cancelled, AuthError::Cancelled));
1634
+ assert_eq!(cancelled.to_string(), "Login cancelled");
1635
+ Ok(())
1636
+ }
1637
+
1638
+ #[tokio::test]
1639
+ async fn unknown_login_method_errors_with_provider_name() -> TestResult {
1640
+ let routes = Arc::new(Mutex::new(vec![(
1641
+ "GET /v1/oauth".to_owned(),
1642
+ 200,
1643
+ sample_config(
1644
+ "https://issuer/token",
1645
+ "https://issuer/device",
1646
+ "https://issuer/authorize",
1647
+ )
1648
+ .to_string(),
1649
+ None,
1650
+ )]));
1651
+ let gateway = spawn_json_server(routes)?;
1652
+ let oauth = radius_with(gateway, free_port()?)?;
1653
+ let interaction = MockInteraction {
1654
+ select: Mutex::new(Some("sms".into())),
1655
+ events: Mutex::new(Vec::new()),
1656
+ signal: None,
1657
+ };
1658
+ let err_value = expect_err(oauth.login(&interaction).await, "unknown")?;
1659
+ assert_eq!(err_value.to_string(), "Unknown Radius sign-in method: sms");
1660
+ Ok(())
1661
+ }
1662
+
1663
+ #[tokio::test]
1664
+ async fn soft_failed_callback_reports_incomplete() -> TestResult {
1665
+ let hold_port = free_port()?;
1666
+ let _hold =
1667
+ TcpListener::bind(format!("127.0.0.1:{hold_port}")).map_err(|e| err(e.to_string()))?;
1668
+ let routes = Arc::new(Mutex::new(vec![(
1669
+ "GET /v1/oauth".to_owned(),
1670
+ 200,
1671
+ sample_config(
1672
+ "https://issuer/token",
1673
+ "https://issuer/device",
1674
+ "https://issuer/authorize",
1675
+ )
1676
+ .to_string(),
1677
+ None,
1678
+ )]));
1679
+ let gateway = spawn_json_server(routes)?;
1680
+ let oauth = radius_with(gateway, hold_port)?;
1681
+ let err_value = expect_err(oauth.login(&MockInteraction::browser()).await, "soft fail")?;
1682
+ assert_eq!(err_value.to_string(), "OAuth callback did not complete.");
1683
+ Ok(())
1684
+ }
1685
+ }