@rivet-health/design-system 41.1.1 → 41.1.2

package/fesm2022/rivet-health-design-system.mjs

@@ -651,8 +651,9 @@ i0.ɵɵngDeclareClassMetadata({
 	}]
 });
 var LockScrollDirective = class LockScrollDirective {
-	constructor(context) {
+	constructor(context, documentRef) {
 		this.context = context;
+		this.documentRef = documentRef;
 		this.lock = false;
 	}
 	ngOnInit() {
@@ -670,16 +671,20 @@ var LockScrollDirective = class LockScrollDirective {
 		}
 	}
 	lockScroll() {
-		this.context.scrollPosition = document.documentElement.scrollTop;
-		document.body.style.position = "fixed";
-		document.body.style.overflow = "hidden";
-		document.body.style.width = "100%";
-		document.body.style.height = "100%";
-		document.body.style.top = `${this.context.scrollPosition * -1}px`;
+		const body = this.documentRef.body;
+		if (!body) return;
+		this.context.scrollPosition = this.documentRef.documentElement.scrollTop;
+		body.style.position = "fixed";
+		body.style.overflow = "hidden";
+		body.style.width = "100%";
+		body.style.height = "100%";
+		body.style.top = `${this.context.scrollPosition * -1}px`;
 	}
 	unlockScroll() {
-		document.body.removeAttribute("style");
-		document.documentElement.scrollTop = this.context.scrollPosition;
+		const body = this.documentRef.body;
+		if (!body) return;
+		body.removeAttribute("style");
+		this.documentRef.documentElement.scrollTop = this.context.scrollPosition;
 	}
 	static {
 		this.ɵfac = i0.ɵɵngDeclareFactory({
@@ -687,7 +692,7 @@ var LockScrollDirective = class LockScrollDirective {
 			version: "22.2.4",
 			ngImport: i0,
 			type: LockScrollDirective,
-			deps: [{ token: LockScrollContextService }],
+			deps: [{ token: LockScrollContextService }, { token: DOCUMENT }],
 			target: i0.ɵɵFactoryTarget.Directive
 		});
 	}
@@ -715,7 +720,13 @@ i0.ɵɵngDeclareClassMetadata({
 			standalone: false
 		}]
 	}],
-	ctorParameters: () => [{ type: LockScrollContextService }],
+	ctorParameters: () => [{ type: LockScrollContextService }, {
+		type: Document,
+		decorators: [{
+			type: Inject,
+			args: [DOCUMENT]
+		}]
+	}],
 	propDecorators: { lock: [{
 		type: Input,
 		args: ["rivLockScroll"]