WIP

packages/client/lib/commands/ROLE.ts

@@ -1,75 +1,67 @@
-export const IS_READ_ONLY = true;
+import { ArrayReply, BlobStringReply, Command, NumberReply } from '../RESP/types';
 
-export function transformArguments(): Array<string> {
+type MasterRole = [
+  role: BlobStringReply<'master'>,
+  replicationOffest: NumberReply,
+  replicas: ArrayReply<[host: BlobStringReply, port: NumberReply, replicationOffest: NumberReply]>,
+];
+
+type SlaveRole = [
+  role: BlobStringReply<'slave'>,
+  masterHost: BlobStringReply,
+  masterPort: NumberReply,
+  state: BlobStringReply<'connect' | 'connecting' | 'sync' | 'connected'>,
+  dataReceived: NumberReply
+];
+
+type SentinelRole = [
+  role: BlobStringReply<'sentinel'>,
+  masterNames: ArrayReply<BlobStringReply>
+];
+
+type Role = MasterRole | SlaveRole | SentinelRole;
+
+export default {
+  IS_READ_ONLY: true,
+  FIRST_KEY_INDEX: undefined,
+  transformArguments() {
     return ['ROLE'];
-}
+  },
+  transformReply(reply: Role) {
+    switch (reply[0] as Role[0]['DEFAULT']) {
+      case 'master': {
+        const [role, replicationOffest, replicas] = reply as MasterRole;
+        return {
+          role,
+          replicationOffest,
+          replicas: replicas.map(([host, port, replicationOffest]) => ({
+            host,
+            port,
+            replicationOffest,
+          })),
+        };
+      }
 
-interface RoleReplyInterface<T extends string> {
-    role: T;
-}
+      case 'slave': {
+        const [role, masterHost, masterPort, state, dataReceived] = reply as SlaveRole;
+        return {
+          role,
+          master: {
+            host: masterHost,
+            port: masterPort,
+          },
+          state,
+          dataReceived,
+        };
+      }
 
-type RoleMasterRawReply = ['master', number, Array<[string, string, string]>];
-
-interface RoleMasterReply extends RoleReplyInterface<'master'> {
-    replicationOffest: number;
-    replicas: Array<{
-        ip: string;
-        port: number;
-        replicationOffest: number;
-    }>;
-}
-
-type RoleReplicaState = 'connect' | 'connecting' | 'sync' | 'connected';
-
-type RoleReplicaRawReply = ['slave', string, number, RoleReplicaState, number];
-
-interface RoleReplicaReply extends RoleReplyInterface<'slave'>  {
-    master: {
-        ip: string;
-        port: number;
-    };
-    state: RoleReplicaState;
-    dataReceived: number;
-}
-
-type RoleSentinelRawReply = ['sentinel', Array<string>];
-
-interface RoleSentinelReply extends RoleReplyInterface<'sentinel'>  {
-    masterNames: Array<string>;
-}
-
-type RoleRawReply = RoleMasterRawReply | RoleReplicaRawReply | RoleSentinelRawReply;
-
-type RoleReply = RoleMasterReply | RoleReplicaReply | RoleSentinelReply;
-
-export function transformReply(reply: RoleRawReply): RoleReply {
-    switch (reply[0]) {
-        case 'master':
-            return {
-                role: 'master',
-                replicationOffest: reply[1],
-                replicas: reply[2].map(([ip, port, replicationOffest]) => ({
-                    ip,
-                    port: Number(port),
-                    replicationOffest: Number(replicationOffest)
-                }))
-            };
-
-        case 'slave':
-            return {
-                role: 'slave',
-                master: {
-                    ip: reply[1],
-                    port: reply[2]
-                },
-                state: reply[3],
-                dataReceived: reply[4]
-            };
-
-        case 'sentinel':
-            return {
-                role: 'sentinel',
-                masterNames: reply[1]
-            };
+      case 'sentinel': {
+        const [role, masterNames] = reply as SentinelRole;
+        return {
+          role,
+          masterNames,
+        };
+      }
     }
-}
+  }
+} as const satisfies Command;