fix(consistent-chaining): allow TSNonNullExpression (#41)

Author: zzxming

src/rules/consistent-chaining.test.ts

@@ -31,6 +31,24 @@ const valids: ValidTestCase[] = [
       .split('')
       .map(Number)
   `,
+  $`
+    foo.bar!.baz
+      .toString()
+      .split('')
+      .map(Number)
+  `,
+  $`
+    foo.bar.baz
+      .toString()!
+      .split('')
+      .map(Number)
+  `,
+  $`
+    foo.bar.baz!
+      .toString()
+      .split('')
+      .map(Number)
+  `,
 ]
 
 // Check snapshot for fixed code
@@ -170,6 +188,64 @@ const invalid: InvalidTestCase[] = [
         .map()"
       `),
   },
+  {
+    code: $`
+      foo.bar.bar
+        .filter()!.map()
+    `,
+    output: o => expect(o)
+      .toMatchInlineSnapshot(`
+        "foo.bar.bar
+          .filter()!
+        .map()"
+      `),
+  },
+  {
+    code: $`
+      Math
+        .random()! .toString() .split('')!.map(Number)
+    `,
+    output: o => expect(o)
+      .toMatchInlineSnapshot(`
+        "Math
+          .random()!
+         .toString()
+         .split('')!
+        .map(Number)"
+      `),
+  },
+  {
+    code: $`
+      foo({
+        bar: true
+      })!
+        .bar!
+        .baz().boo()
+    `,
+    output: o => expect(o)
+      .toMatchInlineSnapshot(`
+        "foo({
+          bar: true
+        })!
+          .bar!
+          .baz()
+        .boo()"
+      `),
+  },
+  {
+    code: $`
+      this.foo!
+        .toString()
+        .split('').map(Number)
+    `,
+    output: o => expect(o)
+      .toMatchInlineSnapshot(`
+        "this.foo!
+          .toString()
+          .split('')
+        .map(Number)"
+      `),
+  },
 ]
 
 run({

src/rules/consistent-chaining.ts

@@ -70,6 +70,10 @@ export default createEslintRule<Options, MessageIds>({
               current = current.callee
               break
             }
+            case 'TSNonNullExpression': {
+              current = current.expression
+              break
+            }
             default: {
               // Other type of note, that means we are probably reaching out the head
               current = undefined
@@ -85,10 +89,10 @@ export default createEslintRule<Options, MessageIds>({
           const token = context.sourceCode.getTokenBefore(m.property)!
           const tokenBefore = context.sourceCode.getTokenBefore(token)!
           const currentMode: 'single' | 'multi' = token.loc.start.line === tokenBefore.loc.end.line ? 'single' : 'multi'
-
+          const object = m.object.type === 'TSNonNullExpression' ? m.object.expression : m.object
           if (
             leadingPropertyAcccess
-            && (m.object.type === 'ThisExpression' || m.object.type === 'Identifier' || m.object.type === 'MemberExpression' || m.object.type === 'Literal')
+            && (object.type === 'ThisExpression' || object.type === 'Identifier' || object.type === 'MemberExpression' || object.type === 'Literal')
             && currentMode === 'single'
           ) {
             return