Initial commit
commit
cf80b55254
GPG Key ID:
8C7ABCEF704FD728
|
|
@ -0,0 +1,134 @@
|
||||||
|
### Node template
|
||||||
|
# Logs
|
||||||
|
logs
|
||||||
|
*.log
|
||||||
|
npm-debug.log*
|
||||||
|
yarn-debug.log*
|
||||||
|
yarn-error.log*
|
||||||
|
lerna-debug.log*
|
||||||
|
.pnpm-debug.log*
|
||||||
|
|
||||||
|
# Diagnostic reports (https://nodejs.org/api/report.html)
|
||||||
|
report.[0-9]*.[0-9]*.[0-9]*.[0-9]*.json
|
||||||
|
|
||||||
|
# Runtime data
|
||||||
|
pids
|
||||||
|
*.pid
|
||||||
|
*.seed
|
||||||
|
*.pid.lock
|
||||||
|
|
||||||
|
# Directory for instrumented libs generated by jscoverage/JSCover
|
||||||
|
lib-cov
|
||||||
|
|
||||||
|
# Coverage directory used by tools like istanbul
|
||||||
|
coverage
|
||||||
|
*.lcov
|
||||||
|
|
||||||
|
# nyc test coverage
|
||||||
|
.nyc_output
|
||||||
|
|
||||||
|
# Grunt intermediate storage (https://gruntjs.com/creating-plugins#storing-task-files)
|
||||||
|
.grunt
|
||||||
|
|
||||||
|
# Bower dependency directory (https://bower.io/)
|
||||||
|
bower_components
|
||||||
|
|
||||||
|
# node-waf configuration
|
||||||
|
.lock-wscript
|
||||||
|
|
||||||
|
# Compiled binary addons (https://nodejs.org/api/addons.html)
|
||||||
|
build/Release
|
||||||
|
|
||||||
|
# Dependency directories
|
||||||
|
node_modules/
|
||||||
|
jspm_packages/
|
||||||
|
|
||||||
|
# Snowpack dependency directory (https://snowpack.dev/)
|
||||||
|
web_modules/
|
||||||
|
|
||||||
|
# TypeScript cache
|
||||||
|
*.tsbuildinfo
|
||||||
|
|
||||||
|
# Optional npm cache directory
|
||||||
|
.npm
|
||||||
|
|
||||||
|
# Optional eslint cache
|
||||||
|
.eslintcache
|
||||||
|
|
||||||
|
# Optional stylelint cache
|
||||||
|
.stylelintcache
|
||||||
|
|
||||||
|
# Microbundle cache
|
||||||
|
.rpt2_cache/
|
||||||
|
.rts2_cache_cjs/
|
||||||
|
.rts2_cache_es/
|
||||||
|
.rts2_cache_umd/
|
||||||
|
|
||||||
|
# Optional REPL history
|
||||||
|
.node_repl_history
|
||||||
|
|
||||||
|
# Output of 'npm pack'
|
||||||
|
*.tgz
|
||||||
|
|
||||||
|
# Yarn Integrity file
|
||||||
|
.yarn-integrity
|
||||||
|
|
||||||
|
# dotenv environment variable files
|
||||||
|
.env
|
||||||
|
.env.development.local
|
||||||
|
.env.test.local
|
||||||
|
.env.production.local
|
||||||
|
.env.local
|
||||||
|
|
||||||
|
# parcel-bundler cache (https://parceljs.org/)
|
||||||
|
.cache
|
||||||
|
.parcel-cache
|
||||||
|
|
||||||
|
# Next.js build output
|
||||||
|
.next
|
||||||
|
out
|
||||||
|
|
||||||
|
# Nuxt.js build / generate output
|
||||||
|
.nuxt
|
||||||
|
dist
|
||||||
|
|
||||||
|
# Gatsby files
|
||||||
|
.cache/
|
||||||
|
# Comment in the public line in if your project uses Gatsby and not Next.js
|
||||||
|
# https://nextjs.org/blog/next-9-1#public-directory-support
|
||||||
|
# public
|
||||||
|
|
||||||
|
# vuepress build output
|
||||||
|
.vuepress/dist
|
||||||
|
|
||||||
|
# vuepress v2.x temp and cache directory
|
||||||
|
.temp
|
||||||
|
.cache
|
||||||
|
|
||||||
|
# Docusaurus cache and generated files
|
||||||
|
.docusaurus
|
||||||
|
|
||||||
|
# Serverless directories
|
||||||
|
.serverless/
|
||||||
|
|
||||||
|
# FuseBox cache
|
||||||
|
.fusebox/
|
||||||
|
|
||||||
|
# DynamoDB Local files
|
||||||
|
.dynamodb/
|
||||||
|
|
||||||
|
# TernJS port file
|
||||||
|
.tern-port
|
||||||
|
|
||||||
|
# Stores VSCode versions used for testing VSCode extensions
|
||||||
|
.vscode-test
|
||||||
|
|
||||||
|
# yarn v2
|
||||||
|
.yarn/cache
|
||||||
|
.yarn/unplugged
|
||||||
|
.yarn/build-state.yml
|
||||||
|
.yarn/install-state.gz
|
||||||
|
.pnp.*
|
||||||
|
|
||||||
|
config.toml
|
||||||
|
recordings
|
||||||
|
|
@ -0,0 +1,5 @@
|
||||||
|
# Default ignored files
|
||||||
|
/shelf/
|
||||||
|
/workspace.xml
|
||||||
|
# Editor-based HTTP Client requests
|
||||||
|
/httpRequests/
|
||||||
|
|
@ -0,0 +1,58 @@
|
||||||
|
<component name="ProjectCodeStyleConfiguration">
|
||||||
|
<code_scheme name="Project" version="173">
|
||||||
|
<option name="LINE_SEPARATOR" value=" " />
|
||||||
|
<HTMLCodeStyleSettings>
|
||||||
|
<option name="HTML_SPACE_INSIDE_EMPTY_TAG" value="true" />
|
||||||
|
<option name="HTML_ENFORCE_QUOTES" value="true" />
|
||||||
|
</HTMLCodeStyleSettings>
|
||||||
|
<JSCodeStyleSettings version="0">
|
||||||
|
<option name="USE_SEMICOLON_AFTER_STATEMENT" value="false" />
|
||||||
|
<option name="FORCE_SEMICOLON_STYLE" value="true" />
|
||||||
|
<option name="SPACE_BEFORE_FUNCTION_LEFT_PARENTH" value="false" />
|
||||||
|
<option name="FORCE_QUOTE_STYlE" value="true" />
|
||||||
|
<option name="ENFORCE_TRAILING_COMMA" value="Remove" />
|
||||||
|
<option name="SPACES_WITHIN_OBJECT_LITERAL_BRACES" value="true" />
|
||||||
|
<option name="SPACES_WITHIN_IMPORTS" value="true" />
|
||||||
|
</JSCodeStyleSettings>
|
||||||
|
<TypeScriptCodeStyleSettings version="0">
|
||||||
|
<option name="USE_SEMICOLON_AFTER_STATEMENT" value="false" />
|
||||||
|
<option name="FORCE_SEMICOLON_STYLE" value="true" />
|
||||||
|
<option name="SPACE_BEFORE_FUNCTION_LEFT_PARENTH" value="false" />
|
||||||
|
<option name="FORCE_QUOTE_STYlE" value="true" />
|
||||||
|
<option name="ENFORCE_TRAILING_COMMA" value="Remove" />
|
||||||
|
<option name="SPACES_WITHIN_OBJECT_LITERAL_BRACES" value="true" />
|
||||||
|
<option name="SPACES_WITHIN_IMPORTS" value="true" />
|
||||||
|
</TypeScriptCodeStyleSettings>
|
||||||
|
<VueCodeStyleSettings>
|
||||||
|
<option name="INTERPOLATION_NEW_LINE_AFTER_START_DELIMITER" value="false" />
|
||||||
|
<option name="INTERPOLATION_NEW_LINE_BEFORE_END_DELIMITER" value="false" />
|
||||||
|
</VueCodeStyleSettings>
|
||||||
|
<codeStyleSettings language="HTML">
|
||||||
|
<option name="SOFT_MARGINS" value="80" />
|
||||||
|
<indentOptions>
|
||||||
|
<option name="CONTINUATION_INDENT_SIZE" value="4" />
|
||||||
|
<option name="USE_TAB_CHARACTER" value="true" />
|
||||||
|
</indentOptions>
|
||||||
|
</codeStyleSettings>
|
||||||
|
<codeStyleSettings language="JavaScript">
|
||||||
|
<option name="SOFT_MARGINS" value="80" />
|
||||||
|
<indentOptions>
|
||||||
|
<option name="USE_TAB_CHARACTER" value="true" />
|
||||||
|
</indentOptions>
|
||||||
|
</codeStyleSettings>
|
||||||
|
<codeStyleSettings language="TypeScript">
|
||||||
|
<option name="SOFT_MARGINS" value="80" />
|
||||||
|
<indentOptions>
|
||||||
|
<option name="USE_TAB_CHARACTER" value="true" />
|
||||||
|
</indentOptions>
|
||||||
|
</codeStyleSettings>
|
||||||
|
<codeStyleSettings language="Vue">
|
||||||
|
<option name="SOFT_MARGINS" value="80" />
|
||||||
|
<indentOptions>
|
||||||
|
<option name="INDENT_SIZE" value="4" />
|
||||||
|
<option name="TAB_SIZE" value="4" />
|
||||||
|
<option name="USE_TAB_CHARACTER" value="true" />
|
||||||
|
</indentOptions>
|
||||||
|
</codeStyleSettings>
|
||||||
|
</code_scheme>
|
||||||
|
</component>
|
||||||
|
|
@ -0,0 +1,5 @@
|
||||||
|
<component name="ProjectCodeStyleConfiguration">
|
||||||
|
<state>
|
||||||
|
<option name="USE_PER_PROJECT_SETTINGS" value="true" />
|
||||||
|
</state>
|
||||||
|
</component>
|
||||||
|
|
@ -0,0 +1,13 @@
|
||||||
|
<?xml version="1.0" encoding="UTF-8"?>
|
||||||
|
<module type="WEB_MODULE" version="4">
|
||||||
|
<component name="NewModuleRootManager">
|
||||||
|
<content url="file://$MODULE_DIR$">
|
||||||
|
<excludeFolder url="file://$MODULE_DIR$/.tmp" />
|
||||||
|
<excludeFolder url="file://$MODULE_DIR$/temp" />
|
||||||
|
<excludeFolder url="file://$MODULE_DIR$/tmp" />
|
||||||
|
<excludeFolder url="file://$MODULE_DIR$/recordings" />
|
||||||
|
</content>
|
||||||
|
<orderEntry type="inheritedJdk" />
|
||||||
|
<orderEntry type="sourceFolder" forTests="false" />
|
||||||
|
</component>
|
||||||
|
</module>
|
||||||
|
|
@ -0,0 +1,8 @@
|
||||||
|
<?xml version="1.0" encoding="UTF-8"?>
|
||||||
|
<project version="4">
|
||||||
|
<component name="ProjectModuleManager">
|
||||||
|
<modules>
|
||||||
|
<module fileurl="file://$PROJECT_DIR$/.idea/koibot.iml" filepath="$PROJECT_DIR$/.idea/koibot.iml" />
|
||||||
|
</modules>
|
||||||
|
</component>
|
||||||
|
</project>
|
||||||
|
|
@ -0,0 +1,7 @@
|
||||||
|
<?xml version="1.0" encoding="UTF-8"?>
|
||||||
|
<project version="4">
|
||||||
|
<component name="PrettierConfiguration">
|
||||||
|
<option name="myConfigurationMode" value="AUTOMATIC" />
|
||||||
|
<option name="myRunOnSave" value="true" />
|
||||||
|
</component>
|
||||||
|
</project>
|
||||||
|
|
@ -0,0 +1,6 @@
|
||||||
|
<?xml version="1.0" encoding="UTF-8"?>
|
||||||
|
<project version="4">
|
||||||
|
<component name="VcsDirectoryMappings">
|
||||||
|
<mapping directory="$PROJECT_DIR$" vcs="Git" />
|
||||||
|
</component>
|
||||||
|
</project>
|
||||||
|
|
@ -0,0 +1,6 @@
|
||||||
|
{
|
||||||
|
"semi": false,
|
||||||
|
"endOfLine": "lf",
|
||||||
|
"useTabs": true,
|
||||||
|
"tabWidth": 4
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,32 @@
|
||||||
|
{
|
||||||
|
"name": "koibot",
|
||||||
|
"version": "1.0.0",
|
||||||
|
"description": "",
|
||||||
|
"main": "dist/index.js",
|
||||||
|
"scripts": {
|
||||||
|
"dev": "tsx src/index.ts"
|
||||||
|
},
|
||||||
|
"keywords": [],
|
||||||
|
"author": "",
|
||||||
|
"license": "ISC",
|
||||||
|
"devDependencies": {
|
||||||
|
"@types/node": "^20.3.2",
|
||||||
|
"prettier": "^2.8.8",
|
||||||
|
"tsx": "^3.12.7",
|
||||||
|
"typescript": "^5.1.5"
|
||||||
|
},
|
||||||
|
"type": "module",
|
||||||
|
"dependencies": {
|
||||||
|
"@discordjs/collection": "^1.5.1",
|
||||||
|
"@discordjs/core": "^0.6.0",
|
||||||
|
"@discordjs/opus": "^0.9.0",
|
||||||
|
"@discordjs/rest": "^1.7.1",
|
||||||
|
"@discordjs/voice": "^0.16.0",
|
||||||
|
"@discordjs/ws": "^0.8.3",
|
||||||
|
"discord-api-types": "^0.37.46",
|
||||||
|
"libsodium-wrappers": "^0.7.11",
|
||||||
|
"node-crc": "1.3.2",
|
||||||
|
"prism-media": "2.0.0-alpha.0",
|
||||||
|
"toml": "^3.0.0"
|
||||||
|
}
|
||||||
|
}
|
||||||
File diff suppressed because it is too large
Load Diff
|
|
@ -0,0 +1,22 @@
|
||||||
|
import { REST } from "@discordjs/rest"
|
||||||
|
import { WebSocketManager } from "@discordjs/ws"
|
||||||
|
import { Client, GatewayIntentBits } from "@discordjs/core"
|
||||||
|
import { config } from "./config.js"
|
||||||
|
import { VoiceStateManager } from "./utils/wrapper/VoiceState.js"
|
||||||
|
import { RecordingSessionManager } from "./utils/sessions.js"
|
||||||
|
|
||||||
|
export const rest = new REST({ version: "10" }).setToken(config.token)
|
||||||
|
|
||||||
|
export const gateway = new WebSocketManager({
|
||||||
|
token: config.token,
|
||||||
|
intents: GatewayIntentBits.Guilds | GatewayIntentBits.GuildVoiceStates,
|
||||||
|
rest,
|
||||||
|
shardIds: [0],
|
||||||
|
shardCount: 1,
|
||||||
|
})
|
||||||
|
|
||||||
|
export const client = new Client({ rest, gateway })
|
||||||
|
|
||||||
|
export const voiceStateManager = new VoiceStateManager()
|
||||||
|
|
||||||
|
export const sessionManager = new RecordingSessionManager()
|
||||||
|
|
@ -0,0 +1,42 @@
|
||||||
|
import {
|
||||||
|
APIChatInputApplicationCommandInteraction,
|
||||||
|
WithIntrinsicProps,
|
||||||
|
} from "@discordjs/core"
|
||||||
|
import {
|
||||||
|
client,
|
||||||
|
gateway,
|
||||||
|
rest,
|
||||||
|
sessionManager,
|
||||||
|
voiceStateManager,
|
||||||
|
} from "../api.js"
|
||||||
|
import { Routes } from "discord-api-types/v10"
|
||||||
|
import { joinVoiceChannel } from "@discordjs/voice"
|
||||||
|
|
||||||
|
export const startCommand = async ({
|
||||||
|
data,
|
||||||
|
api,
|
||||||
|
shardId,
|
||||||
|
}: WithIntrinsicProps<APIChatInputApplicationCommandInteraction>) => {
|
||||||
|
if (!data.member || !data.guild_id)
|
||||||
|
return api.interactions.reply(data.id, data.token, {
|
||||||
|
content: "이 명령어는 서버에서 사용되어야 해요!",
|
||||||
|
})
|
||||||
|
|
||||||
|
if (sessionManager.hasSession(data.guild_id))
|
||||||
|
return api.interactions.reply(data.id, data.token, {
|
||||||
|
content: "이미 이 서버에서는 녹음이 진행중이에요!",
|
||||||
|
})
|
||||||
|
|
||||||
|
const vc = voiceStateManager.getUserVC(data.guild_id, data.member.user.id)
|
||||||
|
|
||||||
|
if (!vc)
|
||||||
|
return api.interactions.reply(data.id, data.token, {
|
||||||
|
content: "먼저 음성 채널에 들어가주세요!",
|
||||||
|
})
|
||||||
|
|
||||||
|
sessionManager.addSession(data.guild_id, vc, shardId)
|
||||||
|
|
||||||
|
await api.interactions.reply(data.id, data.token, {
|
||||||
|
content: "냥냥",
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,12 @@
|
||||||
|
import toml from "toml"
|
||||||
|
import fs from "fs/promises"
|
||||||
|
|
||||||
|
export interface KoiConfig {
|
||||||
|
token: string
|
||||||
|
guilds: string[]
|
||||||
|
sync: boolean
|
||||||
|
}
|
||||||
|
|
||||||
|
export const config: KoiConfig = toml.parse(
|
||||||
|
await fs.readFile("config.toml", "utf-8")
|
||||||
|
)
|
||||||
|
|
@ -0,0 +1,109 @@
|
||||||
|
import { client, gateway, sessionManager, voiceStateManager } from "./api.js"
|
||||||
|
|
||||||
|
import {
|
||||||
|
ActivityType,
|
||||||
|
APIChatInputApplicationCommandInteraction,
|
||||||
|
ApplicationCommandType,
|
||||||
|
GatewayDispatchEvents,
|
||||||
|
InteractionType,
|
||||||
|
PresenceUpdateStatus,
|
||||||
|
WithIntrinsicProps,
|
||||||
|
} from "@discordjs/core"
|
||||||
|
import { startCommand } from "./commands/start.js"
|
||||||
|
import { config } from "./config.js"
|
||||||
|
import { syncCommands } from "./utils/commands.js"
|
||||||
|
import { VoiceState } from "./utils/wrapper/VoiceState.js"
|
||||||
|
|
||||||
|
process.on("uncaughtException", (error) => console.error(error))
|
||||||
|
process.on("unhandledRejection", (error) => console.error(error))
|
||||||
|
|
||||||
|
if (config.sync) {
|
||||||
|
await syncCommands()
|
||||||
|
}
|
||||||
|
|
||||||
|
client.on(GatewayDispatchEvents.GuildCreate, (args) => {
|
||||||
|
console.log(`ADD GUILD: ${args.data.name}(${args.data.id})`)
|
||||||
|
voiceStateManager.addGuild(args.data.id, args.data.voice_states)
|
||||||
|
})
|
||||||
|
|
||||||
|
client.on(GatewayDispatchEvents.GuildDelete, (args) => {
|
||||||
|
voiceStateManager.removeGuild(args.data.id)
|
||||||
|
})
|
||||||
|
|
||||||
|
client.on(GatewayDispatchEvents.Ready, async (ready) => {
|
||||||
|
console.log("ready", ready.data.user.username)
|
||||||
|
|
||||||
|
await Promise.all(
|
||||||
|
(
|
||||||
|
await gateway.getShardIds()
|
||||||
|
).map((shard) =>
|
||||||
|
client
|
||||||
|
.updatePresence(shard, {
|
||||||
|
afk: false,
|
||||||
|
since: Date.now(),
|
||||||
|
status: PresenceUpdateStatus.Online,
|
||||||
|
activities: [
|
||||||
|
{
|
||||||
|
type: ActivityType.Playing,
|
||||||
|
name: "냥냥",
|
||||||
|
},
|
||||||
|
],
|
||||||
|
})
|
||||||
|
.then(() => {
|
||||||
|
console.log("updated presence for shard", shard)
|
||||||
|
})
|
||||||
|
)
|
||||||
|
)
|
||||||
|
})
|
||||||
|
|
||||||
|
client.on(GatewayDispatchEvents.InteractionCreate, async (args) => {
|
||||||
|
if (args.data.type === InteractionType.ApplicationCommand) {
|
||||||
|
if (args.data.data.type === ApplicationCommandType.ChatInput) {
|
||||||
|
const command = args.data.data.name
|
||||||
|
|
||||||
|
let commandFn:
|
||||||
|
| ((
|
||||||
|
args: WithIntrinsicProps<APIChatInputApplicationCommandInteraction>
|
||||||
|
) => Promise<void>)
|
||||||
|
| null = null
|
||||||
|
|
||||||
|
switch (command) {
|
||||||
|
case "start":
|
||||||
|
commandFn = startCommand
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!commandFn) return
|
||||||
|
|
||||||
|
await commandFn(
|
||||||
|
args as WithIntrinsicProps<APIChatInputApplicationCommandInteraction>
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
client.on(GatewayDispatchEvents.VoiceStateUpdate, async (args) => {
|
||||||
|
if (args.data.guild_id) {
|
||||||
|
const adapter = sessionManager.adapters.get(args.data.guild_id)
|
||||||
|
if (adapter) adapter.onVoiceStateUpdate(args.data)
|
||||||
|
const guild = voiceStateManager.guilds.get(args.data.guild_id)
|
||||||
|
if (guild) {
|
||||||
|
const state = guild.get(args.data.user_id)
|
||||||
|
if (state) {
|
||||||
|
state.update(args.data)
|
||||||
|
} else {
|
||||||
|
guild.set(
|
||||||
|
args.data.user_id,
|
||||||
|
new VoiceState(args.data, args.data.guild_id)
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
})
|
||||||
|
|
||||||
|
client.on(GatewayDispatchEvents.VoiceServerUpdate, (args) => {
|
||||||
|
const adapter = sessionManager.adapters.get(args.data.guild_id)
|
||||||
|
if (adapter) adapter.onVoiceServerUpdate(args.data)
|
||||||
|
})
|
||||||
|
|
||||||
|
await gateway.connect()
|
||||||
|
|
@ -0,0 +1,3 @@
|
||||||
|
export const getAppIdFromToken = (token: string) => {
|
||||||
|
return atob(token.split(".")[0])
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,30 @@
|
||||||
|
import { client, rest } from "../api.js"
|
||||||
|
import {
|
||||||
|
ApplicationCommandType,
|
||||||
|
RESTPutAPIApplicationGuildCommandsJSONBody,
|
||||||
|
Routes,
|
||||||
|
} from "discord-api-types/v10"
|
||||||
|
import { getAppIdFromToken } from "./appid.js"
|
||||||
|
import { config } from "../config.js"
|
||||||
|
|
||||||
|
export const syncCommands = async () => {
|
||||||
|
console.log("Syncing commands...")
|
||||||
|
for (const guild of config.guilds) {
|
||||||
|
console.log("Syncing for", guild)
|
||||||
|
await client.api.applicationCommands.bulkOverwriteGuildCommands(
|
||||||
|
getAppIdFromToken(config.token),
|
||||||
|
guild,
|
||||||
|
commands
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
console.log("------------ Done! ------------")
|
||||||
|
}
|
||||||
|
|
||||||
|
const commands: RESTPutAPIApplicationGuildCommandsJSONBody = [
|
||||||
|
{
|
||||||
|
type: ApplicationCommandType.ChatInput,
|
||||||
|
name: "start",
|
||||||
|
description: "start recording",
|
||||||
|
},
|
||||||
|
]
|
||||||
|
|
@ -0,0 +1,132 @@
|
||||||
|
import { Collection } from "@discordjs/collection"
|
||||||
|
import { gateway, voiceStateManager } from "../api.js"
|
||||||
|
import {
|
||||||
|
AudioReceiveStream,
|
||||||
|
DiscordGatewayAdapterLibraryMethods,
|
||||||
|
EndBehaviorType,
|
||||||
|
joinVoiceChannel,
|
||||||
|
VoiceConnection,
|
||||||
|
} from "@discordjs/voice"
|
||||||
|
import { Snowflake } from "discord-api-types/v10"
|
||||||
|
import * as prism from "prism-media"
|
||||||
|
import { pipeline } from "stream"
|
||||||
|
import { createWriteStream, existsSync, mkdirSync } from "fs"
|
||||||
|
import { getAppIdFromToken } from "./appid.js"
|
||||||
|
import { config } from "../config.js"
|
||||||
|
import path from "path"
|
||||||
|
|
||||||
|
export class RecordingSessionManager {
|
||||||
|
sessionMap = new Collection<Snowflake, RecordingSession>()
|
||||||
|
adapters = new Collection<Snowflake, DiscordGatewayAdapterLibraryMethods>()
|
||||||
|
|
||||||
|
hasSession(guild: Snowflake): boolean {
|
||||||
|
return this.sessionMap.has(guild)
|
||||||
|
}
|
||||||
|
|
||||||
|
addSession(guild: Snowflake, channel: Snowflake, shardId: number) {
|
||||||
|
const session = new RecordingSession(
|
||||||
|
joinVoiceChannel({
|
||||||
|
guildId: guild,
|
||||||
|
channelId: channel,
|
||||||
|
adapterCreator: (methods) => {
|
||||||
|
this.adapters.set(guild, methods)
|
||||||
|
return {
|
||||||
|
sendPayload: (payload) => {
|
||||||
|
gateway.send(shardId, payload)
|
||||||
|
return true
|
||||||
|
},
|
||||||
|
destroy: () => {
|
||||||
|
this.adapters.delete(guild)
|
||||||
|
},
|
||||||
|
}
|
||||||
|
},
|
||||||
|
selfDeaf: false,
|
||||||
|
selfMute: true,
|
||||||
|
debug: true,
|
||||||
|
})
|
||||||
|
)
|
||||||
|
|
||||||
|
this.sessionMap.set(guild, session)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export class RecordingSession {
|
||||||
|
id: string
|
||||||
|
streams = new Collection<Snowflake, UserStream>()
|
||||||
|
startedAt = Date.now()
|
||||||
|
|
||||||
|
constructor(public vc: VoiceConnection) {
|
||||||
|
this.id = Date.now().toString()
|
||||||
|
vc.receiver.speaking.on("start", (userId) => this.addReceiver(userId))
|
||||||
|
vc.receiver.speaking.on("end", (userId) => this.removeReceiver(userId))
|
||||||
|
}
|
||||||
|
|
||||||
|
get receiver() {
|
||||||
|
return this.vc.receiver
|
||||||
|
}
|
||||||
|
|
||||||
|
addReceiver(userId: string) {
|
||||||
|
if (userId === getAppIdFromToken(config.token)) return
|
||||||
|
if (this.streams.has(userId)) return
|
||||||
|
|
||||||
|
console.log("Starting record:", userId)
|
||||||
|
|
||||||
|
const receiver = this.receiver.subscribe(userId)
|
||||||
|
|
||||||
|
this.streams.set(userId, new UserStream(this, receiver, userId))
|
||||||
|
}
|
||||||
|
|
||||||
|
removeReceiver(userId: string) {
|
||||||
|
const stream = this.streams.get(userId)
|
||||||
|
|
||||||
|
if (stream) {
|
||||||
|
console.log("Stopping record:", userId)
|
||||||
|
stream.subscription.push(null)
|
||||||
|
this.streams.delete(userId)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
class UserStream {
|
||||||
|
oggStream: prism.opus.OggLogicalBitstream
|
||||||
|
|
||||||
|
constructor(
|
||||||
|
public session: RecordingSession,
|
||||||
|
public subscription: AudioReceiveStream,
|
||||||
|
public userId: string
|
||||||
|
) {
|
||||||
|
this.oggStream = new prism.opus.OggLogicalBitstream({
|
||||||
|
opusHead: new prism.opus.OpusHead({
|
||||||
|
channelCount: 2,
|
||||||
|
sampleRate: 48000,
|
||||||
|
}),
|
||||||
|
pageSizeControl: {
|
||||||
|
maxPackets: 10,
|
||||||
|
},
|
||||||
|
})
|
||||||
|
|
||||||
|
const dir = path.resolve("recordings", session.id, userId)
|
||||||
|
|
||||||
|
if (!existsSync(dir)) {
|
||||||
|
mkdirSync(dir, {
|
||||||
|
recursive: true,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
const filename = path.join(dir, `${Date.now() - session.startedAt}.ogg`)
|
||||||
|
|
||||||
|
const outputStream = createWriteStream(filename)
|
||||||
|
|
||||||
|
pipeline(subscription, this.oggStream, outputStream, (err) => {
|
||||||
|
outputStream.end()
|
||||||
|
|
||||||
|
if (err) {
|
||||||
|
console.log(`Recording ${userId} ended with error: ${err}`)
|
||||||
|
subscription.destroy(err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
console.log(`Recording ${userId} ended`)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,44 @@
|
||||||
|
import { GatewayVoiceState, Snowflake } from "discord-api-types/v10"
|
||||||
|
import { Collection } from "@discordjs/collection"
|
||||||
|
import { sessionManager } from "../../api.js"
|
||||||
|
|
||||||
|
type GuildVoiceState = Omit<GatewayVoiceState, "guild_id">
|
||||||
|
|
||||||
|
export class VoiceStateManager {
|
||||||
|
guilds = new Collection<string, Collection<string, VoiceState>>()
|
||||||
|
|
||||||
|
addGuild(id: Snowflake, states: GuildVoiceState[]) {
|
||||||
|
const guild = new Collection<Snowflake, VoiceState>()
|
||||||
|
for (const state of states) {
|
||||||
|
guild.set(state.user_id, new VoiceState(state, id))
|
||||||
|
}
|
||||||
|
this.guilds.set(id, guild)
|
||||||
|
}
|
||||||
|
|
||||||
|
removeGuild(id: Snowflake) {
|
||||||
|
this.guilds.delete(id)
|
||||||
|
}
|
||||||
|
|
||||||
|
getUserVC(guild: Snowflake, member: Snowflake): string | null {
|
||||||
|
const guildStates = this.guilds.get(guild)
|
||||||
|
if (!guildStates) return null
|
||||||
|
return guildStates.get(member)?.channelId ?? null
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
export class VoiceState {
|
||||||
|
id!: Snowflake
|
||||||
|
channelId!: Snowflake | null
|
||||||
|
|
||||||
|
mute!: boolean
|
||||||
|
|
||||||
|
constructor(data: GuildVoiceState, public guild: Snowflake) {
|
||||||
|
this.update(data)
|
||||||
|
}
|
||||||
|
|
||||||
|
update(data: GuildVoiceState) {
|
||||||
|
this.channelId = data.channel_id
|
||||||
|
this.id = data.user_id
|
||||||
|
this.mute = data.mute || data.self_mute
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,116 @@
|
||||||
|
{
|
||||||
|
"compilerOptions": {
|
||||||
|
/* Visit https://aka.ms/tsconfig to read more about this file */
|
||||||
|
|
||||||
|
/* Projects */
|
||||||
|
// "incremental": true, /* Save .tsbuildinfo files to allow for incremental compilation of projects. */
|
||||||
|
// "composite": true, /* Enable constraints that allow a TypeScript project to be used with project references. */
|
||||||
|
// "tsBuildInfoFile": "./.tsbuildinfo", /* Specify the path to .tsbuildinfo incremental compilation file. */
|
||||||
|
// "disableSourceOfProjectReferenceRedirect": true, /* Disable preferring source files instead of declaration files when referencing composite projects. */
|
||||||
|
// "disableSolutionSearching": true, /* Opt a project out of multi-project reference checking when editing. */
|
||||||
|
// "disableReferencedProjectLoad": true, /* Reduce the number of projects loaded automatically by TypeScript. */
|
||||||
|
|
||||||
|
/* Language and Environment */
|
||||||
|
"target": "ESNext",
|
||||||
|
/* Set the JavaScript language version for emitted JavaScript and include compatible library declarations. */
|
||||||
|
// "lib": [], /* Specify a set of bundled library declaration files that describe the target runtime environment. */
|
||||||
|
// "jsx": "preserve", /* Specify what JSX code is generated. */
|
||||||
|
// "experimentalDecorators": true, /* Enable experimental support for legacy experimental decorators. */
|
||||||
|
// "emitDecoratorMetadata": true, /* Emit design-type metadata for decorated declarations in source files. */
|
||||||
|
// "jsxFactory": "", /* Specify the JSX factory function used when targeting React JSX emit, e.g. 'React.createElement' or 'h'. */
|
||||||
|
// "jsxFragmentFactory": "", /* Specify the JSX Fragment reference used for fragments when targeting React JSX emit e.g. 'React.Fragment' or 'Fragment'. */
|
||||||
|
// "jsxImportSource": "", /* Specify module specifier used to import the JSX factory functions when using 'jsx: react-jsx*'. */
|
||||||
|
// "reactNamespace": "", /* Specify the object invoked for 'createElement'. This only applies when targeting 'react' JSX emit. */
|
||||||
|
// "noLib": true, /* Disable including any library files, including the default lib.d.ts. */
|
||||||
|
// "useDefineForClassFields": true, /* Emit ECMAScript-standard-compliant class fields. */
|
||||||
|
// "moduleDetection": "auto", /* Control what method is used to detect module-format JS files. */
|
||||||
|
|
||||||
|
/* Modules */
|
||||||
|
"module": "ESNext",
|
||||||
|
/* Specify what module code is generated. */
|
||||||
|
// "rootDir": "./", /* Specify the root folder within your source files. */
|
||||||
|
"moduleResolution": "NodeNext",
|
||||||
|
/* Specify how TypeScript looks up a file from a given module specifier. */
|
||||||
|
// "baseUrl": "./", /* Specify the base directory to resolve non-relative module names. */
|
||||||
|
// "paths": {}, /* Specify a set of entries that re-map imports to additional lookup locations. */
|
||||||
|
// "rootDirs": [], /* Allow multiple folders to be treated as one when resolving modules. */
|
||||||
|
// "typeRoots": [], /* Specify multiple folders that act like './node_modules/@types'. */
|
||||||
|
// "types": [], /* Specify type package names to be included without being referenced in a source file. */
|
||||||
|
// "allowUmdGlobalAccess": true, /* Allow accessing UMD globals from modules. */
|
||||||
|
// "moduleSuffixes": [], /* List of file name suffixes to search when resolving a module. */
|
||||||
|
// "allowImportingTsExtensions": true, /* Allow imports to include TypeScript file extensions. Requires '--moduleResolution bundler' and either '--noEmit' or '--emitDeclarationOnly' to be set. */
|
||||||
|
// "resolvePackageJsonExports": true, /* Use the package.json 'exports' field when resolving package imports. */
|
||||||
|
// "resolvePackageJsonImports": true, /* Use the package.json 'imports' field when resolving imports. */
|
||||||
|
// "customConditions": [], /* Conditions to set in addition to the resolver-specific defaults when resolving imports. */
|
||||||
|
// "resolveJsonModule": true, /* Enable importing .json files. */
|
||||||
|
// "allowArbitraryExtensions": true, /* Enable importing files with any extension, provided a declaration file is present. */
|
||||||
|
// "noResolve": true, /* Disallow 'import's, 'require's or '<reference>'s from expanding the number of files TypeScript should add to a project. */
|
||||||
|
|
||||||
|
/* JavaScript Support */
|
||||||
|
// "allowJs": true, /* Allow JavaScript files to be a part of your program. Use the 'checkJS' option to get errors from these files. */
|
||||||
|
// "checkJs": true, /* Enable error reporting in type-checked JavaScript files. */
|
||||||
|
// "maxNodeModuleJsDepth": 1, /* Specify the maximum folder depth used for checking JavaScript files from 'node_modules'. Only applicable with 'allowJs'. */
|
||||||
|
|
||||||
|
/* Emit */
|
||||||
|
// "declaration": true, /* Generate .d.ts files from TypeScript and JavaScript files in your project. */
|
||||||
|
// "declarationMap": true, /* Create sourcemaps for d.ts files. */
|
||||||
|
// "emitDeclarationOnly": true, /* Only output d.ts files and not JavaScript files. */
|
||||||
|
// "sourceMap": true, /* Create source map files for emitted JavaScript files. */
|
||||||
|
// "inlineSourceMap": true, /* Include sourcemap files inside the emitted JavaScript. */
|
||||||
|
// "outFile": "./", /* Specify a file that bundles all outputs into one JavaScript file. If 'declaration' is true, also designates a file that bundles all .d.ts output. */
|
||||||
|
// "outDir": "./", /* Specify an output folder for all emitted files. */
|
||||||
|
// "removeComments": true, /* Disable emitting comments. */
|
||||||
|
// "noEmit": true, /* Disable emitting files from a compilation. */
|
||||||
|
// "importHelpers": true, /* Allow importing helper functions from tslib once per project, instead of including them per-file. */
|
||||||
|
// "importsNotUsedAsValues": "remove", /* Specify emit/checking behavior for imports that are only used for types. */
|
||||||
|
// "downlevelIteration": true, /* Emit more compliant, but verbose and less performant JavaScript for iteration. */
|
||||||
|
// "sourceRoot": "", /* Specify the root path for debuggers to find the reference source code. */
|
||||||
|
// "mapRoot": "", /* Specify the location where debugger should locate map files instead of generated locations. */
|
||||||
|
// "inlineSources": true, /* Include source code in the sourcemaps inside the emitted JavaScript. */
|
||||||
|
// "emitBOM": true, /* Emit a UTF-8 Byte Order Mark (BOM) in the beginning of output files. */
|
||||||
|
// "newLine": "crlf", /* Set the newline character for emitting files. */
|
||||||
|
// "stripInternal": true, /* Disable emitting declarations that have '@internal' in their JSDoc comments. */
|
||||||
|
// "noEmitHelpers": true, /* Disable generating custom helper functions like '__extends' in compiled output. */
|
||||||
|
// "noEmitOnError": true, /* Disable emitting files if any type checking errors are reported. */
|
||||||
|
// "preserveConstEnums": true, /* Disable erasing 'const enum' declarations in generated code. */
|
||||||
|
// "declarationDir": "./", /* Specify the output directory for generated declaration files. */
|
||||||
|
// "preserveValueImports": true, /* Preserve unused imported values in the JavaScript output that would otherwise be removed. */
|
||||||
|
|
||||||
|
/* Interop Constraints */
|
||||||
|
// "isolatedModules": true, /* Ensure that each file can be safely transpiled without relying on other imports. */
|
||||||
|
// "verbatimModuleSyntax": true, /* Do not transform or elide any imports or exports not marked as type-only, ensuring they are written in the output file's format based on the 'module' setting. */
|
||||||
|
// "allowSyntheticDefaultImports": true, /* Allow 'import x from y' when a module doesn't have a default export. */
|
||||||
|
"esModuleInterop": true,
|
||||||
|
/* Emit additional JavaScript to ease support for importing CommonJS modules. This enables 'allowSyntheticDefaultImports' for type compatibility. */
|
||||||
|
// "preserveSymlinks": true, /* Disable resolving symlinks to their realpath. This correlates to the same flag in node. */
|
||||||
|
"forceConsistentCasingInFileNames": true,
|
||||||
|
/* Ensure that casing is correct in imports. */
|
||||||
|
|
||||||
|
/* Type Checking */
|
||||||
|
"strict": true,
|
||||||
|
/* Enable all strict type-checking options. */
|
||||||
|
// "noImplicitAny": true, /* Enable error reporting for expressions and declarations with an implied 'any' type. */
|
||||||
|
// "strictNullChecks": true, /* When type checking, take into account 'null' and 'undefined'. */
|
||||||
|
// "strictFunctionTypes": true, /* When assigning functions, check to ensure parameters and the return values are subtype-compatible. */
|
||||||
|
// "strictBindCallApply": true, /* Check that the arguments for 'bind', 'call', and 'apply' methods match the original function. */
|
||||||
|
// "strictPropertyInitialization": true, /* Check for class properties that are declared but not set in the constructor. */
|
||||||
|
// "noImplicitThis": true, /* Enable error reporting when 'this' is given the type 'any'. */
|
||||||
|
// "useUnknownInCatchVariables": true, /* Default catch clause variables as 'unknown' instead of 'any'. */
|
||||||
|
// "alwaysStrict": true, /* Ensure 'use strict' is always emitted. */
|
||||||
|
// "noUnusedLocals": true, /* Enable error reporting when local variables aren't read. */
|
||||||
|
// "noUnusedParameters": true, /* Raise an error when a function parameter isn't read. */
|
||||||
|
// "exactOptionalPropertyTypes": true, /* Interpret optional property types as written, rather than adding 'undefined'. */
|
||||||
|
// "noImplicitReturns": true, /* Enable error reporting for codepaths that do not explicitly return in a function. */
|
||||||
|
// "noFallthroughCasesInSwitch": true, /* Enable error reporting for fallthrough cases in switch statements. */
|
||||||
|
// "noUncheckedIndexedAccess": true, /* Add 'undefined' to a type when accessed using an index. */
|
||||||
|
// "noImplicitOverride": true, /* Ensure overriding members in derived classes are marked with an override modifier. */
|
||||||
|
// "noPropertyAccessFromIndexSignature": true, /* Enforces using indexed accessors for keys declared using an indexed type. */
|
||||||
|
// "allowUnusedLabels": true, /* Disable error reporting for unused labels. */
|
||||||
|
// "allowUnreachableCode": true, /* Disable error reporting for unreachable code. */
|
||||||
|
|
||||||
|
/* Completeness */
|
||||||
|
// "skipDefaultLibCheck": true, /* Skip type checking .d.ts files that are included with TypeScript. */
|
||||||
|
"skipLibCheck": true
|
||||||
|
/* Skip type checking all .d.ts files. */
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue