refs #master Merge branch 'origin/master' into 14364-searchbar-arrow-navigation
[arvados-workbench2.git] / src / services / api / filter-builder.ts
1 // Copyright (C) The Arvados Authors. All rights reserved.
2 //
3 // SPDX-License-Identifier: AGPL-3.0
4
5 import * as _ from "lodash";
6
7 export function joinFilters(filters0?: string, filters1?: string) {
8     return [filters0, filters1].filter(s => s).join(",");
9 }
10
11 export class FilterBuilder {
12     constructor(private filters = "") { }
13
14     public addEqual(field: string, value?: string | boolean, resourcePrefix?: string) {
15         return this.addCondition(field, "=", value, "", "", resourcePrefix );
16     }
17
18     public addLike(field: string, value?: string, resourcePrefix?: string) {
19         return this.addCondition(field, "like", value, "%", "%", resourcePrefix);
20     }
21
22     public addILike(field: string, value?: string, resourcePrefix?: string) {
23         return this.addCondition(field, "ilike", value, "%", "%", resourcePrefix);
24     }
25
26     public addIsA(field: string, value?: string | string[], resourcePrefix?: string) {
27         return this.addCondition(field, "is_a", value, "", "", resourcePrefix);
28     }
29
30     public addIn(field: string, value?: string | string[], resourcePrefix?: string) {
31         return this.addCondition(field, "in", value, "", "", resourcePrefix);
32     }
33
34     public addGt(field: string, value?: string, resourcePrefix?: string) {
35         return this.addCondition(field, ">", value, "", "", resourcePrefix);
36     }
37
38     public addGte(field: string, value?: string, resourcePrefix?: string) {
39         return this.addCondition(field, ">=", value, "", "", resourcePrefix);
40     }
41
42     public addLt(field: string, value?: string, resourcePrefix?: string) {
43         return this.addCondition(field, "<", value, "", "", resourcePrefix);
44     }
45
46     public addLte(field: string, value?: string, resourcePrefix?: string) {
47         return this.addCondition(field, "<=", value, "", "", resourcePrefix);
48     }
49
50     public getFilters() {
51         return this.filters;
52     }
53
54     private addCondition(field: string, cond: string, value?: string | string[] | boolean, prefix: string = "", postfix: string = "", resourcePrefix?: string) {
55         if (value) {
56             if (typeof value === "string") {
57                 value = `"${prefix}${value}${postfix}"`;
58             } else if (Array.isArray(value)) {
59                 value = `["${value.join(`","`)}"]`;
60             } else {
61                 value = value ? "true" : "false";
62             }
63
64             const resPrefix = resourcePrefix
65                 ? _.snakeCase(resourcePrefix) + "."
66                 : "";
67
68             this.filters += `${this.filters ? "," : ""}["${resPrefix}${_.snakeCase(field)}","${cond}",${value}]`;
69         }
70         return this;
71     }
72 }