fix: use declaration end. (#4)

Author: knightedcodemonkey

package-lock.json

@@ -1,6 +1,6 @@
 {
   "name": "@knighted/display-name",
-  "version": "1.0.0-alpha.1",
+  "version": "1.0.0-alpha.2",
   "description": "Codemod to add React displayName properties to function components.",
   "type": "module",
   "exports": {

package.json

@@ -101,7 +101,7 @@ const modify = async (source: string, options: Options = defaultOptions) => {
                     (!opts.requirePascal || pascal.test(name))
                   ) {
                     code.appendRight(
-                      declarator.end + 1,
+                      declaration.end,
                       `\n${name}.displayName = '${name}'${opts.insertSemicolon ? ';' : ''}`,
                     )
                   }

src/displayName.ts

@@ -256,4 +256,31 @@ describe('@knighted/displayName', () => {
     `.replace(/\s/g, ''),
     )
   })
+
+  it('works with multiple declarations', async () => {
+    const src = `
+      const A = () => <>a</>,
+      B = () => <>b</>;
+
+      const C = function() {
+        return <>c</>
+      }
+    `
+    const code = await modify(src)
+
+    assert.equal(
+      code.replace(/\s/g, ''),
+      `
+        const A = () => <>a</>,
+        B = () => <>b</>;
+        A.displayName = 'A';
+        B.displayName = 'B';
+
+        const C = function() {
+          return <>c</>
+        }
+        C.displayName = 'C';
+    `.replace(/\s/g, ''),
+    )
+  })
 })