35 Commits

Author SHA1 Message Date
610074e4ac 일단 ffmpeg파일 인코딩/디코딩 예제 코드 완성 2025-09-05 01:47:32 +09:00
e3b5e92164 ffmpeg예제 코드 작성 중.. 2025-09-04 18:03:58 +09:00
b123b2ecdb 테스트 구조 완성, 코딩 스타일 교정 기능 추가 쓸데없는 설정파일들 정리 2025-09-01 02:29:37 +09:00
e26d20a869 스타일 정리 및 컴파일 오류 제거 2025-08-31 02:29:42 +09:00
f3974bb86f 쓸데없는 공유 라이브러리 제거 및 콘솔 실행 커맨드 안정화 2025-08-31 02:22:36 +09:00
f846dd7195 Merge branch 'refactor' of github.com:HappyTanuki/BumbleCee into refactor 2025-08-30 02:53:11 +09:00
0bff043d6e ytdlp 자동 다운로드 및 업데이트 로직 구현 2025-08-30 02:51:58 +09:00
7b75fc8861 윈도우 빌드용 매크로 추가 2025-08-29 05:56:36 +09:00
177221bf73 리눅스 빌드 성공 2025-08-29 05:53:47 +09:00
51a49a4470 ffmpeg 부를 때 무조건 extern c 하기... 2025-08-29 05:44:56 +09:00
a2821394bd 일단 빌드 되니 좋았쓰! 2025-08-29 04:53:35 +09:00
a03ed2dc78 정리 중.. 2025-08-28 21:34:21 +09:00
c59fef1d0b 일단 저장 2025-08-28 20:46:37 +09:00
125fbcc49c TODO 추가가 2025-03-09 17:29:09 +09:00
4915a5c3b8 TODO리스트는 늘어만 간다.. 2025-03-09 17:27:42 +09:00
4435337f40 TODO 추가가 2025-03-07 17:25:58 +09:00
77d16c1cdb 아마도? 쉘 인젝션 취약점 픽스 2025-03-07 17:19:58 +09:00
5760f1afdc todo 추가가 2025-02-13 17:42:17 +09:00
bb6cd89f51 Update README.md 2025-02-13 17:31:17 +09:00
2a238f804d 사람 나갈 때 음악 꺼지는 문제 해결결 2025-02-13 17:22:42 +09:00
ea7d42638e 코드 리팩터링 중.. 2025-02-13 17:18:03 +09:00
30f97e3dfb 3차 코드 리팩터터 2025-02-13 17:01:38 +09:00
17236f32b5 도커파일 이미지 크기최적화 2025-02-13 16:54:10 +09:00
00134ee7b1 2차 리팩터터 2025-02-13 16:50:03 +09:00
52146c1f6a 코드 가독성 리팩터 2025-02-13 16:43:11 +09:00
bf0268c1e9 Merge branch 'master' of github.com:HappyTanuki/BumbleCee 2025-02-03 01:42:06 +09:00
2f5185e0c3 큐 출력 스타일 변경 2025-02-03 01:42:02 +09:00
96a4096971 Update README.md 2025-02-03 01:32:55 +09:00
ec65b550e6 Update README.md 2025-02-03 01:32:19 +09:00
14db2a31e8 와 이제 진짜 진짜 버그 다 고친 듯? 2025-02-02 23:55:47 +09:00
1af829c711 Merge branch 'master' of github.com:HappyTanuki/BumbleCee 2025-02-02 07:51:56 +09:00
46c59d40de 도커허브 업로드 자동화 구현 2025-02-02 07:51:52 +09:00
b08641f87e Update README.md 2025-02-02 07:10:21 +09:00
1ca928c4df Update README.md 2025-02-02 07:09:57 +09:00
2935a844a0 아직 read loop ended는 못잡았지만 아무튼 완성(진) 일단 셔플 빼그 기능은 다 작동하니 된 게 아닐까? 2025-02-02 06:37:31 +09:00
53 changed files with 1081 additions and 1452 deletions

8
.clang-format Normal file
View File

@@ -0,0 +1,8 @@
# Google C/C++ Code Style Settings
Language: Cpp
BasedOnStyle: Google
Standard: c++20
IndentWidth: 2
UseTab: Never
ColumnLimit: 80

5
.gitignore vendored
View File

@@ -3,4 +3,7 @@ Temp/
Music
*.json
yt-dlp
ffmpeg
ffmpeg
password
.vs
out

10
.idea/.gitignore generated vendored
View File

@@ -1,10 +0,0 @@
# Default ignored files
/shelf/
/workspace.xml
# Editor-based HTTP Client requests
/httpRequests/
# Environment-dependent path to Maven home directory
/mavenHomeManager.xml
# Datasource local storage ignored files
/dataSources/
/dataSources.local.xml

9
.idea/BumbleCee.iml generated
View File

@@ -1,9 +0,0 @@
<?xml version="1.0" encoding="UTF-8"?>
<module type="JAVA_MODULE" version="4">
<component name="NewModuleRootManager" inherit-compiler-output="true">
<exclude-output />
<content url="file://$MODULE_DIR$" />
<orderEntry type="inheritedJdk" />
<orderEntry type="sourceFolder" forTests="false" />
</component>
</module>

6
.idea/misc.xml generated
View File

@@ -1,6 +0,0 @@
<?xml version="1.0" encoding="UTF-8"?>
<project version="4">
<component name="ProjectRootManager" version="2" languageLevel="JDK_23" default="true" project-jdk-name="openjdk-23" project-jdk-type="JavaSDK">
<output url="file://$PROJECT_DIR$/out" />
</component>
</project>

8
.idea/modules.xml generated
View File

@@ -1,8 +0,0 @@
<?xml version="1.0" encoding="UTF-8"?>
<project version="4">
<component name="ProjectModuleManager">
<modules>
<module fileurl="file://$PROJECT_DIR$/.idea/BumbleCee.iml" filepath="$PROJECT_DIR$/.idea/BumbleCee.iml" />
</modules>
</component>
</project>

6
.idea/vcs.xml generated
View File

@@ -1,6 +0,0 @@
<?xml version="1.0" encoding="UTF-8"?>
<project version="4">
<component name="VcsDirectoryMappings">
<mapping directory="" vcs="Git" />
</component>
</project>

View File

@@ -1,22 +0,0 @@
{
"configurations": [
{
"name": "Linux",
"includePath": [
"${workspaceFolder}/**",
"${workspaceFolder}/include"
],
"defines": [
"DDPP_CORO=on"
],
"compilerPath": "/usr/bin/gcc",
"cStandard": "c17",
"cppStandard": "c++20",
"intelliSenseMode": "linux-gcc-x64",
"compilerArgs": [
"-DDPP_CORO"
]
}
],
"version": 4
}

13
.vscode/launch.json vendored
View File

@@ -1,13 +1,17 @@
{
// Use IntelliSense to learn about possible attributes.
// Hover to view descriptions of existing attributes.
// For more information, visit: https://go.microsoft.com/fwlink/?linkid=830387
"version": "0.2.0",
"configurations": [
{
"name": "Debug",
"name": "(gdb) Launch",
"type": "cppdbg",
"request": "launch",
"program": "${workspaceFolder}/build/BumbleCee",
"program": "${workspaceFolder}/build/Debug Clang 18.1.3 x86_64-pc-linux-gnu/tests/${fileBasenameNoExtension}",
"args": [],
"stopAtEntry": false,
"cwd": "${workspaceFolder}",
"cwd": "${fileDirname}",
"environment": [],
"externalConsole": false,
"MIMode": "gdb",
@@ -22,8 +26,7 @@
"text": "-gdb-set disassembly-flavor intel",
"ignoreFailures": true
}
],
"preLaunchTask": "${defaultBuildTask}"
]
}
]
}

68
.vscode/settings.json vendored
View File

@@ -1,10 +1,17 @@
{
"editor.rulers": [
{
"column": 80
}
],
"editor.renderWhitespace": "boundary",
"editor.formatOnSave": true,
"cmake.generator": "Ninja",
"files.associations": {
"iosfwd": "cpp",
"sstream": "cpp",
"cctype": "cpp",
"clocale": "cpp",
"cmath": "cpp",
"csetjmp": "cpp",
"csignal": "cpp",
"cstdarg": "cpp",
"cstddef": "cpp",
@@ -18,11 +25,13 @@
"array": "cpp",
"atomic": "cpp",
"strstream": "cpp",
"barrier": "cpp",
"bit": "cpp",
"*.tcc": "cpp",
"bitset": "cpp",
"cfenv": "cpp",
"charconv": "cpp",
"chrono": "cpp",
"cinttypes": "cpp",
"codecvt": "cpp",
"compare": "cpp",
"complex": "cpp",
@@ -37,8 +46,10 @@
"set": "cpp",
"string": "cpp",
"unordered_map": "cpp",
"unordered_set": "cpp",
"vector": "cpp",
"exception": "cpp",
"expected": "cpp",
"algorithm": "cpp",
"functional": "cpp",
"iterator": "cpp",
@@ -48,60 +59,63 @@
"optional": "cpp",
"random": "cpp",
"ratio": "cpp",
"regex": "cpp",
"source_location": "cpp",
"string_view": "cpp",
"system_error": "cpp",
"tuple": "cpp",
"type_traits": "cpp",
"utility": "cpp",
"rope": "cpp",
"slist": "cpp",
"format": "cpp",
"fstream": "cpp",
"future": "cpp",
"initializer_list": "cpp",
"iomanip": "cpp",
"iosfwd": "cpp",
"iostream": "cpp",
"istream": "cpp",
"latch": "cpp",
"limits": "cpp",
"mutex": "cpp",
"new": "cpp",
"numbers": "cpp",
"ostream": "cpp",
"ranges": "cpp",
"scoped_allocator": "cpp",
"semaphore": "cpp",
"shared_mutex": "cpp",
"span": "cpp",
"spanstream": "cpp",
"sstream": "cpp",
"stacktrace": "cpp",
"stdexcept": "cpp",
"stdfloat": "cpp",
"stop_token": "cpp",
"streambuf": "cpp",
"syncstream": "cpp",
"thread": "cpp",
"cinttypes": "cpp",
"typeindex": "cpp",
"typeinfo": "cpp",
"valarray": "cpp",
"variant": "cpp",
"*.ipp": "cpp",
"format": "cpp",
"span": "cpp",
"__bit_reference": "cpp",
"__bits": "cpp",
"__config": "cpp",
"__debug": "cpp",
"__errc": "cpp",
"__hash_table": "cpp",
"__locale": "cpp",
"__mutex_base": "cpp",
"__node_handle": "cpp",
"__nullptr": "cpp",
"__split_buffer": "cpp",
"__string": "cpp",
"__threading_support": "cpp",
"__tree": "cpp",
"__tuple": "cpp",
"ios": "cpp",
"locale": "cpp",
"queue": "cpp",
"hash_map": "cpp",
"hash_set": "cpp",
"regex": "cpp",
"stack": "cpp",
"__memory": "cpp"
"print": "cpp",
"__bit_reference": "cpp",
"__hash_table": "cpp",
"__node_handle": "cpp",
"__split_buffer": "cpp",
"__threading_support": "cpp",
"__verbose_abort": "cpp",
"queue": "cpp"
},
"files.exclude": {
"**/*.rpyc": true,
"**/*.rpa": true,
"**/*.rpymc": true,
"**/cache/": true
}
}

35
.vscode/tasks.json vendored
View File

@@ -1,35 +0,0 @@
{
"version": "2.0.0",
"tasks": [
{
"type": "cppbuild",
"label": "make",
"command": "make",
"args": [],
"options": {
"cwd": "${workspaceFolder}/build/"
},
"group": {
"kind": "build",
"isDefault": true
},
"dependsOn": ["cmake"]
},
{
"type": "shell",
"label": "cmake",
"command": "cmake",
"args": [
".."
],
"options": {
"cwd": "${workspaceFolder}/build/"
},
"group": {
"kind": "build",
"isDefault": false
}
},
]
}

8
BuildDockerAndUpload.sh Executable file
View File

@@ -0,0 +1,8 @@
#!/bin/bash
(
cd build
cmake .. && make
)
docker login -u happytanuki12 --password-stdin < password
docker build --tag happytanuki12/bumblebee:latest .
docker push happytanuki12/bumblebee:latest

View File

@@ -1,40 +1,163 @@
cmake_minimum_required (VERSION 3.6)
list(APPEND CMAKE_MODULE_PATH ${CMAKE_CURRENT_SOURCE_DIR}/cmake)
cmake_minimum_required (VERSION 3.16)
set(BOT_NAME "BumbleCee")
project(${BOT_NAME} LANGUAGES CXX)
project(${BOT_NAME})
aux_source_directory("src" coresrc)
aux_source_directory("src/Audio" commands)
aux_source_directory("src/Commands" commands)
aux_source_directory("src/Queue" settings)
aux_source_directory("src/Settings" settings)
aux_source_directory("src/Utils" settings)
add_executable(${BOT_NAME} ${coresrc} ${commands} ${settings})
set(CMAKE_POSITION_INDEPENDENT_CODE ON)
set(CMAKE_BUILD_TYPE Debug)
set(CMAKE_CXX_STANDARD 20)
set(CMAKE_CXX_STANDARD_REQUIRED ON)
set(CMAKE_CXX_EXTENSIONS OFF)
set(CMAKE_EXPORT_COMPILE_COMMANDS ON)
set(BOOST_EXCLUDE_LIBRARIES nowide)
set(BUILD_TESTING ON)
set(DPP_BUILD_TEST OFF)
enable_testing()
if(WIN32)
set(OPENSSL_ROOT_DIR "C:/Program Files/OpenSSL-Win64")
set(OPENSSL_INCLUDE_DIR "${OPENSSL_ROOT_DIR}/include")
if(CMAKE_BUILD_TYPE STREQUAL Debug)
set(OPENSSL_CRYPTO_LIBRARY "${OPENSSL_ROOT_DIR}/lib/VC/x64/MTd/libcrypto.lib")
set(OPENSSL_SSL_LIBRARY "${OPENSSL_ROOT_DIR}/lib/VC/x64/MTd/libssl.lib")
else()
set(OPENSSL_CRYPTO_LIBRARY "${OPENSSL_ROOT_DIR}/lib/VC/x64/MT/libcrypto.lib")
set(OPENSSL_SSL_LIBRARY "${OPENSSL_ROOT_DIR}/lib/VC/x64/MT/libssl.lib")
endif()
endif()
set(THREADS_PREFER_PTHREAD_FLAG TRUE)
find_package(Threads REQUIRED)
find_package(OpenSSL REQUIRED)
target_include_directories(${BOT_NAME} PUBLIC
${CMAKE_CURRENT_SOURCE_DIR}/include
${OPENSSL_INCLUDE_DIR}
/usr/include/opus
)
include(FetchContent)
target_link_libraries(${BOT_NAME}
set(BUILD_SHARED_LIBS_OLD ${BUILD_SHARED_LIBS})
FetchContent_Declare(
Boost
URL "https://github.com/boostorg/boost/releases/download/boost-1.89.0/boost-1.89.0-cmake.7z"
DOWNLOAD_EXTRACT_TIMESTAMP ON
EXCLUDE_FROM_ALL
)
set(BUILD_SHARED_LIBS OFF CACHE BOOL "Do not build SHARED libraries" FORCE)
message(STATUS "Fetching and making available Boost...")
FetchContent_MakeAvailable(Boost)
FetchContent_Declare(
dpp
opus
ogg
oggz
${CMAKE_THREAD_LIBS_INIT}
${OPENSSL_CRYPTO_LIBRARY}
${OPENSSL_SSL_LIBRARY}
GIT_REPOSITORY "https://github.com/brainboxdotcc/DPP.git"
GIT_TAG "v10.1.3"
GIT_SHALLOW ON
)
set(BUILD_SHARED_LIBS ON CACHE BOOL "Build SHARED libraries" FORCE)
message(STATUS "Fetching and making available dpp...")
FetchContent_MakeAvailable(dpp)
set(BUILD_SHARED_LIBS ${BUILD_SHARED_LIBS_OLD} CACHE BOOL "Type of libraries to build" FORCE)
# -------------------------------------------------------------
# 플랫폼별 FFmpeg 바이너리 다운로드 및 링크
# -------------------------------------------------------------
if(CMAKE_HOST_SYSTEM_NAME STREQUAL "Windows")
set(FFMPEG_URL "https://github.com/BtbN/FFmpeg-Builds/releases/download/latest/ffmpeg-master-latest-win64-lgpl-shared.zip")
set(FFMPEG_ARCHIVE_NAME "ffmpeg-windows")
set(FFMPEG_LIB_DIR "lib")
set(FFMPEG_INCLUDE_DIR "include")
elseif(CMAKE_HOST_SYSTEM_NAME STREQUAL "Linux")
set(FFMPEG_URL "https://github.com/BtbN/FFmpeg-Builds/releases/download/latest/ffmpeg-master-latest-linux64-lgpl-shared.tar.xz")
set(FFMPEG_ARCHIVE_NAME "ffmpeg-linux")
set(FFMPEG_LIB_DIR "lib")
set(FFMPEG_INCLUDE_DIR "include")
else()
message(FATAL_ERROR "Unsupported platform for FFmpeg precompiled binaries.")
endif()
FetchContent_Declare(
ffmpeg
URL ${FFMPEG_URL}
DOWNLOAD_EXTRACT_TIMESTAMP ON
SOURCE_DIR ${CMAKE_BINARY_DIR}/${FFMPEG_ARCHIVE_NAME}
)
message(STATUS "Fetching and making available ffmpeg...")
FetchContent_MakeAvailable(ffmpeg)
set(FFMPEG_LIBRARY_PATH ${ffmpeg_SOURCE_DIR}/${FFMPEG_LIB_DIR})
set(FFMPEG_INCLUDE_PATH ${ffmpeg_SOURCE_DIR}/${FFMPEG_INCLUDE_DIR})
find_library(AVCODEC_LIBRARY NAMES avcodec PATHS ${FFMPEG_LIBRARY_PATH} NO_DEFAULT_PATH)
message(STATUS "AVCODEC_LIBRARY = ${AVCODEC_LIBRARY}")
find_library(AVFORMAT_LIBRARY NAMES avformat PATHS ${FFMPEG_LIBRARY_PATH} NO_DEFAULT_PATH)
message(STATUS "AVFORMAT_LIBRARY = ${AVFORMAT_LIBRARY}")
find_library(AVUTIL_LIBRARY NAMES avutil PATHS ${FFMPEG_LIBRARY_PATH} NO_DEFAULT_PATH)
message(STATUS "AVUTIL_LIBRARY = ${AVUTIL_LIBRARY}")
find_library(SWSCALE_LIBRARY NAMES swscale PATHS ${FFMPEG_LIBRARY_PATH} NO_DEFAULT_PATH)
message(STATUS "SWSCALE_LIBRARY = ${SWSCALE_LIBRARY}")
find_library(SWRESAMPLE_LIBRARY NAMES swresample PATHS ${FFMPEG_LIBRARY_PATH} NO_DEFAULT_PATH)
message(STATUS "SWRESAMPLE_LIBRARY = ${SWRESAMPLE_LIBRARY}")
if(NOT AVCODEC_LIBRARY OR NOT AVFORMAT_LIBRARY OR NOT AVUTIL_LIBRARY OR NOT SWRESAMPLE_LIBRARY)
message(FATAL_ERROR "FFmpeg 라이브러리를 찾을 수 없습니다. 다운로드 경로를 확인해주세요.")
endif()
file(GLOB_RECURSE SOURCES "src/*.cpp" "src/*.cxx" "src/*.cc")
add_library(${BOT_NAME}_lib ${SOURCES})
target_include_directories(${BOT_NAME}_lib PUBLIC ${CMAKE_CURRENT_SOURCE_DIR}/include)
target_include_directories(${BOT_NAME}_lib PUBLIC ${OpenSSL_INCLUDE_DIRS})
target_include_directories(${BOT_NAME}_lib PUBLIC ${FFMPEG_INCLUDE_PATH})
target_precompile_headers(${BOT_NAME}_lib PUBLIC "${CMAKE_CURRENT_SOURCE_DIR}/include/precomp.h")
target_link_libraries(${BOT_NAME}_lib PUBLIC dpp)
target_link_libraries(${BOT_NAME}_lib PUBLIC Boost::filesystem)
target_link_libraries(${BOT_NAME}_lib PUBLIC Boost::process)
target_link_libraries(${BOT_NAME}_lib PUBLIC Boost::log)
target_link_libraries(${BOT_NAME}_lib PUBLIC Boost::beast)
target_link_libraries(${BOT_NAME}_lib PUBLIC OpenSSL::Crypto)
target_link_libraries(${BOT_NAME}_lib PUBLIC OpenSSL::SSL)
target_link_libraries(${BOT_NAME}_lib PUBLIC ${AVUTIL_LIBRARY})
target_link_libraries(${BOT_NAME}_lib PUBLIC ${AVCODEC_LIBRARY})
target_link_libraries(${BOT_NAME}_lib PUBLIC ${AVFORMAT_LIBRARY})
target_link_libraries(${BOT_NAME}_lib PUBLIC ${SWSCALE_LIBRARY})
target_link_libraries(${BOT_NAME}_lib PUBLIC ${SWRESAMPLE_LIBRARY})
if(WIN32)
target_link_libraries(${BOT_NAME}_lib PUBLIC ws2_32)
endif()
if(UNIX AND NOT APPLE)
set_target_properties(${BOT_NAME}_lib PROPERTIES
BUILD_WITH_INSTALL_RPATH TRUE
INSTALL_RPATH "$ORIGIN"
SKIP_BUILD_RPATH FALSE
BUILD_RPATH "$ORIGIN"
)
endif()
add_executable(${BOT_NAME} ${SOURCES})
target_link_libraries(${BOT_NAME} PUBLIC ${BOT_NAME}_lib)
file(GLOB FFMPEG_SHARED
"${CMAKE_BINARY_DIR}/${FFMPEG_ARCHIVE_NAME}/bin/*.dll" # Windows
"${CMAKE_BINARY_DIR}/${FFMPEG_ARCHIVE_NAME}/lib/*.so*" # Linux
)
link_directories(/usr/lib)
foreach(ffmpeg_shared ${FFMPEG_SHARED})
add_custom_command(TARGET ${BOT_NAME} POST_BUILD
COMMAND ${CMAKE_COMMAND} -E copy_if_different
"${ffmpeg_shared}"
"$<TARGET_FILE_DIR:${BOT_NAME}>"
COMMENT "Copying ${ffmpeg_shared} to output directory"
)
endforeach()
add_custom_command(TARGET ${BOT_NAME} POST_BUILD
COMMAND ${CMAKE_COMMAND} -E copy_if_different
"$<TARGET_FILE:dpp>"
"$<TARGET_FILE_DIR:${BOT_NAME}>"
COMMENT "Copying dpp DLL/so files to output directory"
)
add_subdirectory(tests)

View File

@@ -1 +1,21 @@
FROM alpine
FROM debian:sid
WORKDIR /
RUN apt-get update && \
apt-get install -y curl libopus0 tini liboggz2 xz-utils ffmpeg python3 \
python3-pip python3-certifi python3-brotli python3-websockets python3-requests python3-mutagen && \
apt-get clean && \
rm -rf /var/lib/apt/lists/*
RUN pip3 install --break-system-packages --no-cache-dir curl_cffi
RUN pip3 install --break-system-packages --no-cache-dir pycryptodome
RUN curl -Lo dpp.deb https://dl.dpp.dev/latest
RUN curl -Lo dpp-legacy.deb https://github.com/brainboxdotcc/DPP/releases/download/v10.0.35/libdpp-10.0.35-linux-x64.deb
RUN dpkg -i dpp.deb
RUN dpkg -i dpp-legacy.deb
RUN rm dpp.deb
RUN curl -LO https://github.com/yt-dlp/yt-dlp/releases/latest/download/yt-dlp
RUN chmod +x ./yt-dlp
COPY ./build/BumbleCee /BumbleCee
COPY ./streamOpus.sh /streamOpus.sh
RUN chmod +x BumbleCee
RUN chmod +x streamOpus.sh
ENTRYPOINT ["/usr/bin/tini", "--", "./BumbleCee"]

View File

@@ -1,18 +1,25 @@
# 이게 뭔가요?
[![CodeFactor](https://www.codefactor.io/repository/github/happytanuki/bumblecee/badge)](https://www.codefactor.io/repository/github/happytanuki/bumblecee)
C++ Dpp 라이브러리를 이용해서 개발된 간단한 디스코드 음악봇입니다!
<div align="center">
<a href="https://github.com/brainboxdotcc/DPP" alt="DPP"> <img src="DPP-markdown-logo.png" /> </a>
</div>
# 어떻게 써요?
1. 실행파일 경로에 config.json 파일을 만들고 다음과 같이 입력하세요:
```
{"token": "디스코드에서 발급받은 개인 봇 토큰"}
{
"CLEAR_PREVIOUS_COMMAND": true,
"FFMPEG_CMD": "ffmpeg",
"LOGLEVEL": "debug",
"TOKEN": "발급받은 토큰",
"YTDLP_CMD": "./yt-dlp"
}
```
2. Music 디렉터리를 만드세요
3. Music/Archive 파일을 만드세요
4. 봇을 실행시키고 초대하셔서 사용하시면 됩니다.
2. 봇을 초대하시고 사용하시면 됩니다.
# 명령어
## /p
@@ -36,3 +43,6 @@ C++ Dpp 라이브러리를 이용해서 개발된 간단한 디스코드 음악
음성 채팅을 떠납니다.
사용법:
/l
# docker
happytanuki12/bumblebee:latest

7
docker-compose.yml Normal file
View File

@@ -0,0 +1,7 @@
services:
bumblebee:
image: happytanuki12/bumblebee:latest
container_name: BumbleBee
volumes:
- ./config.json:/config.json
restart: unless-stopped

View File

@@ -1,69 +0,0 @@
#pragma once
#ifndef _MUSICPLAYMANAGER_HPP_
#define _MUSICPLAYMANAGER_HPP_
#include <dpp/dpp.h>
#include <Queue/MusicQueue.hpp>
#include <condition_variable>
namespace bumbleBee {
class MusicPlayManager {
public:
MusicPlayManager(std::shared_ptr<dpp::cluster> cluster, std::vector<dpp::snowflake> GIDs) :
cluster(cluster), GIDs(GIDs) {
queueMap = std::unordered_map<dpp::snowflake, std::shared_ptr<MusicQueue>>();
queueEmptyMutex = std::unordered_map<dpp::snowflake, std::shared_ptr<std::mutex>>();
cluster->on_voice_ready([this](const dpp::voice_ready_t &event){on_voice_ready(event);});
cluster->on_voice_track_marker([this](const dpp::voice_track_marker_t &event){on_voice_track_marker(event);});
cluster->on_voice_client_disconnect([this](const dpp::voice_client_disconnect_t& event){on_voice_client_disconnect(event);});
for (auto GID : GIDs) {
queueMap[GID] = std::make_shared<MusicQueue>();
queueEmptyMutex[GID] = std::make_shared<std::mutex>();
}
}
/**
* @brief voice_ready 이벤트 인지시 콜백되는 함수
* @param event const dpp::voice_ready_t&
**/
void on_voice_ready(const dpp::voice_ready_t& event);
/**
* @brief voice_track_marker 이벤트 인지시 콜백되는 함수
* @param event const dpp::voice_track_marker_t&
**/
void on_voice_track_marker(const dpp::voice_track_marker_t& event);
/**
* @brief voice_client_disconnect 이벤트 인지시 콜백되는 함수
* @param event const dpp::voice_client_disconnect_t&
**/
void on_voice_client_disconnect(const dpp::voice_client_disconnect_t& event);
void play(dpp::discord_voice_client* client);
void queue_music(const dpp::snowflake guildId, const std::shared_ptr<MusicQueueElement> music);
void clear(const dpp::snowflake guildId);
std::shared_ptr<MusicQueueElement> remove(const dpp::snowflake guildId, dpp::discord_voice_client* client, int index);
int size(const dpp::snowflake guildId);
void setRepeat(const dpp::snowflake guildId, const bool value);
bool getRepeat(const dpp::snowflake guildId);
std::list<MusicQueueElement> getQueue(const dpp::snowflake guildId);
MusicQueueElement getNowPlaying(const dpp::snowflake guildId);
std::condition_variable queuedCondition;
private:
std::shared_ptr<dpp::cluster> cluster;
std::vector<dpp::snowflake> GIDs;
/// @brief 음악 큐
std::unordered_map<dpp::snowflake, std::shared_ptr<MusicQueue>> queueMap;
std::unordered_map<dpp::snowflake, std::shared_ptr<std::mutex>> queueEmptyMutex;
void send_audio_to_voice(const MusicQueueElement& music, dpp::discord_voice_client* client);
};
}
#endif

View File

@@ -1,53 +0,0 @@
#pragma once
#ifndef _BUMBLEBEE_HPP_
#define _BUMBLEBEE_HPP_
#include <dpp/dpp.h>
#include <dpp/nlohmann/json.hpp>
#include <memory>
#include <Commands/BumbleBeeCommand.hpp>
namespace bumbleBee {
/**
* @file BumbleBee.hpp
* @brief 메인 봇 클래스
**/
class BumbleBee {
public:
/**
* @brief 생성자
**/
BumbleBee();
/**
* @brief 파괴자
* @details BumbleBee의 모든 Property를 책임지고 파괴합니다
**/
~BumbleBee() {}
/**
* @brief 봇 시작
**/
void start();
/**
* @brief slashcommand 이벤트 인지시 콜백되는 함수
* @param event const dpp::slashcommand_t&
**/
void on_slashcommand(const dpp::slashcommand_t& event);
/**
* @brief ready 이벤트 인지시 콜백되는 함수
* @param event const dpp::ready_t&
**/
void on_ready(const dpp::ready_t& event);
/// @brief DPP 기본 클러스터 객체
std::shared_ptr<dpp::cluster> cluster;
/// @brief guild id 배열
std::vector<dpp::snowflake> GIDs;
private:
/// @brief Command 목록
std::unordered_map<std::string, std::shared_ptr<commands::ICommand>> commands;
/// @brief voiceclient 관련 event 처리기 <guild id, 각 길드별 MusicPlayManager 인스턴스>
std::shared_ptr<MusicPlayManager> musicManager;
};
}
#endif

View File

@@ -1,70 +0,0 @@
#pragma once
#ifndef _BUMBLEBEECOMMAND_HPP_
#define _BUMBLEBEECOMMAND_HPP_
#include <dpp/dpp.h>
#include <Audio/MusicPlayManager.hpp>
#include <functional>
namespace bumbleBee::commands {
class ICommand : public dpp::slashcommand {
public:
/**
* @brief 기본 생성자
* @param botID 봇 아이디
* @param manager 음악재생 매니저
**/
ICommand(dpp::snowflake botID, std::shared_ptr<MusicPlayManager> manager) {
this->botID = botID;
this->musicManager = manager;
}
/**
* @brief 명령어 호출 시에 콜백될 메소드
* @param event dpp::slashcommand_t
**/
virtual void execute(const dpp::slashcommand_t &event){};
/// @brief 명령어 별명
std::vector<std::string> aliases;
private:
/// @brief 봇 ID
dpp::snowflake botID;
protected:
/// @brief 음악재생 매니저
std::shared_ptr<MusicPlayManager> musicManager;
protected:
/// @brief concrete class에서 구현해야 하는 init 이벤트트
virtual void init() = 0;
};
}
/**
* @brief 명령어 인자가 없는 명령어의 boilerplate 대체 매크로
* @param name 명령어 이름 및 클래스명
* @param description 명령어 설명
**/
#define _DECLARE_BUMBLEBEE_COMMAND(CLASS, NAME, DESCRIPTION) \
namespace bumbleBee::commands { \
class CLASS : public ICommand { \
public: \
CLASS (dpp::snowflake botID, std::shared_ptr<MusicPlayManager> manager) \
: ICommand(botID, manager) { \
name = #NAME; \
description = DESCRIPTION; \
init(); \
} \
virtual void execute(const dpp::slashcommand_t &event) override; \
protected: \
virtual void init() override; \
}; \
}
_DECLARE_BUMBLEBEE_COMMAND(Delete, d, "큐의 해당하는 번호의 노래를 지웁니다")
_DECLARE_BUMBLEBEE_COMMAND(Leave, l, "음성 채팅방을 떠납니다")
_DECLARE_BUMBLEBEE_COMMAND(Play, p, "노래 재생")
_DECLARE_BUMBLEBEE_COMMAND(Queue, q, "노래 예약 큐를 확인합니다")
_DECLARE_BUMBLEBEE_COMMAND(Repeat, r, "반복을 켜거나 끕니다")
_DECLARE_BUMBLEBEE_COMMAND(Skip, s, "현재 재생중인 곡을 스킵합니다")
_DECLARE_BUMBLEBEE_COMMAND(Shuffle, shuffle, "큐를 섞습니다")
#endif

View File

@@ -1,41 +0,0 @@
#pragma once
#ifndef _MUSICQUEUE_HPP_
#define _MUSICQUEUE_HPP_
#include <memory>
#include <functional>
#include <condition_variable>
#include <list>
#include <dpp/dpp.h>
#include <Queue/MusicQueueElement.hpp>
namespace bumbleBee {
class MusicQueue {
public:
MusicQueue() {
queue = std::list<std::shared_ptr<MusicQueueElement>>();
currentPlayingPosition = queue.begin();
repeat = true;
}
void enqueue(std::shared_ptr<MusicQueueElement> Element);
std::shared_ptr<MusicQueueElement> dequeue();
std::list<std::shared_ptr<MusicQueueElement>>::iterator findById(std::string id);
std::list<std::shared_ptr<MusicQueueElement>>::iterator findByIndex(int index);
std::shared_ptr<MusicQueueElement> nowplaying();
std::list<std::shared_ptr<MusicQueueElement>>::iterator next_music();
std::shared_ptr<MusicQueueElement> jump_to_index(int idx);
void clear();
std::shared_ptr<MusicQueueElement> erase(std::list<std::shared_ptr<MusicQueueElement>>::iterator it);
std::list<std::shared_ptr<MusicQueueElement>> getQueueCopy();
int size();
bool repeat;
std::list<std::shared_ptr<MusicQueueElement>>::iterator currentPlayingPosition;
private:
std::list<std::shared_ptr<MusicQueueElement>> queue;
std::mutex queueMutex;
};
}
#endif

View File

@@ -1,23 +0,0 @@
#pragma once
#ifndef _MUSICQUEUEELEMENT_HPP_
#define _MUSICQUEUEELEMENT_HPP_
#include <dpp/dpp.h>
namespace bumbleBee {
class MusicQueueElement {
public:
MusicQueueElement(
std::string id,
std::string query,
dpp::user issuingUser,
dpp::embed embed) :
id(id), query(query), issuingUser(issuingUser), embed(embed) {}
const std::string id;
const std::string query;
const dpp::user issuingUser;
const dpp::embed embed;
};
}
#endif

View File

@@ -1,37 +0,0 @@
#pragma once
#ifndef _SETTINGSMANAGER_HPP_
#define _SETTINGSMANAGER_HPP_
#include <dpp/dpp.h>
#define _DECLARE_DEFAULT_ACCESSER_STATIC_VARIABLE(type, name, Name)\
private:\
static type name;\
public:\
static type get##Name() {return name;}\
static void set##Name(const type& value) {name = value; saveToFile();}
namespace bumbleBee {
/// @brief 모든 설정은 이 객체를 통해 스태틱하게 제공됨.
class SettingsManager {
/// @brief 봇 토큰
_DECLARE_DEFAULT_ACCESSER_STATIC_VARIABLE(std::string, TOKEN, TOKEN)
/// @brief yt-dlp 실행 명령어
_DECLARE_DEFAULT_ACCESSER_STATIC_VARIABLE(std::string, YTDLP_CMD, YTDLP_CMD)
/// @brief ffmpeg 실행 명령어
_DECLARE_DEFAULT_ACCESSER_STATIC_VARIABLE(std::string, FFMPEG_CMD, FFMPEG_CMD)
/// @brief 로그레벨
_DECLARE_DEFAULT_ACCESSER_STATIC_VARIABLE(dpp::loglevel, LOGLEVEL, LOGLEVEL)
/// @brief 이전에 있던 명령을 지우고 등록할지 선택합니다.
_DECLARE_DEFAULT_ACCESSER_STATIC_VARIABLE(bool, REGISTER_COMMAND, REGISTER_COMMAND)
public:
/// @brief 설정 로드하기, 설정은 이 load()를 부르기 전까지는 적절하지 못한 값을 가지고 있음.
/// @return 로드 성공 시 true, 실패 시 false 반환.
static bool load();
/// @brief 설정 변경사항 저장
static void saveToFile();
/// @brief 토큰이 유효한지 체크합니다.
/// @return 유효한 토큰이면 true, 아니면 false를 반환합니다.
static bool validateToken();
};
}
#endif

View File

@@ -1,60 +0,0 @@
#pragma once
#ifndef _ASYNCDOWNLOADMANAGER_HPP_
#define _ASYNCDOWNLOADMANAGER_HPP_
#include <queue>
#include <string>
#include <thread>
#include <condition_variable>
#include <dpp/dpp.h>
#include <Queue/MusicQueue.hpp>
namespace bumbleBee {
/// @brief 싱글톤 멀티스레딩 다운로드 매니저
class [[deprecated]] AsyncDownloadManager {
public:
static AsyncDownloadManager& getInstance(int worker_count, std::weak_ptr<dpp::cluster> weak_cluster) {
static AsyncDownloadManager dl(worker_count);
dl.weak_cluster = weak_cluster;
return dl;
}
void enqueue(std::pair<std::string, dpp::message> query) {
std::thread th(&bumbleBee::AsyncDownloadManager::enqueueAsyncDL, this, query);
th.detach();
}
void stop() {
auto cluster = weak_cluster.lock();
cluster->log(dpp::ll_info, "AsyncDownloadManager stop/destructor called.");
terminate = true;
dlQueueCondition.notify_all();
for (auto& t : worker_thread) {
t.join();
}
}
~AsyncDownloadManager(){
stop();
}
private:
AsyncDownloadManager(int worker_count){
worker_thread.reserve(worker_count);
terminate = false;
for (int i=0; i<worker_count; i++) {
worker_thread.emplace_back([this](){this->downloadWorker();});
}
}
void enqueueAsyncDL(std::pair<std::string, dpp::message> query);
void downloadWorker();
std::queue<std::pair<std::string, dpp::message>> downloadQueue;
std::condition_variable dlQueueCondition;
std::mutex dlQueueMutex;
std::weak_ptr<dpp::cluster> weak_cluster;
std::vector<std::thread> worker_thread;
bool terminate;
};
}
#endif

View File

@@ -1,40 +0,0 @@
#pragma once
#ifndef _CONSOLEUTILS_HPP_
#define _CONSOLEUTILS_HPP_
#include <iostream>
#include <sstream>
#include <queue>
namespace bumbleBee {
class ConsoleUtils {
public:
/**
* @brief 명령어를 쉘에서 실행하고 결과를 EOF 전까지 읽어 \n을 구분자로 토큰화하여 반환합니다
* @param cmd 실행할 명령
* @return std::queue<std::string> tokens
*/
static std::queue<std::string> getResultFromCommand(std::string cmd) {
std::string result, token;
std::queue<std::string> tokens;
FILE* stream;
const int maxBuffer = 12; // 적당한 크기
char buffer[maxBuffer];
cmd.append(" 2>&1"); // 표준에러를 표준출력으로 redirect
stream = popen(cmd.c_str(), "r"); // 주어진 command를 shell로 실행하고 파이프 연결 (fd 반환)
if (stream) {
while (fgets(buffer, maxBuffer, stream) != NULL) result.append(buffer); // fgets: fd (stream)를 길이 (maxBuffer)만큼 읽어 버퍼 (buffer)에 저장
pclose(stream);
}
std::stringstream ss(result);
while (std::getline(ss, token, '\n')) {
tokens.push(token);
}
return tokens;
}
};
}
#endif

View File

@@ -1,77 +0,0 @@
#pragma once
#ifndef _VERSIONCHECKUTILS_HPP_
#define _VERSIONCHECKUTILS_HPP_
#include <dpp/dpp.h>
#include "ConsoleUtils.hpp"
#include "../Settings/SettingsManager.hpp"
namespace bumbleBee {
class VersionsCheckUtils {
public:
static bool isThereCMD(std::shared_ptr<dpp::cluster> cluster, std::string cmd) {
if (ConsoleUtils::getResultFromCommand("which " + cmd).size() == 0) {
cluster->log(dpp::ll_error, cmd + " is unavaliable. unresolable error please install " + cmd);
return false;
}
else
return true;
}
static void validateYTDLPFFMPEGBinary(std::shared_ptr<dpp::cluster> cluster) {
cluster->log(dpp::ll_info, "Checking if yt-dlp and ffmpeg is available...");
std::queue<std::string> result = ConsoleUtils::getResultFromCommand(SettingsManager::getFFMPEG_CMD() + " -version");
std::string front = result.front();
if (front[0] != 'f' ||
front[1] != 'f' ||
front[2] != 'm' ||
front[3] != 'p' ||
front[4] != 'e' ||
front[5] != 'g') {
cluster->log(dpp::ll_warning, "ffmpeg is unavailable. downloading ffmpeg...");
if (!isThereCMD(cluster, "curl")) {
exit(1);
}
if (!isThereCMD(cluster, "tar")) {
exit(1);
}
system("curl -LO https://github.com/BtbN/FFmpeg-Builds/releases/latest/download/ffmpeg-master-latest-linux64-gpl.tar.xz");
system("tar -xf ffmpeg-master-latest-linux64-gpl.tar.xz");
system("rm ffmpeg-master-latest-linux64-gpl.tar.xz");
system("mv ffmpeg-master-latest-linux64-gpl ffmpeg");
SettingsManager::setFFMPEG_CMD("./ffmpeg/bin/ffmpeg");
}
result = ConsoleUtils::getResultFromCommand(SettingsManager::getYTDLP_CMD() + " --version");
front = result.front();
if ((front[0]-'0' < 0 || front[0]-'0' > 9) ||
(front[1]-'0' < 0 || front[1]-'0' > 9) ||
(front[2]-'0' < 0 || front[2]-'0' > 9) ||
(front[3]-'0' < 0 || front[3]-'0' > 9)) {
cluster->log(dpp::ll_warning, "ytdlp is unavailable. downloading ytdlp...");
if (!isThereCMD(cluster, "curl")) {
exit(1);
}
if (!isThereCMD(cluster, "tar")) {
exit(1);
}
system("curl -LO https://github.com/yt-dlp/yt-dlp/releases/latest/download/yt-dlp");
system("chmod +x ./yt-dlp");
SettingsManager::setYTDLP_CMD("./yt-dlp");
}
}
static void updateytdlp(std::shared_ptr<dpp::cluster> cluster) {
cluster->log(dpp::ll_info, "Checking if yt-dlp update is available...");
std::queue<std::string> result = ConsoleUtils::getResultFromCommand("./yt-dlp -U");
while(!result.empty()) {
std::string front = result.front();
result.pop();
cluster->log(dpp::ll_info, front);
}
}
};
}
#endif

19
include/core/bumblebee.h Normal file
View File

@@ -0,0 +1,19 @@
#ifndef BUMBLEBEE_INCLUDE_CORE_BUMBLEBEE_H_
#define BUMBLEBEE_INCLUDE_CORE_BUMBLEBEE_H_
#include "precomp.h"
namespace bumblebee {
class BumbleBee {
public:
BumbleBee();
~BumbleBee();
private:
dpp::cluster cluster;
};
} // namespace bumblebee
#endif

26
include/precomp.h Normal file
View File

@@ -0,0 +1,26 @@
#ifndef BUMBLEBEE_INCLUDE_PRECOMP_H_
#define BUMBLEBEE_INCLUDE_PRECOMP_H_
#ifdef WIN32
#include <winsock2.h>
#endif
#include <dpp/dpp.h>
#include <iostream>
#include <queue>
#include "boost/asio.hpp"
#include "boost/beast/http.hpp"
#include "boost/beast/ssl.hpp"
#include "boost/log/trivial.hpp"
#include "boost/process.hpp"
extern "C" {
#include "libavcodec/avcodec.h"
#include "libavformat/avformat.h"
#include "libavutil/avutil.h"
#include "libswresample/swresample.h"
#include "libswscale/swscale.h"
}
#endif // BUMBLEBEE_INCLUDE_PRECOMP_H_

41
include/utils/console.h Normal file
View File

@@ -0,0 +1,41 @@
#ifndef BUMBLEBEE_INCLUDE_UTILS_CONSOLE_H_
#define BUMBLEBEE_INCLUDE_UTILS_CONSOLE_H_
#include "precomp.h"
namespace utils {
// @brief 명령어가 실행 가능한지 체크합니다
// @param cmd 실행할 명령
// @return int error_code
int ValidateCommand(
boost::asio::io_context& ctx, std::string cmd,
const std::vector<std::string>& args = std::vector<std::string>());
// @brief 명령어를 쉘에서 실행하고 결과를 EOF 전까지 읽어 반환합니다
// @param cmd 실행할 명령
// @param args 아규먼트
// @return int error_code
int ExecuteCommand(
boost::asio::io_context& ctx, const std::string& cmd,
const std::vector<std::string>& args = std::vector<std::string>());
// @brief 명령어를 쉘에서 실행하고 결과를 EOF 전까지 읽어 반환합니다
// @param cmd 실행할 명령
// @param result 실행결과
// @return int error_code
int ExecuteCommand(boost::asio::io_context& ctx, const std::string& cmd,
std::string& result);
// @brief 명령어를 쉘에서 실행하고 결과를 EOF 전까지 읽어 반환합니다
// @param cmd 실행할 명령
// @param args 아규먼트
// @param result 실행결과
// @return int error_code
int ExecuteCommand(boost::asio::io_context& ctx, const std::string& cmd,
const std::vector<std::string>& args, std::string& result);
// @brief 명령어를 쉘에서 실행하고 결과를 파이프로 연결하여 반환합니다
// @param cmd 실행할 명령
// @param args 아규먼트
// @return boost::process::popen
boost::process::popen OpenPipe(
boost::asio::io_context& ctx, const std::string& cmd,
const std::vector<std::string>& args = std::vector<std::string>());
} // namespace utils
#endif

View File

@@ -0,0 +1,10 @@
#ifndef BUMBLEBEE_INCLUDE_UTILS_FILE_DOWNLOADER_H_
#define BUMBLEBEE_INCLUDE_UTILS_FILE_DOWNLOADER_H_
#include "precomp.h"
namespace utils {
void DownloadFileFromHTTPS(std::string url, std::string filename);
}
#endif

View File

@@ -0,0 +1,14 @@
#ifndef BUMBLEBEE_INCLUDE_UTILS_UPDATE_CHECKER_H_
#define BUMBLEBEE_INCLUDE_UTILS_UPDATE_CHECKER_H_
#include "precomp.h"
namespace utils {
int InstallYtdlp(boost::asio::io_context& ctx);
int CheckUpdate(boost::asio::io_context& ctx);
} // namespace utils
#endif

View File

@@ -1,125 +0,0 @@
#include <Audio/MusicPlayManager.hpp>
#include <ogg/ogg.h>
#include <oggz/oggz.h>
#include <algorithm>
namespace bumbleBee {
void MusicPlayManager::on_voice_ready(const dpp::voice_ready_t& event) {
play(event.voice_client);
}
void MusicPlayManager::on_voice_track_marker(const dpp::voice_track_marker_t& event) {
play(event.voice_client);
}
void MusicPlayManager::on_voice_client_disconnect(const dpp::voice_client_disconnect_t& event) { // 안 불리는 듯?
dpp::snowflake gid = dpp::find_channel(event.voice_client->channel_id)->guild_id;
event.voice_client->stop_audio();
queueMap[gid]->clear();
}
void MusicPlayManager::play(dpp::discord_voice_client* client) {
std::thread t([&](dpp::discord_voice_client* client){
dpp::snowflake gid = dpp::find_channel(client->channel_id)->guild_id;
std::unique_lock<std::mutex> queueEmptyLock(*queueEmptyMutex[gid]);
queuedCondition.wait(queueEmptyLock, [&]{ return queueMap[gid]->size() != 0; });
auto np = queueMap[gid]->next_music();
auto music = **np;
send_audio_to_voice(music, client);
}, client);
t.detach();
}
void MusicPlayManager::queue_music(const dpp::snowflake guildId, const std::shared_ptr<MusicQueueElement> music) {
queueMap[guildId]->enqueue(music);
}
void MusicPlayManager::clear(const dpp::snowflake guildId) {
queueMap[guildId]->clear();
}
std::shared_ptr<MusicQueueElement> MusicPlayManager::remove(const dpp::snowflake guildId, dpp::discord_voice_client* client, int index) {
auto queue = queueMap[guildId];
auto foundIterator = queue->findByIndex(index);
if (queue->currentPlayingPosition == foundIterator) {
auto removed = queue->erase(queue->findByIndex(0));
if (client == nullptr)
return removed;
client->pause_audio(true);
client->stop_audio();
client->pause_audio(false);
client->insert_marker("end");
return removed;
}
return queue->erase(queue->findByIndex(index));
}
int MusicPlayManager::size(const dpp::snowflake guildId) {
return queueMap[guildId]->size();
}
void MusicPlayManager::setRepeat(const dpp::snowflake guildId, const bool value) {
queueMap[guildId]->repeat = value;
}
bool MusicPlayManager::getRepeat(const dpp::snowflake guildId) {
return queueMap[guildId]->repeat;
}
std::list<MusicQueueElement> MusicPlayManager::getQueue(const dpp::snowflake guildId){
std::list<std::shared_ptr<MusicQueueElement>> queue = queueMap[guildId]->getQueueCopy();
std::list<MusicQueueElement> returnValue;
for (auto iter = queue.begin(); iter != queue.end(); iter++)
returnValue.push_back(**iter);
return returnValue;
}
MusicQueueElement MusicPlayManager::getNowPlaying(const dpp::snowflake guildId) {
std::shared_ptr<MusicQueueElement> nowplaying = queueMap[guildId]->nowplaying();
MusicQueueElement returnValue(*nowplaying);
return returnValue;
}
void MusicPlayManager::send_audio_to_voice(const MusicQueueElement& music, dpp::discord_voice_client* client) {
std::string command = "./streamOpus.sh ./yt-dlp ffmpeg https://youtu.be/";
command += music.id;
OGGZ* og = oggz_open_stdio(popen(command.c_str(), "r"), OGGZ_READ);
client->stop_audio();
oggz_set_read_callback(
og, -1,
[](OGGZ *oggz, oggz_packet *packet, long serialno, void *user_data) {
auto voiceConn = (dpp::discord_voice_client *)user_data;
voiceConn->send_audio_opus(packet->op.packet, packet->op.bytes);
return 0;
},
(void *)client
);
while (client && !client->terminating) {
static const constexpr long CHUNK_READ = BUFSIZ * 2;
const long read_bytes = oggz_read(og, CHUNK_READ);
/* break on eof */
if (!read_bytes) {
break;
}
}
client->creator->log(dpp::ll_info, "Sending " + music.id + " complete!");
oggz_close(og);
client->insert_marker();
}
}

View File

@@ -1,80 +0,0 @@
#include <BumbleBee.hpp>
#include <memory>
#include <Settings/SettingsManager.hpp>
#include <Utils/VersionsCheckUtils.hpp>
#include <Audio/MusicPlayManager.hpp>
#include <Commands/BumbleBeeCommand.hpp>
namespace bumbleBee{
BumbleBee::BumbleBee() {
if (!SettingsManager::load()) {
std::cout << "Please configure confing.json" << std::endl;
exit(1);
}
cluster = std::make_shared<dpp::cluster>(SettingsManager::getTOKEN());
cluster->on_log([](const dpp::log_t& event) {
if (event.severity >= SettingsManager::getLOGLEVEL()) {
std::cout << "[" << dpp::utility::current_date_time() << "] " << dpp::utility::loglevel(event.severity) << ": " << event.message << "\n";
}
});
cluster->on_slashcommand([this](const dpp::slashcommand_t& event){on_slashcommand(event);});
cluster->on_ready([this](const dpp::ready_t &event){on_ready(event);});
VersionsCheckUtils::validateYTDLPFFMPEGBinary(cluster);
VersionsCheckUtils::updateytdlp(cluster);
}
void BumbleBee::start() { this->cluster->start(dpp::st_wait); }
void BumbleBee::on_slashcommand(const dpp::slashcommand_t &event) {
if (commands.find(event.command.get_command_name()) != commands.end()) {
event.thinking();
auto command = commands.at(event.command.get_command_name());
std::thread t([](const dpp::slashcommand_t &event, std::shared_ptr<commands::ICommand> command){
command->execute(event);
}, event, command);
t.detach();
}
}
void BumbleBee::on_ready(const dpp::ready_t &event) {
GIDs = event.guilds;
musicManager = std::make_shared<MusicPlayManager>(cluster, GIDs);
commands["d"] = std::make_shared<commands::Delete> (cluster->cluster_id, musicManager);
commands["l"] = std::make_shared<commands::Leave> (cluster->cluster_id, musicManager);
commands["p"] = std::make_shared<commands::Play> (cluster->cluster_id, musicManager);
commands["q"] = std::make_shared<commands::Queue> (cluster->cluster_id, musicManager);
commands["r"] = std::make_shared<commands::Repeat> (cluster->cluster_id, musicManager);
commands["s"] = std::make_shared<commands::Skip> (cluster->cluster_id, musicManager);
commands["shuffle"] = std::make_shared<commands::Shuffle> (cluster->cluster_id, musicManager);
if (event.guilds.size() == 0) {
cluster->log(dpp::loglevel::ll_info, "Bot is not on any server! Please invite this bot to any server.");
exit(1);
}
if (dpp::run_once<struct register_bot_commands>()) {
if (SettingsManager::getREGISTER_COMMAND()) {
cluster->log(dpp::loglevel::ll_info, "Clear Pre-installed commands");
cluster->global_bulk_command_create({
*commands["d"],
*commands["l"],
*commands["p"],
*commands["q"],
*commands["r"],
*commands["s"],
*commands["shuffle"]
}, [&](const dpp::confirmation_callback_t &t){
cluster->log(dpp::loglevel::ll_info, "Command created.");
SettingsManager::setREGISTER_COMMAND(false);
});
}
}
cluster->log(dpp::loglevel::ll_info, "Bot ready.");
}
}

View File

@@ -1,39 +0,0 @@
#include <Commands/BumbleBeeCommand.hpp>
#include <format>
namespace bumbleBee::commands {
void Delete::execute(const dpp::slashcommand_t &event) {
if (std::holds_alternative<std::monostate>(event.get_parameter("pos"))) // 여기 들어올 일 있나?
{
event.edit_original_response(dpp::message("위치를 제공하여 주십시오"));
event.reply("");
return;
}
int pos = std::get<std::int64_t>(event.get_parameter("pos"));
if (pos < 0 || pos > musicManager->size(event.command.guild_id))
{
event.edit_original_response(dpp::message(std::string("이상한 인덱스 위치. Pos :") + std::to_string(pos)));
return;
}
dpp::voiceconn* v = event.from->get_voice(event.command.guild_id);
std::shared_ptr<MusicQueueElement> removed;
if (!v) // v-> 로 nullptr을 참조하면 안 되므로.
removed = musicManager->remove(event.command.guild_id, nullptr, pos);
else
removed = musicManager->remove(event.command.guild_id, v->voiceclient, pos);
dpp::message msg("다음 항목을 큐에서 삭제했습니다!:");
msg.add_embed(removed->embed);
event.edit_original_response(msg);
}
void Delete::init() {
add_option(dpp::command_option(dpp::co_integer, "pos", "지울 위치(위치는 1부터 시작, 현재 재생곡은 0)", true));
}
}

View File

@@ -1,19 +0,0 @@
#include <Commands/BumbleBeeCommand.hpp>
namespace bumbleBee::commands {
void Leave::execute(const dpp::slashcommand_t &event) { // 왜 read loop ended가 뜨는가...
dpp::voiceconn* v = event.from->get_voice(event.command.guild_id);
if (!v || !v->voiceclient || !v->voiceclient->is_ready()) {
return;
}
musicManager->clear(event.command.guild_id);
event.from->disconnect_voice(event.command.guild_id);
event.edit_original_response(dpp::message("음성 채팅방을 떠납니다!"));
}
void Leave::init() {
}
}

View File

@@ -1,167 +0,0 @@
#include <Commands/BumbleBeeCommand.hpp>
#include <Utils/ConsoleUtils.hpp>
#include <Settings/SettingsManager.hpp>
#include <dpp/nlohmann/json.hpp>
#include <variant>
namespace bumbleBee::commands {
void Play::execute(const dpp::slashcommand_t &event) {
dpp::guild *g = dpp::find_guild(event.command.guild_id);
if (!g) { //wtf?
event.edit_original_response(dpp::message("GUILD NOT FOUND!! WHAT IS THIS SORCERY??"));
return;
}
if (std::holds_alternative<std::monostate>(event.get_parameter("query"))) // 이거 필요한가???
{
event.reply("노래를 재생하려면 검색어 또는 링크를 입력해 주십시오.");
return;
}
if (!event.from->get_voice(event.command.guild_id) && !g->connect_member_voice(event.command.usr.id)) {
event.edit_original_response(dpp::message("노래를 재생할 음성 채팅방에 먼저 참가하고 신청해야 합니다!"));
return;
}
std::string query = std::get<std::string>(event.get_parameter("query"));
std::queue<std::string> ids = ConsoleUtils::getResultFromCommand(SettingsManager::getYTDLP_CMD() + " --default-search ytsearch --flat-playlist --skip-download --quiet --ignore-errors --print id " + query);
std::queue<std::shared_ptr<MusicQueueElement>> musics;
int initialQueuedCount = musicManager->size(event.command.guild_id);
dpp::message msg;
if (musicManager->size(event.command.guild_id) == 0)
msg.content = "다음 곡을 재생합니다:";
else
msg.content = "큐에 다음 곡을 추가했습니다:";
if (ids.size() >= 2) {
event.from->creator->log(dpp::ll_info, "Playlist detected.");
while (!ids.empty()) {
if (ids.front() == "") {
ids.pop();
continue;
}
FILE* file = popen((SettingsManager::getYTDLP_CMD() + " --default-search ytsearch --flat-playlist --skip-download --quiet --ignore-errors -J http://youtu.be/" + ids.front()).c_str(), "r");
std::ostringstream oss;
char buffer[1024];
size_t bytesRead;
while ((bytesRead = fread(buffer, 1, sizeof(buffer), file)) > 0) {
oss.write(buffer, bytesRead);
}
fclose(file);
std::istringstream iss(oss.str());
nlohmann::json videoDataJson;
iss >> videoDataJson;
// std::string dump = videoDataJson.dump(4);
time_t SongLength = int(videoDataJson["duration"]);
char SongLengthStr[10];
tm t;
t.tm_mday = SongLength / 86400;
t.tm_hour = (SongLength % 86400)/3600;
t.tm_min = (SongLength % 3600)/60;
t.tm_sec = SongLength%60;
strftime(SongLengthStr, sizeof(SongLengthStr), "%X", &t);
dpp::embed embed = dpp::embed()
.set_color(dpp::colors::sti_blue)
.set_title(std::string(videoDataJson["title"]))
.set_description(std::string(videoDataJson["uploader"]))
.set_url(std::string(videoDataJson["webpage_url"]))
.set_image(std::string(videoDataJson["thumbnail"]))
.add_field(
"길이",
SongLengthStr,
true
);
musics.push(std::make_shared<MusicQueueElement>(ids.front(), query, event.command.usr, embed));
ids.pop();
}
}
if (!ids.empty()) {
FILE* file = popen((SettingsManager::getYTDLP_CMD() + " --default-search ytsearch --flat-playlist --skip-download --quiet --ignore-errors -J http://youtu.be/" + ids.front()).c_str(), "r");
std::ostringstream oss;
char buffer[1024];
size_t bytesRead;
while ((bytesRead = fread(buffer, 1, sizeof(buffer), file)) > 0) {
oss.write(buffer, bytesRead);
}
fclose(file);
std::istringstream iss(oss.str());
nlohmann::json videoDataJson;
iss >> videoDataJson;
time_t SongLength = int(videoDataJson["duration"]);
char SongLengthStr[10];
tm t;
t.tm_mday = SongLength / 86400;
t.tm_hour = (SongLength % 86400)/3600;
t.tm_min = (SongLength % 3600)/60;
t.tm_sec = SongLength%60;
strftime(SongLengthStr, sizeof(SongLengthStr), "%X", &t);
dpp::embed embed = dpp::embed()
.set_color(dpp::colors::sti_blue)
.set_title(std::string(videoDataJson["title"]))
.set_description(std::string(videoDataJson["uploader"]))
.set_url(std::string(videoDataJson["webpage_url"]))
.set_image(std::string(videoDataJson["thumbnail"]))
.add_field(
"길이",
SongLengthStr,
true
);
musics.push(std::make_shared<MusicQueueElement>(ids.front(), query, event.command.usr, embed));
}
if (musics.size() == 1) {
event.from->creator->log(dpp::ll_info, "Enqueuing " + musics.front()->id);
musicManager->queue_music(event.command.guild_id, musics.front());
msg.add_embed(musics.front()->embed);
musics.pop();
event.edit_original_response(msg);
musicManager->queuedCondition.notify_all();
}
else if (musics.size() > 1) {
event.from->creator->log(dpp::ll_info, "Enqueuing " + musics.front()->id);
musicManager->queue_music(event.command.guild_id, musics.front());
msg.add_embed(musics.front()->embed);
musics.pop();
event.edit_original_response(msg);
musicManager->queuedCondition.notify_all();
while (!musics.empty()) {
event.from->creator->log(dpp::ll_info, "Enqueuing " + musics.front()->id);
dpp::message followMsg(event.command.channel_id, "");
followMsg.add_embed(musics.front()->embed);
event.from->creator->message_create(followMsg); // 어차피 원래 메시지를 지정해서 수정할 것이기 때문에 먼저 팔로잉 메시지를 작성해도 상관없음.
musicManager->queue_music(event.command.guild_id, musics.front());
musics.pop();
}
}
else { // ??
event.from->creator->log(dpp::ll_error, "??? not queueed any music");
}
}
void Play::init() {
add_option(dpp::command_option(dpp::co_string, "query", "링크 또는 검색어", true));
}
}

View File

@@ -1,34 +0,0 @@
#include <Commands/BumbleBeeCommand.hpp>
namespace bumbleBee::commands {
void Queue::execute(const dpp::slashcommand_t &event) {
auto queue = musicManager->getQueue(event.command.guild_id);
auto nowplaying = musicManager->getNowPlaying(event.command.guild_id);
dpp::message msg;
dpp::embed embed;
if (queue.size() == 0) {
embed
.set_title("큐가 비었습니다!")
.set_timestamp(time(0));
msg.add_embed(embed);
event.edit_original_response(msg);
return;
}
msg.content = "지금 재생 중:";
msg.add_embed(nowplaying.embed);
for (auto iter = queue.begin(); iter != queue.end(); iter++) {
dpp::message followMsg(event.command.channel_id, "");
followMsg.add_embed(iter->embed);
event.from->creator->message_create(followMsg); // 어차피 원래 메시지를 지정해서 수정할 것이기 때문에 먼저 팔로잉 메시지를 작성해도 상관없음.
}
event.edit_original_response(msg);
}
void Queue::init() {
}
}

View File

@@ -1,16 +0,0 @@
#include <Commands/BumbleBeeCommand.hpp>
namespace bumbleBee::commands {
void Repeat::execute(const dpp::slashcommand_t &event) {
if (musicManager->getRepeat(event.command.guild_id)) {
event.edit_original_response(dpp::message("반복을 껐습니다."));
musicManager->setRepeat(event.command.guild_id, false);
}
else {
event.edit_original_response(dpp::message("반복을 켰습니다."));
musicManager->setRepeat(event.command.guild_id, true);
}
}
void Repeat::init() {}
}

View File

@@ -1,10 +0,0 @@
#include <Commands/BumbleBeeCommand.hpp>
namespace bumbleBee::commands {
void Shuffle::execute(const dpp::slashcommand_t &event) {
event.edit_original_response(dpp::message("shuffle"));
}
void Shuffle::init() {
}
}

View File

@@ -1,20 +0,0 @@
#include <Commands/BumbleBeeCommand.hpp>
namespace bumbleBee::commands {
void Skip::execute(const dpp::slashcommand_t &event) {
dpp::voiceconn* v = event.from->get_voice(event.command.guild_id);
if (!v || !v->voiceclient || !v->voiceclient->is_ready()) {
return;
}
v->voiceclient->pause_audio(true);
v->voiceclient->stop_audio();
v->voiceclient->pause_audio(false);
v->voiceclient->insert_marker("end");
event.edit_original_response(dpp::message("스킵했습니다!"));
}
void Skip::init() {
}
}

View File

@@ -1,97 +0,0 @@
#include <Queue/MusicQueue.hpp>
#include <iostream>
#include <algorithm>
namespace bumbleBee {
void MusicQueue::enqueue(std::shared_ptr<MusicQueueElement> Element) {
std::lock_guard<std::mutex> lock(queueMutex);
queue.push_back(Element);
if (queue.size() == 1) // 이전에 하나도 없었다는 말이므로.
currentPlayingPosition = queue.begin(); // 첫번째 곡으로 iterator를 옮겨준다.
}
std::shared_ptr<MusicQueueElement> MusicQueue::dequeue() {
std::lock_guard<std::mutex> lock(queueMutex);
auto value = queue.front();
queue.pop_front();
return value;
}
std::list<std::shared_ptr<MusicQueueElement>>::iterator MusicQueue::findById(std::string id) {
std::lock_guard<std::mutex> lock(queueMutex);
int index = 0;
for (auto iter = queue.begin(); iter != queue.end(); iter++) {
if ((*iter).get()->id == id)
return iter;
}
return queue.end();
}
std::list<std::shared_ptr<MusicQueueElement>>::iterator MusicQueue::findByIndex(int index) {
std::lock_guard<std::mutex> lock(queueMutex);
if (index < 0)
return queue.end();
if (index == 0)
return currentPlayingPosition;
return std::next(queue.begin(), index - 1);
}
std::shared_ptr<MusicQueueElement> MusicQueue::nowplaying() {
std::lock_guard<std::mutex> lock(queueMutex);
return *currentPlayingPosition;
}
std::list<std::shared_ptr<MusicQueueElement>>::iterator MusicQueue::next_music() {
std::lock_guard<std::mutex> lock(queueMutex);
if (currentPlayingPosition == --queue.end() && !repeat)
return queue.end();
if (currentPlayingPosition == --queue.end() && repeat)
currentPlayingPosition = queue.begin();
else
++currentPlayingPosition;
return currentPlayingPosition;
}
std::shared_ptr<MusicQueueElement> MusicQueue::jump_to_index(int idx) {
std::lock_guard<std::mutex> lock(queueMutex);
int index = 0;
for (auto iter = queue.begin(); iter != queue.end(); iter++) {
if (idx == index++) {
currentPlayingPosition = iter;
return *iter;
}
}
return std::shared_ptr<MusicQueueElement>();
}
void MusicQueue::clear() {
std::lock_guard<std::mutex> lock(queueMutex);
queue.clear();
currentPlayingPosition = queue.begin();
}
std::shared_ptr<MusicQueueElement> MusicQueue::erase(std::list<std::shared_ptr<MusicQueueElement>>::iterator it) {
std::lock_guard<std::mutex> lock(queueMutex);
if (it == queue.end())
return nullptr;
if (currentPlayingPosition == it) {
auto removedValue = *it;
auto after = queue.erase(currentPlayingPosition++);
if (after == queue.end())
currentPlayingPosition = --queue.end();
return removedValue;
}
else {
auto removedValue = *it;
queue.erase(it);
return removedValue;
}
}
std::list<std::shared_ptr<MusicQueueElement>> MusicQueue::getQueueCopy(){
std::lock_guard<std::mutex> lock(queueMutex);
std::list<std::shared_ptr<MusicQueueElement>> returnValue;
std::copy(queue.begin(), queue.end(), std::back_inserter(returnValue));
return returnValue;
}
int MusicQueue::size() {
std::lock_guard<std::mutex> lock(queueMutex);
return queue.size();
}
}

View File

@@ -1,118 +0,0 @@
#include <Settings/SettingsManager.hpp>
#include <dpp/nlohmann/json.hpp>
#include <Utils/ConsoleUtils.hpp>
#include <fstream>
#include <sstream>
namespace bumbleBee {
std::string SettingsManager::TOKEN = "";
std::string SettingsManager::YTDLP_CMD = "./yt-dlp";
std::string SettingsManager::FFMPEG_CMD = "./ffmpeg/bin/ffmpeg";
dpp::loglevel SettingsManager::LOGLEVEL = dpp::ll_debug;
bool SettingsManager::REGISTER_COMMAND = false;
bool SettingsManager::validateToken() {
nlohmann::json response;
if (ConsoleUtils::getResultFromCommand("which curl").size() == 0) {
std::cout << "curl is unavaliable. unresolable error please install curl." << std::endl;
return false;
}
std::string stresult = ConsoleUtils::getResultFromCommand("curl -sX GET \"https://discord.com/api/v10/users/@me\" -H \"Authorization: Bot " +
TOKEN + "\"").front();
std::stringstream ss(stresult);
ss >> response;
if (response.contains("message") && response["message"] == "401: Unauthorized") {
std::cout << "Token is invalid" << std::endl;
return false;
}
else
return true;
}
bool SettingsManager::load() {
nlohmann::json configdocument;
try {
std::ifstream configfile("config.json");
if (!configfile) {
saveToFile();
return false;
}
configfile >> configdocument;
TOKEN = configdocument["TOKEN"];
if (!validateToken())
return false;
YTDLP_CMD = configdocument["YTDLP_CMD"];
FFMPEG_CMD = configdocument["FFMPEG_CMD"];
std::string level = configdocument["LOGLEVEL"];
std::transform(level.begin(), level.end(), level.begin(), ::tolower);
if (level == "trace")
LOGLEVEL = dpp::ll_trace;
else if (level == "debug")
LOGLEVEL = dpp::ll_debug;
else if (level == "warning")
LOGLEVEL = dpp::ll_warning;
else if (level == "error")
LOGLEVEL = dpp::ll_error;
else if (level == "critical")
LOGLEVEL = dpp::ll_critical;
else // 값이 병신같을때 기본값으로 ll_info 부여
LOGLEVEL = dpp::ll_info;
REGISTER_COMMAND = configdocument["CLEAR_PREVIOUS_COMMAND"];
}
catch (const nlohmann::json::type_error& e) {
saveToFile();
return load();
}
catch (const nlohmann::json::parse_error& e) {
saveToFile();
return load();
}
return true;
}
void SettingsManager::saveToFile() {
nlohmann::json configdocument;
configdocument["TOKEN"] = TOKEN;
configdocument["YTDLP_CMD"] = YTDLP_CMD;
configdocument["FFMPEG_CMD"] = FFMPEG_CMD;
switch (LOGLEVEL) {
case dpp::ll_trace:
configdocument["LOGLEVEL"] = "trace";
break;
case dpp::ll_debug:
configdocument["LOGLEVEL"] = "debug";
break;
case dpp::ll_info:
configdocument["LOGLEVEL"] = "info";
break;
case dpp::ll_warning:
configdocument["LOGLEVEL"] = "warning";
break;
case dpp::ll_error:
configdocument["LOGLEVEL"] = "error";
break;
default:
configdocument["LOGLEVEL"] = "critical";
break;
}
configdocument["CLEAR_PREVIOUS_COMMAND"] = REGISTER_COMMAND;
std::ofstream configFile("config.json");
if (configFile.is_open()) {
configFile << configdocument.dump(4);
configFile.close();
}
}
}

View File

@@ -1,86 +0,0 @@
#include <Utils/AsyncDownloadManager.hpp>
#include <sstream>
#include "Utils/ConsoleUtils.hpp"
#include "Settings/SettingsManager.hpp"
#include <ogg/ogg.h>
#include <oggz/oggz.h>
#include <opus/opusfile.h>
#include <memory>
namespace bumbleBee {
void AsyncDownloadManager::enqueueAsyncDL(std::pair<std::string, dpp::message> query) {
std::lock_guard<std::mutex> lock(dlQueueMutex);
downloadQueue.push(query);
dlQueueCondition.notify_one();
}
void AsyncDownloadManager::downloadWorker() {
std::ostringstream tid;
tid << std::this_thread::get_id();
while (true) {
//mutex lock
std::unique_lock<std::mutex> dlQueueLock(dlQueueMutex);
dlQueueCondition.wait(dlQueueLock, [&]{ return !downloadQueue.empty() || terminate; });
auto cluster = weak_cluster.lock();
if (weak_cluster.expired()) {
cluster->log(dpp::ll_error, "Missing cluster, terminating thread " + tid.str());
break;
}
if (terminate) {
cluster->log(dpp::ll_info, "Terminating thread " + tid.str());
break;
}
std::string query = downloadQueue.front().first;
dpp::message oRes = downloadQueue.front().second;
downloadQueue.pop();
dlQueueLock.unlock();
cluster->log(dpp::ll_info, "AsyncDownloadManager: " + query + " accepted.");
std::queue<std::string> ids =
ConsoleUtils::getResultFromCommand(SettingsManager::getYTDLP_CMD() +
" --default-search ytsearch --flat-playlist --skip-download --quiet --ignore-errors --print id " + query);
if (ids.size() >= 2) {
cluster->log(dpp::ll_info, query + " is playlist");
while (!ids.empty()) {
if (ids.front() == "") {
ids.pop();
continue;
}
cluster->log(dpp::ll_info, "Enqueuing playlist element " + ids.front());
enqueue(std::make_pair("https://youtu.be/" + ids.front(), oRes));
ids.pop();
}
continue;
}
std::queue<std::string> urls =
ConsoleUtils::getResultFromCommand(SettingsManager::getYTDLP_CMD() +
" -f ba* --print urls https://youtu.be/" + ids.front());
cluster->log(dpp::ll_debug, "url: " + urls.front());
FILE* stream;
// musicQueue->enqueue(std::make_shared<MusicQueueElement>(oRes, ids.front(), urls.front(), stream));
std::string downloadID = ids.front();
std::thread th([&, downloadID](){
if (terminate)
return;
std::ostringstream tid;
tid << std::this_thread::get_id();
cluster->log(dpp::ll_info, "Thread id: " + tid.str() + ": " + downloadID + " accepted.");
std::string command = std::string("./streamOpus.sh " + SettingsManager::getYTDLP_CMD() + " " + downloadID + " " + SettingsManager::getFFMPEG_CMD());
stream = popen(command.c_str(), "r");
cluster->log(dpp::ll_info, "Thread id: " + tid.str() + " Opened stream: " + downloadID);
});
th.detach();
}
}
}

10
src/main.cc Normal file
View File

@@ -0,0 +1,10 @@
#include "precomp.h"
#include "utils/console.h"
#include "utils/update_checker.h"
int main(int argc, char* argv[]) {
boost::asio::io_context ctx;
utils::CheckUpdate(ctx);
return 0;
}

View File

@@ -1,8 +0,0 @@
#include <iostream>
#include <BumbleBee.hpp>
#include <thread>
int main() {
bumbleBee::BumbleBee bot;
bot.start();
}

53
src/utils/console.cc Normal file
View File

@@ -0,0 +1,53 @@
#include "utils/console.h"
#include "precomp.h"
namespace utils {
int ValidateCommand(boost::asio::io_context& ctx, std::string cmd,
const std::vector<std::string>& args) {
try {
ExecuteCommand(ctx, cmd, args);
} catch (const boost::process::system_error& e) {
return -1;
}
return 0;
}
int ExecuteCommand(boost::asio::io_context& ctx, const std::string& cmd,
const std::vector<std::string>& args) {
std::string ignored;
return ExecuteCommand(ctx, cmd, args, ignored);
}
int ExecuteCommand(boost::asio::io_context& ctx, const std::string& cmd,
std::string& result) {
std::vector<std::string> ignored;
return ExecuteCommand(ctx, cmd, ignored, result);
}
int ExecuteCommand(boost::asio::io_context& ctx, const std::string& cmd,
const std::vector<std::string>& args, std::string& result) {
try {
boost::system::error_code ec;
boost::process::popen proc(
ctx, cmd, args, boost::process::process_stdio{{}, nullptr, nullptr});
boost::asio::read(proc, boost::asio::dynamic_buffer(result), ec);
proc.wait();
return proc.exit_code();
} catch (const boost::process::system_error& e) {
return -1;
}
}
boost::process::popen OpenPipe(boost::asio::io_context& ctx,
const std::string& cmd,
const std::vector<std::string>& args) {
return boost::process::popen(
ctx, cmd, args, boost::process::process_stdio{nullptr, nullptr, nullptr});
};
} // namespace utils

View File

@@ -0,0 +1,119 @@
#include "utils/file_downloader.h"
#include <fstream>
#include "boost/asio/ssl.hpp"
#include "boost/beast.hpp"
#include "precomp.h"
namespace utils {
void DownloadFileFromHTTPS(std::string url, std::string filename) {
int max_redirects = 5;
std::string current_host;
std::string current_path;
if (url.find("https://") == 0) {
url.erase(0, 8); // https:// 제거
auto pos = url.find('/');
if (pos != std::string::npos) {
current_host = url.substr(0, pos);
current_path = url.substr(pos);
} else {
current_host = url;
current_path = "/";
}
} else {
// 상대 경로일 경우
current_path = url;
}
for (int redirect = 0; redirect < max_redirects; ++redirect) {
boost::asio::io_context ioc;
boost::asio::ssl::context ctx(boost::asio::ssl::context::tlsv12_client);
ctx.set_default_verify_paths();
boost::beast::ssl_stream<boost::beast::tcp_stream> stream(ioc, ctx);
// DNS 해석
boost::asio::ip::tcp::resolver resolver(ioc);
auto const results = resolver.resolve(current_host, "443");
// TCP 연결
boost::beast::get_lowest_layer(stream).connect(results);
// SSL 핸드셰이크
stream.handshake(boost::asio::ssl::stream_base::client);
// HTTP GET 요청
boost::beast::http::request<boost::beast::http::string_body> req{
boost::beast::http::verb::get, current_path, 11};
req.set(boost::beast::http::field::host, current_host);
req.set(boost::beast::http::field::user_agent, "Mozilla/5.0");
boost::beast::http::write(stream, req);
boost::beast::flat_buffer buffer;
// 응답 파서 만들기
boost::beast::http::response_parser<boost::beast::http::dynamic_body>
parser;
// body 제한 해제 (무제한)
parser.body_limit((std::numeric_limits<std::uint64_t>::max)());
// 응답 읽기
boost::beast::http::read(stream, buffer, parser);
auto res = parser.release();
int status = res.result_int();
if (status / 100 == 3) {
// 리다이렉트 처리
auto loc = res[boost::beast::http::field::location];
if (loc.empty()) {
throw std::runtime_error("Redirect without Location header");
}
// 새 호스트/경로 파싱
std::string new_url(loc);
if (new_url.find("https://") == 0) {
new_url.erase(0, 8); // https:// 제거
auto pos = new_url.find('/');
if (pos != std::string::npos) {
current_host = new_url.substr(0, pos);
current_path = new_url.substr(pos);
} else {
current_host = new_url;
current_path = "/";
}
} else {
// 상대 경로일 경우
current_path = new_url;
}
continue;
}
if (status != 200) {
throw std::runtime_error("HTTP request failed with status " +
std::to_string(status));
}
// 파일 저장
std::ofstream file(filename, std::ios::binary);
if (!file) {
throw std::runtime_error("Could not open file for writing: " + filename);
}
for (auto const& buffer_part : res.body().data()) {
file.write(reinterpret_cast<const char*>(buffer_part.data()),
buffer_part.size());
}
return;
}
throw std::runtime_error("Too many redirects");
}
} // namespace utils

View File

@@ -0,0 +1,71 @@
#include "utils/update_checker.h"
#include "precomp.h"
#include "utils/console.h"
#include "utils/file_downloader.h"
namespace utils {
int InstallYtdlp(boost::asio::io_context& ctx) {
BOOST_LOG_TRIVIAL(warning) << "ytdlp is unavailable. downloading ytdlp...";
#ifdef WIN32
DownloadFileFromHTTPS(
"https://github.com/yt-dlp/yt-dlp/releases/latest/download/yt-dlp.exe",
"yt-dlp.exe");
#else
DownloadFileFromHTTPS(
"https://github.com/yt-dlp/yt-dlp/releases/latest/download/yt-dlp",
"yt-dlp");
ExecuteCommand(ctx,
boost::process::environment::find_executable("chmod").string(),
{"+x", "yt-dlp"});
#endif
return 0;
}
static int FindNewLinePos(std::string& string, int start_pos) {
int newline_pos = start_pos;
while (newline_pos < string.size()) {
if (string[newline_pos] == '\n') {
return newline_pos;
}
newline_pos++;
}
return newline_pos;
}
int CheckUpdate(boost::asio::io_context& ctx) {
std::string output = "";
int old_newline_pos = 0;
int newline_pos = 0;
#ifdef WIN32
if (ExecuteCommand(ctx, "yt-dlp.exe", {"--version", "--newline"}, output) !=
0) {
InstallYtdlp(ctx);
ExecuteCommand(ctx, "yt-dlp.exe", {"--version", "--newline"}, output);
}
BOOST_LOG_TRIVIAL(info) << "yt-dlp version: "
<< output.substr(0, output.size() - 1);
output = "";
ExecuteCommand(ctx, "yt-dlp.exe", {"-U", "--newline"}, output);
#else
if (ExecuteCommand(ctx, "yt-dlp", {"--version", "--newline"}, output) != 0) {
InstallYtdlp(ctx);
ExecuteCommand(ctx, "yt-dlp", {"--version", "--newline"}, output);
}
BOOST_LOG_TRIVIAL(info) << "yt-dlp version: "
<< output.substr(0, output.size() - 1);
output = "";
ExecuteCommand(ctx, "yt-dlp", {"-U", "--newline"}, output);
#endif
while (newline_pos < output.size()) {
old_newline_pos = newline_pos;
newline_pos = FindNewLinePos(output, newline_pos);
BOOST_LOG_TRIVIAL(info) << output.substr(old_newline_pos, newline_pos - 1);
newline_pos++;
}
return 0;
}
} // namespace utils

8
tests/CMakeLists.txt Normal file
View File

@@ -0,0 +1,8 @@
file(GLOB_RECURSE TEST_SOURCES "*.cpp" "*.cxx" "*.cc")
foreach(test_src ${TEST_SOURCES})
get_filename_component(test_name ${test_src} NAME_WE)
add_executable(${test_name} ${test_src})
target_link_libraries(${test_name} PRIVATE ${BOT_NAME}_lib)
add_test(NAME ${test_name} COMMAND ${test_name})
endforeach()

View File

@@ -0,0 +1,52 @@
#include "precomp.h"
#include "utils/console.h"
#include "utils/update_checker.h"
int main() {
boost::asio::io_context ctx;
boost::system::error_code ec;
// utils::CheckUpdate(ctx);
char buf[8192];
#ifdef WIN32
/*try {
auto ytdlp_pipe = utils::OpenPipe(ctx,
"yt-dlp.exe", { "-o", "-", "--quiet", "--ignore-errors", "-f",
"bestaudio", "https://youtu.be/9_bTl2vvYQg?si=IVhvpDhnpPvziwQR" });
while (true) {
boost::system::error_code read_ec;
size_t bytes_read =
boost::asio::read(ytdlp_pipe, boost::asio::buffer(buf,
8192), read_ec); if (bytes_read > 0) { std::cout.write(buf, bytes_read);
}
if (read_ec == boost::asio::error::eof || read_ec) {
break;
}
}
}
catch (const boost::process::system_error& e) {
std::string error = e.what();
return -1;
}*/
#else
auto ytdlp_pipe = utils::OpenPipe(
ctx, "yt-dlp",
{"-o", "-", "--quiet", "--ignore-errors", "-f", "bestaudio",
"https://youtu.be/9_bTl2vvYQg?si=IVhvpDhnpPvziwQR"});
while (true) {
boost::system::error_code read_ec;
size_t bytes_read =
boost::asio::read(ytdlp_pipe, boost::asio::buffer(buf, 8192), read_ec);
if (bytes_read > 0) {
std::cout.write(buf, bytes_read);
}
if (read_ec == boost::asio::error::eof || read_ec) {
break;
}
}
#endif
return 0;
}

232
tests/ffmpeg_any_to_opus.cc Normal file
View File

@@ -0,0 +1,232 @@
#include "precomp.h"
#define OPUS_FRAME_SIZE 960 // 20ms @ 48kHz
int main() {
const char* input_filename = "golden.webm";
const char* output_filename = "output.opus";
AVFormatContext* fmt_ctx = NULL;
AVCodecContext* dec_ctx = NULL;
AVCodecContext* enc_ctx = NULL;
const AVCodec* decoder = NULL;
const AVCodec* encoder = NULL;
AVPacket* packet = NULL;
AVFrame* frame = NULL;
AVFrame* enc_frame = NULL;
SwrContext* swr_ctx = NULL;
FILE* outfile = NULL;
av_log_set_level(AV_LOG_ERROR);
if (avformat_open_input(&fmt_ctx, input_filename, NULL, NULL) < 0) {
fprintf(stderr, "Could not open input file\n");
return -1;
}
if (avformat_find_stream_info(fmt_ctx, NULL) < 0) {
fprintf(stderr, "Could not find stream info\n");
return -1;
}
int stream_index = -1;
for (unsigned i = 0; i < fmt_ctx->nb_streams; i++) {
if (fmt_ctx->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_AUDIO) {
stream_index = i;
break;
}
}
if (stream_index == -1) {
fprintf(stderr, "No audio stream found\n");
return -1;
}
decoder =
avcodec_find_decoder(fmt_ctx->streams[stream_index]->codecpar->codec_id);
if (!decoder) {
fprintf(stderr, "Decoder not found\n");
return -1;
}
dec_ctx = avcodec_alloc_context3(decoder);
avcodec_parameters_to_context(dec_ctx,
fmt_ctx->streams[stream_index]->codecpar);
avcodec_open2(dec_ctx, decoder, NULL);
encoder = avcodec_find_encoder(AV_CODEC_ID_OPUS);
if (!encoder) {
fprintf(stderr, "Opus encoder not found\n");
return -1;
}
enc_ctx = avcodec_alloc_context3(encoder);
AVChannelLayout enc_layout;
av_channel_layout_default(&enc_layout, 2); // 스테레오
av_channel_layout_copy(&enc_ctx->ch_layout, &enc_layout);
enc_ctx->sample_rate = 48000;
enc_ctx->sample_fmt = AV_SAMPLE_FMT_FLT;
enc_ctx->bit_rate = 128000;
avcodec_open2(enc_ctx, encoder, NULL);
swr_ctx = NULL;
if (swr_alloc_set_opts2(&swr_ctx, &enc_ctx->ch_layout, enc_ctx->sample_fmt,
enc_ctx->sample_rate, &dec_ctx->ch_layout,
dec_ctx->sample_fmt, dec_ctx->sample_rate, 0,
NULL) < 0) {
fprintf(stderr, "Failed to allocate SwrContext\n");
return -1;
}
swr_init(swr_ctx);
packet = av_packet_alloc();
frame = av_frame_alloc();
enc_frame = av_frame_alloc();
outfile = fopen(output_filename, "wb");
if (!outfile) {
fprintf(stderr, "Could not open output file\n");
return -1;
}
// 임시 PCM 버퍼 (float, 스테레오)
float* pcm_buffer = (float*)malloc(sizeof(float) * 2 * OPUS_FRAME_SIZE *
4); // 충분히 큰 버퍼
int buffered_samples = 0;
while (av_read_frame(fmt_ctx, packet) >= 0) {
if (packet->stream_index != stream_index) {
av_packet_unref(packet);
continue;
}
avcodec_send_packet(dec_ctx, packet);
while (avcodec_receive_frame(dec_ctx, frame) == 0) {
int max_out = av_rescale_rnd(
swr_get_delay(swr_ctx, dec_ctx->sample_rate) + frame->nb_samples,
enc_ctx->sample_rate, dec_ctx->sample_rate, AV_ROUND_UP);
uint8_t** out_data = NULL;
int out_linesize = 0;
av_samples_alloc_array_and_samples(&out_data, &out_linesize, 2, max_out,
enc_ctx->sample_fmt, 0);
int converted =
swr_convert(swr_ctx, out_data, max_out, (const uint8_t**)frame->data,
frame->nb_samples);
// float PCM으로 임시 버퍼에 추가
memcpy(pcm_buffer + buffered_samples * 2, out_data[0],
converted * 2 * sizeof(float));
buffered_samples += converted;
av_freep(&out_data[0]);
free(out_data);
// OPUS_FRAME_SIZE 단위로 인코딩
while (buffered_samples >= OPUS_FRAME_SIZE) {
enc_frame->nb_samples = OPUS_FRAME_SIZE;
enc_frame->format = enc_ctx->sample_fmt;
enc_frame->sample_rate = enc_ctx->sample_rate;
av_channel_layout_copy(&enc_frame->ch_layout, &enc_ctx->ch_layout);
enc_frame->data[0] = (uint8_t*)pcm_buffer;
AVPacket* out_pkt = av_packet_alloc();
avcodec_send_frame(enc_ctx, enc_frame);
while (avcodec_receive_packet(enc_ctx, out_pkt) == 0) {
fwrite(out_pkt->data, 1, out_pkt->size, outfile);
av_packet_unref(out_pkt);
}
av_packet_free(&out_pkt);
// 버퍼 이동
memmove(pcm_buffer, pcm_buffer + OPUS_FRAME_SIZE * 2,
(buffered_samples - OPUS_FRAME_SIZE) * 2 * sizeof(float));
buffered_samples -= OPUS_FRAME_SIZE;
}
}
av_packet_unref(packet);
}
// 디코더 플러시
avcodec_send_packet(dec_ctx, NULL);
while (avcodec_receive_frame(dec_ctx, frame) == 0) {
int max_out = av_rescale_rnd(
swr_get_delay(swr_ctx, dec_ctx->sample_rate) + frame->nb_samples,
enc_ctx->sample_rate, dec_ctx->sample_rate, AV_ROUND_UP);
uint8_t** out_data = NULL;
int out_linesize = 0;
av_samples_alloc_array_and_samples(&out_data, &out_linesize, 2, max_out,
enc_ctx->sample_fmt, 0);
int converted =
swr_convert(swr_ctx, out_data, max_out, (const uint8_t**)frame->data,
frame->nb_samples);
memcpy(pcm_buffer + buffered_samples * 2, out_data[0],
converted * 2 * sizeof(float));
buffered_samples += converted;
av_freep(&out_data[0]);
free(out_data);
while (buffered_samples >= OPUS_FRAME_SIZE) {
enc_frame->nb_samples = OPUS_FRAME_SIZE;
enc_frame->format = enc_ctx->sample_fmt;
enc_frame->sample_rate = enc_ctx->sample_rate;
av_channel_layout_copy(&enc_frame->ch_layout, &enc_ctx->ch_layout);
enc_frame->data[0] = (uint8_t*)pcm_buffer;
AVPacket* out_pkt = av_packet_alloc();
avcodec_send_frame(enc_ctx, enc_frame);
while (avcodec_receive_packet(enc_ctx, out_pkt) == 0) {
fwrite(out_pkt->data, 1, out_pkt->size, outfile);
av_packet_unref(out_pkt);
}
av_packet_free(&out_pkt);
memmove(pcm_buffer, pcm_buffer + OPUS_FRAME_SIZE * 2,
(buffered_samples - OPUS_FRAME_SIZE) * 2 * sizeof(float));
buffered_samples -= OPUS_FRAME_SIZE;
}
}
// 마지막 남은 샘플 인코딩
if (buffered_samples > 0) {
enc_frame->nb_samples = buffered_samples;
enc_frame->format = enc_ctx->sample_fmt;
enc_frame->sample_rate = enc_ctx->sample_rate;
av_channel_layout_copy(&enc_frame->ch_layout, &enc_ctx->ch_layout);
enc_frame->data[0] = (uint8_t*)pcm_buffer;
AVPacket* out_pkt = av_packet_alloc();
avcodec_send_frame(enc_ctx, enc_frame);
while (avcodec_receive_packet(enc_ctx, out_pkt) == 0) {
fwrite(out_pkt->data, 1, out_pkt->size, outfile);
av_packet_unref(out_pkt);
}
av_packet_free(&out_pkt);
}
// 인코더 플러시
avcodec_send_frame(enc_ctx, NULL);
AVPacket* out_pkt = av_packet_alloc();
while (avcodec_receive_packet(enc_ctx, out_pkt) == 0) {
fwrite(out_pkt->data, 1, out_pkt->size, outfile);
av_packet_unref(out_pkt);
}
av_packet_free(&out_pkt);
fclose(outfile);
free(pcm_buffer);
swr_free(&swr_ctx);
av_frame_free(&frame);
av_frame_free(&enc_frame);
av_packet_free(&packet);
avcodec_free_context(&dec_ctx);
avcodec_free_context(&enc_ctx);
avformat_close_input(&fmt_ctx);
printf("Encoding finished: %s\n", output_filename);
return 0;
}

View File

@@ -0,0 +1,155 @@
#include <fstream>
#include "ffmpeg/libavcodec.h"
#include "precomp.h"
int main() {
const char* input_filename = "golden.webm";
const char* output_filename = "output.pcm";
AVFormatContext* fmt_ctx = NULL;
AVCodecContext* codec_ctx = NULL;
const AVCodec* codec = NULL;
AVPacket* packet = NULL;
AVFrame* frame = NULL;
SwrContext* swr_ctx = NULL;
FILE* outfile = NULL;
if (avformat_open_input(&fmt_ctx, input_filename, NULL, NULL) < 0) {
fprintf(stderr, "Could not open input file\n");
return -1;
}
if (avformat_find_stream_info(fmt_ctx, NULL) < 0) {
fprintf(stderr, "Could not find stream info\n");
return -1;
}
int stream_index = -1;
for (unsigned i = 0; i < fmt_ctx->nb_streams; i++) {
if (fmt_ctx->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_AUDIO) {
stream_index = i;
break;
}
}
if (stream_index == -1) {
fprintf(stderr, "Could not find audio stream\n");
return -1;
}
codec =
avcodec_find_decoder(fmt_ctx->streams[stream_index]->codecpar->codec_id);
if (!codec) {
fprintf(stderr, "Could not find decoder\n");
return -1;
}
codec_ctx = avcodec_alloc_context3(codec);
if (!codec_ctx) {
fprintf(stderr, "Could not allocate codec context\n");
return -1;
}
if (avcodec_parameters_to_context(
codec_ctx, fmt_ctx->streams[stream_index]->codecpar) < 0) {
fprintf(stderr, "Failed to copy codec parameters\n");
return -1;
}
if (avcodec_open2(codec_ctx, codec, NULL) < 0) {
fprintf(stderr, "Could not open codec\n");
return -1;
}
packet = av_packet_alloc();
frame = av_frame_alloc();
if (!packet || !frame) {
fprintf(stderr, "Could not allocate packet or frame\n");
return -1;
}
outfile = fopen(output_filename, "wb");
if (!outfile) {
fprintf(stderr, "Could not open output file\n");
return -1;
}
// AVChannelLayout 초기화
AVChannelLayout in_layout, out_layout;
if (av_channel_layout_copy(&in_layout, &codec_ctx->ch_layout) < 0) {
fprintf(stderr, "Failed to copy channel layout\n");
return -1;
}
av_channel_layout_default(&out_layout, 2); // 스테레오
swr_ctx = NULL; // 먼저 NULL로 선언
if (swr_alloc_set_opts2(&swr_ctx, &out_layout, AV_SAMPLE_FMT_S16, 48000,
&in_layout, codec_ctx->sample_fmt,
codec_ctx->sample_rate, 0, NULL) < 0) {
fprintf(stderr, "Failed to allocate and set SwrContext\n");
return -1;
}
if (swr_init(swr_ctx) < 0) {
fprintf(stderr, "Failed to initialize SwrContext\n");
return -1;
}
while (av_read_frame(fmt_ctx, packet) >= 0) {
if (packet->stream_index == stream_index) {
if (avcodec_send_packet(codec_ctx, packet) == 0) {
while (avcodec_receive_frame(codec_ctx, frame) == 0) {
int out_samples =
av_rescale_rnd(swr_get_delay(swr_ctx, codec_ctx->sample_rate) +
frame->nb_samples,
48000, codec_ctx->sample_rate, AV_ROUND_UP);
uint8_t** out_buf = NULL;
int out_linesize = 0;
av_samples_alloc_array_and_samples(&out_buf, &out_linesize, 2,
out_samples, AV_SAMPLE_FMT_S16, 0);
int converted_samples =
swr_convert(swr_ctx, out_buf, out_samples,
(const uint8_t**)frame->data, frame->nb_samples);
fwrite(out_buf[0], 1, converted_samples * 2 * 2, outfile);
av_freep(&out_buf[0]);
free(out_buf);
}
}
}
av_packet_unref(packet);
}
// 디코더 플러시
avcodec_send_packet(codec_ctx, NULL);
while (avcodec_receive_frame(codec_ctx, frame) == 0) {
int out_samples = av_rescale_rnd(
swr_get_delay(swr_ctx, codec_ctx->sample_rate) + frame->nb_samples,
48000, codec_ctx->sample_rate, AV_ROUND_UP);
uint8_t** out_buf = NULL;
int out_linesize = 0;
av_samples_alloc_array_and_samples(&out_buf, &out_linesize, 2, out_samples,
AV_SAMPLE_FMT_S16, 0);
int converted_samples =
swr_convert(swr_ctx, out_buf, out_samples, (const uint8_t**)frame->data,
frame->nb_samples);
fwrite(out_buf[0], 1, converted_samples * 2 * 2, outfile);
av_freep(&out_buf[0]);
free(out_buf);
}
fclose(outfile);
swr_free(&swr_ctx);
av_frame_free(&frame);
av_packet_free(&packet);
avcodec_free_context(&codec_ctx);
avformat_close_input(&fmt_ctx);
printf("Decoding finished, output saved to %s\n", output_filename);
return 0;
}

8
tests/update.cc Normal file
View File

@@ -0,0 +1,8 @@
#include "precomp.h"
#include "utils/update_checker.h"
int main() {
boost::asio::io_context ctx;
utils::CheckUpdate(ctx);
return 0;
}

BIN
yt-dlp

Binary file not shown.