import LazyMobXCache from '../../../shared/lib/LazyMobXCache';
import {
    CopyNumberCount,
    CopyNumberCountIdentifier,
} from 'cbioportal-ts-api-client';
import internalClient from 'shared/api/cbioportalInternalClientInstance';

function getKey<T extends { entrezGeneId: number; alteration: number }>(
    obj: T
): string {
    return obj.entrezGeneId + '~' + obj.alteration;
}

function fetch(
    queries: CopyNumberCountIdentifier[],
    molecularProfileIdDiscrete: string | undefined
): Promise<CopyNumberCount[]> {
    if (!molecularProfileIdDiscrete) {
        return Promise.reject('No discrete CNA molecular profile id given');
    } else {
        if (queries.length) {
            return internalClient.fetchCopyNumberCountsUsingPOST({
                molecularProfileId: molecularProfileIdDiscrete,
                copyNumberCountIdentifiers: queries,
            });
        } else {
            return Promise.resolve([]);
        }
    }
}

export default class CopyNumberCountCache extends LazyMobXCache<
    CopyNumberCount,
    CopyNumberCountIdentifier
> {
    constructor(molecularProfileIdDiscrete: string | undefined) {
        super(getKey, getKey, fetch, molecularProfileIdDiscrete);
    }
}