src/common/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, 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 getFilters() {
  35         return this.filters;
  36     }
  37
  38     private addCondition(field: string, cond: string, value?: string | string[], prefix: string = "", postfix: string = "", resourcePrefix?: string) {
  39         if (value) {
  40             value = typeof value === "string"
  41                 ? `"${prefix}${value}${postfix}"`
  42                 : `["${value.join(`","`)}"]`;
  43
  44             const resPrefix = resourcePrefix
  45                 ? _.snakeCase(resourcePrefix) + "."
  46                 : "";
  47
  48             this.filters += `${this.filters ? "," : ""}["${resPrefix}${_.snakeCase(field)}","${cond}",${value}]`;
  49         }
  50         return this;
  51     }
  52 }