import { transformReplyNumberInfinity } from './generic-transformers';
import { ZMember } from './ZADD';

export const FIRST_KEY_INDEX = 1;

export function transformArguments(key: string | Array<string>, timeout: number): Array<string> {
    const args = ['BZPOPMAX'];

    if (typeof key === 'string') {
        args.push(key);
    } else {
        args.push(...key);
    }

    args.push(timeout.toString());

    return args;
}

interface ZMemberWithKey extends ZMember {
    key: string;
}

type BZPopMaxReply = ZMemberWithKey | null;

export function transformReply(reply: [key: string, value: string, score: string] | null): BZPopMaxReply | null {
    if (!reply) return null;

    return {
        key: reply[0],
        value: reply[1],
        score: transformReplyNumberInfinity(reply[2])
    };
}