From 8e5cc66dae10bbb9646303fa343cfd53fec1a3de Mon Sep 17 00:00:00 2001
From: shamoon <4887959+shamoon@users.noreply.github.com>
Date: Sun, 8 Sep 2024 11:06:54 -0700
Subject: [PATCH] Basic doclink contains ui but need to fix the dropdown

---
 ...ustom-fields-query-dropdown.component.html |  4 +++
 src-ui/src/app/data/custom-field-query.ts     | 25 +++++++++++--------
 2 files changed, 19 insertions(+), 10 deletions(-)

diff --git a/src-ui/src/app/components/common/custom-fields-query-dropdown/custom-fields-query-dropdown.component.html b/src-ui/src/app/components/common/custom-fields-query-dropdown/custom-fields-query-dropdown.component.html
index fc1c44727..fc0fae824 100644
--- a/src-ui/src/app/components/common/custom-fields-query-dropdown/custom-fields-query-dropdown.component.html
+++ b/src-ui/src/app/components/common/custom-fields-query-dropdown/custom-fields-query-dropdown.component.html
@@ -78,6 +78,10 @@
       @case (CustomFieldQueryOperator.LessThan) {
         <ng-container *ngTemplateOutlet="comparisonValueTemplate; context: { atom: atom }"></ng-container>
       }
+      @case (CustomFieldQueryOperator.Contains) {
+        <!-- TODO: needs visual work -->
+        <pngx-input-document-link [(ngModel)]="atom.value" [horizontal]="true"></pngx-input-document-link>
+      }
       @default {
         <input class="w-25 form-control rounded-end" type="text" [(ngModel)]="atom.value" [disabled]="disabled">
       }
diff --git a/src-ui/src/app/data/custom-field-query.ts b/src-ui/src/app/data/custom-field-query.ts
index c01d3297e..5a9a98b4d 100644
--- a/src-ui/src/app/data/custom-field-query.ts
+++ b/src-ui/src/app/data/custom-field-query.ts
@@ -115,10 +115,8 @@ export const CUSTOM_FIELD_QUERY_VALUE_TYPES_BY_OPERATOR = {
   [CustomFieldQueryOperator.LessThanOrEqual]: 'string|number',
   [CustomFieldQueryOperator.GreaterThan]: 'number',
   [CustomFieldQueryOperator.LessThan]: 'number',
-  // TODO: Implement these
-  // [CustomFieldQueryOperator.In]: 'array',
-  // [CustomFieldQueryOperator.Contains]: 'string',
-  // [CustomFieldQueryOperator.Range]: 'array',
+  [CustomFieldQueryOperator.Contains]: 'array',
+  // [CustomFieldQueryOperator.In]: 'array', // TODO: Implement
 }
 
 export const CUSTOM_FIELD_QUERY_MAX_DEPTH = 4
@@ -132,7 +130,9 @@ export enum CustomFieldQueryElementType {
 export class CustomFieldQueryElement {
   public readonly type: CustomFieldQueryElementType
   public changed: Subject<CustomFieldQueryElement>
-  protected valueModelChanged: Subject<string | CustomFieldQueryElement[]>
+  protected valueModelChanged: Subject<
+    string | string[] | number[] | CustomFieldQueryElement[]
+  >
   public depth: number = 0
   public id: string = uuidv4()
 
@@ -163,12 +163,15 @@ export class CustomFieldQueryElement {
     return this._operator
   }
 
-  protected _value: string | CustomFieldQueryElement[] = null
-  public set value(value: string | CustomFieldQueryElement[]) {
+  protected _value: string | string[] | number[] | CustomFieldQueryElement[] =
+    null
+  public set value(
+    value: string | string[] | number[] | CustomFieldQueryElement[]
+  ) {
     this._value = value
     this.valueModelChanged.next(value)
   }
-  public get value(): string | CustomFieldQueryElement[] {
+  public get value(): string | string[] | number[] | CustomFieldQueryElement[] {
     return this._value
   }
 }
@@ -205,7 +208,9 @@ export class CustomFieldQueryAtom extends CustomFieldQueryElement {
               this.value = null
             }
             break
-          // TODO: Implement all
+          case 'array':
+            this.value = []
+            break
           default:
             this.value = null
             break
@@ -249,7 +254,7 @@ export class CustomFieldQueryAtom extends CustomFieldQueryElement {
   }
 
   public override serialize() {
-    return [this._field, this._operator, this._value.toString()]
+    return [this._field, this._operator, this._value]
   }
 }