import { CarrierAccountWithExtraInfo, CarriersEnum } from "../src/models/components";
import { Shippo } from "../src/sdk";
import { HTTPClient, Fetcher } from "../src/lib/http";

export class MockShippoSDK extends Shippo {
    constructor(fetcher?: Fetcher) {
        const httpClient = new HTTPClient({fetcher: fetcher});
        super({
            apiKeyHeader: process.env.SHIPPO_TOKEN,
            httpClient: httpClient
        });
    }
}


export const shippoSdk = new Shippo({apiKeyHeader: process.env.SHIPPO_TOKEN});

export async function getCarrierAccounts(carrier: CarriersEnum): Promise<Array<CarrierAccountWithExtraInfo>> {
    const carrierAccountsResponse = await shippoSdk.carrierAccounts.list({ carrier: carrier });
    const carrierAccounts = carrierAccountsResponse?.results;
    if (!carrierAccounts) {
        throw new Error('No carrier accounts found');
    } else {
        return carrierAccounts;
    }
}

export async function getDefaultCarrierAccountObjectId(): Promise<string> {
    const carrierAccount = await getCarrierAccount(CarriersEnum.Usps);
    return carrierAccount.objectId;
}

export async function getCarrierAccount(carrier: CarriersEnum): Promise<CarrierAccountWithExtraInfo> {
    const carrierAccounts = await getCarrierAccounts(carrier)
    return carrierAccounts[0];
}