assay/
context.rs

1/// Context output formatter for module metadata.
2/// Renders prompt-ready Markdown text from module metadata.
3
4#[derive(Debug, Clone)]
5pub struct ModuleContextEntry {
6    pub module_name: String,
7    pub description: String,
8    pub env_vars: Vec<String>,
9    pub quickrefs: Vec<QuickRefEntry>,
10}
11
12#[derive(Debug, Clone)]
13pub struct QuickRefEntry {
14    pub signature: String,
15    pub return_hint: String,
16    pub description: String,
17}
18
19/// Format module context entries into prompt-ready Markdown.
20///
21/// Output includes:
22/// - Module list with descriptions, env vars, and method signatures
23/// - Built-in functions section (always present)
24///
25/// Lines are kept under 120 chars where practical.
26pub fn format_context(entries: &[ModuleContextEntry]) -> String {
27    let mut output = String::new();
28
29    output.push_str("# Assay Module Context\n\n");
30
31    if entries.is_empty() {
32        output.push_str("No matching modules found.\n\n");
33    } else {
34        output.push_str("## Matching Modules\n\n");
35
36        for entry in entries {
37            output.push_str(&format!("### {}\n", entry.module_name));
38            output.push_str(&format!("{}\n", entry.description));
39
40            if !entry.env_vars.is_empty() {
41                output.push_str(&format!("Env: {}\n", entry.env_vars.join(", ")));
42            }
43
44            if !entry.quickrefs.is_empty() {
45                output.push_str("Methods:\n");
46                for qr in &entry.quickrefs {
47                    output.push_str(&format!(
48                        "  {} -> {} | {}\n",
49                        qr.signature, qr.return_hint, qr.description
50                    ));
51                }
52            }
53
54            output.push('\n');
55        }
56    }
57
58    output.push_str("## Built-in Functions (always available, no require needed)\n");
59    output.push_str("http.get(url, opts?) -> {status, body, headers}\n");
60    output.push_str("json.parse(str) -> table | json.encode(tbl) -> str\n");
61    output.push_str("yaml.parse(str) -> table | yaml.encode(tbl) -> str\n");
62    output.push_str("toml.parse(str) -> table | toml.encode(tbl) -> str\n");
63    output.push_str("base64.encode(str) -> str | base64.decode(str) -> str\n");
64    output.push_str("crypto.jwt_sign(claims, key, alg) -> token\n");
65    output.push_str("crypto.hash(str, alg) -> str | crypto.hmac(key, data, alg?) -> str\n");
66    output.push_str("crypto.random(len) -> str\n");
67    output.push_str("regex.match(pat, str) -> bool | regex.find(pat, str) -> str\n");
68    output.push_str("regex.find_all(pat, str) -> [str] | regex.replace(pat, str, repl) -> str\n");
69    output.push_str("fs.read(path) -> str | fs.write(path, str)\n");
70    output.push_str("db.connect(url) -> conn | db.query(conn, sql, params?) -> [row]\n");
71    output.push_str("db.execute(conn, sql, params?) -> count | db.close(conn)\n");
72    output.push_str("ws.connect(url) -> conn | ws.send(conn, msg)\n");
73    output.push_str("ws.recv(conn) -> msg | ws.close(conn)\n");
74    output.push_str("template.render(path, vars) -> str\n");
75    output.push_str("template.render_string(tmpl, vars) -> str\n");
76    output.push_str("async.spawn(fn) -> handle | async.spawn_interval(fn, ms) -> handle\n");
77    output.push_str("handle:await() | handle:cancel()\n");
78    output.push_str("assert.eq(a, b, msg?) | assert.gt(a, b, msg?) | assert.lt(a, b, msg?)\n");
79    output.push_str("assert.contains(str, sub, msg?) | assert.not_nil(val, msg?)\n");
80    output.push_str("assert.matches(str, pat, msg?)\n");
81    output.push_str("log.info(msg) | log.warn(msg) | log.error(msg)\n");
82    output.push_str("env.get(key) -> str | sleep(secs) | time() -> int\n");
83
84    output
85}
assay/context.rs

assay/
context.rs
