@esportsplus/reactivity 0.23.0 → 0.23.1

package/build/transformer/detector.js

@@ -0,0 +1,38 @@
+import { mightNeedTransform as checkTransform } from '@esportsplus/typescript/transformer';
+import ts from 'typescript';
+const REACTIVE_REGEX = /\breactive\b/;
+function visit(ctx, node) {
+    if (ctx.hasImport && ctx.hasUsage) {
+        return;
+    }
+    if (ts.isImportDeclaration(node) &&
+        node.importClause?.namedBindings &&
+        ts.isNamedImports(node.importClause.namedBindings)) {
+        let elements = node.importClause.namedBindings.elements;
+        for (let i = 0, n = elements.length; i < n; i++) {
+            let el = elements[i], name = el.propertyName?.text ?? el.name.text;
+            if (name === 'reactive') {
+                ctx.hasImport = true;
+                break;
+            }
+        }
+    }
+    if (ts.isCallExpression(node) &&
+        ts.isIdentifier(node.expression) &&
+        node.expression.text === 'reactive') {
+        ctx.hasUsage = true;
+    }
+    ts.forEachChild(node, n => visit(ctx, n));
+}
+const mightNeedTransform = (code) => {
+    if (!checkTransform(code, { regex: REACTIVE_REGEX })) {
+        return false;
+    }
+    let ctx = {
+        hasImport: false,
+        hasUsage: false
+    };
+    visit(ctx, ts.createSourceFile('detect.ts', code, ts.ScriptTarget.Latest, false));
+    return ctx.hasImport && ctx.hasUsage;
+};
+export { mightNeedTransform };

package/build/transformer/{core/index.d.ts → index.d.ts}

@@ -1,4 +1,4 @@
-import type { TransformOptions, TransformResult } from '../../types.js';
+import type { TransformOptions, TransformResult } from '../types.js';
 import { mightNeedTransform } from './detector.js';
 import ts from 'typescript';
 declare const createTransformer: (options?: TransformOptions) => ts.TransformerFactory<ts.SourceFile>;

package/build/transformer/plugins/esbuild.js

@@ -1,11 +1,12 @@
-import { mightNeedTransform, transform } from '../../transformer/core/index.js';
+import { TRANSFORM_PATTERN } from '@esportsplus/typescript/transformer';
+import { mightNeedTransform, transform } from '../../transformer/index.js';
 import fs from 'fs';
 import ts from 'typescript';
 export default (options) => {
     return {
         name: '@esportsplus/reactivity/plugin-esbuild',
         setup(build) {
-            build.onLoad({ filter: /\.[tj]sx?$/ }, async (args) => {
+            build.onLoad({ filter: TRANSFORM_PATTERN }, async (args) => {
                 let code = await fs.promises.readFile(args.path, 'utf8');
                 if (!mightNeedTransform(code)) {
                     return null;

package/build/transformer/plugins/tsc.js

@@ -1,4 +1,4 @@
-import { createTransformer } from '../../transformer/core/index.js';
+import { createTransformer } from '../../transformer/index.js';
 export default (_program) => {
     return createTransformer();
 };

package/build/transformer/plugins/vite.js

@@ -1,6 +1,6 @@
-import { mightNeedTransform, transform } from '../../transformer/core/index.js';
+import { TRANSFORM_PATTERN } from '@esportsplus/typescript/transformer';
+import { mightNeedTransform, transform } from '../../transformer/index.js';
 import ts from 'typescript';
-const TRANSFORM_PATTERN = /\.[tj]sx?$/;
 export default (options) => {
     return {
         enforce: 'pre',

package/build/transformer/transforms/auto-dispose.js

@@ -0,0 +1,119 @@
+import { uid, TRAILING_SEMICOLON } from '@esportsplus/typescript/transformer';
+import { applyReplacements } from './utilities.js';
+import ts from 'typescript';
+function processFunction(node, sourceFile, edits) {
+    if (!node.body || !ts.isBlock(node.body)) {
+        return;
+    }
+    let ctx = {
+        disposables: [],
+        effectsToCapture: [],
+        parentBody: node.body,
+        returnStatement: null
+    };
+    visitBody(ctx, node.body);
+    if (ctx.disposables.length === 0 || !ctx.returnStatement || !ctx.returnStatement.expression) {
+        return;
+    }
+    let cleanupFn = ctx.returnStatement.expression;
+    if (!cleanupFn.body) {
+        return;
+    }
+    let disposeStatements = [];
+    for (let i = ctx.disposables.length - 1; i >= 0; i--) {
+        let d = ctx.disposables[i];
+        if (d.type === 'reactive') {
+            disposeStatements.push(`${d.name}.dispose();`);
+        }
+        else {
+            disposeStatements.push(`${d.name}();`);
+        }
+    }
+    let disposeCode = disposeStatements.join('\n');
+    if (ts.isBlock(cleanupFn.body)) {
+        edits.push({
+            cleanupBodyEnd: cleanupFn.body.statements[0]?.pos ?? cleanupFn.body.end - 1,
+            cleanupBodyStart: cleanupFn.body.pos + 1,
+            disposeCode,
+            effectsToCapture: ctx.effectsToCapture
+        });
+    }
+    else {
+        edits.push({
+            cleanupBodyEnd: cleanupFn.body.end,
+            cleanupBodyStart: cleanupFn.body.pos,
+            disposeCode: `{ ${disposeCode}\n return ${cleanupFn.body.getText(sourceFile)}; }`,
+            effectsToCapture: ctx.effectsToCapture
+        });
+    }
+}
+function visitBody(ctx, node) {
+    if (ts.isVariableDeclaration(node) &&
+        ts.isIdentifier(node.name) &&
+        node.initializer &&
+        ts.isCallExpression(node.initializer) &&
+        ts.isIdentifier(node.initializer.expression) &&
+        node.initializer.expression.text === 'reactive') {
+        ctx.disposables.push({ name: node.name.text, type: 'reactive' });
+    }
+    if (ts.isCallExpression(node) &&
+        ts.isIdentifier(node.expression) &&
+        node.expression.text === 'effect') {
+        if (ts.isVariableDeclaration(node.parent) && ts.isIdentifier(node.parent.name)) {
+            ctx.disposables.push({ name: node.parent.name.text, type: 'effect' });
+        }
+        else if (ts.isExpressionStatement(node.parent)) {
+            let name = uid('effect');
+            ctx.effectsToCapture.push({
+                end: node.parent.end,
+                name,
+                start: node.parent.pos
+            });
+            ctx.disposables.push({ name, type: 'effect' });
+        }
+    }
+    if (ts.isReturnStatement(node) &&
+        node.expression &&
+        (ts.isArrowFunction(node.expression) || ts.isFunctionExpression(node.expression)) &&
+        node.parent === ctx.parentBody) {
+        ctx.returnStatement = node;
+    }
+    ts.forEachChild(node, n => visitBody(ctx, n));
+}
+function visitMain(ctx, node) {
+    if (ts.isFunctionDeclaration(node) ||
+        ts.isFunctionExpression(node) ||
+        ts.isArrowFunction(node)) {
+        processFunction(node, ctx.sourceFile, ctx.edits);
+    }
+    ts.forEachChild(node, n => visitMain(ctx, n));
+}
+const injectAutoDispose = (sourceFile) => {
+    let code = sourceFile.getFullText(), ctx = {
+        edits: [],
+        sourceFile
+    };
+    visitMain(ctx, sourceFile);
+    if (ctx.edits.length === 0) {
+        return code;
+    }
+    let replacements = [];
+    for (let i = 0, n = ctx.edits.length; i < n; i++) {
+        let edit = ctx.edits[i], effects = edit.effectsToCapture;
+        for (let j = 0, m = effects.length; j < m; j++) {
+            let effect = effects[j], original = code.substring(effect.start, effect.end).trim();
+            replacements.push({
+                end: effect.end,
+                newText: `const ${effect.name} = ${original.replace(TRAILING_SEMICOLON, '')}`,
+                start: effect.start
+            });
+        }
+        replacements.push({
+            end: edit.cleanupBodyEnd,
+            newText: `\n${edit.disposeCode}`,
+            start: edit.cleanupBodyStart
+        });
+    }
+    return applyReplacements(code, replacements);
+};
+export { injectAutoDispose };

package/build/transformer/{core/transforms → transforms}/reactive-array.d.ts

@@ -1,4 +1,4 @@
-import type { Bindings } from '../../../types.js';
+import type { Bindings } from '../../types.js';
 import ts from 'typescript';
 declare const transformReactiveArrays: (sourceFile: ts.SourceFile, bindings: Bindings) => string;
 export { transformReactiveArrays };

package/build/transformer/transforms/reactive-array.js

@@ -0,0 +1,93 @@
+import { applyReplacements } from './utilities.js';
+import ts from 'typescript';
+function getExpressionName(node) {
+    if (ts.isIdentifier(node)) {
+        return node.text;
+    }
+    if (ts.isPropertyAccessExpression(node)) {
+        return getPropertyPath(node);
+    }
+    return null;
+}
+function getPropertyPath(node) {
+    let current = node, parts = [];
+    while (ts.isPropertyAccessExpression(current)) {
+        parts.unshift(current.name.text);
+        current = current.expression;
+    }
+    if (ts.isIdentifier(current)) {
+        parts.unshift(current.text);
+        return parts.join('.');
+    }
+    return null;
+}
+function isAssignmentTarget(node) {
+    let parent = node.parent;
+    if ((ts.isBinaryExpression(parent) && parent.left === node) ||
+        ts.isPostfixUnaryExpression(parent) ||
+        ts.isPrefixUnaryExpression(parent)) {
+        return true;
+    }
+    return false;
+}
+function visit(ctx, node) {
+    if (ts.isVariableDeclaration(node) && ts.isIdentifier(node.name) && node.initializer) {
+        if (ts.isIdentifier(node.initializer) && ctx.bindings.get(node.initializer.text) === 'array') {
+            ctx.bindings.set(node.name.text, 'array');
+        }
+        if (ts.isPropertyAccessExpression(node.initializer)) {
+            let path = getPropertyPath(node.initializer);
+            if (path && ctx.bindings.get(path) === 'array') {
+                ctx.bindings.set(node.name.text, 'array');
+            }
+        }
+    }
+    if ((ts.isFunctionDeclaration(node) || ts.isArrowFunction(node)) && node.parameters) {
+        for (let i = 0, n = node.parameters.length; i < n; i++) {
+            let param = node.parameters[i];
+            if ((ts.isIdentifier(param.name) && param.type) &&
+                ts.isTypeReferenceNode(param.type) &&
+                ts.isIdentifier(param.type.typeName) &&
+                param.type.typeName.text === 'ReactiveArray') {
+                ctx.bindings.set(param.name.text, 'array');
+            }
+        }
+    }
+    if (ts.isPropertyAccessExpression(node) &&
+        node.name.text === 'length' &&
+        !isAssignmentTarget(node)) {
+        let name = getExpressionName(node.expression);
+        if (name && ctx.bindings.get(name) === 'array') {
+            let objText = node.expression.getText(ctx.sourceFile);
+            ctx.replacements.push({
+                end: node.end,
+                newText: `${objText}.$length()`,
+                start: node.pos
+            });
+        }
+    }
+    if (ts.isBinaryExpression(node) &&
+        node.operatorToken.kind === ts.SyntaxKind.EqualsToken &&
+        ts.isElementAccessExpression(node.left)) {
+        let elemAccess = node.left, objName = getExpressionName(elemAccess.expression);
+        if (objName && ctx.bindings.get(objName) === 'array') {
+            let indexText = elemAccess.argumentExpression.getText(ctx.sourceFile), objText = elemAccess.expression.getText(ctx.sourceFile), valueText = node.right.getText(ctx.sourceFile);
+            ctx.replacements.push({
+                end: node.end,
+                newText: `${objText}.$set(${indexText}, ${valueText})`,
+                start: node.pos
+            });
+        }
+    }
+    ts.forEachChild(node, n => visit(ctx, n));
+}
+const transformReactiveArrays = (sourceFile, bindings) => {
+    let code = sourceFile.getFullText(), ctx = {
+        bindings,
+        replacements: [],
+        sourceFile
+    };
+    visit(ctx, sourceFile);
+    return applyReplacements(code, ctx.replacements);
+};
+export { transformReactiveArrays };

package/build/transformer/{core/transforms → transforms}/reactive-object.d.ts

@@ -1,4 +1,4 @@
-import type { Bindings } from '../../../types.js';
+import type { Bindings } from '../../types.js';
 import ts from 'typescript';
 declare const transformReactiveObjects: (sourceFile: ts.SourceFile, bindings: Bindings) => string;
 export { transformReactiveObjects };

package/build/transformer/transforms/reactive-object.js

@@ -0,0 +1,164 @@
+import { uid } from '@esportsplus/typescript/transformer';
+import { addMissingImports, applyReplacements } from './utilities.js';
+import ts from 'typescript';
+const CLASS_NAME_REGEX = /class (\w+)/;
+const EXTRA_IMPORTS = [
+    { module: '@esportsplus/reactivity/constants', specifier: 'REACTIVE_OBJECT' },
+    { module: '@esportsplus/reactivity/reactive/array', specifier: 'ReactiveArray' }
+];
+function analyzeProperty(prop, sourceFile) {
+    if (!ts.isPropertyAssignment(prop)) {
+        return null;
+    }
+    let key;
+    if (ts.isIdentifier(prop.name) || ts.isStringLiteral(prop.name)) {
+        key = prop.name.text;
+    }
+    else {
+        return null;
+    }
+    let value = prop.initializer, valueText = value.getText(sourceFile);
+    if (ts.isArrowFunction(value) || ts.isFunctionExpression(value)) {
+        return { key, type: 'computed', valueText };
+    }
+    if (ts.isArrayLiteralExpression(value)) {
+        return { key, type: 'array', valueText };
+    }
+    return { key, type: 'signal', valueText };
+}
+function buildClassCode(className, properties) {
+    let accessors = [], disposeStatements = [], fields = [];
+    fields.push(`[REACTIVE_OBJECT] = true;`);
+    for (let i = 0, n = properties.length; i < n; i++) {
+        let { key, type, valueText } = properties[i];
+        if (type === 'signal') {
+            let param = uid('v');
+            fields.push(`#${key} = signal(${valueText});`);
+            accessors.push(`get ${key}() { return read(this.#${key}); }`);
+            accessors.push(`set ${key}(${param}) { set(this.#${key}, ${param}); }`);
+        }
+        else if (type === 'array') {
+            let elements = valueText.slice(1, -1);
+            fields.push(`${key} = new ReactiveArray(${elements});`);
+            disposeStatements.push(`this.${key}.dispose();`);
+        }
+        else if (type === 'computed') {
+            fields.push(`#${key}: Computed<unknown> | null = null;`);
+            accessors.push(`get ${key}() { return read(this.#${key} ??= computed(${valueText})); }`);
+            disposeStatements.push(`if (this.#${key}) dispose(this.#${key});`);
+        }
+    }
+    let disposeBody = disposeStatements.length > 0 ? disposeStatements.join('\n') : '';
+    return `
+        class ${className} {
+            ${fields.join('\n')}
+            ${accessors.join('\n')}
+
+            dispose() {
+                ${disposeBody}
+            }
+        }
+    `;
+}
+function visit(ctx, node) {
+    if (ts.isImportDeclaration(node)) {
+        ctx.lastImportEnd = node.end;
+        if (ts.isStringLiteral(node.moduleSpecifier) &&
+            node.moduleSpecifier.text.includes('@esportsplus/reactivity')) {
+            let clause = node.importClause;
+            if (clause?.namedBindings && ts.isNamedImports(clause.namedBindings)) {
+                let elements = clause.namedBindings.elements;
+                for (let i = 0, n = elements.length; i < n; i++) {
+                    if (elements[i].name.text === 'reactive') {
+                        ctx.hasReactiveImport = true;
+                        break;
+                    }
+                }
+            }
+        }
+    }
+    if (ctx.hasReactiveImport &&
+        ts.isCallExpression(node) &&
+        ts.isIdentifier(node.expression) &&
+        node.expression.text === 'reactive') {
+        let arg = node.arguments[0];
+        if (arg && ts.isObjectLiteralExpression(arg)) {
+            let varName = null;
+            if (ts.isVariableDeclaration(node.parent) && ts.isIdentifier(node.parent.name)) {
+                varName = node.parent.name.text;
+                ctx.bindings.set(varName, 'object');
+            }
+            let needsImports = new Set(), properties = [];
+            needsImports.add('REACTIVE_OBJECT');
+            let props = arg.properties;
+            for (let i = 0, n = props.length; i < n; i++) {
+                let prop = props[i];
+                if (ts.isSpreadAssignment(prop)) {
+                    ts.forEachChild(node, n => visit(ctx, n));
+                    return;
+                }
+                let analyzed = analyzeProperty(prop, ctx.sourceFile);
+                if (!analyzed) {
+                    ts.forEachChild(node, n => visit(ctx, n));
+                    return;
+                }
+                properties.push(analyzed);
+                if (analyzed.type === 'signal') {
+                    needsImports.add('read');
+                    needsImports.add('set');
+                    needsImports.add('signal');
+                }
+                else if (analyzed.type === 'array') {
+                    needsImports.add('ReactiveArray');
+                    if (varName) {
+                        ctx.bindings.set(`${varName}.${analyzed.key}`, 'array');
+                    }
+                }
+                else if (analyzed.type === 'computed') {
+                    needsImports.add('computed');
+                    needsImports.add('dispose');
+                    needsImports.add('read');
+                }
+            }
+            needsImports.forEach(imp => ctx.allNeededImports.add(imp));
+            ctx.calls.push({
+                end: node.end,
+                generatedClass: buildClassCode(uid('ReactiveObject'), properties),
+                needsImports,
+                start: node.pos,
+                varName
+            });
+        }
+    }
+    ts.forEachChild(node, n => visit(ctx, n));
+}
+const transformReactiveObjects = (sourceFile, bindings) => {
+    let code = sourceFile.getFullText(), ctx = {
+        allNeededImports: new Set(),
+        bindings,
+        calls: [],
+        hasReactiveImport: false,
+        lastImportEnd: 0,
+        sourceFile
+    };
+    visit(ctx, sourceFile);
+    if (ctx.calls.length === 0) {
+        return code;
+    }
+    let replacements = [];
+    replacements.push({
+        end: ctx.lastImportEnd,
+        newText: code.substring(0, ctx.lastImportEnd) + '\n' + ctx.calls.map(c => c.generatedClass).join('\n') + '\n',
+        start: 0
+    });
+    for (let i = 0, n = ctx.calls.length; i < n; i++) {
+        let call = ctx.calls[i], classMatch = call.generatedClass.match(CLASS_NAME_REGEX);
+        replacements.push({
+            end: call.end,
+            newText: ` new ${classMatch ? classMatch[1] : 'ReactiveObject'}()`,
+            start: call.start
+        });
+    }
+    return addMissingImports(applyReplacements(code, replacements), ctx.allNeededImports, EXTRA_IMPORTS);
+};
+export { transformReactiveObjects };

package/build/transformer/{core/transforms → transforms}/reactive-primitives.d.ts

@@ -1,4 +1,4 @@
-import type { Bindings } from '../../../types.js';
+import type { Bindings } from '../../types.js';
 import ts from 'typescript';
 declare const transformReactivePrimitives: (sourceFile: ts.SourceFile, bindings: Bindings) => string;
 export { transformReactivePrimitives };