@esportsplus/reactivity 0.23.0 → 0.23.2

package/src/transformer/plugins/vite.ts

@@ -1,13 +1,10 @@
-import { mightNeedTransform, transform } from '~/transformer/core';
+import { TRANSFORM_PATTERN } from '@esportsplus/typescript/transformer';
+import { mightNeedTransform, transform } from '~/transformer';
 import type { Plugin } from 'vite';
-import type { TransformOptions } from '~/types';
 import ts from 'typescript';
 
 
-const TRANSFORM_PATTERN = /\.[tj]sx?$/;
-
-
-export default (options?: TransformOptions): Plugin => {
+export default (): Plugin => {
     return {
         enforce: 'pre',
         name: '@esportsplus/reactivity/plugin-vite',
@@ -23,7 +20,7 @@ export default (options?: TransformOptions): Plugin => {
 
             try {
                 let sourceFile = ts.createSourceFile(id, code, ts.ScriptTarget.Latest, true),
-                    result = transform(sourceFile, options);
+                    result = transform(sourceFile);
 
                 if (!result.transformed) {
                     return null;
@@ -38,4 +35,3 @@ export default (options?: TransformOptions): Plugin => {
         }
     };
 };
-export type { TransformOptions as PluginOptions };

package/src/transformer/transforms/array.ts

@@ -0,0 +1,143 @@
+import type { Bindings } from '~/types';
+import { applyReplacements, Replacement } from './utilities';
+import ts from 'typescript';
+
+
+interface TransformContext {
+    bindings: Bindings;
+    replacements: Replacement[];
+    sourceFile: ts.SourceFile;
+}
+
+
+function getExpressionName(node: ts.Expression): string | null {
+    if (ts.isIdentifier(node)) {
+        return node.text;
+    }
+
+    if (ts.isPropertyAccessExpression(node)) {
+        return getPropertyPath(node);
+    }
+
+    return null;
+}
+
+function getPropertyPath(node: ts.PropertyAccessExpression): string | null {
+    let current: ts.Node = node,
+        parts: string[] = [];
+
+    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: ts.Node): boolean {
+    let parent = node.parent;
+
+    if (
+        (ts.isBinaryExpression(parent) && parent.left === node) ||
+        ts.isPostfixUnaryExpression(parent) ||
+        ts.isPrefixUnaryExpression(parent)
+    ) {
+        return true;
+    }
+
+    return false;
+}
+
+function visit(ctx: TransformContext, node: ts.Node): void {
+    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: ts.SourceFile, bindings: Bindings): string => {
+    let code = sourceFile.getFullText(),
+        ctx: TransformContext = {
+            bindings,
+            replacements: [],
+            sourceFile
+        };
+
+    visit(ctx, sourceFile);
+
+    return applyReplacements(code, ctx.replacements);
+};
+
+
+export { transformReactiveArrays };

package/src/transformer/transforms/object.ts

@@ -0,0 +1,251 @@
+import { uid } from '@esportsplus/typescript/transformer';
+import type { Bindings } from '~/types';
+import { addMissingImports, applyReplacements, ExtraImport, Replacement } from './utilities';
+import ts from 'typescript';
+
+
+const CLASS_NAME_REGEX = /class (\w+)/;
+
+const EXTRA_IMPORTS: ExtraImport[] = [
+    { module: '@esportsplus/reactivity/constants', specifier: 'REACTIVE_OBJECT' },
+    { module: '@esportsplus/reactivity/reactive/array', specifier: 'ReactiveArray' }
+];
+
+
+interface AnalyzedProperty {
+    key: string;
+    type: 'array' | 'computed' | 'signal';
+    valueText: string;
+}
+
+interface ReactiveObjectCall {
+    end: number;
+    generatedClass: string;
+    needsImports: Set<string>;
+    start: number;
+    varName: string | null;
+}
+
+interface TransformContext {
+    allNeededImports: Set<string>;
+    bindings: Bindings;
+    calls: ReactiveObjectCall[];
+    hasReactiveImport: boolean;
+    lastImportEnd: number;
+    sourceFile: ts.SourceFile;
+}
+
+
+function analyzeProperty(prop: ts.ObjectLiteralElementLike, sourceFile: ts.SourceFile): AnalyzedProperty | null {
+    if (!ts.isPropertyAssignment(prop)) {
+        return null;
+    }
+
+    let key: string;
+
+    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: string, properties: AnalyzedProperty[]): string {
+    let accessors: string[] = [],
+        disposeStatements: string[] = [],
+        fields: string[] = [];
+
+    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});`);
+        }
+    }
+
+    return `
+        class ${className} {
+            ${fields.join('\n')}
+            ${accessors.join('\n')}
+
+            dispose() {
+                ${disposeStatements.length > 0 ? disposeStatements.join('\n') : ''}
+            }
+        }
+    `;
+}
+
+function visit(ctx: TransformContext, node: ts.Node): void {
+    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: string | null = 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<string>(),
+                properties: AnalyzedProperty[] = [];
+
+            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: ts.SourceFile, bindings: Bindings): string => {
+    let code = sourceFile.getFullText(),
+        ctx: TransformContext = {
+            allNeededImports: new Set<string>(),
+            bindings,
+            calls: [],
+            hasReactiveImport: false,
+            lastImportEnd: 0,
+            sourceFile
+        };
+
+    visit(ctx, sourceFile);
+
+    if (ctx.calls.length === 0) {
+        return code;
+    }
+
+    let replacements: Replacement[] = [];
+
+    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 };