140 lines
		
	
	
		
			4.3 KiB
		
	
	
	
		
			TypeScript
		
	
	
			
		
		
	
	
			140 lines
		
	
	
		
			4.3 KiB
		
	
	
	
		
			TypeScript
		
	
	
/*
 | 
						|
Copyright 2022 The Matrix.org Foundation C.I.C.
 | 
						|
 | 
						|
Licensed under the Apache License, Version 2.0 (the "License");
 | 
						|
you may not use this file except in compliance with the License.
 | 
						|
You may obtain a copy of the License at
 | 
						|
 | 
						|
    http://www.apache.org/licenses/LICENSE-2.0
 | 
						|
 | 
						|
Unless required by applicable law or agreed to in writing, software
 | 
						|
distributed under the License is distributed on an "AS IS" BASIS,
 | 
						|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 | 
						|
See the License for the specific language governing permissions and
 | 
						|
limitations under the License.
 | 
						|
*/
 | 
						|
 | 
						|
import { Optional } from "matrix-events-sdk";
 | 
						|
import { EventType, IContent, MatrixEvent, MsgType, RelationType, Room, RoomMember } from "matrix-js-sdk/src/matrix";
 | 
						|
 | 
						|
import { SdkContextClass } from "../../../src/contexts/SDKContext";
 | 
						|
import {
 | 
						|
    VoiceBroadcastPlayback,
 | 
						|
    VoiceBroadcastPreRecording,
 | 
						|
    VoiceBroadcastRecording,
 | 
						|
} from "../../../src/voice-broadcast";
 | 
						|
import {
 | 
						|
    VoiceBroadcastChunkEventType,
 | 
						|
    VoiceBroadcastInfoEventType,
 | 
						|
    VoiceBroadcastInfoState,
 | 
						|
} from "../../../src/voice-broadcast/types";
 | 
						|
import { mkEvent } from "../../test-utils";
 | 
						|
 | 
						|
// timestamp incremented on each call to prevent duplicate timestamp
 | 
						|
let timestamp = new Date().getTime();
 | 
						|
 | 
						|
export const mkVoiceBroadcastInfoStateEvent = (
 | 
						|
    roomId: Optional<string>,
 | 
						|
    state: Optional<VoiceBroadcastInfoState>,
 | 
						|
    senderId: Optional<string>,
 | 
						|
    senderDeviceId: Optional<string>,
 | 
						|
    startedInfoEvent?: MatrixEvent,
 | 
						|
    lastChunkSequence?: number,
 | 
						|
): MatrixEvent => {
 | 
						|
    const relationContent: IContent = {};
 | 
						|
 | 
						|
    if (startedInfoEvent) {
 | 
						|
        relationContent["m.relates_to"] = {
 | 
						|
            event_id: startedInfoEvent.getId(),
 | 
						|
            rel_type: "m.reference",
 | 
						|
        };
 | 
						|
    }
 | 
						|
 | 
						|
    const lastChunkSequenceContent = lastChunkSequence ? { last_chunk_sequence: lastChunkSequence } : {};
 | 
						|
 | 
						|
    return mkEvent({
 | 
						|
        event: true,
 | 
						|
        // @ts-ignore allow everything here for edge test cases
 | 
						|
        room: roomId,
 | 
						|
        // @ts-ignore allow everything here for edge test cases
 | 
						|
        user: senderId,
 | 
						|
        type: VoiceBroadcastInfoEventType,
 | 
						|
        // @ts-ignore allow everything here for edge test cases
 | 
						|
        skey: senderId,
 | 
						|
        content: {
 | 
						|
            state,
 | 
						|
            device_id: senderDeviceId,
 | 
						|
            ...relationContent,
 | 
						|
            ...lastChunkSequenceContent,
 | 
						|
        },
 | 
						|
        ts: timestamp++,
 | 
						|
    });
 | 
						|
};
 | 
						|
 | 
						|
export const mkVoiceBroadcastChunkEvent = (
 | 
						|
    infoEventId: string,
 | 
						|
    userId: string,
 | 
						|
    roomId: string,
 | 
						|
    duration: number,
 | 
						|
    sequence?: number,
 | 
						|
    timestamp?: number,
 | 
						|
): MatrixEvent => {
 | 
						|
    return mkEvent({
 | 
						|
        event: true,
 | 
						|
        user: userId,
 | 
						|
        room: roomId,
 | 
						|
        type: EventType.RoomMessage,
 | 
						|
        content: {
 | 
						|
            msgtype: MsgType.Audio,
 | 
						|
            ["org.matrix.msc1767.audio"]: {
 | 
						|
                duration,
 | 
						|
            },
 | 
						|
            info: {
 | 
						|
                duration,
 | 
						|
            },
 | 
						|
            [VoiceBroadcastChunkEventType]: {
 | 
						|
                ...(sequence ? { sequence } : {}),
 | 
						|
            },
 | 
						|
            ["m.relates_to"]: {
 | 
						|
                rel_type: RelationType.Reference,
 | 
						|
                event_id: infoEventId,
 | 
						|
            },
 | 
						|
        },
 | 
						|
        ts: timestamp,
 | 
						|
    });
 | 
						|
};
 | 
						|
 | 
						|
export const mkVoiceBroadcastPlayback = (stores: SdkContextClass): VoiceBroadcastPlayback => {
 | 
						|
    const infoEvent = mkVoiceBroadcastInfoStateEvent(
 | 
						|
        "!room:example.com",
 | 
						|
        VoiceBroadcastInfoState.Started,
 | 
						|
        "@user:example.com",
 | 
						|
        "ASD123",
 | 
						|
    );
 | 
						|
    return new VoiceBroadcastPlayback(infoEvent, stores.client!, stores.voiceBroadcastRecordingsStore);
 | 
						|
};
 | 
						|
 | 
						|
export const mkVoiceBroadcastRecording = (stores: SdkContextClass): VoiceBroadcastRecording => {
 | 
						|
    const infoEvent = mkVoiceBroadcastInfoStateEvent(
 | 
						|
        "!room:example.com",
 | 
						|
        VoiceBroadcastInfoState.Started,
 | 
						|
        "@user:example.com",
 | 
						|
        "ASD123",
 | 
						|
    );
 | 
						|
    return new VoiceBroadcastRecording(infoEvent, stores.client!);
 | 
						|
};
 | 
						|
 | 
						|
export const mkVoiceBroadcastPreRecording = (stores: SdkContextClass): VoiceBroadcastPreRecording => {
 | 
						|
    const roomId = "!room:example.com";
 | 
						|
    const userId = "@user:example.com";
 | 
						|
    const room = new Room(roomId, stores.client!, userId);
 | 
						|
    const roomMember = new RoomMember(roomId, userId);
 | 
						|
    return new VoiceBroadcastPreRecording(
 | 
						|
        room,
 | 
						|
        roomMember,
 | 
						|
        stores.client!,
 | 
						|
        stores.voiceBroadcastPlaybacksStore,
 | 
						|
        stores.voiceBroadcastRecordingsStore,
 | 
						|
    );
 | 
						|
};
 |