change inline if to switch case

Khan291 · Jun 13, 2022 · 9981812 · 9981812
1 parent f75cfa9
commit 9981812
Showing 1 changed file with 16 additions and 8 deletions.
diff --git a/npm/ng-packs/packages/theme-shared/extensions/src/lib/pipes/create-injector.pipe.ts b/npm/ng-packs/packages/theme-shared/extensions/src/lib/pipes/create-injector.pipe.ts
@@ -13,14 +13,22 @@ export class CreateInjectorPipe<R> implements PipeTransform {
   ): Injector {
     const get = <T>(token: Type<T> | InjectionToken<T>, notFoundValue?: T, flags?: InjectFlags) => {
       const componentData = context.getData();
-      return token === EXTENSIONS_ACTION_DATA
-        ? componentData
-        : token === EXTENSIONS_ACTION_CALLBACK
-        ? data => {
-            data = data ?? context.getData();
-            return action.action(data);
-          }
-        : context.getInjected.call(context.injector, token, notFoundValue, flags);
+      const componentDataCallback = data => {
+        data = data ?? context.getData();
+        return action.action(data);
+      };
+      let extensionData;
+      switch (token) {
+        case EXTENSIONS_ACTION_DATA:
+          extensionData = componentData;
+          break;
+        case EXTENSIONS_ACTION_CALLBACK:
+          extensionData = componentDataCallback;
+          break;
+        default:
+          extensionData = context.getInjected.call(context.injector, token, notFoundValue, flags);
+      }
+      return extensionData;
     };
     return { get };
   }