eslint-plugin-drizzle 0.2.2 → 0.2.3-a4d758d

package/package.json

@@ -1,6 +1,6 @@
 {
     "name": "eslint-plugin-drizzle",
-    "version": "0.2.2",
+    "version": "0.2.3-a4d758d",
     "description": "Eslint plugin for drizzle users to avoid common pitfalls",
     "main": "src/index.js",
     "scripts": {
@@ -22,6 +22,7 @@
     },
     "license": "Apache-2.0",
     "devDependencies": {
+        "@types/node": "^20.10.1",
         "@typescript-eslint/parser": "^6.10.0",
         "@typescript-eslint/rule-tester": "^6.10.0",
         "@typescript-eslint/utils": "^6.10.0",

package/readme.md

@@ -9,7 +9,9 @@ For cases where it's impossible to perform type checks for specific scenarios, o
 ```sh
 [ npm | yarn | pnpm | bun ] install eslint eslint-plugin-drizzle
 ```
+
 You can install those packages for typescript support in your IDE
+
 ```sh
 [ npm | yarn | pnpm | bun ] install @typescript-eslint/eslint-plugin @typescript-eslint/parser
 ```
@@ -65,6 +67,7 @@ plugins:
 Optionally, you can define a `drizzleObjectName` in the plugin options that accept a `string` or `string[]`. This is useful when you have objects or classes with a delete method that's not from Drizzle. Such a `delete` method will trigger the ESLint rule. To avoid that, you can define the name of the Drizzle object that you use in your codebase (like db) so that the rule would only trigger if the delete method comes from this object:
 
 Example, config 1:
+
 ```json
 "rules": {
   "drizzle/enforce-delete-with-where": ["error"]
@@ -89,11 +92,13 @@ db.delete()
 ```
 
 Example, config 2:
+
 ```json
 "rules": {
   "drizzle/enforce-delete-with-where": ["error", { "drizzleObjectName": ["db"] }],
 }
 ```
+
 ```ts
 class MyClass {
   public delete() {
@@ -116,6 +121,7 @@ db.delete()
 Optionally, you can define a `drizzleObjectName` in the plugin options that accept a `string` or `string[]`. This is useful when you have objects or classes with a delete method that's not from Drizzle. Such as `update` method will trigger the ESLint rule. To avoid that, you can define the name of the Drizzle object that you use in your codebase (like db) so that the rule would only trigger if the delete method comes from this object:
 
 Example, config 1:
+
 ```json
 "rules": {
   "drizzle/enforce-update-with-where": ["error"]
@@ -140,11 +146,13 @@ db.update()
 ```
 
 Example, config 2:
+
 ```json
 "rules": {
   "drizzle/enforce-update-with-where": ["error", { "drizzleObjectName": ["db"] }],
 }
 ```
+
 ```ts
 class MyClass {
   public update() {

package/src/enforce-delete-with-where.js

@@ -1,6 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 const utils_1 = require("@typescript-eslint/utils");
+const ast_1 = require("./utils/ast");
 const options_1 = require("./utils/options");
 const createRule = utils_1.ESLintUtils.RuleCreator(() => 'https://github.com/drizzle-team/eslint-plugin-drizzle');
 let lastNodeName = '';
@@ -14,7 +15,7 @@ const deleteRule = createRule({
         },
         fixable: 'code',
         messages: {
-            enforceDeleteWithWhere: "Without `.where(...)` you will delete all the rows in a table. If you didn't want to do it, please use `db.delete(...).where(...)` instead. Otherwise you can ignore this rule here",
+            enforceDeleteWithWhere: "Without `.where(...)` you will delete all the rows in a table. If you didn't want to do it, please use `{{ drizzleObjName }}.delete(...).where(...)` instead. Otherwise you can ignore this rule here",
         },
         schema: [{
                 type: 'object',
@@ -34,6 +35,9 @@ const deleteRule = createRule({
                         context.report({
                             node,
                             messageId: 'enforceDeleteWithWhere',
+                            data: {
+                                drizzleObjName: (0, ast_1.resolveMemberExpressionPath)(node),
+                            },
                         });
                     }
                     lastNodeName = node.property.name;

package/src/enforce-update-with-where.js

@@ -1,6 +1,7 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 const utils_1 = require("@typescript-eslint/utils");
+const ast_1 = require("./utils/ast");
 const options_1 = require("./utils/options");
 const createRule = utils_1.ESLintUtils.RuleCreator(() => 'https://github.com/drizzle-team/eslint-plugin-drizzle');
 let lastNodeName = '';
@@ -14,7 +15,7 @@ const updateRule = createRule({
         },
         fixable: 'code',
         messages: {
-            enforceUpdateWithWhere: 'Without `.where(...)` you will update all the rows in a table. If you didn\'t want to do it, please use `db.update(...).set(...).where(...)` instead. Otherwise you can ignore this rule here'
+            enforceUpdateWithWhere: "Without `.where(...)` you will update all the rows in a table. If you didn't want to do it, please use `{{ drizzleObjName }}.update(...).set(...).where(...)` instead. Otherwise you can ignore this rule here",
         },
         schema: [{
                 type: 'object',
@@ -40,6 +41,9 @@ const updateRule = createRule({
                         context.report({
                             node,
                             messageId: 'enforceUpdateWithWhere',
+                            data: {
+                                drizzleObjName: (0, ast_1.resolveMemberExpressionPath)(node.object.callee),
+                            },
                         });
                     }
                     lastNodeName = node.property.name;

package/src/utils/ast.js

@@ -0,0 +1,42 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.resolveMemberExpressionPath = void 0;
+const resolveMemberExpressionPath = (node) => {
+    let objectExpression = node.object;
+    let fullName = '';
+    const addToFullName = (name) => {
+        const prefix = fullName ? '.' : '';
+        fullName = `${name}${prefix}${fullName}`;
+    };
+    while (objectExpression) {
+        if (objectExpression.type === 'MemberExpression') {
+            if (objectExpression.property.type === 'Identifier') {
+                addToFullName(objectExpression.property.name);
+            }
+            objectExpression = objectExpression.object;
+        }
+        else if (objectExpression.type === 'CallExpression' && objectExpression.callee.type === 'Identifier') {
+            addToFullName(`${objectExpression.callee.name}(...)`);
+            break;
+        }
+        else if (objectExpression.type === 'CallExpression' && objectExpression.callee.type === 'MemberExpression') {
+            if (objectExpression.callee.property.type === 'Identifier') {
+                addToFullName(`${objectExpression.callee.property.name}(...)`);
+            }
+            objectExpression = objectExpression.callee.object;
+        }
+        else if (objectExpression.type === 'Identifier') {
+            addToFullName(objectExpression.name);
+            break;
+        }
+        else if (objectExpression.type === 'ThisExpression') {
+            addToFullName('this');
+            break;
+        }
+        else {
+            break;
+        }
+    }
+    return fullName;
+};
+exports.resolveMemberExpressionPath = resolveMemberExpressionPath;

package/src/utils/options.js

@@ -1,34 +1,32 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.isDrizzleObj = void 0;
+const isDrizzleObjName = (name, drizzleObjectName) => {
+    if (typeof drizzleObjectName === 'string') {
+        return name === drizzleObjectName;
+    }
+    if (Array.isArray(drizzleObjectName)) {
+        if (drizzleObjectName.length === 0) {
+            return true;
+        }
+        return drizzleObjectName.includes(name);
+    }
+    return false;
+};
 const isDrizzleObj = (node, options) => {
     const drizzleObjectName = options[0].drizzleObjectName;
     if (node.object.type === 'Identifier') {
-        if (typeof drizzleObjectName === 'string'
-            && node.object.name === drizzleObjectName) {
-            return true;
-        }
-        if (Array.isArray(drizzleObjectName)) {
-            if (drizzleObjectName.length === 0) {
-                return true;
-            }
-            if (drizzleObjectName.includes(node.object.name)) {
-                return true;
-            }
-        }
+        return isDrizzleObjName(node.object.name, drizzleObjectName);
     }
     else if (node.object.type === 'MemberExpression' && node.object.property.type === 'Identifier') {
-        if (typeof drizzleObjectName === 'string'
-            && node.object.property.name === drizzleObjectName) {
-            return true;
+        return isDrizzleObjName(node.object.property.name, drizzleObjectName);
+    }
+    else if (node.object.type === 'CallExpression') {
+        if (node.object.callee.type === 'Identifier') {
+            return isDrizzleObjName(node.object.callee.name, drizzleObjectName);
         }
-        if (Array.isArray(drizzleObjectName)) {
-            if (drizzleObjectName.length === 0) {
-                return true;
-            }
-            if (drizzleObjectName.includes(node.object.property.name)) {
-                return true;
-            }
+        else if (node.object.callee.type === 'MemberExpression' && node.object.callee.property.type === 'Identifier') {
+            return isDrizzleObjName(node.object.callee.property.name, drizzleObjectName);
         }
     }
     return false;