Merge branch 'main' into km/pm-11528/move-lock-to-km

libs/angular/src/auth/functions/unauth-ui-refresh-redirect.spec.ts

@@ -40,15 +40,14 @@ describe("unauthUiRefreshRedirect", () => {
   it("returns UrlTree when UnauthenticatedExtensionUIRefresh flag is enabled and preserves query params", async () => {
     configService.getFeatureFlag.mockResolvedValue(true);
 
-    const queryParams = { test: "test" };
+    const urlTree = new UrlTree();
+    urlTree.queryParams = { test: "test" };
 
     const navigation: Navigation = {
-      extras: {
-        queryParams: queryParams,
-      },
+      extras: {},
       id: 0,
       initialUrl: new UrlTree(),
-      extractedUrl: new UrlTree(),
+      extractedUrl: urlTree,
       trigger: "imperative",
       previousNavigation: undefined,
     };
@@ -60,6 +59,8 @@ describe("unauthUiRefreshRedirect", () => {
     expect(configService.getFeatureFlag).toHaveBeenCalledWith(
       FeatureFlag.UnauthenticatedExtensionUIRefresh,
     );
-    expect(router.createUrlTree).toHaveBeenCalledWith(["/redirect"], { queryParams });
+    expect(router.createUrlTree).toHaveBeenCalledWith(["/redirect"], {
+      queryParams: urlTree.queryParams,
+    });
   });
 });

libs/angular/src/auth/functions/unauth-ui-refresh-redirect.ts

@@ -17,7 +17,7 @@ export function unauthUiRefreshRedirect(redirectUrl: string): () => Promise<bool
     );
     if (shouldRedirect) {
       const currentNavigation = router.getCurrentNavigation();
-      const queryParams = currentNavigation?.extras?.queryParams || {};
+      const queryParams = currentNavigation?.extractedUrl?.queryParams || {};
 
       // Preserve query params when redirecting as it is likely that the refreshed component
       // will be consuming the same query params.

libs/angular/src/auth/functions/unauth-ui-refresh-route-swap.ts

@@ -15,6 +15,7 @@ import { componentRouteSwap } from "../../utils/component-route-swap";
  * @param defaultComponent - The current non-refreshed component to render.
  * @param refreshedComponent - The new refreshed component to render.
  * @param options - The shared route options to apply to both components.
+ * @param altOptions - The alt route options to apply to the alt component. If not provided, the base options will be used.
  */
 export function unauthUiRefreshSwap(
   defaultComponent: Type<any>,

libs/angular/src/directives/text-drag.directive.ts

@@ -17,6 +17,6 @@ export class TextDragDirective {
 
   @HostListener("dragstart", ["$event"])
   onDragStart(event: DragEvent) {
-    event.dataTransfer.setData("text", this.data);
+    event.dataTransfer?.setData("text", this.data);
   }
 }

libs/angular/src/utils/extension-refresh-redirect.spec.ts

@@ -0,0 +1,62 @@
+import { TestBed } from "@angular/core/testing";
+import { Navigation, Router, UrlTree } from "@angular/router";
+import { mock, MockProxy } from "jest-mock-extended";
+
+import { FeatureFlag } from "@bitwarden/common/enums/feature-flag.enum";
+import { ConfigService } from "@bitwarden/common/platform/abstractions/config/config.service";
+
+import { extensionRefreshRedirect } from "./extension-refresh-redirect";
+
+describe("extensionRefreshRedirect", () => {
+  let configService: MockProxy<ConfigService>;
+  let router: MockProxy<Router>;
+
+  beforeEach(() => {
+    configService = mock<ConfigService>();
+    router = mock<Router>();
+
+    TestBed.configureTestingModule({
+      providers: [
+        { provide: ConfigService, useValue: configService },
+        { provide: Router, useValue: router },
+      ],
+    });
+  });
+
+  it("returns true when ExtensionRefresh flag is disabled", async () => {
+    configService.getFeatureFlag.mockResolvedValue(false);
+
+    const result = await TestBed.runInInjectionContext(() =>
+      extensionRefreshRedirect("/redirect")(),
+    );
+
+    expect(result).toBe(true);
+    expect(configService.getFeatureFlag).toHaveBeenCalledWith(FeatureFlag.ExtensionRefresh);
+    expect(router.parseUrl).not.toHaveBeenCalled();
+  });
+
+  it("returns UrlTree when ExtensionRefresh flag is enabled and preserves query params", async () => {
+    configService.getFeatureFlag.mockResolvedValue(true);
+
+    const urlTree = new UrlTree();
+    urlTree.queryParams = { test: "test" };
+
+    const navigation: Navigation = {
+      extras: {},
+      id: 0,
+      initialUrl: new UrlTree(),
+      extractedUrl: urlTree,
+      trigger: "imperative",
+      previousNavigation: undefined,
+    };
+
+    router.getCurrentNavigation.mockReturnValue(navigation);
+
+    await TestBed.runInInjectionContext(() => extensionRefreshRedirect("/redirect")());
+
+    expect(configService.getFeatureFlag).toHaveBeenCalledWith(FeatureFlag.ExtensionRefresh);
+    expect(router.createUrlTree).toHaveBeenCalledWith(["/redirect"], {
+      queryParams: urlTree.queryParams,
+    });
+  });
+});

libs/angular/src/utils/extension-refresh-redirect.ts

@@ -16,7 +16,7 @@ export function extensionRefreshRedirect(redirectUrl: string): () => Promise<boo
     const shouldRedirect = await configService.getFeatureFlag(FeatureFlag.ExtensionRefresh);
     if (shouldRedirect) {
       const currentNavigation = router.getCurrentNavigation();
-      const queryParams = currentNavigation?.extras?.queryParams || {};
+      const queryParams = currentNavigation?.extractedUrl?.queryParams || {};
 
       // Preserve query params when redirecting as it is likely that the refreshed component
       // will be consuming the same query params.

libs/angular/src/vault/components/add-edit.component.ts

@@ -25,7 +25,6 @@ import { LogService } from "@bitwarden/common/platform/abstractions/log.service"
 import { MessagingService } from "@bitwarden/common/platform/abstractions/messaging.service";
 import { PlatformUtilsService } from "@bitwarden/common/platform/abstractions/platform-utils.service";
 import { Utils } from "@bitwarden/common/platform/misc/utils";
-import { SendApiService } from "@bitwarden/common/tools/send/services/send-api.service.abstraction";
 import { CollectionId, UserId } from "@bitwarden/common/types/guid";
 import { CipherService } from "@bitwarden/common/vault/abstractions/cipher.service";
 import { FolderService } from "@bitwarden/common/vault/abstractions/folder/folder.service.abstraction";
@@ -126,7 +125,6 @@ export class AddEditComponent implements OnInit, OnDestroy {
     protected logService: LogService,
     protected passwordRepromptService: PasswordRepromptService,
     private organizationService: OrganizationService,
-    protected sendApiService: SendApiService,
     protected dialogService: DialogService,
     protected win: Window,
     protected datePipe: DatePipe,