mirror of
https://github.com/directus/directus.git
synced 2026-01-30 19:18:04 -05:00
43 lines
1.2 KiB
TypeScript
43 lines
1.2 KiB
TypeScript
import { Knex } from 'knex';
|
|
import { FunctionsHelper } from '../database/functions';
|
|
import { REGEX_BETWEEN_PARENS } from '@directus/shared/constants';
|
|
import { applyFunctionToColumnName } from './apply-function-to-column-name';
|
|
|
|
/**
|
|
* Return column prefixed by table. If column includes functions (like `year(date_created)`, the
|
|
* column is replaced with the appropriate SQL)
|
|
*
|
|
* @param knex Current knex / transaction instance
|
|
* @param collection Collection or alias in which column resides
|
|
* @param field name of the column
|
|
* @param alias Whether or not to add a SQL AS statement
|
|
* @returns Knex raw instance
|
|
*/
|
|
export function getColumn(
|
|
knex: Knex,
|
|
table: string,
|
|
column: string,
|
|
alias = applyFunctionToColumnName(column)
|
|
): Knex.Raw {
|
|
const fn = FunctionsHelper(knex);
|
|
|
|
if (column.includes('(') && column.includes(')')) {
|
|
const functionName = column.split('(')[0];
|
|
const columnName = column.match(REGEX_BETWEEN_PARENS)![1];
|
|
|
|
if (functionName in fn) {
|
|
const result = fn[functionName as keyof typeof fn](table, columnName);
|
|
|
|
if (alias) {
|
|
return knex.raw(result + ' AS ??', [alias]);
|
|
}
|
|
|
|
return result;
|
|
} else {
|
|
throw new Error(`Invalid function specified "${functionName}"`);
|
|
}
|
|
}
|
|
|
|
return knex.raw('??.??', [table, column]);
|
|
}
|