import { combineLatest, map, switchMap } from "rxjs"; import { ApiService } from "@bitwarden/common/abstractions/api.service"; import { OrganizationService } from "@bitwarden/common/admin-console/abstractions/organization/organization.service.abstraction"; import { FeatureFlag } from "@bitwarden/common/enums/feature-flag.enum"; import { ListResponse } from "@bitwarden/common/models/response/list.response"; import { ConfigService } from "@bitwarden/common/platform/abstractions/config/config.service"; import { StateProvider } from "@bitwarden/common/platform/state"; import { SecurityTaskId, UserId } from "@bitwarden/common/types/guid"; import { filterOutNullish, perUserCache$ } from "../../utils/observable-utilities"; import { TaskService } from "../abstractions/task.service"; import { SecurityTaskStatus } from "../enums"; import { SecurityTask, SecurityTaskData, SecurityTaskResponse } from "../models"; import { SECURITY_TASKS } from "../state/security-task.state"; export class DefaultTaskService implements TaskService { constructor( private stateProvider: StateProvider, private apiService: ApiService, private organizationService: OrganizationService, private configService: ConfigService, ) {} tasksEnabled$ = perUserCache$((userId) => { return combineLatest([ this.organizationService .organizations$(userId) .pipe(map((orgs) => orgs.some((o) => o.useRiskInsights))), this.configService.getFeatureFlag$(FeatureFlag.SecurityTasks), ]).pipe(map(([atLeastOneOrgEnabled, flagEnabled]) => atLeastOneOrgEnabled && flagEnabled)); }); tasks$ = perUserCache$((userId) => { return this.taskState(userId).state$.pipe( switchMap(async (tasks) => { if (tasks == null) { await this.fetchTasksFromApi(userId); } return tasks; }), filterOutNullish(), map((tasks) => tasks.map((t) => new SecurityTask(t))), ); }); pendingTasks$ = perUserCache$((userId) => { return this.tasks$(userId).pipe( map((tasks) => tasks.filter((t) => t.status === SecurityTaskStatus.Pending)), ); }); async refreshTasks(userId: UserId): Promise { await this.fetchTasksFromApi(userId); } async clear(userId: UserId): Promise { await this.updateTaskState(userId, []); } async markAsComplete(taskId: SecurityTaskId, userId: UserId): Promise { await this.apiService.send("PATCH", `/tasks/${taskId}/complete`, null, true, false); await this.refreshTasks(userId); } /** * Fetches the tasks from the API and updates the local state * @param userId * @private */ private async fetchTasksFromApi(userId: UserId): Promise { const r = await this.apiService.send("GET", "/tasks", null, true, true); const response = new ListResponse(r, SecurityTaskResponse); const taskData = response.data.map((t) => new SecurityTaskData(t)); await this.updateTaskState(userId, taskData); } /** * Returns the local state for the tasks * @param userId * @private */ private taskState(userId: UserId) { return this.stateProvider.getUser(userId, SECURITY_TASKS); } /** * Updates the local state with the provided tasks and returns the updated state * @param userId * @param tasks * @private */ private updateTaskState( userId: UserId, tasks: SecurityTaskData[], ): Promise { return this.taskState(userId).update(() => tasks); } }