Initial commit

master
paring 2023-06-28 23:03:11 +09:00
commit 6dece81a93
Signed by: pikokr
GPG Key ID: 8C7ABCEF704FD728
13 changed files with 1684 additions and 0 deletions

135
.gitignore vendored Normal file
View File

@ -0,0 +1,135 @@
### 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
.idea

6
.prettierrc Normal file
View File

@ -0,0 +1,6 @@
{
"semi": false,
"endOfLine": "lf",
"useTabs": true,
"tabWidth": 4
}

32
package.json Normal file
View File

@ -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"
}
}

1001
pnpm-lock.yaml Normal file

File diff suppressed because it is too large Load Diff

22
src/api.ts Normal file
View File

@ -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()

42
src/commands/start.ts Normal file
View File

@ -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: "냥냥",
})
}

12
src/config.ts Normal file
View File

@ -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")
)

109
src/index.ts Normal file
View File

@ -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()

3
src/utils/appid.ts Normal file
View File

@ -0,0 +1,3 @@
export const getAppIdFromToken = (token: string) => {
return atob(token.split(".")[0])
}

30
src/utils/commands.ts Normal file
View File

@ -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",
},
]

132
src/utils/sessions.ts Normal file
View File

@ -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`)
})
}
}

View File

@ -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
}
}

116
tsconfig.json Normal file
View File

@ -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. */
}
}