zitadel/console/src/app/services/interceptors/grpc-org.interceptor.ts

import { Injectable } from '@angular/core';
import { Metadata } from 'grpc-web';

import { Org } from '../../proto/generated/auth_pb';
import { GrpcHandler } from '../grpc-handler';
import { StorageService } from '../storage.service';
import { GrpcInterceptor } from './grpc-interceptor';

const orgKey = 'x-zitadel-orgid';

@Injectable({ providedIn: 'root' })
export class GrpcOrgInterceptor implements GrpcInterceptor {
    constructor(private readonly storageService: StorageService) { }

    public async intercept(
        req: unknown,
        metadata: Metadata,
        next: GrpcHandler,
    ): Promise<any> {
        const org: Org.AsObject | null = (this.storageService.getItem('organization'));
        if (!metadata[orgKey] && org) {
            metadata[orgKey] = org.id ?? '';
        }
        return await next.handle(req, metadata);
    }
}