mirror of
https://github.com/bitwarden/browser
synced 2025-12-14 23:33:31 +00:00
[PM-19479] Client-Managed SDK state definition (#14839)
* [PM-19479] Client-Managed SDK state definition * Remove test code * Update based on latest sdk * Add DB config * Remove uuid conversion step * Move mapper into separate file * Revert to client managed state * Move mapper to Cipher * Typo --------- Co-authored-by: Andreas Coroiu <andreas.coroiu@gmail.com>
This commit is contained in:
@@ -0,0 +1,64 @@
|
||||
import { firstValueFrom, map } from "rxjs";
|
||||
|
||||
import { UserId } from "@bitwarden/common/types/guid";
|
||||
import { CipherRecordMapper } from "@bitwarden/common/vault/models/domain/cipher-sdk-mapper";
|
||||
import { StateClient, Repository } from "@bitwarden/sdk-internal";
|
||||
|
||||
import { StateProvider, UserKeyDefinition } from "../../state";
|
||||
|
||||
export async function initializeState(
|
||||
userId: UserId,
|
||||
stateClient: StateClient,
|
||||
stateProvider: StateProvider,
|
||||
): Promise<void> {
|
||||
await stateClient.register_cipher_repository(
|
||||
new RepositoryRecord(userId, stateProvider, new CipherRecordMapper()),
|
||||
);
|
||||
}
|
||||
|
||||
export interface SdkRecordMapper<ClientType, SdkType> {
|
||||
userKeyDefinition(): UserKeyDefinition<Record<string, ClientType>>;
|
||||
toSdk(value: ClientType): SdkType;
|
||||
fromSdk(value: SdkType): ClientType;
|
||||
}
|
||||
|
||||
class RepositoryRecord<ClientType, SdkType> implements Repository<SdkType> {
|
||||
constructor(
|
||||
private userId: UserId,
|
||||
private stateProvider: StateProvider,
|
||||
private mapper: SdkRecordMapper<ClientType, SdkType>,
|
||||
) {}
|
||||
|
||||
async get(id: string): Promise<SdkType | null> {
|
||||
const prov = this.stateProvider.getUser(this.userId, this.mapper.userKeyDefinition());
|
||||
const data = await firstValueFrom(prov.state$.pipe(map((data) => data ?? {})));
|
||||
const element = data[id];
|
||||
if (!element) {
|
||||
return null;
|
||||
}
|
||||
return this.mapper.toSdk(element);
|
||||
}
|
||||
|
||||
async list(): Promise<SdkType[]> {
|
||||
const prov = this.stateProvider.getUser(this.userId, this.mapper.userKeyDefinition());
|
||||
const elements = await firstValueFrom(prov.state$.pipe(map((data) => data ?? {})));
|
||||
return Object.values(elements).map((element) => this.mapper.toSdk(element));
|
||||
}
|
||||
|
||||
async set(id: string, value: SdkType): Promise<void> {
|
||||
const prov = this.stateProvider.getUser(this.userId, this.mapper.userKeyDefinition());
|
||||
const elements = await firstValueFrom(prov.state$.pipe(map((data) => data ?? {})));
|
||||
elements[id] = this.mapper.fromSdk(value);
|
||||
await prov.update(() => elements);
|
||||
}
|
||||
|
||||
async remove(id: string): Promise<void> {
|
||||
const prov = this.stateProvider.getUser(this.userId, this.mapper.userKeyDefinition());
|
||||
const elements = await firstValueFrom(prov.state$.pipe(map((data) => data ?? {})));
|
||||
if (!elements[id]) {
|
||||
return;
|
||||
}
|
||||
delete elements[id];
|
||||
await prov.update(() => elements);
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user