wikimedia/mediawiki-extensions-Wikibase

import {
    convertNoSuchEntityError,
    getApiEntity,
} from '@/data-access/ApiWbgetentities';
import { ReadingApi } from '@/definitions/data-access/Api';
import { EntityWithLabels } from '@/definitions/data-access/ApiWbgetentities';
import EntityLabelRepository from '@/definitions/data-access/EntityLabelRepository';
import Term from '@/datamodel/Term';
import EntityWithoutLabelInLanguageException from '@/data-access/error/EntityWithoutLabelInLanguageException';

export default class ApiEntityLabelRepository implements EntityLabelRepository {
    private readonly forLanguageCode: string;
    private readonly api: ReadingApi;

    public constructor( forLanguageCode: string, api: ReadingApi ) {
        this.forLanguageCode = forLanguageCode;
        this.api = api;
    }

    public async getLabel( entityId: string ): Promise<Term> {
        const response = await this.api.get( {
            action: 'wbgetentities',
            props: new Set( [ 'labels' ] ),
            ids: new Set( [ entityId ] ),
            languages: new Set( [ this.forLanguageCode ] ),
            languagefallback: true,
            errorformat: 'raw',
            formatversion: 2,
        } ).catch( convertNoSuchEntityError );
        const entity = getApiEntity( response, entityId ) as EntityWithLabels;

        if ( !( this.forLanguageCode in entity.labels ) ) {
            throw new EntityWithoutLabelInLanguageException(
                `Could not find label for language '${this.forLanguageCode}'.`,
            );
        }
        const label = entity.labels[ this.forLanguageCode ];

        return {
            value: label.value,
            language: label.language,
        };
    }
}