kysely
Version:
Type safe SQL query builder
68 lines (67 loc) • 2.8 kB
JavaScript
/// <reference types="./order-by-parser.d.ts" />
import { isDynamicReferenceBuilder, } from '../dynamic/dynamic-reference-builder.js';
import { isExpression } from '../expression/expression.js';
import { OrderByItemNode } from '../operation-node/order-by-item-node.js';
import { RawNode } from '../operation-node/raw-node.js';
import { OrderByItemBuilder } from '../query-builder/order-by-item-builder.js';
import { logOnce } from '../util/log-once.js';
import { isExpressionOrFactory, parseExpression, } from './expression-parser.js';
import { parseStringReference, } from './reference-parser.js';
export function isOrderByDirection(thing) {
return thing === 'asc' || thing === 'desc';
}
export function parseOrderBy(args) {
if (args.length === 2) {
return [parseOrderByItem(args[0], args[1])];
}
if (args.length === 1) {
const [orderBy] = args;
if (Array.isArray(orderBy)) {
logOnce('orderBy(array) is deprecated, use multiple orderBy calls instead.');
return orderBy.map((item) => parseOrderByItem(item));
}
return [parseOrderByItem(orderBy)];
}
throw new Error(`Invalid number of arguments at order by! expected 1-2, received ${args.length}`);
}
export function parseOrderByItem(expr, modifiers) {
const parsedRef = parseOrderByExpression(expr);
if (OrderByItemNode.is(parsedRef)) {
if (modifiers) {
throw new Error('Cannot specify direction twice!');
}
return parsedRef;
}
return parseOrderByWithModifiers(parsedRef, modifiers);
}
function parseOrderByExpression(expr) {
if (isExpressionOrFactory(expr)) {
return parseExpression(expr);
}
if (isDynamicReferenceBuilder(expr)) {
return expr.toOperationNode();
}
const [ref, direction] = expr.split(' ');
if (direction) {
logOnce("`orderBy('column asc')` is deprecated. Use `orderBy('column', 'asc')` instead.");
return parseOrderByWithModifiers(parseStringReference(ref), direction);
}
return parseStringReference(expr);
}
function parseOrderByWithModifiers(expr, modifiers) {
if (typeof modifiers === 'string') {
if (!isOrderByDirection(modifiers)) {
throw new Error(`Invalid order by direction: ${modifiers}`);
}
return OrderByItemNode.create(expr, RawNode.createWithSql(modifiers));
}
if (isExpression(modifiers)) {
logOnce("`orderBy(..., expr)` is deprecated. Use `orderBy(..., 'asc')` or `orderBy(..., (ob) => ...)` instead.");
return OrderByItemNode.create(expr, modifiers.toOperationNode());
}
const node = OrderByItemNode.create(expr);
if (!modifiers) {
return node;
}
return modifiers(new OrderByItemBuilder({ node })).toOperationNode();
}