wikimedia/mediawiki-extensions-Wikibase

import {
    Snak,
    Statement,
    StatementMap,
} from '@wmde/wikibase-datamodel-types';
import EntityId from '@/datamodel/EntityId';
import { PathToSnak } from '@/store/statements/PathToSnak';
import { PathToStatement } from '@/store/statements/PathToStatement';
import { PathToStatementGroup } from '@/store/statements/PathToStatementGroup';

export class MainSnakPath implements PathToStatement, PathToSnak, PathToStatementGroup {

    private readonly entityId: EntityId;
    private readonly propertyId: EntityId;
    private readonly index: number;

    public constructor(
        entityId: EntityId,
        propertyId: EntityId,
        index: number,
    ) {
        this.entityId = entityId;
        this.propertyId = propertyId;
        this.index = index;
    }

    public resolveStatementGroup( state: Record<EntityId, StatementMap> ): Statement[] | null {
        return state?.[ this.entityId ]?.[ this.propertyId ] ?? null;
    }

    public resolveStatement( state: Record<EntityId, StatementMap> ): Statement | null {
        return this.resolveStatementGroup( state )?.[ this.index ] ?? null;
    }

    public resolveSnakInStatement( state: Record<EntityId, StatementMap> ): Snak|null {
        return this.resolveStatement( state )?.mainsnak ?? null;
    }
}