fix: use node.forParseResult if present (#18)

Author: danielroe

src/utils/template.ts

@@ -109,12 +109,15 @@ function handleNode(
     }
     case NodeTypes.DIRECTIVE: {
       const nodes = [
-        node.exp,
+        ...node.forParseResult
+          ? [
+              node.forParseResult?.source,
+              node.forParseResult?.value,
+              node.forParseResult?.key,
+              node.forParseResult?.index,
+            ]
+          : [node.exp],
         // node.arg,
-        // node.forParseResult?.source,
-        // node.forParseResult?.value,
-        // node.forParseResult?.key,
-        // node.forParseResult?.index,
         ...node.modifiers,
       ].filter(item => !!item)
       for (const child of nodes) {

test/template.test.ts

@@ -11,6 +11,9 @@ describe('transform typescript template', () => {
 
     expect(await fixture(`<div v-for="(item as string, index) in items as unknown[]" :key="item" :index>{{ item }}</div>`))
       .toEqual(`<div v-for="(item, index) in items" :key="item" :index>{{ item }}</div>`)
+
+    expect(await fixture(`<div v-for="(item, index) of items" />`))
+      .toEqual(`<div v-for="(item, index) of items" />`)
   })
 
   it('v-if', async () => {