use harn_lexer::Span;
use harn_parser::visit;
use harn_parser::{DiagnosticCode as Code, DictEntry, Node, SNode};
use super::dict_keys::{entry_for_key, key_name};
use crate::diagnostic::{LintDiagnostic, LintSeverity};
const RULE_NAME: &str = "schema-shaped-tool-parameters";
const JSON_SCHEMA_DOCUMENT_KEYS: &[&str] = &["type", "properties", "required"];
pub(crate) fn check_schema_shaped_tool_parameters(
program: &[SNode],
diagnostics: &mut Vec<LintDiagnostic>,
) {
visit::walk_program(program, &mut |node| {
let Node::DictLiteral(entries) = &node.node else {
return;
};
let Some(parameters) = entry_for_key(entries, "parameters") else {
return;
};
let Node::DictLiteral(inner) = ¶meters.value.node else {
return;
};
if schema_shaped(inner) {
diagnostics.push(make_diagnostic(parameters.value.span));
}
});
}
fn schema_shaped(entries: &[DictEntry]) -> bool {
!entries.is_empty()
&& entries.iter().all(|entry| {
key_name(&entry.key)
.is_some_and(|key| JSON_SCHEMA_DOCUMENT_KEYS.contains(&key.as_str()))
})
}
fn make_diagnostic(span: Span) -> LintDiagnostic {
LintDiagnostic {
code: Code::LintSchemaShapedToolParameters,
rule: RULE_NAME.into(),
message: format!(
"this `parameters` map has only JSON Schema document keys (`{}`), so a reader that \
treats `parameters` as a per-parameter map sees parameters with those names.",
JSON_SCHEMA_DOCUMENT_KEYS.join("`, `")
),
span,
severity: LintSeverity::Error,
suggestion: Some(
"spell a complete schema as `inputSchema`, and keep `parameters` for the \
per-parameter map."
.to_string(),
),
fix: None,
}
}