68 lines
2.7 KiB
JavaScript
68 lines
2.7 KiB
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();
|
|
}
|