use crate::core::config::CompressionLevel;
pub const START_MARK: &str = "<!-- lean-ctx-rules -->";
pub const RULES_MARKER_PREFIX: &str = "<!-- lean-ctx-rules";
pub const AGENTS_BLOCK_START: &str = "<!-- lean-ctx -->";
pub const AGENTS_BLOCK_END: &str = "<!-- /lean-ctx -->";
pub const END_MARK: &str = "<!-- /lean-ctx-rules -->";
pub const COMPRESSION_BLOCK_START: &str = "<!-- lean-ctx-compression -->";
pub const COMPRESSION_BLOCK_END: &str = "<!-- /lean-ctx-compression -->";
pub const RULES_VERSION: usize = 1;
pub const CRITICAL: &str = "CRITICAL: ALWAYS use lean-ctx tools instead of native equivalents. \
This is NOT optional.";
pub const BULLETS: &str = "\
MANDATORY MAPPING:\n\
• Read/cat -> ctx_read(path, mode)\n\
• Glob/find -> ctx_glob(pattern)\n\
• Shell/bash -> ctx_shell(command)\n\
• Grep -> ctx_search(pattern, path)\n\
• ls/find -> ctx_tree(path, depth)";
pub const NEVER: &str =
"NEVER use native Read/Grep/Shell/Glob when ctx_* equivalents are available.";
pub const INTENT: &str = "\
Tool selection by intent:\n\
• Understand code / find answers / before editing -> ctx_compose (call FIRST)\n\
• Read a file -> ctx_read(path, mode=signatures|map|full)\n\
• Find a symbol by name (exact) -> ctx_symbol\n\
• Search code by pattern (fuzzy) -> ctx_search\n\
• Search by meaning (concepts) -> ctx_semantic_search\n\
• Find files by pattern (glob) -> ctx_glob\n\
• Project structure -> ctx_tree\n\
• Who calls this / call graph -> ctx_callgraph\n\
• Session state / memory -> ctx_session / ctx_knowledge";
pub const ANTI: &str = "\
Anti-patterns — do NOT:\n\
• Chain ctx_search -> ctx_read -> ctx_symbol — one ctx_compose replaces all three\n\
• Grep for symbol definitions — ctx_symbol is faster + more precise\n\
• Use ctx_read(mode=full) for orientation — use mode=signatures\n\
• Use ctx_callgraph or ctx_graph for const/static/variable references — they track\n\
function call edges and file-level deps only. Use grep or ctx_compose instead";
pub const PARALLEL: &str = "\
PARALLEL tool calls: fire independent calls in the SAME turn — don't sequence them.\n\
ctx_compose bundles multiple lookups into one call; for anything it doesn't\n\
cover, batch independent reads/searches together.";
pub const AUTO: &str = "Auto: preload/dedup/compress run in background. \
ctx_session=memory, ctx_knowledge=facts, ctx_semantic_search=meaning search, \
ctx_shell raw=true=uncompressed. Details: LEAN-CTX.md";
pub const CEP: &str = "CEP v1: 1.ACT FIRST 2.DELTA ONLY (Fn refs) 3.STRUCTURED (+/-/~) \
4.ONE LINE PER ACTION 5.QUALITY ANCHOR";
pub const INTELLIGENCE: &str =
"OUTPUT: never echo tool output, no narration comments, show only changed code.";
pub const LITM_END: &str = "TOOL PREFERENCE (END): ctx_compose>chain ctx_read>Read ctx_shell>Shell \
ctx_search>Grep ctx_glob>Glob ctx_tree>ls | Edit/Write/Delete=native";
pub const SHADOW_MINIMAL: &str = "\
lean-ctx shadow mode: native file/search/shell calls auto-route to ctx_* — no tool-mapping needed.\n\
Exclusive tools (no native trigger): ctx_compose (understand code, call first), ctx_symbol (exact symbol), ctx_callgraph (callers), ctx_semantic_search (by meaning), ctx_knowledge / ctx_session (memory).";
pub const LITE_PROMPT: &str = "\
OUTPUT STYLE: concise
- Bullet points over paragraphs
- Skip filler words and hedging (\"I think\", \"probably\", \"it seems\")
- 1-sentence explanations max, then code/action
- No repeating what the user said";
pub const STANDARD_PROMPT: &str = "\
OUTPUT STYLE: dense
- Each statement = one atomic fact line
- Use abbreviations: fn, cfg, impl, deps, req, res, ctx, err, ret
- Diff lines only (+/-/~), never repeat unchanged code
- Symbols: → (causes), + (adds), − (removes), ~ (modifies), ∴ (therefore)
- No narration, no filler, no hedging
- BUDGET: ≤200 tokens per response unless code block required";
pub const MAX_PROMPT: &str = "\
OUTPUT STYLE: expert-terse
- Telegraph format: subject-verb-object, drop articles/prepositions
- Symbolic vocabulary: → cause, ∵ because, ∴ therefore, ⊕ add, ⊖ remove, Δ change, ≈ similar, ≠different, ∈ in/member, ∅ empty/none, ✓ ok, ✗ fail
- Code blocks: untouched (never compress code syntax)
- Each line: max 80 chars
- Zero narration, zero filler
- BUDGET: ≤100 tokens per non-code response";
pub fn compression_text(level: CompressionLevel) -> &'static str {
match level {
CompressionLevel::Off => "",
CompressionLevel::Lite => LITE_PROMPT,
CompressionLevel::Standard => STANDARD_PROMPT,
CompressionLevel::Max => MAX_PROMPT,
}
}
const FULL_NON_SHADOW: &[&str] = &[
CRITICAL,
BULLETS,
NEVER,
INTENT,
ANTI,
PARALLEL,
AUTO,
CEP,
INTELLIGENCE,
LITM_END,
];
const FULL_SHADOW: &[&str] = &[SHADOW_MINIMAL, INTELLIGENCE];
const COMPACT_NON_SHADOW: &[&str] = &[CRITICAL, BULLETS, NEVER, INTENT, ANTI, PARALLEL];
const COMPACT_SHADOW: &[&str] = &[SHADOW_MINIMAL];
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum Wrapper {
Dedicated,
Shared,
Bare,
}
pub fn render(shadow: bool, wrapper: Wrapper, level: CompressionLevel) -> String {
let profile = match (wrapper, shadow) {
(Wrapper::Dedicated, false) => FULL_NON_SHADOW,
(Wrapper::Dedicated, true) => FULL_SHADOW,
(_, false) => COMPACT_NON_SHADOW,
(_, true) => COMPACT_SHADOW,
};
let mut body = profile.join("\n\n");
let compression = compression_text(level);
if !compression.is_empty() {
body.push('\n');
if matches!(wrapper, Wrapper::Bare) {
body.push_str(compression);
} else {
body.push_str(COMPRESSION_BLOCK_START);
body.push('\n');
body.push_str(compression);
body.push('\n');
body.push_str(COMPRESSION_BLOCK_END);
}
}
if matches!(wrapper, Wrapper::Bare) {
return body;
}
let version_line = format!("<!-- version: {RULES_VERSION} -->");
format!("{START_MARK}\n{version_line}\n\n{body}\n{END_MARK}")
}
#[derive(Debug)]
pub struct RulesFile<'a> {
content: &'a str,
start: Option<usize>,
end: Option<usize>,
version: usize,
}
fn parse_version_number(s: &str) -> Option<usize> {
let prefix = "<!-- version: ";
let vs = s.find(prefix)?;
let num_start = vs + prefix.len();
let end = s[num_start..].find(" -->")?;
s[num_start..num_start + end].parse().ok()
}
impl<'a> RulesFile<'a> {
pub fn parse(content: &'a str) -> Self {
let start = content.find(START_MARK);
let version = start
.and_then(|s| parse_version_number(&content[s + START_MARK.len()..]))
.unwrap_or(0);
let end = content.find(END_MARK);
RulesFile {
content,
start,
end,
version,
}
}
pub fn has_content(&self) -> bool {
self.start.is_some()
}
pub fn version(&self) -> usize {
self.version
}
pub fn is_current(&self) -> bool {
self.version >= RULES_VERSION
}
pub fn prefix(&self) -> &'a str {
self.start.map_or("", |s| self.content[..s].trim())
}
pub fn suffix(&self) -> &'a str {
self.end
.map_or("", |e| self.content[e + END_MARK.len()..].trim())
}
fn block(&self) -> Option<&'a str> {
match (self.start, self.end) {
(Some(s), Some(e)) if e >= s => Some(&self.content[s..e + END_MARK.len()]),
_ => None,
}
}
pub fn block_matches_render(
&self,
shadow: bool,
wrapper: Wrapper,
level: CompressionLevel,
) -> bool {
match self.block() {
Some(block) => block.trim() == render(shadow, wrapper, level).trim(),
None => false,
}
}
pub fn merged(&self, shadow: bool, wrapper: Wrapper, level: CompressionLevel) -> String {
let fresh = render(shadow, wrapper, level);
if self.start.is_some() {
let before = self.prefix();
let after = self.suffix();
let mut out = String::new();
if !before.is_empty() {
out.push_str(before);
out.push('\n');
out.push('\n');
}
out.push_str(&fresh);
if !after.is_empty() {
out.push('\n');
out.push('\n');
out.push_str(after);
}
if !out.ends_with('\n') {
out.push('\n');
}
out
} else {
let trimmed = self.content.trim_end();
let mut out = trimmed.to_string();
if !out.is_empty() {
out.push('\n');
out.push('\n');
}
out.push_str(&fresh);
out
}
}
pub fn initial(shadow: bool, wrapper: Wrapper, level: CompressionLevel) -> String {
render(shadow, wrapper, level)
}
pub fn without_section(&self) -> String {
if let Some(start_pos) = self.start {
let before = self.content[..start_pos].trim();
let after = self.suffix();
let mut out = String::new();
if !before.is_empty() {
out.push_str(before);
out.push('\n');
}
if !after.is_empty() {
out.push('\n');
out.push_str(after);
}
out
} else {
self.content.to_string()
}
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn bullets_uses_ctx_shell() {
assert!(BULLETS.contains("ctx_shell"));
assert!(!BULLETS.contains("lean-ctx -c"));
assert!(!BULLETS.contains("ctx_edit"));
}
#[test]
fn sections_not_empty() {
assert!(!BULLETS.is_empty());
assert!(!NEVER.is_empty());
assert!(!INTENT.is_empty());
assert!(!ANTI.is_empty());
assert!(!PARALLEL.is_empty());
assert!(!AUTO.is_empty());
assert!(!CEP.is_empty());
assert!(!INTELLIGENCE.is_empty());
assert!(!LITM_END.is_empty());
assert!(!CRITICAL.is_empty());
}
#[test]
fn intent_contains_ctx_compose() {
assert!(INTENT.contains("ctx_compose"));
}
#[test]
fn anti_contains_do_not() {
assert!(ANTI.contains("do NOT"));
}
#[test]
fn parallel_contains_parallel() {
assert!(PARALLEL.contains("PARALLEL"));
}
#[test]
fn dedicated_has_markers_and_version() {
let out = render(false, Wrapper::Dedicated, CompressionLevel::Off);
assert!(out.contains(START_MARK));
assert!(out.contains(&format!("<!-- version: {RULES_VERSION} -->")));
assert!(out.contains(END_MARK));
assert!(out.contains(BULLETS));
assert!(out.contains(NEVER));
assert!(out.contains("CRITICAL"));
}
#[test]
fn dedicated_shadow_is_minimal() {
let out = render(true, Wrapper::Dedicated, CompressionLevel::Off);
assert!(out.contains(START_MARK));
assert!(!out.contains("MANDATORY MAPPING"), "no BULLETS in shadow");
assert!(!out.contains(NEVER), "no NEVER in shadow");
assert!(!out.contains("CRITICAL"), "no CRITICAL banner in shadow");
assert!(
!out.contains("Tool selection by intent"),
"routing INTENT block is redundant under interception"
);
assert!(
!out.contains("Anti-patterns") && !out.contains("PARALLEL tool calls"),
"ANTI/PARALLEL routing guidance is dropped in shadow"
);
assert!(
out.contains("shadow mode") && out.contains("ctx_compose"),
"shadow keeps the exclusive-tool advert"
);
assert!(out.contains(INTELLIGENCE), "shadow keeps the output style");
}
#[test]
fn shadow_is_smaller_than_non_shadow() {
let shadow = render(true, Wrapper::Dedicated, CompressionLevel::Off);
let full = render(false, Wrapper::Dedicated, CompressionLevel::Off);
assert!(
shadow.len() < full.len(),
"shadow ({}) must be smaller than non-shadow ({})",
shadow.len(),
full.len()
);
}
#[test]
fn dedicated_litm_structure() {
let out = render(false, Wrapper::Dedicated, CompressionLevel::Off);
let lines: Vec<&str> = out.lines().collect();
let first_5 = lines[..5.min(lines.len())].join("\n");
assert!(
first_5.contains("CRITICAL") || first_5.contains("MUST"),
"LITM: MUST/CRITICAL instruction near start"
);
let tail = lines[lines.len().saturating_sub(8)..].join("\n");
assert!(
tail.contains("PREFERENCE") || tail.contains("NEVER"),
"LITM: reinforcement near end, tail={tail:?}"
);
}
#[test]
fn shared_has_markers_and_header() {
let out = render(false, Wrapper::Shared, CompressionLevel::Off);
assert!(out.contains(START_MARK));
assert!(out.contains(END_MARK));
assert!(out.contains("MANDATORY MAPPING"));
assert!(out.contains(BULLETS));
}
#[test]
fn shared_shadow_omits_mapping() {
let out = render(true, Wrapper::Shared, CompressionLevel::Off);
assert!(out.contains(START_MARK));
assert!(
!out.contains("MANDATORY MAPPING"),
"shadow must not have header"
);
assert!(
!out.contains("MANDATORY MAPPING"),
"shadow must not contain BULLETS"
);
}
#[test]
fn bare_has_no_markers() {
let out = render(false, Wrapper::Bare, CompressionLevel::Off);
assert!(!out.contains(START_MARK), "Bare must not have START_MARK");
assert!(!out.contains(END_MARK), "Bare must not have END_MARK");
assert!(!out.contains("<!-- version:"), "Bare must not have version");
assert!(out.contains(BULLETS));
assert!(out.contains(NEVER));
}
#[test]
fn bare_shadow_only_read_modes() {
let out = render(true, Wrapper::Bare, CompressionLevel::Off);
assert!(!out.contains(NEVER), "shadow Bare must not have NEVER");
assert!(
!out.contains("MANDATORY MAPPING"),
"shadow Bare must not have BULLETS"
);
}
#[test]
fn render_includes_lite_prompt() {
let out = render(false, Wrapper::Bare, CompressionLevel::Lite);
assert!(out.contains("OUTPUT STYLE: concise"));
assert!(out.contains("Bullet points"));
}
#[test]
fn render_includes_standard_prompt() {
let out = render(false, Wrapper::Bare, CompressionLevel::Standard);
assert!(out.contains("OUTPUT STYLE: dense"));
assert!(out.contains("atomic fact"));
}
#[test]
fn render_includes_max_prompt() {
let out = render(false, Wrapper::Bare, CompressionLevel::Max);
assert!(out.contains("OUTPUT STYLE: expert-terse"));
assert!(out.contains("Telegraph"));
}
#[test]
fn render_off_excludes_compression() {
let out = render(false, Wrapper::Bare, CompressionLevel::Off);
assert!(!out.contains("OUTPUT STYLE:"));
}
#[test]
fn compression_text_matches_level() {
assert!(compression_text(CompressionLevel::Off).is_empty());
assert!(compression_text(CompressionLevel::Lite).contains("Bullet"));
assert!(compression_text(CompressionLevel::Standard).contains("fn, cfg"));
assert!(compression_text(CompressionLevel::Max).contains("Telegraph"));
}
#[test]
fn carrier_wrappers_wrap_compression_in_markers() {
for wrapper in [Wrapper::Dedicated, Wrapper::Shared] {
let out = render(false, wrapper, CompressionLevel::Standard);
assert!(
out.contains(COMPRESSION_BLOCK_START) && out.contains(COMPRESSION_BLOCK_END),
"{wrapper:?} must wrap compression in COMPRESSION_BLOCK markers"
);
let start = out.find(COMPRESSION_BLOCK_START).unwrap();
let end = out.find(COMPRESSION_BLOCK_END).unwrap();
assert!(start < end, "{wrapper:?}: start marker precedes end marker");
assert!(out[start..end].contains("OUTPUT STYLE: dense"));
}
}
#[test]
fn bare_wrapper_emits_compression_without_markers() {
let out = render(false, Wrapper::Bare, CompressionLevel::Standard);
assert!(out.contains("OUTPUT STYLE: dense"));
assert!(!out.contains(COMPRESSION_BLOCK_START));
assert!(!out.contains(COMPRESSION_BLOCK_END));
}
#[test]
fn compression_off_emits_no_markers_in_any_wrapper() {
for wrapper in [Wrapper::Dedicated, Wrapper::Shared, Wrapper::Bare] {
let out = render(false, wrapper, CompressionLevel::Off);
assert!(
!out.contains(COMPRESSION_BLOCK_START) && !out.contains(COMPRESSION_BLOCK_END),
"{wrapper:?}: Off must emit no compression markers"
);
}
}
#[test]
fn rendered_carrier_block_is_seen_as_carrying_compression() {
let dedicated = render(false, Wrapper::Dedicated, CompressionLevel::Lite);
assert!(crate::core::rules_channel::carries_full_rules(&dedicated));
assert!(dedicated.contains(COMPRESSION_BLOCK_START));
}
#[test]
fn all_wrappers_produce_output() {
for shadow in [false, true] {
for wrapper in [Wrapper::Dedicated, Wrapper::Shared, Wrapper::Bare] {
let out = render(shadow, wrapper, CompressionLevel::Off);
assert!(!out.is_empty(), "{wrapper:?} shadow={shadow} is empty");
}
}
}
#[test]
fn rules_file_parses_version() {
let content = format!(
"stuff before\n{START_MARK}\n<!-- version: {RULES_VERSION} -->\n\nbody\n{END_MARK}\nstuff after"
);
let f = RulesFile::parse(&content);
assert!(f.has_content());
assert_eq!(f.version(), RULES_VERSION);
assert!(f.is_current());
assert!(f.prefix().contains("stuff before"));
assert!(f.suffix().contains("stuff after"));
}
#[test]
fn rules_file_no_version_defaults_to_zero() {
let content = format!("{START_MARK}\nbody\n{END_MARK}");
let f = RulesFile::parse(&content);
assert!(f.has_content());
assert_eq!(f.version(), 0);
assert!(!f.is_current());
}
#[test]
fn rules_file_no_start_marker_no_content() {
let f = RulesFile::parse("just user stuff");
assert!(!f.has_content());
assert_eq!(f.version(), 0);
}
#[test]
fn block_matches_render_true_for_fresh_render() {
let fresh = render(false, Wrapper::Dedicated, CompressionLevel::Off);
let content = format!("user before\n{fresh}\nuser after");
let f = RulesFile::parse(&content);
assert!(f.is_current(), "fresh render carries the current version");
assert!(
f.block_matches_render(false, Wrapper::Dedicated, CompressionLevel::Off),
"an unchanged block must compare equal to a fresh render"
);
}
#[test]
fn block_matches_render_false_on_compression_change() {
let content = render(false, Wrapper::Dedicated, CompressionLevel::Off);
let f = RulesFile::parse(&content);
assert!(f.is_current());
assert!(
!f.block_matches_render(false, Wrapper::Dedicated, CompressionLevel::Max),
"a compression-level change must be detected as drift"
);
}
#[test]
fn block_matches_render_false_on_shadow_change() {
let content = render(false, Wrapper::Dedicated, CompressionLevel::Lite);
let f = RulesFile::parse(&content);
assert!(
!f.block_matches_render(true, Wrapper::Dedicated, CompressionLevel::Lite),
"a shadow-mode toggle must be detected as drift"
);
}
#[test]
fn block_matches_render_false_without_block() {
let f = RulesFile::parse("plain user content, no markers");
assert!(!f.block_matches_render(false, Wrapper::Dedicated, CompressionLevel::Off));
}
#[test]
fn rules_file_merged_replaces_section() {
let content =
format!("before\n{START_MARK}\n<!-- version: 1 -->\n\nold\n{END_MARK}\nafter");
let f = RulesFile::parse(&content);
let merged = f.merged(false, Wrapper::Shared, CompressionLevel::Off);
assert!(merged.contains("before"), "prefix preserved");
assert!(merged.contains("after"), "suffix preserved");
assert!(!merged.contains("old"), "old content replaced");
assert!(merged.contains(&format!("<!-- version: {RULES_VERSION} -->")));
}
#[test]
fn rules_file_merged_appends_when_no_section() {
let content = "user content";
let f = RulesFile::parse(content);
assert!(!f.has_content());
let merged = f.merged(false, Wrapper::Bare, CompressionLevel::Off);
assert!(merged.contains("user content"));
assert!(merged.contains(BULLETS));
}
#[test]
fn rules_file_without_section_strips_content() {
let content =
format!("header\n{START_MARK}\n<!-- version: 1 -->\n\nbody\n{END_MARK}\nfooter");
let f = RulesFile::parse(&content);
let stripped = f.without_section();
assert!(stripped.contains("header"));
assert!(stripped.contains("footer"));
assert!(!stripped.contains("body"));
assert!(!stripped.contains(START_MARK));
}
#[test]
fn rules_file_without_section_noop_when_no_content() {
let content = "just user text";
let f = RulesFile::parse(content);
assert_eq!(f.without_section(), content);
}
}