@expo/entity-codemod 0.40.0

package/README.md

@@ -0,0 +1,26 @@
+# @expo/entity-codemod
+
+A package containing jscodeshift codemods for @expo/entity upgrades.
+
+Codemods are transformations that run on your codebase programmatically. This allows for a large amount of changes to be applied without having to manually go through every file.
+
+## Usage
+
+These should be used via the [jscodeshift CLI](https://github.com/facebook/jscodeshift?tab=readme-ov-file#usage-cli).
+
+For example:
+
+```
+yarn add -D @expo/entity-codemod
+```
+
+Then, in package.json scripts:
+
+```json
+"jscodeshift": "jscodeshift --extensions=ts --parser=ts"
+```
+
+And finally:
+```sh
+yarn jscodeshift src -t node_modules/@expo/entity-codemod/build/v0.39.0-v0.40.0.js
+```

package/build/index.js

@@ -0,0 +1,2 @@
+"use strict";
+//# sourceMappingURL=index.js.map

package/build/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":""}

package/build/transforms/__tests__/v0.39.0-v0.40.0-test.d.ts

@@ -0,0 +1 @@
+export {};

package/build/transforms/__tests__/v0.39.0-v0.40.0-test.js

@@ -0,0 +1,16 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const fs_1 = require("fs");
+const path_1 = require("path");
+jest.autoMockOff();
+const defineTest = require('jscodeshift/dist/testUtils').defineTest;
+const fixtureDir = 'v0.39.0-v0.40.0';
+const fixtureDirPath = (0, path_1.join)(__dirname, '..', '__testfixtures__', fixtureDir);
+const fixtures = (0, fs_1.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.39.0-v0.40.0', null, prefix, { parser: 'ts' });
+}
+//# sourceMappingURL=v0.39.0-v0.40.0-test.js.map

package/build/transforms/__tests__/v0.39.0-v0.40.0-test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"v0.39.0-v0.40.0-test.js","sourceRoot":"","sources":["../../../src/transforms/__tests__/v0.39.0-v0.40.0-test.ts"],"names":[],"mappings":";;AAAA,2BAAiC;AACjC,+BAA4B;AAE5B,IAAI,CAAC,WAAW,EAAE,CAAC;AACnB,MAAM,UAAU,GAAG,OAAO,CAAC,4BAA4B,CAAC,CAAC,UAAU,CAAC;AAEpE,MAAM,UAAU,GAAG,iBAAiB,CAAC;AACrC,MAAM,cAAc,GAAG,IAAA,WAAI,EAAC,SAAS,EAAE,IAAI,EAAE,kBAAkB,EAAE,UAAU,CAAC,CAAC;AAC7E,MAAM,QAAQ,GAAG,IAAA,gBAAW,EAAC,cAAc,CAAC;KACzC,MAAM,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;KAC5C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC,CAAC;AAEhD,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;IAC/B,MAAM,MAAM,GAAG,GAAG,UAAU,IAAI,OAAO,EAAE,CAAC;IAC1C,UAAU,CAAC,SAAS,EAAE,iBAAiB,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;AAC3E,CAAC"}

package/build/transforms/v0.39.0-v0.40.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/transforms/v0.39.0-v0.40.0.js

@@ -0,0 +1,128 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.default = transformer;
+function transformEnforcingEntityChainMethod(j, root, type, crudType, methodBefore, methodAfter) {
+    // Find all entity creator expressions of the form
+    // `TestEntity.creator(viewerContext).setField('wat', 2).enforceCreateAsync()`
+    root
+        .find(j.CallExpression, {
+        callee: {
+            type: 'MemberExpression',
+            property: {
+                name: methodBefore,
+            },
+        },
+    })
+        .forEach((path) => {
+        const enforceCreateAsyncCallExpression = path.node; // TestEntity.creator(viewerContext).setField('wat', 2).enforceCreateAsync()
+        const enforceCreateAsyncCallee = enforceCreateAsyncCallExpression.callee; // TestEntity.creator(viewerContext).setField('wat', 2).enforceCreateAsync
+        if (enforceCreateAsyncCallee.type !== 'MemberExpression') {
+            return;
+        }
+        if (enforceCreateAsyncCallee.property.type !== 'Identifier') {
+            return;
+        }
+        // traverse in until we find the first non-setField call
+        let lastCallee = enforceCreateAsyncCallee;
+        while (true) {
+            // TestEntity.creator(viewerContext).setField('wat', 2)
+            const maybeSetFieldObjectCallExpression = lastCallee.object;
+            if (maybeSetFieldObjectCallExpression.type !== 'CallExpression') {
+                break;
+            }
+            // TestEntity.creator(viewerContext).setField
+            const maybeSetFieldObjectCallee = maybeSetFieldObjectCallExpression.callee;
+            if (maybeSetFieldObjectCallee.type !== 'MemberExpression') {
+                break;
+            }
+            const maybeSetFieldObjectCalleeProperty = maybeSetFieldObjectCallee.property;
+            if (maybeSetFieldObjectCalleeProperty.type !== 'Identifier') {
+                break;
+            }
+            if (maybeSetFieldObjectCalleeProperty.name === 'setField') {
+                lastCallee = maybeSetFieldObjectCallee;
+            }
+            else {
+                break;
+            }
+        }
+        // TestEntity.creator(viewerContext)
+        const maybeCreatorCallExpression = lastCallee.object;
+        if (maybeCreatorCallExpression.type !== 'CallExpression') {
+            return;
+        }
+        if (maybeCreatorCallExpression.callee.type !== 'MemberExpression' ||
+            maybeCreatorCallExpression.callee.property.type !== 'Identifier' ||
+            maybeCreatorCallExpression.callee.property.name !== crudType) {
+            return;
+        }
+        // TestEntity.creator(viewerContext).enforcing()
+        const newCreatorCallExpression = j.callExpression(j.memberExpression(maybeCreatorCallExpression, j.identifier(type)), []);
+        // TestEntity.creator(viewerContext).enforcing().setField('wat', 2).createAsync()
+        lastCallee.object = newCreatorCallExpression;
+        // change enforceCreateAsync to createAsync at the end so if this returns early it doesn't mutate
+        enforceCreateAsyncCallee.property.name = methodAfter;
+    });
+}
+function transformEnforcingEntityDeleteMethod(j, root, type, methodBefore, methodAfter) {
+    // Find all entity creator expressions of the form
+    // `TestEntity.enforceDeleteAsync(entity)`
+    root
+        .find(j.CallExpression, {
+        callee: {
+            type: 'MemberExpression',
+            property: {
+                name: methodBefore,
+            },
+        },
+    })
+        .forEach((path) => {
+        const enforceDeleteAsyncCallExpression = path.node; // TestEntity.enforceDeleteAsync(entity)
+        const args = enforceDeleteAsyncCallExpression.arguments; // [entity]
+        enforceDeleteAsyncCallExpression.arguments = [];
+        const enforceDeleteAsyncCallee = enforceDeleteAsyncCallExpression.callee; // TestEntity.enforceDeleteAsync
+        if (enforceDeleteAsyncCallee.type !== 'MemberExpression') {
+            return;
+        }
+        // change enforceDeleteAsync to deleteAsync
+        if (enforceDeleteAsyncCallee.property.type !== 'Identifier') {
+            return;
+        }
+        enforceDeleteAsyncCallee.property.name = methodAfter;
+        enforceDeleteAsyncCallee.object = j.callExpression(j.memberExpression(j.callExpression(j.memberExpression(enforceDeleteAsyncCallee.object, j.identifier('deleter')), args), j.identifier(type)), []);
+    });
+}
+function transformAssociationLoaderMethod(j, root) {
+    // Find all entity associationLoader expressions of the form
+    // `this.associationLoader()`
+    root
+        .find(j.CallExpression, {
+        callee: {
+            type: 'MemberExpression',
+            property: {
+                name: 'associationLoader',
+            },
+        },
+    })
+        .forEach((path) => {
+        const associationLoaderCallExpression = path.node; // this.associationLoader()
+        // replace with this.associationLoader().withAuthorizationResults()
+        path.replace(j.callExpression(j.memberExpression(associationLoaderCallExpression, j.identifier('withAuthorizationResults')), []));
+    });
+}
+function transformer(file, api, _options) {
+    const j = api.jscodeshift;
+    const root = j.withParser('ts')(file.source);
+    // do authorization results first since it uses the same detection pattern as enforcing post-transform (i.e. it looks for createAsync)
+    transformEnforcingEntityChainMethod(j, root, 'withAuthorizationResults', 'creator', 'createAsync', 'createAsync');
+    transformEnforcingEntityChainMethod(j, root, 'withAuthorizationResults', 'updater', 'updateAsync', 'updateAsync');
+    transformEnforcingEntityDeleteMethod(j, root, 'withAuthorizationResults', 'deleteAsync', 'deleteAsync');
+    // now do enforcing
+    transformEnforcingEntityChainMethod(j, root, 'enforcing', 'creator', 'enforceCreateAsync', 'createAsync');
+    transformEnforcingEntityChainMethod(j, root, 'enforcing', 'updater', 'enforceUpdateAsync', 'updateAsync');
+    transformEnforcingEntityDeleteMethod(j, root, 'enforcing', 'enforceDeleteAsync', 'deleteAsync');
+    // association loader
+    transformAssociationLoaderMethod(j, root);
+    return root.toSource();
+}
+//# sourceMappingURL=v0.39.0-v0.40.0.js.map

package/build/transforms/v0.39.0-v0.40.0.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"v0.39.0-v0.40.0.js","sourceRoot":"","sources":["../../src/transforms/v0.39.0-v0.40.0.ts"],"names":[],"mappings":";;AAgKA,8BAoDC;AAlND,SAAS,mCAAmC,CAC1C,CAAqB,EACrB,IAAqB,EACrB,IAA8C,EAC9C,QAA+B,EAC/B,YAAoB,EACpB,WAAmB;IAEnB,kDAAkD;IAClD,8EAA8E;IAC9E,IAAI;SACD,IAAI,CAAC,CAAC,CAAC,cAAc,EAAE;QACtB,MAAM,EAAE;YACN,IAAI,EAAE,kBAAkB;YACxB,QAAQ,EAAE;gBACR,IAAI,EAAE,YAAY;aACnB;SACF;KACF,CAAC;SACD,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;QAChB,MAAM,gCAAgC,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,4EAA4E;QAChI,MAAM,wBAAwB,GAAG,gCAAgC,CAAC,MAAM,CAAC,CAAC,0EAA0E;QACpJ,IAAI,wBAAwB,CAAC,IAAI,KAAK,kBAAkB,EAAE,CAAC;YACzD,OAAO;QACT,CAAC;QAED,IAAI,wBAAwB,CAAC,QAAQ,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YAC5D,OAAO;QACT,CAAC;QAED,wDAAwD;QACxD,IAAI,UAAU,GAAG,wBAAwB,CAAC;QAC1C,OAAO,IAAI,EAAE,CAAC;YACZ,uDAAuD;YACvD,MAAM,iCAAiC,GAAG,UAAU,CAAC,MAAM,CAAC;YAC5D,IAAI,iCAAiC,CAAC,IAAI,KAAK,gBAAgB,EAAE,CAAC;gBAChE,MAAM;YACR,CAAC;YAED,6CAA6C;YAC7C,MAAM,yBAAyB,GAAG,iCAAiC,CAAC,MAAM,CAAC;YAC3E,IAAI,yBAAyB,CAAC,IAAI,KAAK,kBAAkB,EAAE,CAAC;gBAC1D,MAAM;YACR,CAAC;YAED,MAAM,iCAAiC,GAAG,yBAAyB,CAAC,QAAQ,CAAC;YAC7E,IAAI,iCAAiC,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;gBAC5D,MAAM;YACR,CAAC;YAED,IAAI,iCAAiC,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBAC1D,UAAU,GAAG,yBAAyB,CAAC;YACzC,CAAC;iBAAM,CAAC;gBACN,MAAM;YACR,CAAC;QACH,CAAC;QAED,oCAAoC;QACpC,MAAM,0BAA0B,GAAG,UAAU,CAAC,MAAM,CAAC;QACrD,IAAI,0BAA0B,CAAC,IAAI,KAAK,gBAAgB,EAAE,CAAC;YACzD,OAAO;QACT,CAAC;QACD,IACE,0BAA0B,CAAC,MAAM,CAAC,IAAI,KAAK,kBAAkB;YAC7D,0BAA0B,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,KAAK,YAAY;YAChE,0BAA0B,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,KAAK,QAAQ,EAC5D,CAAC;YACD,OAAO;QACT,CAAC;QAED,gDAAgD;QAChD,MAAM,wBAAwB,GAAG,CAAC,CAAC,cAAc,CAC/C,CAAC,CAAC,gBAAgB,CAAC,0BAA0B,EAAE,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,EAClE,EAAE,CACH,CAAC;QAEF,iFAAiF;QACjF,UAAU,CAAC,MAAM,GAAG,wBAAwB,CAAC;QAE7C,iGAAiG;QACjG,wBAAwB,CAAC,QAAQ,CAAC,IAAI,GAAG,WAAW,CAAC;IACvD,CAAC,CAAC,CAAC;AACP,CAAC;AAED,SAAS,oCAAoC,CAC3C,CAAqB,EACrB,IAAqB,EACrB,IAA8C,EAC9C,YAAoB,EACpB,WAAmB;IAEnB,kDAAkD;IAClD,0CAA0C;IAC1C,IAAI;SACD,IAAI,CAAC,CAAC,CAAC,cAAc,EAAE;QACtB,MAAM,EAAE;YACN,IAAI,EAAE,kBAAkB;YACxB,QAAQ,EAAE;gBACR,IAAI,EAAE,YAAY;aACnB;SACF;KACF,CAAC;SACD,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;QAChB,MAAM,gCAAgC,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,wCAAwC;QAC5F,MAAM,IAAI,GAAG,gCAAgC,CAAC,SAAS,CAAC,CAAC,WAAW;QACpE,gCAAgC,CAAC,SAAS,GAAG,EAAE,CAAC;QAChD,MAAM,wBAAwB,GAAG,gCAAgC,CAAC,MAAM,CAAC,CAAC,gCAAgC;QAC1G,IAAI,wBAAwB,CAAC,IAAI,KAAK,kBAAkB,EAAE,CAAC;YACzD,OAAO;QACT,CAAC;QAED,2CAA2C;QAC3C,IAAI,wBAAwB,CAAC,QAAQ,CAAC,IAAI,KAAK,YAAY,EAAE,CAAC;YAC5D,OAAO;QACT,CAAC;QACD,wBAAwB,CAAC,QAAQ,CAAC,IAAI,GAAG,WAAW,CAAC;QAErD,wBAAwB,CAAC,MAAM,GAAG,CAAC,CAAC,cAAc,CAChD,CAAC,CAAC,gBAAgB,CAChB,CAAC,CAAC,cAAc,CACd,CAAC,CAAC,gBAAgB,CAAC,wBAAwB,CAAC,MAAM,EAAE,CAAC,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC,EAC5E,IAAI,CACL,EACD,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,CACnB,EACD,EAAE,CACH,CAAC;IACJ,CAAC,CAAC,CAAC;AACP,CAAC;AAED,SAAS,gCAAgC,CAAC,CAAqB,EAAE,IAAqB;IACpF,4DAA4D;IAC5D,6BAA6B;IAC7B,IAAI;SACD,IAAI,CAAC,CAAC,CAAC,cAAc,EAAE;QACtB,MAAM,EAAE;YACN,IAAI,EAAE,kBAAkB;YACxB,QAAQ,EAAE;gBACR,IAAI,EAAE,mBAAmB;aAC1B;SACF;KACF,CAAC;SACD,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;QAChB,MAAM,+BAA+B,GAAG,IAAI,CAAC,IAAI,CAAC,CAAC,2BAA2B;QAE9E,mEAAmE;QACnE,IAAI,CAAC,OAAO,CACV,CAAC,CAAC,cAAc,CACd,CAAC,CAAC,gBAAgB,CAChB,+BAA+B,EAC/B,CAAC,CAAC,UAAU,CAAC,0BAA0B,CAAC,CACzC,EACD,EAAE,CACH,CACF,CAAC;IACJ,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,sIAAsI;IACtI,mCAAmC,CACjC,CAAC,EACD,IAAI,EACJ,0BAA0B,EAC1B,SAAS,EACT,aAAa,EACb,aAAa,CACd,CAAC;IACF,mCAAmC,CACjC,CAAC,EACD,IAAI,EACJ,0BAA0B,EAC1B,SAAS,EACT,aAAa,EACb,aAAa,CACd,CAAC;IACF,oCAAoC,CAClC,CAAC,EACD,IAAI,EACJ,0BAA0B,EAC1B,aAAa,EACb,aAAa,CACd,CAAC;IAEF,mBAAmB;IACnB,mCAAmC,CACjC,CAAC,EACD,IAAI,EACJ,WAAW,EACX,SAAS,EACT,oBAAoB,EACpB,aAAa,CACd,CAAC;IACF,mCAAmC,CACjC,CAAC,EACD,IAAI,EACJ,WAAW,EACX,SAAS,EACT,oBAAoB,EACpB,aAAa,CACd,CAAC;IACF,oCAAoC,CAAC,CAAC,EAAE,IAAI,EAAE,WAAW,EAAE,oBAAoB,EAAE,aAAa,CAAC,CAAC;IAEhG,qBAAqB;IACrB,gCAAgC,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;IAE1C,OAAO,IAAI,CAAC,QAAQ,EAAE,CAAC;AACzB,CAAC"}

package/package.json

@@ -0,0 +1,40 @@
+{
+  "name": "@expo/entity-codemod",
+  "version": "0.40.0",
+  "description": "jscodeshift codemods for @expo/entity upgrades",
+  "files": [
+    "build",
+    "src"
+  ],
+  "main": "build/index.js",
+  "types": "build/index.d.ts",
+  "scripts": {
+    "tsc": "tsc",
+    "clean": "rm -rf build coverage coverage-integration",
+    "lint": "eslint src",
+    "lint-fix": "eslint src --fix",
+    "test": "jest --rootDir . --config ../../resources/jest.config.js"
+  },
+  "engines": {
+    "node": ">=16"
+  },
+  "keywords": [
+    "entity"
+  ],
+  "author": "Expo",
+  "license": "MIT",
+  "devDependencies": {
+    "@types/jest": "^29.5.12",
+    "@types/jscodeshift": "^0.12.0",
+    "@types/node": "^20.14.1",
+    "eslint": "^8.57.1",
+    "eslint-config-universe": "^14.0.0",
+    "eslint-plugin-tsdoc": "^0.3.0",
+    "jest": "^29.7.0",
+    "jscodeshift": "^17.1.2",
+    "prettier": "^3.3.3",
+    "ts-jest": "^29.2.5",
+    "ts-mockito": "^2.6.1",
+    "typescript": "^5.7.3"
+  }
+}

package/src/transforms/__testfixtures__/v0.39.0-v0.40.0/test1.input.ts

@@ -0,0 +1,79 @@
+/* eslint-disable */
+
+import { ViewerContext } from '@expo/entity';
+
+async function testWithAuthorizationResults(viewerContext: ViewerContext): Promise<void> {
+  // creator
+  const entityResult = await TestEntity.creator(viewerContext).setField('wat', 2).createAsync();
+
+  // updater
+  const updatedEntityResult = await TestEntity.updater(entityResult.enforceValue()).setField('wat', 3).updateAsync();
+
+  // deleter
+  await TestEntity.deleteAsync(updatedEntityResult.enforceValue());
+}
+
+async function testEnforcing(viewerContext: ViewerContext): Promise<void> {
+  // creator
+  const entity = await TestEntity.creator(viewerContext).setField('wat', 2).enforceCreateAsync();
+  const entity2 = await TestEntity.creator(viewerContext).setField('wat', 2).setField('who', 3).enforceCreateAsync();
+  const entity3 = await TestEntity.creator(viewerContext)
+    .setField('wat', 2)
+    .setField('who', 3)
+    .setField('who', 4)
+    .setField('who', 5)
+    .setField('who', 6)
+    .setField('who', 7)
+    .setField('who', 8)
+    .setField('who', 9)
+    .enforceCreateAsync();
+
+  // updater
+  const updatedEntity = await TestEntity.updater(entity).setField('wat', 3).enforceUpdateAsync();
+  const updatedEntity2 = await TestEntity.updater(entity2).setField('wat', 3).setField('who', 4).enforceUpdateAsync();
+  const updatedEntity3 = await TestEntity.updater(entity3)
+    .setField('wat', 3)
+    .setField('who', 4)
+    .setField('who', 5)
+    .setField('who', 6)
+    .setField('who', 7)
+    .setField('who', 8)
+    .setField('who', 9)
+    .enforceUpdateAsync();
+
+  // deleter
+  await TestEntity.enforceDeleteAsync(entity);
+}
+
+async function testNoSetField(viewerContext: ViewerContext): Promise<void> {
+  // creator
+  const entity = await TestEntity.creator(viewerContext).enforceCreateAsync();
+
+  // updater
+  const updatedEntity = await TestEntity.updater(entity).enforceUpdateAsync();
+}
+
+async function testEnforcingWithQueryContext(viewerContext: ViewerContext): Promise<void> {
+  // creator
+  const entity = await TestEntity.creator(viewerContext, queryContext).setField('wat', 2).enforceCreateAsync();
+
+  // updater
+  const updatedEntity = await TestEntity.updater(entity, queryContext).setField('wat', 3).enforceUpdateAsync();
+
+  // deleter
+  await TestEntity.enforceDeleteAsync(entity, queryContext);
+}
+
+async function testAssociationLoader(): Promise<void> {
+  await this.associationLoader().loadAssociatedEntityAsync(
+    'another_id',
+    AnotherEntity,
+    queryContext
+  );
+
+  await entity.associationLoader().loadAssociatedEntityAsync(
+    'another_id',
+    AnotherEntity,
+    queryContext
+  )
+}

package/src/transforms/__testfixtures__/v0.39.0-v0.40.0/test1.output.ts

@@ -0,0 +1,79 @@
+/* eslint-disable */
+
+import { ViewerContext } from '@expo/entity';
+
+async function testWithAuthorizationResults(viewerContext: ViewerContext): Promise<void> {
+  // creator
+  const entityResult = await TestEntity.creator(viewerContext).withAuthorizationResults().setField('wat', 2).createAsync();
+
+  // updater
+  const updatedEntityResult = await TestEntity.updater(entityResult.enforceValue()).withAuthorizationResults().setField('wat', 3).updateAsync();
+
+  // deleter
+  await TestEntity.deleter(updatedEntityResult.enforceValue()).withAuthorizationResults().deleteAsync();
+}
+
+async function testEnforcing(viewerContext: ViewerContext): Promise<void> {
+  // creator
+  const entity = await TestEntity.creator(viewerContext).enforcing().setField('wat', 2).createAsync();
+  const entity2 = await TestEntity.creator(viewerContext).enforcing().setField('wat', 2).setField('who', 3).createAsync();
+  const entity3 = await TestEntity.creator(viewerContext).enforcing()
+    .setField('wat', 2)
+    .setField('who', 3)
+    .setField('who', 4)
+    .setField('who', 5)
+    .setField('who', 6)
+    .setField('who', 7)
+    .setField('who', 8)
+    .setField('who', 9)
+    .createAsync();
+
+  // updater
+  const updatedEntity = await TestEntity.updater(entity).enforcing().setField('wat', 3).updateAsync();
+  const updatedEntity2 = await TestEntity.updater(entity2).enforcing().setField('wat', 3).setField('who', 4).updateAsync();
+  const updatedEntity3 = await TestEntity.updater(entity3).enforcing()
+    .setField('wat', 3)
+    .setField('who', 4)
+    .setField('who', 5)
+    .setField('who', 6)
+    .setField('who', 7)
+    .setField('who', 8)
+    .setField('who', 9)
+    .updateAsync();
+
+  // deleter
+  await TestEntity.deleter(entity).enforcing().deleteAsync();
+}
+
+async function testNoSetField(viewerContext: ViewerContext): Promise<void> {
+  // creator
+  const entity = await TestEntity.creator(viewerContext).enforcing().createAsync();
+
+  // updater
+  const updatedEntity = await TestEntity.updater(entity).enforcing().updateAsync();
+}
+
+async function testEnforcingWithQueryContext(viewerContext: ViewerContext): Promise<void> {
+  // creator
+  const entity = await TestEntity.creator(viewerContext, queryContext).enforcing().setField('wat', 2).createAsync();
+
+  // updater
+  const updatedEntity = await TestEntity.updater(entity, queryContext).enforcing().setField('wat', 3).updateAsync();
+
+  // deleter
+  await TestEntity.deleter(entity, queryContext).enforcing().deleteAsync();
+}
+
+async function testAssociationLoader(): Promise<void> {
+  await this.associationLoader().withAuthorizationResults().loadAssociatedEntityAsync(
+    'another_id',
+    AnotherEntity,
+    queryContext
+  );
+
+  await entity.associationLoader().withAuthorizationResults().loadAssociatedEntityAsync(
+    'another_id',
+    AnotherEntity,
+    queryContext
+  )
+}

package/src/transforms/__testfixtures__/v0.39.0-v0.40.0/test2.input.ts

@@ -0,0 +1,10 @@
+/* eslint-disable */
+
+async function testAsync(viewerContext: ViewerContext): Promise<void> {
+  const testTrigger = await TestTriggerEntity.createAsync(actorViewerContext, undefined, {
+    appId,
+  });
+
+  await TestTriggerEntity.enforceDeleteAsync(testTrigger);
+  return testTrigger;
+};

package/src/transforms/__testfixtures__/v0.39.0-v0.40.0/test2.output.ts

@@ -0,0 +1,10 @@
+/* eslint-disable */
+
+async function testAsync(viewerContext: ViewerContext): Promise<void> {
+  const testTrigger = await TestTriggerEntity.createAsync(actorViewerContext, undefined, {
+    appId,
+  });
+
+  await TestTriggerEntity.deleter(testTrigger).enforcing().deleteAsync();
+  return testTrigger;
+};

package/src/transforms/__testfixtures__/v0.39.0-v0.40.0/test3.input.ts

@@ -0,0 +1,5 @@
+/* eslint-disable */
+
+async function anotherCase(updater: any): Promise<any> {
+  return await updater.enforceUpdateAsync();
+}

package/src/transforms/__testfixtures__/v0.39.0-v0.40.0/test3.output.ts

@@ -0,0 +1,5 @@
+/* eslint-disable */
+
+async function anotherCase(updater: any): Promise<any> {
+  return await updater.enforceUpdateAsync();
+}

package/src/transforms/__tests__/v0.39.0-v0.40.0-test.ts

@@ -0,0 +1,16 @@
+import { readdirSync } from 'fs';
+import { join } from 'path';
+
+jest.autoMockOff();
+const defineTest = require('jscodeshift/dist/testUtils').defineTest;
+
+const fixtureDir = 'v0.39.0-v0.40.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.39.0-v0.40.0', null, prefix, { parser: 'ts' });
+}

package/src/transforms/v0.39.0-v0.40.0.ts

@@ -0,0 +1,213 @@
+import { API, Collection, FileInfo, Options } from 'jscodeshift';
+
+function transformEnforcingEntityChainMethod(
+  j: API['jscodeshift'],
+  root: Collection<any>,
+  type: 'enforcing' | 'withAuthorizationResults',
+  crudType: 'creator' | 'updater',
+  methodBefore: string,
+  methodAfter: string,
+): void {
+  // Find all entity creator expressions of the form
+  // `TestEntity.creator(viewerContext).setField('wat', 2).enforceCreateAsync()`
+  root
+    .find(j.CallExpression, {
+      callee: {
+        type: 'MemberExpression',
+        property: {
+          name: methodBefore,
+        },
+      },
+    })
+    .forEach((path) => {
+      const enforceCreateAsyncCallExpression = path.node; // TestEntity.creator(viewerContext).setField('wat', 2).enforceCreateAsync()
+      const enforceCreateAsyncCallee = enforceCreateAsyncCallExpression.callee; // TestEntity.creator(viewerContext).setField('wat', 2).enforceCreateAsync
+      if (enforceCreateAsyncCallee.type !== 'MemberExpression') {
+        return;
+      }
+
+      if (enforceCreateAsyncCallee.property.type !== 'Identifier') {
+        return;
+      }
+
+      // traverse in until we find the first non-setField call
+      let lastCallee = enforceCreateAsyncCallee;
+      while (true) {
+        // TestEntity.creator(viewerContext).setField('wat', 2)
+        const maybeSetFieldObjectCallExpression = lastCallee.object;
+        if (maybeSetFieldObjectCallExpression.type !== 'CallExpression') {
+          break;
+        }
+
+        // TestEntity.creator(viewerContext).setField
+        const maybeSetFieldObjectCallee = maybeSetFieldObjectCallExpression.callee;
+        if (maybeSetFieldObjectCallee.type !== 'MemberExpression') {
+          break;
+        }
+
+        const maybeSetFieldObjectCalleeProperty = maybeSetFieldObjectCallee.property;
+        if (maybeSetFieldObjectCalleeProperty.type !== 'Identifier') {
+          break;
+        }
+
+        if (maybeSetFieldObjectCalleeProperty.name === 'setField') {
+          lastCallee = maybeSetFieldObjectCallee;
+        } else {
+          break;
+        }
+      }
+
+      // TestEntity.creator(viewerContext)
+      const maybeCreatorCallExpression = lastCallee.object;
+      if (maybeCreatorCallExpression.type !== 'CallExpression') {
+        return;
+      }
+      if (
+        maybeCreatorCallExpression.callee.type !== 'MemberExpression' ||
+        maybeCreatorCallExpression.callee.property.type !== 'Identifier' ||
+        maybeCreatorCallExpression.callee.property.name !== crudType
+      ) {
+        return;
+      }
+
+      // TestEntity.creator(viewerContext).enforcing()
+      const newCreatorCallExpression = j.callExpression(
+        j.memberExpression(maybeCreatorCallExpression, j.identifier(type)),
+        [],
+      );
+
+      // TestEntity.creator(viewerContext).enforcing().setField('wat', 2).createAsync()
+      lastCallee.object = newCreatorCallExpression;
+
+      // change enforceCreateAsync to createAsync at the end so if this returns early it doesn't mutate
+      enforceCreateAsyncCallee.property.name = methodAfter;
+    });
+}
+
+function transformEnforcingEntityDeleteMethod(
+  j: API['jscodeshift'],
+  root: Collection<any>,
+  type: 'enforcing' | 'withAuthorizationResults',
+  methodBefore: string,
+  methodAfter: string,
+): void {
+  // Find all entity creator expressions of the form
+  // `TestEntity.enforceDeleteAsync(entity)`
+  root
+    .find(j.CallExpression, {
+      callee: {
+        type: 'MemberExpression',
+        property: {
+          name: methodBefore,
+        },
+      },
+    })
+    .forEach((path) => {
+      const enforceDeleteAsyncCallExpression = path.node; // TestEntity.enforceDeleteAsync(entity)
+      const args = enforceDeleteAsyncCallExpression.arguments; // [entity]
+      enforceDeleteAsyncCallExpression.arguments = [];
+      const enforceDeleteAsyncCallee = enforceDeleteAsyncCallExpression.callee; // TestEntity.enforceDeleteAsync
+      if (enforceDeleteAsyncCallee.type !== 'MemberExpression') {
+        return;
+      }
+
+      // change enforceDeleteAsync to deleteAsync
+      if (enforceDeleteAsyncCallee.property.type !== 'Identifier') {
+        return;
+      }
+      enforceDeleteAsyncCallee.property.name = methodAfter;
+
+      enforceDeleteAsyncCallee.object = j.callExpression(
+        j.memberExpression(
+          j.callExpression(
+            j.memberExpression(enforceDeleteAsyncCallee.object, j.identifier('deleter')),
+            args,
+          ),
+          j.identifier(type),
+        ),
+        [],
+      );
+    });
+}
+
+function transformAssociationLoaderMethod(j: API['jscodeshift'], root: Collection<any>): void {
+  // Find all entity associationLoader expressions of the form
+  // `this.associationLoader()`
+  root
+    .find(j.CallExpression, {
+      callee: {
+        type: 'MemberExpression',
+        property: {
+          name: 'associationLoader',
+        },
+      },
+    })
+    .forEach((path) => {
+      const associationLoaderCallExpression = path.node; // this.associationLoader()
+
+      // replace with this.associationLoader().withAuthorizationResults()
+      path.replace(
+        j.callExpression(
+          j.memberExpression(
+            associationLoaderCallExpression,
+            j.identifier('withAuthorizationResults'),
+          ),
+          [],
+        ),
+      );
+    });
+}
+
+export default function transformer(file: FileInfo, api: API, _options: Options): string {
+  const j = api.jscodeshift;
+  const root = j.withParser('ts')(file.source);
+
+  // do authorization results first since it uses the same detection pattern as enforcing post-transform (i.e. it looks for createAsync)
+  transformEnforcingEntityChainMethod(
+    j,
+    root,
+    'withAuthorizationResults',
+    'creator',
+    'createAsync',
+    'createAsync',
+  );
+  transformEnforcingEntityChainMethod(
+    j,
+    root,
+    'withAuthorizationResults',
+    'updater',
+    'updateAsync',
+    'updateAsync',
+  );
+  transformEnforcingEntityDeleteMethod(
+    j,
+    root,
+    'withAuthorizationResults',
+    'deleteAsync',
+    'deleteAsync',
+  );
+
+  // now do enforcing
+  transformEnforcingEntityChainMethod(
+    j,
+    root,
+    'enforcing',
+    'creator',
+    'enforceCreateAsync',
+    'createAsync',
+  );
+  transformEnforcingEntityChainMethod(
+    j,
+    root,
+    'enforcing',
+    'updater',
+    'enforceUpdateAsync',
+    'updateAsync',
+  );
+  transformEnforcingEntityDeleteMethod(j, root, 'enforcing', 'enforceDeleteAsync', 'deleteAsync');
+
+  // association loader
+  transformAssociationLoaderMethod(j, root);
+
+  return root.toSource();
+}