NodeRedis/node-redis

import { CommandParser } from '../client/parser';
import { Command, TuplesToMapReply, BlobStringReply, NullReply, NumberReply, MapReply, Resp2Reply, UnwrapReply } from '../RESP/types';
import { isNullReply } from './generic-transformers';

type RunningScript = NullReply | TuplesToMapReply<[
  [BlobStringReply<'name'>, BlobStringReply],
  [BlobStringReply<'command'>, BlobStringReply],
  [BlobStringReply<'duration_ms'>, NumberReply]
]>;

type Engine = TuplesToMapReply<[
  [BlobStringReply<'libraries_count'>, NumberReply],
  [BlobStringReply<'functions_count'>, NumberReply]
]>;

type Engines = MapReply<BlobStringReply, Engine>;

type FunctionStatsReply = TuplesToMapReply<[
  [BlobStringReply<'running_script'>, RunningScript],
  [BlobStringReply<'engines'>, Engines]
]>;

export default {
  NOT_KEYED_COMMAND: true,
  IS_READ_ONLY: true,
  parseCommand(parser: CommandParser) {
    parser.push('FUNCTION', 'STATS');
  },
  transformReply: {
    2: (reply: UnwrapReply<Resp2Reply<FunctionStatsReply>>) => {
      return {
        running_script: transformRunningScript(reply[1]),
        engines: transformEngines(reply[3])
      };
    },
    3: undefined as unknown as () => FunctionStatsReply
  }
} as const satisfies Command;

function transformRunningScript(reply: Resp2Reply<RunningScript>) {
  if (isNullReply(reply)) {
    return null;
  }

  const unwraped = reply as unknown as UnwrapReply<typeof reply>;
  return {
    name: unwraped[1],
    command: unwraped[3],
    duration_ms: unwraped[5]
  };
}

function transformEngines(reply: Resp2Reply<Engines>) {
  const unwraped = reply as unknown as UnwrapReply<typeof reply>;

  const engines: Record<string, {
    libraries_count: NumberReply;
    functions_count: NumberReply;
  }> = Object.create(null);
  for (let i = 0; i < unwraped.length; i++) {
    const name = unwraped[i] as BlobStringReply,
      stats = unwraped[++i] as Resp2Reply<Engine>,
      unwrapedStats = stats as unknown as UnwrapReply<typeof stats>;
    engines[name.toString()] = {
      libraries_count: unwrapedStats[1],
      functions_count: unwrapedStats[3]
    };
  }

  return engines;
}