import { EResult, JsPbMessage, OperationResponse, Unregisterable } from "./shared";

export interface Updates {
    /**
     * @param base64 Serialized base64 message from `CMsgSystemUpdateApplyParams`.
     */
    ApplyUpdates(base64: string): Promise<OperationResponse>;

    CheckForUpdates(): Promise<OperationResponse>; // Checks for software updates

    GetCurrentOSBranch(): Promise<OSBranch>;

    GetOSBranchList(): Promise<any[]>;

    /**
     * If `data` is deserialized, returns {@link MsgSystemUpdateState}.
     * @returns a ProtoBuf message.
     */
    RegisterForUpdateStateChanges(callback: (data: ArrayBuffer) => void): Unregisterable;

    /**
     * @param base64 Serialized base64 message from `CMsgSelectOSBranchParams`.
     */
    SelectOSBranch(base64: string): Promise<OperationResponse>;
}


export interface OSBranch {
    eBranch: EOSBranch; // 1 - Stable
    sRawName: string;
}

export enum EOSBranch {
    Unknown,
    Release,
    ReleaseCandidate,
    Beta,
    BetaCandidate,
    Preview,
    PreviewCandidate,
    Main,
    Staging,
}

/**
 * CMsgSystemUpdateState
 */
export interface MsgSystemUpdateState extends JsPbMessage {
    state(): EUpdaterState | undefined;

    progress(): UpdateProgress | undefined;

    supports_os_updates(): boolean | undefined;

    update_apply_results(): UpdateApplyResult[];

    update_check_results(): UpdateCheckResult[];
}

export interface UpdateApplyResult {
    type: EUpdaterType;
    eresult: EResult;
    requires_client_restart: boolean;
    requires_system_restart: boolean;
}

export interface UpdateCheckResult {
    type: EUpdaterType;
    eresult: EResult;
    rtime_checked: number;
    available: boolean;
}

export interface UpdateProgress {
    stage_progress: number | undefined;
    stage_size_bytes: number | undefined;
    rtime_estimated_completion: number | undefined;
}

export enum EUpdaterState {
    Invalid,
    // ty valve
    UpToDate = 2,
    Checking,
    Available,
    Applying,
    ClientRestartPending,
    SystemRestartPending,
    RollBack,
}

export enum EUpdaterType {
    Invalid,
    Client,
    OS,
    BIOS,
    Aggregated,
    Test1,
    Test2,
    Dummy,
}