@real-router/angular 0.2.2 → 0.4.0

package/src/providers.ts

@@ -1,13 +1,18 @@
 import {
+  DestroyRef,
   InjectionToken,
+  inject,
   makeEnvironmentProviders,
+  provideEnvironmentInitializer,
   type EnvironmentProviders,
 } from "@angular/core";
 import { getNavigator, type Router, type Navigator } from "@real-router/core";
 import { createRouteSource } from "@real-router/sources";
 
+import { createScrollRestoration } from "./dom-utils";
 import { sourceToSignal } from "./sourceToSignal";
 
+import type { ScrollRestorationOptions } from "./dom-utils";
 import type { RouteSignals } from "./types";
 
 export const ROUTER = new InjectionToken<Router>("ROUTER");
@@ -16,10 +21,17 @@ export const NAVIGATOR = new InjectionToken<Navigator>("NAVIGATOR");
 
 export const ROUTE = new InjectionToken<RouteSignals>("ROUTE");
 
-export function provideRealRouter(router: Router): EnvironmentProviders {
+export interface RealRouterOptions {
+  scrollRestoration?: ScrollRestorationOptions;
+}
+
+export function provideRealRouter(
+  router: Router,
+  options?: RealRouterOptions,
+): EnvironmentProviders {
   const navigator = getNavigator(router);
 
-  return makeEnvironmentProviders([
+  const providers: Parameters<typeof makeEnvironmentProviders>[0] = [
     { provide: ROUTER, useValue: router },
     { provide: NAVIGATOR, useValue: navigator },
     {
@@ -29,5 +41,21 @@ export function provideRealRouter(router: Router): EnvironmentProviders {
         navigator,
       }),
     },
-  ]);
+  ];
+
+  if (options?.scrollRestoration) {
+    const scrollOpts = options.scrollRestoration;
+
+    providers.push(
+      provideEnvironmentInitializer(() => {
+        const sr = createScrollRestoration(router, scrollOpts);
+
+        inject(DestroyRef).onDestroy(() => {
+          sr.destroy();
+        });
+      }),
+    );
+  }
+
+  return makeEnvironmentProviders(providers);
 }