From 535a61f676b5e181a7a9a0038f0bce3e69b75a44 Mon Sep 17 00:00:00 2001
From: bendtherules <abhasbhattacharya2@gmail.com>
Date: Sat, 25 Jan 2020 07:42:04 +0530
Subject: [PATCH] example - App - Create AST, consume ASTVisualizer and
 transform ast after some time

---
 example/src/App.js | 50 ++++++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 48 insertions(+), 2 deletions(-)

diff --git a/example/src/App.js b/example/src/App.js
index 2782ba1..7c00cb1 100644
--- a/example/src/App.js
+++ b/example/src/App.js
@@ -1,12 +1,58 @@
 import React, { Component } from "react";
 
-import ExampleComponent from "ts-transformer-visualizer";
+import {
+  ASTVisualizer,
+  initSourceFile,
+  updateSourceFile
+} from "ts-transformer-visualizer";
+import * as ts from "typescript";
+
+const source = "x = 1;";
+
+const mainFile = ts.createSourceFile("/test", source, ts.ScriptTarget.ES2015);
+
+const transformer = context => {
+  return sourceFile => {
+    /**
+     *
+     * @param {ts.Node} node
+     * @returns {ts.VisitResult<ts.Node>}
+     */
+    const visitor = node => {
+      // If it is a expression statement,
+      if (ts.isExpressionStatement(node)) {
+        // Return it twice.
+        // Effectively duplicating the statement
+        return [node, ts.getMutableClone(node)];
+      }
+
+      return ts.visitEachChild(node, visitor, context);
+    };
+    return ts.visitNode(sourceFile, visitor);
+  };
+};
+
+// const printer = ts.createPrinter();
+// const result = ts.transform(mainFile, [transformer]);
+// const transformedSourceFile = result.transformed[0];
+// const newContent = printer.printFile(transformedSourceFile);
+// result.dispose();
+
+// console.log(newContent);
 
 export default class App extends Component {
+  componentDidMount() {
+    initSourceFile(mainFile);
+
+    setTimeout(() => {
+      const modifiedFile = ts.transform(mainFile, [transformer]).transformed[0];
+      updateSourceFile(modifiedFile);
+    }, 5000);
+  }
   render() {
     return (
       <div>
-        <ExampleComponent text="Modern React component module" />
+        <ASTVisualizer />
       </div>
     );
   }