use super::Context;
use super::LintRule;
use swc_ecmascript::visit::Node;
use swc_ecmascript::visit::Visit;
use std::sync::Arc;
pub struct NoNonNullAssertion;
impl LintRule for NoNonNullAssertion {
fn new() -> Box<Self> {
Box::new(NoNonNullAssertion)
}
fn code(&self) -> &'static str {
"no-non-null-assertion"
}
fn lint_module(
&self,
context: Arc<Context>,
module: &swc_ecmascript::ast::Module,
) {
let mut visitor = NoNonNullAssertionVisitor::new(context);
visitor.visit_module(module, module);
}
}
struct NoNonNullAssertionVisitor {
context: Arc<Context>,
}
impl NoNonNullAssertionVisitor {
fn new(context: Arc<Context>) -> Self {
Self { context }
}
}
impl Visit for NoNonNullAssertionVisitor {
fn visit_ts_non_null_expr(
&mut self,
non_null_expr: &swc_ecmascript::ast::TsNonNullExpr,
_parent: &dyn Node,
) {
self.context.add_diagnostic(
non_null_expr.span,
"no-non-null-assertion",
"do not use non-null assertion",
);
}
}
#[cfg(test)]
mod tests {
use super::*;
use crate::test_util::*;
#[test]
fn should_ok() {
assert_lint_ok::<NoNonNullAssertion>("instance.doWork();");
assert_lint_ok::<NoNonNullAssertion>("foo.bar?.includes('baz')");
assert_lint_ok::<NoNonNullAssertion>("x;");
assert_lint_ok::<NoNonNullAssertion>("x.y;");
assert_lint_ok::<NoNonNullAssertion>("x.y.z;");
assert_lint_ok::<NoNonNullAssertion>("x?.y.z;");
assert_lint_ok::<NoNonNullAssertion>("x?.y?.z;");
assert_lint_ok::<NoNonNullAssertion>("!x;");
}
#[test]
fn should_err() {
assert_lint_err::<NoNonNullAssertion>("instance!.doWork()", 0);
assert_lint_err::<NoNonNullAssertion>("foo.bar!.includes('baz');", 0);
assert_lint_err::<NoNonNullAssertion>("x.y.z!?.();", 0);
assert_lint_err::<NoNonNullAssertion>("x!?.y.z;", 0);
assert_lint_err::<NoNonNullAssertion>("x!?.[y].z;", 0);
assert_lint_err::<NoNonNullAssertion>("x.y.z!!();", 0);
assert_lint_err::<NoNonNullAssertion>("x.y!!;", 0);
assert_lint_err::<NoNonNullAssertion>("x!!.y;", 0);
assert_lint_err::<NoNonNullAssertion>("x!!!;", 0);
assert_lint_err::<NoNonNullAssertion>("x.y?.z!();", 0);
assert_lint_err::<NoNonNullAssertion>("x.y.z!();", 0);
assert_lint_err::<NoNonNullAssertion>("x![y]?.z;", 0);
assert_lint_err::<NoNonNullAssertion>("x![y];", 0);
assert_lint_err::<NoNonNullAssertion>("!x!.y;", 1);
assert_lint_err::<NoNonNullAssertion>("x!.y?.z;", 0);
assert_lint_err::<NoNonNullAssertion>("x.y!;", 0);
assert_lint_err::<NoNonNullAssertion>("x!.y;", 0);
assert_lint_err::<NoNonNullAssertion>("x!;", 0);
}
}