mirror of
https://github.com/GuillemCastro/spotify-dl.git
synced 2024-11-10 05:20:25 +01:00
updated to 2021 toolchain + metadata is know saved
This commit is contained in:
parent
227eeeb11b
commit
d8e06c79e6
6 changed files with 1484 additions and 927 deletions
1
.gitignore
vendored
1
.gitignore
vendored
|
@ -1,6 +1,7 @@
|
|||
/target
|
||||
.vscode
|
||||
*.flac
|
||||
bin
|
||||
|
||||
debug/
|
||||
target/
|
||||
|
|
2296
Cargo.lock
generated
2296
Cargo.lock
generated
File diff suppressed because it is too large
Load diff
|
@ -6,16 +6,16 @@ edition = "2018"
|
|||
links = "FLAC"
|
||||
|
||||
[dependencies]
|
||||
librespot = { path = "./librespot" }
|
||||
tokio-core = "0.1.17"
|
||||
futures = "0.1"
|
||||
futures-state-stream = "0.1"
|
||||
structopt = { version = "0.3", default-features = false }
|
||||
rpassword = "5.0"
|
||||
indicatif = "0.15.0"
|
||||
|
||||
[dependencies.flac-bound]
|
||||
version = "0.2.0"
|
||||
librespot = "0.3.1"
|
||||
tokio = { version = "1.18.2", features = ["rt"] }
|
||||
flac-bound = { version = "0.2.0" }
|
||||
audiotags = "0.2.7182"
|
||||
|
||||
[package.metadata.deb]
|
||||
depends="libflac-dev"
|
||||
|
|
|
@ -1 +0,0 @@
|
|||
Subproject commit 1b1c22b6bc1e330fef731f5b3bc3b130d52a5bff
|
|
@ -1,22 +1,34 @@
|
|||
use librespot::playback::audio_backend::{Open, Sink};
|
||||
use std::io::{self};
|
||||
use std::path::Path;
|
||||
|
||||
extern crate flac_bound;
|
||||
use audiotags::{Tag, TagType};
|
||||
use librespot::{playback::{audio_backend::{Open, Sink, SinkError}, config::AudioFormat, decoder::AudioPacket, convert::Converter}};
|
||||
|
||||
// extern crate flac_bound;
|
||||
|
||||
use flac_bound::{FlacEncoder};
|
||||
|
||||
use crate::TrackMetadata;
|
||||
|
||||
pub struct FileSink {
|
||||
sink: String,
|
||||
content: Vec<i32>
|
||||
content: Vec<i32>,
|
||||
metadata: Option<TrackMetadata>
|
||||
}
|
||||
|
||||
impl FileSink {
|
||||
pub fn add_metadata(&mut self, meta: TrackMetadata) {
|
||||
self.metadata = Some(meta);
|
||||
}
|
||||
}
|
||||
|
||||
impl Open for FileSink {
|
||||
fn open(path: Option<String>) -> Self {
|
||||
fn open(path: Option<String>, _audio_format: AudioFormat) -> Self {
|
||||
if let Some(path) = path {
|
||||
let file = path;
|
||||
FileSink {
|
||||
sink: file,
|
||||
content: Vec::new()
|
||||
content: Vec::new(),
|
||||
metadata: None
|
||||
}
|
||||
} else {
|
||||
panic!();
|
||||
|
@ -25,20 +37,35 @@ impl Open for FileSink {
|
|||
}
|
||||
|
||||
impl Sink for FileSink {
|
||||
fn start(&mut self) -> io::Result<()> {
|
||||
fn start(&mut self) -> Result<(), SinkError> {
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn stop(&mut self) -> io::Result<()> {
|
||||
let mut encoder = FlacEncoder::new().unwrap().channels(2).bits_per_sample(16).compression_level(0).init_file(&self.sink).unwrap();
|
||||
fn stop(&mut self) -> Result<(), SinkError> {
|
||||
let mut encoder = FlacEncoder::new().unwrap().channels(2).bits_per_sample(16).compression_level(4).init_file(&self.sink).unwrap();
|
||||
encoder.process_interleaved(self.content.as_slice(), (self.content.len()/2) as u32).unwrap();
|
||||
encoder.finish().unwrap();
|
||||
|
||||
match &self.metadata {
|
||||
Some(meta) => {
|
||||
let mut tag = Tag::new().with_tag_type(TagType::Flac).read_from_path(Path::new(&self.sink)).unwrap();
|
||||
|
||||
tag.set_album_title(&meta.album);
|
||||
for artist in &meta.artists {
|
||||
tag.add_artist(artist);
|
||||
}
|
||||
tag.set_title(&meta.track_name);
|
||||
tag.write_to_path(&self.sink).expect("Failed to write metadata");
|
||||
},
|
||||
None => (),
|
||||
}
|
||||
Ok(())
|
||||
}
|
||||
|
||||
fn write(&mut self, data: &[i16]) -> io::Result<()> {
|
||||
let mut input: Vec<i32> = data.iter().map(|el| i32::from(*el)).collect();
|
||||
self.content.append(&mut input);
|
||||
fn write(&mut self, packet: &AudioPacket, converter: &mut Converter) -> Result<(), SinkError> {
|
||||
let data = converter.f64_to_s16(packet.samples().unwrap());
|
||||
let mut data32: Vec<i32> = data.iter().map(|el| i32::from(*el)).collect();
|
||||
self.content.append(&mut data32);
|
||||
Ok(())
|
||||
}
|
||||
}
|
50
src/main.rs
50
src/main.rs
|
@ -3,7 +3,6 @@ mod file_sink;
|
|||
extern crate rpassword;
|
||||
|
||||
use std::{path::PathBuf};
|
||||
use tokio_core::reactor::Core;
|
||||
|
||||
use librespot::{core::authentication::Credentials, metadata::Playlist};
|
||||
use librespot::core::config::SessionConfig;
|
||||
|
@ -14,7 +13,7 @@ use librespot::playback::config::PlayerConfig;
|
|||
use librespot::playback::player::Player;
|
||||
use librespot::playback::audio_backend::{Open};
|
||||
|
||||
use librespot::metadata::{Track, Artist, Metadata};
|
||||
use librespot::metadata::{Track, Artist, Metadata, Album};
|
||||
|
||||
use structopt::StructOpt;
|
||||
|
||||
|
@ -33,14 +32,20 @@ struct Opt {
|
|||
destination: String
|
||||
}
|
||||
|
||||
fn create_session(core: &mut Core, credentials: Credentials) -> Session {
|
||||
#[derive(Clone)]
|
||||
pub struct TrackMetadata {
|
||||
artists: Vec<String>,
|
||||
track_name: String,
|
||||
album: String,
|
||||
}
|
||||
|
||||
async fn create_session(credentials: Credentials) -> Session {
|
||||
let session_config = SessionConfig::default();
|
||||
let session = core.run(Session::connect(session_config, credentials, None, core.handle()))
|
||||
.unwrap();
|
||||
let session = Session::connect(session_config, credentials, None).await.unwrap();
|
||||
session
|
||||
}
|
||||
|
||||
fn download_tracks(core: &mut Core, session: &Session, destination: PathBuf, tracks: Vec<SpotifyId>) {
|
||||
async fn download_tracks(session: &Session, destination: PathBuf, tracks: Vec<SpotifyId>) {
|
||||
let player_config = PlayerConfig::default();
|
||||
let bar_style = ProgressStyle::default_bar()
|
||||
.template("[{elapsed_precise}] {bar:40.cyan/blue} {pos:>7}/{len:7} (ETA: {eta}) {msg}")
|
||||
|
@ -48,49 +53,58 @@ fn download_tracks(core: &mut Core, session: &Session, destination: PathBuf, tra
|
|||
let bar = ProgressBar::new(tracks.len() as u64);
|
||||
bar.set_style(bar_style);
|
||||
bar.enable_steady_tick(500);
|
||||
|
||||
for track in tracks {
|
||||
let track_item = core.run(Track::get(&session, track)).unwrap();
|
||||
let track_item = Track::get(&session, track).await.unwrap();
|
||||
let artist_name: String;
|
||||
|
||||
let mut metadata = TrackMetadata {
|
||||
artists: Vec::new(),
|
||||
track_name: track_item.name,
|
||||
album: Album::get(session, track_item.album).await.unwrap().name
|
||||
};
|
||||
if track_item.artists.len() > 1 {
|
||||
let mut tmp: String = String::new();
|
||||
for artist in track_item.artists {
|
||||
let artist_item = core.run(Artist::get(&session, artist)).unwrap();
|
||||
let artist_item = Artist::get(&session, artist).await.unwrap();
|
||||
metadata.artists.push(artist_item.name.clone());
|
||||
tmp.push_str(artist_item.name.as_str());
|
||||
tmp.push_str(", ");
|
||||
}
|
||||
artist_name = String::from(tmp.trim_end_matches(", "));
|
||||
}
|
||||
else {
|
||||
artist_name = core.run(Artist::get(&session, track_item.artists[0])).unwrap().name;
|
||||
artist_name = Artist::get(&session, track_item.artists[0]).await.unwrap().name;
|
||||
metadata.artists.push(artist_name.clone());
|
||||
}
|
||||
let full_track_name = format!("{} - {}", artist_name, track_item.name);
|
||||
let full_track_name = format!("{} - {}", artist_name, metadata.track_name);
|
||||
let filename = format!("{}.flac", full_track_name);
|
||||
let joined_path = destination.join(&filename);
|
||||
let path = joined_path.to_str().unwrap();
|
||||
bar.set_message(full_track_name.as_str());
|
||||
let file_sink = file_sink::FileSink::open(Some(path.to_owned()));
|
||||
let mut file_sink = file_sink::FileSink::open(Some(path.to_owned()), librespot::playback::config::AudioFormat::S16);
|
||||
file_sink.add_metadata(metadata);
|
||||
let (mut player, _) = Player::new(player_config.clone(), session.clone(), None, move || {
|
||||
Box::new(file_sink)
|
||||
});
|
||||
player.load(track, true, 0);
|
||||
core.run(player.get_end_of_track_future()).unwrap();
|
||||
player.await_end_of_track().await;
|
||||
player.stop();
|
||||
bar.inc(1);
|
||||
}
|
||||
bar.finish();
|
||||
}
|
||||
|
||||
fn main() {
|
||||
#[tokio::main]
|
||||
async fn main() {
|
||||
|
||||
let opt = Opt::from_args();
|
||||
|
||||
let mut core = Core::new().unwrap();
|
||||
|
||||
let username = opt.username;
|
||||
let password = opt.password.unwrap_or_else(|| rpassword::read_password_from_tty(Some("Password: ")).unwrap());
|
||||
let credentials = Credentials::with_password(username, password);
|
||||
|
||||
let session = create_session(&mut core, credentials.clone());
|
||||
let session = create_session(credentials.clone()).await;
|
||||
|
||||
let mut tracks: Vec<SpotifyId> = Vec::new();
|
||||
|
||||
|
@ -104,7 +118,7 @@ fn main() {
|
|||
tracks.push(track);
|
||||
}
|
||||
librespot::core::spotify_id::SpotifyAudioType::NonPlayable => {
|
||||
match core.run(Playlist::get(&session, track)) {
|
||||
match Playlist::get(&session, track).await {
|
||||
Ok(mut playlist) => {
|
||||
println!("Adding all songs from playlist {} (by {}) to the queue", &playlist.name, &playlist.user);
|
||||
tracks.append(&mut playlist.tracks);
|
||||
|
@ -117,6 +131,6 @@ fn main() {
|
|||
}
|
||||
}
|
||||
|
||||
download_tracks(&mut core, &session, PathBuf::from(opt.destination), tracks);
|
||||
download_tracks(&session, PathBuf::from(opt.destination), tracks).await;
|
||||
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue