Merge pull request #1188 from Mikunj/revert-group-id

Added strict type checking to group id

ts/session/messages/outgoing/content/data/group/ClosedGroupChatMessage.ts

@@ -1,10 +1,11 @@
 import { SignalService } from '../../../../../../protobuf';
 import { ChatMessage } from '../ChatMessage';
 import { ClosedGroupMessage } from './ClosedGroupMessage';
+import { PubKey } from '../../../../../types';
 
 interface ClosedGroupChatMessageParams {
   identifier?: string;
-  groupId: string;
+  groupId: string | PubKey;
   chatMessage: ChatMessage;
 }
 

ts/session/messages/outgoing/content/data/group/ClosedGroupMessage.ts

@@ -2,26 +2,28 @@ import { DataMessage } from '../DataMessage';
 import { SignalService } from '../../../../../../protobuf';
 import { TextEncoder } from 'util';
 import { MessageParams } from '../../../Message';
+import { PubKey } from '../../../../../types';
 
 interface ClosedGroupMessageParams extends MessageParams {
-  groupId: string;
+  groupId: string | PubKey;
 }
 
 export abstract class ClosedGroupMessage extends DataMessage {
-  public readonly groupId: string;
+  public readonly groupId: PubKey;
 
   constructor(params: ClosedGroupMessageParams) {
     super({
       timestamp: params.timestamp,
       identifier: params.identifier,
     });
-    this.groupId = params.groupId;
+    const { groupId } = params;
+    this.groupId = typeof groupId === 'string' ? new PubKey(groupId) : groupId;
   }
 
   protected abstract groupContextType(): SignalService.GroupContext.Type;
 
   protected groupContext(): SignalService.GroupContext {
-    const id = new TextEncoder().encode(this.groupId);
+    const id = new TextEncoder().encode(this.groupId.key);
     const type = this.groupContextType();
 
     return new SignalService.GroupContext({ id, type });

ts/session/sending/MessageQueue.ts

@@ -139,7 +139,7 @@ export class MessageQueue implements MessageQueueInterface {
   public async processPending(device: PubKey) {
     const messages = await this.pendingMessageCache.getForDevice(device);
 
-    const isMediumGroup = GroupUtils.isMediumGroup(device.key);
+    const isMediumGroup = GroupUtils.isMediumGroup(device);
     const hasSession = await SessionProtocol.hasSession(device);
 
     if (!isMediumGroup && !hasSession) {

ts/session/utils/Groups.ts

@@ -1,11 +1,11 @@
 import _ from 'lodash';
-import { PrimaryPubKey } from '../types';
+import { PrimaryPubKey, PubKey } from '../types';
 import { MultiDeviceProtocol } from '../protocols';
 
 export async function getGroupMembers(
-  groupId: string
+  groupId: PubKey
 ): Promise<Array<PrimaryPubKey>> {
-  const groupConversation = window.ConversationController.get(groupId);
+  const groupConversation = window.ConversationController.get(groupId.key);
   const groupMembers = groupConversation
     ? groupConversation.attributes.members
     : undefined;
@@ -22,8 +22,8 @@ export async function getGroupMembers(
   return _.uniqWith(primaryDevices, (a, b) => a.isEqual(b));
 }
 
-export function isMediumGroup(groupId: string): boolean {
-  const conversation = window.ConversationController.get(groupId);
+export function isMediumGroup(groupId: PubKey): boolean {
+  const conversation = window.ConversationController.get(groupId.key);
 
   if (!conversation) {
     return false;

ts/test/session/sending/MessageQueue_test.ts

@@ -43,7 +43,7 @@ describe('MessageQueue', () => {
   // Message Sender Stubs
   let sendStub: sinon.SinonStub<[RawMessage, (number | undefined)?]>;
   // Utils Stubs
-  let isMediumGroupStub: sinon.SinonStub<[string], boolean>;
+  let isMediumGroupStub: sinon.SinonStub<[PubKey], boolean>;
   // Session Protocol Stubs
   let hasSessionStub: sinon.SinonStub<[PubKey]>;
   let sendSessionRequestIfNeededStub: sinon.SinonStub<[PubKey], Promise<void>>;