136 lines
3.3 KiB
TypeScript
136 lines
3.3 KiB
TypeScript
import {
|
|
genVoiceToken,
|
|
Payload,
|
|
WebSocket,
|
|
generateStreamKey,
|
|
} from "@spacebar/gateway";
|
|
import {
|
|
Channel,
|
|
Config,
|
|
emitEvent,
|
|
Member,
|
|
Snowflake,
|
|
Stream,
|
|
StreamCreateEvent,
|
|
StreamCreateSchema,
|
|
StreamServerUpdateEvent,
|
|
StreamSession,
|
|
VoiceState,
|
|
VoiceStateUpdateEvent,
|
|
} from "@spacebar/util";
|
|
import { check } from "./instanceOf";
|
|
|
|
export async function onStreamCreate(this: WebSocket, data: Payload) {
|
|
const startTime = Date.now();
|
|
check.call(this, StreamCreateSchema, data.d);
|
|
const body = data.d as StreamCreateSchema;
|
|
|
|
if (body.channel_id.trim().length === 0) return;
|
|
|
|
// first check if we are in a voice channel already. cannot create a stream if there's no existing voice connection
|
|
const voiceState = await VoiceState.findOne({
|
|
where: { user_id: this.user_id },
|
|
});
|
|
|
|
if (!voiceState || !voiceState.channel_id) return;
|
|
|
|
if (body.guild_id) {
|
|
voiceState.member = await Member.findOneOrFail({
|
|
where: { id: voiceState.user_id, guild_id: voiceState.guild_id },
|
|
relations: ["user", "roles"],
|
|
});
|
|
}
|
|
|
|
// TODO: permissions check - if it's a guild, check if user is allowed to create stream in this guild
|
|
|
|
const channel = await Channel.findOne({
|
|
where: { id: body.channel_id },
|
|
});
|
|
|
|
if (
|
|
!channel ||
|
|
(body.type === "guild" && channel.guild_id != body.guild_id)
|
|
)
|
|
return this.close(4000, "invalid channel");
|
|
|
|
// TODO: actually apply preferred_region from the event payload
|
|
const regions = Config.get().regions;
|
|
const guildRegion = regions.available.filter(
|
|
(r) => r.id === regions.default,
|
|
)[0];
|
|
|
|
// first make sure theres no other streams for this user that somehow didnt get cleared
|
|
await Stream.delete({
|
|
owner_id: this.user_id,
|
|
});
|
|
|
|
// create a new entry in db containing the token for authenticating user in stream gateway IDENTIFY
|
|
const stream = Stream.create({
|
|
id: Snowflake.generate(),
|
|
owner_id: this.user_id,
|
|
channel_id: body.channel_id,
|
|
endpoint: guildRegion.endpoint,
|
|
});
|
|
|
|
await stream.save();
|
|
|
|
const token = genVoiceToken();
|
|
|
|
const streamSession = StreamSession.create({
|
|
stream_id: stream.id,
|
|
user_id: this.user_id,
|
|
session_id: this.session_id,
|
|
token,
|
|
});
|
|
|
|
await streamSession.save();
|
|
|
|
const streamKey = generateStreamKey(
|
|
body.type,
|
|
body.guild_id,
|
|
body.channel_id,
|
|
this.user_id,
|
|
);
|
|
|
|
await emitEvent({
|
|
event: "STREAM_CREATE",
|
|
data: {
|
|
stream_key: streamKey,
|
|
rtc_server_id: stream.id, // for voice connections in guilds it is guild_id, for dm voice calls it seems to be DM channel id, for GoLive streams a generated number
|
|
viewer_ids: [],
|
|
region: guildRegion.name,
|
|
paused: false,
|
|
},
|
|
user_id: this.user_id,
|
|
} as StreamCreateEvent);
|
|
|
|
await emitEvent({
|
|
event: "STREAM_SERVER_UPDATE",
|
|
data: {
|
|
token: streamSession.token,
|
|
stream_key: streamKey,
|
|
guild_id: null, // not sure why its always null
|
|
endpoint: stream.endpoint,
|
|
},
|
|
user_id: this.user_id,
|
|
} as StreamServerUpdateEvent);
|
|
|
|
voiceState.self_stream = true;
|
|
await voiceState.save();
|
|
|
|
await emitEvent({
|
|
event: "VOICE_STATE_UPDATE",
|
|
data: voiceState.toPublicVoiceState(),
|
|
guild_id: voiceState.guild_id,
|
|
channel_id: voiceState.channel_id,
|
|
} as VoiceStateUpdateEvent);
|
|
|
|
console.log(
|
|
`[Gateway] STREAM_CREATE for user ${this.user_id} in channel ${body.channel_id} with stream key ${streamKey} in ${Date.now() - startTime}ms`,
|
|
);
|
|
}
|
|
|
|
//stream key:
|
|
// guild:${guild_id}:${channel_id}:${user_id}
|
|
// call:${channel_id}:${user_id}
|