mirror of
https://github.com/GuillemCastro/spotify-dl.git
synced 2024-11-09 21:20:24 +01:00
add option for compression
This commit is contained in:
parent
84a63e1f7e
commit
8a22a6659d
6 changed files with 53 additions and 14 deletions
2
.gitignore
vendored
2
.gitignore
vendored
|
@ -12,3 +12,5 @@ target/
|
|||
|
||||
# These are backup files generated by rustfmt
|
||||
**/*.rs.bk
|
||||
|
||||
.DS_Store
|
||||
|
|
2
Cargo.lock
generated
2
Cargo.lock
generated
|
@ -1916,7 +1916,7 @@ dependencies = [
|
|||
|
||||
[[package]]
|
||||
name = "spotify-dl"
|
||||
version = "0.1.0"
|
||||
version = "0.1.1"
|
||||
dependencies = [
|
||||
"audiotags",
|
||||
"flac-bound",
|
||||
|
|
|
@ -1,6 +1,6 @@
|
|||
[package]
|
||||
name = "spotify-dl"
|
||||
version = "0.1.0"
|
||||
version = "0.1.1"
|
||||
authors = ["Guillem Castro <guillemcastro4@gmail.com>"]
|
||||
edition = "2018"
|
||||
#links = "FLAC"
|
||||
|
|
|
@ -28,7 +28,7 @@ brew install flac
|
|||
## Usage
|
||||
|
||||
```
|
||||
spotify-dl 0.1.0
|
||||
spotify-dl 0.1.1
|
||||
A commandline utility to download music directly from Spotify
|
||||
|
||||
USAGE:
|
||||
|
@ -36,10 +36,13 @@ USAGE:
|
|||
|
||||
FLAGS:
|
||||
-h, --help Prints help information
|
||||
-o, --ordered Download songs in the order they are in the playlist, prfixing the filename with the track number
|
||||
-o, --ordered Prefixing the filename with its index in the playlist
|
||||
-V, --version Prints version information
|
||||
|
||||
OPTIONS:
|
||||
-c, --compression <compression> Setting the flac compression level from 0 (fastest, least compression) to
|
||||
8 (slowest, most compression). A value larger than 8 will be reated as 8.
|
||||
Defaults to 4.
|
||||
-d, --destination <destination> The directory where the songs will be downloaded [default: .]
|
||||
-p, --password <password> Your Spotify password
|
||||
-u, --username <username> Your Spotify username
|
||||
|
@ -56,4 +59,4 @@ The usage of this software may infringe Spotify's ToS and/or your local legislat
|
|||
|
||||
## License
|
||||
|
||||
spotify-dl is licensed under the MIT license. See [LICENSE](LICENSE).
|
||||
spotify-dl is lic:ewensed under the MIT license. See [LICENSE](LICENSE).
|
||||
|
|
|
@ -1,24 +1,33 @@
|
|||
use std::path::Path;
|
||||
|
||||
use audiotags::{Tag, TagType};
|
||||
use librespot::{playback::{audio_backend::{Open, Sink, SinkError}, config::AudioFormat, decoder::AudioPacket, convert::Converter}};
|
||||
use librespot::playback::{
|
||||
audio_backend::{Open, Sink, SinkError},
|
||||
config::AudioFormat,
|
||||
convert::Converter,
|
||||
decoder::AudioPacket,
|
||||
};
|
||||
|
||||
// extern crate flac_bound;
|
||||
|
||||
use flac_bound::{FlacEncoder};
|
||||
use flac_bound::FlacEncoder;
|
||||
|
||||
use crate::TrackMetadata;
|
||||
|
||||
pub struct FileSink {
|
||||
sink: String,
|
||||
content: Vec<i32>,
|
||||
metadata: Option<TrackMetadata>
|
||||
metadata: Option<TrackMetadata>,
|
||||
compression: u32,
|
||||
}
|
||||
|
||||
impl FileSink {
|
||||
pub fn add_metadata(&mut self, meta: TrackMetadata) {
|
||||
self.metadata = Some(meta);
|
||||
}
|
||||
pub fn set_compression(&mut self, compression: u32) {
|
||||
self.compression = compression;
|
||||
}
|
||||
}
|
||||
|
||||
impl Open for FileSink {
|
||||
|
@ -28,7 +37,8 @@ impl Open for FileSink {
|
|||
FileSink {
|
||||
sink: file,
|
||||
content: Vec::new(),
|
||||
metadata: None
|
||||
metadata: None,
|
||||
compression: 4,
|
||||
}
|
||||
} else {
|
||||
panic!();
|
||||
|
@ -42,21 +52,33 @@ impl Sink for FileSink {
|
|||
}
|
||||
|
||||
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();
|
||||
let mut encoder = FlacEncoder::new()
|
||||
.unwrap()
|
||||
.channels(2)
|
||||
.bits_per_sample(16)
|
||||
.compression_level(*&self.compression)
|
||||
.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();
|
||||
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");
|
||||
},
|
||||
tag.write_to_path(&self.sink)
|
||||
.expect("Failed to write metadata");
|
||||
}
|
||||
None => (),
|
||||
}
|
||||
Ok(())
|
||||
|
|
12
src/main.rs
12
src/main.rs
|
@ -45,6 +45,13 @@ struct Opt {
|
|||
help = "Prefixing the filename with its index in the playlist"
|
||||
)]
|
||||
ordered: bool,
|
||||
#[structopt(
|
||||
short = "c",
|
||||
long = "compression",
|
||||
help = "Setting the flac compression level from 0 (fastest, least compression) to
|
||||
8 (slowest, most compression). A value larger than 8 will be reated as 8. Defaults to 4."
|
||||
)]
|
||||
compression: Option<u32>,
|
||||
}
|
||||
|
||||
#[derive(Clone)]
|
||||
|
@ -67,6 +74,7 @@ async fn download_tracks(
|
|||
destination: PathBuf,
|
||||
tracks: Vec<SpotifyId>,
|
||||
ordered: bool,
|
||||
compression: Option<u32>,
|
||||
) {
|
||||
let player_config = PlayerConfig::default();
|
||||
let bar_style = ProgressStyle::default_bar()
|
||||
|
@ -116,6 +124,9 @@ async fn download_tracks(
|
|||
librespot::playback::config::AudioFormat::S16,
|
||||
);
|
||||
file_sink.add_metadata(metadata);
|
||||
if let Some(compression) = compression {
|
||||
file_sink.set_compression(compression);
|
||||
}
|
||||
let (mut player, _) =
|
||||
Player::new(player_config.clone(), session.clone(), None, move || {
|
||||
Box::new(file_sink)
|
||||
|
@ -184,6 +195,7 @@ async fn main() {
|
|||
PathBuf::from(opt.destination),
|
||||
tracks,
|
||||
opt.ordered,
|
||||
opt.compression,
|
||||
)
|
||||
.await;
|
||||
}
|
||||
|
|
Loading…
Reference in a new issue