1use crate::surface::{build_surface, build_surface_cached, SurfaceMode, SurfacePolicy, SurfaceRequest};
10use crate::ContextCompiler;
11use scc_core::kinds;
12use scc_core::{estimate_tokens, ContextArtifact, ContextBudget, ContextLedger};
13use serde::{Deserialize, Serialize};
14use std::collections::{BTreeMap, BTreeSet};
15
16pub const RENDERER_VERSION: &str = env!("CARGO_PKG_VERSION");
19
20pub struct StartupContext {
26 pub atlas: String,
27 pub atlas_budget_used: usize,
32 pub skeleton: String,
35 pub surface: String,
36 pub important: String,
40 pub surface_render: scc_core::SurfaceRenderResult,
41 pub coverage: Vec<String>,
42 pub omissions: Vec<String>,
43 pub artifact: ContextArtifact,
44}
45#[derive(Debug, Clone, PartialEq, Serialize, Deserialize)]
61pub struct GlobalRankCache {
63 pub epoch: String,
65 pub policy: String,
67 pub salt: String,
69 pub global_vector: Vec<f64>,
72 pub node_symbol_map: BTreeMap<String, f64>,
75 pub candidates_epoch: String,
78 #[serde(default, skip_serializing_if = "Vec::is_empty")]
81 pub candidate_ids: Vec<String>,
82 #[serde(default)]
85 pub hits: u64,
86}
87
88pub fn allocate_startup_budget(
102 compiler: &ContextCompiler,
103 target_tokens: Option<usize>,
104) -> ContextBudget {
105 let total = target_tokens.unwrap_or(compiler.settings.startup_tokens);
106 let view = &compiler.view;
107 let entity_count = view.entities().count();
108 let component_count = view.components().len();
109 let flow_count = view.flows().len();
110 let surface_candidates = view.entities_of_kind(scc_core::kinds::SYMBOL).len();
111 scc_core::ContextBudget::adaptive(total, entity_count, component_count, flow_count, surface_candidates)
112}
113
114fn coverage_lines(
116 compiler: &ContextCompiler,
117 render_ids: usize,
118 candidates: usize,
119 render_tokens: usize,
120 surface_budget: usize,
121) -> Vec<String> {
122 let mut coverage = Vec::new();
123 for w in compiler_warnings(compiler) {
124 coverage.push(w);
125 }
126 let mut stale: Vec<String> = compiler.stale_paths.clone();
127 stale.sort();
128 stale.dedup();
129 const MAX_STALE_LINES: usize = 10;
132 for p in stale.iter().take(MAX_STALE_LINES) {
133 coverage.push(format!("stale: {p}"));
134 }
135 if stale.len() > MAX_STALE_LINES {
136 coverage.push(format!(
137 "stale: …and {} more changed file(s) not yet re-indexed",
138 stale.len() - MAX_STALE_LINES
139 ));
140 }
141 coverage.push(format!(
142 "surface map: {} of {} entries rendered, {} tokens (budget {})",
143 render_ids,
144 candidates,
145 render_tokens,
146 surface_budget
147 ));
148 coverage
149}
150
151fn omission_lines(
156 render_omissions: &[scc_core::SurfaceOmission],
157 omitted_len: usize,
158 atlas_dropped: &[String],
159 atlas_hard_truncated: bool,
160 atlas_exceeded: bool,
161) -> Vec<String> {
162 let mut omissions = Vec::new();
163 for o in render_omissions {
164 omissions.push(format!("surface: {} ({})", o.kind, o.reason));
165 }
166 if omitted_len > 0 {
167 omissions.push(format!(
168 "surface: {} lower-ranked definitions omitted",
169 omitted_len
170 ));
171 }
172 for d in atlas_dropped {
173 omissions.push(format!("atlas section dropped: {d}"));
174 }
175 if atlas_hard_truncated {
176 omissions.push("atlas hard-truncated mid-section".into());
177 }
178 if atlas_exceeded {
179 omissions.push("atlas exceeded soft budget: kept complete, over budget".into());
180 }
181 if omissions.is_empty() {
182 omissions.push("none".into());
183 }
184 omissions
185}
186
187pub fn build_startup(
194 compiler: &ContextCompiler,
195 budget: &ContextBudget,
196 renderer_version: &str,
197) -> StartupContext {
198 let epoch = compiler
199 .store
200 .cache_epoch()
201 .unwrap_or_else(|_| "no-epoch".into());
202
203 let atlas_pack = compiler.system_atlas(Some(budget.atlas));
206 let atlas = atlas_pack.content.clone();
207
208 let paths: Vec<String> = compiler
213 .store
214 .all_files()
215 .unwrap_or_default()
216 .into_iter()
217 .map(|(p, _, _, _, _)| p)
218 .collect();
219 let mut skeleton =
220 crate::skeleton::build_skeleton(&paths, crate::skeleton::skeleton_budget(budget.total)).text;
221
222 let mut rank_cache = load_global_rank_cache(compiler);
230 let cache_hit = rank_cache.is_some();
231 let render = build_surface_cached(
232 compiler,
233 SurfaceRequest {
234 mode: SurfaceMode::Global,
235 budget: budget.surface,
236 explain: false,
237 policy: SurfacePolicy::defaults(budget.surface),
238 semantic: None,
239 },
240 &mut rank_cache,
241 );
242 let startup_hard_max = budget.total.saturating_add((budget.total / 5).max(500));
249
250 let mut render = render;
251 let mut atlas_pack = atlas_pack;
252 let mut atlas = atlas;
253 let mut atlas_budget = budget.atlas;
254 let mut coverage = coverage_lines(
259 compiler,
260 render.rendered_ids.len(),
261 render.rendered_ids.len() + render.omitted_ids.len(),
262 render.token_count,
263 budget.surface,
264 );
265
266 const BLOCK_HEADER_OVERHEAD: usize = 40; let mut iterations = 0;
289 loop {
290 let omissions_probe = omission_lines(
291 &render.omissions,
292 render.omitted_ids.len(),
293 &atlas_pack.dropped_sections,
294 atlas_pack.hard_truncated,
295 atlas_pack.exceeded_soft_budget,
296 );
297 let fused = assemble_body(&atlas, &skeleton, &render.text, "", &coverage, &omissions_probe);
298 if BLOCK_HEADER_OVERHEAD + estimate_tokens(&fused) <= startup_hard_max {
299 break;
300 }
301 iterations += 1;
302 if iterations > 64 {
303 break;
309 }
310 let overhead = estimate_tokens(&assemble_body(&atlas, &skeleton, "", "", &coverage, &omissions_probe));
311 let room = startup_hard_max.saturating_sub(overhead);
312 if room >= 64 {
313 let policy = SurfacePolicy {
316 quotas: true,
317 mmr: true,
318 coverage: true,
319 hard_max: room,
320 };
321 render = build_surface_cached(
322 compiler,
323 SurfaceRequest {
324 mode: SurfaceMode::Global,
325 budget: room,
326 explain: false,
327 policy,
328 semantic: None,
329 },
330 &mut rank_cache,
331 );
332 coverage = coverage_lines(
333 compiler,
334 render.rendered_ids.len(),
335 render.rendered_ids.len() + render.omitted_ids.len(),
336 render.token_count,
337 room,
338 );
339 } else {
340 let smaller = atlas_budget.saturating_mul(3) / 4;
342 if smaller == atlas_budget {
343 break; }
345 atlas_budget = smaller;
346 atlas_pack = compiler.system_atlas(Some(atlas_budget));
347 atlas = atlas_pack.content.clone();
348 }
349 }
350
351 if let Some(c) = &mut rank_cache {
355 if cache_hit {
356 c.hits += 1;
357 }
358 store_global_rank_cache(compiler, c);
359 }
360 let mut surface = render.text.clone();
361 let mut atlas_budget_used = atlas_budget;
362
363 let emergency_omissions: Option<Vec<String>>;
373 {
374 let omissions_probe0 = omission_lines(
375 &render.omissions,
376 render.omitted_ids.len(),
377 &atlas_pack.dropped_sections,
378 atlas_pack.hard_truncated,
379 atlas_pack.exceeded_soft_budget,
380 );
381 if BLOCK_HEADER_OVERHEAD
382 + estimate_tokens(&assemble_body(
383 &atlas,
384 &skeleton,
385 &surface,
386 "",
387 &coverage,
388 &omissions_probe0,
389 ))
390 <= startup_hard_max
391 {
392 emergency_omissions = None;
393 } else {
394
395 atlas_budget_used = 256;
396 atlas_pack = compiler.system_atlas(Some(atlas_budget_used));
397 atlas = atlas_pack.content.clone();
398 render.rendered_ids.clear();
399 render.omitted_ids.clear();
400 render.omissions.clear();
401 render.text = String::new();
402 render.token_count = 0;
403 surface = String::new();
404 coverage = compiler_warnings(compiler);
407 if compiler.stale_paths.is_empty() {
408 coverage.push("model: current".into());
409 } else {
410 coverage.push(format!(
411 "model stale: {} changed file(s) not yet re-indexed (paths omitted over hard max)",
412 compiler.stale_paths.len()
413 ));
414 }
415 coverage.push("surface map: omitted over hard max (0 rendered)".into());
416 let mut skel_budget = crate::skeleton::skeleton_budget(budget.total);
417 loop {
418 let probe = crate::skeleton::build_skeleton(&paths, skel_budget);
419 let omissions_probe = {
420 let mut o = omission_lines(
421 &[],
422 0,
423 &atlas_pack.dropped_sections,
424 atlas_pack.hard_truncated,
425 atlas_pack.exceeded_soft_budget,
426 );
427 o.push(
428 "startup emergency compression: atlas essentials + skeleton only (surface omitted over hard max)".into(),
429 );
430 o };
431 let fused = assemble_body(&atlas, &probe.text, "", "", &coverage, &omissions_probe);
432 if BLOCK_HEADER_OVERHEAD + estimate_tokens(&fused) <= startup_hard_max
433 || skel_budget == 0
434 {
435 skeleton = probe.text;
436 emergency_omissions = Some(omissions_probe);
437 break;
438 }
439 skel_budget /= 2;
440 }
441 }
442 }
443
444 let important_n = (startup_hard_max / 500).clamp(1, 15);
455 let mut important = if render.rendered_ids.is_empty() {
456 "## SYSTEM-CRITICAL SYMBOLS\n(surface omitted over hard max)\n".to_string()
457 } else {
458 let top = crate::surface::important_symbols(
459 compiler,
460 crate::surface::SurfaceMode::Global,
461 important_n,
462 );
463 crate::surface::render_important(&top, false)
464 };
465 {
466 let probe_om = omission_lines(
467 &render.omissions,
468 render.omitted_ids.len(),
469 &atlas_pack.dropped_sections,
470 atlas_pack.hard_truncated,
471 atlas_pack.exceeded_soft_budget,
472 );
473 let fused = assemble_body(&atlas, &skeleton, &render.text, &important, &coverage, &probe_om);
474 if BLOCK_HEADER_OVERHEAD + estimate_tokens(&fused) > startup_hard_max && !render.rendered_ids.is_empty() {
475 let over = BLOCK_HEADER_OVERHEAD + estimate_tokens(&fused) - startup_hard_max;
476 let room = render
477 .token_count
478 .saturating_sub(over)
479 .saturating_sub(estimate_tokens(&important));
480 if room >= 64 {
481 let policy = SurfacePolicy {
482 quotas: true,
483 mmr: true,
484 coverage: true,
485 hard_max: room,
486 };
487 render = build_surface_cached(
488 compiler,
489 SurfaceRequest {
490 mode: SurfaceMode::Global,
491 budget: room,
492 explain: false,
493 policy,
494 semantic: None,
495 },
496 &mut rank_cache,
497 );
498 coverage = coverage_lines(
499 compiler,
500 render.rendered_ids.len(),
501 render.rendered_ids.len() + render.omitted_ids.len(),
502 render.token_count,
503 room,
504 );
505 important = if render.rendered_ids.is_empty() {
506 "## SYSTEM-CRITICAL SYMBOLS\n(surface omitted over hard max)\n".to_string()
507 } else {
508 let top = crate::surface::important_symbols(
509 compiler,
510 crate::surface::SurfaceMode::Global,
511 important_n,
512 );
513 crate::surface::render_important(&top, false)
514 };
515 let probe_om2 = omission_lines(
518 &render.omissions,
519 render.omitted_ids.len(),
520 &atlas_pack.dropped_sections,
521 atlas_pack.hard_truncated,
522 atlas_pack.exceeded_soft_budget,
523 );
524 let fused2 = assemble_body(&atlas, &skeleton, &render.text, &important, &coverage, &probe_om2);
525 if BLOCK_HEADER_OVERHEAD + estimate_tokens(&fused2) > startup_hard_max {
526 important = "## SYSTEM-CRITICAL SYMBOLS\n(omitted over hard max)\n".to_string();
527 }
528 }
529 } else {
530 important = "## SYSTEM-CRITICAL SYMBOLS\n(omitted over hard max)\n".to_string();
532 }
533 }
534
535 let omissions = emergency_omissions.unwrap_or_else(|| {
539 omission_lines(
540 &render.omissions,
541 render.omitted_ids.len(),
542 &atlas_pack.dropped_sections,
543 atlas_pack.hard_truncated,
544 atlas_pack.exceeded_soft_budget,
545 )
546 });
547
548 let trust_policy = trust_policy_str(compiler.view.policy());
549
550 let mut h = blake3::Hasher::new();
554 h.update(b"startup-artifact-v1");
555 h.update(epoch.as_bytes());
556 h.update(renderer_version.as_bytes());
557 h.update(trust_policy.as_bytes());
558 h.update(budget.total.to_string().as_bytes());
559 h.update(budget.atlas.to_string().as_bytes());
560 h.update(budget.surface.to_string().as_bytes());
561 h.update(budget.task_delta.to_string().as_bytes());
562 h.update(budget.structural_source.to_string().as_bytes());
563 let sha256 = h.finalize().to_hex().to_string();
564
565 let body = assemble_body(&atlas, &skeleton, &surface, &important, &coverage, &omissions);
570 let mut ch = blake3::Hasher::new();
571 ch.update(b"startup-content-v1");
572 ch.update(epoch.as_bytes());
573 ch.update(renderer_version.as_bytes());
574 ch.update(trust_policy.as_bytes());
575 ch.update(budget.total.to_string().as_bytes());
576 ch.update(budget.atlas.to_string().as_bytes());
577 ch.update(budget.surface.to_string().as_bytes());
578 ch.update(budget.task_delta.to_string().as_bytes());
579 ch.update(budget.structural_source.to_string().as_bytes());
580 ch.update(body.as_bytes());
581 let content_hash = ch.finalize().to_hex().to_string();
582
583 let mut artifact = ContextArtifact {
584 kind: "startup".into(),
585 epoch,
586 renderer_version: renderer_version.to_string(),
587 trust_policy,
588 budget: budget.clone(),
589 sha256,
590 content_hash,
591 text: String::new(),
592 };
593 artifact.text = assemble_block(&atlas, &skeleton, &surface, &important, &coverage, &omissions, &artifact);
594
595 StartupContext {
596 atlas,
597 atlas_budget_used,
598 skeleton,
599 surface,
600 important,
601 surface_render: render,
602 coverage,
603 omissions,
604 artifact,
605 }
606}
607fn global_rank_key(epoch: &str, policy: &str, salt: &str) -> String {
614 let mut h = blake3::Hasher::new();
615 h.update(b"rank:global:v1");
616 h.update(epoch.as_bytes());
617 h.update(b"\0");
618 h.update(policy.as_bytes());
619 h.update(b"\0");
620 h.update(salt.as_bytes());
621 format!("rank:global:{}", &h.finalize().to_hex()[..20])
622}
623
624pub fn load_global_rank_cache(compiler: &ContextCompiler) -> Option<GlobalRankCache> {
631 let epoch = compiler.store.cache_epoch().ok()?;
632 let policy = trust_policy_str(compiler.view.policy());
633 let salt = &compiler.settings.rank_salt;
634 let key = global_rank_key(&epoch, &policy, salt);
635 let cached = compiler.store.cache_get(&key, &epoch).ok().flatten()?;
636 let c: GlobalRankCache = serde_json::from_str(&cached).ok()?;
637 if c.epoch != epoch || c.policy != policy || c.salt != *salt {
638 return None;
639 }
640 Some(c)
641}
642
643pub fn store_global_rank_cache(compiler: &ContextCompiler, cache: &GlobalRankCache) {
647 let epoch = compiler
648 .store
649 .cache_epoch()
650 .unwrap_or_else(|_| "no-epoch".into());
651 let policy = trust_policy_str(compiler.view.policy());
652 let salt = &compiler.settings.rank_salt;
653 let key = global_rank_key(&epoch, &policy, salt);
654 if let Ok(json) = serde_json::to_string(cache) {
655 let _ = compiler.store.cache_put(&key, &json, &epoch);
656 }
657}
658
659fn symbol_id_of(entry_id: &str) -> String {
665 entry_id
666 .rsplit_once("#overload")
667 .map(|(logical, _)| logical)
668 .unwrap_or(entry_id)
669 .to_string()
670}
671
672pub fn render_startup(s: &StartupContext) -> String {
676 assemble_block(&s.atlas, &s.skeleton, &s.surface, &s.important, &s.coverage, &s.omissions, &s.artifact)
677}
678
679fn assemble_body(atlas: &str, skeleton: &str, surface: &str, important: &str, coverage: &[String], omissions: &[String]) -> String {
683 let mut out = String::new();
684 out.push_str("# SCC SYSTEM CONTEXT\n");
685 out.push_str("## SYSTEM ATLAS\n");
686 out.push_str(atlas.trim_end());
687 out.push_str("\n\n## REPOSITORY SKELETON\n");
688 out.push_str(skeleton.trim_end());
689 out.push_str("\n\n## SYSTEM SURFACE MAP\n");
690 out.push_str(surface.trim_end());
691 out.push_str("\n\n");
692 out.push_str(important.trim_end());
693 out.push_str("\n\n## MODEL COVERAGE\n");
694 if coverage.is_empty() {
695 out.push_str("(no warnings)\n");
696 } else {
697 for c in coverage {
698 out.push_str(c);
699 out.push('\n');
700 }
701 }
702 out.push_str("\n## OMISSIONS\n");
703 for o in omissions {
704 out.push_str(o);
705 out.push('\n');
706 }
707 out
708}
709
710fn assemble_block(
712 atlas: &str,
713 skeleton: &str,
714 surface: &str,
715 important: &str,
716 coverage: &[String],
717 omissions: &[String],
718 artifact: &ContextArtifact,
719) -> String {
720 let mut out = String::new();
721 out.push_str("# SCC SYSTEM CONTEXT\n");
722 out.push_str(&format!(
723 "<!-- artifact sha256:{} content_hash:{} epoch:{} renderer:{} -->\n\n",
724 artifact.sha256, artifact.content_hash, artifact.epoch, artifact.renderer_version
725 ));
726 out.push_str(&assemble_body(atlas, skeleton, surface, important, coverage, omissions));
727 out
728}
729
730fn compiler_warnings(compiler: &ContextCompiler) -> Vec<String> {
734 let mut w = Vec::new();
735 if compiler.store.snapshot_status().ok().flatten().is_none() {
736 w.push("Repository is not indexed — run `scc index`.".into());
737 }
738 if !compiler.stale_paths.is_empty() {
739 w.push(format!(
740 "Model is stale: {} changed file(s) not yet re-indexed.",
741 compiler.stale_paths.len()
742 ));
743 }
744 if let Ok(findings) = compiler.store.drift_findings(true) {
745 for (_, kind, sev, msg, _) in findings {
746 if sev == "high" || sev == "critical" {
747 w.push(format!("Drift [{kind}]: {msg}"));
748 }
749 }
750 }
751 w.truncate(6);
752 w
753}
754
755pub(crate) fn trust_policy_str(p: &scc_graph::TrustPolicy) -> String {
757 format!(
758 "extracted={} resolved={} observed={} declared={} inferred={} floor={}",
759 p.allow_extracted,
760 p.allow_resolved,
761 p.allow_observed,
762 p.allow_declared,
763 p.allow_inferred,
764 p.min_inferred_confidence
765 )
766}
767
768pub fn visible_ids_from_startup(
777 compiler: &ContextCompiler,
778 startup: &StartupContext,
779) -> (BTreeSet<String>, BTreeSet<String>, BTreeSet<String>, BTreeSet<String>) {
780 let mut symbols = BTreeSet::new();
781 let mut files = BTreeSet::new();
782 let mut components = BTreeSet::new();
783 let mut flows = BTreeSet::new();
784
785 let atlas_pack = compiler.system_atlas(Some(startup.atlas_budget_used));
789
790 let view = &compiler.view;
796 let rendered_symbols: BTreeSet<String> = startup
797 .surface_render
798 .rendered_ids
799 .iter()
800 .map(|id| symbol_id_of(id))
801 .collect();
802 let mut comp_of: BTreeMap<String, String> = BTreeMap::new();
806 for c in view.components() {
807 for r in sorted_rels(view.out_pred(&c.id, scc_core::predicates::CONTAINS)) {
808 for sr in sorted_rels(view.out_pred(&r.object, scc_core::predicates::CONTAINS)) {
809 if rendered_symbols.contains(&sr.object) {
810 comp_of
811 .entry(sr.object.clone())
812 .or_insert_with(|| c.name.clone());
813 }
814 }
815 }
816 }
817 for id in &startup.surface_render.rendered_ids {
818 let symbol_id = symbol_id_of(id);
819 symbols.insert(symbol_id.clone());
820 let ent = view.entity(id).or_else(|| view.entity(&symbol_id));
823 if let Some(e) = ent {
824 if let Some(f) = e.attributes.get("file").and_then(|v| v.as_str()) {
825 files.insert(f.to_string());
826 }
827 }
828 if let Some(c) = comp_of.get(&symbol_id) {
829 components.insert(c.clone());
830 }
831 }
832 for id in &atlas_pack.entity_ids {
834 if let Some(e) = compiler.view.entity(id) {
835 match e.kind.as_str() {
836 kinds::SYMBOL => {
837 symbols.insert(id.clone());
838 }
839 kinds::FILE => {
840 files.insert(e.name.clone());
841 }
842 kinds::COMPONENT => {
843 components.insert(id.clone());
844 }
845 kinds::FLOW => {
846 flows.insert(id.clone());
847 }
848 _ => {}
849 }
850 }
851 }
852 (symbols, files, components, flows)
853}
854
855fn sorted_rels(rels: Vec<&scc_core::Relationship>) -> Vec<&scc_core::Relationship> {
859 let mut v = rels;
860 v.sort_by(|a, b| {
861 a.id.cmp(&b.id)
862 .then_with(|| a.subject.cmp(&b.subject))
863 .then_with(|| a.object.cmp(&b.object))
864 });
865 v
866}
867
868pub fn task_delta_with_ids(
878 compiler: &ContextCompiler,
879 goal: &str,
880 visible: &ContextLedger,
881 budget_tokens: usize,
882 semantic: Option<&dyn crate::rank::SemanticScorer>,
883) -> (String, Vec<String>) {
884 let render = build_surface(
885 compiler,
886 SurfaceRequest {
887 mode: SurfaceMode::Task {
888 goal,
889 visible: Some(visible),
890 },
891 budget: budget_tokens,
892 explain: false,
893 policy: SurfacePolicy::defaults(budget_tokens),
894 semantic,
895 },
896 );
897 let critical = crate::surface::important_symbols(
900 compiler,
901 SurfaceMode::Task { goal, visible: Some(visible) },
902 8,
903 );
904 let mut out = String::new();
905 out.push_str("# SCC TASK DELTA\n");
906 out.push_str(&format!("TASK-FOCUS: {goal}\n"));
907 out.push_str(crate::surface::render_important(&critical, true).as_str());
908 out.push_str("Relevant APIs not already visible:\n");
909 let body = render
910 .text
911 .strip_prefix("SCC SYSTEM SURFACE MAP\n\n")
912 .unwrap_or(&render.text);
913 out.push_str(body.trim_end());
914 out.push('\n');
915 (out, render.rendered_ids)
916}
917
918pub fn task_surface(
923 compiler: &ContextCompiler,
924 goal: &str,
925 budget_tokens: usize,
926 explain: bool,
927 semantic: Option<&dyn crate::rank::SemanticScorer>,
928) -> String {
929 task_surface_with_ids(compiler, goal, budget_tokens, explain, semantic).0
930}
931
932pub fn task_surface_with_ids(
937 compiler: &ContextCompiler,
938 goal: &str,
939 budget_tokens: usize,
940 explain: bool,
941 semantic: Option<&dyn crate::rank::SemanticScorer>,
942) -> (String, Vec<String>) {
943 let render = build_surface(
944 compiler,
945 SurfaceRequest {
946 mode: SurfaceMode::Task {
947 goal,
948 visible: None,
949 },
950 budget: budget_tokens,
951 explain,
952 policy: SurfacePolicy::defaults(budget_tokens),
953 semantic,
954 },
955 );
956 let mut out = String::new();
957 out.push_str(&format!("# SYSTEM SURFACE MAP (task-personalized: {goal})\n"));
958 let body = render
959 .text
960 .strip_prefix("SCC SYSTEM SURFACE MAP\n\n")
961 .unwrap_or(&render.text);
962 out.push_str(body.trim_end());
963 out.push('\n');
964 (out, render.rendered_ids)
965}
966
967#[cfg(test)]
968mod tests {
969 use super::*;
970
971fn fixture_compiler() -> (tempfile::TempDir, crate::ContextCompiler<'static>) {
973 let dir = tempfile::TempDir::new().unwrap();
974 let root = dir.path().join("repo");
975 std::fs::create_dir_all(&root).unwrap();
976 let store = Box::leak(Box::new(
977 scc_store::Store::open(&dir.path().join("scc.db"), &root).unwrap(),
978 ));
979 let graph = Box::leak(Box::new(scc_graph::RealityGraph::load(store).unwrap()));
980 let settings = crate::ContextSettings::default();
981 let comp = crate::ContextCompiler::new(store, graph, settings, Vec::new());
982 (dir, comp)
983 }
984
985 #[test]
986fn render_startup_emits_spec_headers() {
988 let sc = StartupContext {
989 atlas: "ATLAS-BODY".into(),
990 atlas_budget_used: 6000,
991 skeleton: "SKELETON-BODY".into(),
992 surface: "SURFACE-BODY".into(),
993 important: "IMPORTANT-BODY".into(),
994 surface_render: scc_core::SurfaceRenderResult {
995 text: "SURFACE-BODY".into(),
996 rendered_ids: vec![],
997 rendered_entries: vec![],
998 omitted_ids: vec![],
999 omissions: vec![],
1000 token_count: 0,
1001 critical_drops: vec![],
1002 },
1003 coverage: vec!["stale: a.py".into()],
1004 omissions: vec!["none".into()],
1005 artifact: ContextArtifact {
1006 kind: "startup".into(),
1007 epoch: "epoch:test".into(),
1008 renderer_version: "test".into(),
1009 trust_policy: "floor=0.85".into(),
1010 budget: ContextBudget::default(),
1011 sha256: "abc".into(),
1012 content_hash: "def".into(),
1013 text: String::new(),
1014 },
1015 };
1016 let out = render_startup(&sc);
1017 assert!(out.contains("# SCC SYSTEM CONTEXT"));
1018 assert!(out.contains("## SYSTEM ATLAS"));
1019 assert!(out.contains("## REPOSITORY SKELETON"));
1020 assert!(out.contains("## SYSTEM SURFACE MAP"));
1021 assert!(out.contains("ATLAS-BODY"));
1022 assert!(out.contains("SURFACE-BODY"));
1023 assert!(out.contains("SKELETON-BODY"));
1024 assert!(out.contains("## SYSTEM-CRITICAL SYMBOLS") || out.contains("IMPORTANT-BODY"));
1025 let (ia, ik, is) = (
1026 out.find("## SYSTEM ATLAS").unwrap(),
1027 out.find("## REPOSITORY SKELETON").unwrap(),
1028 out.find("## SYSTEM SURFACE MAP").unwrap(),
1029 );
1030 assert!(ia < ik && ik < is, "skeleton grounds before architecture");
1031 assert!(out.contains("stale: a.py"));
1032 }
1033
1034 #[test]
1035fn artifact_text_equals_rendered_block() {
1037 let (_dir, comp) = fixture_compiler();
1038 let budget = ContextBudget::default();
1039 let sc = build_startup(&comp, &budget, "test-renderer");
1040 assert_eq!(sc.artifact.text, render_startup(&sc));
1041 assert_eq!(sc.artifact.sha256.len(), 64);
1042 assert_eq!(sc.artifact.content_hash.len(), 64);
1043 assert_ne!(sc.artifact.content_hash, sc.artifact.sha256);
1044 assert!(render_startup(&sc).contains("content_hash:"));
1045 assert_eq!(sc.artifact.epoch, comp.store.cache_epoch().unwrap_or_default());
1046 }
1047
1048 #[test]
1049fn global_rank_cache_roundtrips_through_the_store() {
1051 let (_dir, comp) = fixture_compiler();
1052 assert!(load_global_rank_cache(&comp).is_none());
1054 let cache = GlobalRankCache {
1055 epoch: comp.store.cache_epoch().unwrap_or_else(|_| "no-epoch".into()),
1056 policy: trust_policy_str(comp.view.policy()),
1057 salt: comp.settings.rank_salt.clone(),
1058 global_vector: vec![0.1, 0.2, 0.3],
1059 node_symbol_map: BTreeMap::from([("repo://r/symbol/a.py/serve".into(), 0.42)]),
1060 candidates_epoch: comp.store.cache_epoch().unwrap_or_default(),
1061 candidate_ids: vec!["repo://r/symbol/a.py/serve".into()],
1062 hits: 1,
1063 };
1064 store_global_rank_cache(&comp, &cache);
1065 let loaded = load_global_rank_cache(&comp).expect("cache entry present after store");
1066 assert_eq!(loaded, cache);
1067 assert_eq!(loaded.hits, 1);
1068 assert_eq!(
1069 loaded.node_symbol_map.get("repo://r/symbol/a.py/serve"),
1070 Some(&0.42)
1071 );
1072 assert_eq!(loaded.candidates_epoch, loaded.epoch);
1073 }
1074
1075 #[test]
1076fn visible_ids_consume_the_same_render_the_artifact_printed() {
1078 let (_dir, comp) = fixture_compiler();
1085 let budget = ContextBudget::default();
1086 let sc = build_startup(&comp, &budget, "test-renderer");
1087 assert!(sc.artifact.text.contains(&sc.surface));
1088 let (syms, _files, _comps, _flows) = visible_ids_from_startup(&comp, &sc);
1089 for id in &sc.surface_render.rendered_ids {
1090 let symbol_id = symbol_id_of(id);
1091 assert!(
1092 syms.contains(&symbol_id),
1093 "rendered entry {id} must be marked visible (symbol {symbol_id})"
1094 );
1095 }
1096 }
1097}