From bf90ea1b7f7293c5f411908336a1960a1376b5d8 Mon Sep 17 00:00:00 2001
From: HoikanChan <hoikan.chen@gmail.com>
Date: Thu, 11 Apr 2024 10:38:03 +0800
Subject: [PATCH] feat(watch): switch to label watch

---
 .../class-transformer/src/pluginProvider.ts   | 24 +++++++++++++------
 .../src/test/fn2Class.test.ts                 |  6 ++---
 2 files changed, 20 insertions(+), 10 deletions(-)

diff --git a/packages/transpiler/class-transformer/src/pluginProvider.ts b/packages/transpiler/class-transformer/src/pluginProvider.ts
index 878b50cf..50549fde 100644
--- a/packages/transpiler/class-transformer/src/pluginProvider.ts
+++ b/packages/transpiler/class-transformer/src/pluginProvider.ts
@@ -54,7 +54,7 @@ export class PluginProvider {
       }
 
       // handle watch
-      if (this.t.isStatement(node)) {
+      if (this.t.isLabeledStatement(node) && node.label.name === 'watch') {
         // transform the watch statement to watch method
         classTransformer.transformWatch(node);
         return;
@@ -113,7 +113,7 @@ class ClassComponentTransformer {
   // transform function component to class component extends View
   genClassComponent(name: string) {
     // generate ctor and push this.initExpressions to ctor
-    let nestedDestructuringBindingsMethod: t.ClassMethod;
+    let nestedDestructuringBindingsMethod: t.ClassMethod | null = null;
     if (this.nestedDestructuringBindings.length) {
       nestedDestructuringBindingsMethod = this.t.classMethod(
         'method',
@@ -205,10 +205,20 @@ class ClassComponentTransformer {
     throw new Error('Unsupported props type, please use object destructuring or identifier.');
   }
 
-  // transform node to method with watch decorator
-  transformWatch(node: ToWatchNode) {
+  /**
+   * transform node to watch label to watch decorator
+   * e.g.
+   * 
+   * watch: console.log(state)
+   * // transform into
+   * @Watch
+   * _watch() {
+   *   console.log(state)
+   * }
+   */
+  transformWatch(node: t.LabeledStatement) {
     const id = this.functionScope.generateUidIdentifier(DECORATOR_WATCH.toLowerCase());
-    const method = this.t.classMethod('method', id, [], this.t.blockStatement([node]), false, false);
+    const method = this.t.classMethod('method', id, [], this.t.blockStatement([node.body]), false, false);
     method.decorators = [this.t.decorator(this.t.identifier(DECORATOR_WATCH))];
     this.addProperty(method);
   }
@@ -250,7 +260,7 @@ class ClassComponentTransformer {
         let key = prop.key;
         let defaultVal: t.Expression;
         if (this.t.isIdentifier(key)) {
-          let alias: t.Identifier;
+          let alias: t.Identifier | null = null;
           if (this.t.isAssignmentPattern(prop.value)) {
             const propName = prop.value.left;
             defaultVal = prop.value.right;
@@ -320,7 +330,7 @@ class ClassComponentTransformer {
   }
 
   // add prop to class, like @prop name = '';
-  private addClassProperty(key: t.Identifier, decorator: string, defaultValue?: t.Expression) {
+  private addClassProperty(key: t.Identifier, decorator: string | null, defaultValue?: t.Expression) {
     // clone the key to avoid reference issue
     const id = this.t.cloneNode(key);
     this.addProperty(
diff --git a/packages/transpiler/class-transformer/src/test/fn2Class.test.ts b/packages/transpiler/class-transformer/src/test/fn2Class.test.ts
index 72536df3..f74a22cf 100644
--- a/packages/transpiler/class-transformer/src/test/fn2Class.test.ts
+++ b/packages/transpiler/class-transformer/src/test/fn2Class.test.ts
@@ -178,7 +178,7 @@ describe('fn2Class', () => {
       transform(`
         export default function CountComp() {
           let count = 0;
-          console.log(count);
+          watch: console.log(count);
 
           return <div>{count}</div>;
         }
@@ -204,7 +204,7 @@ describe('fn2Class', () => {
       transform(`
         export default function CountComp() {
           let count = 0;
-          for (let i = 0; i < count; i++) {
+          watch: for (let i = 0; i < count; i++) {
             console.log(\`The count change to: \${i}\`);
           }
           return <>
@@ -242,7 +242,7 @@ describe('fn2Class', () => {
       transform(`
         export default function CountComp() {
           let count = 0;
-          if (count > 0) {
+          watch: if (count > 0) {
             console.log(\`The count is greater than 0\`);
           }