zitadel/console/src/app/services/interceptors/exhausted.grpc.interceptor.ts

import { Injectable } from '@angular/core';
import { Request, StatusCode, UnaryInterceptor, UnaryResponse } from 'grpc-web';
import { EnvironmentService } from '../environment.service';
import { ExhaustedService } from '../exhausted.service';

/**
 * ExhaustedGrpcInterceptor shows the exhausted dialog after receiving a gRPC response status 8.
 */
@Injectable({ providedIn: 'root' })
export class ExhaustedGrpcInterceptor<TReq = unknown, TResp = unknown> implements UnaryInterceptor<TReq, TResp> {
  constructor(private exhaustedSvc: ExhaustedService, private envSvc: EnvironmentService) {}

  public async intercept(
    request: Request<TReq, TResp>,
    invoker: (request: Request<TReq, TResp>) => Promise<UnaryResponse<TReq, TResp>>,
  ): Promise<UnaryResponse<TReq, TResp>> {
    return invoker(request).catch((error: any) => {
      if (error.code === StatusCode.RESOURCE_EXHAUSTED) {
        return this.exhaustedSvc
          .showExhaustedDialog(this.envSvc.env)
          .toPromise()
          .then(() => {
            throw error;
          });
      }
      throw error;
    });
  }
}