use crate::context::LintContext;
use crate::diagnostic::Severity;
use crate::rule::{Rule, RuleCategory, RuleMeta};
use vize_relief::ast::{DirectiveNode, ElementNode, ExpressionNode, PropNode};
static META: RuleMeta = RuleMeta {
name: "vue/require-v-for-key",
description: "Require `v-bind:key` with `v-for` directives",
category: RuleCategory::Essential,
fixable: false,
default_severity: Severity::Error,
};
pub struct RequireVForKey;
impl Rule for RequireVForKey {
fn meta(&self) -> &'static RuleMeta {
&META
}
fn check_directive<'a>(
&self,
ctx: &mut LintContext<'a>,
element: &ElementNode<'a>,
directive: &DirectiveNode<'a>,
) {
if directive.name.as_str() != "for" {
return;
}
if element.tag.as_str() == "template" {
return;
}
let has_key = element.props.iter().any(|prop| match prop {
PropNode::Attribute(attr) => attr.name.as_str() == "key",
PropNode::Directive(dir) => {
if dir.name.as_str() == "bind" {
if let Some(ExpressionNode::Simple(s)) = &dir.arg {
return s.content.as_str() == "key";
}
}
false
}
});
if !has_key {
ctx.error_with_help(
ctx.t_fmt(
"vue/require-v-for-key.message",
&[("tag", element.tag.as_str())],
),
&directive.loc,
ctx.t("vue/require-v-for-key.help"),
);
}
}
}
#[cfg(test)]
mod tests {
use super::RequireVForKey;
use crate::linter::Linter;
use crate::rule::RuleRegistry;
fn create_linter() -> Linter {
let mut registry = RuleRegistry::new();
registry.register(Box::new(RequireVForKey));
Linter::with_registry(registry)
}
#[test]
fn test_valid_v_for_with_key() {
let linter = create_linter();
let result = linter.lint_template(
r#"<ul><li v-for="item in items" :key="item.id">{{ item.name }}</li></ul>"#,
"test.vue",
);
assert_eq!(result.error_count, 0);
}
#[test]
fn test_invalid_v_for_without_key() {
let linter = create_linter();
let result = linter.lint_template(
r#"<ul><li v-for="item in items">{{ item.name }}</li></ul>"#,
"test.vue",
);
assert_eq!(result.error_count, 1);
insta::assert_debug_snapshot!(result.diagnostics);
}
#[test]
fn test_valid_v_for_with_static_key() {
let linter = create_linter();
let result = linter.lint_template(
r#"<div v-for="item in items" key="static"></div>"#,
"test.vue",
);
assert_eq!(result.error_count, 0);
}
#[test]
fn test_template_v_for_ignored() {
let linter = create_linter();
let result = linter.lint_template(
r#"<template v-for="item in items"><div :key="item.id">{{ item }}</div></template>"#,
"test.vue",
);
assert_eq!(result.error_count, 0);
}
}