import { Model, Sequelize } from 'sequelize-typescript'
import validator from 'validator'
import { Col } from 'sequelize/types/lib/utils'
-import { literal, OrderItem } from 'sequelize'
+import { literal, OrderItem, Op } from 'sequelize'
+
+type Primitive = string | Function | number | boolean | Symbol | undefined | null
+type DeepOmitHelper<T, K extends keyof T> = {
+ [P in K]: // extra level of indirection needed to trigger homomorhic behavior
+ T[P] extends infer TP // distribute over unions
+ ? TP extends Primitive
+ ? TP // leave primitives and functions alone
+ : TP extends any[]
+ ? DeepOmitArray<TP, K> // Array special handling
+ : DeepOmit<TP, K>
+ : never
+}
+type DeepOmit<T, K> = T extends Primitive ? T : DeepOmitHelper<T, Exclude<keyof T, K>>
+
+type DeepOmitArray<T extends any[], K> = {
+ [P in keyof T]: DeepOmit<T[P], K>
+}
type SortType = { sortModel: string, sortValue: string }
}
const createSafeIn = (model: typeof Model, stringArr: (string | number)[]) => {
- return stringArr.map(t => model.sequelize.escape('' + t))
- .join(', ')
+ return stringArr.map(t => {
+ return t === null
+ ? null
+ : model.sequelize.escape('' + t)
+ }).join(', ')
}
function buildLocalAccountIdsIn () {
)
}
+function buildDirectionAndField (value: string) {
+ let field: string
+ let direction: 'ASC' | 'DESC'
+
+ if (value.substring(0, 1) === '-') {
+ direction = 'DESC'
+ field = value.substring(1)
+ } else {
+ direction = 'ASC'
+ field = value
+ }
+
+ return { direction, field }
+}
+
+function searchAttribute (sourceField, targetField) {
+ if (sourceField) {
+ return {
+ [targetField]: {
+ [Op.iLike]: `%${sourceField}%`
+ }
+ }
+ } else {
+ return {}
+ }
+}
+
+interface QueryStringFilterPrefixes {
+ [key: string]: string | { prefix: string, handler: Function, multiple?: boolean }
+}
+
+function parseQueryStringFilter (q: string, prefixes: QueryStringFilterPrefixes): {
+ search: string
+ [key: string]: string | number | string[] | number[]
+} {
+ const tokens = q // tokenize only if we have a querystring
+ ? [].concat.apply([], q.split('"').map((v, i) => i % 2 ? v : v.split(' '))).filter(Boolean) // split by space unless using double quotes
+ : []
+
+ const objectMap = (obj, fn) => Object.fromEntries(
+ Object.entries(obj).map(
+ ([ k, v ], i) => [ k, fn(v, k, i) ]
+ )
+ )
+
+ return {
+ // search is the querystring minus defined filters
+ search: tokens.filter(e => !Object.values(prefixes).some(p => {
+ if (typeof p === 'string') {
+ return e.startsWith(p)
+ } else {
+ return e.startsWith(p.prefix)
+ }
+ })).join(' '),
+ // filters defined in prefixes are added under their own name
+ ...objectMap(prefixes, p => {
+ if (typeof p === 'string') {
+ return tokens.filter(e => e.startsWith(p)).map(e => e.slice(p.length)) // we keep the matched item, and remove its prefix
+ } else {
+ const _tokens = tokens.filter(e => e.startsWith(p.prefix)).map(e => e.slice(p.prefix.length)).map(p.handler)
+ // multiple is false by default, meaning we usually just keep the first occurence of a given prefix
+ if (!p.multiple && _tokens.length > 0) {
+ return _tokens[0]
+ } else if (!p.multiple) {
+ return ''
+ }
+ return _tokens
+ }
+ })
+ }
+}
+
// ---------------------------------------------------------------------------
export {
+ DeepOmit,
buildBlockedAccountSQL,
buildLocalActorIdsIn,
SortType,
isOutdated,
parseAggregateResult,
getFollowsSort,
- createSafeIn
+ buildDirectionAndField,
+ createSafeIn,
+ searchAttribute,
+ parseQueryStringFilter
}
// ---------------------------------------------------------------------------
function searchTrigramNormalizeCol (col: string) {
return Sequelize.fn('lower', Sequelize.fn('immutable_unaccent', Sequelize.col(col)))
}
-
-function buildDirectionAndField (value: string) {
- let field: string
- let direction: 'ASC' | 'DESC'
-
- if (value.substring(0, 1) === '-') {
- direction = 'DESC'
- field = value.substring(1)
- } else {
- direction = 'ASC'
- field = value
- }
-
- return { direction, field }
-}