packages/agent-core-v2/src/app/flag/flagRegistryService.ts

import { Disposable, type IDisposable } from '#/_base/di/lifecycle';
import { LifecycleScope } from '#/app/scopes';
import { ScopeActivation, registerScopedService } from '#/_base/di/scope';
import { BugIndicatingError } from '#/errors';

import {
  type FlagDefinitionInput,
  type FlagId,
  getContributedFlags,
  IFlagRegistry,
} from './flagRegistry';

export class FlagRegistryService extends Disposable implements IFlagRegistry {
  declare readonly _serviceBrand: undefined;
  private readonly byId = new Map<FlagId, FlagDefinitionInput>();

  constructor() {
    super();
    for (const def of getContributedFlags()) {
      this.add(def);
    }
  }

  register(definition: FlagDefinitionInput): IDisposable {
    this.add(definition);
    return this._register({
      dispose: () => {
        this.byId.delete(definition.id);
      },
    });
  }

  get(id: FlagId): FlagDefinitionInput | undefined {
    return this.byId.get(id);
  }

  list(): readonly FlagDefinitionInput[] {
    return [...this.byId.values()];
  }

  private add(definition: FlagDefinitionInput): void {
    if (this.byId.has(definition.id)) {
      throw new BugIndicatingError(`Flag '${definition.id}' is already registered`);
    }
    this.byId.set(definition.id, definition);
  }
}

registerScopedService(
  LifecycleScope.App,
  IFlagRegistry,
  FlagRegistryService,
  ScopeActivation.OnScopeCreated,
  'flag',
);