Use fold function for template context

fboeller · Mar 8, 2021 · 1410317 · 1410317
1 parent d8ca804
commit 1410317
Showing 1 changed file with 10 additions and 7 deletions.
diff --git a/projects/ngx-loading-customizer/src/lib/angular/loadable/loadable.component.ts b/projects/ngx-loading-customizer/src/lib/angular/loadable/loadable.component.ts
@@ -11,6 +11,7 @@ import {
   ViewChild,
   ViewContainerRef,
 } from '@angular/core';
+import { fold } from '../../functions/fold.function';
 import { idle } from '../../loadable.constructors';
 import {
   getLoadingState,
@@ -34,13 +35,15 @@ export function setComponentInputs(
 }
 
 function loadableTemplateContext(loadable: Loadable<unknown>): object {
-  if (isLoaded(loadable)) {
-    return { value: loadable.value };
-  } else if (hasErrored(loadable)) {
-    return { error: loadable.error };
-  } else {
-    return {};
-  }
+  return fold(
+    {
+      idle: () => ({}),
+      loading: () => ({}),
+      loaded: (value) => ({ value }),
+      error: (error) => ({ error }),
+    },
+    loadable
+  );
 }
 
 export type TemplateRefs = {