mirror of
https://github.com/HappyTanuki/BumbleCee.git
synced 2025-10-26 01:45:15 +00:00
Compare commits
15 Commits
4435337f40
...
refactor
| Author | SHA1 | Date | |
|---|---|---|---|
| 610074e4ac | |||
| e3b5e92164 | |||
| b123b2ecdb | |||
| e26d20a869 | |||
| f3974bb86f | |||
| f846dd7195 | |||
| 0bff043d6e | |||
| 7b75fc8861 | |||
| 177221bf73 | |||
| 51a49a4470 | |||
| a2821394bd | |||
| a03ed2dc78 | |||
| c59fef1d0b | |||
| 125fbcc49c | |||
| 4915a5c3b8 |
8
.clang-format
Normal file
8
.clang-format
Normal file
@@ -0,0 +1,8 @@
|
||||
# Google C/C++ Code Style Settings
|
||||
|
||||
Language: Cpp
|
||||
BasedOnStyle: Google
|
||||
Standard: c++20
|
||||
IndentWidth: 2
|
||||
UseTab: Never
|
||||
ColumnLimit: 80
|
||||
4
.gitignore
vendored
4
.gitignore
vendored
@@ -4,4 +4,6 @@ Music
|
||||
*.json
|
||||
yt-dlp
|
||||
ffmpeg
|
||||
password
|
||||
password
|
||||
.vs
|
||||
out
|
||||
10
.idea/.gitignore
generated
vendored
10
.idea/.gitignore
generated
vendored
@@ -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
9
.idea/BumbleCee.iml
generated
@@ -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
6
.idea/misc.xml
generated
@@ -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
8
.idea/modules.xml
generated
@@ -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
6
.idea/vcs.xml
generated
@@ -1,6 +0,0 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<project version="4">
|
||||
<component name="VcsDirectoryMappings">
|
||||
<mapping directory="" vcs="Git" />
|
||||
</component>
|
||||
</project>
|
||||
22
.vscode/c_cpp_properties.json
vendored
22
.vscode/c_cpp_properties.json
vendored
@@ -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
13
.vscode/launch.json
vendored
@@ -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
68
.vscode/settings.json
vendored
@@ -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,62 +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",
|
||||
"print": "cpp"
|
||||
"queue": "cpp"
|
||||
},
|
||||
"files.exclude": {
|
||||
"**/*.rpyc": true,
|
||||
"**/*.rpa": true,
|
||||
"**/*.rpymc": true,
|
||||
"**/cache/": true
|
||||
}
|
||||
}
|
||||
35
.vscode/tasks.json
vendored
35
.vscode/tasks.json
vendored
@@ -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
|
||||
}
|
||||
},
|
||||
]
|
||||
}
|
||||
185
CMakeLists.txt
185
CMakeLists.txt
@@ -1,46 +1,163 @@
|
||||
cmake_minimum_required (VERSION 3.6)
|
||||
|
||||
INCLUDE_DIRECTORIES(include . ${Boost_INCLUDE_DIR})
|
||||
|
||||
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)
|
||||
add_definitions( -DBOOST_ALL_NO_LIB )
|
||||
set( Boost_USE_STATIC_LIBS ON )
|
||||
find_package(Threads REQUIRED)
|
||||
find_package(OpenSSL REQUIRED)
|
||||
find_package(Boost REQUIRED)
|
||||
|
||||
target_include_directories(${BOT_NAME} PUBLIC
|
||||
${CMAKE_CURRENT_SOURCE_DIR}/include
|
||||
${OPENSSL_INCLUDE_DIR}
|
||||
/usr/include/opus
|
||||
include(FetchContent)
|
||||
|
||||
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)
|
||||
|
||||
target_link_libraries(${BOT_NAME}
|
||||
FetchContent_Declare(
|
||||
dpp
|
||||
opus
|
||||
ogg
|
||||
oggz
|
||||
${CMAKE_THREAD_LIBS_INIT}
|
||||
${OPENSSL_CRYPTO_LIBRARY}
|
||||
${OPENSSL_SSL_LIBRARY}
|
||||
${Boost_LIBRARIES}
|
||||
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)
|
||||
@@ -7,8 +7,10 @@ RUN apt-get update && \
|
||||
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/
|
||||
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
|
||||
@@ -16,4 +18,4 @@ COPY ./build/BumbleCee /BumbleCee
|
||||
COPY ./streamOpus.sh /streamOpus.sh
|
||||
RUN chmod +x BumbleCee
|
||||
RUN chmod +x streamOpus.sh
|
||||
ENTRYPOINT ["/usr/bin/tini", "--", "./BumbleCee"]
|
||||
ENTRYPOINT ["/usr/bin/tini", "--", "./BumbleCee"]
|
||||
|
||||
1303
flamegraph.pl
1303
flamegraph.pl
File diff suppressed because it is too large
Load Diff
@@ -1,67 +0,0 @@
|
||||
#!/bin/bash
|
||||
|
||||
# 기본값 설정
|
||||
n_flag=20
|
||||
i_flag=0
|
||||
process_args=()
|
||||
|
||||
# 입력값 파싱
|
||||
while [[ $# -gt 0 ]]; do
|
||||
case "$1" in
|
||||
-n)
|
||||
shift
|
||||
if [[ ! "$1" =~ ^[0-9]+$ ]]; then
|
||||
echo "오류: -n 옵션 뒤에는 숫자가 와야 합니다." >&2
|
||||
exit 1
|
||||
fi
|
||||
n_flag="$1"
|
||||
;;
|
||||
-i)
|
||||
shift
|
||||
if [[ ! "$1" =~ ^[0-9]+$ ]]; then
|
||||
echo "오류: -i 옵션 뒤에는 숫자가 와야 합니다." >&2
|
||||
exit 1
|
||||
fi
|
||||
i_flag="$1"
|
||||
;;
|
||||
*)
|
||||
# 프로세스 명 또는 PID 검증
|
||||
if [[ "$1" =~ ^[0-9]+$ ]]; then
|
||||
# 숫자인 경우, PID 존재 여부 확인
|
||||
if ! ps -p "$1" > /dev/null 2>&1; then
|
||||
echo "오류: PID $1 에 해당하는 프로세스가 존재하지 않습니다." >&2
|
||||
exit 1
|
||||
fi
|
||||
else
|
||||
# 문자열인 경우, 프로세스 이름 존재 여부 확인
|
||||
if ! pgrep -x "$1" > /dev/null 2>&1; then
|
||||
echo "오류: 프로세스 '$1' 가 실행 중이지 않습니다." >&2
|
||||
exit 1
|
||||
fi
|
||||
fi
|
||||
process_args+=("$1")
|
||||
;;
|
||||
esac
|
||||
shift
|
||||
done
|
||||
|
||||
# 프로세스 인자 검증
|
||||
if [[ ${#process_args[@]} -eq 0 ]]; then
|
||||
echo "오류: 최소한 하나 이상의 프로세스 이름 또는 PID를 입력해야 합니다." >&2
|
||||
exit 1
|
||||
fi
|
||||
|
||||
pid=$(pidof $process_args)
|
||||
|
||||
for x in $(seq 1 $n_flag)
|
||||
do
|
||||
sudo gdb -ex "set pagination 0" \
|
||||
-ex "thread apply all bt" \
|
||||
-batch -p $pid >> out.gdb 2> /dev/null
|
||||
sleep $i_flag
|
||||
done
|
||||
|
||||
./stackcollapse-gdb.pl out.gdb > out.folded
|
||||
rm out.gdb
|
||||
./flamegraph.pl out.folded > stackflame.svg
|
||||
rm out.folded
|
||||
@@ -1,67 +0,0 @@
|
||||
#pragma once
|
||||
#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::pair<std::shared_ptr<std::list<std::shared_ptr<MusicQueueElement>>>, std::list<std::shared_ptr<MusicQueueElement>>::iterator> 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(std::shared_ptr<bumbleBee::MusicQueueElement> music, dpp::discord_voice_client* client);
|
||||
};
|
||||
}
|
||||
@@ -1,54 +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
|
||||
@@ -1,69 +0,0 @@
|
||||
#pragma once
|
||||
#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(); \
|
||||
} \
|
||||
void execute(const dpp::slashcommand_t &event) override; \
|
||||
\
|
||||
protected: \
|
||||
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, "큐를 섞습니다")
|
||||
@@ -1,51 +0,0 @@
|
||||
#pragma once
|
||||
#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::pair<std::shared_ptr<std::list<std::shared_ptr<MusicQueueElement>>>, std::list<std::shared_ptr<MusicQueueElement>>::iterator>
|
||||
getQueueCopy();
|
||||
int
|
||||
size();
|
||||
std::list<std::shared_ptr<MusicQueueElement>>::iterator
|
||||
end();
|
||||
|
||||
bool repeat;
|
||||
|
||||
std::list<std::shared_ptr<MusicQueueElement>>::iterator currentPlayingPosition;
|
||||
|
||||
private:
|
||||
std::list<std::shared_ptr<MusicQueueElement>> queue;
|
||||
std::mutex queueMutex;
|
||||
};
|
||||
}
|
||||
@@ -1,20 +0,0 @@
|
||||
#pragma once
|
||||
#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;
|
||||
};
|
||||
}
|
||||
@@ -1,34 +0,0 @@
|
||||
#pragma once
|
||||
#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();
|
||||
};
|
||||
}
|
||||
@@ -1,56 +0,0 @@
|
||||
#pragma once
|
||||
#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;
|
||||
};
|
||||
}
|
||||
@@ -1,46 +0,0 @@
|
||||
#pragma once
|
||||
#include <iostream>
|
||||
#include <sstream>
|
||||
#include <queue>
|
||||
#include <regex>
|
||||
#include <boost/process.hpp>
|
||||
|
||||
namespace bumbleBee {
|
||||
class ConsoleUtils {
|
||||
public:
|
||||
/**
|
||||
* @brief 명령어를 쉘에서 실행하고 결과를 EOF 전까지 읽어 \n을 구분자로 토큰화하여 반환합니다
|
||||
* @param cmd 실행할 명령
|
||||
* @param args 아규먼트
|
||||
* @return std::queue<std::string> tokens
|
||||
*/
|
||||
static std::queue<std::string> safe_execute_command(const std::string& cmd, const std::vector<std::string>& args) {
|
||||
std::queue<std::string> tokens;
|
||||
try {
|
||||
boost::process::ipstream output; // 명령의 출력을 받을 스트림
|
||||
boost::process::child c(cmd, boost::process::args(args), boost::process::std_out > output);
|
||||
|
||||
std::string line;
|
||||
while (!output.eof() && std::getline(output, line))
|
||||
tokens.push(line);
|
||||
|
||||
c.wait(); // 프로세스가 종료될 때까지 대기
|
||||
return tokens;
|
||||
} catch (const std::exception& e) {
|
||||
return tokens;
|
||||
}
|
||||
}
|
||||
/**
|
||||
* @brief 명령어를 쉘에서 실행하고 결과를 파이프로 연결하여 반환합니다
|
||||
* @param cmd 실행할 명령
|
||||
* @param args 아규먼트
|
||||
* @return FILE* fd
|
||||
*/
|
||||
static FILE* safe_open_pipe(const std::string& cmd, const std::vector<std::string>& args) {
|
||||
boost::process::pipe pipe;
|
||||
boost::process::child c(cmd, boost::process::args(args), boost::process::std_out > pipe);
|
||||
|
||||
return fdopen(pipe.native_source(), "r");
|
||||
}
|
||||
};
|
||||
}
|
||||
@@ -1,77 +0,0 @@
|
||||
#pragma once
|
||||
#include <dpp/dpp.h>
|
||||
#include <Queue/MusicQueueElement.hpp>
|
||||
#include <cmath>
|
||||
|
||||
namespace bumbleBee {
|
||||
class QueuedMusicListEmbedProvider {
|
||||
public:
|
||||
static std::queue<dpp::embed> makeEmbed(std::shared_ptr<std::list<std::shared_ptr<MusicQueueElement>>> queue, std::list<std::shared_ptr<MusicQueueElement>>::iterator np, bool repeat) {
|
||||
std::queue<dpp::embed> returnValue;
|
||||
std::list<std::shared_ptr<MusicQueueElement>>::iterator it = queue->begin();
|
||||
if (queue->size() == 0) {
|
||||
dpp::embed embed = makeEmbedPart(queue, np, repeat, it);
|
||||
returnValue.push(embed);
|
||||
return returnValue;
|
||||
}
|
||||
for (int i = 0; i < std::ceil(queue->size() / 5.0) && it != queue->end(); i++) {
|
||||
dpp::embed embed = makeEmbedPart(queue, np, repeat, it);
|
||||
returnValue.push(embed);
|
||||
}
|
||||
return returnValue;
|
||||
}
|
||||
|
||||
private:
|
||||
static dpp::embed makeEmbedPart(
|
||||
std::shared_ptr<std::list<std::shared_ptr<MusicQueueElement>>> queue,
|
||||
std::list<std::shared_ptr<MusicQueueElement>>::iterator np,
|
||||
bool repeat,
|
||||
std::list<std::shared_ptr<MusicQueueElement>>::iterator &startIter) {
|
||||
dpp::embed embed = dpp::embed()
|
||||
.set_color(dpp::colors::sti_blue);
|
||||
|
||||
if (queue->begin() == queue->end()) {
|
||||
embed
|
||||
.set_title("큐가 비었습니다!")
|
||||
.set_timestamp(time(0));
|
||||
|
||||
if (repeat)
|
||||
embed.add_field(":repeat:","");
|
||||
|
||||
return embed;
|
||||
}
|
||||
|
||||
int startIndex = std::distance(queue->begin(), startIter) + 1;
|
||||
int index = startIndex;
|
||||
|
||||
for (; (index < startIndex + 5) && startIter != queue->end() && index < queue->size()+1; startIter++, index++) { //iter로 순회하면 왠지 모르게 이상한 값을 읽을 때가 있음 이유는 나도 몰?루
|
||||
if (*startIter == *np)
|
||||
embed.add_field (
|
||||
"np",
|
||||
"",
|
||||
true
|
||||
);
|
||||
else
|
||||
embed.add_field (
|
||||
std::to_string(index),
|
||||
"",
|
||||
true
|
||||
);
|
||||
embed.add_field (
|
||||
(*startIter)->embed.title,
|
||||
(*startIter)->embed.description,
|
||||
true
|
||||
)
|
||||
.add_field("","");
|
||||
}
|
||||
|
||||
if (startIter == queue->end() || index >= queue->size()+1) {
|
||||
embed.set_timestamp(time(0));
|
||||
if (repeat)
|
||||
embed.add_field(":repeat:","");
|
||||
}
|
||||
|
||||
return embed;
|
||||
}
|
||||
};
|
||||
}
|
||||
@@ -1,83 +0,0 @@
|
||||
#pragma once
|
||||
#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::safe_execute_command("which", {cmd}).size() == 0) {
|
||||
cluster->log(dpp::ll_error, cmd + " is unavaliable. unresolable error please install " + cmd);
|
||||
return false;
|
||||
}
|
||||
else
|
||||
return true;
|
||||
}
|
||||
|
||||
static void validateFFMPEG(std::shared_ptr<dpp::cluster> cluster) {
|
||||
std::queue<std::string> result = ConsoleUtils::safe_execute_command(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/bin/ffmpeg .");
|
||||
system("rm -rf ffmpeg-master-latest-linux64-gpl");
|
||||
SettingsManager::setFFMPEG_CMD("./ffmpeg");
|
||||
}
|
||||
}
|
||||
|
||||
static void validateYTDLP(std::shared_ptr<dpp::cluster> cluster) {
|
||||
std::queue<std::string> result = ConsoleUtils::safe_execute_command(SettingsManager::getYTDLP_CMD(), {"--version"});
|
||||
std::string 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 validateYTDLPFFMPEGBinary(std::shared_ptr<dpp::cluster> cluster) {
|
||||
cluster->log(dpp::ll_info, "Checking if yt-dlp and ffmpeg is available...");
|
||||
validateFFMPEG(cluster);
|
||||
validateYTDLP(cluster);
|
||||
}
|
||||
|
||||
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::safe_execute_command("./yt-dlp", {"-U"});
|
||||
while(!result.empty()) {
|
||||
std::string front = result.front();
|
||||
result.pop();
|
||||
cluster->log(dpp::ll_info, front);
|
||||
}
|
||||
}
|
||||
};
|
||||
}
|
||||
19
include/core/bumblebee.h
Normal file
19
include/core/bumblebee.h
Normal 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
26
include/precomp.h
Normal 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
41
include/utils/console.h
Normal 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
|
||||
10
include/utils/file_downloader.h
Normal file
10
include/utils/file_downloader.h
Normal 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
|
||||
14
include/utils/update_checker.h
Normal file
14
include/utils/update_checker.h
Normal 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
|
||||
@@ -1,138 +0,0 @@
|
||||
#include <Audio/MusicPlayManager.hpp>
|
||||
#include <ogg/ogg.h>
|
||||
#include <oggz/oggz.h>
|
||||
#include <algorithm>
|
||||
#include <Settings/SettingsManager.hpp>
|
||||
|
||||
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) {
|
||||
dpp::snowflake gid = dpp::find_channel(event.voice_client->channel_id)->guild_id;
|
||||
queueMap[gid]->next_music(); // TODO("repeat가 꺼져 있을 때 노래 큐에서 지우기")
|
||||
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 && queueMap[gid]->currentPlayingPosition != queueMap[gid]->end();
|
||||
});
|
||||
|
||||
auto next = queueMap[gid]->currentPlayingPosition;
|
||||
send_audio_to_voice(*next, 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::pair<std::shared_ptr<std::list<std::shared_ptr<MusicQueueElement>>>, std::list<std::shared_ptr<MusicQueueElement>>::iterator>
|
||||
MusicPlayManager::getQueue(const dpp::snowflake guildId){
|
||||
auto returnValue = queueMap[guildId]->getQueueCopy();
|
||||
return returnValue;
|
||||
}
|
||||
|
||||
MusicQueueElement MusicPlayManager::getNowPlaying(const dpp::snowflake guildId) {
|
||||
std::shared_ptr<MusicQueueElement> nowplaying = queueMap[guildId]->nowplaying();
|
||||
if (nowplaying == nullptr)
|
||||
return MusicQueueElement("", "", dpp::user(), dpp::embed());
|
||||
MusicQueueElement returnValue(*nowplaying);
|
||||
return returnValue;
|
||||
}
|
||||
|
||||
void MusicPlayManager::send_audio_to_voice(std::shared_ptr<bumbleBee::MusicQueueElement> music, dpp::discord_voice_client* client) {
|
||||
std::thread t([](std::shared_ptr<bumbleBee::MusicQueueElement> music, dpp::discord_voice_client* client) { // TODO: thread 벡터 만들고 delete, leave, skip시에 전송 중지하도록 만들기
|
||||
std::string command = "./streamOpus.sh ";
|
||||
command += SettingsManager::getYTDLP_CMD() + " ";
|
||||
command += SettingsManager::getFFMPEG_CMD() + " ";
|
||||
command += "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 && music != nullptr) {
|
||||
static const constexpr long CHUNK_READ = BUFSIZ * 2;
|
||||
|
||||
const long read_bytes = oggz_read(og, CHUNK_READ);
|
||||
|
||||
/* break on eof */
|
||||
if (!read_bytes) {
|
||||
break;
|
||||
}
|
||||
}
|
||||
|
||||
if (music != nullptr)
|
||||
client->creator->log(dpp::ll_info, "Sending " + music->embed.title + " - " + music->id + " complete!");
|
||||
else {
|
||||
client->stop_audio();
|
||||
client->pause_audio(true);
|
||||
}
|
||||
|
||||
oggz_close(og);
|
||||
|
||||
client->insert_marker();
|
||||
}, music, client);
|
||||
t.detach();
|
||||
}
|
||||
}
|
||||
|
||||
@@ -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 << std::endl;
|
||||
}
|
||||
});
|
||||
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.");
|
||||
}
|
||||
}
|
||||
@@ -1,32 +0,0 @@
|
||||
#include <Commands/BumbleBeeCommand.hpp>
|
||||
#include <format>
|
||||
|
||||
namespace bumbleBee::commands {
|
||||
void Delete::execute(const dpp::slashcommand_t &event) {
|
||||
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));
|
||||
}
|
||||
}
|
||||
@@ -1,23 +0,0 @@
|
||||
#include <Commands/BumbleBeeCommand.hpp>
|
||||
|
||||
namespace bumbleBee::commands {
|
||||
void Leave::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()) {
|
||||
event.edit_original_response(dpp::message("현재 음성 채팅방에 있는 상태가 아닙니다!"));
|
||||
return;
|
||||
}
|
||||
musicManager->clear(event.command.guild_id);
|
||||
|
||||
v->voiceclient->stop_audio();
|
||||
v->voiceclient->pause_audio(true);
|
||||
event.from->clear_queue();
|
||||
event.from->disconnect_voice(event.command.guild_id);
|
||||
|
||||
event.edit_original_response(dpp::message("음성 채팅방을 떠납니다!"));
|
||||
}
|
||||
|
||||
void Leave::init() {
|
||||
}
|
||||
}
|
||||
@@ -1,240 +0,0 @@
|
||||
#include <Commands/BumbleBeeCommand.hpp>
|
||||
#include <Utils/ConsoleUtils.hpp>
|
||||
#include <Settings/SettingsManager.hpp>
|
||||
#include <dpp/nlohmann/json.hpp>
|
||||
#include <Utils/QueuedMusicListEmbedProvider.hpp>
|
||||
#include <Utils/ConsoleUtils.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"));
|
||||
// query = "\"" + query + "\"";
|
||||
|
||||
std::queue<std::string> ids =
|
||||
ConsoleUtils::safe_execute_command(
|
||||
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.empty()) {
|
||||
if (ids.front() == "") {
|
||||
ids.pop();
|
||||
}
|
||||
|
||||
std::string jsonData = ConsoleUtils::safe_execute_command(SettingsManager::getYTDLP_CMD(), {
|
||||
"--default-search",
|
||||
"ytsearch",
|
||||
"--flat-playlist",
|
||||
"--skip-download",
|
||||
"--quiet",
|
||||
"--ignore-errors",
|
||||
"-J",
|
||||
"http://youtu.be/" + ids.front()
|
||||
}).front();
|
||||
|
||||
std::istringstream iss(jsonData);
|
||||
nlohmann::json videoDataJson;
|
||||
iss >> videoDataJson;
|
||||
|
||||
time_t SongLength = int(videoDataJson["duration"]);
|
||||
char SongLengthStr[13];
|
||||
tm t;
|
||||
t.tm_mday = SongLength / 86400;
|
||||
t.tm_hour = (SongLength % 86400)/3600;
|
||||
t.tm_min = (SongLength % 3600)/60;
|
||||
t.tm_sec = SongLength%60;
|
||||
if (t.tm_mday > 0)
|
||||
strftime(SongLengthStr, sizeof(SongLengthStr), "%d:%H:%M:%S", &t);
|
||||
else if (t.tm_hour > 0)
|
||||
strftime(SongLengthStr, sizeof(SongLengthStr), "%H:%M:%S", &t);
|
||||
else
|
||||
strftime(SongLengthStr, sizeof(SongLengthStr), "%M:%S", &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()) {
|
||||
std::thread t([](
|
||||
std::queue<std::string> ids,
|
||||
dpp::snowflake guildId,
|
||||
dpp::snowflake channelId,
|
||||
std::string query,
|
||||
dpp::user user,
|
||||
dpp::cluster* cluster,
|
||||
std::shared_ptr<MusicPlayManager> manager
|
||||
) {
|
||||
while (!ids.empty()) {
|
||||
if (ids.front() == "") {
|
||||
ids.pop();
|
||||
continue;
|
||||
}
|
||||
|
||||
std::string jsonData = ConsoleUtils::safe_execute_command(SettingsManager::getYTDLP_CMD(), {
|
||||
"--default-search",
|
||||
"ytsearch",
|
||||
"--flat-playlist",
|
||||
"--skip-download",
|
||||
"--quiet",
|
||||
"--ignore-errors",
|
||||
"-J",
|
||||
"http://youtu.be/" + ids.front()
|
||||
}).front();
|
||||
|
||||
std::istringstream iss(jsonData);
|
||||
nlohmann::json videoDataJson;
|
||||
iss >> videoDataJson;
|
||||
|
||||
time_t SongLength = int(videoDataJson["duration"]);
|
||||
char SongLengthStr[13];
|
||||
tm t;
|
||||
t.tm_mday = SongLength / 86400;
|
||||
t.tm_hour = (SongLength % 86400)/3600;
|
||||
t.tm_min = (SongLength % 3600)/60;
|
||||
t.tm_sec = SongLength%60;
|
||||
if (t.tm_mday > 0)
|
||||
strftime(SongLengthStr, sizeof(SongLengthStr), "%d:%H:%M:%S", &t);
|
||||
else if (t.tm_hour > 0)
|
||||
strftime(SongLengthStr, sizeof(SongLengthStr), "%H:%M:%S", &t);
|
||||
else
|
||||
strftime(SongLengthStr, sizeof(SongLengthStr), "%M:%S", &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
|
||||
);
|
||||
|
||||
auto music = std::make_shared<MusicQueueElement>(ids.front(), query, user, embed);
|
||||
|
||||
cluster->log(dpp::ll_info, "Enqueuing " + music->embed.title + " - " + music->id);
|
||||
manager->queue_music(guildId, music);
|
||||
ids.pop();
|
||||
}
|
||||
|
||||
std::mutex messageorder;
|
||||
std::unique_lock lock(messageorder);
|
||||
std::condition_variable messageSentCondition; // 개씨발 코드 개더러워 ;;
|
||||
bool messagesent = false;
|
||||
auto queue = manager->getQueue(guildId);
|
||||
auto queued = QueuedMusicListEmbedProvider::makeEmbed(queue.first, queue.second, manager->getRepeat(guildId));
|
||||
if (!queued.empty()) {
|
||||
dpp::message followMsg(channelId, "현재 큐에 있는 항목:");
|
||||
|
||||
followMsg.add_embed(queued.front());
|
||||
messagesent = false;
|
||||
cluster->message_create(followMsg, [&](const dpp::confirmation_callback_t &callback){
|
||||
messagesent = true;
|
||||
messageSentCondition.notify_all();
|
||||
});
|
||||
|
||||
messageSentCondition.wait(lock, [&](){ return messagesent; });
|
||||
queued.pop();
|
||||
}
|
||||
while (!queued.empty()) {
|
||||
dpp::message followMsg(channelId, "");
|
||||
|
||||
followMsg.add_embed(queued.front());
|
||||
messagesent = false;
|
||||
cluster->message_create(followMsg, [&](const dpp::confirmation_callback_t &callback){
|
||||
messagesent = true;
|
||||
messageSentCondition.notify_all();
|
||||
});
|
||||
|
||||
messageSentCondition.wait(lock, [&](){ return messagesent; });
|
||||
queued.pop();
|
||||
}
|
||||
},
|
||||
ids,
|
||||
event.command.guild_id,
|
||||
event.command.channel_id,
|
||||
query,
|
||||
event.command.usr,
|
||||
event.from->creator,
|
||||
musicManager);
|
||||
|
||||
t.detach();
|
||||
}
|
||||
|
||||
if (!musics.empty()) {
|
||||
event.from->creator->log(dpp::ll_info, "Enqueuing " + musics.front()->embed.title + " - " + 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 {
|
||||
msg.content = "검색 결과가 없습니다";
|
||||
event.edit_original_response(msg);
|
||||
}
|
||||
|
||||
if (musicManager->getNowPlaying(event.command.guild_id).id == "") {
|
||||
dpp::voiceconn* v = event.from->get_voice(event.command.guild_id);
|
||||
|
||||
if (!v || !v->voiceclient || !v->voiceclient->is_ready()) {
|
||||
event.edit_original_response(dpp::message("현재 음성 채팅방에 있는 상태가 아닙니다!"));
|
||||
return;
|
||||
}
|
||||
v->voiceclient->insert_marker();
|
||||
v->voiceclient->pause_audio(false);
|
||||
}
|
||||
}
|
||||
|
||||
void Play::init() {
|
||||
add_option(dpp::command_option(dpp::co_string, "query", "링크 또는 검색어", true));
|
||||
}
|
||||
}
|
||||
@@ -1,73 +0,0 @@
|
||||
#include <Commands/BumbleBeeCommand.hpp>
|
||||
#include <Utils/QueuedMusicListEmbedProvider.hpp>
|
||||
|
||||
namespace bumbleBee::commands {
|
||||
void Queue::execute(const dpp::slashcommand_t &event) {
|
||||
auto queue = musicManager->getQueue(event.command.guild_id);
|
||||
|
||||
dpp::message msg;
|
||||
dpp::embed embed;
|
||||
|
||||
auto queued = QueuedMusicListEmbedProvider::makeEmbed(queue.first, queue.second, musicManager->getRepeat(event.command.guild_id));
|
||||
|
||||
// if (queue.first.size() == 0) {
|
||||
// msg.add_embed(queued.front());
|
||||
// event.edit_original_response(msg);
|
||||
// return;
|
||||
// }
|
||||
|
||||
// &&
|
||||
// queue.first.size() != 0 &&
|
||||
// *queue.second != nullptr &&
|
||||
// (*queue.second)->id != ""
|
||||
|
||||
if (queue.first->size() != 0 && queue.first->end() != queue.second && (*queue.second)->id != "") {
|
||||
msg.content = "지금 재생 중:";
|
||||
msg.add_embed((*queue.second)->embed);
|
||||
event.edit_original_response(msg);
|
||||
}
|
||||
else {
|
||||
msg.content = "재생 중인 노래가 없습니다";
|
||||
event.edit_original_response(msg);
|
||||
}
|
||||
|
||||
std::thread t([](std::queue<dpp::embed> queued, dpp::snowflake channel_id, dpp::cluster* cluster) {
|
||||
std::mutex messageorder;
|
||||
std::unique_lock lock(messageorder);
|
||||
std::condition_variable messageSentCondition;
|
||||
bool messagesent = false;
|
||||
if (!queued.empty()) {
|
||||
dpp::message followMsg(channel_id, "현재 큐에 있는 항목:");
|
||||
|
||||
followMsg.add_embed(queued.front());
|
||||
messagesent = false;
|
||||
cluster->message_create(followMsg, [&](const dpp::confirmation_callback_t &callback){
|
||||
messagesent = true;
|
||||
messageSentCondition.notify_all();
|
||||
});
|
||||
|
||||
messageSentCondition.wait(lock, [&](){ return messagesent; });
|
||||
|
||||
queued.pop();
|
||||
}
|
||||
while (!queued.empty()) {
|
||||
dpp::message followMsg(channel_id, "");
|
||||
|
||||
followMsg.add_embed(queued.front());
|
||||
|
||||
messagesent = false;
|
||||
cluster->message_create(followMsg, [&](const dpp::confirmation_callback_t &callback){
|
||||
messagesent = true;
|
||||
messageSentCondition.notify_all();
|
||||
});
|
||||
|
||||
messageSentCondition.wait(lock, [&](){ return messagesent; });
|
||||
queued.pop();
|
||||
}
|
||||
}, queued, event.command.channel_id, event.from->creator);
|
||||
t.detach();
|
||||
}
|
||||
|
||||
void Queue::init() {
|
||||
}
|
||||
}
|
||||
@@ -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() {}
|
||||
}
|
||||
@@ -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() {
|
||||
}
|
||||
}
|
||||
@@ -1,19 +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()) {
|
||||
event.edit_original_response(dpp::message("스킵하려면 음악을 재생중이어야 합니다!"));
|
||||
return;
|
||||
}
|
||||
v->voiceclient->stop_audio();
|
||||
v->voiceclient->insert_marker();
|
||||
|
||||
event.edit_original_response(dpp::message("스킵했습니다!"));
|
||||
}
|
||||
|
||||
void Skip::init() {
|
||||
}
|
||||
}
|
||||
@@ -1,115 +0,0 @@
|
||||
#include <Queue/MusicQueue.hpp>
|
||||
#include <iostream>
|
||||
#include <algorithm>
|
||||
#include <Utils/QueuedMusicListEmbedProvider.hpp>
|
||||
|
||||
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);
|
||||
if (currentPlayingPosition == queue.end())
|
||||
return nullptr;
|
||||
else
|
||||
return *currentPlayingPosition;
|
||||
}
|
||||
std::list<std::shared_ptr<MusicQueueElement>>::iterator MusicQueue::next_music() {
|
||||
std::lock_guard<std::mutex> lock(queueMutex);
|
||||
if (currentPlayingPosition == --queue.end() && repeat)
|
||||
currentPlayingPosition = queue.begin();
|
||||
else if (currentPlayingPosition == --queue.end() && !repeat) // 반복이 꺼져있을 때 큐 재생이 끝난 경우
|
||||
currentPlayingPosition = queue.end();
|
||||
else if (currentPlayingPosition == queue.end() && !repeat) // 반복이 꺼져있고 현재 재생 곡이 없는데 새 곡이 들어왔을 경우
|
||||
currentPlayingPosition = --queue.end();
|
||||
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::pair<std::shared_ptr<std::list<std::shared_ptr<MusicQueueElement>>>, std::list<std::shared_ptr<MusicQueueElement>>::iterator>
|
||||
MusicQueue::getQueueCopy() {
|
||||
std::lock_guard<std::mutex> lock(queueMutex);
|
||||
std::shared_ptr<std::list<std::shared_ptr<MusicQueueElement>>> returnValue = std::make_shared<std::list<std::shared_ptr<MusicQueueElement>>>();
|
||||
|
||||
for (auto i = queue.begin(); i != queue.end(); i++)
|
||||
returnValue->push_back(*i);
|
||||
|
||||
if (returnValue->begin() == returnValue->end() || currentPlayingPosition == queue.end())
|
||||
return std::make_pair(returnValue, returnValue->end());
|
||||
|
||||
std::list<std::shared_ptr<MusicQueueElement>>::iterator iter = returnValue->begin();
|
||||
std::advance(iter, std::distance(queue.begin(), currentPlayingPosition));
|
||||
|
||||
return std::make_pair(returnValue, iter);
|
||||
}
|
||||
int MusicQueue::size() {
|
||||
std::lock_guard<std::mutex> lock(queueMutex);
|
||||
return queue.size();
|
||||
}
|
||||
std::list<std::shared_ptr<MusicQueueElement>>::iterator MusicQueue::end() {
|
||||
std::lock_guard<std::mutex> lock(queueMutex);
|
||||
return queue.end();
|
||||
}
|
||||
}
|
||||
@@ -1,124 +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";
|
||||
dpp::loglevel SettingsManager::LOGLEVEL = dpp::ll_debug;
|
||||
bool SettingsManager::REGISTER_COMMAND = false;
|
||||
|
||||
bool SettingsManager::validateToken() {
|
||||
nlohmann::json response;
|
||||
std::string curl = ConsoleUtils::safe_execute_command("/usr/bin/which", {"curl"}).front();
|
||||
if (curl == "") {
|
||||
std::cout << "curl is unavaliable. unresolable error please install curl." << std::endl;
|
||||
return false;
|
||||
}
|
||||
|
||||
std::string stresult = ConsoleUtils::safe_execute_command(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();
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -1,103 +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::safe_execute_command(
|
||||
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::safe_execute_command(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.");
|
||||
|
||||
stream = ConsoleUtils::safe_open_pipe("./streamOpus.sh", {
|
||||
SettingsManager::getYTDLP_CMD(),
|
||||
downloadID,
|
||||
SettingsManager::getFFMPEG_CMD()
|
||||
});
|
||||
|
||||
cluster->log(dpp::ll_info, "Thread id: " + tid.str() + " Opened stream: " + downloadID);
|
||||
});
|
||||
th.detach();
|
||||
}
|
||||
}
|
||||
}
|
||||
10
src/main.cc
Normal file
10
src/main.cc
Normal 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;
|
||||
}
|
||||
@@ -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
53
src/utils/console.cc
Normal 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
|
||||
119
src/utils/file_downloader.cc
Normal file
119
src/utils/file_downloader.cc
Normal 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
|
||||
71
src/utils/update_checker.cc
Normal file
71
src/utils/update_checker.cc
Normal 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
|
||||
@@ -1,72 +0,0 @@
|
||||
#!/usr/bin/perl -ws
|
||||
#
|
||||
# stackcollapse-gdb Collapse GDB backtraces
|
||||
#
|
||||
# Parse a list of GDB backtraces as generated with the poor man's
|
||||
# profiler [1]:
|
||||
#
|
||||
# for x in $(seq 1 500); do
|
||||
# gdb -ex "set pagination 0" -ex "thread apply all bt" -batch -p $pid 2> /dev/null
|
||||
# sleep 0.01
|
||||
# done
|
||||
#
|
||||
# [1] http://poormansprofiler.org/
|
||||
#
|
||||
# Copyright 2014 Gabriel Corona. All rights reserved.
|
||||
#
|
||||
# CDDL HEADER START
|
||||
#
|
||||
# The contents of this file are subject to the terms of the
|
||||
# Common Development and Distribution License (the "License").
|
||||
# You may not use this file except in compliance with the License.
|
||||
#
|
||||
# You can obtain a copy of the license at docs/cddl1.txt or
|
||||
# http://opensource.org/licenses/CDDL-1.0.
|
||||
# See the License for the specific language governing permissions
|
||||
# and limitations under the License.
|
||||
#
|
||||
# When distributing Covered Code, include this CDDL HEADER in each
|
||||
# file and include the License file at docs/cddl1.txt.
|
||||
# If applicable, add the following below this CDDL HEADER, with the
|
||||
# fields enclosed by brackets "[]" replaced with your own identifying
|
||||
# information: Portions Copyright [yyyy] [name of copyright owner]
|
||||
#
|
||||
# CDDL HEADER END
|
||||
|
||||
use strict;
|
||||
|
||||
my $current = "";
|
||||
my $previous_function = "";
|
||||
|
||||
my %stacks;
|
||||
|
||||
while(<>) {
|
||||
chomp;
|
||||
if (m/^Thread/) {
|
||||
$current=""
|
||||
}
|
||||
elsif(m/^#[0-9]* *([^ ]*) ([^ ]*) ([^ ]*) ([^ ]*)/) {
|
||||
my $function = $3;
|
||||
my $alt = $1;
|
||||
if(not($1 =~ /0x[a-zA-Z0-9]*/)) {
|
||||
$function = $alt;
|
||||
}
|
||||
if ($current eq "") {
|
||||
$current = $function;
|
||||
} else {
|
||||
$current = $function . ";" . $current;
|
||||
}
|
||||
} elsif(!($current eq "")) {
|
||||
$stacks{$current} += 1;
|
||||
$current = "";
|
||||
}
|
||||
}
|
||||
|
||||
if(!($current eq "")) {
|
||||
$stacks{$current} += 1;
|
||||
$current = "";
|
||||
}
|
||||
|
||||
foreach my $k (sort { $a cmp $b } keys %stacks) {
|
||||
print "$k $stacks{$k}\n";
|
||||
}
|
||||
@@ -1,435 +0,0 @@
|
||||
#!/usr/bin/perl -w
|
||||
#
|
||||
# stackcollapse-perf.pl collapse perf samples into single lines.
|
||||
#
|
||||
# Parses a list of multiline stacks generated by "perf script", and
|
||||
# outputs a semicolon separated stack followed by a space and a count.
|
||||
# If memory addresses (+0xd) are present, they are stripped, and resulting
|
||||
# identical stacks are colased with their counts summed.
|
||||
#
|
||||
# USAGE: ./stackcollapse-perf.pl [options] infile > outfile
|
||||
#
|
||||
# Run "./stackcollapse-perf.pl -h" to list options.
|
||||
#
|
||||
# Example input:
|
||||
#
|
||||
# swapper 0 [000] 158665.570607: cpu-clock:
|
||||
# ffffffff8103ce3b native_safe_halt ([kernel.kallsyms])
|
||||
# ffffffff8101c6a3 default_idle ([kernel.kallsyms])
|
||||
# ffffffff81013236 cpu_idle ([kernel.kallsyms])
|
||||
# ffffffff815bf03e rest_init ([kernel.kallsyms])
|
||||
# ffffffff81aebbfe start_kernel ([kernel.kallsyms].init.text)
|
||||
# [...]
|
||||
#
|
||||
# Example output:
|
||||
#
|
||||
# swapper;start_kernel;rest_init;cpu_idle;default_idle;native_safe_halt 1
|
||||
#
|
||||
# Input may be created and processed using:
|
||||
#
|
||||
# perf record -a -g -F 997 sleep 60
|
||||
# perf script | ./stackcollapse-perf.pl > out.stacks-folded
|
||||
#
|
||||
# The output of "perf script" should include stack traces. If these are missing
|
||||
# for you, try manually selecting the perf script output; eg:
|
||||
#
|
||||
# perf script -f comm,pid,tid,cpu,time,event,ip,sym,dso,trace | ...
|
||||
#
|
||||
# This is also required for the --pid or --tid options, so that the output has
|
||||
# both the PID and TID.
|
||||
#
|
||||
# Copyright 2012 Joyent, Inc. All rights reserved.
|
||||
# Copyright 2012 Brendan Gregg. All rights reserved.
|
||||
#
|
||||
# CDDL HEADER START
|
||||
#
|
||||
# The contents of this file are subject to the terms of the
|
||||
# Common Development and Distribution License (the "License").
|
||||
# You may not use this file except in compliance with the License.
|
||||
#
|
||||
# You can obtain a copy of the license at docs/cddl1.txt or
|
||||
# http://opensource.org/licenses/CDDL-1.0.
|
||||
# See the License for the specific language governing permissions
|
||||
# and limitations under the License.
|
||||
#
|
||||
# When distributing Covered Code, include this CDDL HEADER in each
|
||||
# file and include the License file at docs/cddl1.txt.
|
||||
# If applicable, add the following below this CDDL HEADER, with the
|
||||
# fields enclosed by brackets "[]" replaced with your own identifying
|
||||
# information: Portions Copyright [yyyy] [name of copyright owner]
|
||||
#
|
||||
# CDDL HEADER END
|
||||
#
|
||||
# 02-Mar-2012 Brendan Gregg Created this.
|
||||
# 02-Jul-2014 " " Added process name to stacks.
|
||||
|
||||
use strict;
|
||||
use Getopt::Long;
|
||||
|
||||
my %collapsed;
|
||||
|
||||
sub remember_stack {
|
||||
my ($stack, $count) = @_;
|
||||
$collapsed{$stack} += $count;
|
||||
}
|
||||
my $annotate_kernel = 0; # put an annotation on kernel function
|
||||
my $annotate_jit = 0; # put an annotation on jit symbols
|
||||
my $annotate_all = 0; # enale all annotations
|
||||
my $include_pname = 1; # include process names in stacks
|
||||
my $include_pid = 0; # include process ID with process name
|
||||
my $include_tid = 0; # include process & thread ID with process name
|
||||
my $include_addrs = 0; # include raw address where a symbol can't be found
|
||||
my $tidy_java = 1; # condense Java signatures
|
||||
my $tidy_generic = 1; # clean up function names a little
|
||||
my $target_pname; # target process name from perf invocation
|
||||
my $event_filter = ""; # event type filter, defaults to first encountered event
|
||||
my $event_defaulted = 0; # whether we defaulted to an event (none provided)
|
||||
my $event_warning = 0; # if we printed a warning for the event
|
||||
|
||||
my $show_inline = 0;
|
||||
my $show_context = 0;
|
||||
|
||||
my $srcline_in_input = 0; # if there are extra lines with source location (perf script -F+srcline)
|
||||
GetOptions('inline' => \$show_inline,
|
||||
'context' => \$show_context,
|
||||
'srcline' => \$srcline_in_input,
|
||||
'pid' => \$include_pid,
|
||||
'kernel' => \$annotate_kernel,
|
||||
'jit' => \$annotate_jit,
|
||||
'all' => \$annotate_all,
|
||||
'tid' => \$include_tid,
|
||||
'addrs' => \$include_addrs,
|
||||
'event-filter=s' => \$event_filter)
|
||||
or die <<USAGE_END;
|
||||
USAGE: $0 [options] infile > outfile\n
|
||||
--pid # include PID with process names [1]
|
||||
--tid # include TID and PID with process names [1]
|
||||
--inline # un-inline using addr2line
|
||||
--all # all annotations (--kernel --jit)
|
||||
--kernel # annotate kernel functions with a _[k]
|
||||
--jit # annotate jit functions with a _[j]
|
||||
--context # adds source context to --inline
|
||||
--srcline # parses output of 'perf script -F+srcline' and adds source context
|
||||
--addrs # include raw addresses where symbols can't be found
|
||||
--event-filter=EVENT # event name filter\n
|
||||
[1] perf script must emit both PID and TIDs for these to work; eg, Linux < 4.1:
|
||||
perf script -f comm,pid,tid,cpu,time,event,ip,sym,dso,trace
|
||||
for Linux >= 4.1:
|
||||
perf script -F comm,pid,tid,cpu,time,event,ip,sym,dso,trace
|
||||
If you save this output add --header on Linux >= 3.14 to include perf info.
|
||||
USAGE_END
|
||||
|
||||
if ($annotate_all) {
|
||||
$annotate_kernel = $annotate_jit = 1;
|
||||
}
|
||||
|
||||
my %inlineCache;
|
||||
|
||||
my %nmCache;
|
||||
|
||||
sub inlineCacheAdd {
|
||||
my ($pc, $mod, $result) = @_;
|
||||
if (defined($inlineCache{$pc})) {
|
||||
$inlineCache{$pc}{$mod} = $result;
|
||||
} else {
|
||||
$inlineCache{$pc} = {$mod => $result};
|
||||
}
|
||||
}
|
||||
|
||||
# for the --inline option
|
||||
sub inline {
|
||||
my ($pc, $rawfunc, $mod) = @_;
|
||||
|
||||
return $inlineCache{$pc}{$mod} if defined($inlineCache{$pc}{$mod});
|
||||
|
||||
# capture addr2line output
|
||||
my $a2l_output = `addr2line -a $pc -e $mod -i -f -s -C`;
|
||||
|
||||
# remove first line
|
||||
$a2l_output =~ s/^(.*\n){1}//;
|
||||
|
||||
if ($a2l_output =~ /\?\?\n\?\?:0/) {
|
||||
# if addr2line fails and rawfunc is func+offset, then fall back to it
|
||||
if ($rawfunc =~ /^(.+)\+0x([0-9a-f]+)$/) {
|
||||
my $func = $1;
|
||||
my $addr = hex $2;
|
||||
|
||||
$nmCache{$mod}=`nm $mod` unless defined $nmCache{$mod};
|
||||
|
||||
if ($nmCache{$mod} =~ /^([0-9a-f]+) . \Q$func\E$/m) {
|
||||
my $base = hex $1;
|
||||
my $newPc = sprintf "0x%x", $base+$addr;
|
||||
my $result = inline($newPc, '', $mod);
|
||||
inlineCacheAdd($pc, $mod, $result);
|
||||
return $result;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
my @fullfunc;
|
||||
my $one_item = "";
|
||||
for (split /^/, $a2l_output) {
|
||||
chomp $_;
|
||||
|
||||
# remove discriminator info if exists
|
||||
$_ =~ s/ \(discriminator \S+\)//;
|
||||
|
||||
if ($one_item eq "") {
|
||||
$one_item = $_;
|
||||
} else {
|
||||
if ($show_context == 1) {
|
||||
unshift @fullfunc, $one_item . ":$_";
|
||||
} else {
|
||||
unshift @fullfunc, $one_item;
|
||||
}
|
||||
$one_item = "";
|
||||
}
|
||||
}
|
||||
|
||||
my $result = join ";" , @fullfunc;
|
||||
|
||||
inlineCacheAdd($pc, $mod, $result);
|
||||
|
||||
return $result;
|
||||
}
|
||||
|
||||
my @stack;
|
||||
my $pname;
|
||||
my $m_pid;
|
||||
my $m_tid;
|
||||
my $m_period;
|
||||
|
||||
#
|
||||
# Main loop
|
||||
#
|
||||
while (defined($_ = <>)) {
|
||||
|
||||
# find the name of the process launched by perf, by stepping backwards
|
||||
# over the args to find the first non-option (no dash):
|
||||
if (/^# cmdline/) {
|
||||
my @args = split ' ', $_;
|
||||
foreach my $arg (reverse @args) {
|
||||
if ($arg !~ /^-/) {
|
||||
$target_pname = $arg;
|
||||
$target_pname =~ s:.*/::; # strip pathname
|
||||
last;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
# skip remaining comments
|
||||
next if m/^#/;
|
||||
chomp;
|
||||
|
||||
# end of stack. save cached data.
|
||||
if (m/^$/) {
|
||||
# ignore filtered samples
|
||||
next if not $pname;
|
||||
|
||||
if ($include_pname) {
|
||||
if (defined $pname) {
|
||||
unshift @stack, $pname;
|
||||
} else {
|
||||
unshift @stack, "";
|
||||
}
|
||||
}
|
||||
remember_stack(join(";", @stack), $m_period) if @stack;
|
||||
undef @stack;
|
||||
undef $pname;
|
||||
next;
|
||||
}
|
||||
|
||||
#
|
||||
# event record start
|
||||
#
|
||||
if (/^(\S.+?)\s+(\d+)\/*(\d+)*\s+/) {
|
||||
# default "perf script" output has TID but not PID
|
||||
# eg, "java 25607 4794564.109216: 1 cycles:"
|
||||
# eg, "java 12688 [002] 6544038.708352: 235 cpu-clock:"
|
||||
# eg, "V8 WorkerThread 25607 4794564.109216: 104345 cycles:"
|
||||
# eg, "java 24636/25607 [000] 4794564.109216: 1 cycles:"
|
||||
# eg, "java 12688/12764 6544038.708352: 10309278 cpu-clock:"
|
||||
# eg, "V8 WorkerThread 24636/25607 [000] 94564.109216: 100 cycles:"
|
||||
# other combinations possible
|
||||
my ($comm, $pid, $tid, $period) = ($1, $2, $3, "");
|
||||
if (not $tid) {
|
||||
$tid = $pid;
|
||||
$pid = "?";
|
||||
}
|
||||
|
||||
if (/:\s*(\d+)*\s+(\S+):\s*$/) {
|
||||
$period = $1;
|
||||
my $event = $2;
|
||||
|
||||
if ($event_filter eq "") {
|
||||
# By default only show events of the first encountered
|
||||
# event type. Merging together different types, such as
|
||||
# instructions and cycles, produces misleading results.
|
||||
$event_filter = $event;
|
||||
$event_defaulted = 1;
|
||||
} elsif ($event ne $event_filter) {
|
||||
if ($event_defaulted and $event_warning == 0) {
|
||||
# only print this warning if necessary:
|
||||
# when we defaulted and there was
|
||||
# multiple event types.
|
||||
print STDERR "Filtering for events of type: $event\n";
|
||||
$event_warning = 1;
|
||||
}
|
||||
next;
|
||||
}
|
||||
}
|
||||
|
||||
if (not $period) {
|
||||
$period = 1
|
||||
}
|
||||
($m_pid, $m_tid, $m_period) = ($pid, $tid, $period);
|
||||
|
||||
if ($include_tid) {
|
||||
$pname = "$comm-$m_pid/$m_tid";
|
||||
} elsif ($include_pid) {
|
||||
$pname = "$comm-$m_pid";
|
||||
} else {
|
||||
$pname = "$comm";
|
||||
}
|
||||
$pname =~ tr/ /_/;
|
||||
|
||||
#
|
||||
# stack line
|
||||
#
|
||||
} elsif (/^\s*(\w+)\s*(.+) \((.*)\)/) {
|
||||
# ignore filtered samples
|
||||
next if not $pname;
|
||||
|
||||
my ($pc, $rawfunc, $mod) = ($1, $2, $3);
|
||||
|
||||
if ($show_inline == 1 && $mod !~ m/(perf-\d+.map|kernel\.|\[[^\]]+\])/) {
|
||||
my $inlineRes = inline($pc, $rawfunc, $mod);
|
||||
# - empty result this happens e.g., when $mod does not exist or is a path to a compressed kernel module
|
||||
# if this happens, the user will see error message from addr2line written to stderr
|
||||
# - if addr2line results in "??" , then it's much more sane to fall back than produce a '??' in graph
|
||||
if($inlineRes ne "" and $inlineRes ne "??" and $inlineRes ne "??:??:0" ) {
|
||||
unshift @stack, $inlineRes;
|
||||
next;
|
||||
}
|
||||
}
|
||||
|
||||
# Linux 4.8 included symbol offsets in perf script output by default, eg:
|
||||
# 7fffb84c9afc cpu_startup_entry+0x800047c022ec ([kernel.kallsyms])
|
||||
# strip these off:
|
||||
$rawfunc =~ s/\+0x[\da-f]+$//;
|
||||
|
||||
next if $rawfunc =~ /^\(/; # skip process names
|
||||
|
||||
my $is_unknown=0;
|
||||
my @inline;
|
||||
for (split /\->/, $rawfunc) {
|
||||
my $func = $_;
|
||||
|
||||
if ($func eq "[unknown]") {
|
||||
if ($mod ne "[unknown]") { # use module name instead, if known
|
||||
$func = $mod;
|
||||
$func =~ s/.*\///;
|
||||
} else {
|
||||
$func = "unknown";
|
||||
$is_unknown=1;
|
||||
}
|
||||
|
||||
if ($include_addrs) {
|
||||
$func = "\[$func \<$pc\>\]";
|
||||
} else {
|
||||
$func = "\[$func\]";
|
||||
}
|
||||
}
|
||||
|
||||
if ($tidy_generic) {
|
||||
$func =~ s/;/:/g;
|
||||
if ($func !~ m/\.\(.*\)\./) {
|
||||
# This doesn't look like a Go method name (such as
|
||||
# "net/http.(*Client).Do"), so everything after the first open
|
||||
# paren (that is not part of an "(anonymous namespace)") is
|
||||
# just noise.
|
||||
$func =~ s/\((?!anonymous namespace\)).*//;
|
||||
}
|
||||
# now tidy this horrible thing:
|
||||
# 13a80b608e0a RegExp:[&<>\"\'] (/tmp/perf-7539.map)
|
||||
$func =~ tr/"\'//d;
|
||||
# fall through to $tidy_java
|
||||
}
|
||||
|
||||
if ($tidy_java and $pname =~ m/^java/) {
|
||||
# along with $tidy_generic, converts the following:
|
||||
# Lorg/mozilla/javascript/ContextFactory;.call(Lorg/mozilla/javascript/ContextAction;)Ljava/lang/Object;
|
||||
# Lorg/mozilla/javascript/ContextFactory;.call(Lorg/mozilla/javascript/C
|
||||
# Lorg/mozilla/javascript/MemberBox;.<init>(Ljava/lang/reflect/Method;)V
|
||||
# into:
|
||||
# org/mozilla/javascript/ContextFactory:.call
|
||||
# org/mozilla/javascript/ContextFactory:.call
|
||||
# org/mozilla/javascript/MemberBox:.init
|
||||
$func =~ s/^L// if $func =~ m:/:;
|
||||
}
|
||||
|
||||
#
|
||||
# Annotations
|
||||
#
|
||||
# detect inlined from the @inline array
|
||||
# detect kernel from the module name; eg, frames to parse include:
|
||||
# ffffffff8103ce3b native_safe_halt ([kernel.kallsyms])
|
||||
# 8c3453 tcp_sendmsg (/lib/modules/4.3.0-rc1-virtual/build/vmlinux)
|
||||
# 7d8 ipv4_conntrack_local+0x7f8f80b8 ([nf_conntrack_ipv4])
|
||||
# detect jit from the module name; eg:
|
||||
# 7f722d142778 Ljava/io/PrintStream;::print (/tmp/perf-19982.map)
|
||||
if (scalar(@inline) > 0) {
|
||||
$func .= "_[i]" unless $func =~ m/\_\[i\]/; # inlined
|
||||
} elsif ($annotate_kernel == 1 && $mod =~ m/(^\[|vmlinux$)/ && $mod !~ /unknown/) {
|
||||
$func .= "_[k]"; # kernel
|
||||
} elsif ($annotate_jit == 1 && $mod =~ m:/tmp/perf-\d+\.map:) {
|
||||
$func .= "_[j]" unless $func =~ m/\_\[j\]/; # jitted
|
||||
}
|
||||
|
||||
#
|
||||
# Source lines
|
||||
#
|
||||
#
|
||||
# Sample outputs:
|
||||
# | a.out 35081 252436.005167: 667783 cycles:
|
||||
# | 408ebb some_method_name+0x8b (/full/path/to/a.out)
|
||||
# | uniform_int_dist.h:300
|
||||
# | 4069f5 main+0x935 (/full/path/to/a.out)
|
||||
# | file.cpp:137
|
||||
# | 7f6d2148eb25 __libc_start_main+0xd5 (/lib64/libc-2.33.so)
|
||||
# | libc-2.33.so[27b25]
|
||||
#
|
||||
# | a.out 35081 252435.738165: 306459 cycles:
|
||||
# | 7f6d213c2750 [unknown] (/usr/lib64/libkmod.so.2.3.6)
|
||||
# | libkmod.so.2.3.6[6750]
|
||||
#
|
||||
# | a.out 35081 252435.738373: 315813 cycles:
|
||||
# | 7f6d215ca51b __strlen_avx2+0x4b (/lib64/libc-2.33.so)
|
||||
# | libc-2.33.so[16351b]
|
||||
# | 7ffc71ee9580 [unknown] ([unknown])
|
||||
# |
|
||||
#
|
||||
# | a.out 35081 252435.718940: 247984 cycles:
|
||||
# | ffffffff814f9302 up_write+0x32 ([kernel.kallsyms])
|
||||
# | [kernel.kallsyms][ffffffff814f9302]
|
||||
if($srcline_in_input and not $is_unknown){
|
||||
$_ = <>;
|
||||
chomp;
|
||||
s/\[.*?\]//g;
|
||||
s/^\s*//g;
|
||||
s/\s*$//g;
|
||||
$func.=':'.$_ unless $_ eq "";
|
||||
}
|
||||
|
||||
push @inline, $func;
|
||||
}
|
||||
|
||||
unshift @stack, @inline;
|
||||
} else {
|
||||
warn "Unrecognized line: $_";
|
||||
}
|
||||
}
|
||||
|
||||
foreach my $k (sort { $a cmp $b } keys %collapsed) {
|
||||
print "$k $collapsed{$k}\n";
|
||||
}
|
||||
862
stackflame.svg
862
stackflame.svg
@@ -1,862 +0,0 @@
|
||||
<?xml version="1.0" standalone="no"?>
|
||||
<!DOCTYPE svg PUBLIC "-//W3C//DTD SVG 1.1//EN" "http://www.w3.org/Graphics/SVG/1.1/DTD/svg11.dtd">
|
||||
<svg version="1.1" width="1200" height="390" onload="init(evt)" viewBox="0 0 1200 390" xmlns="http://www.w3.org/2000/svg" xmlns:xlink="http://www.w3.org/1999/xlink">
|
||||
<!-- Flame graph stack visualization. See https://github.com/brendangregg/FlameGraph for latest version, and http://www.brendangregg.com/flamegraphs.html for examples. -->
|
||||
<!-- NOTES: -->
|
||||
<defs>
|
||||
<linearGradient id="background" y1="0" y2="1" x1="0" x2="0" >
|
||||
<stop stop-color="#eeeeee" offset="5%" />
|
||||
<stop stop-color="#eeeeb0" offset="95%" />
|
||||
</linearGradient>
|
||||
</defs>
|
||||
<style type="text/css">
|
||||
text { font-family:Verdana; font-size:12px; fill:rgb(0,0,0); }
|
||||
#search, #ignorecase { opacity:0.1; cursor:pointer; }
|
||||
#search:hover, #search.show, #ignorecase:hover, #ignorecase.show { opacity:1; }
|
||||
#subtitle { text-anchor:middle; font-color:rgb(160,160,160); }
|
||||
#title { text-anchor:middle; font-size:17px}
|
||||
#unzoom { cursor:pointer; }
|
||||
#frames > *:hover { stroke:black; stroke-width:0.5; cursor:pointer; }
|
||||
.hide { display:none; }
|
||||
.parent { opacity:0.5; }
|
||||
</style>
|
||||
<script type="text/ecmascript">
|
||||
<![CDATA[
|
||||
"use strict";
|
||||
var details, searchbtn, unzoombtn, matchedtxt, svg, searching, currentSearchTerm, ignorecase, ignorecaseBtn;
|
||||
function init(evt) {
|
||||
details = document.getElementById("details").firstChild;
|
||||
searchbtn = document.getElementById("search");
|
||||
ignorecaseBtn = document.getElementById("ignorecase");
|
||||
unzoombtn = document.getElementById("unzoom");
|
||||
matchedtxt = document.getElementById("matched");
|
||||
svg = document.getElementsByTagName("svg")[0];
|
||||
searching = 0;
|
||||
currentSearchTerm = null;
|
||||
|
||||
// use GET parameters to restore a flamegraphs state.
|
||||
var params = get_params();
|
||||
if (params.x && params.y)
|
||||
zoom(find_group(document.querySelector('[x="' + params.x + '"][y="' + params.y + '"]')));
|
||||
if (params.s) search(params.s);
|
||||
}
|
||||
|
||||
// event listeners
|
||||
window.addEventListener("click", function(e) {
|
||||
var target = find_group(e.target);
|
||||
if (target) {
|
||||
if (target.nodeName == "a") {
|
||||
if (e.ctrlKey === false) return;
|
||||
e.preventDefault();
|
||||
}
|
||||
if (target.classList.contains("parent")) unzoom(true);
|
||||
zoom(target);
|
||||
if (!document.querySelector('.parent')) {
|
||||
// we have basically done a clearzoom so clear the url
|
||||
var params = get_params();
|
||||
if (params.x) delete params.x;
|
||||
if (params.y) delete params.y;
|
||||
history.replaceState(null, null, parse_params(params));
|
||||
unzoombtn.classList.add("hide");
|
||||
return;
|
||||
}
|
||||
|
||||
// set parameters for zoom state
|
||||
var el = target.querySelector("rect");
|
||||
if (el && el.attributes && el.attributes.y && el.attributes._orig_x) {
|
||||
var params = get_params()
|
||||
params.x = el.attributes._orig_x.value;
|
||||
params.y = el.attributes.y.value;
|
||||
history.replaceState(null, null, parse_params(params));
|
||||
}
|
||||
}
|
||||
else if (e.target.id == "unzoom") clearzoom();
|
||||
else if (e.target.id == "search") search_prompt();
|
||||
else if (e.target.id == "ignorecase") toggle_ignorecase();
|
||||
}, false)
|
||||
|
||||
// mouse-over for info
|
||||
// show
|
||||
window.addEventListener("mouseover", function(e) {
|
||||
var target = find_group(e.target);
|
||||
if (target) details.nodeValue = "Function: " + g_to_text(target);
|
||||
}, false)
|
||||
|
||||
// clear
|
||||
window.addEventListener("mouseout", function(e) {
|
||||
var target = find_group(e.target);
|
||||
if (target) details.nodeValue = ' ';
|
||||
}, false)
|
||||
|
||||
// ctrl-F for search
|
||||
// ctrl-I to toggle case-sensitive search
|
||||
window.addEventListener("keydown",function (e) {
|
||||
if (e.keyCode === 114 || (e.ctrlKey && e.keyCode === 70)) {
|
||||
e.preventDefault();
|
||||
search_prompt();
|
||||
}
|
||||
else if (e.ctrlKey && e.keyCode === 73) {
|
||||
e.preventDefault();
|
||||
toggle_ignorecase();
|
||||
}
|
||||
}, false)
|
||||
|
||||
// functions
|
||||
function get_params() {
|
||||
var params = {};
|
||||
var paramsarr = window.location.search.substr(1).split('&');
|
||||
for (var i = 0; i < paramsarr.length; ++i) {
|
||||
var tmp = paramsarr[i].split("=");
|
||||
if (!tmp[0] || !tmp[1]) continue;
|
||||
params[tmp[0]] = decodeURIComponent(tmp[1]);
|
||||
}
|
||||
return params;
|
||||
}
|
||||
function parse_params(params) {
|
||||
var uri = "?";
|
||||
for (var key in params) {
|
||||
uri += key + '=' + encodeURIComponent(params[key]) + '&';
|
||||
}
|
||||
if (uri.slice(-1) == "&")
|
||||
uri = uri.substring(0, uri.length - 1);
|
||||
if (uri == '?')
|
||||
uri = window.location.href.split('?')[0];
|
||||
return uri;
|
||||
}
|
||||
function find_child(node, selector) {
|
||||
var children = node.querySelectorAll(selector);
|
||||
if (children.length) return children[0];
|
||||
}
|
||||
function find_group(node) {
|
||||
var parent = node.parentElement;
|
||||
if (!parent) return;
|
||||
if (parent.id == "frames") return node;
|
||||
return find_group(parent);
|
||||
}
|
||||
function orig_save(e, attr, val) {
|
||||
if (e.attributes["_orig_" + attr] != undefined) return;
|
||||
if (e.attributes[attr] == undefined) return;
|
||||
if (val == undefined) val = e.attributes[attr].value;
|
||||
e.setAttribute("_orig_" + attr, val);
|
||||
}
|
||||
function orig_load(e, attr) {
|
||||
if (e.attributes["_orig_"+attr] == undefined) return;
|
||||
e.attributes[attr].value = e.attributes["_orig_" + attr].value;
|
||||
e.removeAttribute("_orig_"+attr);
|
||||
}
|
||||
function g_to_text(e) {
|
||||
var text = find_child(e, "title").firstChild.nodeValue;
|
||||
return (text)
|
||||
}
|
||||
function g_to_func(e) {
|
||||
var func = g_to_text(e);
|
||||
// if there's any manipulation we want to do to the function
|
||||
// name before it's searched, do it here before returning.
|
||||
return (func);
|
||||
}
|
||||
function update_text(e) {
|
||||
var r = find_child(e, "rect");
|
||||
var t = find_child(e, "text");
|
||||
var w = parseFloat(r.attributes.width.value) -3;
|
||||
var txt = find_child(e, "title").textContent.replace(/\([^(]*\)$/,"");
|
||||
t.attributes.x.value = parseFloat(r.attributes.x.value) + 3;
|
||||
|
||||
// Smaller than this size won't fit anything
|
||||
if (w < 2 * 12 * 0.59) {
|
||||
t.textContent = "";
|
||||
return;
|
||||
}
|
||||
|
||||
t.textContent = txt;
|
||||
var sl = t.getSubStringLength(0, txt.length);
|
||||
// check if only whitespace or if we can fit the entire string into width w
|
||||
if (/^ *$/.test(txt) || sl < w)
|
||||
return;
|
||||
|
||||
// this isn't perfect, but gives a good starting point
|
||||
// and avoids calling getSubStringLength too often
|
||||
var start = Math.floor((w/sl) * txt.length);
|
||||
for (var x = start; x > 0; x = x-2) {
|
||||
if (t.getSubStringLength(0, x + 2) <= w) {
|
||||
t.textContent = txt.substring(0, x) + "..";
|
||||
return;
|
||||
}
|
||||
}
|
||||
t.textContent = "";
|
||||
}
|
||||
|
||||
// zoom
|
||||
function zoom_reset(e) {
|
||||
if (e.attributes != undefined) {
|
||||
orig_load(e, "x");
|
||||
orig_load(e, "width");
|
||||
}
|
||||
if (e.childNodes == undefined) return;
|
||||
for (var i = 0, c = e.childNodes; i < c.length; i++) {
|
||||
zoom_reset(c[i]);
|
||||
}
|
||||
}
|
||||
function zoom_child(e, x, ratio) {
|
||||
if (e.attributes != undefined) {
|
||||
if (e.attributes.x != undefined) {
|
||||
orig_save(e, "x");
|
||||
e.attributes.x.value = (parseFloat(e.attributes.x.value) - x - 10) * ratio + 10;
|
||||
if (e.tagName == "text")
|
||||
e.attributes.x.value = find_child(e.parentNode, "rect[x]").attributes.x.value + 3;
|
||||
}
|
||||
if (e.attributes.width != undefined) {
|
||||
orig_save(e, "width");
|
||||
e.attributes.width.value = parseFloat(e.attributes.width.value) * ratio;
|
||||
}
|
||||
}
|
||||
|
||||
if (e.childNodes == undefined) return;
|
||||
for (var i = 0, c = e.childNodes; i < c.length; i++) {
|
||||
zoom_child(c[i], x - 10, ratio);
|
||||
}
|
||||
}
|
||||
function zoom_parent(e) {
|
||||
if (e.attributes) {
|
||||
if (e.attributes.x != undefined) {
|
||||
orig_save(e, "x");
|
||||
e.attributes.x.value = 10;
|
||||
}
|
||||
if (e.attributes.width != undefined) {
|
||||
orig_save(e, "width");
|
||||
e.attributes.width.value = parseInt(svg.width.baseVal.value) - (10 * 2);
|
||||
}
|
||||
}
|
||||
if (e.childNodes == undefined) return;
|
||||
for (var i = 0, c = e.childNodes; i < c.length; i++) {
|
||||
zoom_parent(c[i]);
|
||||
}
|
||||
}
|
||||
function zoom(node) {
|
||||
var attr = find_child(node, "rect").attributes;
|
||||
var width = parseFloat(attr.width.value);
|
||||
var xmin = parseFloat(attr.x.value);
|
||||
var xmax = parseFloat(xmin + width);
|
||||
var ymin = parseFloat(attr.y.value);
|
||||
var ratio = (svg.width.baseVal.value - 2 * 10) / width;
|
||||
|
||||
// XXX: Workaround for JavaScript float issues (fix me)
|
||||
var fudge = 0.0001;
|
||||
|
||||
unzoombtn.classList.remove("hide");
|
||||
|
||||
var el = document.getElementById("frames").children;
|
||||
for (var i = 0; i < el.length; i++) {
|
||||
var e = el[i];
|
||||
var a = find_child(e, "rect").attributes;
|
||||
var ex = parseFloat(a.x.value);
|
||||
var ew = parseFloat(a.width.value);
|
||||
var upstack;
|
||||
// Is it an ancestor
|
||||
if (0 == 0) {
|
||||
upstack = parseFloat(a.y.value) > ymin;
|
||||
} else {
|
||||
upstack = parseFloat(a.y.value) < ymin;
|
||||
}
|
||||
if (upstack) {
|
||||
// Direct ancestor
|
||||
if (ex <= xmin && (ex+ew+fudge) >= xmax) {
|
||||
e.classList.add("parent");
|
||||
zoom_parent(e);
|
||||
update_text(e);
|
||||
}
|
||||
// not in current path
|
||||
else
|
||||
e.classList.add("hide");
|
||||
}
|
||||
// Children maybe
|
||||
else {
|
||||
// no common path
|
||||
if (ex < xmin || ex + fudge >= xmax) {
|
||||
e.classList.add("hide");
|
||||
}
|
||||
else {
|
||||
zoom_child(e, xmin, ratio);
|
||||
update_text(e);
|
||||
}
|
||||
}
|
||||
}
|
||||
search();
|
||||
}
|
||||
function unzoom(dont_update_text) {
|
||||
unzoombtn.classList.add("hide");
|
||||
var el = document.getElementById("frames").children;
|
||||
for(var i = 0; i < el.length; i++) {
|
||||
el[i].classList.remove("parent");
|
||||
el[i].classList.remove("hide");
|
||||
zoom_reset(el[i]);
|
||||
if(!dont_update_text) update_text(el[i]);
|
||||
}
|
||||
search();
|
||||
}
|
||||
function clearzoom() {
|
||||
unzoom();
|
||||
|
||||
// remove zoom state
|
||||
var params = get_params();
|
||||
if (params.x) delete params.x;
|
||||
if (params.y) delete params.y;
|
||||
history.replaceState(null, null, parse_params(params));
|
||||
}
|
||||
|
||||
// search
|
||||
function toggle_ignorecase() {
|
||||
ignorecase = !ignorecase;
|
||||
if (ignorecase) {
|
||||
ignorecaseBtn.classList.add("show");
|
||||
} else {
|
||||
ignorecaseBtn.classList.remove("show");
|
||||
}
|
||||
reset_search();
|
||||
search();
|
||||
}
|
||||
function reset_search() {
|
||||
var el = document.querySelectorAll("#frames rect");
|
||||
for (var i = 0; i < el.length; i++) {
|
||||
orig_load(el[i], "fill")
|
||||
}
|
||||
var params = get_params();
|
||||
delete params.s;
|
||||
history.replaceState(null, null, parse_params(params));
|
||||
}
|
||||
function search_prompt() {
|
||||
if (!searching) {
|
||||
var term = prompt("Enter a search term (regexp " +
|
||||
"allowed, eg: ^ext4_)"
|
||||
+ (ignorecase ? ", ignoring case" : "")
|
||||
+ "\nPress Ctrl-i to toggle case sensitivity", "");
|
||||
if (term != null) search(term);
|
||||
} else {
|
||||
reset_search();
|
||||
searching = 0;
|
||||
currentSearchTerm = null;
|
||||
searchbtn.classList.remove("show");
|
||||
searchbtn.firstChild.nodeValue = "Search"
|
||||
matchedtxt.classList.add("hide");
|
||||
matchedtxt.firstChild.nodeValue = ""
|
||||
}
|
||||
}
|
||||
function search(term) {
|
||||
if (term) currentSearchTerm = term;
|
||||
if (currentSearchTerm === null) return;
|
||||
|
||||
var re = new RegExp(currentSearchTerm, ignorecase ? 'i' : '');
|
||||
var el = document.getElementById("frames").children;
|
||||
var matches = new Object();
|
||||
var maxwidth = 0;
|
||||
for (var i = 0; i < el.length; i++) {
|
||||
var e = el[i];
|
||||
var func = g_to_func(e);
|
||||
var rect = find_child(e, "rect");
|
||||
if (func == null || rect == null)
|
||||
continue;
|
||||
|
||||
// Save max width. Only works as we have a root frame
|
||||
var w = parseFloat(rect.attributes.width.value);
|
||||
if (w > maxwidth)
|
||||
maxwidth = w;
|
||||
|
||||
if (func.match(re)) {
|
||||
// highlight
|
||||
var x = parseFloat(rect.attributes.x.value);
|
||||
orig_save(rect, "fill");
|
||||
rect.attributes.fill.value = "rgb(230,0,230)";
|
||||
|
||||
// remember matches
|
||||
if (matches[x] == undefined) {
|
||||
matches[x] = w;
|
||||
} else {
|
||||
if (w > matches[x]) {
|
||||
// overwrite with parent
|
||||
matches[x] = w;
|
||||
}
|
||||
}
|
||||
searching = 1;
|
||||
}
|
||||
}
|
||||
if (!searching)
|
||||
return;
|
||||
var params = get_params();
|
||||
params.s = currentSearchTerm;
|
||||
history.replaceState(null, null, parse_params(params));
|
||||
|
||||
searchbtn.classList.add("show");
|
||||
searchbtn.firstChild.nodeValue = "Reset Search";
|
||||
|
||||
// calculate percent matched, excluding vertical overlap
|
||||
var count = 0;
|
||||
var lastx = -1;
|
||||
var lastw = 0;
|
||||
var keys = Array();
|
||||
for (k in matches) {
|
||||
if (matches.hasOwnProperty(k))
|
||||
keys.push(k);
|
||||
}
|
||||
// sort the matched frames by their x location
|
||||
// ascending, then width descending
|
||||
keys.sort(function(a, b){
|
||||
return a - b;
|
||||
});
|
||||
// Step through frames saving only the biggest bottom-up frames
|
||||
// thanks to the sort order. This relies on the tree property
|
||||
// where children are always smaller than their parents.
|
||||
var fudge = 0.0001; // JavaScript floating point
|
||||
for (var k in keys) {
|
||||
var x = parseFloat(keys[k]);
|
||||
var w = matches[keys[k]];
|
||||
if (x >= lastx + lastw - fudge) {
|
||||
count += w;
|
||||
lastx = x;
|
||||
lastw = w;
|
||||
}
|
||||
}
|
||||
// display matched percent
|
||||
matchedtxt.classList.remove("hide");
|
||||
var pct = 100 * count / maxwidth;
|
||||
if (pct != 100) pct = pct.toFixed(1)
|
||||
matchedtxt.firstChild.nodeValue = "Matched: " + pct + "%";
|
||||
}
|
||||
]]>
|
||||
</script>
|
||||
<rect x="0.0" y="0" width="1200.0" height="390.0" fill="url(#background)" />
|
||||
<text id="title" x="600.00" y="24" >Flame Graph</text>
|
||||
<text id="details" x="10.00" y="373" > </text>
|
||||
<text id="unzoom" x="10.00" y="24" class="hide">Reset Zoom</text>
|
||||
<text id="search" x="1090.00" y="24" >Search</text>
|
||||
<text id="ignorecase" x="1174.00" y="24" >ic</text>
|
||||
<text id="matched" x="1090.00" y="373" > </text>
|
||||
<g id="frames">
|
||||
<g >
|
||||
<title>_IO_new_file_underflow (2 samples, 0.52%)</title><rect x="1122.4" y="101" width="6.1" height="15.0" fill="rgb(214,42,10)" rx="2" ry="2" />
|
||||
<text x="1125.40" y="111.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>dpp::ssl_client::read_loop() (4 samples, 1.04%)</title><rect x="919.6" y="213" width="12.3" height="15.0" fill="rgb(243,175,41)" rx="2" ry="2" />
|
||||
<text x="922.58" y="223.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>lll_mutex_lock_optimized (1 samples, 0.26%)</title><rect x="117.6" y="213" width="3.0" height="15.0" fill="rgb(209,22,5)" rx="2" ry="2" />
|
||||
<text x="120.55" y="223.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>dpp::https_client::connect() (4 samples, 1.04%)</title><rect x="919.6" y="229" width="12.3" height="15.0" fill="rgb(251,211,50)" rx="2" ry="2" />
|
||||
<text x="922.58" y="239.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>___pthread_cond_wait (20 samples, 5.21%)</title><rect x="1128.5" y="277" width="61.5" height="15.0" fill="rgb(252,219,52)" rx="2" ry="2" />
|
||||
<text x="1131.54" y="287.5" >___pth..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>___pthread_mutex_lock (1 samples, 0.26%)</title><rect x="117.6" y="229" width="3.0" height="15.0" fill="rgb(206,7,1)" rx="2" ry="2" />
|
||||
<text x="120.55" y="239.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>all (384 samples, 100%)</title><rect x="10.0" y="341" width="1180.0" height="15.0" fill="rgb(213,39,9)" rx="2" ry="2" />
|
||||
<text x="13.00" y="351.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__GI___underflow (2 samples, 0.52%)</title><rect x="1122.4" y="117" width="6.1" height="15.0" fill="rgb(221,74,17)" rx="2" ry="2" />
|
||||
<text x="1125.40" y="127.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>nlohmann::json_abi_v3_11_2::detail::parser<nlohmann::json_abi_v3_11_2::basic_json<std::map, (1 samples, 0.26%)</title><rect x="1079.4" y="149" width="3.0" height="15.0" fill="rgb(217,59,14)" rx="2" ry="2" />
|
||||
<text x="1082.38" y="159.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__futex_abstimed_wait_common (256 samples, 66.67%)</title><rect x="132.9" y="197" width="786.7" height="15.0" fill="rgb(205,0,0)" rx="2" ry="2" />
|
||||
<text x="135.92" y="207.5" >__futex_abstimed_wait_common</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>___pthread_cond_clockwait64 (40 samples, 10.42%)</title><rect x="931.9" y="261" width="122.9" height="15.0" fill="rgb(233,129,30)" rx="2" ry="2" />
|
||||
<text x="934.88" y="271.5" >___pthread_cond..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>?? (2 samples, 0.52%)</title><rect x="1122.4" y="165" width="6.1" height="15.0" fill="rgb(249,205,49)" rx="2" ry="2" />
|
||||
<text x="1125.40" y="175.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__GI___libc_free (1 samples, 0.26%)</title><rect x="1054.8" y="133" width="3.1" height="15.0" fill="rgb(252,218,52)" rx="2" ry="2" />
|
||||
<text x="1057.79" y="143.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>_IO_new_file_underflow (3 samples, 0.78%)</title><rect x="1057.9" y="117" width="9.2" height="15.0" fill="rgb(214,42,10)" rx="2" ry="2" />
|
||||
<text x="1060.86" y="127.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>std::thread::_Invoker<std::tuple<bumbleBee::MusicPlayManager::play(dpp::discord_voice_client*)::<lambda(dpp::discord_voice_client*)>, (13 samples, 3.39%)</title><rect x="1082.4" y="261" width="40.0" height="15.0" fill="rgb(208,13,3)" rx="2" ry="2" />
|
||||
<text x="1085.45" y="271.5" >std..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__GI__IO_fread (3 samples, 0.78%)</title><rect x="1057.9" y="165" width="9.2" height="15.0" fill="rgb(212,34,8)" rx="2" ry="2" />
|
||||
<text x="1060.86" y="175.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>bumbleBee::MusicPlayManager::remove (1 samples, 0.26%)</title><rect x="1054.8" y="165" width="3.1" height="15.0" fill="rgb(230,117,28)" rx="2" ry="2" />
|
||||
<text x="1057.79" y="175.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__GI__IO_default_xsgetn (3 samples, 0.78%)</title><rect x="1057.9" y="149" width="9.2" height="15.0" fill="rgb(236,145,34)" rx="2" ry="2" />
|
||||
<text x="1060.86" y="159.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>___pthread_cond_clockwait64 (40 samples, 10.42%)</title><rect x="931.9" y="245" width="122.9" height="15.0" fill="rgb(233,129,30)" rx="2" ry="2" />
|
||||
<text x="934.88" y="255.5" >___pthread_cond..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__GI__IO_getline (4 samples, 1.04%)</title><rect x="1067.1" y="133" width="12.3" height="15.0" fill="rgb(226,100,23)" rx="2" ry="2" />
|
||||
<text x="1070.08" y="143.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>_int_free (1 samples, 0.26%)</title><rect x="1054.8" y="117" width="3.1" height="15.0" fill="rgb(247,196,46)" rx="2" ry="2" />
|
||||
<text x="1057.79" y="127.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__GI___libc_read (2 samples, 0.52%)</title><rect x="1122.4" y="69" width="6.1" height="15.0" fill="rgb(241,166,39)" rx="2" ry="2" />
|
||||
<text x="1125.40" y="79.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>dpp::in_thread::in_loop(unsigned (260 samples, 67.71%)</title><rect x="132.9" y="277" width="799.0" height="15.0" fill="rgb(208,15,3)" rx="2" ry="2" />
|
||||
<text x="135.92" y="287.5" >dpp::in_thread::in_loop(unsigned</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>bumbleBee::BumbleBee::start (20 samples, 5.21%)</title><rect x="1128.5" y="309" width="61.5" height="15.0" fill="rgb(252,216,51)" rx="2" ry="2" />
|
||||
<text x="1131.54" y="319.5" >bumble..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__futex_abstimed_wait_common64 (13 samples, 3.39%)</title><rect x="1082.4" y="101" width="40.0" height="15.0" fill="rgb(220,71,17)" rx="2" ry="2" />
|
||||
<text x="1085.45" y="111.5" >__f..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>dpp::discord_voice_client::write_ready() (4 samples, 1.04%)</title><rect x="120.6" y="245" width="12.3" height="15.0" fill="rgb(245,186,44)" rx="2" ry="2" />
|
||||
<text x="123.62" y="255.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>operator() (2 samples, 0.52%)</title><rect x="1122.4" y="197" width="6.1" height="15.0" fill="rgb(231,121,29)" rx="2" ry="2" />
|
||||
<text x="1125.40" y="207.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__pthread_cond_wait_common (40 samples, 10.42%)</title><rect x="931.9" y="229" width="122.9" height="15.0" fill="rgb(210,24,5)" rx="2" ry="2" />
|
||||
<text x="934.88" y="239.5" >__pthread_cond_..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__GI__IO_default_xsgetn (2 samples, 0.52%)</title><rect x="1122.4" y="133" width="6.1" height="15.0" fill="rgb(236,145,34)" rx="2" ry="2" />
|
||||
<text x="1125.40" y="143.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>nlohmann::json_abi_v3_11_2::operator>> (1 samples, 0.26%)</title><rect x="1079.4" y="165" width="3.0" height="15.0" fill="rgb(210,23,5)" rx="2" ry="2" />
|
||||
<text x="1082.38" y="175.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__futex_abstimed_wait_common (40 samples, 10.42%)</title><rect x="931.9" y="197" width="122.9" height="15.0" fill="rgb(205,0,0)" rx="2" ry="2" />
|
||||
<text x="934.88" y="207.5" >__futex_abstime..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>___pthread_cond_wait (13 samples, 3.39%)</title><rect x="1082.4" y="165" width="40.0" height="15.0" fill="rgb(252,219,52)" rx="2" ry="2" />
|
||||
<text x="1085.45" y="175.5" >___..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>std::__invoke_impl<void, (13 samples, 3.39%)</title><rect x="1082.4" y="213" width="40.0" height="15.0" fill="rgb(244,179,42)" rx="2" ry="2" />
|
||||
<text x="1085.45" y="223.5" >std..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>main (20 samples, 5.21%)</title><rect x="1128.5" y="325" width="61.5" height="15.0" fill="rgb(243,179,42)" rx="2" ry="2" />
|
||||
<text x="1131.54" y="335.5" >main</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>std::thread::_Invoker<std::tuple<bumbleBee::BumbleBee::on_slashcommand(const (9 samples, 2.34%)</title><rect x="1054.8" y="245" width="27.6" height="15.0" fill="rgb(251,213,50)" rx="2" ry="2" />
|
||||
<text x="1057.79" y="255.5" >s..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>std::__invoke<bumbleBee::MusicPlayManager::send_audio_to_voice(std::shared_ptr<bumbleBee::MusicQueueElement>, (2 samples, 0.52%)</title><rect x="1122.4" y="229" width="6.1" height="15.0" fill="rgb(217,59,14)" rx="2" ry="2" />
|
||||
<text x="1125.40" y="239.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>dpp::request_queue::out_loop() (40 samples, 10.42%)</title><rect x="931.9" y="277" width="122.9" height="15.0" fill="rgb(227,105,25)" rx="2" ry="2" />
|
||||
<text x="934.88" y="287.5" >dpp::request_qu..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>std::thread::_Invoker<std::tuple<bumbleBee::MusicPlayManager::send_audio_to_voice(std::shared_ptr<bumbleBee::MusicQueueElement>, (2 samples, 0.52%)</title><rect x="1122.4" y="245" width="6.1" height="15.0" fill="rgb(216,50,12)" rx="2" ry="2" />
|
||||
<text x="1125.40" y="255.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>std::__invoke_impl<void, (9 samples, 2.34%)</title><rect x="1054.8" y="213" width="27.6" height="15.0" fill="rgb(244,179,42)" rx="2" ry="2" />
|
||||
<text x="1057.79" y="223.5" >s..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>std::thread::_Invoker<std::tuple<bumbleBee::BumbleBee::on_slashcommand(const (9 samples, 2.34%)</title><rect x="1054.8" y="261" width="27.6" height="15.0" fill="rgb(251,213,50)" rx="2" ry="2" />
|
||||
<text x="1057.79" y="271.5" >s..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>std::thread::_State_impl<std::thread::_Invoker<std::tuple<bumbleBee::MusicPlayManager::play(dpp::discord_voice_client*)::<lambda(dpp::discord_voice_client*)>, (13 samples, 3.39%)</title><rect x="1082.4" y="277" width="40.0" height="15.0" fill="rgb(205,4,1)" rx="2" ry="2" />
|
||||
<text x="1085.45" y="287.5" >std..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>bumbleBee::commands::Play::execute (8 samples, 2.08%)</title><rect x="1057.9" y="181" width="24.5" height="15.0" fill="rgb(217,56,13)" rx="2" ry="2" />
|
||||
<text x="1060.86" y="191.5" >b..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__GI___poll (20 samples, 5.21%)</title><rect x="10.0" y="245" width="61.5" height="15.0" fill="rgb(228,110,26)" rx="2" ry="2" />
|
||||
<text x="13.00" y="255.5" >__GI__..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>_IO_fgets (4 samples, 1.04%)</title><rect x="1067.1" y="149" width="12.3" height="15.0" fill="rgb(235,139,33)" rx="2" ry="2" />
|
||||
<text x="1070.08" y="159.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>std::__invoke<bumbleBee::MusicPlayManager::play(dpp::discord_voice_client*)::<lambda(dpp::discord_voice_client*)>, (13 samples, 3.39%)</title><rect x="1082.4" y="229" width="40.0" height="15.0" fill="rgb(209,22,5)" rx="2" ry="2" />
|
||||
<text x="1085.45" y="239.5" >std..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>std::thread::_State_impl<std::thread::_Invoker<std::tuple<bumbleBee::MusicPlayManager::send_audio_to_voice(std::shared_ptr<bumbleBee::MusicQueueElement>, (2 samples, 0.52%)</title><rect x="1122.4" y="277" width="6.1" height="15.0" fill="rgb(213,41,9)" rx="2" ry="2" />
|
||||
<text x="1125.40" y="287.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__GI___libc_read (2 samples, 0.52%)</title><rect x="1122.4" y="85" width="6.1" height="15.0" fill="rgb(241,166,39)" rx="2" ry="2" />
|
||||
<text x="1125.40" y="95.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>___pthread_cond_clockwait64 (256 samples, 66.67%)</title><rect x="132.9" y="245" width="786.7" height="15.0" fill="rgb(233,129,30)" rx="2" ry="2" />
|
||||
<text x="135.92" y="255.5" >___pthread_cond_clockwait64</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>operator() (9 samples, 2.34%)</title><rect x="1054.8" y="197" width="27.6" height="15.0" fill="rgb(231,121,29)" rx="2" ry="2" />
|
||||
<text x="1057.79" y="207.5" >o..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__GI___clock_nanosleep (1 samples, 0.26%)</title><rect x="120.6" y="213" width="3.1" height="15.0" fill="rgb(232,126,30)" rx="2" ry="2" />
|
||||
<text x="123.62" y="223.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__futex_abstimed_wait_common (13 samples, 3.39%)</title><rect x="1082.4" y="117" width="40.0" height="15.0" fill="rgb(205,0,0)" rx="2" ry="2" />
|
||||
<text x="1085.45" y="127.5" >__f..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__GI___poll (4 samples, 1.04%)</title><rect x="919.6" y="197" width="12.3" height="15.0" fill="rgb(228,110,26)" rx="2" ry="2" />
|
||||
<text x="922.58" y="207.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__pthread_cond_wait_common (20 samples, 5.21%)</title><rect x="1128.5" y="261" width="61.5" height="15.0" fill="rgb(210,24,5)" rx="2" ry="2" />
|
||||
<text x="1131.54" y="271.5" >__pthr..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>nlohmann::json_abi_v3_11_2::detail::lexer<nlohmann::json_abi_v3_11_2::basic_json<std::map, (1 samples, 0.26%)</title><rect x="1079.4" y="85" width="3.0" height="15.0" fill="rgb(222,78,18)" rx="2" ry="2" />
|
||||
<text x="1082.38" y="95.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>nlohmann::json_abi_v3_11_2::detail::parser<nlohmann::json_abi_v3_11_2::basic_json<std::map, (1 samples, 0.26%)</title><rect x="1079.4" y="133" width="3.0" height="15.0" fill="rgb(217,59,14)" rx="2" ry="2" />
|
||||
<text x="1082.38" y="143.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>dpp::discord_voice_client::thread_run() (20 samples, 5.21%)</title><rect x="71.5" y="277" width="61.4" height="15.0" fill="rgb(238,153,36)" rx="2" ry="2" />
|
||||
<text x="74.46" y="287.5" >dpp::d..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>dpp::discord_client::thread_run() (20 samples, 5.21%)</title><rect x="10.0" y="277" width="61.5" height="15.0" fill="rgb(251,214,51)" rx="2" ry="2" />
|
||||
<text x="13.00" y="287.5" >dpp::d..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>nlohmann::json_abi_v3_11_2::detail::lexer<nlohmann::json_abi_v3_11_2::basic_json<std::map, (1 samples, 0.26%)</title><rect x="1079.4" y="101" width="3.0" height="15.0" fill="rgb(222,78,18)" rx="2" ry="2" />
|
||||
<text x="1082.38" y="111.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__GI___poll (15 samples, 3.91%)</title><rect x="71.5" y="245" width="46.1" height="15.0" fill="rgb(228,110,26)" rx="2" ry="2" />
|
||||
<text x="74.46" y="255.5" >__GI..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__GI___underflow (3 samples, 0.78%)</title><rect x="1057.9" y="133" width="9.2" height="15.0" fill="rgb(221,74,17)" rx="2" ry="2" />
|
||||
<text x="1060.86" y="143.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>std::thread::_State_impl<std::thread::_Invoker<std::tuple<bumbleBee::BumbleBee::on_slashcommand(const (9 samples, 2.34%)</title><rect x="1054.8" y="277" width="27.6" height="15.0" fill="rgb(249,203,48)" rx="2" ry="2" />
|
||||
<text x="1057.79" y="287.5" >s..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__GI___futex_abstimed_wait_cancelable64 (13 samples, 3.39%)</title><rect x="1082.4" y="133" width="40.0" height="15.0" fill="rgb(215,48,11)" rx="2" ry="2" />
|
||||
<text x="1085.45" y="143.5" >__G..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>___pthread_cond_clockwait64 (256 samples, 66.67%)</title><rect x="132.9" y="261" width="786.7" height="15.0" fill="rgb(233,129,30)" rx="2" ry="2" />
|
||||
<text x="135.92" y="271.5" >___pthread_cond_clockwait64</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>dpp::https_client::https_client(std::__cxx11::basic_string<char, (4 samples, 1.04%)</title><rect x="919.6" y="245" width="12.3" height="15.0" fill="rgb(206,4,1)" rx="2" ry="2" />
|
||||
<text x="922.58" y="255.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>nlohmann::json_abi_v3_11_2::detail::lexer<nlohmann::json_abi_v3_11_2::basic_json<std::map, (1 samples, 0.26%)</title><rect x="1079.4" y="69" width="3.0" height="15.0" fill="rgb(222,78,18)" rx="2" ry="2" />
|
||||
<text x="1082.38" y="79.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>dpp::ssl_client::read_loop() (20 samples, 5.21%)</title><rect x="71.5" y="261" width="61.4" height="15.0" fill="rgb(243,175,41)" rx="2" ry="2" />
|
||||
<text x="74.46" y="271.5" >dpp::s..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__GI___futex_abstimed_wait_cancelable64 (40 samples, 10.42%)</title><rect x="931.9" y="213" width="122.9" height="15.0" fill="rgb(215,48,11)" rx="2" ry="2" />
|
||||
<text x="934.88" y="223.5" >__GI___futex_ab..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__futex_abstimed_wait_common64 (40 samples, 10.42%)</title><rect x="931.9" y="181" width="122.9" height="15.0" fill="rgb(220,71,17)" rx="2" ry="2" />
|
||||
<text x="934.88" y="191.5" >__futex_abstime..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>futex_wait (1 samples, 0.26%)</title><rect x="117.6" y="181" width="3.0" height="15.0" fill="rgb(235,138,33)" rx="2" ry="2" />
|
||||
<text x="120.55" y="191.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__pthread_cond_wait_common (256 samples, 66.67%)</title><rect x="132.9" y="229" width="786.7" height="15.0" fill="rgb(210,24,5)" rx="2" ry="2" />
|
||||
<text x="135.92" y="239.5" >__pthread_cond_wait_common</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>std::vector<char, (1 samples, 0.26%)</title><rect x="1079.4" y="53" width="3.0" height="15.0" fill="rgb(222,81,19)" rx="2" ry="2" />
|
||||
<text x="1082.38" y="63.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>std::condition_variable::wait<bumbleBee::MusicPlayManager::play(dpp::discord_voice_client*)::<lambda(dpp::discord_voice_client*)>::<lambda()> (13 samples, 3.39%)</title><rect x="1082.4" y="181" width="40.0" height="15.0" fill="rgb(242,172,41)" rx="2" ry="2" />
|
||||
<text x="1085.45" y="191.5" >std..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__GI___libc_read (3 samples, 0.78%)</title><rect x="1057.9" y="85" width="9.2" height="15.0" fill="rgb(241,166,39)" rx="2" ry="2" />
|
||||
<text x="1060.86" y="95.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__GI__IO_fread (2 samples, 0.52%)</title><rect x="1122.4" y="149" width="6.1" height="15.0" fill="rgb(212,34,8)" rx="2" ry="2" />
|
||||
<text x="1125.40" y="159.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>?? (364 samples, 94.79%)</title><rect x="10.0" y="293" width="1118.5" height="15.0" fill="rgb(249,205,49)" rx="2" ry="2" />
|
||||
<text x="13.00" y="303.5" >??</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__futex_abstimed_wait_common (20 samples, 5.21%)</title><rect x="1128.5" y="229" width="61.5" height="15.0" fill="rgb(205,0,0)" rx="2" ry="2" />
|
||||
<text x="1131.54" y="239.5" >__fute..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>start_thread (364 samples, 94.79%)</title><rect x="10.0" y="309" width="1118.5" height="15.0" fill="rgb(212,34,8)" rx="2" ry="2" />
|
||||
<text x="13.00" y="319.5" >start_thread</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__GI___lll_lock_wait (1 samples, 0.26%)</title><rect x="117.6" y="197" width="3.0" height="15.0" fill="rgb(226,96,23)" rx="2" ry="2" />
|
||||
<text x="120.55" y="207.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>clone3 (364 samples, 94.79%)</title><rect x="10.0" y="325" width="1118.5" height="15.0" fill="rgb(216,54,12)" rx="2" ry="2" />
|
||||
<text x="13.00" y="335.5" >clone3</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>dpp::http_request::run(dpp::cluster*) (4 samples, 1.04%)</title><rect x="919.6" y="261" width="12.3" height="15.0" fill="rgb(230,119,28)" rx="2" ry="2" />
|
||||
<text x="922.58" y="271.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__GI__IO_default_uflow (4 samples, 1.04%)</title><rect x="1067.1" y="101" width="12.3" height="15.0" fill="rgb(234,135,32)" rx="2" ry="2" />
|
||||
<text x="1070.08" y="111.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>std::thread::_Invoker<std::tuple<bumbleBee::MusicPlayManager::play(dpp::discord_voice_client*)::<lambda(dpp::discord_voice_client*)>, (13 samples, 3.39%)</title><rect x="1082.4" y="245" width="40.0" height="15.0" fill="rgb(208,13,3)" rx="2" ry="2" />
|
||||
<text x="1085.45" y="255.5" >std..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__GI___libc_read (4 samples, 1.04%)</title><rect x="1067.1" y="69" width="12.3" height="15.0" fill="rgb(241,166,39)" rx="2" ry="2" />
|
||||
<text x="1070.08" y="79.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__GI___libc_read (3 samples, 0.78%)</title><rect x="1057.9" y="101" width="9.2" height="15.0" fill="rgb(241,166,39)" rx="2" ry="2" />
|
||||
<text x="1060.86" y="111.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>_IO_new_file_underflow (4 samples, 1.04%)</title><rect x="1067.1" y="85" width="12.3" height="15.0" fill="rgb(214,42,10)" rx="2" ry="2" />
|
||||
<text x="1070.08" y="95.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__GI___libc_read (4 samples, 1.04%)</title><rect x="1067.1" y="53" width="12.3" height="15.0" fill="rgb(241,166,39)" rx="2" ry="2" />
|
||||
<text x="1070.08" y="63.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>dpp::discord_voice_client::stop_audio() (1 samples, 0.26%)</title><rect x="1054.8" y="149" width="3.1" height="15.0" fill="rgb(217,57,13)" rx="2" ry="2" />
|
||||
<text x="1057.79" y="159.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>std::__invoke<bumbleBee::BumbleBee::on_slashcommand(const (9 samples, 2.34%)</title><rect x="1054.8" y="229" width="27.6" height="15.0" fill="rgb(253,222,53)" rx="2" ry="2" />
|
||||
<text x="1057.79" y="239.5" >s..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>dpp::ssl_client::read_loop() (20 samples, 5.21%)</title><rect x="10.0" y="261" width="61.5" height="15.0" fill="rgb(243,175,41)" rx="2" ry="2" />
|
||||
<text x="13.00" y="271.5" >dpp::s..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__GI__IO_getline_info (4 samples, 1.04%)</title><rect x="1067.1" y="117" width="12.3" height="15.0" fill="rgb(248,199,47)" rx="2" ry="2" />
|
||||
<text x="1070.08" y="127.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>bumbleBee::commands::Delete::execute (1 samples, 0.26%)</title><rect x="1054.8" y="181" width="3.1" height="15.0" fill="rgb(246,190,45)" rx="2" ry="2" />
|
||||
<text x="1057.79" y="191.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>std::__invoke_impl<void, (2 samples, 0.52%)</title><rect x="1122.4" y="213" width="6.1" height="15.0" fill="rgb(244,179,42)" rx="2" ry="2" />
|
||||
<text x="1125.40" y="223.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>dpp::cluster::start(bool) (20 samples, 5.21%)</title><rect x="1128.5" y="293" width="61.5" height="15.0" fill="rgb(205,0,0)" rx="2" ry="2" />
|
||||
<text x="1131.54" y="303.5" >dpp::c..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>std::vector<dpp::voice_out_packet, (3 samples, 0.78%)</title><rect x="123.7" y="229" width="9.2" height="15.0" fill="rgb(222,82,19)" rx="2" ry="2" />
|
||||
<text x="126.70" y="239.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>dpp::discord_voice_client::want_write() (1 samples, 0.26%)</title><rect x="117.6" y="245" width="3.0" height="15.0" fill="rgb(233,130,31)" rx="2" ry="2" />
|
||||
<text x="120.55" y="255.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__pthread_cond_wait_common (13 samples, 3.39%)</title><rect x="1082.4" y="149" width="40.0" height="15.0" fill="rgb(210,24,5)" rx="2" ry="2" />
|
||||
<text x="1085.45" y="159.5" >__p..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__futex_abstimed_wait_common64 (256 samples, 66.67%)</title><rect x="132.9" y="181" width="786.7" height="15.0" fill="rgb(220,71,17)" rx="2" ry="2" />
|
||||
<text x="135.92" y="191.5" >__futex_abstimed_wait_common64</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__GI___nanosleep (1 samples, 0.26%)</title><rect x="120.6" y="229" width="3.1" height="15.0" fill="rgb(231,119,28)" rx="2" ry="2" />
|
||||
<text x="123.62" y="239.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>oggz_read (2 samples, 0.52%)</title><rect x="1122.4" y="181" width="6.1" height="15.0" fill="rgb(252,218,52)" rx="2" ry="2" />
|
||||
<text x="1125.40" y="191.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>operator() (13 samples, 3.39%)</title><rect x="1082.4" y="197" width="40.0" height="15.0" fill="rgb(231,121,29)" rx="2" ry="2" />
|
||||
<text x="1085.45" y="207.5" >ope..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>std::thread::_Invoker<std::tuple<bumbleBee::MusicPlayManager::send_audio_to_voice(std::shared_ptr<bumbleBee::MusicQueueElement>, (2 samples, 0.52%)</title><rect x="1122.4" y="261" width="6.1" height="15.0" fill="rgb(216,50,12)" rx="2" ry="2" />
|
||||
<text x="1125.40" y="271.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__GI___futex_abstimed_wait_cancelable64 (256 samples, 66.67%)</title><rect x="132.9" y="213" width="786.7" height="15.0" fill="rgb(215,48,11)" rx="2" ry="2" />
|
||||
<text x="135.92" y="223.5" >__GI___futex_abstimed_wait_cancelable64</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>nlohmann::json_abi_v3_11_2::detail::parser<nlohmann::json_abi_v3_11_2::basic_json<std::map, (1 samples, 0.26%)</title><rect x="1079.4" y="117" width="3.0" height="15.0" fill="rgb(217,59,14)" rx="2" ry="2" />
|
||||
<text x="1082.38" y="127.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>std::move<char&> (1 samples, 0.26%)</title><rect x="1079.4" y="37" width="3.0" height="15.0" fill="rgb(232,124,29)" rx="2" ry="2" />
|
||||
<text x="1082.38" y="47.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>bumbleBee::ConsoleUtils::getResultFromCommand (4 samples, 1.04%)</title><rect x="1067.1" y="165" width="12.3" height="15.0" fill="rgb(223,85,20)" rx="2" ry="2" />
|
||||
<text x="1070.08" y="175.5" ></text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__futex_abstimed_wait_common64 (20 samples, 5.21%)</title><rect x="1128.5" y="213" width="61.5" height="15.0" fill="rgb(220,71,17)" rx="2" ry="2" />
|
||||
<text x="1131.54" y="223.5" >__fute..</text>
|
||||
</g>
|
||||
<g >
|
||||
<title>__GI___futex_abstimed_wait_cancelable64 (20 samples, 5.21%)</title><rect x="1128.5" y="245" width="61.5" height="15.0" fill="rgb(215,48,11)" rx="2" ry="2" />
|
||||
<text x="1131.54" y="255.5" >__GI__..</text>
|
||||
</g>
|
||||
</g>
|
||||
</svg>
|
||||
|
Before Width: | Height: | Size: 36 KiB |
@@ -1,4 +0,0 @@
|
||||
sudo echo recording...
|
||||
sudo perf record -F 99 -a -g -- sleep 60
|
||||
sudo perf script | ./stackcollapse-perf.pl | ./flamegraph.pl > stackflame.svg
|
||||
sudo rm perf.data
|
||||
8
tests/CMakeLists.txt
Normal file
8
tests/CMakeLists.txt
Normal 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()
|
||||
52
tests/basic_yt_dlp_download.cc
Normal file
52
tests/basic_yt_dlp_download.cc
Normal 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
232
tests/ffmpeg_any_to_opus.cc
Normal 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;
|
||||
}
|
||||
155
tests/ffmpeg_decode_audio.cc
Normal file
155
tests/ffmpeg_decode_audio.cc
Normal 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
8
tests/update.cc
Normal 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;
|
||||
}
|
||||
Reference in New Issue
Block a user