1use indexmap::IndexMap;
9use lex_ast::canonicalize_program;
10use lex_bytecode::{compile_program, vm::Vm, Value};
11use lex_runtime::{check_program as check_policy, DefaultHandler, Policy};
12use lex_store::Store;
13use crate::publish_examples::record_examples_for_publish;
14use lex_syntax::{load_package, load_program_from_str, Manifest};
15use lex_vcs::{MergeSession, MergeSessionId};
16use serde::{Deserialize, Serialize};
17use std::collections::{BTreeMap, BTreeSet, HashMap};
18use std::path::PathBuf;
19use std::sync::{Arc, Mutex};
20use std::time::{SystemTime, UNIX_EPOCH};
21use tiny_http::{Header, Method, Request, Response};
22
23fn stage_fns(stages: &[lex_ast::Stage]) -> BTreeMap<String, lex_ast::FnDecl> {
25 stages.iter().filter_map(|s| match s {
26 lex_ast::Stage::FnDecl(fd) => Some((fd.name.clone(), fd.clone())),
27 _ => None,
28 }).collect()
29}
30
31fn stage_types(stages: &[lex_ast::Stage]) -> BTreeMap<String, lex_ast::TypeDecl> {
34 stages.iter().filter_map(|s| match s {
35 lex_ast::Stage::TypeDecl(td) => Some((td.name.clone(), td.clone())),
36 _ => None,
37 }).collect()
38}
39
40pub struct State {
41 pub store: Mutex<Store>,
42 pub root: PathBuf,
47 pub sessions: Mutex<HashMap<MergeSessionId, ApiMergeSession>>,
54 pub policy_ceiling: Option<Policy>,
75}
76
77pub struct ApiMergeSession {
84 pub inner: MergeSession,
85 pub src_branch: String,
86 pub dst_branch: String,
87}
88
89impl State {
90 pub fn open(root: PathBuf) -> anyhow::Result<Self> {
91 Self::open_with_ceiling(root, None)
92 }
93
94 pub fn open_with_ceiling(
99 root: PathBuf,
100 policy_ceiling: Option<Policy>,
101 ) -> anyhow::Result<Self> {
102 Ok(Self {
103 store: Mutex::new(Store::open(&root)?),
104 root,
105 sessions: Mutex::new(HashMap::new()),
106 policy_ceiling,
107 })
108 }
109
110 pub fn new_with_tenant(tenant_id: &str, store_root: PathBuf) -> anyhow::Result<Self> {
120 validate_tenant_id(tenant_id)?;
121 Self::open(store_root.join(tenant_id))
122 }
123
124 pub fn new_with_tenant_and_ceiling(
129 tenant_id: &str,
130 store_root: PathBuf,
131 policy_ceiling: Option<Policy>,
132 ) -> anyhow::Result<Self> {
133 validate_tenant_id(tenant_id)?;
134 Self::open_with_ceiling(store_root.join(tenant_id), policy_ceiling)
135 }
136}
137
138fn clamp_policy(requested: Policy, ceiling: &Policy) -> Policy {
154 let allow_effects: BTreeSet<String> = requested
155 .allow_effects
156 .intersection(&ceiling.allow_effects)
157 .cloned()
158 .collect();
159 let budget = match (requested.budget, ceiling.budget) {
160 (Some(r), Some(c)) => Some(r.min(c)),
161 (None, Some(c)) => Some(c),
162 (Some(r), None) => Some(r),
163 (None, None) => None,
164 };
165 Policy {
166 allow_effects,
167 allow_fs_read: ceiling.allow_fs_read.clone(),
168 allow_fs_write: ceiling.allow_fs_write.clone(),
169 allow_net_host: ceiling.allow_net_host.clone(),
170 allow_proc: ceiling.allow_proc.clone(),
171 allow_approval: ceiling.allow_approval.clone(),
172 budget,
173 }
174}
175
176fn validate_tenant_id(tenant_id: &str) -> anyhow::Result<()> {
177 if tenant_id.is_empty() {
178 anyhow::bail!("tenant_id must not be empty");
179 }
180 if tenant_id.len() > 64 {
181 anyhow::bail!("tenant_id must be at most 64 bytes");
182 }
183 if !tenant_id
184 .bytes()
185 .all(|b| b.is_ascii_alphanumeric() || b == b'_' || b == b'-')
186 {
187 anyhow::bail!(
188 "tenant_id {tenant_id:?} contains characters outside [A-Za-z0-9_-]"
189 );
190 }
191 Ok(())
192}
193
194#[derive(Debug, Serialize, Deserialize)]
195struct ErrorEnvelope {
196 error: String,
197 #[serde(skip_serializing_if = "Option::is_none")]
198 detail: Option<serde_json::Value>,
199}
200
201pub(crate) fn json_response(status: u16, body: &serde_json::Value) -> Response<std::io::Cursor<Vec<u8>>> {
202 let bytes = serde_json::to_vec(body).unwrap_or_else(|_| b"{}".to_vec());
203 Response::from_data(bytes)
204 .with_status_code(status)
205 .with_header(Header::from_bytes(&b"Content-Type"[..], &b"application/json"[..]).unwrap())
206}
207
208pub(crate) fn error_response(status: u16, msg: impl Into<String>) -> Response<std::io::Cursor<Vec<u8>>> {
209 json_response(status, &serde_json::to_value(ErrorEnvelope {
210 error: msg.into(), detail: None,
211 }).unwrap())
212}
213
214pub(crate) fn error_with_detail(status: u16, msg: impl Into<String>, detail: serde_json::Value)
215 -> Response<std::io::Cursor<Vec<u8>>>
216{
217 json_response(status, &serde_json::to_value(ErrorEnvelope {
218 error: msg.into(), detail: Some(detail),
219 }).unwrap())
220}
221
222fn write_error_response(prefix: &str, err: lex_store::StoreError)
228 -> Response<std::io::Cursor<Vec<u8>>>
229{
230 if let lex_store::StoreError::Contention { branch, attempts } = &err {
231 let body = serde_json::to_vec(&ErrorEnvelope {
232 error: format!("{prefix}: branch '{branch}' is contended (attempts={attempts})"),
233 detail: Some(serde_json::json!({
234 "kind": "contention",
235 "branch": branch,
236 "attempts": attempts,
237 })),
238 }).unwrap_or_else(|_| b"{}".to_vec());
239 return Response::from_data(body)
240 .with_status_code(503)
241 .with_header(Header::from_bytes(&b"Content-Type"[..], &b"application/json"[..]).unwrap())
242 .with_header(Header::from_bytes(&b"Retry-After"[..], &b"1"[..]).unwrap());
243 }
244 if let lex_store::StoreError::BudgetExceeded { session_id, cap, spent_after } = &err {
252 let body = serde_json::to_vec(&ErrorEnvelope {
253 error: format!(
254 "{prefix}: session `{session_id}` budget exceeded \
255 (spent_after={spent_after}, cap={cap})"
256 ),
257 detail: Some(serde_json::json!({
258 "kind": "budget_exceeded",
259 "session_id": session_id,
260 "cap": cap,
261 "spent_after": spent_after,
262 })),
263 }).unwrap_or_else(|_| b"{}".to_vec());
264 return Response::from_data(body)
265 .with_status_code(503)
266 .with_header(Header::from_bytes(&b"Content-Type"[..], &b"application/json"[..]).unwrap())
267 .with_header(Header::from_bytes(&b"Retry-After"[..], &b"0"[..]).unwrap());
268 }
269 error_response(500, format!("{prefix}: {err}"))
270}
271
272pub fn handle(state: Arc<State>, mut req: Request) -> std::io::Result<()> {
273 let method = req.method().clone();
274 let url = req.url().to_string();
275 let path = url.split('?').next().unwrap_or("").to_string();
276 let query = url.split_once('?').map(|(_, q)| q.to_string()).unwrap_or_default();
277
278 let x_lex_user = req.headers().iter()
283 .find(|h| h.field.equiv("x-lex-user"))
284 .map(|h| h.value.as_str().to_string());
285
286 if matches!(method, Method::Post) && path == "/v1/pkg/publish" {
288 let mut body_bytes: Vec<u8> = Vec::new();
289 let _ = req.as_reader().read_to_end(&mut body_bytes);
290 let resp = pkg_publish_handler(&state, &body_bytes);
291 return req.respond(resp);
292 }
293
294 let mut body = String::new();
295 let _ = req.as_reader().read_to_string(&mut body);
296
297 let resp = route(&state, &method, &path, &query, &body, x_lex_user.as_deref());
298 req.respond(resp)
299}
300
301pub fn handle_with_auth<F>(state: Arc<State>, req: Request, auth: F) -> std::io::Result<()>
305where
306 F: FnOnce(&str, &[Header]) -> bool,
307{
308 let path = req.url().split('?').next().unwrap_or("").to_string();
309 if !auth(&path, req.headers()) {
310 return req.respond(
311 Response::from_data(br#"{"error":"unauthorized"}"#.to_vec())
312 .with_status_code(401)
313 .with_header(
314 Header::from_bytes(&b"Content-Type"[..], &b"application/json"[..]).unwrap(),
315 ),
316 );
317 }
318 handle(state, req)
319}
320
321fn route(
322 state: &State,
323 method: &Method,
324 path: &str,
325 query: &str,
326 body: &str,
327 x_lex_user: Option<&str>,
328) -> Response<std::io::Cursor<Vec<u8>>> {
329 match (method, path) {
330 (Method::Get, "/") => crate::web::activity_handler(state),
332 (Method::Get, "/web/branches") => crate::web::branches_handler(state),
333 (Method::Get, "/web/trust") => crate::web::trust_handler(state),
334 (Method::Get, "/web/attention") => crate::web::attention_handler(state),
335 (Method::Get, p) if p.starts_with("/web/branch/") => {
336 let name = &p["/web/branch/".len()..];
337 crate::web::branch_handler(state, name)
338 }
339 (Method::Get, p) if p.starts_with("/web/stage/") => {
340 let id = &p["/web/stage/".len()..];
341 crate::web::stage_html_handler(state, id)
342 }
343 (Method::Post, p) if p.starts_with("/web/stage/") && (
348 p.ends_with("/pin") || p.ends_with("/defer")
349 || p.ends_with("/block") || p.ends_with("/unblock")
350 ) => {
351 let prefix_len = "/web/stage/".len();
352 let last_slash = p.rfind('/').unwrap_or(p.len());
353 let id = &p[prefix_len..last_slash];
354 let verb = &p[last_slash + 1..];
355 let decision = match verb {
356 "pin" => crate::web::WebStageDecision::Pin,
357 "defer" => crate::web::WebStageDecision::Defer,
358 "block" => crate::web::WebStageDecision::Block,
359 "unblock" => crate::web::WebStageDecision::Unblock,
360 _ => unreachable!("matched in outer guard"),
361 };
362 crate::web::stage_decision_handler(state, id, body, decision, x_lex_user)
363 }
364 (Method::Get, "/v1/health") => json_response(200, &serde_json::json!({"ok": true})),
366 (Method::Post, "/v1/parse") => parse_handler(body),
367 (Method::Post, "/v1/check") => check_handler(body),
368 (Method::Post, "/v1/publish") => publish_handler(state, body),
369 (Method::Post, "/v1/patch") => patch_handler(state, body),
370 (Method::Get, p) if p.starts_with("/v1/stage/") => {
371 let suffix = &p["/v1/stage/".len()..];
372 if let Some(id) = suffix.strip_suffix("/attestations") {
375 stage_attestations_handler(state, id)
376 } else {
377 stage_handler(state, suffix)
378 }
379 }
380 (Method::Post, "/v1/run") => run_handler(state, body, false),
381 (Method::Post, "/v1/replay") => run_handler(state, body, true),
382 (Method::Get, p) if p.starts_with("/v1/trace/") => {
383 let id = &p["/v1/trace/".len()..];
384 trace_handler(state, id)
385 }
386 (Method::Get, "/v1/diff") => diff_handler(state, query),
387 (Method::Post, "/v1/merge/start") => merge_start_handler(state, body),
388 (Method::Post, p) if p.starts_with("/v1/merge/") && p.ends_with("/resolve") => {
389 let id = &p["/v1/merge/".len()..p.len() - "/resolve".len()];
390 merge_resolve_handler(state, id, body)
391 }
392 (Method::Post, p) if p.starts_with("/v1/merge/") && p.ends_with("/commit") => {
393 let id = &p["/v1/merge/".len()..p.len() - "/commit".len()];
394 merge_commit_handler(state, id)
395 }
396 (Method::Post, "/v1/ops/batch") => ops_batch_handler(state, body),
398 (Method::Post, "/v1/attestations/batch") => attestations_batch_handler(state, body),
399 (Method::Post, "/v1/stages/batch") => crate::sync_http::stages_batch_handler(state, body),
402 (Method::Post, "/v1/stages/fetch") => crate::sync_http::stages_fetch_handler(state, body),
403 (Method::Post, "/v1/intents/batch") => crate::sync_http::intents_batch_handler(state, body),
404 (Method::Post, "/v1/intents/fetch") => crate::sync_http::intents_fetch_handler(state, body),
405 (Method::Get, "/v1/review/inbox") => crate::review_http::review_inbox_handler(state, query),
409 (Method::Post, "/v1/review/verdict") => crate::review_http::review_verdict_handler(state, body),
410 (Method::Get, "/v1/branches") => crate::branches_http::branches_list_handler(state),
411 (Method::Post, "/v1/branches") => crate::branches_http::branch_create_handler(state, body),
412 (Method::Post, p) if p.starts_with("/v1/branches/") && p.ends_with("/checkout") => {
413 let name = &p["/v1/branches/".len()..p.len() - "/checkout".len()];
414 crate::branches_http::branch_checkout_handler(state, name)
415 }
416 (Method::Get, p) if p.starts_with("/v1/branches/") && p.ends_with("/head") => {
419 let name = &p["/v1/branches/".len()..p.len() - "/head".len()];
420 crate::branches_http::branch_head_handler(state, name)
421 }
422 (Method::Post, p) if p.starts_with("/v1/branches/") && p.ends_with("/head") => {
423 let name = &p["/v1/branches/".len()..p.len() - "/head".len()];
424 crate::branches_http::branch_advance_head_handler(state, name, body)
425 }
426 (Method::Get, "/v1/ops/since") => ops_since_handler(state, query),
430 (Method::Get, "/v1/attestations/since") => attestations_since_handler(state, query),
431 (Method::Get, "/v1/pkg") => pkg_list_handler(state),
435 (Method::Put, p) if p.starts_with("/v1/pkg/") && p.ends_with("/visibility") => {
439 let name = &p["/v1/pkg/".len()..p.len() - "/visibility".len()];
440 pkg_set_visibility_handler(state, name, body)
441 }
442 (Method::Post, p) if p.starts_with("/v1/pkg/") && p.ends_with("/release") => {
445 let name = &p["/v1/pkg/".len()..p.len() - "/release".len()];
446 pkg_release_handler(state, name, body)
447 }
448 (Method::Get, p) if p.starts_with("/v1/pkg/") && p.ends_with("/head") => {
449 let name = &p["/v1/pkg/".len()..p.len() - "/head".len()];
450 pkg_head_handler(state, name)
451 }
452 (Method::Get, p) if p.starts_with("/v1/pkg/") && p.ends_with("/versions") => {
453 let name = &p["/v1/pkg/".len()..p.len() - "/versions".len()];
454 pkg_versions_handler(state, name)
455 }
456 (Method::Get, p) if p.starts_with("/v1/pkg/") && p.ends_with("/api-diff") => {
457 let name = &p["/v1/pkg/".len()..p.len() - "/api-diff".len()];
458 pkg_api_diff_handler(state, name, query)
459 }
460 (Method::Get, p) if p.starts_with("/v1/pkg/") && p.ends_with("/archive") => {
462 let inner = &p["/v1/pkg/".len()..p.len() - "/archive".len()];
463 if let Some((name, version)) = inner.split_once('/') {
465 pkg_archive_handler(state, name, version)
466 } else {
467 error_response(400, "expected /v1/pkg/{name}/{version}/archive")
468 }
469 }
470 (Method::Get, p) if p.starts_with("/v1/pkg/") && p["/v1/pkg/".len()..].contains('/') => {
472 let inner = &p["/v1/pkg/".len()..];
473 if let Some((name, version)) = inner.split_once('/') {
474 pkg_get_version_handler(state, name, version)
475 } else {
476 error_response(400, "expected /v1/pkg/{name}/{version}")
477 }
478 }
479 (Method::Get, p) if p.starts_with("/v1/pkg/") => {
480 let name = &p["/v1/pkg/".len()..];
481 pkg_get_handler(state, name)
482 }
483 (Method::Delete, p) if p.starts_with("/v1/pkg/") => {
484 let name = &p["/v1/pkg/".len()..];
485 pkg_delete_handler(state, name)
486 }
487 _ => error_response(404, format!("unknown route: {method:?} {path}")),
488 }
489}
490
491#[derive(Deserialize)]
492struct ParseReq { source: String }
493
494fn parse_handler(body: &str) -> Response<std::io::Cursor<Vec<u8>>> {
495 let req: ParseReq = match serde_json::from_str(body) {
496 Ok(r) => r, Err(e) => return error_response(400, format!("bad request: {e}")),
497 };
498 match load_program_from_str(&req.source) {
499 Ok(prog) => {
500 let stages = canonicalize_program(&prog);
501 json_response(200, &serde_json::to_value(&stages).unwrap())
502 }
503 Err(e) => error_response(400, format!("syntax error: {e}")),
504 }
505}
506
507pub(crate) fn check_handler(body: &str) -> Response<std::io::Cursor<Vec<u8>>> {
508 let req: ParseReq = match serde_json::from_str(body) {
509 Ok(r) => r, Err(e) => return error_response(400, format!("bad request: {e}")),
510 };
511 let prog = match load_program_from_str(&req.source) {
512 Ok(p) => p, Err(e) => return error_response(400, format!("syntax error: {e}")),
513 };
514 let stages = canonicalize_program(&prog);
515 match lex_types::check_program(&stages) {
516 Ok(_) => json_response(200, &serde_json::json!({"ok": true})),
517 Err(errs) => json_response(422, &serde_json::to_value(&errs).unwrap()),
518 }
519}
520
521#[derive(Deserialize)]
522struct PublishReq { source: String, #[serde(default)] activate: bool }
523
524pub(crate) fn publish_handler(state: &State, body: &str) -> Response<std::io::Cursor<Vec<u8>>> {
525 let req: PublishReq = match serde_json::from_str(body) {
526 Ok(r) => r, Err(e) => return error_response(400, format!("bad request: {e}")),
527 };
528 let prog = match load_program_from_str(&req.source) {
529 Ok(p) => p, Err(e) => return error_response(400, format!("syntax error: {e}")),
530 };
531 let mut stages = canonicalize_program(&prog);
535 if let Err(errs) = lex_types::check_and_rewrite_program(&mut stages) {
536 return error_with_detail(422, "type errors", serde_json::to_value(&errs).unwrap());
537 }
538 let example_errors = lex_runtime::evaluate_examples(&stages);
544 if !example_errors.is_empty() {
545 return error_with_detail(422, "example mismatch",
546 serde_json::to_value(&example_errors).unwrap_or_default());
547 }
548
549 let store = state.store.lock().unwrap();
550 let branch = store.current_branch();
551
552 let old_head = match store.branch_head(&branch) {
554 Ok(h) => h,
555 Err(e) => return error_response(500, format!("branch_head: {e}")),
556 };
557 let old_head_stages: Vec<lex_ast::Stage> =
559 old_head.values().filter_map(|stg| store.get_ast(stg).ok()).collect();
560 let old_fns = stage_fns(&old_head_stages);
561 let new_fns = stage_fns(&stages);
562 let old_types = stage_types(&old_head_stages);
563 let new_types = stage_types(&stages);
564 let report =
565 lex_vcs::compute_diff_with_types(&old_fns, &new_fns, &old_types, &new_types, false);
566
567 let mut new_imports: lex_vcs::ImportMap = lex_vcs::ImportMap::new();
569 {
570 let entry = new_imports.entry("<source>".into()).or_default();
571 for s in &stages {
572 if let lex_ast::Stage::Import(im) = s {
573 entry.insert(lex_vcs::ImportRef {
574 reference: im.reference.clone(),
575 alias: im.alias.clone(),
576 });
577 }
578 }
579 }
580
581 match store.publish_program(&branch, &stages, &report, &new_imports, req.activate) {
582 Ok(outcome) => {
583 record_examples_for_publish(&store, &stages, &outcome);
587 json_response(200, &serde_json::json!({
588 "ops": outcome.ops,
589 "head_op": outcome.head_op,
590 }))
591 }
592 Err(lex_store::StoreError::TypeError(errs)) => {
600 error_with_detail(422, "type errors", serde_json::to_value(&errs).unwrap())
601 }
602 Err(e) => write_error_response("publish_program", e),
603 }
604}
605
606#[derive(Deserialize)]
607struct PatchReq {
608 stage_id: String,
609 patch: lex_ast::Patch,
610 #[serde(default)] activate: bool,
611}
612
613fn patch_handler(state: &State, body: &str) -> Response<std::io::Cursor<Vec<u8>>> {
616 let req: PatchReq = match serde_json::from_str(body) {
617 Ok(r) => r, Err(e) => return error_response(400, format!("bad request: {e}")),
618 };
619 let store = state.store.lock().unwrap();
620
621 let original = match store.get_ast(&req.stage_id) {
623 Ok(s) => s, Err(e) => return error_response(404, format!("stage: {e}")),
624 };
625
626 let patched = match lex_ast::apply_patch(&original, &req.patch) {
628 Ok(s) => s,
629 Err(e) => return error_with_detail(422, "patch failed",
630 serde_json::to_value(&e).unwrap_or_default()),
631 };
632
633 let branch = store.current_branch();
644
645 let sig = match lex_ast::sig_id(&patched) {
647 Some(s) => s,
648 None => return error_response(500, "patched stage has no sig_id"),
649 };
650
651 let new_id = match store.publish(&patched) {
654 Ok(id) => id, Err(e) => return error_response(500, format!("publish: {e}")),
655 };
656
657 let original_effects: std::collections::BTreeSet<String> = match &original {
659 lex_ast::Stage::FnDecl(fd) => fd.effects.iter().map(|e| e.name.clone()).collect(),
660 _ => std::collections::BTreeSet::new(),
661 };
662 let patched_effects: std::collections::BTreeSet<String> = match &patched {
663 lex_ast::Stage::FnDecl(fd) => fd.effects.iter().map(|e| e.name.clone()).collect(),
664 _ => std::collections::BTreeSet::new(),
665 };
666 let head_now = match store.get_branch(&branch) {
667 Ok(b) => b.and_then(|b| b.head_op),
668 Err(e) => return error_response(500, format!("get_branch: {e}")),
669 };
670 let kind = if original_effects != patched_effects {
671 let from_budget = lex_vcs::operation_budget_from_effects(&original_effects);
678 let to_budget = lex_vcs::operation_budget_from_effects(&patched_effects);
679 lex_vcs::OperationKind::ChangeEffectSig {
680 sig_id: sig.clone(),
681 from_stage_id: req.stage_id.clone(),
682 to_stage_id: new_id.clone(),
683 from_effects: original_effects,
684 to_effects: patched_effects,
685 from_budget,
686 to_budget,
687 }
688 } else {
689 let budget = lex_vcs::operation_budget_from_effects(&original_effects);
690 lex_vcs::OperationKind::ModifyBody {
691 sig_id: sig.clone(),
692 from_stage_id: req.stage_id.clone(),
693 to_stage_id: new_id.clone(),
694 from_budget: budget,
695 to_budget: budget,
696 }
697 };
698 let transition = lex_vcs::StageTransition::Replace {
699 sig_id: sig.clone(),
700 from: req.stage_id.clone(),
701 to: new_id.clone(),
702 };
703 let op = lex_vcs::Operation::new(
704 kind,
705 head_now.into_iter().collect::<Vec<_>>(),
706 );
707 let op_id = match store.apply_operation_gated(&branch, op, transition) {
708 Ok(id) => id,
709 Err(lex_store::StoreError::TypeError(errs)) => return error_with_detail(
710 422, "type errors after patch", serde_json::to_value(&errs).unwrap_or_default()),
711 Err(e) => return write_error_response("apply_operation_gated", e),
712 };
713 if req.activate {
714 if let Err(e) = store.activate(&new_id) {
715 return error_response(500, format!("activate: {e}"));
716 }
717 }
718
719 let status = format!("{:?}",
720 store.get_status(&new_id).unwrap_or(lex_store::StageStatus::Draft)).to_lowercase();
721 json_response(200, &serde_json::json!({
722 "old_stage_id": req.stage_id,
723 "new_stage_id": new_id,
724 "sig_id": sig,
725 "status": status,
726 "op_id": op_id,
727 }))
728}
729
730pub(crate) fn stage_handler(state: &State, id: &str) -> Response<std::io::Cursor<Vec<u8>>> {
731 let store = state.store.lock().unwrap();
732 let meta = match store.get_metadata(id) {
733 Ok(m) => m, Err(e) => return error_response(404, format!("{e}")),
734 };
735 let ast = match store.get_ast(id) {
736 Ok(a) => a, Err(e) => return error_response(404, format!("{e}")),
737 };
738 let status = format!("{:?}", store.get_status(id).unwrap_or(lex_store::StageStatus::Draft)).to_lowercase();
739 json_response(200, &serde_json::json!({
740 "metadata": meta,
741 "ast": ast,
742 "status": status,
743 }))
744}
745
746pub(crate) fn stage_attestations_handler(state: &State, id: &str) -> Response<std::io::Cursor<Vec<u8>>> {
755 let store = state.store.lock().unwrap();
756 if let Err(e) = store.get_metadata(id) {
757 return error_response(404, format!("{e}"));
758 }
759 let log = match store.attestation_log() {
760 Ok(l) => l,
761 Err(e) => return error_response(500, format!("attestation log: {e}")),
762 };
763 let mut listing = match log.list_for_stage(&id.to_string()) {
764 Ok(v) => v,
765 Err(e) => return error_response(500, format!("list_for_stage: {e}")),
766 };
767 listing.sort_by_key(|a| std::cmp::Reverse(a.timestamp));
768 json_response(200, &serde_json::json!({"attestations": listing}))
769}
770
771#[derive(Deserialize, Default)]
772struct PolicyJson {
773 #[serde(default)] allow_effects: Vec<String>,
774 #[serde(default)] allow_fs_read: Vec<String>,
775 #[serde(default)] allow_fs_write: Vec<String>,
776 #[serde(default)] budget: Option<u64>,
777}
778
779impl PolicyJson {
780 fn into_policy(self) -> Policy {
781 Policy {
782 allow_effects: self.allow_effects.into_iter().collect::<BTreeSet<_>>(),
783 allow_fs_read: self.allow_fs_read.into_iter().map(PathBuf::from).collect(),
784 allow_fs_write: self.allow_fs_write.into_iter().map(PathBuf::from).collect(),
785 allow_net_host: Vec::new(),
786 allow_proc: Vec::new(),
787 allow_approval: Vec::new(),
788 budget: self.budget,
789 }
790 }
791}
792
793#[derive(Deserialize)]
794struct RunReq {
795 source: String,
796 #[serde(rename = "fn")] func: String,
797 #[serde(default)] args: Vec<serde_json::Value>,
798 #[serde(default)] policy: PolicyJson,
799 #[serde(default)] overrides: IndexMap<String, serde_json::Value>,
800}
801
802pub(crate) fn run_handler(state: &State, body: &str, with_overrides: bool) -> Response<std::io::Cursor<Vec<u8>>> {
803 let req: RunReq = match serde_json::from_str(body) {
804 Ok(r) => r, Err(e) => return error_response(400, format!("bad request: {e}")),
805 };
806 let prog = match load_program_from_str(&req.source) {
807 Ok(p) => p, Err(e) => return error_response(400, format!("syntax error: {e}")),
808 };
809 let stages = canonicalize_program(&prog);
810 if let Err(errs) = lex_types::check_program(&stages) {
811 return error_with_detail(422, "type errors", serde_json::to_value(&errs).unwrap());
812 }
813 let bc = compile_program(&stages);
814 let mut policy = req.policy.into_policy();
815 if let Some(ceiling) = &state.policy_ceiling {
821 policy = clamp_policy(policy, ceiling);
822 }
823 if let Err(violations) = check_policy(&bc, &policy) {
824 return error_with_detail(403, "policy violation", serde_json::to_value(&violations).unwrap());
825 }
826
827 let mut recorder = lex_trace::Recorder::new();
828 if with_overrides && !req.overrides.is_empty() {
829 recorder = recorder.with_overrides(req.overrides);
830 }
831 let handle = recorder.handle();
832 let handler = DefaultHandler::new(policy);
833 let mut vm = Vm::with_handler(&bc, Box::new(handler));
834 vm.set_tracer(Box::new(recorder));
835
836 let vargs: Vec<Value> = req.args.iter().map(json_to_value).collect();
837 let started = std::time::SystemTime::now().duration_since(std::time::UNIX_EPOCH).unwrap().as_secs();
838 let result = vm.call(&req.func, vargs);
839 let ended = std::time::SystemTime::now().duration_since(std::time::UNIX_EPOCH).unwrap().as_secs();
840
841 let store = state.store.lock().unwrap();
842 let (root_out, root_err, status) = match &result {
843 Ok(v) => (Some(value_to_json(v)), None, 200u16),
844 Err(e) => (None, Some(format!("{e}")), 200u16),
845 };
846 let tree = handle.finalize(req.func.clone(), serde_json::Value::Null,
847 root_out.clone(), root_err.clone(), started, ended);
848 let run_id = match store.save_trace(&tree) {
849 Ok(id) => id,
850 Err(e) => return error_response(500, format!("save_trace: {e}")),
851 };
852
853 let mut body = serde_json::json!({
854 "run_id": run_id,
855 "output": root_out,
856 });
857 if let Some(err) = root_err {
858 body["error"] = serde_json::Value::String(err);
859 }
860 json_response(status, &body)
861}
862
863fn trace_handler(state: &State, id: &str) -> Response<std::io::Cursor<Vec<u8>>> {
864 let store = state.store.lock().unwrap();
865 match store.load_trace(id) {
866 Ok(t) => json_response(200, &serde_json::to_value(&t).unwrap()),
867 Err(e) => error_response(404, format!("{e}")),
868 }
869}
870
871fn diff_handler(state: &State, query: &str) -> Response<std::io::Cursor<Vec<u8>>> {
872 let mut a = None;
873 let mut b = None;
874 for kv in query.split('&') {
875 if let Some((k, v)) = kv.split_once('=') {
876 match k { "a" => a = Some(v.to_string()), "b" => b = Some(v.to_string()), _ => {} }
877 }
878 }
879 let (Some(a), Some(b)) = (a, b) else {
880 return error_response(400, "missing a or b query params");
881 };
882 let store = state.store.lock().unwrap();
883 let ta = match store.load_trace(&a) { Ok(t) => t, Err(e) => return error_response(404, format!("a: {e}")) };
884 let tb = match store.load_trace(&b) { Ok(t) => t, Err(e) => return error_response(404, format!("b: {e}")) };
885 match lex_trace::diff_runs(&ta, &tb) {
886 Some(d) => json_response(200, &serde_json::to_value(&d).unwrap()),
887 None => json_response(200, &serde_json::json!({"divergence": null})),
888 }
889}
890
891fn json_to_value(v: &serde_json::Value) -> Value { Value::from_json(v) }
892
893fn value_to_json(v: &Value) -> serde_json::Value { v.to_json() }
894
895#[derive(Deserialize)]
896struct MergeStartReq {
897 src_branch: String,
898 dst_branch: String,
899}
900
901fn merge_start_handler(state: &State, body: &str) -> Response<std::io::Cursor<Vec<u8>>> {
912 let req: MergeStartReq = match serde_json::from_str(body) {
913 Ok(r) => r, Err(e) => return error_response(400, format!("bad request: {e}")),
914 };
915 let store = state.store.lock().unwrap();
916 let src_head = match store.get_branch(&req.src_branch) {
917 Ok(Some(b)) => b.head_op,
918 Ok(None) => return error_response(404, format!("unknown src branch `{}`", req.src_branch)),
919 Err(e) => return error_response(500, format!("src branch read: {e}")),
920 };
921 let dst_head = match store.get_branch(&req.dst_branch) {
922 Ok(Some(b)) => b.head_op,
923 Ok(None) => return error_response(404, format!("unknown dst branch `{}`", req.dst_branch)),
924 Err(e) => return error_response(500, format!("dst branch read: {e}")),
925 };
926 let log = match lex_vcs::OpLog::open(store.root()) {
927 Ok(l) => l,
928 Err(e) => return error_response(500, format!("op log: {e}")),
929 };
930 let merge_id = mint_merge_id();
934 let session = match MergeSession::start(
935 merge_id.clone(),
936 &log,
937 src_head.as_ref(),
938 dst_head.as_ref(),
939 ) {
940 Ok(s) => s,
941 Err(e) => return error_response(500, format!("merge start: {e}")),
942 };
943 let conflicts: Vec<&lex_vcs::ConflictRecord> = session.remaining_conflicts();
944 let auto_resolved_count = session.auto_resolved.len();
945 let body = serde_json::json!({
946 "merge_id": merge_id,
947 "src_head": session.src_head,
948 "dst_head": session.dst_head,
949 "lca": session.lca,
950 "conflicts": conflicts,
951 "auto_resolved_count": auto_resolved_count,
952 });
953 drop(conflicts);
954 drop(store);
955 let wrapped = ApiMergeSession {
956 inner: session,
957 src_branch: req.src_branch,
958 dst_branch: req.dst_branch,
959 };
960 state.sessions.lock().unwrap().insert(merge_id, wrapped);
961 json_response(200, &body)
962}
963
964#[derive(Deserialize)]
965struct MergeResolveReq {
966 resolutions: Vec<MergeResolveEntry>,
971}
972
973#[derive(Deserialize)]
974struct MergeResolveEntry {
975 conflict_id: String,
976 resolution: lex_vcs::Resolution,
977}
978
979fn merge_resolve_handler(
990 state: &State,
991 merge_id: &str,
992 body: &str,
993) -> Response<std::io::Cursor<Vec<u8>>> {
994 let req: MergeResolveReq = match serde_json::from_str(body) {
995 Ok(r) => r, Err(e) => return error_response(400, format!("bad request: {e}")),
996 };
997 let mut sessions = state.sessions.lock().unwrap();
998 let Some(wrapped) = sessions.get_mut(merge_id) else {
999 return error_response(404, format!("unknown merge_id `{merge_id}`"));
1000 };
1001 let pairs: Vec<(String, lex_vcs::Resolution)> = req.resolutions.into_iter()
1002 .map(|e| (e.conflict_id, e.resolution))
1003 .collect();
1004 let store = state.store.lock().unwrap();
1009 let checker = lex_store::MergeResolutionChecker::new(&store, wrapped.dst_branch.clone());
1010 let verdicts = wrapped.inner.resolve_checked(pairs, &checker);
1011 drop(store);
1012 let remaining: Vec<&lex_vcs::ConflictRecord> = wrapped.inner.remaining_conflicts();
1013 let body = serde_json::json!({
1014 "verdicts": verdicts,
1015 "remaining_conflicts": remaining,
1016 });
1017 json_response(200, &body)
1018}
1019
1020fn merge_commit_handler(
1039 state: &State,
1040 merge_id: &str,
1041) -> Response<std::io::Cursor<Vec<u8>>> {
1042 use std::collections::BTreeMap;
1043 let wrapped = match state.sessions.lock().unwrap().remove(merge_id) {
1044 Some(w) => w,
1045 None => return error_response(404, format!("unknown merge_id `{merge_id}`")),
1046 };
1047 let dst_branch = wrapped.dst_branch.clone();
1048 let src_head = wrapped.inner.src_head.clone();
1049 let dst_head = wrapped.inner.dst_head.clone();
1050 let auto_resolved = wrapped.inner.auto_resolved.clone();
1051
1052 let mut entries: BTreeMap<lex_vcs::SigId, Option<lex_vcs::StageId>> = BTreeMap::new();
1055
1056 for outcome in &auto_resolved {
1058 if let lex_vcs::MergeOutcome::Src { sig_id, stage_id } = outcome {
1059 entries.insert(sig_id.clone(), stage_id.clone());
1060 }
1061 }
1062
1063 let resolved = match wrapped.inner.commit() {
1065 Ok(r) => r,
1066 Err(lex_vcs::CommitError::ConflictsRemaining(ids)) => {
1067 return error_with_detail(
1071 422,
1072 "conflicts remaining",
1073 serde_json::json!({"unresolved": ids}),
1074 );
1075 }
1076 };
1077
1078 for (conflict_id, resolution) in resolved {
1079 match resolution {
1080 lex_vcs::Resolution::TakeOurs => {
1081 }
1083 lex_vcs::Resolution::TakeTheirs => {
1084 match resolve_take_theirs(state, &src_head, &conflict_id) {
1094 Ok(stage_id) => {
1095 entries.insert(conflict_id.clone(), stage_id);
1096 }
1097 Err(e) => return error_response(500, format!("resolve take_theirs: {e}")),
1098 }
1099 }
1100 lex_vcs::Resolution::Custom { op } => {
1101 match op.kind.merge_target() {
1110 Some((sig, stage)) => {
1111 if sig != conflict_id {
1112 return error_with_detail(
1113 422,
1114 "custom op targets a different sig than the conflict",
1115 serde_json::json!({
1116 "conflict_id": conflict_id,
1117 "op_targets": sig,
1118 }),
1119 );
1120 }
1121 entries.insert(conflict_id, stage);
1122 }
1123 None => {
1124 return error_with_detail(
1125 422,
1126 "custom op kind doesn't yield a single sig→stage delta",
1127 serde_json::json!({
1128 "conflict_id": conflict_id,
1129 "kind": serde_json::to_value(&op.kind).unwrap_or(serde_json::Value::Null),
1130 }),
1131 );
1132 }
1133 }
1134 }
1135 lex_vcs::Resolution::Defer => {
1136 return error_response(500, "internal: Defer slipped past commit gate");
1138 }
1139 }
1140 }
1141
1142 let resolved_count = entries.len();
1143 let mut parents: Vec<lex_vcs::OpId> = Vec::new();
1144 if let Some(d) = dst_head { parents.push(d); }
1145 if let Some(s) = src_head { parents.push(s); }
1146 let op = lex_vcs::Operation::new(
1147 lex_vcs::OperationKind::Merge { resolved: resolved_count },
1148 parents,
1149 );
1150 let transition = lex_vcs::StageTransition::Merge { entries };
1151 let store = state.store.lock().unwrap();
1152 match store.apply_merge_op_gated(&dst_branch, op, transition) {
1155 Ok(new_head_op) => json_response(200, &serde_json::json!({
1156 "new_head_op": new_head_op,
1157 "dst_branch": dst_branch,
1158 })),
1159 Err(lex_store::StoreError::TypeError(errs)) => error_with_detail(
1160 422, "merged program has type errors", serde_json::to_value(&errs).unwrap_or_default()),
1161 Err(e) => write_error_response("apply merge op", e),
1162 }
1163}
1164
1165fn resolve_take_theirs(
1170 state: &State,
1171 src_head: &Option<lex_vcs::OpId>,
1172 sig: &lex_vcs::SigId,
1173) -> std::io::Result<Option<lex_vcs::StageId>> {
1174 let store = state.store.lock().unwrap();
1175 let log = lex_vcs::OpLog::open(store.root())?;
1176 let Some(head) = src_head.as_ref() else { return Ok(None); };
1177 let mut current: Option<lex_vcs::StageId> = None;
1180 for record in log.walk_forward(head, None)? {
1181 match &record.produces {
1182 lex_vcs::StageTransition::Create { sig_id, stage_id }
1183 if sig_id == sig => { current = Some(stage_id.clone()); }
1184 lex_vcs::StageTransition::Replace { sig_id, to, .. }
1185 if sig_id == sig => { current = Some(to.clone()); }
1186 lex_vcs::StageTransition::Remove { sig_id, .. }
1187 if sig_id == sig => { current = None; }
1188 lex_vcs::StageTransition::Rename { from, to, body_stage_id }
1189 if from == sig || to == sig => {
1190 if from == sig { current = None; }
1191 if to == sig { current = Some(body_stage_id.clone()); }
1192 }
1193 lex_vcs::StageTransition::Merge { entries } => {
1194 if let Some(opt) = entries.get(sig) {
1195 current = opt.clone();
1196 }
1197 }
1198 _ => {}
1199 }
1200 }
1201 Ok(current)
1202}
1203
1204fn mint_merge_id() -> MergeSessionId {
1205 use std::sync::atomic::{AtomicU64, Ordering};
1206 static COUNTER: AtomicU64 = AtomicU64::new(0);
1207 let nanos = SystemTime::now()
1208 .duration_since(UNIX_EPOCH)
1209 .map(|d| d.as_nanos())
1210 .unwrap_or(0);
1211 let n = COUNTER.fetch_add(1, Ordering::Relaxed);
1212 format!("merge_{nanos:x}_{n:x}")
1213}
1214
1215pub(crate) fn ops_batch_handler(state: &State, body: &str)
1246 -> Response<std::io::Cursor<Vec<u8>>>
1247{
1248 let records: Vec<lex_vcs::OperationRecord> = match serde_json::from_str(body) {
1249 Ok(r) => r,
1250 Err(e) => return error_response(400,
1251 format!("body must be a JSON array of OperationRecord: {e}")),
1252 };
1253 let store = state.store.lock().unwrap();
1254 let log = match lex_vcs::OpLog::open(store.root()) {
1255 Ok(l) => l,
1256 Err(e) => return error_response(500, format!("opening op log: {e}")),
1257 };
1258
1259 let mut batch_ids: std::collections::BTreeSet<lex_vcs::OpId> =
1267 std::collections::BTreeSet::new();
1268 for rec in &records {
1269 let expected = rec.op.op_id();
1270 if expected != rec.op_id {
1271 return error_with_detail(409, "OpIdMismatch", serde_json::json!({
1272 "supplied": rec.op_id,
1273 "expected": expected,
1274 }));
1275 }
1276 for parent in &rec.op.parents {
1277 let known = match log.get(parent) {
1278 Ok(Some(_)) => true,
1279 Ok(None) => false,
1280 Err(e) => return error_response(500, format!("op log read: {e}")),
1281 };
1282 if !known && !batch_ids.contains(parent) {
1283 return error_with_detail(422, "MissingParent", serde_json::json!({
1284 "op_id": rec.op_id,
1285 "missing_parent": parent,
1286 }));
1287 }
1288 }
1289 batch_ids.insert(rec.op_id.clone());
1290 }
1291
1292 let mut added = 0usize;
1295 let mut added_ids: Vec<&lex_vcs::OpId> = Vec::new();
1296 for rec in &records {
1297 let already_present = matches!(log.get(&rec.op_id), Ok(Some(_)));
1298 match log.put(rec) {
1299 Ok(()) => {
1300 if !already_present {
1301 added += 1;
1302 added_ids.push(&rec.op_id);
1303 }
1304 }
1305 Err(e) => return error_response(500, format!("op log write: {e}")),
1306 }
1307 }
1308
1309 json_response(200, &serde_json::json!({
1310 "received": records.len(),
1311 "added": added,
1312 "skipped": records.len() - added,
1313 "added_ids": added_ids,
1314 }))
1315}
1316
1317pub(crate) fn attestations_batch_handler(state: &State, body: &str)
1339 -> Response<std::io::Cursor<Vec<u8>>>
1340{
1341 let attestations: Vec<lex_vcs::Attestation> = match serde_json::from_str(body) {
1342 Ok(a) => a,
1343 Err(e) => return error_response(400,
1344 format!("body must be a JSON array of Attestation: {e}")),
1345 };
1346 let store = state.store.lock().unwrap();
1347 let log = match store.attestation_log() {
1348 Ok(l) => l,
1349 Err(e) => return error_response(500, format!("opening attestation log: {e}")),
1350 };
1351 let op_log = match lex_vcs::OpLog::open(store.root()) {
1352 Ok(l) => l,
1353 Err(e) => return error_response(500, format!("opening op log: {e}")),
1354 };
1355
1356 for att in &attestations {
1358 let expected = lex_vcs::Attestation::with_timestamp(
1361 att.stage_id.clone(),
1362 att.op_id.clone(),
1363 att.intent_id.clone(),
1364 att.kind.clone(),
1365 att.result.clone(),
1366 att.produced_by.clone(),
1367 att.cost.clone(),
1368 att.timestamp,
1369 ).attestation_id;
1370 if expected != att.attestation_id {
1371 return error_with_detail(409, "AttestationIdMismatch", serde_json::json!({
1372 "supplied": att.attestation_id,
1373 "expected": expected,
1374 }));
1375 }
1376 if let Some(op_id) = &att.op_id {
1380 match op_log.get(op_id) {
1381 Ok(Some(_)) => {}
1382 Ok(None) => return error_with_detail(422, "UnknownOp", serde_json::json!({
1383 "attestation_id": att.attestation_id,
1384 "op_id": op_id,
1385 })),
1386 Err(e) => return error_response(500, format!("op log read: {e}")),
1387 }
1388 }
1389 }
1390
1391 let mut added = 0usize;
1395 let mut added_ids: Vec<&lex_vcs::AttestationId> = Vec::new();
1396 for att in &attestations {
1397 let already_present = matches!(log.get(&att.attestation_id), Ok(Some(_)));
1398 match log.put(att) {
1399 Ok(()) => {
1400 if !already_present {
1401 added += 1;
1402 added_ids.push(&att.attestation_id);
1403 }
1404 }
1405 Err(e) => return error_response(500, format!("attestation log write: {e}")),
1406 }
1407 }
1408
1409 json_response(200, &serde_json::json!({
1410 "received": attestations.len(),
1411 "added": added,
1412 "skipped": attestations.len() - added,
1413 "added_ids": added_ids,
1414 }))
1415}
1416
1417pub(crate) fn ops_since_handler(state: &State, query: &str)
1449 -> Response<std::io::Cursor<Vec<u8>>>
1450{
1451 let mut after: Option<String> = None;
1452 let mut branch = String::from("main");
1453 let mut limit: Option<usize> = None;
1454 for kv in query.split('&') {
1455 let Some((k, v)) = kv.split_once('=') else { continue };
1456 match k {
1457 "after" => after = Some(v.to_string()),
1458 "branch" => branch = v.to_string(),
1459 "limit" => {
1460 limit = Some(match v.parse::<usize>() {
1461 Ok(n) => n,
1462 Err(_) => return error_response(400,
1463 format!("limit must be a positive integer, got `{v}`")),
1464 });
1465 }
1466 _ => {}
1467 }
1468 }
1469
1470 let store = state.store.lock().unwrap();
1471 let log = match lex_vcs::OpLog::open(store.root()) {
1472 Ok(l) => l,
1473 Err(e) => return error_response(500, format!("opening op log: {e}")),
1474 };
1475 let head = match store.get_branch(&branch) {
1476 Ok(Some(b)) => b.head_op,
1477 Ok(None) => None,
1478 Err(e) => return error_response(500, format!("get_branch: {e}")),
1479 };
1480 let Some(head) = head else {
1481 return json_response(200, &serde_json::json!([]));
1482 };
1483
1484 let ops_since = match log.ops_since(&head, after.as_ref()) {
1485 Ok(o) => o,
1486 Err(e) => return error_response(500, format!("ops_since: {e}")),
1487 };
1488 let mut ops = ops_since;
1492 ops.reverse();
1493 if let Some(n) = limit {
1494 ops.truncate(n);
1495 }
1496
1497 json_response(200, &serde_json::to_value(&ops).unwrap_or_default())
1498}
1499
1500pub(crate) fn attestations_since_handler(state: &State, query: &str)
1513 -> Response<std::io::Cursor<Vec<u8>>>
1514{
1515 let mut after_op: Option<String> = None;
1516 let mut limit: Option<usize> = None;
1517 for kv in query.split('&') {
1518 let Some((k, v)) = kv.split_once('=') else { continue };
1519 match k {
1520 "after-op" => after_op = Some(v.to_string()),
1521 "limit" => {
1522 limit = Some(match v.parse::<usize>() {
1523 Ok(n) => n,
1524 Err(_) => return error_response(400,
1525 format!("limit must be a positive integer, got `{v}`")),
1526 });
1527 }
1528 _ => {}
1529 }
1530 }
1531
1532 let store = state.store.lock().unwrap();
1533 let log = match store.attestation_log() {
1534 Ok(l) => l,
1535 Err(e) => return error_response(500, format!("opening attestation log: {e}")),
1536 };
1537
1538 let exclude: std::collections::BTreeSet<String> = match &after_op {
1542 None => std::collections::BTreeSet::new(),
1543 Some(cutoff) => {
1544 let op_log = match lex_vcs::OpLog::open(store.root()) {
1545 Ok(l) => l,
1546 Err(e) => return error_response(500, format!("opening op log: {e}")),
1547 };
1548 match op_log.walk_back(cutoff, None) {
1549 Ok(records) => records.into_iter().map(|r| r.op_id).collect(),
1550 Err(_) => {
1551 std::collections::BTreeSet::new()
1555 }
1556 }
1557 }
1558 };
1559
1560 let all = match log.list_all() {
1561 Ok(v) => v,
1562 Err(e) => return error_response(500, format!("listing attestations: {e}")),
1563 };
1564 let mut filtered: Vec<lex_vcs::Attestation> = all
1565 .into_iter()
1566 .filter(|a| match &a.op_id {
1567 Some(op_id) => !exclude.contains(op_id),
1568 None => true,
1572 })
1573 .collect();
1574 filtered.sort_by(|a, b| {
1578 a.timestamp.cmp(&b.timestamp)
1579 .then_with(|| a.attestation_id.cmp(&b.attestation_id))
1580 });
1581 if let Some(n) = limit {
1582 filtered.truncate(n);
1583 }
1584
1585 json_response(200, &serde_json::to_value(&filtered).unwrap_or_default())
1586}
1587
1588#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
1592struct PkgRecord {
1593 name: String,
1594 version: String,
1595 head_op: Option<String>,
1596 published_at: u64,
1597 function_names: Vec<String>,
1599 ops: Vec<serde_json::Value>,
1601}
1602
1603#[derive(Debug, Clone, Copy, PartialEq, Eq, serde::Serialize, serde::Deserialize, Default)]
1610#[serde(rename_all = "lowercase")]
1611pub enum Visibility {
1612 #[default]
1613 Private,
1614 Public,
1615}
1616
1617#[derive(Debug, Clone, serde::Serialize, serde::Deserialize, Default)]
1622struct PkgIndex {
1623 latest: Option<String>,
1625 versions: Vec<PkgVersionSummary>,
1627 #[serde(default)]
1631 visibility: Visibility,
1632}
1633
1634#[derive(Debug, Clone, serde::Serialize, serde::Deserialize)]
1635struct PkgVersionSummary {
1636 version: String,
1637 head_op: Option<String>,
1638 published_at: u64,
1639}
1640
1641fn pkg_name_dir(root: &std::path::Path, name: &str) -> PathBuf {
1642 root.join("packages").join(name)
1643}
1644
1645fn pkg_index_path(root: &std::path::Path, name: &str) -> PathBuf {
1646 pkg_name_dir(root, name).join("index.json")
1647}
1648
1649fn pkg_version_path(root: &std::path::Path, name: &str, version: &str) -> PathBuf {
1650 pkg_name_dir(root, name).join(format!("{version}.json"))
1651}
1652
1653fn pkg_archive_path(root: &std::path::Path, name: &str, version: &str) -> PathBuf {
1654 pkg_name_dir(root, name).join(format!("{version}.tar.gz"))
1655}
1656
1657fn load_pkg_index(root: &std::path::Path, name: &str) -> Option<PkgIndex> {
1658 let bytes = std::fs::read(pkg_index_path(root, name)).ok()?;
1659 serde_json::from_slice(&bytes).ok()
1660}
1661
1662fn load_pkg_record(root: &std::path::Path, name: &str, version: &str) -> Option<PkgRecord> {
1663 let bytes = std::fs::read(pkg_version_path(root, name, version)).ok()?;
1664 serde_json::from_slice(&bytes).ok()
1665}
1666
1667fn load_latest_pkg_record(root: &std::path::Path, name: &str) -> Option<PkgRecord> {
1668 let index = load_pkg_index(root, name)?;
1669 let latest = index.latest.clone()?;
1670 load_pkg_record(root, name, &latest)
1671}
1672
1673fn pkg_is_public(root: &std::path::Path, name: &str) -> bool {
1677 load_pkg_index(root, name).map(|i| i.visibility) == Some(Visibility::Public)
1678}
1679
1680fn valid_pkg_segment(s: &str) -> bool {
1685 !s.is_empty()
1686 && s.len() <= 128
1687 && s != "."
1688 && s != ".."
1689 && s.chars().all(|c| c.is_ascii_alphanumeric() || matches!(c, '.' | '_' | '-'))
1690}
1691
1692#[derive(Deserialize)]
1693struct VisibilityReq {
1694 visibility: Visibility,
1695}
1696
1697fn pkg_set_visibility_handler(
1704 state: &State,
1705 name: &str,
1706 body: &str,
1707) -> Response<std::io::Cursor<Vec<u8>>> {
1708 if !valid_pkg_segment(name) {
1709 return error_response(400, format!("invalid package name {name:?}"));
1710 }
1711 let req: VisibilityReq = match serde_json::from_str(body) {
1712 Ok(r) => r,
1713 Err(e) => return error_response(400, format!("bad request: {e}")),
1714 };
1715 let mut index = match load_pkg_index(&state.root, name) {
1716 Some(i) => i,
1717 None => return error_response(404, format!("package {name:?} not found")),
1718 };
1719 index.visibility = req.visibility;
1720 let bytes = serde_json::to_vec_pretty(&index).unwrap_or_default();
1721 match std::fs::write(pkg_index_path(&state.root, name), bytes) {
1722 Ok(()) => json_response(
1723 200,
1724 &serde_json::json!({ "name": name, "visibility": index.visibility }),
1725 ),
1726 Err(e) => error_response(500, format!("write index: {e}")),
1727 }
1728}
1729
1730#[derive(serde::Deserialize)]
1731struct ReleaseReq {
1732 version: String,
1733 #[serde(default)]
1734 branch: Option<String>,
1735}
1736
1737fn pkg_release_handler(state: &State, name: &str, body: &str) -> Response<std::io::Cursor<Vec<u8>>> {
1746 if !valid_pkg_segment(name) {
1747 return error_response(400, format!("invalid package name {name:?}"));
1748 }
1749 let req: ReleaseReq = match serde_json::from_str(body) {
1750 Ok(r) => r,
1751 Err(e) => return error_response(400, format!("bad request: {e}")),
1752 };
1753 let version = req.version.trim().to_string();
1754 if version.is_empty() || !valid_pkg_segment(&version) {
1755 return error_response(400, "version must be a non-empty, path-safe string (e.g. 1.2.0)");
1756 }
1757 if load_pkg_record(&state.root, name, &version).is_some() {
1759 return error_response(
1760 409,
1761 format!("{name}@{version} already released; releases are immutable — bump the version"),
1762 );
1763 }
1764
1765 let store = state.store.lock().unwrap();
1766 let branch = req.branch.unwrap_or_else(|| store.current_branch());
1767 let head_op = match store.get_branch(&branch) {
1768 Ok(Some(b)) => b.head_op,
1769 Ok(None) => return error_response(404, format!("unknown branch {branch:?}")),
1770 Err(e) => return error_response(500, format!("get_branch: {e}")),
1771 };
1772 let Some(head_op) = head_op else {
1773 return error_response(400, format!("branch {branch:?} has no commits to release"));
1774 };
1775
1776 let predecessor = load_pkg_index(&state.root, name)
1787 .map(|i| i.versions)
1788 .unwrap_or_default()
1789 .into_iter()
1790 .filter_map(|v| lex_syntax::semver::parse_exact(&v.version).map(|p| (p, v)))
1791 .filter(|(p, _)| lex_syntax::semver::parse_exact(&version).map(|n| *p < n).unwrap_or(false))
1792 .max_by_key(|(p, _)| *p)
1793 .map(|(_, v)| v);
1794 if let Some(prev) = predecessor {
1795 if let (Some(prev_head), Some(declared)) = (
1796 prev.head_op.clone(),
1797 lex_syntax::semver::bump_between(&prev.version, &version),
1798 ) {
1799 if let (Ok(prev_api), Ok(new_api)) = (
1800 lex_store::api::public_api_at_op(&store, &prev_head),
1801 lex_store::api::public_api_at_op(&store, &head_op),
1802 ) {
1803 use lex_store::api::ApiChange;
1804 use lex_syntax::semver::Bump;
1805 let (required, why) = match lex_store::api::classify_api_change(&prev_api, &new_api) {
1806 ApiChange::Breaking(d) => (Bump::Major, d),
1807 ApiChange::Additive(d) => (Bump::Minor, d),
1808 ApiChange::None => (Bump::Patch, String::new()),
1809 };
1810 if declared < required {
1811 let need = match required {
1812 Bump::Major => "major",
1813 Bump::Minor => "minor",
1814 Bump::Patch => "patch",
1815 };
1816 return error_response(
1817 422,
1818 format!(
1819 "version bump too small: {} → {version} is a {declared:?} bump, \
1820 but the API change ({why}) requires a {need} bump",
1821 prev.version
1822 ),
1823 );
1824 }
1825 }
1826 }
1827 }
1828
1829 let head = store.branch_head(&branch).unwrap_or_default();
1832 let pairs: Vec<(String, String)> = head.iter().map(|(s, st)| (s.clone(), st.clone())).collect();
1833 let function_names: Vec<String> = store
1834 .get_asts_for_sigs_bulk(&pairs)
1835 .into_iter()
1836 .filter_map(|r| r.ok())
1837 .filter_map(|s| match s {
1838 lex_ast::Stage::FnDecl(fd) => Some(fd.name),
1839 _ => None,
1840 })
1841 .collect();
1842 drop(store);
1843
1844 let published_at = std::time::SystemTime::now()
1845 .duration_since(std::time::UNIX_EPOCH)
1846 .map(|d| d.as_secs())
1847 .unwrap_or(0);
1848 let record = PkgRecord {
1849 name: name.to_string(),
1850 version: version.clone(),
1851 head_op: Some(head_op.clone()),
1852 published_at,
1853 function_names,
1854 ops: Vec::new(),
1855 };
1856 if let Err(e) = save_pkg_record(&state.root, &record, None) {
1857 return error_response(500, format!("write release: {e}"));
1858 }
1859 json_response(
1860 201,
1861 &serde_json::json!({
1862 "name": name,
1863 "version": version,
1864 "head_op": head_op,
1865 "branch": branch,
1866 }),
1867 )
1868}
1869
1870fn public_pkg_names(root: &std::path::Path) -> Vec<String> {
1873 list_pkg_names(root)
1874 .into_iter()
1875 .filter(|name| pkg_is_public(root, name))
1876 .collect()
1877}
1878
1879fn public_pkg_list_handler(state: &State) -> Response<std::io::Cursor<Vec<u8>>> {
1883 let packages: Vec<serde_json::Value> = public_pkg_names(&state.root)
1884 .iter()
1885 .filter_map(|name| {
1886 let r = load_latest_pkg_record(&state.root, name)?;
1887 Some(serde_json::json!({
1888 "name": r.name,
1889 "version": r.version,
1890 "head_op": r.head_op,
1891 "published_at": r.published_at,
1892 }))
1893 })
1894 .collect();
1895 json_response(200, &serde_json::json!({ "packages": packages }))
1896}
1897
1898#[derive(Debug, PartialEq, Eq)]
1903enum PublicTarget {
1904 List,
1905 Latest(String),
1906 Versions(String),
1907 ApiDiff(String),
1908 Head(String),
1909 Version(String, String),
1910 Archive(String, String),
1911}
1912
1913impl PublicTarget {
1914 fn pkg_name(&self) -> Option<&str> {
1916 match self {
1917 PublicTarget::List => None,
1918 PublicTarget::Latest(n)
1919 | PublicTarget::Versions(n)
1920 | PublicTarget::ApiDiff(n)
1921 | PublicTarget::Head(n)
1922 | PublicTarget::Version(n, _)
1923 | PublicTarget::Archive(n, _) => Some(n),
1924 }
1925 }
1926}
1927
1928fn resolve_public(method: &Method, path: &str) -> Result<PublicTarget, u16> {
1933 if !matches!(method, Method::Get) {
1934 return Err(405);
1935 }
1936 let rest = path.trim_matches('/');
1937 if rest.is_empty() {
1938 return Ok(PublicTarget::List);
1939 }
1940 let segs: Vec<&str> = rest.split('/').collect();
1941 if !segs.iter().all(|s| valid_pkg_segment(s)) {
1942 return Err(404);
1943 }
1944 match segs.as_slice() {
1945 [n] => Ok(PublicTarget::Latest(n.to_string())),
1946 [n, "versions"] => Ok(PublicTarget::Versions(n.to_string())),
1947 [n, "api-diff"] => Ok(PublicTarget::ApiDiff(n.to_string())),
1948 [n, "head"] => Ok(PublicTarget::Head(n.to_string())),
1949 [n, v, "archive"] => Ok(PublicTarget::Archive(n.to_string(), v.to_string())),
1950 [n, v] => Ok(PublicTarget::Version(n.to_string(), v.to_string())),
1951 _ => Err(404),
1952 }
1953}
1954
1955pub fn route_public(
1966 state: &State,
1967 method: &Method,
1968 path: &str,
1969 query: &str,
1970) -> Response<std::io::Cursor<Vec<u8>>> {
1971 let target = match resolve_public(method, path) {
1972 Ok(t) => t,
1973 Err(405) => return error_response(405, "public read is GET-only"),
1974 Err(_) => return error_response(404, "not found"),
1975 };
1976 if let PublicTarget::List = target {
1978 return public_pkg_list_handler(state);
1979 }
1980 if let Some(name) = target.pkg_name() {
1982 if !pkg_is_public(&state.root, name) {
1983 return error_response(404, format!("package {name:?} not found"));
1984 }
1985 }
1986 match target {
1987 PublicTarget::List => unreachable!("handled above"),
1988 PublicTarget::Latest(n) => pkg_get_handler(state, &n),
1989 PublicTarget::Versions(n) => pkg_versions_handler(state, &n),
1990 PublicTarget::ApiDiff(n) => pkg_api_diff_handler(state, &n, query),
1991 PublicTarget::Head(n) => pkg_head_handler(state, &n),
1992 PublicTarget::Version(n, v) => pkg_get_version_handler(state, &n, &v),
1993 PublicTarget::Archive(n, v) => pkg_archive_handler(state, &n, &v),
1994 }
1995}
1996
1997fn save_pkg_record(
1998 root: &std::path::Path,
1999 record: &PkgRecord,
2000 archive: Option<&[u8]>,
2003) -> std::io::Result<()> {
2004 let dir = pkg_name_dir(root, &record.name);
2005 std::fs::create_dir_all(&dir)?;
2006
2007 let rec_bytes = serde_json::to_vec_pretty(record).unwrap_or_default();
2009 std::fs::write(pkg_version_path(root, &record.name, &record.version), rec_bytes)?;
2010
2011 if let Some(archive) = archive {
2013 std::fs::write(pkg_archive_path(root, &record.name, &record.version), archive)?;
2014 }
2015
2016 let mut index = load_pkg_index(root, &record.name).unwrap_or_default();
2018 index.latest = Some(record.version.clone());
2019 if !index.versions.iter().any(|v| v.version == record.version) {
2020 index.versions.push(PkgVersionSummary {
2021 version: record.version.clone(),
2022 head_op: record.head_op.clone(),
2023 published_at: record.published_at,
2024 });
2025 }
2026 let idx_bytes = serde_json::to_vec_pretty(&index).unwrap_or_default();
2027 std::fs::write(pkg_index_path(root, &record.name), idx_bytes)
2028}
2029
2030fn list_pkg_names(root: &std::path::Path) -> Vec<String> {
2031 let dir = root.join("packages");
2032 let Ok(entries) = std::fs::read_dir(&dir) else {
2033 return Vec::new();
2034 };
2035 let mut names: Vec<String> = entries
2036 .filter_map(|e| e.ok())
2037 .filter(|e| e.path().is_dir())
2038 .filter_map(|e| e.file_name().into_string().ok())
2039 .collect();
2040 names.sort();
2041 names
2042}
2043
2044fn collect_lex_files(dir: &std::path::Path, out: &mut Vec<PathBuf>) {
2045 let Ok(entries) = std::fs::read_dir(dir) else { return };
2046 let mut entries: Vec<_> = entries.filter_map(|e| e.ok()).collect();
2047 entries.sort_by_key(|e| e.path());
2048 for entry in entries {
2049 let path = entry.path();
2050 if path.is_dir() {
2051 collect_lex_files(&path, out);
2052 } else if path.extension().and_then(|x| x.to_str()) == Some("lex") {
2053 out.push(path);
2054 }
2055 }
2056}
2057
2058fn pkg_publish_handler(state: &State, body: &[u8]) -> Response<std::io::Cursor<Vec<u8>>> {
2061 let tmp = match tempfile::TempDir::new() {
2062 Ok(t) => t,
2063 Err(e) => return error_response(500, format!("create temp dir: {e}")),
2064 };
2065 {
2066 let gz = flate2::read::GzDecoder::new(std::io::Cursor::new(body));
2067 let mut ar = tar::Archive::new(gz);
2068 if let Err(e) = ar.unpack(tmp.path()) {
2069 return error_response(400, format!("unpack archive: {e}"));
2070 }
2071 }
2072
2073 let toml_path = tmp.path().join("lex.toml");
2074 if !toml_path.exists() {
2075 return error_response(400, "archive must contain lex.toml at root");
2076 }
2077 let manifest = match Manifest::load(&toml_path) {
2078 Ok(m) => m,
2079 Err(e) => return error_response(400, format!("lex.toml: {e}")),
2080 };
2081 let (pkg_name, pkg_version) = match &manifest.package {
2082 Some(m) => (m.name.clone(), m.version.clone()),
2083 None => return error_response(400, "lex.toml must have a [package] section"),
2084 };
2085
2086 if load_pkg_record(&state.root, &pkg_name, &pkg_version).is_some() {
2090 return error_response(
2091 409,
2092 format!(
2093 "package {pkg_name}@{pkg_version} already published; \
2094 bump the version in lex.toml to publish a new release"
2095 ),
2096 );
2097 }
2098
2099 let src_dir = tmp.path().join("src");
2100 if !src_dir.exists() {
2101 return error_response(400, "archive must contain a src/ directory");
2102 }
2103 let mut lex_files: Vec<PathBuf> = Vec::new();
2104 collect_lex_files(&src_dir, &mut lex_files);
2105 if lex_files.is_empty() {
2106 return error_response(400, "no .lex files found in src/");
2107 }
2108
2109 let store = state.store.lock().unwrap();
2110 let branch = store.current_branch();
2111
2112 let old_head = match store.branch_head(&branch) {
2132 Ok(h) => h,
2133 Err(e) => return error_response(500, format!("branch_head: {e}")),
2134 };
2135 let old_pairs: Vec<(String, String)> =
2142 old_head.iter().map(|(sig, stage)| (sig.clone(), stage.clone())).collect();
2143 let mut old_fns_by_name: BTreeMap<String, Vec<lex_ast::FnDecl>> = BTreeMap::new();
2144 for fd in store.get_asts_for_sigs_bulk(&old_pairs)
2145 .into_iter()
2146 .filter_map(|r| r.ok())
2147 .filter_map(|s| match s { lex_ast::Stage::FnDecl(fd) => Some(fd), _ => None })
2148 {
2149 old_fns_by_name.entry(fd.name.clone()).or_default().push(fd);
2150 }
2151 let mut old_types_by_name: BTreeMap<String, lex_ast::TypeDecl> = BTreeMap::new();
2154 for td in store.get_asts_for_sigs_bulk(&old_pairs)
2155 .into_iter()
2156 .filter_map(|r| r.ok())
2157 .filter_map(|s| match s { lex_ast::Stage::TypeDecl(td) => Some(td), _ => None })
2158 {
2159 old_types_by_name.insert(td.name.clone(), td);
2160 }
2161 fn structural_key(fd: &lex_ast::FnDecl) -> Option<String> {
2169 let mut anon = fd.clone();
2170 anon.name = String::new();
2171 lex_ast::sig_id(&lex_ast::Stage::FnDecl(anon))
2172 }
2173
2174 fn take_matching(
2184 map: &mut BTreeMap<String, Vec<lex_ast::FnDecl>>,
2185 name: &str,
2186 new_fd: &lex_ast::FnDecl,
2187 ) -> Option<lex_ast::FnDecl> {
2188 let candidates = map.get_mut(name)?;
2189 let idx = match candidates.len() {
2190 0 => return None,
2191 1 => 0,
2192 _ => {
2193 let want = structural_key(new_fd);
2194 candidates.iter().position(|c| structural_key(c) == want)?
2195 }
2196 };
2197 let matched = candidates.remove(idx);
2198 if candidates.is_empty() {
2199 map.remove(name);
2200 }
2201 Some(matched)
2202 }
2203
2204 let loaded = match load_package(&lex_files, tmp.path(), &pkg_name) {
2224 Ok(p) => p,
2225 Err(e) => return error_response(400, format!("load package: {e}")),
2226 };
2227 let mut stages = canonicalize_program(&loaded.program);
2228 if let Err(errs) = lex_types::check_and_rewrite_program(&mut stages) {
2234 return error_with_detail(
2235 422,
2236 format!("type errors in package {pkg_name}"),
2237 serde_json::to_value(&errs).unwrap(),
2238 );
2239 }
2240 let new_fns = stage_fns(&stages);
2241 let all_function_names: Vec<String> = new_fns.keys().cloned().collect();
2242
2243 let mut old_fns: BTreeMap<String, lex_ast::FnDecl> = BTreeMap::new();
2248 for (name, new_fd) in &new_fns {
2249 if let Some(fd) = take_matching(&mut old_fns_by_name, name, new_fd) {
2250 old_fns.insert(name.clone(), fd);
2251 }
2252 }
2253 let new_types = stage_types(&stages);
2254 let old_types: BTreeMap<String, lex_ast::TypeDecl> = new_types
2258 .keys()
2259 .filter_map(|n| old_types_by_name.get(n).map(|td| (n.clone(), td.clone())))
2260 .collect();
2261 let report =
2262 lex_vcs::compute_diff_with_types(&old_fns, &new_fns, &old_types, &new_types, false);
2263
2264 let mut new_imports = lex_vcs::ImportMap::new();
2274 for (file, modules) in &loaded.imports_by_file {
2275 let entry = new_imports.entry(file.clone()).or_default();
2276 for m in modules {
2277 entry.insert(lex_vcs::ImportRef {
2278 reference: m.clone(),
2279 alias: lex_vcs::default_import_alias(m),
2280 });
2281 }
2282 }
2283
2284 let outcome = match store.publish_program_with_intent(
2287 &branch,
2288 &stages,
2289 &report,
2290 &new_imports,
2291 false,
2292 None,
2293 None,
2294 &loaded.module_prefixes,
2295 ) {
2296 Ok(outcome) => outcome,
2297 Err(lex_store::StoreError::TypeError(errs)) => {
2298 return error_with_detail(422, "type errors", serde_json::to_value(&errs).unwrap());
2299 }
2300 Err(e) => return write_error_response("publish_program", e),
2301 };
2302 let all_ops: Vec<serde_json::Value> = match serde_json::to_value(&outcome.ops) {
2303 Ok(serde_json::Value::Array(arr)) => arr,
2304 _ => Vec::new(),
2305 };
2306 let final_head_op = outcome.head_op;
2307
2308 let now = SystemTime::now()
2333 .duration_since(UNIX_EPOCH)
2334 .map(|d| d.as_secs())
2335 .unwrap_or(0);
2336 let record = PkgRecord {
2337 name: pkg_name.clone(),
2338 version: pkg_version,
2339 head_op: final_head_op.clone(),
2340 published_at: now,
2341 function_names: all_function_names,
2342 ops: all_ops.clone(),
2343 };
2344 if let Err(e) = save_pkg_record(&state.root, &record, Some(body)) {
2345 return error_response(500, format!("save package index: {e}"));
2346 }
2347
2348 json_response(200, &serde_json::json!({
2349 "package": pkg_name,
2350 "ops": all_ops,
2351 "head_op": final_head_op,
2352 }))
2353}
2354
2355fn pkg_list_handler(state: &State) -> Response<std::io::Cursor<Vec<u8>>> {
2357 let names = list_pkg_names(&state.root);
2358 let packages: Vec<serde_json::Value> = names.iter()
2359 .filter_map(|name| {
2360 let idx = load_pkg_index(&state.root, name)?;
2361 let latest = idx.latest.as_deref()?;
2362 let r = load_pkg_record(&state.root, name, latest)?;
2363 Some(serde_json::json!({
2364 "name": r.name,
2365 "version": r.version,
2366 "head_op": r.head_op,
2367 "published_at": r.published_at,
2368 }))
2369 })
2370 .collect();
2371 json_response(200, &serde_json::json!({ "packages": packages }))
2372}
2373
2374fn pkg_get_handler(state: &State, name: &str) -> Response<std::io::Cursor<Vec<u8>>> {
2376 match load_latest_pkg_record(&state.root, name) {
2377 Some(r) => json_response(200, &serde_json::json!({
2378 "name": r.name,
2379 "version": r.version,
2380 "head_op": r.head_op,
2381 "published_at": r.published_at,
2382 "function_names": r.function_names,
2383 "ops": r.ops,
2384 })),
2385 None => error_response(404, format!("package {name:?} not found")),
2386 }
2387}
2388
2389fn pkg_versions_handler(state: &State, name: &str) -> Response<std::io::Cursor<Vec<u8>>> {
2391 match load_pkg_index(&state.root, name) {
2392 Some(idx) => json_response(200, &serde_json::json!({
2393 "name": name,
2394 "latest": idx.latest,
2395 "versions": idx.versions,
2396 })),
2397 None => error_response(404, format!("package {name:?} not found")),
2398 }
2399}
2400
2401fn pkg_api_diff_handler(state: &State, name: &str, query: &str) -> Response<std::io::Cursor<Vec<u8>>> {
2406 let mut from: Option<String> = None;
2407 let mut to: Option<String> = None;
2408 for kv in query.split('&') {
2409 match kv.split_once('=') {
2410 Some(("from", v)) => from = Some(v.to_string()),
2411 Some(("to", v)) => to = Some(v.to_string()),
2412 _ => {}
2413 }
2414 }
2415 let (Some(from), Some(to)) = (from, to) else {
2416 return error_response(400, "api-diff requires ?from=<version>&to=<version>");
2417 };
2418 let head_of = |v: &str| load_pkg_record(&state.root, name, v).and_then(|r| r.head_op);
2419 let (Some(from_head), Some(to_head)) = (head_of(&from), head_of(&to)) else {
2420 return error_response(404, format!("{name}: unknown release in {from}..{to}"));
2421 };
2422
2423 let store = state.store.lock().unwrap();
2424 let (prev_api, new_api) = match (
2425 lex_store::api::public_api_at_op(&store, &from_head),
2426 lex_store::api::public_api_at_op(&store, &to_head),
2427 ) {
2428 (Ok(a), Ok(b)) => (a, b),
2429 _ => return error_response(500, "could not read package APIs for the given releases"),
2430 };
2431 let (change, detail) = match lex_store::api::classify_api_change(&prev_api, &new_api) {
2432 lex_store::api::ApiChange::Breaking(d) => ("breaking", d),
2433 lex_store::api::ApiChange::Additive(d) => ("additive", d),
2434 lex_store::api::ApiChange::None => ("none", String::new()),
2435 };
2436 let renames = lex_store::api::detect_renames(&prev_api, &new_api);
2437 json_response(200, &serde_json::json!({
2438 "name": name, "from": from, "to": to,
2439 "change": change, "detail": detail, "renames": renames,
2440 }))
2441}
2442
2443fn pkg_get_version_handler(state: &State, name: &str, version: &str) -> Response<std::io::Cursor<Vec<u8>>> {
2445 match load_pkg_record(&state.root, name, version) {
2446 Some(r) => json_response(200, &serde_json::json!({
2447 "name": r.name,
2448 "version": r.version,
2449 "head_op": r.head_op,
2450 "published_at": r.published_at,
2451 "function_names": r.function_names,
2452 "ops": r.ops,
2453 })),
2454 None => error_response(404, format!("package {name:?}@{version:?} not found")),
2455 }
2456}
2457
2458fn pkg_archive_handler(state: &State, name: &str, version: &str) -> Response<std::io::Cursor<Vec<u8>>> {
2460 let gzip = |bytes: Vec<u8>| {
2461 Response::from_data(bytes).with_status_code(200).with_header(
2462 tiny_http::Header::from_bytes(&b"Content-Type"[..], &b"application/gzip"[..]).unwrap(),
2463 )
2464 };
2465
2466 if let Ok(bytes) = std::fs::read(pkg_archive_path(&state.root, name, version)) {
2468 return gzip(bytes);
2469 }
2470
2471 if let Some(head_op) = load_pkg_record(&state.root, name, version).and_then(|r| r.head_op) {
2475 match render_op_log_archive(state, name, version, &head_op) {
2476 Ok(bytes) => return gzip(bytes),
2477 Err(e) => {
2478 return error_response(500, format!("rendering archive for {name:?}@{version:?}: {e}"));
2479 }
2480 }
2481 }
2482
2483 error_response(404, format!("archive for {name:?}@{version:?} not found"))
2484}
2485
2486fn render_op_log_archive(
2492 state: &State,
2493 name: &str,
2494 version: &str,
2495 head_op: &str,
2496) -> Result<Vec<u8>, String> {
2497 let files: Vec<(String, String)> = {
2502 let store = state.store.lock().unwrap();
2503 let head = lex_store::render::package_head_at_op(&store, head_op)
2504 .map_err(|e| format!("reading head {head_op}: {e}"))?;
2505 match lex_store::render::render_source(&store, &head)
2506 .map_err(|e| format!("rendering source at {head_op}: {e}"))?
2507 {
2508 lex_store::render::RenderedSource::Single(src) => vec![("src/lib.lex".to_string(), src)],
2509 lex_store::render::RenderedSource::Multi(tree) => tree.into_iter().collect(),
2510 }
2511 };
2512
2513 let manifest = format!("[package]\nname = \"{name}\"\nversion = \"{version}\"\n");
2514 let mut enc = flate2::write::GzEncoder::new(Vec::new(), flate2::Compression::default());
2515 {
2516 let mut ar = tar::Builder::new(&mut enc);
2517 let mut append = |p: &str, data: &[u8]| -> std::io::Result<()> {
2518 let mut h = tar::Header::new_gnu();
2519 h.set_size(data.len() as u64);
2520 h.set_mode(0o644);
2521 h.set_cksum();
2522 ar.append_data(&mut h, p, data)
2523 };
2524 append("lex.toml", manifest.as_bytes()).map_err(|e| e.to_string())?;
2525 for (path, src) in &files {
2526 append(path, src.as_bytes()).map_err(|e| e.to_string())?;
2527 }
2528 ar.finish().map_err(|e| e.to_string())?;
2529 }
2530 enc.finish().map_err(|e| e.to_string())
2531}
2532
2533fn pkg_head_handler(state: &State, name: &str) -> Response<std::io::Cursor<Vec<u8>>> {
2535 match load_latest_pkg_record(&state.root, name) {
2536 Some(r) => json_response(200, &serde_json::json!({
2537 "name": r.name,
2538 "version": r.version,
2539 "head_op": r.head_op,
2540 })),
2541 None => error_response(404, format!("package {name:?} not found")),
2542 }
2543}
2544
2545fn pkg_delete_handler(state: &State, name: &str) -> Response<std::io::Cursor<Vec<u8>>> {
2547 let record = match load_latest_pkg_record(&state.root, name) {
2548 Some(r) => r,
2549 None => return error_response(404, format!("package {name:?} not found")),
2550 };
2551
2552 let store = state.store.lock().unwrap();
2553 let branch = store.current_branch();
2554
2555 let head = match store.branch_head(&branch) {
2556 Ok(h) => h,
2557 Err(e) => return error_response(500, format!("branch_head: {e}")),
2558 };
2559
2560 let head_pairs: Vec<(String, String)> = head
2567 .iter()
2568 .map(|(sig, stage)| (sig.clone(), stage.clone()))
2569 .collect();
2570 let old_fns: BTreeMap<String, lex_ast::FnDecl> = store
2571 .get_asts_for_sigs_bulk(&head_pairs)
2572 .into_iter()
2573 .filter_map(|r| r.ok())
2574 .filter_map(|s| match s {
2575 lex_ast::Stage::FnDecl(fd)
2576 if record.function_names.contains(&fd.name) => Some((fd.name.clone(), fd)),
2577 _ => None,
2578 })
2579 .collect();
2580
2581 let new_fns: BTreeMap<String, lex_ast::FnDecl> = BTreeMap::new();
2582 let report = lex_vcs::compute_diff(&old_fns, &new_fns, false);
2583 let empty_imports = lex_vcs::ImportMap::new();
2584
2585 match store.publish_program(&branch, &[], &report, &empty_imports, false) {
2586 Ok(outcome) => {
2587 let ver = record.version.clone();
2589 let _ = std::fs::remove_file(pkg_version_path(&state.root, name, &ver));
2590 let _ = std::fs::remove_file(pkg_archive_path(&state.root, name, &ver));
2591 if let Some(mut idx) = load_pkg_index(&state.root, name) {
2593 idx.versions.retain(|v| v.version != ver);
2594 idx.latest = idx.versions.last().map(|v| v.version.clone());
2595 if idx.versions.is_empty() {
2596 let _ = std::fs::remove_dir_all(pkg_name_dir(&state.root, name));
2597 } else {
2598 let bytes = serde_json::to_vec_pretty(&idx).unwrap_or_default();
2599 let _ = std::fs::write(pkg_index_path(&state.root, name), bytes);
2600 }
2601 }
2602 json_response(200, &serde_json::json!({
2603 "deleted": name,
2604 "version": ver,
2605 "ops": outcome.ops,
2606 "head_op": outcome.head_op,
2607 }))
2608 }
2609 Err(lex_store::StoreError::TypeError(errs)) => {
2610 error_with_detail(422, "type errors", serde_json::to_value(&errs).unwrap())
2611 }
2612 Err(e) => write_error_response("retract package", e),
2613 }
2614}
2615
2616#[cfg(test)]
2617mod policy_ceiling_tests {
2618 use super::*;
2619 use lex_runtime::Policy;
2620 use std::path::PathBuf;
2621
2622 fn permissive_request() -> Policy {
2626 Policy {
2627 allow_effects: ["io", "fs_read", "fs_write", "net", "proc"]
2628 .iter()
2629 .map(|s| s.to_string())
2630 .collect(),
2631 allow_fs_read: vec![PathBuf::from("/")],
2632 allow_fs_write: vec![PathBuf::from("/")],
2633 allow_net_host: Vec::new(),
2634 allow_proc: Vec::new(),
2635 allow_approval: Vec::new(),
2636 budget: None,
2637 }
2638 }
2639
2640 #[test]
2641 fn ceiling_drops_effects_the_caller_was_not_granted() {
2642 let ceiling = Policy {
2643 allow_effects: ["io", "time"].iter().map(|s| s.to_string()).collect(),
2644 ..Policy::default()
2645 };
2646 let got = clamp_policy(permissive_request(), &ceiling);
2647 assert!(got.allow_effects.contains("io"));
2648 assert!(!got.allow_effects.contains("proc"), "proc must not survive a ceiling without it");
2649 assert!(!got.allow_effects.contains("fs_write"));
2650 assert!(!got.allow_effects.contains("net"));
2651 assert!(!got.allow_effects.contains("time"));
2653 }
2654
2655 #[test]
2656 fn ceiling_scopes_override_caller_scopes() {
2657 let ceiling = Policy {
2658 allow_effects: ["fs_read"].iter().map(|s| s.to_string()).collect(),
2659 allow_fs_read: vec![PathBuf::from("/srv/tenant")],
2660 ..Policy::default()
2661 };
2662 let got = clamp_policy(permissive_request(), &ceiling);
2663 assert_eq!(got.allow_fs_read, vec![PathBuf::from("/srv/tenant")]);
2666 assert!(got.allow_fs_write.is_empty());
2667 assert!(got.allow_proc.is_empty());
2668 assert!(got.allow_net_host.is_empty());
2669 }
2670
2671 #[test]
2672 fn ceiling_caps_budget_and_prefers_the_smaller() {
2673 let mut req = permissive_request();
2675 req.budget = None;
2676 let ceiling = Policy { budget: Some(1_000), ..Policy::default() };
2677 assert_eq!(clamp_policy(req, &ceiling).budget, Some(1_000));
2678
2679 let mut req2 = permissive_request();
2681 req2.budget = Some(50);
2682 let ceiling2 = Policy { budget: Some(1_000), ..Policy::default() };
2683 assert_eq!(clamp_policy(req2, &ceiling2).budget, Some(50));
2684 }
2685
2686 #[test]
2687 fn empty_ceiling_is_pure_only() {
2688 let got = clamp_policy(permissive_request(), &Policy::default());
2689 assert!(got.allow_effects.is_empty(), "an empty ceiling grants nothing");
2690 assert!(got.allow_proc.is_empty());
2691 assert!(got.allow_fs_write.is_empty());
2692 }
2693}
2694
2695#[cfg(test)]
2696mod public_read_tests {
2697 use super::*;
2698
2699 fn seed_pkg(root: &std::path::Path, name: &str, version: &str) {
2702 let record = PkgRecord {
2703 name: name.to_string(),
2704 version: version.to_string(),
2705 head_op: Some(format!("op-{name}")),
2706 published_at: 1,
2707 function_names: vec![format!("{name}.f")],
2708 ops: vec![],
2709 };
2710 save_pkg_record(root, &record, Some(format!("ARCHIVE:{name}@{version}").as_bytes()))
2711 .expect("seed package");
2712 }
2713
2714 #[test]
2715 fn new_package_defaults_to_private() {
2716 let tmp = tempfile::TempDir::new().unwrap();
2717 seed_pkg(tmp.path(), "lex-schema", "0.9.2");
2718 assert!(!pkg_is_public(tmp.path(), "lex-schema"));
2719 assert!(!pkg_is_public(tmp.path(), "does-not-exist"));
2721 }
2722
2723 #[test]
2724 fn set_visibility_round_trips_and_index_persists() {
2725 let tmp = tempfile::TempDir::new().unwrap();
2726 let state = State::open(tmp.path().to_path_buf()).unwrap();
2727 seed_pkg(tmp.path(), "lex-schema", "0.9.2");
2728
2729 let _ = pkg_set_visibility_handler(&state, "lex-schema", r#"{"visibility":"public"}"#);
2730 assert!(pkg_is_public(tmp.path(), "lex-schema"));
2731 let idx = load_pkg_index(tmp.path(), "lex-schema").unwrap();
2733 assert_eq!(idx.latest.as_deref(), Some("0.9.2"));
2734 assert_eq!(idx.versions.len(), 1);
2735
2736 let _ = pkg_set_visibility_handler(&state, "lex-schema", r#"{"visibility":"private"}"#);
2737 assert!(!pkg_is_public(tmp.path(), "lex-schema"));
2738 }
2739
2740 #[test]
2741 fn set_visibility_on_unknown_package_is_a_noop() {
2742 let tmp = tempfile::TempDir::new().unwrap();
2743 let state = State::open(tmp.path().to_path_buf()).unwrap();
2744 let _ = pkg_set_visibility_handler(&state, "ghost", r#"{"visibility":"public"}"#);
2746 assert!(load_pkg_index(tmp.path(), "ghost").is_none());
2747 }
2748
2749 #[test]
2750 fn public_listing_omits_private_packages() {
2751 let tmp = tempfile::TempDir::new().unwrap();
2752 let state = State::open(tmp.path().to_path_buf()).unwrap();
2753 seed_pkg(tmp.path(), "pub-pkg", "1.0.0");
2754 seed_pkg(tmp.path(), "priv-pkg", "1.0.0");
2755 let _ = pkg_set_visibility_handler(&state, "pub-pkg", r#"{"visibility":"public"}"#);
2756
2757 let names = public_pkg_names(tmp.path());
2758 assert_eq!(names, vec!["pub-pkg".to_string()]);
2759 }
2760
2761 #[test]
2762 fn resolve_public_maps_routes() {
2763 let get = Method::Get;
2764 assert_eq!(resolve_public(&get, "").unwrap(), PublicTarget::List);
2765 assert_eq!(resolve_public(&get, "/").unwrap(), PublicTarget::List);
2766 assert_eq!(
2767 resolve_public(&get, "/lex-schema").unwrap(),
2768 PublicTarget::Latest("lex-schema".into())
2769 );
2770 assert_eq!(
2771 resolve_public(&get, "/lex-schema/versions").unwrap(),
2772 PublicTarget::Versions("lex-schema".into())
2773 );
2774 assert_eq!(
2775 resolve_public(&get, "/lex-schema/head").unwrap(),
2776 PublicTarget::Head("lex-schema".into())
2777 );
2778 assert_eq!(
2779 resolve_public(&get, "/lex-schema/0.9.2").unwrap(),
2780 PublicTarget::Version("lex-schema".into(), "0.9.2".into())
2781 );
2782 assert_eq!(
2783 resolve_public(&get, "/lex-schema/0.9.2/archive").unwrap(),
2784 PublicTarget::Archive("lex-schema".into(), "0.9.2".into())
2785 );
2786 }
2787
2788 #[test]
2789 fn resolve_public_rejects_bad_method_and_traversal() {
2790 assert_eq!(resolve_public(&Method::Put, "/lex-schema"), Err(405));
2792 assert_eq!(resolve_public(&Method::Post, "").err(), Some(405));
2793 assert_eq!(resolve_public(&Method::Get, "/.."), Err(404));
2795 assert_eq!(resolve_public(&Method::Get, "/lex-schema/../etc"), Err(404));
2796 assert_eq!(resolve_public(&Method::Get, "/a/b/c/d"), Err(404));
2797 assert!(resolve_public(&Method::Get, "/lex schema").is_err());
2799 }
2800}