@expo/entity-codemod 0.54.0 → 0.57.0

package/build/src/transforms/v0.55.0-v0.56.0.d.ts

@@ -0,0 +1,2 @@
+import { API, FileInfo, Options } from 'jscodeshift';
+export default function transformer(file: FileInfo, api: API, _options: Options): string;

package/build/src/transforms/v0.55.0-v0.56.0.js

@@ -0,0 +1,144 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.default = transformer;
+const KNEX_SPECIFIC_METHODS = [
+    'loadFirstByFieldEqualityConjunctionAsync',
+    'loadManyByFieldEqualityConjunctionAsync',
+    'loadManyByRawWhereClauseAsync',
+];
+function isKnexSpecificMethodUsed(j, node) {
+    // Check if this loader call is followed by a knex-specific method
+    // We need to traverse the AST to find usages of the loader result
+    const parent = node.parent;
+    // Check if the loader call is directly chained with a knex method
+    if (parent?.value.type === 'MemberExpression' && parent.value.object === node.value) {
+        const grandParent = parent.parent;
+        if (grandParent?.value.type === 'CallExpression' && grandParent.value.callee === parent.value) {
+            if (parent.value.property.type === 'Identifier' &&
+                KNEX_SPECIFIC_METHODS.includes(parent.value.property.name)) {
+                return true;
+            }
+        }
+    }
+    // Check if the loader is assigned to a variable and then used with knex methods
+    if (parent?.value.type === 'VariableDeclarator' && parent.value.init === node.value) {
+        const variableName = parent.value.id.name;
+        const scope = parent.scope;
+        // Find all references to this variable in the same scope
+        const references = j(scope.path)
+            .find(j.Identifier, { name: variableName })
+            .filter((path) => {
+            // Check if this identifier is used as object in member expression
+            const parentNode = path.parent.value;
+            if (parentNode.type === 'MemberExpression' && parentNode.object === path.value) {
+                const prop = parentNode.property;
+                if (prop.type === 'Identifier' && KNEX_SPECIFIC_METHODS.includes(prop.name)) {
+                    return true;
+                }
+            }
+            return false;
+        });
+        if (references.size() > 0) {
+            return true;
+        }
+    }
+    // Check await expressions
+    if (parent?.value.type === 'AwaitExpression' && parent.value.argument === node.value) {
+        const awaitParent = parent.parent;
+        if (awaitParent?.value.type === 'VariableDeclarator') {
+            const variableName = awaitParent.value.id.name;
+            const scope = awaitParent.scope;
+            // Find all references to this variable in the same scope
+            const references = j(scope.path)
+                .find(j.Identifier, { name: variableName })
+                .filter((path) => {
+                const parentNode = path.parent.value;
+                if (parentNode.type === 'MemberExpression' && parentNode.object === path.value) {
+                    const prop = parentNode.property;
+                    if (prop.type === 'Identifier' && KNEX_SPECIFIC_METHODS.includes(prop.name)) {
+                        return true;
+                    }
+                }
+                return false;
+            });
+            if (references.size() > 0) {
+                return true;
+            }
+        }
+    }
+    return false;
+}
+function transformLoaderToKnexLoader(j, root) {
+    // Find all entity expressions of the form `Entity.loader(viewerContext)`
+    root
+        .find(j.CallExpression, {
+        callee: {
+            type: 'MemberExpression',
+            property: {
+                name: 'loader',
+            },
+        },
+    })
+        .forEach((path) => {
+        const loaderCallExpression = path.node; // Entity.loader(viewerContext)
+        const loaderCallee = loaderCallExpression.callee; // Entity.loader
+        if (loaderCallee.type !== 'MemberExpression') {
+            return;
+        }
+        // Make sure this is a static method call on an entity (not on an instance)
+        // Typically entity names start with uppercase letter
+        if (loaderCallee.object.type === 'Identifier') {
+            const firstChar = loaderCallee.object.name[0];
+            if (firstChar === firstChar?.toUpperCase()) {
+                // Check if this loader uses knex-specific methods
+                if (isKnexSpecificMethodUsed(j, path)) {
+                    // Rename loader to knexLoader
+                    if (loaderCallee.property.type === 'Identifier') {
+                        loaderCallee.property.name = 'knexLoader';
+                    }
+                }
+            }
+        }
+    });
+}
+function transformLoaderWithAuthorizationResultsToKnexLoaderWithAuthorizationResults(j, root) {
+    // Find all entity expressions of the form `Entity.loaderWithAuthorizationResults(viewerContext)`
+    root
+        .find(j.CallExpression, {
+        callee: {
+            type: 'MemberExpression',
+            property: {
+                name: 'loaderWithAuthorizationResults',
+            },
+        },
+    })
+        .forEach((path) => {
+        const loaderCallExpression = path.node; // Entity.loaderWithAuthorizationResults(viewerContext)
+        const loaderCallee = loaderCallExpression.callee; // Entity.loaderWithAuthorizationResults
+        if (loaderCallee.type !== 'MemberExpression') {
+            return;
+        }
+        // Make sure this is a static method call on an entity (not on an instance)
+        // Typically entity names start with uppercase letter
+        if (loaderCallee.object.type === 'Identifier') {
+            const firstChar = loaderCallee.object.name[0];
+            if (firstChar === firstChar?.toUpperCase()) {
+                // Check if this loader uses knex-specific methods
+                if (isKnexSpecificMethodUsed(j, path)) {
+                    // Rename loaderWithAuthorizationResults to knexLoaderWithAuthorizationResults
+                    if (loaderCallee.property.type === 'Identifier') {
+                        loaderCallee.property.name = 'knexLoaderWithAuthorizationResults';
+                    }
+                }
+            }
+        }
+    });
+}
+function transformer(file, api, _options) {
+    const j = api.jscodeshift;
+    const root = j.withParser('ts')(file.source);
+    transformLoaderToKnexLoader(j, root);
+    transformLoaderWithAuthorizationResultsToKnexLoaderWithAuthorizationResults(j, root);
+    return root.toSource();
+}
+//# sourceMappingURL=v0.55.0-v0.56.0.js.map

package/build/src/transforms/v0.55.0-v0.56.0.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"v0.55.0-v0.56.0.js","sourceRoot":"","sources":["../../../src/transforms/v0.55.0-v0.56.0.ts"],"names":[],"mappings":";;AA4JA,8BAQC;AAlKD,MAAM,qBAAqB,GAAG;IAC5B,0CAA0C;IAC1C,yCAAyC;IACzC,+BAA+B;CAChC,CAAC;AAEF,SAAS,wBAAwB,CAAC,CAAqB,EAAE,IAAS;IAChE,kEAAkE;IAClE,kEAAkE;IAClE,MAAM,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;IAE3B,kEAAkE;IAClE,IAAI,MAAM,EAAE,KAAK,CAAC,IAAI,KAAK,kBAAkB,IAAI,MAAM,CAAC,KAAK,CAAC,MAAM,KAAK,IAAI,CAAC,KAAK,EAAE,CAAC;QACpF,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,CAAC;QAClC,IAAI,WAAW,EAAE,KAAK,CAAC,IAAI,KAAK,gBAAgB,IAAI,WAAW,CAAC,KAAK,CAAC,MAAM,KAAK,MAAM,CAAC,KAAK,EAAE,CAAC;YAC9F,IACE,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,KAAK,YAAY;gBAC3C,qBAAqB,CAAC,QAAQ,CAAC,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,EAC1D,CAAC;gBACD,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;IACH,CAAC;IAED,gFAAgF;IAChF,IAAI,MAAM,EAAE,KAAK,CAAC,IAAI,KAAK,oBAAoB,IAAI,MAAM,CAAC,KAAK,CAAC,IAAI,KAAK,IAAI,CAAC,KAAK,EAAE,CAAC;QACpF,MAAM,YAAY,GAAG,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC;QAC1C,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC;QAE3B,yDAAyD;QACzD,MAAM,UAAU,GAAG,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC;aAC7B,IAAI,CAAC,CAAC,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;aAC1C,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE;YACf,kEAAkE;YAClE,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;YACrC,IAAI,UAAU,CAAC,IAAI,KAAK,kBAAkB,IAAI,UAAU,CAAC,MAAM,KAAK,IAAI,CAAC,KAAK,EAAE,CAAC;gBAC/E,MAAM,IAAI,GAAG,UAAU,CAAC,QAAQ,CAAC;gBACjC,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,IAAI,qBAAqB,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;oBAC5E,OAAO,IAAI,CAAC;gBACd,CAAC;YACH,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC,CAAC,CAAC;QAEL,IAAI,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,CAAC;YAC1B,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED,0BAA0B;IAC1B,IAAI,MAAM,EAAE,KAAK,CAAC,IAAI,KAAK,iBAAiB,IAAI,MAAM,CAAC,KAAK,CAAC,QAAQ,KAAK,IAAI,CAAC,KAAK,EAAE,CAAC;QACrF,MAAM,WAAW,GAAG,MAAM,CAAC,MAAM,CAAC;QAClC,IAAI,WAAW,EAAE,KAAK,CAAC,IAAI,KAAK,oBAAoB,EAAE,CAAC;YACrD,MAAM,YAAY,GAAG,WAAW,CAAC,KAAK,CAAC,EAAE,CAAC,IAAI,CAAC;YAC/C,MAAM,KAAK,GAAG,WAAW,CAAC,KAAK,CAAC;YAEhC,yDAAyD;YACzD,MAAM,UAAU,GAAG,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC;iBAC7B,IAAI,CAAC,CAAC,CAAC,UAAU,EAAE,EAAE,IAAI,EAAE,YAAY,EAAE,CAAC;iBAC1C,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE;gBACf,MAAM,UAAU,GAAG,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC;gBACrC,IAAI,UAAU,CAAC,IAAI,KAAK,kBAAkB,IAAI,UAAU,CAAC,MAAM,KAAK,IAAI,CAAC,KAAK,EAAE,CAAC;oBAC/E,MAAM,IAAI,GAAG,UAAU,CAAC,QAAQ,CAAC;oBACjC,IAAI,IAAI,CAAC,IAAI,KAAK,YAAY,IAAI,qBAAqB,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC;wBAC5E,OAAO,IAAI,CAAC;oBACd,CAAC;gBACH,CAAC;gBACD,OAAO,KAAK,CAAC;YACf,CAAC,CAAC,CAAC;YAEL,IAAI,UAAU,CAAC,IAAI,EAAE,GAAG,CAAC,EAAE,CAAC;gBAC1B,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,2BAA2B,CAAC,CAAqB,EAAE,IAAqB;IAC/E,yEAAyE;IACzE,IAAI;SACD,IAAI,CAAC,CAAC,CAAC,cAAc,EAAE;QACtB,MAAM,EAAE;YACN,IAAI,EAAE,kBAAkB;YACxB,QAAQ,EAAE;gBACR,IAAI,EAAE,QAAQ;aACf;SACF;KACF,CAAC;SACD,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;QAChB,MAAM,oBAAoB,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,+BAA+B;QACvE,MAAM,YAAY,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC,gBAAgB;QAElE,IAAI,YAAY,CAAC,IAAI,KAAK,kBAAkB,EAAE,CAAC;YAC7C,OAAO;QACT,CAAC;QAED,2EAA2E;QAC3E,qDAAqD;QACrD,IAAI,YAAY,CAAC,MAAM,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YAC9C,MAAM,SAAS,GAAG,YAAY,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC9C,IAAI,SAAS,KAAK,SAAS,EAAE,WAAW,EAAE,EAAE,CAAC;gBAC3C,kDAAkD;gBAClD,IAAI,wBAAwB,CAAC,CAAC,EAAE,IAAI,CAAC,EAAE,CAAC;oBACtC,8BAA8B;oBAC9B,IAAI,YAAY,CAAC,QAAQ,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;wBAChD,YAAY,CAAC,QAAQ,CAAC,IAAI,GAAG,YAAY,CAAC;oBAC5C,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;AACP,CAAC;AAED,SAAS,2EAA2E,CAClF,CAAqB,EACrB,IAAqB;IAErB,iGAAiG;IACjG,IAAI;SACD,IAAI,CAAC,CAAC,CAAC,cAAc,EAAE;QACtB,MAAM,EAAE;YACN,IAAI,EAAE,kBAAkB;YACxB,QAAQ,EAAE;gBACR,IAAI,EAAE,gCAAgC;aACvC;SACF;KACF,CAAC;SACD,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;QAChB,MAAM,oBAAoB,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,uDAAuD;QAC/F,MAAM,YAAY,GAAG,oBAAoB,CAAC,MAAM,CAAC,CAAC,wCAAwC;QAE1F,IAAI,YAAY,CAAC,IAAI,KAAK,kBAAkB,EAAE,CAAC;YAC7C,OAAO;QACT,CAAC;QAED,2EAA2E;QAC3E,qDAAqD;QACrD,IAAI,YAAY,CAAC,MAAM,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YAC9C,MAAM,SAAS,GAAG,YAAY,CAAC,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;YAC9C,IAAI,SAAS,KAAK,SAAS,EAAE,WAAW,EAAE,EAAE,CAAC;gBAC3C,kDAAkD;gBAClD,IAAI,wBAAwB,CAAC,CAAC,EAAE,IAAI,CAAC,EAAE,CAAC;oBACtC,8EAA8E;oBAC9E,IAAI,YAAY,CAAC,QAAQ,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;wBAChD,YAAY,CAAC,QAAQ,CAAC,IAAI,GAAG,oCAAoC,CAAC;oBACpE,CAAC;gBACH,CAAC;YACH,CAAC;QACH,CAAC;IACH,CAAC,CAAC,CAAC;AACP,CAAC;AAED,SAAwB,WAAW,CAAC,IAAc,EAAE,GAAQ,EAAE,QAAiB;IAC7E,MAAM,CAAC,GAAG,GAAG,CAAC,WAAW,CAAC;IAC1B,MAAM,IAAI,GAAG,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IAE7C,2BAA2B,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IACrC,2EAA2E,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IAErF,OAAO,IAAI,CAAC,QAAQ,EAAE,CAAC;AACzB,CAAC"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@expo/entity-codemod",
-  "version": "0.54.0",
+  "version": "0.57.0",
   "description": "jscodeshift codemods for @expo/entity upgrades",
   "files": [
     "build",
@@ -29,9 +29,9 @@
   "devDependencies": {
     "@jest/globals": "30.2.0",
     "@types/jscodeshift": "17.3.0",
-    "@types/node": "24.10.9",
+    "@types/node": "24.10.13",
     "jscodeshift": "17.3.0",
     "typescript": "5.9.3"
   },
-  "gitHead": "283c02eedbd4f5721afad96f31ada85bf92afb41"
+  "gitHead": "e876cfb27bb9b0004d81b40c9067481e3e0c2beb"
 }

package/src/transforms/__testfixtures__/v0.55.0-v0.56.0/test1.input.ts

@@ -0,0 +1,37 @@
+import { ViewerContext } from '@expo/entity';
+import { UserEntity } from './entities/UserEntity';
+import { PostEntity } from './entities/PostEntity';
+
+async function loadUser(viewerContext: ViewerContext) {
+  // Basic loader calls - only transformed when using knex-specific methods
+  const userLoader = UserEntity.loader(viewerContext);
+  const postLoader = PostEntity.loader(viewerContext);
+
+  // These use knex-specific methods, so they should be transformed
+  const posts = await postLoader.loadManyByFieldEqualityConjunctionAsync([
+    { fieldName: 'status', fieldValue: 'published' }
+  ]);
+  const firstPost = await postLoader.loadFirstByFieldEqualityConjunctionAsync([
+    { fieldName: 'id', fieldValue: '123' }
+  ]);
+
+  // Loader with authorization results - only transformed when using knex methods
+  const userLoaderWithAuth = UserEntity.loaderWithAuthorizationResults(viewerContext);
+  const rawResults = await userLoaderWithAuth.loadManyByRawWhereClauseAsync('age > ?', [18]);
+
+  // Loader that doesn't use knex methods - should NOT be transformed
+  const standardLoader = PostEntity.loader(viewerContext);
+  const post = await standardLoader.loadByIDAsync('456');
+
+  // Should not transform instance methods or other properties
+  const user = await userLoader.loadByIDAsync('123');
+  const userLoadMethod = user.loader; // This should not be transformed
+
+  // Should not transform lowercase object methods
+  const customLoader = {
+    loader: (ctx: any) => ctx,
+  };
+  customLoader.loader(viewerContext);
+
+  return user;
+}

package/src/transforms/__testfixtures__/v0.55.0-v0.56.0/test1.output.ts

@@ -0,0 +1,37 @@
+import { ViewerContext } from '@expo/entity';
+import { UserEntity } from './entities/UserEntity';
+import { PostEntity } from './entities/PostEntity';
+
+async function loadUser(viewerContext: ViewerContext) {
+  // Basic loader calls - only transformed when using knex-specific methods
+  const userLoader = UserEntity.loader(viewerContext);
+  const postLoader = PostEntity.knexLoader(viewerContext);
+
+  // These use knex-specific methods, so they should be transformed
+  const posts = await postLoader.loadManyByFieldEqualityConjunctionAsync([
+    { fieldName: 'status', fieldValue: 'published' }
+  ]);
+  const firstPost = await postLoader.loadFirstByFieldEqualityConjunctionAsync([
+    { fieldName: 'id', fieldValue: '123' }
+  ]);
+
+  // Loader with authorization results - only transformed when using knex methods
+  const userLoaderWithAuth = UserEntity.knexLoaderWithAuthorizationResults(viewerContext);
+  const rawResults = await userLoaderWithAuth.loadManyByRawWhereClauseAsync('age > ?', [18]);
+
+  // Loader that doesn't use knex methods - should NOT be transformed
+  const standardLoader = PostEntity.loader(viewerContext);
+  const post = await standardLoader.loadByIDAsync('456');
+
+  // Should not transform instance methods or other properties
+  const user = await userLoader.loadByIDAsync('123');
+  const userLoadMethod = user.loader; // This should not be transformed
+
+  // Should not transform lowercase object methods
+  const customLoader = {
+    loader: (ctx: any) => ctx,
+  };
+  customLoader.loader(viewerContext);
+
+  return user;
+}

package/src/transforms/__testfixtures__/v0.55.0-v0.56.0/test2.input.ts

@@ -0,0 +1,33 @@
+import { ViewerContext } from '@expo/entity';
+import { CommentEntity } from './entities/CommentEntity';
+
+// Chained calls
+const loadComments = async (viewerContext: ViewerContext) => {
+  // Direct chaining with knex-specific method
+  const comments = await CommentEntity.loader(viewerContext)
+    .loadManyByFieldEqualityConjunctionAsync([
+      { fieldName: 'postId', fieldValue: '123' }
+    ]);
+
+  // Direct chaining with regular method - should NOT be transformed
+  const singleComment = await CommentEntity
+    .loader(viewerContext)
+    .loadByIDAsync('456');
+
+  // With authorization results and knex method
+  const commentsWithAuth = await CommentEntity
+    .loaderWithAuthorizationResults(viewerContext)
+    .loadManyByRawWhereClauseAsync('postId = ?', ['456']);
+
+  // Edge cases - these should NOT be transformed
+  const anotherEntity = {
+    loader: (ctx: any) => ctx, // This is not an entity class
+  };
+  anotherEntity.loader(viewerContext); // Should NOT be transformed (lowercase object)
+
+  // Complex chaining with regular method - should NOT be transformed
+  return CommentEntity
+    .loader(viewerContext)
+    .withAuthenticationResults()
+    .loadByIDAsync('789');
+};

package/src/transforms/__testfixtures__/v0.55.0-v0.56.0/test2.output.ts

@@ -0,0 +1,33 @@
+import { ViewerContext } from '@expo/entity';
+import { CommentEntity } from './entities/CommentEntity';
+
+// Chained calls
+const loadComments = async (viewerContext: ViewerContext) => {
+  // Direct chaining with knex-specific method
+  const comments = await CommentEntity.knexLoader(viewerContext)
+    .loadManyByFieldEqualityConjunctionAsync([
+      { fieldName: 'postId', fieldValue: '123' }
+    ]);
+
+  // Direct chaining with regular method - should NOT be transformed
+  const singleComment = await CommentEntity
+    .loader(viewerContext)
+    .loadByIDAsync('456');
+
+  // With authorization results and knex method
+  const commentsWithAuth = await CommentEntity
+    .knexLoaderWithAuthorizationResults(viewerContext)
+    .loadManyByRawWhereClauseAsync('postId = ?', ['456']);
+
+  // Edge cases - these should NOT be transformed
+  const anotherEntity = {
+    loader: (ctx: any) => ctx, // This is not an entity class
+  };
+  anotherEntity.loader(viewerContext); // Should NOT be transformed (lowercase object)
+
+  // Complex chaining with regular method - should NOT be transformed
+  return CommentEntity
+    .loader(viewerContext)
+    .withAuthenticationResults()
+    .loadByIDAsync('789');
+};

package/src/transforms/__tests__/v0.55.0-v0.56.0-test.ts

@@ -0,0 +1,17 @@
+import { jest } from '@jest/globals';
+import { readdirSync } from 'fs';
+import { join } from 'path';
+
+jest.autoMockOff();
+const defineTest = require('jscodeshift/dist/testUtils').defineTest;
+
+const fixtureDir = 'v0.55.0-v0.56.0';
+const fixtureDirPath = join(__dirname, '..', '__testfixtures__', fixtureDir);
+const fixtures = readdirSync(fixtureDirPath)
+  .filter((file) => file.endsWith('.input.ts'))
+  .map((file) => file.replace('.input.ts', ''));
+
+for (const fixture of fixtures) {
+  const prefix = `${fixtureDir}/${fixture}`;
+  defineTest(__dirname, 'v0.55.0-v0.56.0', null, prefix, { parser: 'ts' });
+}

package/src/transforms/v0.55.0-v0.56.0.ts

@@ -0,0 +1,165 @@
+import { API, Collection, FileInfo, Options } from 'jscodeshift';
+
+const KNEX_SPECIFIC_METHODS = [
+  'loadFirstByFieldEqualityConjunctionAsync',
+  'loadManyByFieldEqualityConjunctionAsync',
+  'loadManyByRawWhereClauseAsync',
+];
+
+function isKnexSpecificMethodUsed(j: API['jscodeshift'], node: any): boolean {
+  // Check if this loader call is followed by a knex-specific method
+  // We need to traverse the AST to find usages of the loader result
+  const parent = node.parent;
+
+  // Check if the loader call is directly chained with a knex method
+  if (parent?.value.type === 'MemberExpression' && parent.value.object === node.value) {
+    const grandParent = parent.parent;
+    if (grandParent?.value.type === 'CallExpression' && grandParent.value.callee === parent.value) {
+      if (
+        parent.value.property.type === 'Identifier' &&
+        KNEX_SPECIFIC_METHODS.includes(parent.value.property.name)
+      ) {
+        return true;
+      }
+    }
+  }
+
+  // Check if the loader is assigned to a variable and then used with knex methods
+  if (parent?.value.type === 'VariableDeclarator' && parent.value.init === node.value) {
+    const variableName = parent.value.id.name;
+    const scope = parent.scope;
+
+    // Find all references to this variable in the same scope
+    const references = j(scope.path)
+      .find(j.Identifier, { name: variableName })
+      .filter((path) => {
+        // Check if this identifier is used as object in member expression
+        const parentNode = path.parent.value;
+        if (parentNode.type === 'MemberExpression' && parentNode.object === path.value) {
+          const prop = parentNode.property;
+          if (prop.type === 'Identifier' && KNEX_SPECIFIC_METHODS.includes(prop.name)) {
+            return true;
+          }
+        }
+        return false;
+      });
+
+    if (references.size() > 0) {
+      return true;
+    }
+  }
+
+  // Check await expressions
+  if (parent?.value.type === 'AwaitExpression' && parent.value.argument === node.value) {
+    const awaitParent = parent.parent;
+    if (awaitParent?.value.type === 'VariableDeclarator') {
+      const variableName = awaitParent.value.id.name;
+      const scope = awaitParent.scope;
+
+      // Find all references to this variable in the same scope
+      const references = j(scope.path)
+        .find(j.Identifier, { name: variableName })
+        .filter((path) => {
+          const parentNode = path.parent.value;
+          if (parentNode.type === 'MemberExpression' && parentNode.object === path.value) {
+            const prop = parentNode.property;
+            if (prop.type === 'Identifier' && KNEX_SPECIFIC_METHODS.includes(prop.name)) {
+              return true;
+            }
+          }
+          return false;
+        });
+
+      if (references.size() > 0) {
+        return true;
+      }
+    }
+  }
+
+  return false;
+}
+
+function transformLoaderToKnexLoader(j: API['jscodeshift'], root: Collection<any>): void {
+  // Find all entity expressions of the form `Entity.loader(viewerContext)`
+  root
+    .find(j.CallExpression, {
+      callee: {
+        type: 'MemberExpression',
+        property: {
+          name: 'loader',
+        },
+      },
+    })
+    .forEach((path) => {
+      const loaderCallExpression = path.node; // Entity.loader(viewerContext)
+      const loaderCallee = loaderCallExpression.callee; // Entity.loader
+
+      if (loaderCallee.type !== 'MemberExpression') {
+        return;
+      }
+
+      // Make sure this is a static method call on an entity (not on an instance)
+      // Typically entity names start with uppercase letter
+      if (loaderCallee.object.type === 'Identifier') {
+        const firstChar = loaderCallee.object.name[0];
+        if (firstChar === firstChar?.toUpperCase()) {
+          // Check if this loader uses knex-specific methods
+          if (isKnexSpecificMethodUsed(j, path)) {
+            // Rename loader to knexLoader
+            if (loaderCallee.property.type === 'Identifier') {
+              loaderCallee.property.name = 'knexLoader';
+            }
+          }
+        }
+      }
+    });
+}
+
+function transformLoaderWithAuthorizationResultsToKnexLoaderWithAuthorizationResults(
+  j: API['jscodeshift'],
+  root: Collection<any>,
+): void {
+  // Find all entity expressions of the form `Entity.loaderWithAuthorizationResults(viewerContext)`
+  root
+    .find(j.CallExpression, {
+      callee: {
+        type: 'MemberExpression',
+        property: {
+          name: 'loaderWithAuthorizationResults',
+        },
+      },
+    })
+    .forEach((path) => {
+      const loaderCallExpression = path.node; // Entity.loaderWithAuthorizationResults(viewerContext)
+      const loaderCallee = loaderCallExpression.callee; // Entity.loaderWithAuthorizationResults
+
+      if (loaderCallee.type !== 'MemberExpression') {
+        return;
+      }
+
+      // Make sure this is a static method call on an entity (not on an instance)
+      // Typically entity names start with uppercase letter
+      if (loaderCallee.object.type === 'Identifier') {
+        const firstChar = loaderCallee.object.name[0];
+        if (firstChar === firstChar?.toUpperCase()) {
+          // Check if this loader uses knex-specific methods
+          if (isKnexSpecificMethodUsed(j, path)) {
+            // Rename loaderWithAuthorizationResults to knexLoaderWithAuthorizationResults
+            if (loaderCallee.property.type === 'Identifier') {
+              loaderCallee.property.name = 'knexLoaderWithAuthorizationResults';
+            }
+          }
+        }
+      }
+    });
+}
+
+export default function transformer(file: FileInfo, api: API, _options: Options): string {
+  const j = api.jscodeshift;
+  const root = j.withParser('ts')(file.source);
+
+  transformLoaderToKnexLoader(j, root);
+  transformLoaderWithAuthorizationResultsToKnexLoaderWithAuthorizationResults(j, root);
+
+  return root.toSource();
+}