support cjs,mjs,cts,mts extensions (#235)

* extensions

* tests

* docs

Author: JasonShin

book/docs/cli/README.mdx

@@ -106,7 +106,7 @@ sqlx-ts [OPTIONS] <PATH>
     Primary DB user
 
     --ext <EXT>
-    file extensions [possible values: ts, js, sql]
+    file extensions [possible values: ts, js, sql, mjs, cjs, mts, cts]
 
     -g, --generate-types
     generate types of raw SQLs using default configuration
@@ -246,7 +246,7 @@ user of the database that you connect to
 
 ### --ext
 
-File extensions to search. It can be either `ts` | `js` - [default: ts]
+File extensions to search. Supported extensions: `ts` | `js` | `sql` | `mjs` | `cjs` | `mts` | `cts` - [default: ts]
 
 <Tabs>
     <TabItem value="npm" label="npm" default>

src/common/cli.rs

@@ -8,6 +8,10 @@ impl fmt::Display for FileExtension {
       FileExtension::Ts => ".ts".to_string(),
       FileExtension::Js => ".js".to_string(),
       FileExtension::Sql => ".sql".to_string(),
+      FileExtension::Mts => ".mts".to_string(),
+      FileExtension::Cts => ".cts".to_string(),
+      FileExtension::Mjs => ".mjs".to_string(),
+      FileExtension::Cjs => ".cjs".to_string(),
     };
     write!(f, "{extension}")
   }

src/common/types.rs

@@ -7,6 +7,10 @@ pub enum FileExtension {
   Ts,
   Js,
   Sql,
+  Mts,
+  Cts,
+  Mjs,
+  Cjs,
 }
 
 #[derive(ValueEnum, Debug, Clone, Serialize, Deserialize)]

src/parser/mod.rs

@@ -15,7 +15,7 @@ use swc_common::errors::Handler;
 pub fn parse_source(path: &PathBuf) -> Result<(HashMap<PathBuf, Vec<SQL>>, Handler)> {
   let extension = path.extension().and_then(|s| s.to_str()).unwrap_or("");
   let result = match extension {
-    "ts" | "js" => parse_js_file(path),
+    "ts" | "js" | "mts" | "cts" | "mjs" | "cjs" => parse_js_file(path),
     "sql" => parse_sql_file(path),
     _ => {
       return Err(color_eyre::eyre::eyre!("Unsupported file extension: {}", extension));

tests/demo/file_extensions/queries-cjs.cjs

@@ -0,0 +1,17 @@
+const { sql } = require('sqlx-ts')
+
+// .cjs file extension test (CommonJS JavaScript)
+const cjsQuery = sql`
+-- @name: cjs query
+SELECT id, name FROM items WHERE rarity = $1
+`
+
+module.exports = {
+  cjsQuery,
+  cjsFunction: () => {
+    return sql`
+-- @name: cjs function query
+SELECT id, name FROM items WHERE id = $1
+    `
+  }
+}

tests/demo/file_extensions/queries-cjs.queries.ts

@@ -0,0 +1,23 @@
+export type CjsQueryParams = [string | null];
+
+export interface ICjsQueryResult {
+	id: number;
+	name: string;
+}
+
+export interface ICjsQueryQuery {
+	params: CjsQueryParams;
+	result: ICjsQueryResult;
+}
+
+export type CjsFunctionQueryParams = [number];
+
+export interface ICjsFunctionQueryResult {
+	id: number;
+	name: string;
+}
+
+export interface ICjsFunctionQueryQuery {
+	params: CjsFunctionQueryParams;
+	result: ICjsFunctionQueryResult;
+}

tests/demo/file_extensions/queries-cjs.snapshot.ts

@@ -0,0 +1,24 @@
+export type CjsQueryParams = [string | null];
+
+export interface ICjsQueryResult {
+	id: number;
+	name: string;
+}
+
+export interface ICjsQueryQuery {
+	params: CjsQueryParams;
+	result: ICjsQueryResult;
+}
+
+export type CjsFunctionQueryParams = [number];
+
+export interface ICjsFunctionQueryResult {
+	id: number;
+	name: string;
+}
+
+export interface ICjsFunctionQueryQuery {
+	params: CjsFunctionQueryParams;
+	result: ICjsFunctionQueryResult;
+}
+

tests/demo/file_extensions/queries-cts.cts

@@ -0,0 +1,17 @@
+import { sql } from 'sqlx-ts'
+
+// .cts file extension test (CommonJS TypeScript)
+const ctsQuery = sql`
+-- @name: cts query
+SELECT id, name FROM items WHERE rarity = $1
+`
+
+module.exports = {
+  ctsQuery,
+  ctsFunction: () => {
+    return sql`
+-- @name: cts function query
+SELECT id, name FROM items WHERE id = $1
+    `
+  }
+}

tests/demo/file_extensions/queries-cts.queries.ts

@@ -0,0 +1,23 @@
+export type CtsQueryParams = [string | null];
+
+export interface ICtsQueryResult {
+	id: number;
+	name: string;
+}
+
+export interface ICtsQueryQuery {
+	params: CtsQueryParams;
+	result: ICtsQueryResult;
+}
+
+export type CtsFunctionQueryParams = [number];
+
+export interface ICtsFunctionQueryResult {
+	id: number;
+	name: string;
+}
+
+export interface ICtsFunctionQueryQuery {
+	params: CtsFunctionQueryParams;
+	result: ICtsFunctionQueryResult;
+}

tests/demo/file_extensions/queries-cts.snapshot.ts

@@ -0,0 +1,24 @@
+export type CtsQueryParams = [string | null];
+
+export interface ICtsQueryResult {
+	id: number;
+	name: string;
+}
+
+export interface ICtsQueryQuery {
+	params: CtsQueryParams;
+	result: ICtsQueryResult;
+}
+
+export type CtsFunctionQueryParams = [number];
+
+export interface ICtsFunctionQueryResult {
+	id: number;
+	name: string;
+}
+
+export interface ICtsFunctionQueryQuery {
+	params: CtsFunctionQueryParams;
+	result: ICtsFunctionQueryResult;
+}
+