97 lines
2.8 KiB
TypeScript
97 lines
2.8 KiB
TypeScript
import { AuthService } from "@bitwarden/common/abstractions/auth.service";
|
|
import { CipherService } from "@bitwarden/common/abstractions/cipher.service";
|
|
import { PlatformUtilsService } from "@bitwarden/common/abstractions/platformUtils.service";
|
|
import { AuthenticationStatus } from "@bitwarden/common/enums/authenticationStatus";
|
|
import { UriMatchType } from "@bitwarden/common/enums/uriMatchType";
|
|
|
|
export default class WebRequestBackground {
|
|
private pendingAuthRequests: any[] = [];
|
|
private webRequest: any;
|
|
private isFirefox: boolean;
|
|
|
|
constructor(
|
|
platformUtilsService: PlatformUtilsService,
|
|
private cipherService: CipherService,
|
|
private authService: AuthService
|
|
) {
|
|
this.webRequest = (window as any).chrome.webRequest;
|
|
this.isFirefox = platformUtilsService.isFirefox();
|
|
}
|
|
|
|
async init() {
|
|
if (!this.webRequest || !this.webRequest.onAuthRequired) {
|
|
return;
|
|
}
|
|
|
|
this.webRequest.onAuthRequired.addListener(
|
|
async (details: any, callback: any) => {
|
|
if (!details.url || this.pendingAuthRequests.indexOf(details.requestId) !== -1) {
|
|
if (callback) {
|
|
callback();
|
|
}
|
|
return;
|
|
}
|
|
|
|
this.pendingAuthRequests.push(details.requestId);
|
|
|
|
if (this.isFirefox) {
|
|
// eslint-disable-next-line
|
|
return new Promise(async (resolve, reject) => {
|
|
await this.resolveAuthCredentials(details.url, resolve, reject);
|
|
});
|
|
} else {
|
|
await this.resolveAuthCredentials(details.url, callback, callback);
|
|
}
|
|
},
|
|
{ urls: ["http://*/*", "https://*/*"] },
|
|
[this.isFirefox ? "blocking" : "asyncBlocking"]
|
|
);
|
|
|
|
this.webRequest.onCompleted.addListener((details: any) => this.completeAuthRequest(details), {
|
|
urls: ["http://*/*"],
|
|
});
|
|
this.webRequest.onErrorOccurred.addListener(
|
|
(details: any) => this.completeAuthRequest(details),
|
|
{
|
|
urls: ["http://*/*"],
|
|
}
|
|
);
|
|
}
|
|
|
|
// eslint-disable-next-line
|
|
private async resolveAuthCredentials(domain: string, success: Function, error: Function) {
|
|
if ((await this.authService.getAuthStatus()) < AuthenticationStatus.Unlocked) {
|
|
error();
|
|
return;
|
|
}
|
|
|
|
try {
|
|
const ciphers = await this.cipherService.getAllDecryptedForUrl(
|
|
domain,
|
|
null,
|
|
UriMatchType.Host
|
|
);
|
|
if (ciphers == null || ciphers.length !== 1) {
|
|
error();
|
|
return;
|
|
}
|
|
|
|
success({
|
|
authCredentials: {
|
|
username: ciphers[0].login.username,
|
|
password: ciphers[0].login.password,
|
|
},
|
|
});
|
|
} catch {
|
|
error();
|
|
}
|
|
}
|
|
|
|
private completeAuthRequest(details: any) {
|
|
const i = this.pendingAuthRequests.indexOf(details.requestId);
|
|
if (i > -1) {
|
|
this.pendingAuthRequests.splice(i, 1);
|
|
}
|
|
}
|
|
}
|