minor: added firestoreDateBlockRange()

dereekb · Aug 10, 2022 · 67f2dba · 67f2dba
1 parent b5a16ec
commit 67f2dba
Showing 1 changed file with 32 additions and 1 deletion.
diff --git a/packages/firebase/src/lib/common/firestore/snapshot/snapshot.field.ts b/packages/firebase/src/lib/common/firestore/snapshot/snapshot.field.ts
@@ -1,6 +1,6 @@
 import { UNKNOWN_WEBSITE_LINK_TYPE, WebsiteLink, GrantedRole } from '@dereekb/model';
 import { FirestoreModelKey } from '../collection/collection';
-import { nowISODateString, toISODateString, toJsDate } from '@dereekb/date';
+import { DateBlockRange, nowISODateString, toISODateString, toJsDate } from '@dereekb/date';
 import {
   ModelFieldMapFunctionsConfig,
   GetterOrValue,
@@ -582,6 +582,37 @@ export function firestoreWebsiteLinkArray() {
   });
 }
 
+// MARK: DateBlockRange
+export const DEFAULT_DATE_BLOCK_RANGE: DateBlockRange = {
+  i: 0
+};
+
+export const assignDateBlockRangeFunction = assignValuesToPOJOFunction<DateBlockRange>({ keysFilter: ['i', 'to'], valueFilter: KeyValueTypleValueFilter.NULL });
+export const firestoreDateBlockRangeAssignFn: MapFunction<DateBlockRange, DateBlockRange> = (input) => {
+  const block = assignDateBlockRangeFunction(DEFAULT_DATE_BLOCK_RANGE, input);
+
+  if (block.to == null) {
+    block.to = block.i;
+  }
+
+  return block;
+};
+
+export function firestoreDateBlockRange() {
+  return firestoreField<DateBlockRange, DateBlockRange>({
+    default: DEFAULT_DATE_BLOCK_RANGE,
+    fromData: firestoreDateBlockRangeAssignFn,
+    toData: firestoreDateBlockRangeAssignFn
+  });
+}
+
+// MARK: DateBlockRange Array
+export function firestoreDateBlockRangeArray() {
+  return firestoreObjectArray({
+    firestoreField: firestoreDateBlockRange()
+  });
+}
+
 // MARK: Deprecated
 export type FirestoreSetFieldConfig<T extends string | number> = DefaultMapConfiguredFirestoreFieldConfig<Set<T>, T[]>;