Compare commits
57 Commits
database
...
renovate/a
| Author | SHA1 | Date | |
|---|---|---|---|
| 00d2f2365f | |||
| de4260dd84 | |||
| 335626312b | |||
| 07a4da73c1 | |||
| fa1d720d0f | |||
| 1ec358ec8f | |||
| 624e08f3aa | |||
| f4f6a486c9 | |||
| c354ec0562 | |||
| e0f1abcd9c | |||
| aa302e4bb5 | |||
| c461671bfb | |||
| 271ec4d323 | |||
| 1e807c121d | |||
| 4293775824 | |||
| fa8f5f5b04 | |||
| 4ecf9bf205 | |||
|
5b5ca52c04
|
|||
|
ebb66e21cc
|
|||
| f240b8f0f6 | |||
| 65c09c5e85 | |||
| c9837e456e | |||
| 9efc372f8f | |||
|
7ff991206b
|
|||
|
b1d6069d81
|
|||
| de3262c072 | |||
| 8e88cadbb8 | |||
|
c6e34cf000
|
|||
| 3b135894eb | |||
| 14cab9d833 | |||
|
f2f4396674
|
|||
|
b60cce35ca
|
|||
|
9f1ed63011
|
|||
| 6096e625a2 | |||
| 5c42fb9480 | |||
|
d1df70a888
|
|||
| c4bc5503a5 | |||
| 16f6c97eec | |||
| cd42401280 | |||
| cc0af054f8 | |||
| aff5e1dd8d | |||
| ed9956c5a5 | |||
|
456a51bc70
|
|||
|
b9b3709bb4
|
|||
|
27168c36cf
|
|||
|
88d9c1d841
|
|||
|
d3a4114347
|
|||
|
a3e9175ea8
|
|||
|
6601396a88
|
|||
|
687f110416
|
|||
|
9146ac06a6
|
|||
|
a25e8ef65f
|
|||
|
f0337050c1
|
|||
|
440201e153
|
|||
|
f7dca8ff65
|
|||
|
1e523502eb
|
|||
|
dfd0f9c8c8
|
@@ -13,47 +13,47 @@ jobs:
|
|||||||
fmt:
|
fmt:
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4
|
||||||
- run: sudo apt-get update
|
- run: sudo apt-get update
|
||||||
- run: sudo apt-get install -y cmake
|
- run: sudo apt-get install -y cmake
|
||||||
- uses: actions-rust-lang/setup-rust-toolchain@v1
|
- uses: actions-rust-lang/setup-rust-toolchain@9399c7bb15d4c7d47b27263d024f0a4978346ba4 # v1
|
||||||
with:
|
with:
|
||||||
components: rustfmt
|
components: rustfmt
|
||||||
- name: Rustfmt Check
|
- name: Rustfmt Check
|
||||||
uses: actions-rust-lang/rustfmt@v1
|
uses: actions-rust-lang/rustfmt@559aa3035a47390ba96088dffa783b5d26da9326 # v1
|
||||||
|
|
||||||
build:
|
build:
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4
|
||||||
- run: sudo apt-get update
|
- run: sudo apt-get update
|
||||||
- run: sudo apt-get install -y cmake
|
- run: sudo apt-get install -y cmake
|
||||||
- uses: actions-rust-lang/setup-rust-toolchain@v1
|
- uses: actions-rust-lang/setup-rust-toolchain@9399c7bb15d4c7d47b27263d024f0a4978346ba4 # v1
|
||||||
- name: Build
|
- name: Build
|
||||||
run: cargo build --verbose
|
run: cargo build --verbose
|
||||||
|
|
||||||
test:
|
test:
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4
|
||||||
- run: sudo apt-get update
|
- run: sudo apt-get update
|
||||||
- run: sudo apt-get install -y cmake
|
- run: sudo apt-get install -y cmake
|
||||||
- uses: actions/setup-python@v5
|
- uses: actions/setup-python@8d9ed9ac5c53483de85588cdf95a591a75ab9f55 # v5
|
||||||
with:
|
with:
|
||||||
python-version: '3.12'
|
python-version: '3.13'
|
||||||
- name: Install yt-dlp
|
- name: Install yt-dlp
|
||||||
run: pip install yt-dlp
|
run: pip install yt-dlp
|
||||||
- uses: actions-rust-lang/setup-rust-toolchain@v1
|
- uses: actions-rust-lang/setup-rust-toolchain@9399c7bb15d4c7d47b27263d024f0a4978346ba4 # v1
|
||||||
- name: Run tests
|
- name: Run tests
|
||||||
run: cargo test --verbose --all-features
|
run: cargo test --verbose --all-features
|
||||||
|
|
||||||
clippy:
|
clippy:
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4
|
||||||
- run: sudo apt-get update
|
- run: sudo apt-get update
|
||||||
- run: sudo apt-get install -y cmake
|
- run: sudo apt-get install -y cmake
|
||||||
- uses: actions-rust-lang/setup-rust-toolchain@v1
|
- uses: actions-rust-lang/setup-rust-toolchain@9399c7bb15d4c7d47b27263d024f0a4978346ba4 # v1
|
||||||
with:
|
with:
|
||||||
components: clippy
|
components: clippy
|
||||||
- name: Run clippy
|
- name: Run clippy
|
||||||
@@ -62,13 +62,13 @@ jobs:
|
|||||||
pre-commit:
|
pre-commit:
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4
|
||||||
- run: sudo apt-get update
|
- run: sudo apt-get update
|
||||||
- run: sudo apt-get install -y cmake
|
- run: sudo apt-get install -y cmake
|
||||||
- uses: actions/setup-python@v5
|
- uses: actions/setup-python@8d9ed9ac5c53483de85588cdf95a591a75ab9f55 # v5
|
||||||
with:
|
with:
|
||||||
python-version: '3.12'
|
python-version: '3.13'
|
||||||
- uses: actions-rust-lang/setup-rust-toolchain@v1
|
- uses: actions-rust-lang/setup-rust-toolchain@9399c7bb15d4c7d47b27263d024f0a4978346ba4 # v1
|
||||||
with:
|
with:
|
||||||
components: rustfmt
|
components: rustfmt
|
||||||
- uses: pre-commit/action@v3.0.1
|
- uses: pre-commit/action@2c7b3805fd2a0fd8c1884dcaebf91fc102a13ecd # v3.0.1
|
||||||
|
|||||||
@@ -13,21 +13,23 @@ jobs:
|
|||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
steps:
|
steps:
|
||||||
- name: Set up QEMU
|
- name: Set up QEMU
|
||||||
uses: docker/setup-qemu-action@v3
|
uses: docker/setup-qemu-action@29109295f81e9208d7d86ff1c6c12d2833863392 # v3
|
||||||
|
|
||||||
- name: Set up Docker BuildX
|
- name: Set up Docker BuildX
|
||||||
uses: docker/setup-buildx-action@v3
|
uses: docker/setup-buildx-action@b5ca514318bd6ebac0fb2aedd5d36ec1b5c232a2 # v3
|
||||||
|
|
||||||
- name: Login to DockerHub
|
- name: Login to DockerHub
|
||||||
uses: docker/login-action@v3
|
uses: docker/login-action@74a5d142397b4f367a81961eba4e8cd7edddf772 # v3
|
||||||
with:
|
with:
|
||||||
username: ${{ secrets.DOCKER_USERNAME }}
|
username: ${{ secrets.DOCKER_USERNAME }}
|
||||||
password: ${{ secrets.DOCKER_PASSWORD }}
|
password: ${{ secrets.DOCKER_PASSWORD }}
|
||||||
|
|
||||||
- name: Build and push
|
- name: Build and push
|
||||||
uses: docker/build-push-action@v5
|
uses: docker/build-push-action@471d1dc4e07e5cdedd4c2171150001c434f0b7a4 # v6
|
||||||
with:
|
with:
|
||||||
platforms: |
|
platforms: |
|
||||||
linux/amd64
|
linux/amd64
|
||||||
push: true
|
push: true
|
||||||
tags: jheuel/ohrwurm:latest
|
tags: jheuel/ohrwurm:latest
|
||||||
|
cache-from: type=gha
|
||||||
|
cache-to: type=gha,mode=max
|
||||||
|
|||||||
32
.gitea/workflows/renovate.yaml
Normal file
32
.gitea/workflows/renovate.yaml
Normal file
@@ -0,0 +1,32 @@
|
|||||||
|
---
|
||||||
|
# yamllint disable rule:line-length
|
||||||
|
|
||||||
|
name: renovate
|
||||||
|
|
||||||
|
on: # yamllint disable-line rule:truthy
|
||||||
|
schedule:
|
||||||
|
- cron: "@daily"
|
||||||
|
push:
|
||||||
|
branches:
|
||||||
|
- main
|
||||||
|
|
||||||
|
jobs:
|
||||||
|
renovate:
|
||||||
|
runs-on: [nas]
|
||||||
|
steps:
|
||||||
|
- name: Checkout
|
||||||
|
uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
|
||||||
|
- name: Self-hosted Renovate
|
||||||
|
uses: renovatebot/github-action@c21017a4a2fc9f42953bcc907e375a5a544557ec # v41.0.18
|
||||||
|
with:
|
||||||
|
token: ${{ secrets.RENOVATE_TOKEN }}
|
||||||
|
env-regex: "^(?:RENOVATE_\\w+|LOG_LEVEL|GITHUB_COM_TOKEN)$"
|
||||||
|
env:
|
||||||
|
LOG_LEVEL: "info"
|
||||||
|
GITHUB_COM_TOKEN: ${{ secrets.CI_GITHUB_TOKEN }}
|
||||||
|
RENOVATE_PLATFORM: "gitea"
|
||||||
|
RENOVATE_GIT_AUTHOR: "Renovate Bot <renovate@boosted.science>"
|
||||||
|
RENOVATE_GIT_EMAIL: "renovate@boosted.science"
|
||||||
|
RENOVATE_ENDPOINT: "https://gitea.jheuel.boosted.science/api/v1/"
|
||||||
|
RENOVATE_AUTODISCOVER: "true"
|
||||||
|
RENOVATE_OPTIMIZE_FOR_DISABLED: "true"
|
||||||
32
.github/workflows/ci.yml
vendored
32
.github/workflows/ci.yml
vendored
@@ -12,47 +12,47 @@ jobs:
|
|||||||
fmt:
|
fmt:
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4
|
||||||
- run: sudo apt-get update
|
- run: sudo apt-get update
|
||||||
- run: sudo apt-get install -y cmake
|
- run: sudo apt-get install -y cmake
|
||||||
- uses: actions-rust-lang/setup-rust-toolchain@v1
|
- uses: actions-rust-lang/setup-rust-toolchain@9399c7bb15d4c7d47b27263d024f0a4978346ba4 # v1
|
||||||
with:
|
with:
|
||||||
components: rustfmt
|
components: rustfmt
|
||||||
- name: Rustfmt Check
|
- name: Rustfmt Check
|
||||||
uses: actions-rust-lang/rustfmt@v1
|
uses: actions-rust-lang/rustfmt@559aa3035a47390ba96088dffa783b5d26da9326 # v1
|
||||||
|
|
||||||
build:
|
build:
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4
|
||||||
- run: sudo apt-get update
|
- run: sudo apt-get update
|
||||||
- run: sudo apt-get install -y cmake
|
- run: sudo apt-get install -y cmake
|
||||||
- uses: actions-rust-lang/setup-rust-toolchain@v1
|
- uses: actions-rust-lang/setup-rust-toolchain@9399c7bb15d4c7d47b27263d024f0a4978346ba4 # v1
|
||||||
- name: Build
|
- name: Build
|
||||||
run: cargo build --verbose
|
run: cargo build --verbose
|
||||||
|
|
||||||
test:
|
test:
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4
|
||||||
- run: sudo apt-get update
|
- run: sudo apt-get update
|
||||||
- run: sudo apt-get install -y cmake
|
- run: sudo apt-get install -y cmake
|
||||||
- uses: actions/setup-python@v5
|
- uses: actions/setup-python@8d9ed9ac5c53483de85588cdf95a591a75ab9f55 # v5
|
||||||
with:
|
with:
|
||||||
python-version: '3.12'
|
python-version: '3.13'
|
||||||
- name: Install yt-dlp
|
- name: Install yt-dlp
|
||||||
run: pip install yt-dlp
|
run: pip install yt-dlp
|
||||||
- uses: actions-rust-lang/setup-rust-toolchain@v1
|
- uses: actions-rust-lang/setup-rust-toolchain@9399c7bb15d4c7d47b27263d024f0a4978346ba4 # v1
|
||||||
- name: Run tests
|
- name: Run tests
|
||||||
run: cargo test --verbose --all-features
|
run: cargo test --verbose --all-features
|
||||||
|
|
||||||
clippy:
|
clippy:
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4
|
||||||
- run: sudo apt-get update
|
- run: sudo apt-get update
|
||||||
- run: sudo apt-get install -y cmake
|
- run: sudo apt-get install -y cmake
|
||||||
- uses: actions-rust-lang/setup-rust-toolchain@v1
|
- uses: actions-rust-lang/setup-rust-toolchain@9399c7bb15d4c7d47b27263d024f0a4978346ba4 # v1
|
||||||
with:
|
with:
|
||||||
components: clippy
|
components: clippy
|
||||||
- name: Run clippy
|
- name: Run clippy
|
||||||
@@ -61,13 +61,13 @@ jobs:
|
|||||||
pre-commit:
|
pre-commit:
|
||||||
runs-on: ubuntu-latest
|
runs-on: ubuntu-latest
|
||||||
steps:
|
steps:
|
||||||
- uses: actions/checkout@v4
|
- uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4
|
||||||
- run: sudo apt-get update
|
- run: sudo apt-get update
|
||||||
- run: sudo apt-get install -y cmake
|
- run: sudo apt-get install -y cmake
|
||||||
- uses: actions/setup-python@v5
|
- uses: actions/setup-python@8d9ed9ac5c53483de85588cdf95a591a75ab9f55 # v5
|
||||||
with:
|
with:
|
||||||
python-version: '3.12'
|
python-version: '3.13'
|
||||||
- uses: actions-rust-lang/setup-rust-toolchain@v1
|
- uses: actions-rust-lang/setup-rust-toolchain@9399c7bb15d4c7d47b27263d024f0a4978346ba4 # v1
|
||||||
with:
|
with:
|
||||||
components: rustfmt
|
components: rustfmt
|
||||||
- uses: pre-commit/action@v3.0.1
|
- uses: pre-commit/action@2c7b3805fd2a0fd8c1884dcaebf91fc102a13ecd # v3.0.1
|
||||||
|
|||||||
2451
Cargo.lock
generated
2451
Cargo.lock
generated
File diff suppressed because it is too large
Load Diff
22
Cargo.toml
22
Cargo.toml
@@ -6,21 +6,21 @@ edition = "2021"
|
|||||||
license = "MIT"
|
license = "MIT"
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
symphonia = { version = "0.5.2", features=["all"] }
|
symphonia = { version = "0.5.2", features=["all", "opt-simd"] }
|
||||||
songbird = { version = "0.4.0", features = ["driver", "gateway", "twilight", "rustls", "builtin-queue"] }
|
songbird = { version = "0.5.0", features = ["driver", "gateway", "twilight", "rustls", "builtin-queue"] }
|
||||||
tokio = { features = ["macros", "rt-multi-thread", "signal", "sync"], version = "1" }
|
tokio = { features = ["macros", "rt-multi-thread", "signal", "sync"], version = "1" }
|
||||||
sqlx = { version = "0.7", features = [ "runtime-tokio", "sqlite", "chrono", "migrate"] }
|
sqlx = { version = "0.8", features = [ "runtime-tokio", "sqlite", "chrono", "migrate"] }
|
||||||
regex = { version = "1", features = ["unicode-case"] }
|
regex = { version = "1", features = ["unicode-case"] }
|
||||||
futures = "0.3"
|
futures = "0.3"
|
||||||
reqwest = "0.11"
|
reqwest = "0.12"
|
||||||
tracing = "0.1"
|
tracing = "0.1"
|
||||||
tracing-subscriber = "0.2"
|
tracing-subscriber = "0.3"
|
||||||
twilight-gateway = "0.15"
|
twilight-gateway = "0.16"
|
||||||
twilight-http = "0.15"
|
twilight-http = "0.16"
|
||||||
twilight-model = "0.15"
|
twilight-model = "0.16"
|
||||||
twilight-standby = "0.15"
|
twilight-standby = "0.16"
|
||||||
twilight-cache-inmemory = "0.15"
|
twilight-cache-inmemory = "0.16"
|
||||||
twilight-util = { version = "0.15", features=["builder"] }
|
twilight-util = { version = "0.16", features=["builder"] }
|
||||||
dotenv = "0.15.0"
|
dotenv = "0.15.0"
|
||||||
serde = { version = "1.0", features = ["derive"] }
|
serde = { version = "1.0", features = ["derive"] }
|
||||||
serde_json = "1.0"
|
serde_json = "1.0"
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
# Build image
|
# Build image
|
||||||
FROM rust:slim-bullseye as build
|
FROM rust:slim-bullseye@sha256:1eca9af45f393ac4669b9b63659529638359575f6268cbd4e6543ddc46c53803 as build
|
||||||
|
|
||||||
RUN apt-get update && apt-get install -y \
|
RUN apt-get update && apt-get install -y \
|
||||||
build-essential autoconf automake cmake libtool libssl-dev pkg-config
|
build-essential autoconf automake cmake libtool libssl-dev pkg-config
|
||||||
@@ -18,9 +18,9 @@ RUN touch src/main.rs
|
|||||||
RUN cargo build --release --locked
|
RUN cargo build --release --locked
|
||||||
|
|
||||||
# Release image
|
# Release image
|
||||||
FROM debian:bullseye-slim
|
FROM debian:bullseye-slim@sha256:e4b93db6aad977a95aa103917f3de8a2b16ead91cf255c3ccdb300c5d20f3015
|
||||||
|
|
||||||
RUN apt-get update && apt-get install -y python3-pip ffmpeg
|
RUN apt-get update && apt-get install -y python3-pip
|
||||||
RUN pip install -U yt-dlp
|
RUN pip install -U yt-dlp
|
||||||
|
|
||||||
COPY --from=build /app/target/release/ohrwurm .
|
COPY --from=build /app/target/release/ohrwurm .
|
||||||
|
|||||||
@@ -1,4 +1,3 @@
|
|||||||
-- Add migration script here
|
|
||||||
CREATE TABLE IF NOT EXISTS tracks
|
CREATE TABLE IF NOT EXISTS tracks
|
||||||
(
|
(
|
||||||
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
id INTEGER PRIMARY KEY AUTOINCREMENT,
|
||||||
|
|||||||
25
renovate.json
Normal file
25
renovate.json
Normal file
@@ -0,0 +1,25 @@
|
|||||||
|
{
|
||||||
|
"$schema": "https://docs.renovatebot.com/renovate-schema.json",
|
||||||
|
"timezone": "Europe/Berlin",
|
||||||
|
"extends": [
|
||||||
|
"config:recommended",
|
||||||
|
"helpers:pinGitHubActionDigests",
|
||||||
|
":configMigration",
|
||||||
|
":pinDevDependencies",
|
||||||
|
":dependencyDashboard",
|
||||||
|
"docker:pinDigests",
|
||||||
|
"preview:dockerCompose",
|
||||||
|
"preview:dockerVersions"
|
||||||
|
],
|
||||||
|
"packageRules": [
|
||||||
|
{
|
||||||
|
"matchUpdateTypes": [
|
||||||
|
"minor",
|
||||||
|
"patch",
|
||||||
|
"pin",
|
||||||
|
"digest"
|
||||||
|
],
|
||||||
|
"automerge": true
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
@@ -26,7 +26,7 @@ pub(crate) async fn delete(
|
|||||||
.http
|
.http
|
||||||
.channel_messages(msg.channel_id)
|
.channel_messages(msg.channel_id)
|
||||||
.before(msg.id)
|
.before(msg.id)
|
||||||
.limit(n)?
|
.limit(n)
|
||||||
.await?
|
.await?
|
||||||
.model()
|
.model()
|
||||||
.await?;
|
.await?;
|
||||||
|
|||||||
@@ -1,4 +1,5 @@
|
|||||||
use crate::state::State;
|
use crate::state::State;
|
||||||
|
use anyhow::Context;
|
||||||
use std::error::Error;
|
use std::error::Error;
|
||||||
use tracing::debug;
|
use tracing::debug;
|
||||||
use twilight_model::{
|
use twilight_model::{
|
||||||
@@ -22,7 +23,7 @@ pub(crate) async fn join_channel(
|
|||||||
let channel_id = state
|
let channel_id = state
|
||||||
.cache
|
.cache
|
||||||
.voice_state(user_id, guild_id)
|
.voice_state(user_id, guild_id)
|
||||||
.ok_or("Cannot get voice state for user")?
|
.context("Could not get voice state for user")?
|
||||||
.channel_id();
|
.channel_id();
|
||||||
|
|
||||||
// join the voice channel
|
// join the voice channel
|
||||||
@@ -30,12 +31,12 @@ pub(crate) async fn join_channel(
|
|||||||
.songbird
|
.songbird
|
||||||
.join(guild_id.cast(), channel_id)
|
.join(guild_id.cast(), channel_id)
|
||||||
.await
|
.await
|
||||||
.map_err(|e| format!("Could not join voice channel: {:?}", e))?;
|
.context("Could not join voice channel")?;
|
||||||
|
|
||||||
// signal that we are not listening
|
// signal that we are not listening
|
||||||
if let Some(call_lock) = state.songbird.get(guild_id.cast()) {
|
if let Some(call_lock) = state.songbird.get(guild_id.cast()) {
|
||||||
let mut call = call_lock.lock().await;
|
let mut call = call_lock.lock().await;
|
||||||
call.deafen(true).await?;
|
call.deafen(true).await.context("Could not deafen")?;
|
||||||
}
|
}
|
||||||
|
|
||||||
// create guild config
|
// create guild config
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
use crate::metadata::MetadataMap;
|
use crate::metadata::Metadata;
|
||||||
use crate::state::{State, StateRef};
|
use crate::state::{State, StateRef};
|
||||||
use async_trait::async_trait;
|
use async_trait::async_trait;
|
||||||
|
use songbird::tracks::Track;
|
||||||
use songbird::{Event, EventContext, EventHandler, TrackEvent};
|
use songbird::{Event, EventContext, EventHandler, TrackEvent};
|
||||||
use std::ops::Sub;
|
use std::ops::Sub;
|
||||||
use std::time::Duration;
|
use std::time::Duration;
|
||||||
@@ -99,12 +100,11 @@ impl EventHandler for TrackEndNotifier {
|
|||||||
let mut call = call_lock.lock().await;
|
let mut call = call_lock.lock().await;
|
||||||
|
|
||||||
// get metadata from finished track
|
// get metadata from finished track
|
||||||
let old_typemap_lock = track_handle.typemap().read().await;
|
let old_metadata = track_handle.data::<Metadata>();
|
||||||
let old_metadata = old_typemap_lock.get::<MetadataMap>().unwrap();
|
|
||||||
|
|
||||||
// enqueue track
|
// enqueue track
|
||||||
let handle = call.enqueue_with_preload(
|
let track = Track::new_with_data(old_metadata.src.clone().into(), old_metadata.clone());
|
||||||
old_metadata.src.clone().into(),
|
let _handle = call.enqueue_with_preload(
|
||||||
|
track,
|
||||||
old_metadata.duration.map(|duration| -> Duration {
|
old_metadata.duration.map(|duration| -> Duration {
|
||||||
if duration.as_secs() > 5 {
|
if duration.as_secs() > 5 {
|
||||||
duration.sub(Duration::from_secs(5))
|
duration.sub(Duration::from_secs(5))
|
||||||
@@ -113,10 +113,6 @@ impl EventHandler for TrackEndNotifier {
|
|||||||
}
|
}
|
||||||
}),
|
}),
|
||||||
);
|
);
|
||||||
|
|
||||||
// insert metadata into new track
|
|
||||||
let mut new_typemap = handle.typemap().write().await;
|
|
||||||
new_typemap.insert::<MetadataMap>(old_metadata.clone());
|
|
||||||
}
|
}
|
||||||
None
|
None
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,27 +1,39 @@
|
|||||||
use crate::commands::join::join_channel;
|
use crate::commands::join::join_channel;
|
||||||
use crate::metadata::{Metadata, MetadataMap};
|
use crate::metadata::Metadata;
|
||||||
use crate::state::State;
|
use crate::state::State;
|
||||||
use crate::{colors, db};
|
use crate::{colors, db};
|
||||||
|
|
||||||
|
use anyhow::Context;
|
||||||
use serde::{Deserialize, Serialize};
|
use serde::{Deserialize, Serialize};
|
||||||
use songbird::input::cached::Memory;
|
|
||||||
use songbird::input::{Compose, YoutubeDl};
|
use songbird::input::{Compose, YoutubeDl};
|
||||||
use songbird::tracks::Track;
|
use songbird::tracks::Track;
|
||||||
use std::io::{BufRead, BufReader};
|
use std::sync::Arc;
|
||||||
use std::ops::Sub;
|
|
||||||
use std::{error::Error, time::Duration};
|
use std::{error::Error, time::Duration};
|
||||||
|
use std::{
|
||||||
|
io::{BufRead, BufReader},
|
||||||
|
ops::Sub,
|
||||||
|
};
|
||||||
use tokio::process::Command;
|
use tokio::process::Command;
|
||||||
use tracing::debug;
|
use tracing::debug;
|
||||||
use twilight_model::channel::message::embed::{
|
use twilight_model::channel::message::embed::{
|
||||||
EmbedAuthor, EmbedField, EmbedFooter, EmbedThumbnail,
|
EmbedAuthor, EmbedField, EmbedFooter, EmbedThumbnail,
|
||||||
};
|
};
|
||||||
use twilight_model::channel::message::MessageFlags;
|
use twilight_model::channel::message::{Embed, MessageFlags};
|
||||||
use twilight_model::gateway::payload::incoming::InteractionCreate;
|
use twilight_model::gateway::payload::incoming::InteractionCreate;
|
||||||
use twilight_model::http::interaction::{InteractionResponse, InteractionResponseType};
|
use twilight_model::http::interaction::{InteractionResponse, InteractionResponseType};
|
||||||
use twilight_util::builder::embed::EmbedBuilder;
|
use twilight_util::builder::embed::EmbedBuilder;
|
||||||
use twilight_util::builder::InteractionResponseDataBuilder;
|
use twilight_util::builder::InteractionResponseDataBuilder;
|
||||||
use url::Url;
|
use url::Url;
|
||||||
|
|
||||||
|
#[derive(Debug)]
|
||||||
|
struct TrackType {
|
||||||
|
url: String,
|
||||||
|
title: Option<String>,
|
||||||
|
duration_string: String,
|
||||||
|
channel: String,
|
||||||
|
thumbnail: Option<String>,
|
||||||
|
}
|
||||||
|
|
||||||
#[derive(Debug, Serialize, Deserialize)]
|
#[derive(Debug, Serialize, Deserialize)]
|
||||||
struct YouTubeTrack {
|
struct YouTubeTrack {
|
||||||
url: Option<String>,
|
url: Option<String>,
|
||||||
@@ -31,6 +43,7 @@ struct YouTubeTrack {
|
|||||||
playlist: Option<String>,
|
playlist: Option<String>,
|
||||||
playlist_id: Option<String>,
|
playlist_id: Option<String>,
|
||||||
duration_string: String,
|
duration_string: String,
|
||||||
|
thumbnail: Option<String>,
|
||||||
}
|
}
|
||||||
|
|
||||||
fn build_playlist_url(playlist_id: &str) -> String {
|
fn build_playlist_url(playlist_id: &str) -> String {
|
||||||
@@ -45,12 +58,18 @@ async fn get_tracks(
|
|||||||
.output()
|
.output()
|
||||||
.await?;
|
.await?;
|
||||||
|
|
||||||
|
tracing::info!(
|
||||||
|
"yt-dlp output: {:?}",
|
||||||
|
String::from_utf8_lossy(&output.stdout)
|
||||||
|
);
|
||||||
|
|
||||||
let reader = BufReader::new(output.stdout.as_slice());
|
let reader = BufReader::new(output.stdout.as_slice());
|
||||||
let tracks: Vec<YouTubeTrack> = reader
|
let tracks: Vec<YouTubeTrack> = reader
|
||||||
.lines()
|
.lines()
|
||||||
.map_while(Result::ok)
|
.map_while(Result::ok)
|
||||||
.flat_map(|line| serde_json::from_str(&line))
|
.flat_map(|line| serde_json::from_str(&line))
|
||||||
.collect();
|
.collect();
|
||||||
|
tracing::info!("yt-dlp tracks: {:?}", tracks);
|
||||||
|
|
||||||
if tracks.is_empty() {
|
if tracks.is_empty() {
|
||||||
if let Ok(stderr) = String::from_utf8(output.stderr) {
|
if let Ok(stderr) = String::from_utf8(output.stderr) {
|
||||||
@@ -63,22 +82,210 @@ async fn get_tracks(
|
|||||||
}
|
}
|
||||||
return Err("No tracks found".into());
|
return Err("No tracks found".into());
|
||||||
}
|
}
|
||||||
tracing::debug!("tracks: {:?}", tracks);
|
tracing::info!("tracks: {:?}", tracks);
|
||||||
Ok(tracks)
|
Ok(tracks)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
async fn persistence(
|
||||||
|
interaction: &InteractionCreate,
|
||||||
|
track: &YouTubeTrack,
|
||||||
|
state: State,
|
||||||
|
) -> Result<(), Box<dyn Error + Send + Sync + 'static>> {
|
||||||
|
let Some(guild_id) = interaction.guild_id else {
|
||||||
|
return Ok(());
|
||||||
|
};
|
||||||
|
let Some(user_id) = interaction.author_id() else {
|
||||||
|
return Ok(());
|
||||||
|
};
|
||||||
|
let url = track
|
||||||
|
.original_url
|
||||||
|
.clone()
|
||||||
|
.or(track.url.clone())
|
||||||
|
.ok_or("Could not find url")?;
|
||||||
|
let (author_name, author_global_name) = if let Some(author) = interaction.author() {
|
||||||
|
(author.name.clone(), author.global_name.clone())
|
||||||
|
} else {
|
||||||
|
("".to_string(), None)
|
||||||
|
};
|
||||||
|
|
||||||
|
db::track::insert_guild(&state.pool, db::track::Guild::new(guild_id.to_string()))
|
||||||
|
.await
|
||||||
|
.context("failed to insert guild")?;
|
||||||
|
|
||||||
|
db::track::insert_user(
|
||||||
|
&state.pool,
|
||||||
|
db::track::User::new(user_id.to_string(), author_name, author_global_name),
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.context("failed to insert user")?;
|
||||||
|
|
||||||
|
let track_id = db::track::insert_track(
|
||||||
|
&state.pool,
|
||||||
|
db::track::Track::new(
|
||||||
|
url.clone(),
|
||||||
|
track.title.clone(),
|
||||||
|
track.channel.clone(),
|
||||||
|
track.duration_string.clone(),
|
||||||
|
track.thumbnail.clone().unwrap_or_default(),
|
||||||
|
),
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.context("failed to insert track")?;
|
||||||
|
|
||||||
|
db::track::insert_query(
|
||||||
|
&state.pool,
|
||||||
|
db::track::Query::new(user_id.to_string(), guild_id.to_string(), track_id),
|
||||||
|
)
|
||||||
|
.await
|
||||||
|
.context("failed to insert query")?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
fn build_single_track_added_embeds(tracks_added: &[TrackType]) -> Vec<Embed> {
|
||||||
|
let track = tracks_added.first().unwrap();
|
||||||
|
|
||||||
|
let host = if let Ok(host) = Url::parse(&track.url) {
|
||||||
|
Some(
|
||||||
|
host.host_str()
|
||||||
|
.unwrap_or_default()
|
||||||
|
.trim_start_matches("www.")
|
||||||
|
.to_string(),
|
||||||
|
)
|
||||||
|
} else {
|
||||||
|
None
|
||||||
|
};
|
||||||
|
|
||||||
|
let footer = match host {
|
||||||
|
Some(host) => EmbedFooter {
|
||||||
|
text: format!("Streaming from {}", host),
|
||||||
|
icon_url: Some(format!(
|
||||||
|
"https://www.google.com/s2/favicons?domain={}",
|
||||||
|
host
|
||||||
|
)),
|
||||||
|
proxy_icon_url: None,
|
||||||
|
},
|
||||||
|
None => EmbedFooter {
|
||||||
|
text: String::new(),
|
||||||
|
icon_url: None,
|
||||||
|
proxy_icon_url: None,
|
||||||
|
},
|
||||||
|
};
|
||||||
|
|
||||||
|
let mut embed = EmbedBuilder::new()
|
||||||
|
.author(EmbedAuthor {
|
||||||
|
name: "🔊 Added to queue".to_string(),
|
||||||
|
icon_url: None,
|
||||||
|
proxy_icon_url: None,
|
||||||
|
url: None,
|
||||||
|
})
|
||||||
|
.title(track.title.clone().unwrap_or("Unknown".to_string()))
|
||||||
|
.url(track.url.clone())
|
||||||
|
.color(colors::BLURPLE)
|
||||||
|
.footer(footer)
|
||||||
|
.field(EmbedField {
|
||||||
|
inline: true,
|
||||||
|
name: "Duration".to_string(),
|
||||||
|
value: track.duration_string.clone(),
|
||||||
|
})
|
||||||
|
.field(EmbedField {
|
||||||
|
inline: true,
|
||||||
|
name: "Channel".to_string(),
|
||||||
|
value: track.channel.clone(),
|
||||||
|
})
|
||||||
|
.build();
|
||||||
|
|
||||||
|
if let Some(thumbnail) = &track.thumbnail {
|
||||||
|
embed.thumbnail = Some(EmbedThumbnail {
|
||||||
|
height: None,
|
||||||
|
proxy_url: None,
|
||||||
|
url: thumbnail.to_string(),
|
||||||
|
width: None,
|
||||||
|
});
|
||||||
|
}
|
||||||
|
|
||||||
|
vec![embed]
|
||||||
|
}
|
||||||
|
|
||||||
|
fn build_playlist_added_embeds(tracks: &[YouTubeTrack], num_tracks_added: usize) -> Vec<Embed> {
|
||||||
|
let mut content = String::new();
|
||||||
|
let first_track = tracks.first().unwrap();
|
||||||
|
content.push_str(&format!(
|
||||||
|
"Adding playlist: [{}]({})\n",
|
||||||
|
&first_track
|
||||||
|
.playlist
|
||||||
|
.clone()
|
||||||
|
.unwrap_or("Unknown".to_string()),
|
||||||
|
build_playlist_url(
|
||||||
|
&first_track
|
||||||
|
.playlist_id
|
||||||
|
.clone()
|
||||||
|
.unwrap_or("Unknown".to_string())
|
||||||
|
)
|
||||||
|
));
|
||||||
|
content.push_str(&format!(
|
||||||
|
"Added {} tracks to the queue.\n",
|
||||||
|
num_tracks_added
|
||||||
|
));
|
||||||
|
let embed = EmbedBuilder::new()
|
||||||
|
.description(content)
|
||||||
|
.color(colors::BLURPLE)
|
||||||
|
.build();
|
||||||
|
vec![embed]
|
||||||
|
}
|
||||||
|
|
||||||
|
fn build_embeds(tracks: &[YouTubeTrack], tracks_added: &[TrackType]) -> Vec<Embed> {
|
||||||
|
let num_tracks_added = tracks_added.len();
|
||||||
|
match num_tracks_added {
|
||||||
|
0 => vec![],
|
||||||
|
1 => build_single_track_added_embeds(tracks_added),
|
||||||
|
_ => build_playlist_added_embeds(tracks, num_tracks_added),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
pub(crate) async fn play(
|
pub(crate) async fn play(
|
||||||
interaction: Box<InteractionCreate>,
|
interaction: Box<InteractionCreate>,
|
||||||
state: State,
|
state: State,
|
||||||
query: String,
|
query: String,
|
||||||
) -> Result<(), Box<dyn Error + Send + Sync + 'static>> {
|
) -> Result<(), Box<dyn Error + Send + Sync + 'static>> {
|
||||||
debug!(
|
tracing::info!(
|
||||||
"play command in channel {:?} by {:?}",
|
"play command in channel {:?} by {:?}",
|
||||||
interaction.channel,
|
interaction.channel,
|
||||||
interaction.author(),
|
interaction.author(),
|
||||||
);
|
);
|
||||||
|
match play_inner(&interaction, Arc::clone(&state), query).await {
|
||||||
|
Ok(_) => Ok(()),
|
||||||
|
Err(e) => {
|
||||||
|
tracing::debug!("Search did not result in any tracks: {}", e);
|
||||||
|
let content = "Search did not result in any tracks.".to_string();
|
||||||
|
|
||||||
|
let embeds = vec![EmbedBuilder::new()
|
||||||
|
.description(content)
|
||||||
|
.color(colors::RED)
|
||||||
|
.build()];
|
||||||
|
state
|
||||||
|
.http
|
||||||
|
.interaction(interaction.application_id)
|
||||||
|
.update_response(&interaction.token)
|
||||||
|
.embeds(Some(&embeds))
|
||||||
|
.await?;
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub(crate) async fn play_inner(
|
||||||
|
interaction: &InteractionCreate,
|
||||||
|
state: State,
|
||||||
|
query: String,
|
||||||
|
) -> Result<(), Box<dyn Error + Send + Sync + 'static>> {
|
||||||
|
tracing::info!(
|
||||||
|
"play_inner in channel {:?} by {:?}",
|
||||||
|
interaction.channel,
|
||||||
|
interaction.author(),
|
||||||
|
);
|
||||||
|
|
||||||
let content = format!("Adding track(s) to the queue: {}", query);
|
let content = format!("Adding track(s) to the queue: {}", query);
|
||||||
|
tracing::info!("content: {:?}", content);
|
||||||
let embeds = vec![EmbedBuilder::new()
|
let embeds = vec![EmbedBuilder::new()
|
||||||
.description(content)
|
.description(content)
|
||||||
.color(colors::YELLOW)
|
.color(colors::YELLOW)
|
||||||
@@ -113,25 +320,10 @@ pub(crate) async fn play(
|
|||||||
query
|
query
|
||||||
};
|
};
|
||||||
|
|
||||||
debug!("query: {:?}", query);
|
tracing::info!("query: {:?}", query);
|
||||||
|
|
||||||
let tracks = match get_tracks(query).await {
|
let tracks = get_tracks(query).await?;
|
||||||
Err(e) => {
|
tracing::info!("got tracks: {:?}", tracks);
|
||||||
let content = format!("{}", e);
|
|
||||||
let embeds = vec![EmbedBuilder::new()
|
|
||||||
.description(content)
|
|
||||||
.color(colors::RED)
|
|
||||||
.build()];
|
|
||||||
state
|
|
||||||
.http
|
|
||||||
.interaction(interaction.application_id)
|
|
||||||
.update_response(&interaction.token)
|
|
||||||
.embeds(Some(&embeds))?
|
|
||||||
.await?;
|
|
||||||
return Ok(());
|
|
||||||
}
|
|
||||||
Ok(tracks) => tracks,
|
|
||||||
};
|
|
||||||
|
|
||||||
if tracks.len() > 1 {
|
if tracks.len() > 1 {
|
||||||
let first_track = tracks.first().unwrap();
|
let first_track = tracks.first().unwrap();
|
||||||
@@ -156,21 +348,14 @@ pub(crate) async fn play(
|
|||||||
.http
|
.http
|
||||||
.interaction(interaction.application_id)
|
.interaction(interaction.application_id)
|
||||||
.update_response(&interaction.token)
|
.update_response(&interaction.token)
|
||||||
.embeds(Some(&embeds))?
|
.embeds(Some(&embeds))
|
||||||
.await?;
|
.await
|
||||||
|
.context("Could not send playlist loading message")?;
|
||||||
}
|
}
|
||||||
|
|
||||||
if let Some(call_lock) = state.songbird.get(guild_id) {
|
if let Some(call_lock) = state.songbird.get(guild_id) {
|
||||||
let call = call_lock.lock().await;
|
let call = call_lock.lock().await;
|
||||||
call.queue().resume()?;
|
call.queue().resume().context("Could not resume playing")?;
|
||||||
}
|
|
||||||
|
|
||||||
struct TrackType {
|
|
||||||
url: String,
|
|
||||||
title: Option<String>,
|
|
||||||
duration_string: String,
|
|
||||||
channel: String,
|
|
||||||
thumbnail: Option<String>,
|
|
||||||
}
|
}
|
||||||
|
|
||||||
let mut tracks_added = vec![];
|
let mut tracks_added = vec![];
|
||||||
@@ -180,172 +365,86 @@ pub(crate) async fn play(
|
|||||||
.original_url
|
.original_url
|
||||||
.clone()
|
.clone()
|
||||||
.or(yttrack.url.clone())
|
.or(yttrack.url.clone())
|
||||||
.ok_or("Could not find url")?;
|
.context("Could not find url")?;
|
||||||
|
|
||||||
let mut src = YoutubeDl::new(reqwest::Client::new(), url.clone());
|
let mut src = YoutubeDl::new(state.client.clone(), url.clone());
|
||||||
let src_copy = src.clone();
|
|
||||||
let src_copy2 = src.clone();
|
|
||||||
let _m = Memory::new(src_copy2.into());
|
|
||||||
let track: Track = src_copy.into();
|
|
||||||
|
|
||||||
if let Ok(metadata) = src.aux_metadata().await {
|
match src.aux_metadata().await {
|
||||||
debug!("metadata: {:?}", metadata);
|
Ok(metadata) => {
|
||||||
|
debug!("metadata: {:?}", metadata);
|
||||||
|
|
||||||
let (author_name, author_global_name) = if let Some(author) = interaction.author() {
|
let track: Track = Track::new_with_data(
|
||||||
(author.name.clone(), author.global_name.clone())
|
src.clone().into(),
|
||||||
} else {
|
Arc::new(Metadata {
|
||||||
("".to_string(), None)
|
title: metadata.title.clone(),
|
||||||
};
|
duration: metadata.duration,
|
||||||
|
url: url.clone(),
|
||||||
db::track::insert_guild(&state.pool, db::track::Guild::new(guild_id.to_string()))
|
src,
|
||||||
.await
|
|
||||||
.expect("failed to insert guild: {e}");
|
|
||||||
|
|
||||||
db::track::insert_user(
|
|
||||||
&state.pool,
|
|
||||||
db::track::User::new(user_id.to_string(), author_name, author_global_name),
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
.expect("failed to insert user: {e}");
|
|
||||||
|
|
||||||
let track_id = db::track::insert_track(
|
|
||||||
&state.pool,
|
|
||||||
db::track::Track::new(
|
|
||||||
url.clone(),
|
|
||||||
yttrack.title.clone(),
|
|
||||||
yttrack.channel.clone(),
|
|
||||||
yttrack.duration_string.clone(),
|
|
||||||
metadata.thumbnail.clone().unwrap_or_default(),
|
|
||||||
),
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
.expect("failed to insert track: {e}");
|
|
||||||
|
|
||||||
db::track::insert_query(
|
|
||||||
&state.pool,
|
|
||||||
db::track::Query::new(user_id.to_string(), guild_id.to_string(), track_id),
|
|
||||||
)
|
|
||||||
.await
|
|
||||||
.expect("failed to insert track: {e}");
|
|
||||||
|
|
||||||
tracks_added.push(TrackType {
|
|
||||||
url: url.clone(),
|
|
||||||
title: metadata.title.clone(),
|
|
||||||
duration_string: yttrack.duration_string.clone(),
|
|
||||||
channel: yttrack.channel.clone(),
|
|
||||||
thumbnail: metadata.thumbnail.clone(),
|
|
||||||
});
|
|
||||||
|
|
||||||
if let Some(call_lock) = state.songbird.get(guild_id) {
|
|
||||||
let mut call = call_lock.lock().await;
|
|
||||||
let handle = call.enqueue_with_preload(
|
|
||||||
track,
|
|
||||||
metadata.duration.map(|duration| -> Duration {
|
|
||||||
if duration.as_secs() > 5 {
|
|
||||||
duration.sub(Duration::from_secs(5))
|
|
||||||
} else {
|
|
||||||
duration
|
|
||||||
}
|
|
||||||
}),
|
}),
|
||||||
);
|
);
|
||||||
let mut x = handle.typemap().write().await;
|
|
||||||
x.insert::<MetadataMap>(Metadata {
|
persistence(interaction, yttrack, Arc::clone(&state))
|
||||||
title: metadata.title,
|
.await
|
||||||
duration: metadata.duration,
|
.unwrap_or_else(|e| {
|
||||||
url,
|
tracing::error!("could not persist track: {:?}", e);
|
||||||
src,
|
});
|
||||||
|
|
||||||
|
tracks_added.push(TrackType {
|
||||||
|
url: url.clone(),
|
||||||
|
title: metadata.title.clone(),
|
||||||
|
duration_string: yttrack.duration_string.clone(),
|
||||||
|
channel: yttrack.channel.clone(),
|
||||||
|
thumbnail: metadata.thumbnail.clone(),
|
||||||
});
|
});
|
||||||
|
|
||||||
|
match state.songbird.get(guild_id) {
|
||||||
|
Some(call_lock) => {
|
||||||
|
let mut call = call_lock.lock().await;
|
||||||
|
let _handle = call.enqueue_with_preload(
|
||||||
|
track,
|
||||||
|
metadata.duration.map(|duration| -> Duration {
|
||||||
|
if duration.as_secs() > 5 {
|
||||||
|
duration.sub(Duration::from_secs(5))
|
||||||
|
} else {
|
||||||
|
duration
|
||||||
|
}
|
||||||
|
}),
|
||||||
|
);
|
||||||
|
}
|
||||||
|
None => tracing::error!("could not get call lock"),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Err(e) => {
|
||||||
|
tracing::error!("could not get metadata: {:?}", e);
|
||||||
|
if e.to_string()
|
||||||
|
.contains("Sign in to confirm you’re not a bot.")
|
||||||
|
{
|
||||||
|
let content =
|
||||||
|
"I seem to have been flagged by YouTube as a bot. :-(".to_string();
|
||||||
|
let embeds = vec![EmbedBuilder::new()
|
||||||
|
.description(content)
|
||||||
|
.color(colors::RED)
|
||||||
|
.build()];
|
||||||
|
state
|
||||||
|
.http
|
||||||
|
.interaction(interaction.application_id)
|
||||||
|
.update_response(&interaction.token)
|
||||||
|
.embeds(Some(&embeds))
|
||||||
|
.await?;
|
||||||
|
return Ok(());
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
let mut content = String::new();
|
|
||||||
let num_tracks_added = tracks_added.len();
|
|
||||||
let embeds = match num_tracks_added {
|
|
||||||
0 => {
|
|
||||||
vec![]
|
|
||||||
}
|
|
||||||
1 => {
|
|
||||||
let track = tracks_added.first().unwrap();
|
|
||||||
|
|
||||||
let host = Url::parse(&track.url)?;
|
|
||||||
let host = host
|
|
||||||
.host_str()
|
|
||||||
.unwrap_or_default()
|
|
||||||
.trim_start_matches("www.");
|
|
||||||
let mut embed = EmbedBuilder::new()
|
|
||||||
.author(EmbedAuthor {
|
|
||||||
name: "🔊 Added to queue".to_string(),
|
|
||||||
icon_url: None,
|
|
||||||
proxy_icon_url: None,
|
|
||||||
url: None,
|
|
||||||
})
|
|
||||||
.title(track.title.clone().unwrap_or("Unknown".to_string()))
|
|
||||||
.url(track.url.clone())
|
|
||||||
.color(colors::BLURPLE)
|
|
||||||
.footer(EmbedFooter {
|
|
||||||
text: format!("Streaming from {}", host),
|
|
||||||
icon_url: Some(format!(
|
|
||||||
"https://www.google.com/s2/favicons?domain={}",
|
|
||||||
host
|
|
||||||
)),
|
|
||||||
proxy_icon_url: None,
|
|
||||||
})
|
|
||||||
.field(EmbedField {
|
|
||||||
inline: true,
|
|
||||||
name: "Duration".to_string(),
|
|
||||||
value: track.duration_string.clone(),
|
|
||||||
})
|
|
||||||
.field(EmbedField {
|
|
||||||
inline: true,
|
|
||||||
name: "Channel".to_string(),
|
|
||||||
value: track.channel.clone(),
|
|
||||||
})
|
|
||||||
.build();
|
|
||||||
|
|
||||||
if let Some(thumbnail) = &track.thumbnail {
|
|
||||||
embed.thumbnail = Some(EmbedThumbnail {
|
|
||||||
height: None,
|
|
||||||
proxy_url: None,
|
|
||||||
url: thumbnail.to_string(),
|
|
||||||
width: None,
|
|
||||||
});
|
|
||||||
}
|
|
||||||
|
|
||||||
vec![embed]
|
|
||||||
}
|
|
||||||
_ => {
|
|
||||||
let first_track = tracks.first().unwrap();
|
|
||||||
content.push_str(&format!(
|
|
||||||
"Adding playlist: [{}]({})\n",
|
|
||||||
&first_track
|
|
||||||
.playlist
|
|
||||||
.clone()
|
|
||||||
.unwrap_or("Unknown".to_string()),
|
|
||||||
build_playlist_url(
|
|
||||||
&first_track
|
|
||||||
.playlist_id
|
|
||||||
.clone()
|
|
||||||
.unwrap_or("Unknown".to_string())
|
|
||||||
)
|
|
||||||
));
|
|
||||||
content.push_str(&format!(
|
|
||||||
"Added {} tracks to the queue:\n",
|
|
||||||
num_tracks_added
|
|
||||||
));
|
|
||||||
let embed = EmbedBuilder::new()
|
|
||||||
.description(content)
|
|
||||||
.color(colors::BLURPLE)
|
|
||||||
.build();
|
|
||||||
vec![embed]
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
|
let embeds = build_embeds(&tracks, &tracks_added);
|
||||||
state
|
state
|
||||||
.http
|
.http
|
||||||
.interaction(interaction.application_id)
|
.interaction(interaction.application_id)
|
||||||
.update_response(&interaction.token)
|
.update_response(&interaction.token)
|
||||||
.embeds(Some(&embeds))?
|
.embeds(Some(&embeds))
|
||||||
.await?;
|
.await
|
||||||
|
.context("Could not send final play message")?;
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
use songbird::tracks::TrackHandle;
|
use songbird::tracks::TrackHandle;
|
||||||
use twilight_model::channel::message::component::{ActionRow, Button, ButtonStyle};
|
use twilight_model::channel::message::component::{ActionRow, Button, ButtonStyle};
|
||||||
use twilight_model::channel::message::{Component, Embed, MessageFlags, ReactionType};
|
use twilight_model::channel::message::{Component, Embed, EmojiReactionType, MessageFlags};
|
||||||
use twilight_model::gateway::payload::incoming::InteractionCreate;
|
use twilight_model::gateway::payload::incoming::InteractionCreate;
|
||||||
use twilight_model::http::interaction::InteractionResponse;
|
use twilight_model::http::interaction::InteractionResponse;
|
||||||
use twilight_model::http::interaction::InteractionResponseType;
|
use twilight_model::http::interaction::InteractionResponseType;
|
||||||
@@ -8,7 +8,8 @@ use twilight_util::builder::embed::EmbedBuilder;
|
|||||||
use twilight_util::builder::InteractionResponseDataBuilder;
|
use twilight_util::builder::InteractionResponseDataBuilder;
|
||||||
|
|
||||||
use crate::colors;
|
use crate::colors;
|
||||||
use crate::{metadata::MetadataMap, state::State};
|
use crate::metadata::Metadata;
|
||||||
|
use crate::state::State;
|
||||||
use std::error::Error;
|
use std::error::Error;
|
||||||
|
|
||||||
pub(crate) const TRACKS_PER_PAGE: usize = 5;
|
pub(crate) const TRACKS_PER_PAGE: usize = 5;
|
||||||
@@ -38,10 +39,7 @@ pub(crate) async fn build_queue_embeds(queue: &[TrackHandle], page: usize) -> Ve
|
|||||||
.skip(TRACKS_PER_PAGE * page)
|
.skip(TRACKS_PER_PAGE * page)
|
||||||
.take(TRACKS_PER_PAGE)
|
.take(TRACKS_PER_PAGE)
|
||||||
{
|
{
|
||||||
let map = track.typemap().read().await;
|
let metadata = track.data::<Metadata>();
|
||||||
let metadata = map
|
|
||||||
.get::<MetadataMap>()
|
|
||||||
.expect("Could not get metadata map");
|
|
||||||
message.push_str(
|
message.push_str(
|
||||||
format!(
|
format!(
|
||||||
"* [{}]({})",
|
"* [{}]({})",
|
||||||
@@ -59,7 +57,7 @@ pub(crate) async fn build_queue_embeds(queue: &[TrackHandle], page: usize) -> Ve
|
|||||||
}
|
}
|
||||||
message.push('\n');
|
message.push('\n');
|
||||||
|
|
||||||
let n_pages = (queue.len() + TRACKS_PER_PAGE - 1) / TRACKS_PER_PAGE;
|
let n_pages = queue.len().div_ceil(TRACKS_PER_PAGE);
|
||||||
if n_pages > 1 {
|
if n_pages > 1 {
|
||||||
message.push_str(&format!("page {}/{}", 1 + page, n_pages));
|
message.push_str(&format!("page {}/{}", 1 + page, n_pages));
|
||||||
}
|
}
|
||||||
@@ -76,31 +74,34 @@ pub(crate) fn build_action_row(page: usize, n_pages: usize) -> Vec<Component> {
|
|||||||
custom_id: Some(format!("page:{}", page as i32 - 1)),
|
custom_id: Some(format!("page:{}", page as i32 - 1)),
|
||||||
style: ButtonStyle::Primary,
|
style: ButtonStyle::Primary,
|
||||||
label: Some("Previous page".to_string()),
|
label: Some("Previous page".to_string()),
|
||||||
emoji: Some(ReactionType::Unicode {
|
emoji: Some(EmojiReactionType::Unicode {
|
||||||
name: "⬅️".to_string(),
|
name: "⬅️".to_string(),
|
||||||
}),
|
}),
|
||||||
url: None,
|
url: None,
|
||||||
disabled: page == 0,
|
disabled: page == 0,
|
||||||
|
sku_id: None,
|
||||||
}),
|
}),
|
||||||
Component::Button(Button {
|
Component::Button(Button {
|
||||||
custom_id: Some(format!("page:{}", page)),
|
custom_id: Some(format!("page:{}", page)),
|
||||||
style: ButtonStyle::Primary,
|
style: ButtonStyle::Primary,
|
||||||
label: Some("Refresh".to_string()),
|
label: Some("Refresh".to_string()),
|
||||||
emoji: Some(ReactionType::Unicode {
|
emoji: Some(EmojiReactionType::Unicode {
|
||||||
name: "🔄".to_string(),
|
name: "🔄".to_string(),
|
||||||
}),
|
}),
|
||||||
url: None,
|
url: None,
|
||||||
disabled: false,
|
disabled: false,
|
||||||
|
sku_id: None,
|
||||||
}),
|
}),
|
||||||
Component::Button(Button {
|
Component::Button(Button {
|
||||||
custom_id: Some(format!("page:{}", page + 1)),
|
custom_id: Some(format!("page:{}", page + 1)),
|
||||||
style: ButtonStyle::Primary,
|
style: ButtonStyle::Primary,
|
||||||
label: Some("Next page".to_string()),
|
label: Some("Next page".to_string()),
|
||||||
emoji: Some(ReactionType::Unicode {
|
emoji: Some(EmojiReactionType::Unicode {
|
||||||
name: "➡️".to_string(),
|
name: "➡️".to_string(),
|
||||||
}),
|
}),
|
||||||
url: None,
|
url: None,
|
||||||
disabled: page >= n_pages - 1,
|
disabled: page >= n_pages - 1,
|
||||||
|
sku_id: None,
|
||||||
}),
|
}),
|
||||||
],
|
],
|
||||||
})]
|
})]
|
||||||
@@ -146,15 +147,15 @@ pub(crate) async fn queue(
|
|||||||
}
|
}
|
||||||
|
|
||||||
let embeds = build_queue_embeds(&queue, 0).await;
|
let embeds = build_queue_embeds(&queue, 0).await;
|
||||||
let n_pages = (queue.len() + TRACKS_PER_PAGE - 1) / TRACKS_PER_PAGE;
|
let n_pages = queue.len().div_ceil(TRACKS_PER_PAGE);
|
||||||
let action_row = build_action_row(0, n_pages);
|
let action_row = build_action_row(0, n_pages);
|
||||||
|
|
||||||
state
|
state
|
||||||
.http
|
.http
|
||||||
.interaction(interaction.application_id)
|
.interaction(interaction.application_id)
|
||||||
.update_response(&interaction.token)
|
.update_response(&interaction.token)
|
||||||
.embeds(Some(&embeds))?
|
.embeds(Some(&embeds))
|
||||||
.components(Some(&action_row))?
|
.components(Some(&action_row))
|
||||||
.await?;
|
.await?;
|
||||||
|
|
||||||
Ok(())
|
Ok(())
|
||||||
|
|||||||
@@ -36,18 +36,29 @@ pub(crate) async fn insert_track(
|
|||||||
pool: &sqlx::SqlitePool,
|
pool: &sqlx::SqlitePool,
|
||||||
track: Track,
|
track: Track,
|
||||||
) -> Result<i64, sqlx::Error> {
|
) -> Result<i64, sqlx::Error> {
|
||||||
let query =
|
let query = r#"
|
||||||
"INSERT OR REPLACE INTO tracks (url, title, channel, duration, thumbnail, updated) VALUES ($1, $2, $3, $4, $5, $6)";
|
INSERT INTO tracks (url, title, channel, duration, thumbnail, updated)
|
||||||
let res = sqlx::query(query)
|
VALUES (?, ?, ?, ?, ?, ?)
|
||||||
.bind(track.url)
|
ON CONFLICT (url) DO UPDATE SET
|
||||||
|
title = EXCLUDED.title,
|
||||||
|
channel = EXCLUDED.channel,
|
||||||
|
duration = EXCLUDED.duration,
|
||||||
|
thumbnail = EXCLUDED.thumbnail,
|
||||||
|
updated = EXCLUDED.updated
|
||||||
|
RETURNING id
|
||||||
|
"#;
|
||||||
|
|
||||||
|
let id = sqlx::query_scalar(query)
|
||||||
|
.bind(&track.url)
|
||||||
.bind(track.title)
|
.bind(track.title)
|
||||||
.bind(track.channel)
|
.bind(track.channel)
|
||||||
.bind(track.duration)
|
.bind(track.duration)
|
||||||
.bind(track.thumbnail)
|
.bind(track.thumbnail)
|
||||||
.bind(track.updated)
|
.bind(track.updated)
|
||||||
.execute(pool)
|
.fetch_one(pool)
|
||||||
.await?;
|
.await?;
|
||||||
Ok(res.last_insert_rowid())
|
|
||||||
|
Ok(id)
|
||||||
}
|
}
|
||||||
|
|
||||||
#[derive(Debug, FromRow)]
|
#[derive(Debug, FromRow)]
|
||||||
@@ -70,8 +81,14 @@ impl User {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub(crate) async fn insert_user(pool: &sqlx::SqlitePool, user: User) -> Result<(), sqlx::Error> {
|
pub(crate) async fn insert_user(pool: &sqlx::SqlitePool, user: User) -> Result<(), sqlx::Error> {
|
||||||
let query =
|
let query = r#"
|
||||||
"INSERT OR REPLACE INTO users (id, name, global_name, updated) VALUES ($1, $2, $3, $4)";
|
INSERT INTO users (id, name, global_name, updated)
|
||||||
|
VALUES (?, ?, ?, ?)
|
||||||
|
ON CONFLICT (id) DO UPDATE SET
|
||||||
|
name = EXCLUDED.name,
|
||||||
|
global_name = EXCLUDED.global_name,
|
||||||
|
updated = EXCLUDED.updated
|
||||||
|
"#;
|
||||||
sqlx::query(query)
|
sqlx::query(query)
|
||||||
.bind(user.id)
|
.bind(user.id)
|
||||||
.bind(user.name)
|
.bind(user.name)
|
||||||
@@ -105,8 +122,10 @@ impl Query {
|
|||||||
}
|
}
|
||||||
|
|
||||||
pub(crate) async fn insert_query(pool: &sqlx::SqlitePool, q: Query) -> Result<i64, sqlx::Error> {
|
pub(crate) async fn insert_query(pool: &sqlx::SqlitePool, q: Query) -> Result<i64, sqlx::Error> {
|
||||||
let query =
|
let query = r#"
|
||||||
"INSERT OR REPLACE INTO queries (user_id, guild_id, track_id, updated) VALUES ($1, $2, $3, $4)";
|
INSERT INTO queries (user_id, guild_id, track_id, updated)
|
||||||
|
VALUES (?, ?, ?, ?)
|
||||||
|
"#;
|
||||||
let res = sqlx::query(query)
|
let res = sqlx::query(query)
|
||||||
.bind(q.user_id)
|
.bind(q.user_id)
|
||||||
.bind(q.guild_id)
|
.bind(q.guild_id)
|
||||||
@@ -132,15 +151,17 @@ impl Guild {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pub(crate) async fn insert_guild(
|
pub(crate) async fn insert_guild(pool: &sqlx::SqlitePool, guild: Guild) -> Result<(), sqlx::Error> {
|
||||||
pool: &sqlx::SqlitePool,
|
let query = r#"
|
||||||
guild: Guild,
|
INSERT INTO guilds (id, updated)
|
||||||
) -> Result<i64, sqlx::Error> {
|
VALUES (?, ?)
|
||||||
let query = "INSERT OR REPLACE INTO guilds (id, updated) VALUES ($1, $2)";
|
ON CONFLICT (id) DO UPDATE SET
|
||||||
let res = sqlx::query(query)
|
updated = EXCLUDED.updated
|
||||||
|
"#;
|
||||||
|
sqlx::query(query)
|
||||||
.bind(guild.id)
|
.bind(guild.id)
|
||||||
.bind(guild.updated)
|
.bind(guild.updated)
|
||||||
.execute(pool)
|
.execute(pool)
|
||||||
.await?;
|
.await?;
|
||||||
Ok(res.last_insert_rowid())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -14,6 +14,7 @@ use twilight_model::gateway::payload::incoming::InteractionCreate;
|
|||||||
use twilight_model::http::interaction::{InteractionResponse, InteractionResponseType};
|
use twilight_model::http::interaction::{InteractionResponse, InteractionResponseType};
|
||||||
use twilight_util::builder::InteractionResponseDataBuilder;
|
use twilight_util::builder::InteractionResponseDataBuilder;
|
||||||
|
|
||||||
|
#[derive(Debug, Clone)]
|
||||||
pub(crate) struct Handler {
|
pub(crate) struct Handler {
|
||||||
state: State,
|
state: State,
|
||||||
}
|
}
|
||||||
@@ -112,7 +113,7 @@ impl Handler {
|
|||||||
let call = call_lock.lock().await;
|
let call = call_lock.lock().await;
|
||||||
queue = call.queue().current_queue();
|
queue = call.queue().current_queue();
|
||||||
}
|
}
|
||||||
let n_pages = (queue.len() + TRACKS_PER_PAGE - 1) / TRACKS_PER_PAGE;
|
let n_pages = queue.len().div_ceil(TRACKS_PER_PAGE);
|
||||||
let page = page.min(n_pages - 1).max(0);
|
let page = page.min(n_pages - 1).max(0);
|
||||||
let embeds = build_queue_embeds(&queue, page).await;
|
let embeds = build_queue_embeds(&queue, page).await;
|
||||||
let action_row = build_action_row(page, n_pages);
|
let action_row = build_action_row(page, n_pages);
|
||||||
|
|||||||
113
src/main.rs
113
src/main.rs
@@ -12,18 +12,12 @@ mod utils;
|
|||||||
|
|
||||||
use crate::commands::get_chat_commands;
|
use crate::commands::get_chat_commands;
|
||||||
use dotenv::dotenv;
|
use dotenv::dotenv;
|
||||||
use futures::StreamExt;
|
|
||||||
use signal::signal_handler;
|
|
||||||
use songbird::{shards::TwilightMap, Songbird};
|
use songbird::{shards::TwilightMap, Songbird};
|
||||||
use state::StateRef;
|
use state::StateRef;
|
||||||
use std::{env, error::Error, sync::Arc};
|
use std::{env, error::Error, str::FromStr, sync::Arc, time::Duration};
|
||||||
use tokio::select;
|
|
||||||
use tracing::{debug, info};
|
use tracing::{debug, info};
|
||||||
use twilight_cache_inmemory::InMemoryCache;
|
use twilight_cache_inmemory::InMemoryCache;
|
||||||
use twilight_gateway::{
|
use twilight_gateway::{Event, EventTypeFlags, Intents, Shard, StreamExt as _};
|
||||||
stream::{self, ShardEventStream},
|
|
||||||
Intents, Shard,
|
|
||||||
};
|
|
||||||
use twilight_http::Client as HttpClient;
|
use twilight_http::Client as HttpClient;
|
||||||
use twilight_model::id::Id;
|
use twilight_model::id::Id;
|
||||||
use twilight_standby::Standby;
|
use twilight_standby::Standby;
|
||||||
@@ -34,16 +28,15 @@ async fn main() -> Result<(), Box<dyn Error + Send + Sync + 'static>> {
|
|||||||
|
|
||||||
println!("Starting up...");
|
println!("Starting up...");
|
||||||
|
|
||||||
// Initialize the tracing subscriber.
|
|
||||||
tracing_subscriber::fmt::init();
|
tracing_subscriber::fmt::init();
|
||||||
|
|
||||||
info!("Starting up...");
|
info!("Starting up...");
|
||||||
|
|
||||||
let (mut shards, state) = {
|
let (shards, state) = {
|
||||||
let db = env::var("DATABASE_URL").map_err(|_| "DATABASE_URL is not set")?;
|
let db = env::var("DATABASE_URL").map_err(|_| "DATABASE_URL is not set")?;
|
||||||
let options = SqliteConnectOptions::new()
|
let options = SqliteConnectOptions::from_str(&db)
|
||||||
.create_if_missing(true)
|
.expect("could not create options")
|
||||||
.filename(&db);
|
.create_if_missing(true);
|
||||||
let pool = SqlitePoolOptions::new()
|
let pool = SqlitePoolOptions::new()
|
||||||
.max_connections(5)
|
.max_connections(5)
|
||||||
.connect_with(options)
|
.connect_with(options)
|
||||||
@@ -72,7 +65,7 @@ async fn main() -> Result<(), Box<dyn Error + Send + Sync + 'static>> {
|
|||||||
| Intents::MESSAGE_CONTENT;
|
| Intents::MESSAGE_CONTENT;
|
||||||
let config = twilight_gateway::Config::new(token.clone(), intents);
|
let config = twilight_gateway::Config::new(token.clone(), intents);
|
||||||
let shards: Vec<Shard> =
|
let shards: Vec<Shard> =
|
||||||
stream::create_recommended(&http, config, |_, builder| builder.build())
|
twilight_gateway::create_recommended(&http, config, |_, builder| builder.build())
|
||||||
.await?
|
.await?
|
||||||
.collect();
|
.collect();
|
||||||
let senders = TwilightMap::new(
|
let senders = TwilightMap::new(
|
||||||
@@ -83,6 +76,11 @@ async fn main() -> Result<(), Box<dyn Error + Send + Sync + 'static>> {
|
|||||||
);
|
);
|
||||||
let songbird = Songbird::twilight(Arc::new(senders), user_id);
|
let songbird = Songbird::twilight(Arc::new(senders), user_id);
|
||||||
let cache = InMemoryCache::new();
|
let cache = InMemoryCache::new();
|
||||||
|
let client = reqwest::ClientBuilder::new()
|
||||||
|
.connect_timeout(Duration::from_secs(10))
|
||||||
|
.timeout(Duration::from_secs(3600))
|
||||||
|
.build()
|
||||||
|
.expect("could not build http client");
|
||||||
|
|
||||||
(
|
(
|
||||||
shards,
|
shards,
|
||||||
@@ -93,6 +91,7 @@ async fn main() -> Result<(), Box<dyn Error + Send + Sync + 'static>> {
|
|||||||
standby: Standby::new(),
|
standby: Standby::new(),
|
||||||
guild_settings: Default::default(),
|
guild_settings: Default::default(),
|
||||||
pool,
|
pool,
|
||||||
|
client,
|
||||||
}),
|
}),
|
||||||
)
|
)
|
||||||
};
|
};
|
||||||
@@ -100,45 +99,55 @@ async fn main() -> Result<(), Box<dyn Error + Send + Sync + 'static>> {
|
|||||||
info!("Ready to receive events");
|
info!("Ready to receive events");
|
||||||
|
|
||||||
let handler = Handler::new(Arc::clone(&state));
|
let handler = Handler::new(Arc::clone(&state));
|
||||||
let mut stop_rx = signal_handler();
|
// let mut stop_rx = signal_handler();
|
||||||
let mut stream = ShardEventStream::new(shards.iter_mut());
|
let mut set = tokio::task::JoinSet::new();
|
||||||
loop {
|
|
||||||
select! {
|
|
||||||
biased;
|
|
||||||
_ = stop_rx.changed() => {
|
|
||||||
for guild in state.cache.iter().guilds() {
|
|
||||||
if let Some(user) = state.cache.current_user() {
|
|
||||||
if state.cache.voice_state(user.id, guild.id()).is_some() {
|
|
||||||
debug!("Leaving guild {:?}", guild.id());
|
|
||||||
state.songbird.leave(guild.id()).await?;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// need to grab next event to properly leave voice channels
|
|
||||||
stream.next().await;
|
|
||||||
break;
|
|
||||||
},
|
|
||||||
next = stream.next() => {
|
|
||||||
let event = match next {
|
|
||||||
Some((_, Ok(event))) => event,
|
|
||||||
Some((_, Err(source))) => {
|
|
||||||
tracing::warn!(?source, "error receiving event");
|
|
||||||
if source.is_fatal() {
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
continue;
|
|
||||||
}
|
|
||||||
None => break,
|
|
||||||
};
|
|
||||||
debug!("Event: {:?}", &event);
|
|
||||||
|
|
||||||
state.cache.update(&event);
|
for shard in shards {
|
||||||
state.standby.process(&event);
|
set.spawn(tokio::spawn(runner(shard, handler.clone(), state.clone())));
|
||||||
state.songbird.process(&event).await;
|
|
||||||
|
|
||||||
handler.act(event).await?;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
set.join_next().await;
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn runner(mut shard: Shard, handler: Handler, state: Arc<StateRef>) {
|
||||||
|
while let Some(item) = shard.next_event(EventTypeFlags::all()).await {
|
||||||
|
let event = match item {
|
||||||
|
Ok(event) => event,
|
||||||
|
Err(source) => {
|
||||||
|
tracing::warn!(?source, "error receiving event");
|
||||||
|
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
|
tokio::spawn({
|
||||||
|
let state = state.clone();
|
||||||
|
let handler = handler.clone();
|
||||||
|
async move {
|
||||||
|
handle_event(event, handler, state)
|
||||||
|
.await
|
||||||
|
.unwrap_or_else(|source| {
|
||||||
|
tracing::warn!(?source, "error handling event");
|
||||||
|
});
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn handle_event(
|
||||||
|
event: Event,
|
||||||
|
handler: Handler,
|
||||||
|
state: Arc<StateRef>,
|
||||||
|
) -> Result<(), Box<dyn Error>> {
|
||||||
|
state.standby.process(&event);
|
||||||
|
state.songbird.process(&event).await;
|
||||||
|
debug!("Event: {:?}", &event);
|
||||||
|
|
||||||
|
state.cache.update(&event);
|
||||||
|
state.standby.process(&event);
|
||||||
|
state.songbird.process(&event).await;
|
||||||
|
|
||||||
|
handler.act(event).await?;
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
use songbird::{input::YoutubeDl, typemap::TypeMapKey};
|
use songbird::input::YoutubeDl;
|
||||||
use std::time::Duration;
|
use std::time::Duration;
|
||||||
|
|
||||||
#[derive(Clone)]
|
#[derive(Clone)]
|
||||||
@@ -6,10 +6,5 @@ pub(crate) struct Metadata {
|
|||||||
pub(crate) title: Option<String>,
|
pub(crate) title: Option<String>,
|
||||||
pub(crate) duration: Option<Duration>,
|
pub(crate) duration: Option<Duration>,
|
||||||
pub(crate) url: String,
|
pub(crate) url: String,
|
||||||
pub(crate) src: YoutubeDl,
|
pub(crate) src: YoutubeDl<'static>,
|
||||||
}
|
|
||||||
|
|
||||||
pub(crate) struct MetadataMap;
|
|
||||||
impl TypeMapKey for MetadataMap {
|
|
||||||
type Value = Metadata;
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,21 +1,21 @@
|
|||||||
use tokio::{
|
// use tokio::{
|
||||||
select,
|
// select,
|
||||||
signal::unix::{signal, SignalKind},
|
// signal::unix::{signal, SignalKind},
|
||||||
sync::watch,
|
// sync::watch,
|
||||||
};
|
// };
|
||||||
|
|
||||||
pub(crate) fn signal_handler() -> watch::Receiver<()> {
|
// pub(crate) fn signal_handler() -> watch::Receiver<()> {
|
||||||
let (stop_tx, stop_rx) = watch::channel(());
|
// let (stop_tx, stop_rx) = watch::channel(());
|
||||||
tokio::spawn(async move {
|
// tokio::spawn(async move {
|
||||||
let mut sigterm = signal(SignalKind::terminate()).unwrap();
|
// let mut sigterm = signal(SignalKind::terminate()).unwrap();
|
||||||
let mut sigint = signal(SignalKind::interrupt()).unwrap();
|
// let mut sigint = signal(SignalKind::interrupt()).unwrap();
|
||||||
loop {
|
// loop {
|
||||||
select! {
|
// select! {
|
||||||
_ = sigterm.recv() => println!("Receive SIGTERM"),
|
// _ = sigterm.recv() => println!("Receive SIGTERM"),
|
||||||
_ = sigint.recv() => println!("Receive SIGTERM"),
|
// _ = sigint.recv() => println!("Receive SIGTERM"),
|
||||||
};
|
// };
|
||||||
stop_tx.send(()).unwrap();
|
// stop_tx.send(()).unwrap();
|
||||||
}
|
// }
|
||||||
});
|
// });
|
||||||
stop_rx
|
// stop_rx
|
||||||
}
|
// }
|
||||||
|
|||||||
@@ -33,4 +33,5 @@ pub(crate) struct StateRef {
|
|||||||
pub(crate) standby: Standby,
|
pub(crate) standby: Standby,
|
||||||
pub(crate) guild_settings: DashMap<Id<GuildMarker>, Settings>,
|
pub(crate) guild_settings: DashMap<Id<GuildMarker>, Settings>,
|
||||||
pub(crate) pool: sqlx::SqlitePool,
|
pub(crate) pool: sqlx::SqlitePool,
|
||||||
|
pub(crate) client: reqwest::Client,
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user