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 export function joinFilters(...filters: string[]) {
   6     return filters.filter(s => s).join(",");
   7 }
   8
   9 export class FilterBuilder {
  10     constructor(private filters = "") { }
  11
  12     public addEqual(field: string, value?: string | string[] | boolean | null, resourcePrefix?: string) {
  13         return this.addCondition(field, "=", value, "", "", resourcePrefix);
  14     }
  15
  16     public addDistinct(field: string, value?: string | boolean | null, resourcePrefix?: string) {
  17         return this.addCondition(field, "!=", value, "", "", resourcePrefix);
  18     }
  19
  20     public addLike(field: string, value?: string, resourcePrefix?: string) {
  21         return this.addCondition(field, "like", value, "%", "%", resourcePrefix);
  22     }
  23
  24     public addILike(field: string, value?: string, resourcePrefix?: string) {
  25         return this.addCondition(field, "ilike", value, "%", "%", resourcePrefix);
  26     }
  27
  28     public addContains(field: string, value?: string, resourcePrefix?: string) {
  29         return this.addCondition(field, "contains", value, "", "", resourcePrefix);
  30     }
  31
  32     public addIsA(field: string, value?: string | string[], resourcePrefix?: string) {
  33         return this.addCondition(field, "is_a", value, "", "", resourcePrefix);
  34     }
  35
  36     public addIn(field: string, value?: string | string[], resourcePrefix?: string) {
  37         return this.addCondition(field, "in", value, "", "", resourcePrefix);
  38     }
  39
  40     public addNotIn(field: string, value?: string | string[], resourcePrefix?: string) {
  41         return this.addCondition(field, "not in", value, "", "", resourcePrefix);
  42     }
  43
  44     public addGt(field: string, value?: string, resourcePrefix?: string) {
  45         return this.addCondition(field, ">", value, "", "", resourcePrefix);
  46     }
  47
  48     public addGte(field: string, value?: string, resourcePrefix?: string) {
  49         return this.addCondition(field, ">=", value, "", "", resourcePrefix);
  50     }
  51
  52     public addLt(field: string, value?: string, resourcePrefix?: string) {
  53         return this.addCondition(field, "<", value, "", "", resourcePrefix);
  54     }
  55
  56     public addLte(field: string, value?: string, resourcePrefix?: string) {
  57         return this.addCondition(field, "<=", value, "", "", resourcePrefix);
  58     }
  59
  60     public addExists(value?: string, resourcePrefix?: string) {
  61         return this.addCondition("properties", "exists", value, "", "", resourcePrefix);
  62     }
  63     public addDoesNotExist(field: string, resourcePrefix?: string) {
  64         return this.addCondition("properties." + field, "exists", false, "", "", resourcePrefix);
  65     }
  66
  67     public addFullTextSearch(value: string, table?: string) {
  68         const regex = /"[^"]*"/;
  69         const matches: any[] = [];
  70
  71         let match = value.match(regex);
  72
  73         while (match) {
  74             value = value.replace(match[0], "");
  75             matches.push(match[0].replace(/"/g, ''));
  76             match = value.match(regex);
  77         }
  78
  79         let searchIn = 'any';
  80         if (table) {
  81             searchIn = table + ".any";
  82         }
  83
  84         const terms = value.trim().split(/(\s+)/).concat(matches);
  85         terms.forEach(term => {
  86             if (term !== " ") {
  87                 this.addCondition(searchIn, "ilike", term, "%", "%");
  88             }
  89         });
  90         return this;
  91     }
  92
  93     public getFilters() {
  94         return this.filters;
  95     }
  96
  97     private addCondition(field: string, cond: string, value?: string | string[] | boolean | null, prefix: string = "", postfix: string = "", resourcePrefix?: string) {
  98         if (value !== undefined) {
  99             if (typeof value === "string") {
 100                 value = `"${prefix}${value}${postfix}"`;
 101             } else if (Array.isArray(value)) {
 102                 value = `["${value.join(`","`)}"]`;
 103             } else if (value !== null) {
 104                 value = value ? "true" : "false";
 105             }
 106
 107             const resPrefix = resourcePrefix
 108                 ? resourcePrefix + "."
 109                 : "";
 110
 111             this.filters += `${this.filters ? "," : ""}["${resPrefix}${field}","${cond}",${value}]`;
 112         }
 113         return this;
 114     }
 115 }