fix: handle this.prepare and super.prepare

Author: merceyz

src/utils.ts

@@ -4,6 +4,12 @@ export function stringifyNode(
 	node: TSESTree.Expression | TSESTree.PrivateIdentifier,
 ): string | null {
 	switch (node.type) {
+		case TSESTree.AST_NODE_TYPES.Super: {
+			return "super";
+		}
+		case TSESTree.AST_NODE_TYPES.ThisExpression: {
+			return "this";
+		}
 		case TSESTree.AST_NODE_TYPES.Identifier: {
 			return node.name;
 		}

tests/rules/typed-input.test.ts

@@ -29,6 +29,8 @@ ruleTester.run("typed-input", typedInputRule, {
 		'db.prepare<[unknown, {"userID": string}]>("SELECT * FROM users WHERE id = :userID or name = ?")',
 		'db.prepare<{"userID": string}>("SELECT * FROM users WHERE id = :userID")',
 		'db.prepare<[unknown]>(`SELECT * FROM users WHERE id IN (${foo.map(() => "?").join(",")})`)',
+		'this.prepare<[unknown]>("SELECT * FROM users WHERE id = ?")',
+		'super.prepare<[unknown]>("SELECT * FROM users WHERE id = ?")',
 	],
 	invalid: [
 		// No parameters
@@ -139,5 +141,15 @@ ruleTester.run("typed-input", typedInputRule, {
 				'db.prepare<[unknown]>(`SELECT * FROM users WHERE id IN (${foo.map(() => "?").join(",")})`)',
 			errors: [{ messageId: "missingInputType" }],
 		},
+		{
+			code: 'this.prepare("SELECT * FROM users WHERE id = ?")',
+			output: 'this.prepare<[unknown]>("SELECT * FROM users WHERE id = ?")',
+			errors: [{ messageId: "missingInputType" }],
+		},
+		{
+			code: 'super.prepare("SELECT * FROM users WHERE id = ?")',
+			output: 'super.prepare<[unknown]>("SELECT * FROM users WHERE id = ?")',
+			errors: [{ messageId: "missingInputType" }],
+		},
 	],
 });

tests/rules/typed-result.test.ts

@@ -61,6 +61,8 @@ ruleTester.run("typed-result", rule, {
 		`db.prepare<[]>("DELETE FROM foo")`,
 		`db.prepare<[], {"random()": (foo | number), "id": number}>("SELECT random(), id FROM users")`,
 		'db.prepare<[], {"name": string}>(`SELECT name FROM users WHERE id IN (${foo.map(() => "?").join(",")})`)',
+		'this.prepare<[], {"name": string}>(`SELECT name FROM users`)',
+		'super.prepare<[], {"name": string}>(`SELECT name FROM users`)',
 	],
 	invalid: [
 		// Query as string Literal
@@ -190,5 +192,15 @@ ruleTester.run("typed-result", rule, {
 				'db.prepare<[], {"name": string}>(`SELECT name FROM users WHERE id IN (${foo.map(() => "?").join(",")})`)',
 			errors: [{ messageId: "missingResultType" }],
 		},
+		{
+			code: "this.prepare(`SELECT name FROM users`)",
+			output: 'this.prepare<[], {"name": string}>(`SELECT name FROM users`)',
+			errors: [{ messageId: "missingResultType" }],
+		},
+		{
+			code: "super.prepare(`SELECT name FROM users`)",
+			output: 'super.prepare<[], {"name": string}>(`SELECT name FROM users`)',
+			errors: [{ messageId: "missingResultType" }],
+		},
 	],
 });

tests/rules/valid-query.test.ts

@@ -32,6 +32,8 @@ ruleTester.run("valid-query", rule, {
 		"db_users.prepare(`SELECT * FROM users WHERE ${ids.map(() => 'NAME LIKE ? || \\'%\\'').join(' OR ')}`);",
 		"const query = `SELECT * FROM users WHERE ${ids.map(() => 'NAME LIKE ? || \\'%\\'').join(' OR ')}`;db_users.prepare(query);",
 		"db_users.prepare(`SELECT * FROM users WHERE id IN (${ids.map(() => '?').join()})`);",
+		'this.prepare("SELECT * FROM foo")',
+		'super.prepare("SELECT * FROM foo")',
 	],
 	invalid: [
 		{
@@ -116,5 +118,27 @@ ruleTester.run("valid-query", rule, {
 				},
 			],
 		},
+		{
+			code: "this.prepare('SELECT * FROM user')",
+			errors: [
+				{
+					messageId: "invalidQuery",
+					data: {
+						message: `no such table: user`,
+					},
+				},
+			],
+		},
+		{
+			code: "super.prepare('SELECT * FROM user')",
+			errors: [
+				{
+					messageId: "invalidQuery",
+					data: {
+						message: `no such table: user`,
+					},
+				},
+			],
+		},
 	],
 });