metal-orm 1.0.0

package/tests/window-function.test.ts

@@ -0,0 +1,137 @@
+import { describe, it, expect } from 'vitest';
+import { SelectQueryBuilder } from '../src/builder/select';
+import { SqliteDialect } from '../src/dialect/sqlite';
+import { MySqlDialect } from '../src/dialect/mysql';
+import { SqlServerDialect } from '../src/dialect/mssql';
+import { TableDef } from '../src/schema/table';
+import { rowNumber, rank, denseRank, lag, lead, ntile, firstValue, lastValue, windowFunction } from '../src/ast/expression';
+
+const table = (name: string): TableDef => ({ name, columns: {}, relations: {} });
+const col = (name: string, table?: string) => ({ type: 'Column', name, table: table || 'unknown' } as any);
+const lit = (value: any) => ({ type: 'Literal', value } as any);
+
+describe('Window Function Support', () => {
+    const sqlite = new SqliteDialect();
+    const mysql = new MySqlDialect();
+    const mssql = new SqlServerDialect();
+
+    it('should generate ROW_NUMBER() window function', () => {
+        const users = table('users');
+
+        const query = new SelectQueryBuilder(users)
+            .select({
+                id: col('id', 'users'),
+                name: col('name', 'users'),
+                row_num: rowNumber()
+            });
+
+        const expectedSqlite = 'SELECT "users"."id" AS "id", "users"."name" AS "name", ROW_NUMBER() OVER () AS "row_num" FROM "users";';
+        const expectedMysql = 'SELECT `users`.`id` AS `id`, `users`.`name` AS `name`, ROW_NUMBER() OVER () AS `row_num` FROM `users`;';
+        const expectedMssql = 'SELECT [users].[id] AS [id], [users].[name] AS [name], ROW_NUMBER() OVER () AS [row_num] FROM [users];';
+
+        expect(query.toSql(sqlite)).toBe(expectedSqlite);
+        expect(query.toSql(mysql)).toBe(expectedMysql);
+        expect(query.toSql(mssql)).toBe(expectedMssql);
+    });
+
+    it('should generate RANK() with PARTITION BY and ORDER BY', () => {
+        const orders = table('orders');
+
+        const query = new SelectQueryBuilder(orders)
+            .select({
+                id: col('id', 'orders'),
+                customer_id: col('customer_id', 'orders'),
+                amount: col('amount', 'orders'),
+                rank: windowFunction('RANK', [], [col('customer_id', 'orders')], [{ column: col('amount', 'orders'), direction: 'DESC' }])
+            });
+
+        const expectedSqlite = 'SELECT "orders"."id" AS "id", "orders"."customer_id" AS "customer_id", "orders"."amount" AS "amount", RANK() OVER (PARTITION BY "orders"."customer_id" ORDER BY "orders"."amount" DESC) AS "rank" FROM "orders";';
+        const expectedMysql = 'SELECT `orders`.`id` AS `id`, `orders`.`customer_id` AS `customer_id`, `orders`.`amount` AS `amount`, RANK() OVER (PARTITION BY `orders`.`customer_id` ORDER BY `orders`.`amount` DESC) AS `rank` FROM `orders`;';
+        const expectedMssql = 'SELECT [orders].[id] AS [id], [orders].[customer_id] AS [customer_id], [orders].[amount] AS [amount], RANK() OVER (PARTITION BY [orders].[customer_id] ORDER BY [orders].[amount] DESC) AS [rank] FROM [orders];';
+
+        expect(query.toSql(sqlite)).toBe(expectedSqlite);
+        expect(query.toSql(mysql)).toBe(expectedMysql);
+        expect(query.toSql(mssql)).toBe(expectedMssql);
+    });
+
+    it('should generate LAG function with offset and default value', () => {
+        const sales = table('sales');
+
+        const query = new SelectQueryBuilder(sales)
+            .select({
+                date: col('date', 'sales'),
+                amount: col('amount', 'sales'),
+                prev_amount: lag(col('amount', 'sales'), 1, 0)
+            });
+
+        const expectedSqlite = 'SELECT "sales"."date" AS "date", "sales"."amount" AS "amount", LAG("sales"."amount", ?, ?) OVER () AS "prev_amount" FROM "sales";';
+        const expectedMysql = 'SELECT `sales`.`date` AS `date`, `sales`.`amount` AS `amount`, LAG(`sales`.`amount`, ?, ?) OVER () AS `prev_amount` FROM `sales`;';
+        const expectedMssql = 'SELECT [sales].[date] AS [date], [sales].[amount] AS [amount], LAG([sales].[amount], @p1, @p2) OVER () AS [prev_amount] FROM [sales];';
+
+        expect(query.toSql(sqlite)).toBe(expectedSqlite);
+        expect(query.toSql(mysql)).toBe(expectedMysql);
+        expect(query.toSql(mssql)).toBe(expectedMssql);
+    });
+
+    it('should generate LEAD function', () => {
+        const sales = table('sales');
+
+        const query = new SelectQueryBuilder(sales)
+            .select({
+                date: col('date', 'sales'),
+                amount: col('amount', 'sales'),
+                next_amount: lead(col('amount', 'sales'), 1)
+            });
+
+        const expectedSqlite = 'SELECT "sales"."date" AS "date", "sales"."amount" AS "amount", LEAD("sales"."amount", ?) OVER () AS "next_amount" FROM "sales";';
+        const expectedMysql = 'SELECT `sales`.`date` AS `date`, `sales`.`amount` AS `amount`, LEAD(`sales`.`amount`, ?) OVER () AS `next_amount` FROM `sales`;';
+        const expectedMssql = 'SELECT [sales].[date] AS [date], [sales].[amount] AS [amount], LEAD([sales].[amount], @p1) OVER () AS [next_amount] FROM [sales];';
+
+        expect(query.toSql(sqlite)).toBe(expectedSqlite);
+        expect(query.toSql(mysql)).toBe(expectedMysql);
+        expect(query.toSql(mssql)).toBe(expectedMssql);
+    });
+
+    it('should generate window function with both PARTITION BY and ORDER BY', () => {
+        const employees = table('employees');
+
+        const query = new SelectQueryBuilder(employees)
+            .select({
+                id: col('id', 'employees'),
+                name: col('name', 'employees'),
+                department: col('department', 'employees'),
+                salary: col('salary', 'employees'),
+                dept_rank: windowFunction('ROW_NUMBER', [], [col('department', 'employees')], [{ column: col('salary', 'employees'), direction: 'DESC' }])
+            });
+
+        const expectedSqlite = 'SELECT "employees"."id" AS "id", "employees"."name" AS "name", "employees"."department" AS "department", "employees"."salary" AS "salary", ROW_NUMBER() OVER (PARTITION BY "employees"."department" ORDER BY "employees"."salary" DESC) AS "dept_rank" FROM "employees";';
+        const expectedMysql = 'SELECT `employees`.`id` AS `id`, `employees`.`name` AS `name`, `employees`.`department` AS `department`, `employees`.`salary` AS `salary`, ROW_NUMBER() OVER (PARTITION BY `employees`.`department` ORDER BY `employees`.`salary` DESC) AS `dept_rank` FROM `employees`;';
+        const expectedMssql = 'SELECT [employees].[id] AS [id], [employees].[name] AS [name], [employees].[department] AS [department], [employees].[salary] AS [salary], ROW_NUMBER() OVER (PARTITION BY [employees].[department] ORDER BY [employees].[salary] DESC) AS [dept_rank] FROM [employees];';
+
+        expect(query.toSql(sqlite)).toBe(expectedSqlite);
+        expect(query.toSql(mysql)).toBe(expectedMysql);
+        expect(query.toSql(mssql)).toBe(expectedMssql);
+    });
+
+    it('should generate multiple window functions in one query', () => {
+        const employees = table('employees');
+
+        const query = new SelectQueryBuilder(employees)
+            .select({
+                id: col('id', 'employees'),
+                name: col('name', 'employees'),
+                salary: col('salary', 'employees'),
+                row_num: rowNumber(),
+                rank: rank(),
+                dense_rank: denseRank()
+            });
+
+        const expectedSqlite = 'SELECT "employees"."id" AS "id", "employees"."name" AS "name", "employees"."salary" AS "salary", ROW_NUMBER() OVER () AS "row_num", RANK() OVER () AS "rank", DENSE_RANK() OVER () AS "dense_rank" FROM "employees";';
+        const expectedMysql = 'SELECT `employees`.`id` AS `id`, `employees`.`name` AS `name`, `employees`.`salary` AS `salary`, ROW_NUMBER() OVER () AS `row_num`, RANK() OVER () AS `rank`, DENSE_RANK() OVER () AS `dense_rank` FROM `employees`;';
+        const expectedMssql = 'SELECT [employees].[id] AS [id], [employees].[name] AS [name], [employees].[salary] AS [salary], ROW_NUMBER() OVER () AS [row_num], RANK() OVER () AS [rank], DENSE_RANK() OVER () AS [dense_rank] FROM [employees];';
+
+        expect(query.toSql(sqlite)).toBe(expectedSqlite);
+        expect(query.toSql(mysql)).toBe(expectedMysql);
+        expect(query.toSql(mssql)).toBe(expectedMssql);
+    });
+});

package/tsconfig.json

@@ -0,0 +1,30 @@
+{
+  "compilerOptions": {
+    "target": "ES2022",
+    "experimentalDecorators": true,
+    "useDefineForClassFields": false,
+    "module": "ESNext",
+    "lib": [
+      "DOM",
+      "DOM.Iterable",
+      "ES2022"
+    ],
+    "jsx": "react-jsx",
+    "skipLibCheck": true,
+    "types": [
+      "node"
+    ],
+    "moduleResolution": "bundler",
+    "isolatedModules": true,
+    "moduleDetection": "force",
+    "allowJs": true,
+    "baseUrl": "./src",
+    "paths": {
+      "@orm/*": [
+        "./*"
+      ]
+    },
+    "allowImportingTsExtensions": true,
+    "noEmit": true
+  }
+}

package/tsup.config.ts

@@ -0,0 +1,10 @@
+import { defineConfig } from 'tsup';
+
+export default defineConfig({
+    entry: ['src/index.ts'],
+    format: ['cjs', 'esm'],
+    dts: true,
+    splitting: false,
+    sourcemap: true,
+    clean: true,
+});

package/vite.config.ts

@@ -0,0 +1,22 @@
+import 'tsconfig-paths/register';
+import { defineConfig } from 'vite';
+import react from '@vitejs/plugin-react';
+import path from 'path';
+import { fileURLToPath } from 'url';
+import { playgroundApiPlugin } from './playground/api/playground-api';
+
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
+
+export default defineConfig({
+    plugins: [
+        react(),
+        playgroundApiPlugin(),
+    ],
+    resolve: {
+        alias: {
+            '@orm': path.resolve(__dirname, './src'),
+            'playground': path.resolve(__dirname, './playground'),
+        },
+    },
+    root: './playground',
+});

package/vitest.config.ts

@@ -0,0 +1,14 @@
+import { defineConfig } from 'vitest/config';
+import path from 'path';
+
+export default defineConfig({
+    test: {
+        globals: true,
+        environment: 'node',
+    },
+    resolve: {
+        alias: {
+            '@': path.resolve(__dirname, './src'),
+        },
+    },
+});