node-redis/packages/client/lib/commands/ZADD.ts

import { RedisCommandArgument, RedisCommandArguments } from '.';
import { transformNumberInfinityArgument, ZMember } from './generic-transformers';

export const FIRST_KEY_INDEX = 1;

interface NX {
    NX?: true;
}

interface XX {
    XX?: true;
}

interface LT {
    LT?: true;
}

interface GT {
    GT?: true;
}

interface CH {
    CH?: true;
}

interface INCR {
    INCR?: true;
}

type ZAddOptions = (NX | (XX & LT & GT)) & CH & INCR;

export function transformArguments(
    key: RedisCommandArgument,
    members: ZMember | Array<ZMember>,
    options?: ZAddOptions
): RedisCommandArguments {
    const args = ['ZADD', key];

    if ((<NX>options)?.NX) {
        args.push('NX');
    } else {
        if ((<XX>options)?.XX) {
            args.push('XX');
        }

        if ((<GT>options)?.GT) {
            args.push('GT');
        } else if ((<LT>options)?.LT) {
            args.push('LT');
        }
    }

    if ((<CH>options)?.CH) {
        args.push('CH');
    }

    if ((<INCR>options)?.INCR) {
        args.push('INCR');
    }

    for (const { score, value } of (Array.isArray(members) ? members : [members])) {
        args.push(
            transformNumberInfinityArgument(score),
            value
        );
    }

    return args;
}

export { transformNumberInfinityReply as transformReply } from './generic-transformers';