import * as request from 'superagent';
import { ITrial, ITrialMatch } from 'shared/model/MatchMiner.ts';
import { buildCBioPortalAPIUrl } from './urls';

/**
 * Retrieves the trial matches for the query given, if they are in the MatchMiner API.
 */
// It cannot be set globally since it will cause test error: undefined of 'replace'.
// const cbioportalUrl = buildCBioPortalAPIUrl('api/matchminer/api');
export async function fetchTrialMatchesUsingPOST(
    query: object
): Promise<Array<ITrialMatch>> {
    const cbioportalUrl = buildCBioPortalAPIUrl('api/matchminer/api');
    return request
        .post(cbioportalUrl + '/post_trial_match')
        .set('Content-Type', 'application/json')
        .send(query)
        .then(res => {
            const response = JSON.parse(res.text);
            return response.map((record: any) => ({
                id: record.nct_id + '+' + record.protocol_no,
                nctId: record.nct_id,
                protocolNo: record.protocol_no,
                gender: record.gender ? record.gender : '',
                matchType: record.match_type ? record.match_type : '',
                armDescription: record.arm_description
                    ? record.arm_description
                    : '',
                armType: record.arm_type ? record.arm_type : '',
                sampleId: record.sample_id,
                mrn: record.mrn,
                vitalStatus: record.vital_status ? record.vital_status : '',
                genomicAlteration: record.genomic_alteration
                    ? record.genomic_alteration
                    : '',
                trueHugoSymbol: record.true_hugo_symbol
                    ? record.true_hugo_symbol
                    : '',
                trueProteinChange: record.true_protein_change
                    ? record.true_protein_change
                    : '',
                oncotreePrimaryDiagnosisName: record.oncotreePrimaryDiagnosisName
                    ? record.oncotreePrimaryDiagnosisName
                    : '',
                trialAgeNumerical: record.trial_age_numerical
                    ? record.trial_age_numerical
                    : '',
                trialOncotreePrimaryDiagnosis: record.trial_oncotree_primary_diagnosis
                    ? record.trial_oncotree_primary_diagnosis
                    : '',
            }));
        });
}

export async function fetchTrialsByTypeAndId(
    type: string,
    id: string
): Promise<ITrial> {
    const cbioportalUrl = buildCBioPortalAPIUrl('api/matchminer/api');
    return request.get(cbioportalUrl + '/' + type + '/' + id).then(res => {
        const response = JSON.parse(res.text);
        return {
            id: response.nct_id + '+' + response.protocol_no,
            nctId: response.nct_id,
            protocolNo: response.protocol_no,
            principalInvestigator: response.principal_investigator,
            phase: response.phase,
            shortTitle: response.short_title,
            status: response.status,
            treatmentList: response.treatment_list,
        };
    });
}

export async function fetchTrialsUsingPost(
    query: object
): Promise<Array<ITrial>> {
    const cbioportalUrl = buildCBioPortalAPIUrl('api/matchminer/api');
    return request
        .post(cbioportalUrl + '/post_trial')
        .set('Content-Type', 'application/json')
        .send(query)
        .then(res => {
            const response = JSON.parse(res.text);
            return response.map((record: any) => ({
                id: record.nct_id + '+' + record.protocol_no,
                nctId: record.nct_id,
                protocolNo: record.protocol_no,
                phase: record.phase,
                shortTitle: record.short_title,
                status: record.status,
                treatmentList: record.treatment_list,
            }));
        });
}

export async function fetchTrialsById(query: object): Promise<Array<ITrial>> {
    const cbioportalUrl = buildCBioPortalAPIUrl('api/matchminer/api');
    return request
        .post(cbioportalUrl + '/trials')
        .set('Content-Type', 'application/json')
        .send(query)
        .then(res => {
            const response = JSON.parse(res.text);
            return response.map((record: any) => ({
                id: record.nct_id + '+' + record.protocol_no,
                nctId: record.nct_id,
                protocolNo: record.protocol_no,
                principalInvestigator: record.principal_investigator,
                phase: record.phase,
                shortTitle: record.short_title,
                status: record.status,
                treatmentList: record.treatment_list,
            }));
        });
}