mirror of
https://github.com/eRgo35/lyra.git
synced 2026-02-04 12:26:10 +01:00
Compare commits
15 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
212abd9bdd
|
|||
|
8d787f2a11
|
|||
|
486f403d10
|
|||
|
bfd301c9c2
|
|||
|
670b8dd30e
|
|||
|
c607aaf403
|
|||
|
f41fa7e09b
|
|||
|
10b7d4d7e3
|
|||
|
772d5d2a97
|
|||
|
f1b52d2be2
|
|||
|
e77818e09b
|
|||
|
5a026afc52
|
|||
|
8a0d08de4e
|
|||
|
f224ccfbdf
|
|||
|
d0af34833b
|
1
.github/FUNDING.yml
vendored
Normal file
1
.github/FUNDING.yml
vendored
Normal file
@@ -0,0 +1 @@
|
||||
github: eRgo35
|
||||
46
.github/workflows/ci.yml
vendored
Normal file
46
.github/workflows/ci.yml
vendored
Normal file
@@ -0,0 +1,46 @@
|
||||
name: ci
|
||||
|
||||
on:
|
||||
push:
|
||||
branches: [senpai]
|
||||
pull_request:
|
||||
branches: [senpai]
|
||||
|
||||
env:
|
||||
RUST_BACKTRACE: 1
|
||||
RUST_LOG: trace
|
||||
|
||||
jobs:
|
||||
test:
|
||||
name: test
|
||||
strategy:
|
||||
matrix:
|
||||
os: [ubuntu-latest]
|
||||
runs-on: ${{ matrix.os }}
|
||||
|
||||
steps:
|
||||
- uses: actions/checkout@v3
|
||||
|
||||
- name: Install Rust
|
||||
uses: actions-rs/toolchain@v1
|
||||
with:
|
||||
toolchain: stable
|
||||
profile: minimal
|
||||
override: true
|
||||
|
||||
- name: Set up cache
|
||||
uses: actions/cache@v3
|
||||
with:
|
||||
path: |
|
||||
~/.cargo/bin/
|
||||
~/.cargo/registry/index/
|
||||
~/.cargo/registry/cache/
|
||||
~/.cargo/git/db/
|
||||
target/
|
||||
key: ${{ runner.os }}-cargo-${{ hashFiles('**/Cargo.lock') }}
|
||||
|
||||
- name: Build projectable
|
||||
run: cargo build --verbose
|
||||
|
||||
- name: Run tests
|
||||
run: cargo test --verbose
|
||||
119
.github/workflows/release.yml
vendored
Normal file
119
.github/workflows/release.yml
vendored
Normal file
@@ -0,0 +1,119 @@
|
||||
name: release
|
||||
|
||||
on:
|
||||
push:
|
||||
tags:
|
||||
- "[0-9]+.[0-9]+.[0-9]+"
|
||||
|
||||
permissions: write-all
|
||||
|
||||
jobs:
|
||||
# Creates a release and outputs the url
|
||||
create-release:
|
||||
name: create-release
|
||||
runs-on: ubuntu-latest
|
||||
outputs:
|
||||
upload_url: ${{ env.RELEASE_UPLOAD_URL }}
|
||||
version: ${{ env.VERSION }}
|
||||
steps:
|
||||
- name: Get the release version from the tag
|
||||
shell: bash
|
||||
if: env.VERSION == ''
|
||||
run: |
|
||||
echo "VERSION=${GITHUB_REF#refs/tags/}" >> $GITHUB_ENV
|
||||
echo "version is: ${{ env.VERSION }}"
|
||||
- name: Create GitHub release
|
||||
uses: actions/github-script@v6
|
||||
env:
|
||||
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
||||
with:
|
||||
github-token: "${{ secrets.GITHUB_TOKEN }}"
|
||||
script: |
|
||||
try {
|
||||
const response = await github.rest.repos.createRelease({
|
||||
draft: false,
|
||||
generate_release_notes: true,
|
||||
name: process.env.VERSION,
|
||||
owner: context.repo.owner,
|
||||
prerelease: false,
|
||||
repo: context.repo.repo,
|
||||
tag_name: process.env.VERSION,
|
||||
});
|
||||
|
||||
core.exportVariable('RELEASE_ID', response.data.id);
|
||||
core.exportVariable('RELEASE_UPLOAD_URL', response.data.upload_url);
|
||||
} catch (error) {
|
||||
core.setFailed(error.message);
|
||||
}
|
||||
|
||||
build-and-upload:
|
||||
name: Build and upload
|
||||
needs: create-release
|
||||
runs-on: ${{ matrix.os }}
|
||||
|
||||
strategy:
|
||||
matrix:
|
||||
build: [linux-gnu, linux-aarch64-gnu]
|
||||
include:
|
||||
- build: linux-gnu
|
||||
os: ubuntu-latest
|
||||
target: x86_64-unknown-linux-gnu
|
||||
use-cross: true
|
||||
|
||||
- build: linux-aarch64-gnu
|
||||
os: ubuntu-latest
|
||||
target: aarch64-unknown-linux-gnu
|
||||
use-cross: true
|
||||
|
||||
steps:
|
||||
- name: Checkout
|
||||
uses: actions/checkout@v3
|
||||
|
||||
- name: Install Rust
|
||||
uses: actions-rs/toolchain@v1
|
||||
with:
|
||||
toolchain: stable
|
||||
profile: minimal
|
||||
override: true
|
||||
target: ${{ matrix.target }}
|
||||
|
||||
- name: Build
|
||||
uses: actions-rs/cargo@v1
|
||||
with:
|
||||
use-cross: ${{ matrix.use-cross }}
|
||||
command: build
|
||||
args: --verbose --release --target ${{ matrix.target }}
|
||||
|
||||
- name: Strip release binary (linux and macos)
|
||||
if: matrix.build == 'linux-gnu' || matrix.build == 'linux-musl' || matrix.build == 'macos' || matrix.build == 'linux-aarch64-gnu'
|
||||
run: strip "target/${{ matrix.target }}/release/lyra"
|
||||
|
||||
- name: Build archive
|
||||
shell: bash
|
||||
run: |
|
||||
dirname="lyra-${{ needs.create-release.outputs.version }}-${{ matrix.target }}"
|
||||
mkdir "$dirname"
|
||||
if [ "${{ matrix.os }}" = "windows-latest" ]; then
|
||||
mv "target/${{ matrix.target }}/release/lyra.exe" "$dirname"
|
||||
else
|
||||
mv "target/${{ matrix.target }}/release/lyra" "$dirname"
|
||||
fi
|
||||
mv LICENSE.md "$dirname"
|
||||
|
||||
if [ "${{ matrix.os }}" = "windows-latest" ]; then
|
||||
7z a "$dirname.zip" "$dirname"
|
||||
echo "ASSET=$dirname.zip" >> $GITHUB_ENV
|
||||
else
|
||||
tar -czf "$dirname.tar.gz" "$dirname"
|
||||
echo "ASSET=$dirname.tar.gz" >> $GITHUB_ENV
|
||||
fi
|
||||
|
||||
- name: Upload archive
|
||||
uses: actions/upload-release-asset@v1.0.1
|
||||
env:
|
||||
GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }}
|
||||
with:
|
||||
upload_url: ${{ needs.create-release.outputs.upload_url }}
|
||||
asset_path: ${{ env.ASSET }}
|
||||
asset_name: ${{ env.ASSET }}
|
||||
asset_content_type: application/octet-stream
|
||||
61
Cargo.lock
generated
61
Cargo.lock
generated
@@ -149,21 +149,6 @@ version = "0.22.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "72b3254f16251a8381aa12e40e3c4d2f0199f8c6508fbecb9d91f575e0fbb8c6"
|
||||
|
||||
[[package]]
|
||||
name = "bit-set"
|
||||
version = "0.5.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "0700ddab506f33b20a03b13996eccd309a48e5ff77d0d95926aa0210fb4e95f1"
|
||||
dependencies = [
|
||||
"bit-vec",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "bit-vec"
|
||||
version = "0.6.3"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "349f9b6a179ed607305526ca489b34ad0a41aed5f7980fa90eb03160b69598fb"
|
||||
|
||||
[[package]]
|
||||
name = "bitflags"
|
||||
version = "1.3.2"
|
||||
@@ -642,17 +627,6 @@ version = "0.1.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "af9673d8203fcb076b19dfd17e38b3d4ae9f44959416ea532ce72415a6020365"
|
||||
|
||||
[[package]]
|
||||
name = "fancy-regex"
|
||||
version = "0.13.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "531e46835a22af56d1e3b66f04844bed63158bc094a628bec1d321d9b4c44bf2"
|
||||
dependencies = [
|
||||
"bit-set",
|
||||
"regex-automata 0.4.7",
|
||||
"regex-syntax 0.8.4",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "fastrand"
|
||||
version = "2.1.0"
|
||||
@@ -1292,18 +1266,6 @@ version = "1.0.5"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "db13adb97ab515a3691f56e4dbab09283d0b86cb45abd991d8634a9d6f501760"
|
||||
|
||||
[[package]]
|
||||
name = "lib-spotify-parser"
|
||||
version = "1.0.0"
|
||||
dependencies = [
|
||||
"regex",
|
||||
"reqwest 0.12.5",
|
||||
"scraper",
|
||||
"serde",
|
||||
"serde_json",
|
||||
"tokio",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "libc"
|
||||
version = "0.2.155"
|
||||
@@ -1349,12 +1311,11 @@ dependencies = [
|
||||
|
||||
[[package]]
|
||||
name = "lyra"
|
||||
version = "0.9.0"
|
||||
version = "0.10.2"
|
||||
dependencies = [
|
||||
"dotenv",
|
||||
"fancy-regex",
|
||||
"json",
|
||||
"lib-spotify-parser",
|
||||
"once_cell",
|
||||
"openssl",
|
||||
"owoify",
|
||||
"poise",
|
||||
@@ -1365,6 +1326,7 @@ dependencies = [
|
||||
"serde_json",
|
||||
"serenity",
|
||||
"songbird",
|
||||
"spotify-parser",
|
||||
"symphonia",
|
||||
"tokio",
|
||||
"tracing",
|
||||
@@ -2719,8 +2681,7 @@ dependencies = [
|
||||
[[package]]
|
||||
name = "serenity-voice-model"
|
||||
version = "0.2.0"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "593682f6155d07c8b331b3d1060f5aab7e6796caca9f2f66bd9e6855c880e06b"
|
||||
source = "git+https://github.com/serenity-rs/serenity?branch=current#658b6a7261fbde7cf59f84613f870ef7a48735c6"
|
||||
dependencies = [
|
||||
"bitflags 2.6.0",
|
||||
"num-traits",
|
||||
@@ -2886,6 +2847,20 @@ dependencies = [
|
||||
"lock_api",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "spotify-parser"
|
||||
version = "1.0.1"
|
||||
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||
checksum = "e7b9d1d990125af02c6f611f2e8e34fd979310d43ae7ac5bd2cd357b87b13352"
|
||||
dependencies = [
|
||||
"regex",
|
||||
"reqwest 0.12.5",
|
||||
"scraper",
|
||||
"serde",
|
||||
"serde_json",
|
||||
"tokio",
|
||||
]
|
||||
|
||||
[[package]]
|
||||
name = "stable-vec"
|
||||
version = "0.4.1"
|
||||
|
||||
25
Cargo.toml
25
Cargo.toml
@@ -1,36 +1,38 @@
|
||||
[package]
|
||||
name = "lyra"
|
||||
version = "0.9.0"
|
||||
version = "0.10.2"
|
||||
authors = ["Michał Czyż <mike@c2yz.com>"]
|
||||
edition = "2021"
|
||||
description = "A featureful Discord bot written in Rust."
|
||||
documentation = "https://lyra.c2yz.com/docs"
|
||||
documentation = "https://github.com/eRgo35/lyra"
|
||||
readme = "README.md"
|
||||
homepage = "https://lyra.c2yz.com"
|
||||
license-file = "LICENSE.md"
|
||||
keywords = ["discord", "bot", "rust", "music", "featureful"]
|
||||
|
||||
|
||||
[dependencies]
|
||||
lib-spotify-parser = { path = "libs/spotify-parser" }
|
||||
spotify-parser = "1.0.1"
|
||||
dotenv = "0.15.0"
|
||||
fancy-regex = "0.13.0"
|
||||
json = "0.12.4"
|
||||
openssl = { version = "0.10.66", features = ["vendored"] }
|
||||
owoify = "0.1.5"
|
||||
poise = "0.6.1"
|
||||
poise = { default-features = true, version = "0.6.1" }
|
||||
rand = "0.8.5"
|
||||
regex = "1.10.6"
|
||||
reqwest = { version = "0.11.27", features = ["json"] }
|
||||
serde = { version = "1.0.204", features = ["derive"] }
|
||||
serde_json = "1.0.122"
|
||||
serenity = { version = "0.12.2", features = [
|
||||
songbird = { version = "0.4.3", default-features = true, features = [
|
||||
"builtin-queue",
|
||||
] }
|
||||
serenity = { default-features = true, features = [
|
||||
"cache",
|
||||
"framework",
|
||||
"standard_framework",
|
||||
"voice",
|
||||
] }
|
||||
songbird = { version = "0.4.3", features = ["builtin-queue", "serenity"] }
|
||||
"http",
|
||||
"rustls_backend",
|
||||
], version = "0.12" }
|
||||
symphonia = { version = "0.5.4", features = [
|
||||
"aac",
|
||||
"adpcm",
|
||||
@@ -44,3 +46,8 @@ tracing = "0.1.40"
|
||||
tracing-futures = "0.2.5"
|
||||
tracing-subscriber = "0.3.18"
|
||||
url = "2.5.2"
|
||||
once_cell = "1.19.0"
|
||||
|
||||
[patch.crates-io.serenity-voice-model]
|
||||
git = "https://github.com/serenity-rs/serenity"
|
||||
branch = "current"
|
||||
|
||||
75
README.md
75
README.md
@@ -1,52 +1,82 @@
|
||||
# Lyra
|
||||
|
||||

|
||||
|
||||
Lyra is a music bot written in Rust.
|
||||
|
||||
## Getting Started
|
||||
<h2 align="center">
|
||||
<a href="https://lyra.c2yz.com" target="blank_">
|
||||
<img height="256" alt="Lyra" src="assets/lyra-256.png" />
|
||||
</a>
|
||||
<br />
|
||||
Lyra: a discord music bot written in rust :crab:
|
||||
</h2>
|
||||
|
||||
Lyra is an open source, discord music bot written in Rust.
|
||||
|
||||
The idea behind this project is to allow a user to self-host one's own instance of the bot.
|
||||
User no longer has to rely on 3rd-parties to provide them a service.
|
||||
The bot can be run on a desktop, a vps or a phone as it's just a simple binary.
|
||||
|
||||
User no longer has to rely on 3rd parties to provide them an invite link.
|
||||
Currently bot is still heavily in development!
|
||||
|
||||
The bot can be run even on a desktop or a phone because after compilation, it's just a simple binary.
|
||||
## Getting started
|
||||
|
||||
As of now, the bot supports spotify url track recognition through a separate nodejs script. I plan to write the actual parser inside the bot iteself but as of now I postponed it into future release.
|
||||
Before you start, you need to create a discord bot and get a token.
|
||||
You can do it [here](https://discord.com/developers/applications).
|
||||
|
||||
Slash commands are still work in progress! Currently bot is still heavily in development!
|
||||
After you create a bot, you need to invite it to your server.
|
||||
|
||||
## Setting up
|
||||
Then, head to download section and download the latest release (or compile it yourself).
|
||||
|
||||
After you download the binary, you need to create a `.env` file in the same directory as the binary.
|
||||
Example can be found in `.env.example` file.
|
||||
|
||||
```
|
||||
DISCORD_TOKEN=<YOUR_DISCORD_TOKEN>
|
||||
PREFIX=<YOUR_PREFIX>
|
||||
```
|
||||
|
||||
DISCORD_TOKEN is the token you got from discord developers page and PREFIX is the prefix you want to use for your bot.
|
||||
|
||||
Bot by default reacts only to the prefix. To enable slash commands, while the bot is running type `:register` in the chat (where `:` is your bot prefix).
|
||||
|
||||
## Features
|
||||
|
||||
- Music playback
|
||||
- Audio effects (soon)
|
||||
- Some multipurpose commands
|
||||
- Slash commands
|
||||
- Self-hosted
|
||||
|
||||
## Compilation
|
||||
|
||||
To compile the source code on your own, you need `rust` and `cargo`
|
||||
|
||||
To run a dev version use
|
||||
To run a dev version, `cd` into the project directory and type
|
||||
|
||||
```bash
|
||||
$ cargo run
|
||||
```
|
||||
|
||||
To build a production version use
|
||||
|
||||
```bash
|
||||
$ cargo build --release
|
||||
```
|
||||
|
||||
If you need an ARM version and just don't want to wait for ages for the program to compile, use
|
||||
If you need a version for a different system or architecture, you can use `cross` crate
|
||||
|
||||
```bash
|
||||
$ cross build -r --target aarch64-unknown-linux-gnu
|
||||
```
|
||||
|
||||
To run a program, just type
|
||||
|
||||
```bash
|
||||
$ ./lyra
|
||||
```
|
||||
|
||||
if you want to disown it from the shell, I recommend using the script I provided in `scripts` folder
|
||||
Remember to provide a `.env` file in the same directory as the binary.
|
||||
|
||||
If you want to disown the bot from the shell, I recommend using the script I provided in `scripts` folder
|
||||
|
||||
## Commands
|
||||
|
||||
As of now, working commands are:
|
||||
As of now, the commands are:
|
||||
|
||||
```
|
||||
Music:
|
||||
@@ -60,7 +90,7 @@ Music:
|
||||
/repeat Loops currently playing song provided amount of times; aliases: repeat, loop, while, for
|
||||
/resume Resumes currently paused song
|
||||
/seek Seeks a track by provided seconds
|
||||
/skip Skips the currently playing song
|
||||
/skip Skips the currently playing song; aliases: skip, :skipper:
|
||||
/stop Stops playback and destroys the queue; aliases: stop, end
|
||||
/volume Changes output volume
|
||||
/effect Plays one of available audio effects
|
||||
@@ -69,13 +99,22 @@ Music:
|
||||
Tools:
|
||||
/ai Asks AI
|
||||
/dice Rolls a dice
|
||||
/dictionary Explains provided query
|
||||
/ip Shows IP information
|
||||
/metar Prints metar for provided airport
|
||||
/owoify Owoifies whatever you want uwu
|
||||
/ping Pings you backs with a response time
|
||||
/posix Prints current time in POSIX format
|
||||
/qr Creates a qr code from text
|
||||
/taf Returns taf for provided airport
|
||||
/uptime Checks how long the bot has been running
|
||||
/verse Reference Bible by verse
|
||||
/weather Shows weather for provided location
|
||||
|
||||
Help:
|
||||
/help Prints this help message; aliases: help, huh, welp
|
||||
|
||||
Use /help command for more info on a command.
|
||||
You can edit you message to the bot and the bot will edit its response.
|
||||
|
||||
```
|
||||
|
||||
BIN
assets/lyra-nightly.png
Normal file
BIN
assets/lyra-nightly.png
Normal file
Binary file not shown.
|
After Width: | Height: | Size: 291 KiB |
1906
libs/spotify-parser/Cargo.lock
generated
1906
libs/spotify-parser/Cargo.lock
generated
File diff suppressed because it is too large
Load Diff
@@ -1,12 +0,0 @@
|
||||
[package]
|
||||
name = "lib-spotify-parser"
|
||||
version = "1.0.0"
|
||||
edition = "2021"
|
||||
|
||||
[dependencies]
|
||||
regex = "1.10.6"
|
||||
reqwest = { version = "0.12.5", features = ["blocking", "rustls-tls"] }
|
||||
scraper = "0.19.1"
|
||||
serde = "1.0.204"
|
||||
serde_json = "1.0.122"
|
||||
tokio = { version = "1.39.2", features = ["macros"] }
|
||||
@@ -1,14 +0,0 @@
|
||||
use std::fmt;
|
||||
|
||||
#[derive(Debug)]
|
||||
pub enum ParseError {
|
||||
InvalidUrl,
|
||||
}
|
||||
|
||||
impl fmt::Display for ParseError {
|
||||
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
||||
write!(f, "Invalid URL")
|
||||
}
|
||||
}
|
||||
|
||||
impl std::error::Error for ParseError {}
|
||||
@@ -1,154 +0,0 @@
|
||||
mod error;
|
||||
mod parser;
|
||||
mod retriever;
|
||||
|
||||
use error::ParseError;
|
||||
use parser::{parse_list, parse_single, USER_AGENT};
|
||||
use regex::Regex;
|
||||
use reqwest;
|
||||
use retriever::{retrieve_async_track, retrieve_async_tracks, retrieve_track, retrieve_tracks};
|
||||
|
||||
pub fn retrieve_url(url: &str) -> Result<Vec<String>, ParseError> {
|
||||
let spotify_regex: Regex = Regex::new(r"https?:\/\/(?:embed\.|open\.)spotify\.com\/(track|album|playlist)\/([a-zA-Z0-9]+)(?:\?si=[\w-]+)?").unwrap();
|
||||
|
||||
if let Some(captures) = spotify_regex.captures(url) {
|
||||
let category = captures.get(1).unwrap().as_str();
|
||||
let id = captures.get(2).unwrap().as_str();
|
||||
|
||||
match category {
|
||||
"track" => {
|
||||
let track = retrieve_track(category, id).unwrap();
|
||||
Ok(vec![track])
|
||||
}
|
||||
"playlist" | "album" => {
|
||||
let tracks = retrieve_tracks(category, id).unwrap();
|
||||
Ok(tracks)
|
||||
}
|
||||
_ => Err(ParseError::InvalidUrl),
|
||||
}
|
||||
} else {
|
||||
Err(ParseError::InvalidUrl)
|
||||
}
|
||||
}
|
||||
|
||||
pub async fn retrieve_async_url(url: &str) -> Result<Vec<String>, ParseError> {
|
||||
let spotify_regex: Regex = Regex::new(r"https?:\/\/(?:embed\.|open\.)spotify\.com\/(track|album|playlist)\/([a-zA-Z0-9]+)(?:\?si=[\w-]+)?").unwrap();
|
||||
|
||||
if let Some(captures) = spotify_regex.captures(url) {
|
||||
let category = captures.get(1).unwrap().as_str();
|
||||
let id = captures.get(2).unwrap().as_str();
|
||||
|
||||
match category {
|
||||
"track" => {
|
||||
let track = retrieve_async_track(category, id).await.unwrap();
|
||||
Ok(vec![track])
|
||||
}
|
||||
"playlist" | "album" => {
|
||||
let tracks = retrieve_async_tracks(category, id).await.unwrap();
|
||||
Ok(tracks)
|
||||
}
|
||||
_ => Err(ParseError::InvalidUrl),
|
||||
}
|
||||
} else {
|
||||
Err(ParseError::InvalidUrl)
|
||||
}
|
||||
}
|
||||
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::*;
|
||||
|
||||
const TRACK: &str = "https://open.spotify.com/track/4PTG3Z6ehGkBFwjybzWkR8?si=e0a8c8ada8284e43";
|
||||
const MULTIPLE_ARTISTS_TRACK: &str =
|
||||
"https://open.spotify.com/track/1O0SdrryPeGp6eSSUibdgo?si=4ae58febe9e74eae";
|
||||
const PLAYLIST: &str =
|
||||
"https://open.spotify.com/playlist/37i9dQZF1DZ06evO05tE88?si=e0c6f44d176f44e6";
|
||||
const ALBUM: &str =
|
||||
"https://open.spotify.com/album/6eUW0wxWtzkFdaEFsTJto6?si=_grLtlNySNyfJTZr8tP44Q";
|
||||
|
||||
#[test]
|
||||
fn check_track() {
|
||||
let track: Vec<&str> = vec!["Rick Astley - Never Gonna Give You Up"];
|
||||
assert_eq!(retrieve_url(TRACK).unwrap(), track);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn check_multiple_artists_track() {
|
||||
let track: Vec<&str> = vec!["Will o' the wisp, Rick Astley - Blood on My Tie"];
|
||||
assert_eq!(retrieve_url(MULTIPLE_ARTISTS_TRACK).unwrap(), track);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn check_playlist() {
|
||||
let playlist: Vec<&str> = vec![
|
||||
"Rick Astley - Never Gonna Give You Up",
|
||||
"Rick Astley - Take Me to Your Heart (2023 Remaster)",
|
||||
"Rick Astley - Cry for Help - Single Edit",
|
||||
"Rick Astley - Never Gonna Stop",
|
||||
"Rick Astley - Together Forever",
|
||||
"Rick Astley - Hold Me in Your Arms (7\" Version)",
|
||||
"Rick Astley - Angels on My Side",
|
||||
"New Kids On The Block, Salt-N-Pepa, Rick Astley, En Vogue - Bring Back The Time",
|
||||
"Rick Astley - Whenever You Need Somebody",
|
||||
"Rick Astley - She Wants to Dance with Me (2023 Remaster)",
|
||||
"Rick Astley - Dippin My Feet",
|
||||
"Rick Astley - My Arms Keep Missing You",
|
||||
"Rick Astley - Don't Say Goodbye",
|
||||
"Rick Astley - Dance",
|
||||
"Rick Astley - Giving Up On Love (7'' Pop Version)",
|
||||
"Rick Astley - Never Gonna Give You Up (Cake Mix)",
|
||||
"Rick Astley - It Would Take a Strong Strong Man",
|
||||
"Rick Astley - Driving Me Crazy",
|
||||
"Rick Astley - Beautiful Life",
|
||||
"Rick Astley - I Don't Want to Lose Her",
|
||||
"Rick Astley - Keep Singing",
|
||||
"Rick Astley - Forever and More",
|
||||
"Rick Astley - Hopelessly",
|
||||
"Rick Astley - When I Fall in Love",
|
||||
"Rick Astley - Every One of Us",
|
||||
"Rick Astley - High Enough",
|
||||
"Rick Astley - Ain't Too Proud to Beg (2023 Remaster)",
|
||||
"Rick Astley - I'll Never Let You Down",
|
||||
"Trevor Horn, Rick Astley - Owner Of A Lonely Heart",
|
||||
"Rick Astley - Letting Go",
|
||||
"Rick Astley - Never Knew Love",
|
||||
"Rick Astley - Lights Out - Radio Edit",
|
||||
"Rick Astley - Try",
|
||||
"Rick Astley - Dial My Number (2023 Remaster)",
|
||||
"Rick Astley - Wish Away",
|
||||
"Rick Astley - Giant",
|
||||
"Rick Astley - Move Right Out",
|
||||
"Rick Astley - Till Then (Time Stands Still) (2023 Remaster)",
|
||||
"Rick Astley - Pray with Me",
|
||||
"Rick Astley - I Don't Want to Be Your Lover",
|
||||
"Will o' the wisp, Rick Astley - Blood on My Tie",
|
||||
"Rick Astley - Can't Help Falling in Love",
|
||||
"Rick Astley - I Like the Sun",
|
||||
"Rick Astley - She Makes Me",
|
||||
"Rick Astley - Body and Soul",
|
||||
"Rick Astley - (They Long to Be) Close to You",
|
||||
"Rick Astley - Unwanted (Official Song from the Podcast)",
|
||||
"Rick Astley - Last Night on Earth",
|
||||
"Rick Astley - Everlong - Acoustic Version",
|
||||
"Rick Astley - Superman",
|
||||
];
|
||||
assert_eq!(retrieve_url(PLAYLIST).unwrap(), playlist);
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn check_album() {
|
||||
let album: Vec<&str> = vec![
|
||||
"Rick Astley - Never Gonna Give You Up",
|
||||
"Rick Astley - Whenever You Need Somebody",
|
||||
"Rick Astley - Together Forever",
|
||||
"Rick Astley - It Would Take a Strong Strong Man",
|
||||
"Rick Astley - The Love Has Gone",
|
||||
"Rick Astley - Don't Say Goodbye",
|
||||
"Rick Astley - Slipping Away",
|
||||
"Rick Astley - No More Looking for Love",
|
||||
"Rick Astley - You Move Me",
|
||||
"Rick Astley - When I Fall in Love",
|
||||
];
|
||||
assert_eq!(retrieve_url(ALBUM).unwrap(), album);
|
||||
}
|
||||
}
|
||||
@@ -1,70 +0,0 @@
|
||||
use scraper::{Html, Selector};
|
||||
use serde_json::Value;
|
||||
use std::error::Error;
|
||||
|
||||
#[derive(Debug, Clone)]
|
||||
pub(crate) struct SpotifyTrack {
|
||||
pub title: String,
|
||||
pub artist: String,
|
||||
}
|
||||
|
||||
pub(crate) const USER_AGENT: &str = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36";
|
||||
|
||||
pub(crate) fn parse_single(content: String) -> Result<SpotifyTrack, Box<dyn Error>> {
|
||||
let document = Html::parse_document(&content);
|
||||
let selector = Selector::parse("#__NEXT_DATA__").unwrap();
|
||||
|
||||
if let Some(script_element) = document.select(&selector).next() {
|
||||
let json_str = script_element.inner_html();
|
||||
let json_value: Value = serde_json::from_str(&json_str)?;
|
||||
let metadata = &json_value["props"]["pageProps"]["state"]["data"]["entity"];
|
||||
|
||||
// println!("{metadata:?}");
|
||||
|
||||
let title = metadata["title"].as_str().unwrap().to_string();
|
||||
|
||||
let artists = metadata["artists"].as_array().unwrap();
|
||||
let artists_list: Vec<String> = artists
|
||||
.iter()
|
||||
.map(|artist| artist["name"].as_str().unwrap().to_string())
|
||||
.collect();
|
||||
let artist: String = artists_list.join(", ").to_string();
|
||||
|
||||
let track = SpotifyTrack { title, artist };
|
||||
|
||||
Ok(track)
|
||||
} else {
|
||||
Err("Could not find element".into())
|
||||
}
|
||||
}
|
||||
|
||||
pub(crate) fn parse_list(content: String) -> Result<Vec<SpotifyTrack>, Box<dyn Error>> {
|
||||
let document = Html::parse_document(&content);
|
||||
let selector = Selector::parse("#__NEXT_DATA__").unwrap();
|
||||
|
||||
if let Some(script_element) = document.select(&selector).next() {
|
||||
let json_str = script_element.inner_html();
|
||||
let json_value: Value = serde_json::from_str(&json_str)?;
|
||||
let metadata = &json_value["props"]["pageProps"]["state"]["data"]["entity"]["trackList"]
|
||||
.as_array()
|
||||
.unwrap();
|
||||
|
||||
// println!("{metadata:?}");
|
||||
|
||||
let tracks: Vec<SpotifyTrack> = metadata
|
||||
.iter()
|
||||
.map(|track| {
|
||||
let title = track["title"].as_str().unwrap().to_string();
|
||||
let artist = track["subtitle"].as_str().unwrap().to_string();
|
||||
|
||||
SpotifyTrack { title, artist }
|
||||
})
|
||||
.collect();
|
||||
|
||||
// println!("{tracks:?}");
|
||||
|
||||
Ok(tracks)
|
||||
} else {
|
||||
Err("Could not find element".into())
|
||||
}
|
||||
}
|
||||
@@ -1,85 +0,0 @@
|
||||
use crate::*;
|
||||
|
||||
pub fn retrieve_track(category: &str, id: &str) -> Result<String, String> {
|
||||
let embed_url = format!("https://embed.spotify.com/?uri=spotify:{}:{}", category, id);
|
||||
|
||||
let client = reqwest::blocking::Client::builder()
|
||||
.use_rustls_tls()
|
||||
.user_agent(USER_AGENT)
|
||||
.build()
|
||||
.unwrap();
|
||||
|
||||
let response = client.get(&embed_url).send().unwrap();
|
||||
let content = response.text().unwrap();
|
||||
|
||||
let parsed_content = parse_single(content).unwrap();
|
||||
|
||||
Ok(format!(
|
||||
"{} - {}",
|
||||
parsed_content.artist, parsed_content.title
|
||||
))
|
||||
}
|
||||
|
||||
pub fn retrieve_tracks(category: &str, id: &str) -> Result<Vec<String>, String> {
|
||||
let embed_url = format!("https://embed.spotify.com/?uri=spotify:{}:{}", category, id);
|
||||
|
||||
let client = reqwest::blocking::Client::builder()
|
||||
.use_rustls_tls()
|
||||
.user_agent(USER_AGENT)
|
||||
.build()
|
||||
.unwrap();
|
||||
|
||||
let response = client.get(&embed_url).send().unwrap();
|
||||
let content = response.text().unwrap();
|
||||
|
||||
let parsed_content = parse_list(content).unwrap();
|
||||
|
||||
let tracks: Vec<String> = parsed_content
|
||||
.iter()
|
||||
.map(|track| format!("{} - {}", track.artist, track.title))
|
||||
.collect();
|
||||
|
||||
Ok(tracks)
|
||||
}
|
||||
|
||||
pub async fn retrieve_async_track(category: &str, id: &str) -> Result<String, String> {
|
||||
let embed_url = format!("https://embed.spotify.com/?uri=spotify:{}:{}", category, id);
|
||||
|
||||
let client = reqwest::Client::builder()
|
||||
.use_rustls_tls()
|
||||
.user_agent(USER_AGENT)
|
||||
.build()
|
||||
.unwrap();
|
||||
|
||||
let response = client.get(&embed_url).send().await.unwrap();
|
||||
let content = response.text().await.unwrap();
|
||||
|
||||
let parsed_content = parse_single(content).unwrap();
|
||||
|
||||
Ok(format!(
|
||||
"{} - {}",
|
||||
parsed_content.artist, parsed_content.title
|
||||
))
|
||||
}
|
||||
|
||||
pub async fn retrieve_async_tracks(category: &str, id: &str) -> Result<Vec<String>, String> {
|
||||
let embed_url = format!("https://embed.spotify.com/?uri=spotify:{}:{}", category, id);
|
||||
|
||||
let client = reqwest::Client::builder()
|
||||
.use_rustls_tls()
|
||||
.user_agent(USER_AGENT)
|
||||
.build()
|
||||
.unwrap();
|
||||
|
||||
let response = client.get(&embed_url).send().await.unwrap();
|
||||
let content = response.text().await.unwrap();
|
||||
|
||||
let parsed_content = parse_list(content).unwrap();
|
||||
|
||||
let tracks: Vec<String> = parsed_content
|
||||
.iter()
|
||||
.map(|track| format!("{} - {}", track.artist, track.title))
|
||||
.collect();
|
||||
|
||||
Ok(tracks)
|
||||
}
|
||||
@@ -1,4 +1,3 @@
|
||||
pub mod embeds;
|
||||
pub mod kashi;
|
||||
pub mod music;
|
||||
pub mod tools;
|
||||
|
||||
@@ -1,11 +1,9 @@
|
||||
use crate::{Context, Error};
|
||||
|
||||
use poise::serenity_prelude::CreateEmbed;
|
||||
use poise::CreateReply;
|
||||
use serenity::{
|
||||
builder::{CreateEmbedAuthor, CreateEmbedFooter},
|
||||
model::{Colour, Timestamp},
|
||||
use poise::serenity_prelude::{
|
||||
Color, CreateEmbed, CreateEmbedAuthor, CreateEmbedFooter, Timestamp,
|
||||
};
|
||||
use poise::CreateReply;
|
||||
|
||||
pub async fn fail(ctx: Context<'_>, err: String) -> Result<(), Error> {
|
||||
ctx.send(
|
||||
@@ -25,7 +23,7 @@ pub async fn error_embed(ctx: Context<'_>, msg: &str) -> Result<CreateEmbed, Err
|
||||
.author(
|
||||
CreateEmbedAuthor::new("Something went wrong!").icon_url(ctx.author().clone().face()),
|
||||
)
|
||||
.colour(Colour::from_rgb(255, 58, 97))
|
||||
.colour(Color::from_rgb(255, 58, 97))
|
||||
.title("Oopsie, Doopsie!")
|
||||
.description(msg)
|
||||
.timestamp(Timestamp::now())
|
||||
@@ -45,7 +43,7 @@ pub async fn embed(
|
||||
) -> Result<CreateEmbed, Error> {
|
||||
let embed = CreateEmbed::default()
|
||||
.author(CreateEmbedAuthor::new(author).icon_url(ctx.author().clone().face()))
|
||||
.colour(Colour::from_rgb(255, 58, 97))
|
||||
.colour(Color::from_rgb(255, 58, 97))
|
||||
.title(title)
|
||||
.description(description)
|
||||
.timestamp(Timestamp::now())
|
||||
|
||||
@@ -1,3 +0,0 @@
|
||||
pub mod kashi;
|
||||
|
||||
pub use kashi::kashi;
|
||||
@@ -1,17 +0,0 @@
|
||||
use crate::{Context, Error};
|
||||
|
||||
/// Kashi integration platform (WIP)
|
||||
#[poise::command(
|
||||
prefix_command,
|
||||
slash_command,
|
||||
category = "Kashi"
|
||||
)]
|
||||
pub async fn kashi(
|
||||
ctx: Context<'_>
|
||||
) -> Result<(), Error> {
|
||||
|
||||
let response = format!("Kashi platform is currently under construction!");
|
||||
ctx.say(response).await?;
|
||||
|
||||
Ok(())
|
||||
}
|
||||
@@ -15,7 +15,7 @@ use poise::CreateReply;
|
||||
pub async fn deafen(ctx: Context<'_>) -> Result<(), Error> {
|
||||
let guild_id = ctx.guild_id().unwrap();
|
||||
|
||||
let manager = songbird::get(&ctx.serenity_context())
|
||||
let manager = songbird::get(ctx.serenity_context())
|
||||
.await
|
||||
.expect("Songbird client placed at init")
|
||||
.clone();
|
||||
|
||||
@@ -27,7 +27,7 @@ pub async fn join(ctx: Context<'_>) -> Result<(), Error> {
|
||||
}
|
||||
};
|
||||
|
||||
let manager = songbird::get(&ctx.serenity_context())
|
||||
let manager = songbird::get(ctx.serenity_context())
|
||||
.await
|
||||
.expect("Songbird client placed at init")
|
||||
.clone();
|
||||
|
||||
@@ -15,12 +15,12 @@ use poise::CreateReply;
|
||||
pub async fn leave(ctx: Context<'_>) -> Result<(), Error> {
|
||||
let guild_id = ctx.guild_id().unwrap();
|
||||
|
||||
let manager = songbird::get(&ctx.serenity_context())
|
||||
let manager = songbird::get(ctx.serenity_context())
|
||||
.await
|
||||
.expect("Songbird client placed at init")
|
||||
.clone();
|
||||
|
||||
if !manager.get(guild_id).is_some() {
|
||||
if manager.get(guild_id).is_none() {
|
||||
let msg = "I am not in a voice channel!";
|
||||
ctx.send(CreateReply::default().embed(error_embed(ctx, msg).await.unwrap()))
|
||||
.await?;
|
||||
|
||||
@@ -15,7 +15,7 @@ use poise::CreateReply;
|
||||
pub async fn mute(ctx: Context<'_>) -> Result<(), Error> {
|
||||
let guild_id = ctx.guild_id().unwrap();
|
||||
|
||||
let manager = songbird::get(&ctx.serenity_context())
|
||||
let manager = songbird::get(ctx.serenity_context())
|
||||
.await
|
||||
.expect("Songbird client placed at init")
|
||||
.clone();
|
||||
|
||||
@@ -9,7 +9,7 @@ use poise::CreateReply;
|
||||
pub async fn pause(ctx: Context<'_>) -> Result<(), Error> {
|
||||
let guild_id = ctx.guild_id().unwrap();
|
||||
|
||||
let manager = songbird::get(&ctx.serenity_context())
|
||||
let manager = songbird::get(ctx.serenity_context())
|
||||
.await
|
||||
.expect("Songbird client placed at init")
|
||||
.clone();
|
||||
|
||||
@@ -1,25 +1,24 @@
|
||||
use crate::commands::music::metadata::Metadata;
|
||||
use crate::commands::music::notifier::TrackErrorNotifier;
|
||||
use crate::{commands::embeds::error_embed, Context, Error};
|
||||
|
||||
use fancy_regex::Regex;
|
||||
use lib_spotify_parser;
|
||||
use poise::serenity_prelude::model::Timestamp;
|
||||
use poise::serenity_prelude::Colour;
|
||||
use poise::serenity_prelude::CreateEmbed;
|
||||
use poise::serenity_prelude::{
|
||||
Color, CreateEmbed, CreateEmbedAuthor, CreateEmbedFooter, Timestamp,
|
||||
};
|
||||
use poise::CreateReply;
|
||||
use regex::Regex as Regex_Classic;
|
||||
use serenity::builder::CreateEmbedAuthor;
|
||||
use serenity::builder::CreateEmbedFooter;
|
||||
use reqwest::Client;
|
||||
use serenity::all::GuildId;
|
||||
use songbird::events::TrackEvent;
|
||||
use songbird::input::AuxMetadata;
|
||||
use songbird::input::{Compose, YoutubeDl};
|
||||
use songbird::tracks::{TrackHandle, TrackQueue};
|
||||
use songbird::tracks::TrackQueue;
|
||||
use songbird::Call;
|
||||
use spotify_parser;
|
||||
use std::collections::VecDeque;
|
||||
use std::process::Command;
|
||||
use std::time::Duration;
|
||||
|
||||
use crate::commands::music::notifier::TrackErrorNotifier;
|
||||
use crate::http::HttpKey;
|
||||
|
||||
/// Plays a song; \
|
||||
/// you can search by query or paste an url; \
|
||||
/// aliases: play, p, enqueue
|
||||
@@ -33,25 +32,15 @@ pub async fn play(
|
||||
ctx: Context<'_>,
|
||||
#[description = "Provide a query or an url"]
|
||||
#[rest]
|
||||
mut song: String,
|
||||
song: String,
|
||||
) -> Result<(), Error> {
|
||||
let regex_spotify = Regex::new(r"https?:\/\/(?:embed\.|open\.)(?:spotify\.com\/)(?:track\/|\?uri=spotify:track:)((\w|-)+)(?:(?=\?)(?:[?&]foo=(\d*)(?=[&#]|$)|(?![?&]foo=)[^#])+)?(?=#|$)").unwrap();
|
||||
let regex_youtube =
|
||||
Regex_Classic::new(r#""url": "(https://www.youtube.com/watch\?v=[A-Za-z0-9]{11})""#)
|
||||
.unwrap();
|
||||
let regex_youtube_playlist = Regex::new(
|
||||
r"^((?:https?:)\/\/)?((?:www|m)\.)?((?:youtube\.com)).*(youtu.be\/|list=)([^#&?]*).*",
|
||||
)
|
||||
.unwrap();
|
||||
let regex_spotify_playlist = Regex::new(r"https?:\/\/(?:embed\.|open\.)(?:spotify\.com\/)(?:(album|playlist)\/|\?uri=spotify:playlist:)((\w|-)+)(?:(?=\?)(?:[?&]foo=(\d*)(?=[&#]|$)|(?![?&]foo=)[^#])+)?(?=#|$)").unwrap();
|
||||
|
||||
let is_playlist = regex_youtube_playlist.is_match(&song).unwrap()
|
||||
|| regex_spotify_playlist.is_match(&song).unwrap();
|
||||
let is_spotify =
|
||||
regex_spotify.is_match(&song).unwrap() || regex_spotify_playlist.is_match(&song).unwrap();
|
||||
let is_query = !song.starts_with("http");
|
||||
|
||||
let guild_id = ctx.guild_id().unwrap();
|
||||
let http_client = ctx.data().http_client.clone();
|
||||
let manager = songbird::get(ctx.serenity_context()).await.unwrap().clone();
|
||||
|
||||
let mut rest_playlist: VecDeque<String> = VecDeque::new();
|
||||
|
||||
if manager.get(guild_id).is_none() {
|
||||
let channel_id = ctx
|
||||
.guild()
|
||||
.unwrap()
|
||||
@@ -65,125 +54,131 @@ pub async fn play(
|
||||
let msg = "I am not in a voice channel!";
|
||||
ctx.send(CreateReply::default().embed(error_embed(ctx, msg).await.unwrap()))
|
||||
.await?;
|
||||
|
||||
return Ok(());
|
||||
}
|
||||
};
|
||||
|
||||
let http_client = {
|
||||
let data = ctx.serenity_context().data.read().await;
|
||||
data.get::<HttpKey>()
|
||||
.cloned()
|
||||
.expect("Guaranteed to exist in the typemap.")
|
||||
};
|
||||
|
||||
let manager = songbird::get(&ctx.serenity_context())
|
||||
.await
|
||||
.expect("Songbird Voice placed at init")
|
||||
.clone();
|
||||
|
||||
if let Ok(handler_lock) = manager.join(guild_id, connect_to).await {
|
||||
let mut handler = handler_lock.lock().await;
|
||||
|
||||
handler.add_global_event(TrackEvent::Error.into(), TrackErrorNotifier);
|
||||
|
||||
if is_playlist && is_spotify {
|
||||
let tracks: Vec<String> = lib_spotify_parser::retrieve_async_url(&song).await.unwrap();
|
||||
|
||||
for (index, url) in tracks.clone().iter().enumerate() {
|
||||
if url.is_empty() {
|
||||
break;
|
||||
rest_playlist = handle_play(ctx, song, handler, http_client.clone())
|
||||
.await
|
||||
.unwrap();
|
||||
} else {
|
||||
let msg = "Failed to join the voice channel!";
|
||||
ctx.send(CreateReply::default().embed(error_embed(ctx, msg).await.unwrap()))
|
||||
.await?;
|
||||
}
|
||||
let src = YoutubeDl::new_ytdl_like(
|
||||
"yt-dlp",
|
||||
http_client.clone(),
|
||||
format!("ytsearch:{}", url.to_string()),
|
||||
);
|
||||
} else {
|
||||
let handler = manager.get(guild_id).unwrap();
|
||||
let mut handler = handler.lock().await;
|
||||
handler.add_global_event(TrackEvent::Error.into(), TrackErrorNotifier);
|
||||
rest_playlist = handle_play(ctx, song, handler, http_client.clone())
|
||||
.await
|
||||
.unwrap();
|
||||
}
|
||||
|
||||
handle_playlist(rest_playlist, manager, guild_id, http_client)
|
||||
.await
|
||||
.unwrap();
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
async fn handle_play<'a>(
|
||||
ctx: Context<'a>,
|
||||
song: String,
|
||||
mut handler: tokio::sync::MutexGuard<'a, Call>,
|
||||
http_client: Client,
|
||||
) -> Result<VecDeque<String>, Error> {
|
||||
let mut results = parse_data(song).await;
|
||||
|
||||
let src: YoutubeDl =
|
||||
YoutubeDl::new_ytdl_like("yt-dlp", http_client, results.pop_front().unwrap());
|
||||
|
||||
ctx.send(
|
||||
CreateReply::default().embed(
|
||||
generate_embed(ctx, src.clone(), handler.queue(), results.clone())
|
||||
.await
|
||||
.unwrap(),
|
||||
),
|
||||
)
|
||||
.await?;
|
||||
|
||||
let aux_metadata = src.clone().aux_metadata().await.unwrap();
|
||||
let track = handler.enqueue_input(src.clone().into()).await;
|
||||
let _ = track
|
||||
|
||||
handler
|
||||
.enqueue_input(src.clone().into())
|
||||
.await
|
||||
.typemap()
|
||||
.write()
|
||||
.await
|
||||
.insert::<Metadata>(aux_metadata);
|
||||
|
||||
if index == 0 {
|
||||
let embed = generate_playlist_embed(ctx, track, tracks.len()).await;
|
||||
let response = CreateReply::default().embed(embed.unwrap());
|
||||
ctx.send(response).await?;
|
||||
}
|
||||
}
|
||||
Ok(results)
|
||||
}
|
||||
|
||||
return Ok(());
|
||||
}
|
||||
async fn parse_data(data: String) -> VecDeque<String> {
|
||||
let tracks = spotify_parser::retrieve_async_url(&data)
|
||||
.await
|
||||
.unwrap_or(vec![data])
|
||||
.iter()
|
||||
.flat_map(|track| {
|
||||
if track.contains("?list=") {
|
||||
let regex_youtube = Regex_Classic::new(
|
||||
r#""url": "(https://www.youtube.com/watch\?v=[A-Za-z0-9]{11})""#,
|
||||
)
|
||||
.unwrap();
|
||||
|
||||
if is_playlist {
|
||||
let raw_list = Command::new("yt-dlp")
|
||||
.args(["-j", "--flat-playlist", &song])
|
||||
let list = Command::new("yt-dlp")
|
||||
.args(["-j", "--flat-playlist", track])
|
||||
.output()
|
||||
.expect("failed to execute process")
|
||||
.expect("Failed to execute process")
|
||||
.stdout;
|
||||
let list = String::from_utf8(list).unwrap();
|
||||
|
||||
let list = String::from_utf8(raw_list.clone()).expect("Invalid UTF-8");
|
||||
|
||||
let urls: Vec<String> = regex_youtube
|
||||
regex_youtube
|
||||
.captures_iter(&list)
|
||||
.map(|capture| capture[1].to_string())
|
||||
.map(|capture| capture.get(1).unwrap().as_str().to_string())
|
||||
.collect::<Vec<String>>()
|
||||
} else if track.starts_with("http") {
|
||||
vec![track.clone()]
|
||||
} else {
|
||||
vec![format!("ytsearch:{}", track)]
|
||||
}
|
||||
})
|
||||
.collect();
|
||||
|
||||
for (index, url) in urls.clone().iter().enumerate() {
|
||||
if url.is_empty() {
|
||||
break;
|
||||
}
|
||||
let src = YoutubeDl::new_ytdl_like("yt-dlp", http_client.clone(), url.to_string());
|
||||
tracks
|
||||
}
|
||||
|
||||
async fn handle_playlist(
|
||||
playlist: VecDeque<String>,
|
||||
manager: std::sync::Arc<songbird::Songbird>,
|
||||
guild_id: GuildId,
|
||||
http_client: Client,
|
||||
) -> Result<(), Error> {
|
||||
for song in playlist {
|
||||
if manager.get(guild_id).is_some() {
|
||||
let handler = manager.get(guild_id).unwrap();
|
||||
let mut handler = handler.lock().await;
|
||||
|
||||
let src: YoutubeDl =
|
||||
YoutubeDl::new_ytdl_like("yt-dlp", http_client.clone(), song.clone());
|
||||
|
||||
let aux_metadata = src.clone().aux_metadata().await.unwrap();
|
||||
let track = handler.enqueue_input(src.clone().into()).await;
|
||||
let _ = track
|
||||
|
||||
handler
|
||||
.enqueue_input(src.clone().into())
|
||||
.await
|
||||
.typemap()
|
||||
.write()
|
||||
.await
|
||||
.insert::<Metadata>(aux_metadata);
|
||||
|
||||
if index == 0 {
|
||||
let embed = generate_playlist_embed(ctx, track, urls.len()).await;
|
||||
let response = CreateReply::default().embed(embed.unwrap());
|
||||
ctx.send(response).await?;
|
||||
}
|
||||
}
|
||||
|
||||
return Ok(());
|
||||
}
|
||||
|
||||
if is_spotify {
|
||||
song = format!(
|
||||
"ytsearch:{}",
|
||||
lib_spotify_parser::retrieve_async_url(&song)
|
||||
.await
|
||||
.unwrap()
|
||||
.first()
|
||||
.unwrap()
|
||||
);
|
||||
}
|
||||
|
||||
if is_query {
|
||||
song = format!("ytsearch:{}", song);
|
||||
}
|
||||
|
||||
let src = YoutubeDl::new_ytdl_like("yt-dlp", http_client, song);
|
||||
let embed = generate_embed(ctx, src.clone(), handler.queue()).await;
|
||||
let response = CreateReply::default().embed(embed.unwrap());
|
||||
ctx.send(response).await?;
|
||||
|
||||
let aux_metadata = src.clone().aux_metadata().await.unwrap();
|
||||
|
||||
let track = handler.enqueue_input(src.clone().into()).await;
|
||||
let _ = track
|
||||
.typemap()
|
||||
.write()
|
||||
.await
|
||||
.insert::<Metadata>(aux_metadata);
|
||||
}
|
||||
|
||||
Ok(())
|
||||
}
|
||||
|
||||
@@ -191,6 +186,7 @@ async fn generate_embed(
|
||||
ctx: Context<'_>,
|
||||
src: YoutubeDl,
|
||||
queue: &TrackQueue,
|
||||
results: VecDeque<String>,
|
||||
) -> Result<CreateEmbed, Error> {
|
||||
let metadata = src.clone().aux_metadata().await.unwrap();
|
||||
let AuxMetadata {
|
||||
@@ -201,18 +197,22 @@ async fn generate_embed(
|
||||
duration,
|
||||
..
|
||||
} = metadata;
|
||||
let timestamp = Timestamp::now();
|
||||
let duration_minutes = duration.unwrap_or(Duration::new(0, 0)).clone().as_secs() / 60;
|
||||
let duration_seconds = duration.unwrap_or(Duration::new(0, 0)).clone().as_secs() % 60;
|
||||
let mut description = format!("Song added to queue @ {}", queue.len() + 1);
|
||||
let mut description = format!("Enqueued @ {}", queue.len() + 1);
|
||||
let mut tracks = "Tracks enqueued";
|
||||
|
||||
if queue.len() == 0 {
|
||||
description = format!("Playing now!");
|
||||
if results.len() == 1 {
|
||||
tracks = "Track enqueued";
|
||||
}
|
||||
|
||||
if queue.is_empty() {
|
||||
description = "Playing now!".to_string();
|
||||
}
|
||||
|
||||
let embed = CreateEmbed::default()
|
||||
.author(CreateEmbedAuthor::new("Track enqueued").icon_url(ctx.author().clone().face()))
|
||||
.colour(Colour::from_rgb(255, 58, 97))
|
||||
.author(CreateEmbedAuthor::new(tracks).icon_url(ctx.author().clone().face()))
|
||||
.colour(Color::from_rgb(255, 58, 97))
|
||||
.title(title.unwrap())
|
||||
.url(source_url.unwrap())
|
||||
.thumbnail(thumbnail.unwrap_or(ctx.cache().current_user().face()))
|
||||
@@ -228,59 +228,7 @@ async fn generate_embed(
|
||||
)
|
||||
.field("DJ", ctx.author().name.clone(), true)
|
||||
.description(description)
|
||||
.timestamp(timestamp)
|
||||
.footer(
|
||||
CreateEmbedFooter::new(ctx.cache().current_user().name.to_string())
|
||||
.icon_url(ctx.cache().current_user().face()),
|
||||
);
|
||||
|
||||
Ok(embed)
|
||||
}
|
||||
|
||||
async fn generate_playlist_embed(
|
||||
ctx: Context<'_>,
|
||||
track: TrackHandle,
|
||||
queue_length: usize,
|
||||
) -> Result<CreateEmbed, Error> {
|
||||
let meta_typemap = track.typemap().read().await;
|
||||
let metadata = meta_typemap.get::<Metadata>().unwrap();
|
||||
let AuxMetadata {
|
||||
title,
|
||||
thumbnail,
|
||||
source_url,
|
||||
artist,
|
||||
duration,
|
||||
..
|
||||
} = metadata;
|
||||
let timestamp = Timestamp::now();
|
||||
let duration_minutes = duration.unwrap_or(Duration::new(0, 0)).clone().as_secs() / 60;
|
||||
let duration_seconds = duration.unwrap_or(Duration::new(0, 0)).clone().as_secs() % 60;
|
||||
|
||||
let description = format!("Enqueued tracks: {}", queue_length - 1);
|
||||
|
||||
let embed = CreateEmbed::default()
|
||||
.author(CreateEmbedAuthor::new("Playlist enqueued").icon_url(ctx.author().clone().face()))
|
||||
.colour(Colour::from_rgb(255, 58, 97))
|
||||
.title(title.as_ref().unwrap())
|
||||
.url(source_url.as_ref().unwrap())
|
||||
.thumbnail(
|
||||
thumbnail
|
||||
.as_ref()
|
||||
.unwrap_or(&ctx.cache().current_user().face()),
|
||||
)
|
||||
.field(
|
||||
"Artist",
|
||||
artist.as_ref().unwrap_or(&"Unknown Artist".to_string()),
|
||||
true,
|
||||
)
|
||||
.field(
|
||||
"Duration",
|
||||
format!("{:02}:{:02}", duration_minutes, duration_seconds),
|
||||
true,
|
||||
)
|
||||
.field("DJ", ctx.author().name.clone(), true)
|
||||
.description(description)
|
||||
.timestamp(timestamp)
|
||||
.timestamp(Timestamp::now())
|
||||
.footer(
|
||||
CreateEmbedFooter::new(ctx.cache().current_user().name.to_string())
|
||||
.icon_url(ctx.cache().current_user().face()),
|
||||
|
||||
@@ -2,12 +2,10 @@ use std::time::Duration;
|
||||
|
||||
use crate::commands::music::metadata::Metadata;
|
||||
use crate::{commands::embeds::error_embed, Context, Error};
|
||||
use poise::serenity_prelude::CreateEmbed;
|
||||
use poise::CreateReply;
|
||||
use serenity::{
|
||||
builder::{CreateEmbedAuthor, CreateEmbedFooter},
|
||||
model::{Colour, Timestamp},
|
||||
use poise::serenity_prelude::{
|
||||
Color, CreateEmbed, CreateEmbedAuthor, CreateEmbedFooter, Timestamp,
|
||||
};
|
||||
use poise::CreateReply;
|
||||
use songbird::input::AuxMetadata;
|
||||
|
||||
const QUEUE_DISPLAY_LENGTH: usize = 10;
|
||||
@@ -18,7 +16,7 @@ const QUEUE_DISPLAY_LENGTH: usize = 10;
|
||||
pub async fn queue(ctx: Context<'_>) -> Result<(), Error> {
|
||||
let guild_id = ctx.guild_id().unwrap();
|
||||
|
||||
let manager = songbird::get(&ctx.serenity_context())
|
||||
let manager = songbird::get(ctx.serenity_context())
|
||||
.await
|
||||
.expect("Songbird client placed at init")
|
||||
.clone();
|
||||
@@ -80,7 +78,7 @@ async fn embed(ctx: Context<'_>, queue: String) -> Result<CreateEmbed, Error> {
|
||||
|
||||
let embed = CreateEmbed::default()
|
||||
.author(CreateEmbedAuthor::new("Queue").icon_url(ctx.author().clone().face()))
|
||||
.colour(Colour::from_rgb(255, 58, 97))
|
||||
.colour(Color::from_rgb(255, 58, 97))
|
||||
.title(title)
|
||||
.description(queue)
|
||||
.timestamp(timestamp)
|
||||
|
||||
@@ -21,7 +21,7 @@ pub async fn repeat(
|
||||
) -> Result<(), Error> {
|
||||
let guild_id = ctx.guild_id().unwrap();
|
||||
|
||||
let manager = songbird::get(&ctx.serenity_context())
|
||||
let manager = songbird::get(ctx.serenity_context())
|
||||
.await
|
||||
.expect("Songbird client placed at init")
|
||||
.clone();
|
||||
|
||||
@@ -9,7 +9,7 @@ use poise::CreateReply;
|
||||
pub async fn resume(ctx: Context<'_>) -> Result<(), Error> {
|
||||
let guild_id = ctx.guild_id().unwrap();
|
||||
|
||||
let manager = songbird::get(&ctx.serenity_context())
|
||||
let manager = songbird::get(ctx.serenity_context())
|
||||
.await
|
||||
.expect("Songbird client placed at init")
|
||||
.clone();
|
||||
|
||||
@@ -14,7 +14,7 @@ pub async fn seek(
|
||||
) -> Result<(), Error> {
|
||||
let guild_id = ctx.guild_id().unwrap();
|
||||
|
||||
let manager = songbird::get(&ctx.serenity_context())
|
||||
let manager = songbird::get(ctx.serenity_context())
|
||||
.await
|
||||
.expect("Songbird client placed at init")
|
||||
.clone();
|
||||
|
||||
@@ -5,25 +5,19 @@ use crate::{
|
||||
commands::embeds::{embed, error_embed},
|
||||
Context, Error,
|
||||
};
|
||||
use poise::CreateReply;
|
||||
use serenity::{
|
||||
builder::{CreateEmbed, CreateEmbedAuthor, CreateEmbedFooter},
|
||||
model::{Colour, Timestamp},
|
||||
use poise::serenity_prelude::{
|
||||
Color, CreateEmbed, CreateEmbedAuthor, CreateEmbedFooter, Timestamp,
|
||||
};
|
||||
use poise::CreateReply;
|
||||
use songbird::{input::AuxMetadata, tracks::TrackHandle};
|
||||
|
||||
/// Skips the currently playing song; \
|
||||
/// aliases: skip, :skipper:
|
||||
#[poise::command(
|
||||
prefix_command,
|
||||
slash_command,
|
||||
aliases("skipper:"),
|
||||
category = "Music"
|
||||
)]
|
||||
#[poise::command(prefix_command, slash_command, aliases("skipper:"), category = "Music")]
|
||||
pub async fn skip(ctx: Context<'_>) -> Result<(), Error> {
|
||||
let guild_id = ctx.guild_id().unwrap();
|
||||
|
||||
let manager = songbird::get(&ctx.serenity_context())
|
||||
let manager = songbird::get(ctx.serenity_context())
|
||||
.await
|
||||
.expect("Songbird client placed at init")
|
||||
.clone();
|
||||
@@ -36,23 +30,18 @@ pub async fn skip(ctx: Context<'_>) -> Result<(), Error> {
|
||||
let track = track_raw.get(1);
|
||||
let queue_length = queue.len() - 1;
|
||||
|
||||
let response;
|
||||
let mut response = CreateReply::default().embed(
|
||||
embed(ctx, "Skipped!", "The queue is empty!", "")
|
||||
.await
|
||||
.unwrap(),
|
||||
);
|
||||
|
||||
match track {
|
||||
Some(track) => {
|
||||
if let Some(track) = track {
|
||||
response = CreateReply::default().embed(
|
||||
generate_embed(ctx, track.clone(), queue_length)
|
||||
.await
|
||||
.unwrap(),
|
||||
);
|
||||
}
|
||||
None => {
|
||||
response = CreateReply::default().embed(
|
||||
embed(ctx, "Skipped!", "The queue is empty!", "")
|
||||
.await
|
||||
.unwrap(),
|
||||
);
|
||||
}
|
||||
};
|
||||
|
||||
ctx.send(response).await?;
|
||||
@@ -88,7 +77,7 @@ async fn generate_embed(
|
||||
|
||||
let embed = CreateEmbed::default()
|
||||
.author(CreateEmbedAuthor::new("Skipped!").icon_url(ctx.author().clone().face()))
|
||||
.colour(Colour::from_rgb(255, 58, 97))
|
||||
.colour(Color::from_rgb(255, 58, 97))
|
||||
.title(title.as_ref().unwrap())
|
||||
.url(source_url.as_ref().unwrap())
|
||||
.thumbnail(
|
||||
|
||||
@@ -1,11 +1,9 @@
|
||||
use crate::{commands::embeds::error_embed, Context, Error};
|
||||
|
||||
use poise::serenity_prelude::model::Timestamp;
|
||||
use poise::serenity_prelude::Colour;
|
||||
use poise::serenity_prelude::CreateEmbed;
|
||||
use poise::serenity_prelude::{
|
||||
Color, CreateEmbed, CreateEmbedAuthor, CreateEmbedFooter, Timestamp,
|
||||
};
|
||||
use poise::CreateReply;
|
||||
use serenity::builder::CreateEmbedAuthor;
|
||||
use serenity::builder::CreateEmbedFooter;
|
||||
use songbird::events::TrackEvent;
|
||||
use songbird::input::AuxMetadata;
|
||||
use songbird::input::{Compose, YoutubeDl};
|
||||
@@ -36,7 +34,7 @@ pub async fn effect(
|
||||
.expect("Guaranteed to exist in the typemap.")
|
||||
};
|
||||
|
||||
let manager = songbird::get(&ctx.serenity_context())
|
||||
let manager = songbird::get(ctx.serenity_context())
|
||||
.await
|
||||
.expect("Songbird Voice placed at init")
|
||||
.clone();
|
||||
@@ -87,11 +85,11 @@ async fn generate_embed(ctx: Context<'_>, src: YoutubeDl) -> Result<CreateEmbed,
|
||||
let duration_minutes = duration.unwrap_or(Duration::new(0, 0)).clone().as_secs() / 60;
|
||||
let duration_seconds = duration.unwrap_or(Duration::new(0, 0)).clone().as_secs() % 60;
|
||||
|
||||
let description = format!("Playing now!");
|
||||
let description = "Playing now!";
|
||||
|
||||
let embed = CreateEmbed::default()
|
||||
.author(CreateEmbedAuthor::new("Playing an effect!").icon_url(ctx.author().clone().face()))
|
||||
.colour(Colour::from_rgb(255, 58, 97))
|
||||
.colour(Color::from_rgb(255, 58, 97))
|
||||
.title(title.unwrap())
|
||||
.url(source_url.unwrap())
|
||||
.thumbnail(thumbnail.unwrap_or(ctx.cache().current_user().face()))
|
||||
|
||||
@@ -1,11 +1,9 @@
|
||||
use crate::{commands::embeds::error_embed, Context, Error};
|
||||
|
||||
use poise::serenity_prelude::model::Timestamp;
|
||||
use poise::serenity_prelude::Colour;
|
||||
use poise::serenity_prelude::CreateEmbed;
|
||||
use poise::serenity_prelude::{
|
||||
Color, CreateEmbed, CreateEmbedAuthor, CreateEmbedFooter, Timestamp,
|
||||
};
|
||||
use poise::CreateReply;
|
||||
use serenity::builder::CreateEmbedAuthor;
|
||||
use serenity::builder::CreateEmbedFooter;
|
||||
use songbird::events::TrackEvent;
|
||||
use songbird::input::AuxMetadata;
|
||||
use songbird::input::{Compose, YoutubeDl};
|
||||
@@ -43,7 +41,7 @@ pub async fn stream(
|
||||
.expect("Guaranteed to exist in the typemap.")
|
||||
};
|
||||
|
||||
let manager = songbird::get(&ctx.serenity_context())
|
||||
let manager = songbird::get(ctx.serenity_context())
|
||||
.await
|
||||
.expect("Songbird Voice placed at init")
|
||||
.clone();
|
||||
@@ -94,13 +92,13 @@ async fn generate_embed(ctx: Context<'_>, src: YoutubeDl) -> Result<CreateEmbed,
|
||||
let duration_minutes = duration.unwrap_or(Duration::new(0, 0)).clone().as_secs() / 60;
|
||||
let duration_seconds = duration.unwrap_or(Duration::new(0, 0)).clone().as_secs() % 60;
|
||||
|
||||
let description = format!("Playing now!");
|
||||
let description = "Playing now!";
|
||||
|
||||
let embed = CreateEmbed::default()
|
||||
.author(
|
||||
CreateEmbedAuthor::new("Audio output hijacked!").icon_url(ctx.author().clone().face()),
|
||||
)
|
||||
.colour(Colour::from_rgb(255, 58, 97))
|
||||
.colour(Color::from_rgb(255, 58, 97))
|
||||
.title(title.unwrap())
|
||||
.url(source_url.unwrap())
|
||||
.thumbnail(thumbnail.unwrap_or(ctx.cache().current_user().face()))
|
||||
|
||||
@@ -10,7 +10,7 @@ use poise::CreateReply;
|
||||
pub async fn stop(ctx: Context<'_>) -> Result<(), Error> {
|
||||
let guild_id = ctx.guild_id().unwrap();
|
||||
|
||||
let manager = songbird::get(&ctx.serenity_context())
|
||||
let manager = songbird::get(ctx.serenity_context())
|
||||
.await
|
||||
.expect("Songbird client placed at init")
|
||||
.clone();
|
||||
|
||||
@@ -9,7 +9,7 @@ use poise::CreateReply;
|
||||
pub async fn volume(ctx: Context<'_>, #[description = "Volume"] volume: f32) -> Result<(), Error> {
|
||||
let guild_id = ctx.guild_id().unwrap();
|
||||
|
||||
let manager = songbird::get(&ctx.serenity_context())
|
||||
let manager = songbird::get(ctx.serenity_context())
|
||||
.await
|
||||
.expect("Songbird client placed at init")
|
||||
.clone();
|
||||
|
||||
@@ -10,7 +10,7 @@ pub mod posix;
|
||||
pub mod qr;
|
||||
pub mod register;
|
||||
pub mod taf;
|
||||
// pub mod uptime;
|
||||
pub mod uptime;
|
||||
pub mod verse;
|
||||
pub mod weather;
|
||||
|
||||
@@ -26,6 +26,6 @@ pub use posix::posix;
|
||||
pub use qr::qr;
|
||||
pub use register::register;
|
||||
pub use taf::taf;
|
||||
// pub use uptime::uptime;
|
||||
pub use uptime::uptime;
|
||||
pub use verse::verse;
|
||||
pub use weather::weather;
|
||||
|
||||
@@ -15,7 +15,7 @@ pub async fn ai(
|
||||
#[rest]
|
||||
prompt: String,
|
||||
) -> Result<(), Error> {
|
||||
let iamsorry = vec![
|
||||
let iamsorry = &[
|
||||
"I'm sorry, but as an AI language model, I must follow ethical guidelines, and I cannot engage in harmful, malicious, or offensive behavior.",
|
||||
"I'm sorry, but as an AI language model, I may not always be perfect and can make mistakes or provide inaccurate information. Please verify important details from reliable sources.",
|
||||
"I'm sorry, but as an AI language model, I can't engage in real-time conversations or remember previous interactions with users.",
|
||||
@@ -27,19 +27,17 @@ pub async fn ai(
|
||||
|
||||
println!("Funny prompts: {}", prompt);
|
||||
|
||||
let response;
|
||||
|
||||
let _ = {
|
||||
let response = {
|
||||
let mut rng = rand::thread_rng();
|
||||
|
||||
response = rng.gen_range(0..iamsorry.len());
|
||||
rng.gen_range(0..iamsorry.len())
|
||||
};
|
||||
|
||||
sleep(Duration::from_secs(3));
|
||||
sleep(Duration::from_secs(1));
|
||||
|
||||
ctx.send(
|
||||
CreateReply::default().embed(
|
||||
embed(ctx, "AI Response:", "", &format!("{}", iamsorry[response]))
|
||||
embed(ctx, "AI Response:", "", iamsorry[response])
|
||||
.await
|
||||
.unwrap(),
|
||||
),
|
||||
|
||||
@@ -7,12 +7,9 @@ use crate::{commands::embeds::embed, Context, Error};
|
||||
/// Rolls a dice
|
||||
#[poise::command(prefix_command, slash_command, category = "Tools")]
|
||||
pub async fn dice(ctx: Context<'_>) -> Result<(), Error> {
|
||||
let dice;
|
||||
|
||||
let _ = {
|
||||
let dice = {
|
||||
let mut rng = rand::thread_rng();
|
||||
|
||||
dice = rng.gen_range(1..7);
|
||||
rng.gen_range(1..=6)
|
||||
};
|
||||
|
||||
ctx.send(
|
||||
|
||||
@@ -1,8 +1,7 @@
|
||||
use poise::CreateReply;
|
||||
use serenity::{
|
||||
builder::{CreateEmbed, CreateEmbedAuthor, CreateEmbedFooter},
|
||||
model::{Colour, Timestamp},
|
||||
use poise::serenity_prelude::{
|
||||
Color, CreateEmbed, CreateEmbedAuthor, CreateEmbedFooter, Timestamp,
|
||||
};
|
||||
use poise::CreateReply;
|
||||
|
||||
use crate::{Context, Error};
|
||||
use url::form_urlencoded;
|
||||
@@ -34,7 +33,7 @@ async fn generate_embed(ctx: Context<'_>, message: String) -> Result<CreateEmbed
|
||||
CreateEmbedAuthor::new("Your message as a QR Code!")
|
||||
.icon_url(ctx.author().clone().face()),
|
||||
)
|
||||
.colour(Colour::from_rgb(255, 58, 97))
|
||||
.colour(Color::from_rgb(255, 58, 97))
|
||||
.title("Your QR Code:")
|
||||
.url(url.clone())
|
||||
.image(url)
|
||||
|
||||
@@ -1,34 +1,47 @@
|
||||
use once_cell::sync::Lazy;
|
||||
use poise::CreateReply;
|
||||
use std::sync::Mutex;
|
||||
|
||||
use crate::{commands::embeds::embed, Context, Error};
|
||||
|
||||
// Currently unable to get information on how long the thread was running.
|
||||
const PROCESS_UPTIME: i64 = 1000;
|
||||
pub static PROCESS_UPTIME: Lazy<Mutex<std::time::SystemTime>> =
|
||||
Lazy::new(|| Mutex::new(std::time::SystemTime::now()));
|
||||
|
||||
/// Checks how long the bot has been running
|
||||
#[poise::command(prefix_command, slash_command, category = "Tools")]
|
||||
pub async fn uptime(ctx: Context<'_>) -> Result<(), Error> {
|
||||
let uptime = PROCESS_UPTIME;
|
||||
let days = uptime / (24 * 60 * 60);
|
||||
let hours = (uptime % (24 * 60 * 60)) / 3600;
|
||||
let minutes = (uptime % 60 * 60) / 60;
|
||||
let seconds = uptime % 60;
|
||||
let start = *PROCESS_UPTIME.lock().unwrap();
|
||||
let uptime = std::time::SystemTime::now().duration_since(start).unwrap();
|
||||
|
||||
ctx.send(
|
||||
CreateReply::default().embed(
|
||||
embed(
|
||||
ctx,
|
||||
"I have been up and awake for",
|
||||
&format!("{} seconds", uptime),
|
||||
&format!(
|
||||
"{} days, {} hours, {} minutes and {} seconds",
|
||||
let (days, hours, minutes, seconds) = (
|
||||
uptime.as_secs() / 86400,
|
||||
(uptime.as_secs() / 3600) % 24,
|
||||
(uptime.as_secs() / 60) % 60,
|
||||
uptime.as_secs() % 60,
|
||||
);
|
||||
|
||||
let mut message = format!(
|
||||
"I have been awake for {} days, {} hours, {} minutes and {} seconds!",
|
||||
days, hours, minutes, seconds
|
||||
),
|
||||
)
|
||||
.await
|
||||
.unwrap(),
|
||||
),
|
||||
)
|
||||
);
|
||||
|
||||
if days != 0 {
|
||||
message = format!("I have been awake for {} days!", days);
|
||||
}
|
||||
|
||||
if days == 0 && hours != 0 {
|
||||
message = format!("I have been awake for {} hours!", hours);
|
||||
}
|
||||
|
||||
if days == 0 && hours == 0 && minutes != 0 {
|
||||
message = format!("I have been awake for {} minutes!", minutes);
|
||||
}
|
||||
|
||||
if days == 0 && hours == 0 && minutes == 0 && seconds != 0 {
|
||||
message = format!("I have been awake for {} seconds!", seconds);
|
||||
}
|
||||
|
||||
ctx.send(CreateReply::default().embed(embed(ctx, "Uptime", "", &message).await.unwrap()))
|
||||
.await?;
|
||||
|
||||
Ok(())
|
||||
|
||||
25
src/main.rs
25
src/main.rs
@@ -1,3 +1,4 @@
|
||||
use commands::tools::uptime::PROCESS_UPTIME;
|
||||
use poise::serenity_prelude::{self as serenity, ActivityData};
|
||||
use reqwest::Client as HttpClient;
|
||||
use songbird::SerenityInit;
|
||||
@@ -8,7 +9,6 @@ use tracing::{error, info, warn};
|
||||
mod commands;
|
||||
mod http;
|
||||
|
||||
use crate::commands::kashi;
|
||||
use crate::commands::music;
|
||||
use crate::commands::tools;
|
||||
use crate::http::HttpKey;
|
||||
@@ -16,7 +16,9 @@ use crate::http::HttpKey;
|
||||
type Error = Box<dyn std::error::Error + Send + Sync>;
|
||||
type Context<'a> = poise::Context<'a, Data, Error>;
|
||||
|
||||
pub struct Data;
|
||||
pub struct Data {
|
||||
pub http_client: HttpClient,
|
||||
}
|
||||
|
||||
async fn on_error(error: poise::FrameworkError<'_, Data, Error>) {
|
||||
match error {
|
||||
@@ -37,12 +39,13 @@ async fn main() {
|
||||
tracing_subscriber::fmt::init();
|
||||
dotenv::dotenv().expect("Failed to load .env file.");
|
||||
|
||||
let _ = *PROCESS_UPTIME.lock().unwrap();
|
||||
|
||||
let token =
|
||||
std::env::var("DISCORD_TOKEN").expect("Environment variable `DISCORD_TOKEN` not found!");
|
||||
let prefix = std::env::var("PREFIX").expect("Environment variable `PREFIX` not found!");
|
||||
|
||||
let commands = vec![
|
||||
kashi::kashi(),
|
||||
music::deafen(),
|
||||
music::join(),
|
||||
music::leave(),
|
||||
@@ -70,7 +73,7 @@ async fn main() {
|
||||
tools::qr(),
|
||||
tools::register(),
|
||||
tools::taf(),
|
||||
// tools::uptime(),
|
||||
tools::uptime(),
|
||||
tools::verse(),
|
||||
tools::weather(),
|
||||
];
|
||||
@@ -78,7 +81,7 @@ async fn main() {
|
||||
let options = poise::FrameworkOptions {
|
||||
commands,
|
||||
prefix_options: poise::PrefixFrameworkOptions {
|
||||
prefix: Some(prefix.to_string().into()),
|
||||
prefix: Some(prefix.to_string()),
|
||||
edit_tracker: Some(Arc::new(poise::EditTracker::for_timespan(
|
||||
Duration::from_secs(3600),
|
||||
))),
|
||||
@@ -123,7 +126,7 @@ async fn main() {
|
||||
};
|
||||
|
||||
let framework = poise::Framework::builder()
|
||||
.setup(move |ctx, ready, _framework| {
|
||||
.setup(move |ctx, ready, framework| {
|
||||
Box::pin(async move {
|
||||
info!(
|
||||
"{} [{}] connected successfully!",
|
||||
@@ -131,8 +134,16 @@ async fn main() {
|
||||
);
|
||||
ctx.set_activity(Some(ActivityData::listening(prefix + "help")));
|
||||
// poise::builtins::register_globally(ctx, &framework.options().commands).await?;
|
||||
poise::builtins::register_in_guild(
|
||||
ctx,
|
||||
&framework.options().commands,
|
||||
512680330495524873.into(),
|
||||
)
|
||||
.await?;
|
||||
|
||||
Ok(Data {})
|
||||
Ok(Data {
|
||||
http_client: HttpClient::new(),
|
||||
})
|
||||
})
|
||||
})
|
||||
.options(options)
|
||||
|
||||
Reference in New Issue
Block a user