@fairfox/polly 0.13.2 → 0.14.1

package/dist/tools/teach/src/index.js

@@ -236940,6 +236940,39 @@ class HandlerExtractor {
       if (value !== undefined) {
         assignments.push({ field, value });
       }
+      return;
+    }
+    const valueMatch = fieldPath.match(/\.value\.(.+)$/);
+    if (valueMatch && valueMatch[1]) {
+      const field = valueMatch[1];
+      const value = this.extractValue(right);
+      if (value !== undefined) {
+        assignments.push({ field, value });
+      }
+      return;
+    }
+    if (fieldPath.endsWith(".value") && import_ts_morph4.Node.isObjectLiteralExpression(right)) {
+      this.extractObjectLiteralAssignments(right, assignments);
+    }
+  }
+  extractObjectLiteralAssignments(objectLiteral, assignments) {
+    if (!import_ts_morph4.Node.isObjectLiteralExpression(objectLiteral))
+      return;
+    for (const prop of objectLiteral.getProperties()) {
+      if (import_ts_morph4.Node.isPropertyAssignment(prop)) {
+        const name = prop.getName();
+        const initializer = prop.getInitializer();
+        if (!name || !initializer)
+          continue;
+        const value = this.extractValue(initializer);
+        if (value !== undefined) {
+          assignments.push({ field: name, value });
+        }
+      }
+      if (import_ts_morph4.Node.isShorthandPropertyAssignment(prop)) {
+        const name = prop.getName();
+        assignments.push({ field: name, value: "@" });
+      }
     }
   }
   extractElementAccessAssignment(left, right, assignments) {
@@ -236949,9 +236982,9 @@ class HandlerExtractor {
     if (!import_ts_morph4.Node.isPropertyAccessExpression(expr))
       return;
     const fieldPath = this.getPropertyPath(expr);
-    if (!fieldPath.startsWith("state."))
+    const field = this.extractFieldFromStatePath(fieldPath);
+    if (!field)
       return;
-    const field = fieldPath.substring(6);
     const indexExpr = left.getArgumentExpression();
     const index = indexExpr ? indexExpr.getText() : "0";
     const value = this.extractValue(right);
@@ -236963,6 +236996,16 @@ class HandlerExtractor {
       });
     }
   }
+  extractFieldFromStatePath(fieldPath) {
+    if (fieldPath.startsWith("state.")) {
+      return fieldPath.substring(6);
+    }
+    const valueMatch = fieldPath.match(/\.value\.(.+)$/);
+    if (valueMatch && valueMatch[1]) {
+      return valueMatch[1];
+    }
+    return null;
+  }
   extractCompoundAssignment(node, assignments) {
     if (!import_ts_morph4.Node.isBinaryExpression(node))
       return;
@@ -236971,8 +237014,8 @@ class HandlerExtractor {
     const right = node.getRight();
     if (import_ts_morph4.Node.isPropertyAccessExpression(left)) {
       const fieldPath = this.getPropertyPath(left);
-      if (fieldPath.startsWith("state.")) {
-        const field = fieldPath.substring(6);
+      const field = this.extractFieldFromStatePath(fieldPath);
+      if (field) {
         const rightValue = right.getText();
         const tlaOp = operator.slice(0, -1);
         assignments.push({
@@ -237006,8 +237049,8 @@ class HandlerExtractor {
     if (!import_ts_morph4.Node.isPropertyAccessExpression(operand))
       return;
     const fieldPath = this.getPropertyPath(operand);
-    if (fieldPath.startsWith("state.")) {
-      const field = fieldPath.substring(6);
+    const field = this.extractFieldFromStatePath(fieldPath);
+    if (field) {
       const value = operatorText === "++" ? "@ + 1" : "@ - 1";
       assignments.push({ field, value });
     }
@@ -237022,8 +237065,8 @@ class HandlerExtractor {
     const object = expr.getExpression();
     if (import_ts_morph4.Node.isPropertyAccessExpression(object)) {
       const fieldPath = this.getPropertyPath(object);
-      if (fieldPath.startsWith("state.")) {
-        const field = fieldPath.substring(6);
+      const field = this.extractFieldFromStatePath(fieldPath);
+      if (field) {
         const args = node.getArguments().map((arg) => arg.getText());
         const tlaValue = this.translateArrayMethod(methodName, args, fieldPath);
         if (tlaValue) {
@@ -237663,8 +237706,8 @@ class HandlerExtractor {
     if (!import_ts_morph4.Node.isPropertyAccessExpression(left) && !import_ts_morph4.Node.isElementAccessExpression(left))
       return;
     const fieldPath = import_ts_morph4.Node.isPropertyAccessExpression(left) ? this.getPropertyPath(left) : this.getPropertyPath(left.getExpression());
-    if (fieldPath.startsWith("state.")) {
-      const field = fieldPath.substring(6);
+    const field = this.extractFieldFromStatePath(fieldPath);
+    if (field) {
       const line = node.getStartLineNumber();
       const afterAwait = node.getStart() > firstAwaitPos;
       mutations.push({ field, line, afterAwait });
@@ -237681,10 +237724,8 @@ class HandlerExtractor {
     if (!import_ts_morph4.Node.isPropertyAccessExpression(object))
       return;
     const fieldPath = this.getPropertyPath(object);
-    if (!fieldPath.startsWith("state."))
-      return;
-    if (["push", "pop", "shift", "unshift", "splice"].includes(methodName)) {
-      const field = fieldPath.substring(6);
+    const field = this.extractFieldFromStatePath(fieldPath);
+    if (field && ["push", "pop", "shift", "unshift", "splice"].includes(methodName)) {
       const line = node.getStartLineNumber();
       const afterAwait = node.getStart() > firstAwaitPos;
       mutations.push({ field, line, afterAwait });
@@ -240753,4 +240794,4 @@ export {
   generateTeachingMaterial
 };
 
-//# debugId=555FA35403EDF64764756E2164756E21
+//# debugId=CA5D1CF4CE333DEB64756E2164756E21