@eventra_dev/eventra-cli 0.0.5 → 0.0.7

package/src/utils/extract.ts

@@ -2,6 +2,7 @@ import {
   CallExpression,
   Node,
   ObjectLiteralExpression,
+  SyntaxKind,
 } from "ts-morph";
 
 export function extractEvent(
@@ -15,37 +16,59 @@ export function extractEvent(
 
   if (!node) return null;
 
-  for (let i = 1; i < parts.length; i++) {
-    if (
-      Node.isObjectLiteralExpression(node)
-    ) {
-      const obj: ObjectLiteralExpression =
-        node;
+  node = unwrap(node);
 
-      const prop =
-        obj.getProperty(parts[i]);
+  for (let i = 1; i < parts.length; i++) {
+    if (!Node.isObjectLiteralExpression(node)) {
+      return null;
+    }
 
-      if (!prop) return null;
+    const obj: ObjectLiteralExpression =
+      node;
 
-      if (
-        Node.isPropertyAssignment(prop)
-      ) {
-        const initializer =
-          prop.getInitializer();
+    const prop =
+      obj.getProperty(parts[i]);
 
-        if (!initializer) return null;
+    if (!prop) return null;
 
-        node = initializer;
-      }
+    if (!Node.isPropertyAssignment(prop)) {
+      return null;
     }
+
+    const initializer =
+      prop.getInitializer();
+
+    if (!initializer) return null;
+
+    node = unwrap(initializer);
+  }
+
+  if (Node.isStringLiteral(node)) {
+    return node.getLiteralText();
   }
 
   if (
-    node &&
-    Node.isStringLiteral(node)
+    node.getKind() ===
+    SyntaxKind.NoSubstitutionTemplateLiteral
   ) {
-    return node.getLiteralText();
+    return node
+      .getText()
+      .replace(/`/g, "");
   }
 
   return null;
 }
+
+function unwrap(node: Node): Node {
+  let current = node;
+
+  while (
+    Node.isParenthesizedExpression(
+      current
+    )
+    ) {
+    current = current.getExpression();
+  }
+
+  return current;
+}

package/src/utils/parsers/vue.ts

@@ -11,9 +11,15 @@ export function parseVue(
       /<script[\s\S]*?>([\s\S]*?)<\/script>/
     );
 
+  return `
+${script?.[1] ?? ""}
+
+function __vue_template__() {
   return (
-    (template?.[1] ?? "") +
-    "\n" +
-    (script?.[1] ?? "")
+    <>
+      ${template?.[1] ?? ""}
+    </>
   );
 }
+`;
+}

package/src/utils/scanners/component-wrappers.ts

@@ -61,7 +61,6 @@ export function scanComponentWrappers(
 
         if (!init) continue;
 
-        // event="signup"
         if (
           init.getKind() ===
           SyntaxKind.StringLiteral
@@ -76,7 +75,6 @@ export function scanComponentWrappers(
           );
         }
 
-        // event={"signup"}
         if (
           init.getKind() ===
           SyntaxKind.JsxExpression

package/src/utils/scanners/function-wrappers.ts

@@ -1,6 +1,9 @@
 import {
   SourceFile,
   SyntaxKind,
+  Node,
+  CallExpression,
+  PropertyAccessExpression,
 } from "ts-morph";
 
 import { extractEvent } from "../extract";
@@ -10,8 +13,7 @@ export function scanFunctionWrappers(
   source: SourceFile,
   wrappers: FunctionWrapper[]
 ) {
-  const events =
-    new Set<string>();
+  const events = new Set<string>();
 
   const calls =
     source.getDescendantsOfKind(
@@ -19,43 +21,19 @@ export function scanFunctionWrappers(
     );
 
   for (const call of calls) {
-    const expression =
-      call.getExpression();
-
-    let name: string | null = null;
-
-    // trackFeature()
-    if (
-      expression.getKind() ===
-      SyntaxKind.Identifier
-    ) {
-      name =
-        expression.getText();
-    }
-
-    // analytics.trackFeature()
-    if (
-      expression.getKind() ===
-      SyntaxKind.PropertyAccessExpression
-    ) {
-      const prop =
-        expression.asKindOrThrow(
-          SyntaxKind.PropertyAccessExpression
-        );
-
-      name = prop.getName();
-    }
+    const name =
+      getFunctionName(call);
 
     if (!name) continue;
 
     for (const wrapper of wrappers) {
-      if (name !== wrapper.name)
+      if (wrapper.name !== name)
         continue;
 
       const event =
-        extractEvent(
+        extractEventFromArgs(
           call,
-          wrapper.path
+          wrapper.event
         );
 
       if (event)
@@ -65,3 +43,98 @@ export function scanFunctionWrappers(
 
   return events;
 }
+
+function getFunctionName(
+  call: CallExpression
+): string | null {
+  const expression =
+    call.getExpression();
+
+  // trackFeature()
+  if (
+    Node.isIdentifier(
+      expression
+    )
+  ) {
+    return expression.getText();
+  }
+
+  // analytics.trackFeature()
+  if (
+    Node.isPropertyAccessExpression(
+      expression
+    )
+  ) {
+    return getDeepName(
+      expression
+    );
+  }
+
+  return null;
+}
+
+function getDeepName(
+  node: PropertyAccessExpression
+): string {
+  let current:
+    | Node
+    | undefined = node;
+
+  let name = "";
+
+  while (
+    Node.isPropertyAccessExpression(
+      current
+    )
+    ) {
+    name =
+      current.getName();
+
+    current =
+      current.getExpression();
+  }
+
+  return name;
+}
+
+
+function extractEventFromArgs(
+  call: CallExpression,
+  event?: string
+): string | null {
+  const args = call.getArguments();
+
+  for (let i = 0; i < args.length; i++) {
+    const arg = args[i];
+
+    // string case
+    if (!event) {
+      if (Node.isStringLiteral(arg)) {
+        return arg.getLiteralText();
+      }
+
+      // template literal
+      if (
+        arg.getKind() ===
+        SyntaxKind.NoSubstitutionTemplateLiteral
+      ) {
+        return arg
+          .getText()
+          .replace(/`/g, "");
+      }
+    }
+
+    // object case
+    if (event) {
+      const result =
+        extractEvent(
+          call,
+          `${i}.${event}`
+        );
+
+      if (result) return result;
+    }
+  }
+
+  return null;
+}

package/tests/fixtures/backend/express/app.ts

@@ -0,0 +1,78 @@
+/* eslint-disable */
+
+declare function track(event: string): void;
+
+const express = (...args: any[]) => ({
+  get: (...args: any[]) => {},
+  post: (...args: any[]) => {},
+  put: (...args: any[]) => {},
+  delete: (...args: any[]) => {},
+  use: (...args: any[]) => {},
+  listen: (...args: any[]) => {}
+});
+
+const app = express();
+
+// ===== direct =====
+track("express_event");
+
+// ===== middleware =====
+app.use(() => {
+  track("middleware_event");
+});
+
+// ===== get =====
+app.get("/", () => {
+  track("get_event");
+});
+
+// ===== post =====
+app.post("/users", () => {
+  track("post_event");
+});
+
+// ===== put =====
+app.put("/users", () => {
+  track("put_event");
+});
+
+// ===== delete =====
+app.delete("/users", () => {
+  track("delete_event");
+});
+
+// ===== nested =====
+app.get("/nested", () => {
+  if (true) {
+    track("nested_event");
+  }
+});
+
+// ===== async =====
+app.get("/async", async () => {
+  track("async_event");
+});
+
+// ===== arrow =====
+const handler = () => {
+  track("arrow_event");
+};
+
+app.get("/arrow", handler);
+
+// ===== function =====
+function service() {
+  track("function_event");
+}
+
+// ===== class =====
+class Service {
+  run() {
+    track("class_event");
+  }
+}
+
+// ===== listen =====
+app.listen(3000, () => {
+  track("listen_event");
+});

package/tests/fixtures/backend/nest/service.ts

@@ -0,0 +1,70 @@
+/* eslint-disable */
+
+declare function track(event: string): void;
+
+function Injectable(): any {
+  return () => {};
+}
+
+function Controller(): any {
+  return () => {};
+}
+
+function Get(): any {
+  return () => {};
+}
+
+function Post(): any {
+  return () => {};
+}
+
+// ===== service =====
+@Injectable()
+export class TestService {
+  run() {
+    track("nest_event");
+  }
+
+  async asyncRun() {
+    track("async_event");
+  }
+
+  arrow = () => {
+    track("arrow_event");
+  };
+}
+
+// ===== controller =====
+@Controller()
+export class TestController {
+  @Get()
+  get() {
+    track("get_event");
+  }
+
+  @Post()
+  post() {
+    track("post_event");
+  }
+
+  nested() {
+    if (true) {
+      track("nested_event");
+    }
+  }
+}
+
+// ===== function =====
+function service() {
+  track("function_event");
+}
+
+// ===== class =====
+class AnotherService {
+  run() {
+    track("class_event");
+  }
+}
+
+// ===== direct =====
+track("direct_event");

package/tests/fixtures/backend/node/index.ts

@@ -0,0 +1,63 @@
+/* eslint-disable */
+
+declare function track(event: string): void;
+
+const eventName = "variable_event";
+
+// ===== direct =====
+track("node_event");
+
+// ===== multiline =====
+track(
+  "multiline_event"
+);
+
+// ===== template string =====
+track(`template_event`);
+
+// ===== conditional =====
+if (true) {
+  track("conditional_event");
+}
+
+// ===== ternary =====
+true
+  ? track("ternary_a")
+  : track("ternary_b");
+
+// ===== variable =====
+track(eventName);
+
+// ===== function =====
+function run() {
+  track("function_event");
+}
+
+// ===== arrow =====
+const handler = () => {
+  track("arrow_event");
+};
+
+// ===== class =====
+class Service {
+  run() {
+    track("class_event");
+  }
+}
+
+// ===== nested =====
+function outer() {
+  function inner() {
+    track("nested_event");
+  }
+}
+
+// ===== async =====
+async function asyncRun() {
+  track("async_event");
+}
+
+// ===== promise =====
+Promise.resolve().then(() => {
+  track("promise_event");
+});

package/tests/fixtures/frontend/next/page.tsx

@@ -0,0 +1,101 @@
+/* eslint-disable */
+
+declare function track(event: string): void;
+declare function trackFeature(event: string): void;
+
+declare const analytics: any;
+declare const Button: any;
+declare const TrackedButton: any;
+declare const MyButton: any;
+
+const eventName = "variable_event";
+
+export default function Page() {
+  // ===== direct =====
+  track("next_event");
+
+  // ===== multiline =====
+  track(
+    "multiline_event"
+  );
+
+  // ===== template string =====
+  track(`template_event`);
+
+  // ===== wrapper function =====
+  trackFeature("wrapper_function");
+
+  // ===== object =====
+  analytics.track("object_track");
+
+  // ===== nested object =====
+  analytics.events.track("nested_track");
+
+  // ===== variable =====
+  track(eventName);
+
+  // ===== conditional =====
+  if (true) {
+    track("conditional_event");
+  }
+
+  // ===== ternary =====
+  true
+    ? track("ternary_a")
+    : track("ternary_b");
+
+  // ===== arrow function =====
+  const run = () => {
+    track("arrow_event");
+  };
+
+  // ===== function =====
+  function test() {
+    track("function_event");
+  }
+
+  // ===== class =====
+  class Service {
+    run() {
+      track("class_event");
+    }
+  }
+
+  return (
+    <>
+      {/* basic */}
+      <Button event="next_button" />
+
+      {/* expression */}
+      <Button event={"expression_event"} />
+
+      {/* wrapper component */}
+      <TrackedButton event="tracked_button" />
+
+      {/* another wrapper */}
+      <MyButton event="my_button" />
+
+      {/* nested */}
+      <div>
+        <Button event="nested_button" />
+      </div>
+
+      {/* conditional */}
+      {true && (
+        <Button event="conditional_button" />
+      )}
+
+      {/* ternary */}
+      {true
+        ? <Button event="ternary_a_button" />
+        : <Button event="ternary_b_button" />
+      }
+
+      {/* array */}
+      {[
+        <Button key="1" event="array_1" />,
+        <Button key="2" event="array_2" />
+      ]}
+    </>
+  );
+}

package/tests/fixtures/frontend/react/App.tsx

@@ -0,0 +1,104 @@
+/* eslint-disable */
+
+declare function track(event: string): void;
+declare function trackFeature(event: string): void;
+
+declare const analytics: any;
+declare const Button: any;
+declare const TrackedButton: any;
+declare const MyButton: any;
+
+const eventName = "variable_event";
+
+// ===== direct =====
+track("direct_event");
+
+// ===== multiline =====
+track(
+  "multiline_event"
+);
+
+// ===== template string =====
+track(`template_event`);
+
+// ===== conditional =====
+if (true) {
+  track("conditional_event");
+}
+
+// ===== ternary =====
+true
+  ? track("ternary_a")
+  : track("ternary_b");
+
+// ===== wrapper function =====
+trackFeature("wrapper_function");
+
+// ===== object function =====
+analytics.track("object_track");
+
+// ===== nested object =====
+analytics.events.track("nested_track");
+
+// ===== variable (optional support) =====
+track(eventName);
+
+// ===== function =====
+function test() {
+  track("function_event");
+}
+
+// ===== arrow function =====
+const run = () => {
+  track("arrow_event");
+};
+
+// ===== class =====
+class TestService {
+  run() {
+    track("class_event");
+  }
+}
+
+// ===== component return =====
+export default function Page() {
+  track("page_event");
+
+  return (
+    <>
+      {/* basic */}
+      <Button event="button_event" />
+
+      {/* wrapper component */}
+      <TrackedButton event="tracked_button" />
+
+      {/* another wrapper */}
+      <MyButton event="my_button" />
+
+      {/* expression */}
+      <Button event={"expression_event"} />
+
+      {/* nested */}
+      <div>
+        <Button event="nested_button" />
+      </div>
+
+      {/* conditional */}
+      {true && (
+        <Button event="conditional_button" />
+      )}
+
+      {/* ternary */}
+      {true
+        ? <Button event="ternary_a_button" />
+        : <Button event="ternary_b_button" />
+      }
+
+      {/* array */}
+      {[
+        <Button key="1" event="array_1" />,
+        <Button key="2" event="array_2" />
+      ]}
+    </>
+  );
+}