pub const SOURCE: &str = include_str!("source.md");
pub fn extract_surface<'a>(source: &'a str, surface: &str) -> Option<&'a str> {
let open = format!("<!-- @surface {surface} -->\n");
let start = source.find(&open)? + open.len();
let rest = &source[start..];
let end_offset = rest.find("<!-- @end -->")?;
Some(&rest[..end_offset])
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn extracts_server_instructions_byte_for_byte() {
let extracted = extract_surface(SOURCE, "server_instructions")
.expect("server_instructions surface present in source.md");
assert_eq!(
extracted,
crate::prompts::SERVER_INSTRUCTIONS,
"source.md must reproduce server_instructions.md byte-for-byte; \
diff between extract and constant means source.md drifted from the file"
);
}
#[test]
fn extracts_onboarding_prompt_byte_for_byte() {
let extracted = extract_surface(SOURCE, "onboarding_prompt")
.expect("onboarding_prompt surface present in source.md");
assert_eq!(
extracted,
crate::prompts::RAW_ONBOARDING_PROMPT,
"source.md must reproduce onboarding_prompt.md byte-for-byte; \
diff between extract and constant means source.md drifted from the file"
);
}
#[test]
fn unknown_surface_returns_none() {
assert!(extract_surface(SOURCE, "no_such_surface").is_none());
}
#[test]
fn extract_handles_inline_string() {
let src = "<!-- @surface foo -->\nbody\n<!-- @end -->\n";
assert_eq!(extract_surface(src, "foo"), Some("body\n"));
}
#[test]
fn extract_returns_none_when_close_tag_missing() {
let src = "<!-- @surface foo -->\nbody without close tag\n";
assert!(extract_surface(src, "foo").is_none());
}
}