refractor

This commit is contained in:
Gvidas Juknevičius 2024-04-19 00:19:11 +03:00
parent b05a20d724
commit 3cad0b0651
Signed by: MCorange
GPG Key ID: 12B1346D720B7FBB
10 changed files with 2211 additions and 1952 deletions

1
music_mgr/Cargo.lock generated
View File

@ -766,6 +766,7 @@ dependencies = [
"serde",
"serde_json",
"tokio",
"url",
"windows",
"zip-extensions",
]

View File

@ -17,6 +17,8 @@ log = "0.4.21"
reqwest = "0.12.3"
serde = { version = "1.0.197", features = ["derive"] }
serde_json = "1.0.115"
# serde_traitobject = "0.2.8"
tokio = { version = "1.37.0", features = ["macros", "rt-multi-thread", "process", "sync"] }
url = "2.5.0"
windows = { version = "0.56.0", features = ["Win32_Foundation", "Win32_Storage_FileSystem", "Win32_System_Console"] }
zip-extensions = "0.6.2"

View File

@ -1,4 +1,6 @@
use crate::{config::ConfigWrapper, downloader::Downloader, manifest::{Manifest, ManifestSong}};
use std::str::FromStr;
use crate::{config::ConfigWrapper, downloader::Downloader, manifest::{Manifest, ManifestSong}, util::is_supported_host};
@ -23,6 +25,12 @@ pub async fn add(cfg: &ConfigWrapper, manifest: &mut Manifest, downloader: &mut
crate::prompt::simple_prompt("Enter song youtube url, make sure its not a playlist, (yt only for now)")
);
if !is_supported_host(url::Url::from_str(&url)?) {
log::error!("Invalid or unsupported host name");
return Ok(());
}
let name = name.clone().unwrap_or_else( ||
crate::prompt::simple_prompt("Enter song name with like this: {Author} - {Song name}")
);

View File

@ -1,4 +1,4 @@
use std::{collections::HashMap, path::PathBuf, process::Stdio};
use std::{collections::HashMap, path::PathBuf, process::Stdio, str::FromStr};
use lazy_static::lazy_static;
use log::Level;
@ -36,7 +36,7 @@ impl Downloader {
for (genre, songs) in &manifest.genres {
for song in songs {
self.download_song(cfg, &song, &genre, &format).await?;
self.download_song(cfg, song, &genre, &format).await?;
self.wait_for_procs(10).await?;
}
}
@ -45,37 +45,52 @@ impl Downloader {
}
pub async fn download_song(&mut self, cfg: &ConfigWrapper, song: &ManifestSong, genre: &String, format: &String) -> anyhow::Result<()> {
let path = format!("{}/{genre}/{}.{}", cfg.cli.output, song.name, &format);
let dl_dir = format!("{}/{genre}/", cfg.cli.output);
let dl_file = format!("{dl_dir}/{}.{}", song.name, &format);
if PathBuf::from(&path).exists() {
log::debug!("File {path} exists, skipping");
if PathBuf::from(&dl_file).exists() {
log::debug!("File {dl_file} exists, skipping");
return Ok(())
}
log::debug!("File {path} doesnt exist, downloading");
let mut cmd = if song.url.contains("youtube.com") || song.url.contains("youtu.be") {
log::debug!("Song {} is from yotube", song.url);
let mut cmd = tokio::process::Command::new(&cfg.cfg.ytdlp.path);
cmd.args([
"-x",
"--audio-format",
format.as_str(),
"-o",
path.as_str(),
let url = url::Url::from_str(&song.url)?;
let Some(url_host) = url.host() else {
log::error!("Url {} doesnt have a valid host name", &song.url);
return Ok(());
};
log::debug!("File {dl_file} doesnt exist, downloading");
let mut cmd = match url_host.to_string().as_str() {
"youtube.com" |
"youtu.be" => {
log::debug!("Song {} is from yotube", song.url);
let mut cmd = tokio::process::Command::new(&cfg.cfg.ytdlp.path);
cmd.args([
"-x",
"--audio-format",
&format.to_string(),
"-o",
dl_file.as_str(),
song.url.as_str()
]);
cmd
}
"open.spotify.com" => {
let mut cmd = tokio::process::Command::new(&cfg.cfg.spotdl.path);
cmd.args([
"--format",
&format.to_string(),
"--output",
dl_dir.as_str(),
song.url.as_str()
]);
cmd
} else {
let mut cmd = tokio::process::Command::new(&cfg.cfg.spotdl.path);
cmd.args([
"-x",
"--audio-format",
format.as_str(),
"-o",
path.as_str(),
song.url.as_str()
]);
cmd
cmd
}
url => {
log::error!("Unknown or unsupported hostname '{}'", url);
return Ok(());
}
};
if log::max_level() < Level::Debug {
@ -92,10 +107,10 @@ impl Downloader {
PROCESSES.lock().await.write().await.get_mut(&id).unwrap().finished = true;
});
log::info!("Downloading {path}");
log::info!("Downloading {dl_file}");
PROCESSES.lock().await.write().await.insert(id, Proc {
url: song.url.clone(),
path,
path: dl_file,
finished: false,
});
self.id_itr += 1;

View File

@ -1,76 +0,0 @@
use std::{collections::HashMap, fs::read_to_string, path::{Path, PathBuf}};
use anyhow::bail;
use serde::{Deserialize, Serialize};
const ALLOWED_FORMATS: &[&'static str] = &["m4a", "aac", "flac", "mp3", "vaw"];
type Genre = String;
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct Manifest {
#[serde(skip)]
path: PathBuf,
format: String,
pub genres: HashMap<Genre, Vec<ManifestSong>>
}
impl Manifest {
pub fn format(&self) -> anyhow::Result<String> {
if !ALLOWED_FORMATS.contains(&self.format.as_str()) {
log::error!("Unknown format, allowed formats: {}", ALLOWED_FORMATS.join(", "));
bail!("")
}
Ok(self.format.clone())
}
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ManifestSong {
pub name: String,
pub url: String
}
impl Manifest {
fn from_string(s: String) -> anyhow::Result<Self> {
let s = serde_json::from_str(&s)?;
Ok(s)
}
pub fn from_path(p: &Path) -> anyhow::Result<Self> {
let data = read_to_string(p)?;
let mut s = Self::from_string(data)?;
s.path = p.to_path_buf();
Ok(s)
}
pub fn add_song(&mut self, genre: String, name: String, url: String) -> anyhow::Result<()> {
if !self.genres.contains_key(&genre) {
self.genres.insert(genre.clone(), Vec::new());
}
let Some(genre_ref) = self.genres.get_mut(&genre) else {
log::error!("Invalid genre '{}'", genre);
bail!("Invalid genre")
};
genre_ref.push(ManifestSong {
name,
url,
});
Ok(())
}
pub fn save(&self) -> anyhow::Result<()> {
let data = serde_json::to_string_pretty(self)?;
std::fs::write(&self.path, data)?;
Ok(())
}
}

View File

@ -0,0 +1,165 @@
// pub mod v1;
use std::{collections::HashMap, fmt::{Debug, Display}, path::{Path, PathBuf}, str::FromStr};
use anyhow::{bail, Result};
use serde::{Deserialize, Serialize};
const ALLOWED_FORMATS: &[&'static str] = &["m4a", "aac", "flac", "mp3", "vaw"];
type Genre = String;
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct Manifest {
#[serde(skip)]
path: PathBuf,
format: String,
pub genres: HashMap<Genre, Vec<ManifestSong>>
}
impl Manifest {
pub fn format(&self) -> anyhow::Result<String> {
if !ALLOWED_FORMATS.contains(&self.format.as_str()) {
log::error!("Unknown format, allowed formats: {}", ALLOWED_FORMATS.join(", "));
bail!("")
}
Ok(self.format.clone())
}
}
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ManifestSong {
pub name: String,
pub url: String
}
impl Manifest {
fn from_string(s: String) -> anyhow::Result<Self> {
let s = serde_json::from_str(&s)?;
Ok(s)
}
pub fn from_path(p: &Path) -> anyhow::Result<Self> {
let data = std::fs::read_to_string(p)?;
let mut s = Self::from_string(data)?;
s.path = p.to_path_buf();
Ok(s)
}
pub fn add_song(&mut self, genre: String, name: String, url: String) -> anyhow::Result<()> {
if !self.genres.contains_key(&genre) {
self.genres.insert(genre.clone(), Vec::new());
}
let Some(genre_ref) = self.genres.get_mut(&genre) else {
log::error!("Invalid genre '{}'", genre);
bail!("Invalid genre")
};
genre_ref.push(ManifestSong {
name,
url,
});
Ok(())
}
pub fn save(&self) -> anyhow::Result<()> {
let data = serde_json::to_string_pretty(self)?;
std::fs::write(&self.path, data)?;
Ok(())
}
}
// pub type GenreName = String;
// pub type SongName = String;
// pub type Genre = HashMap<SongName, Box<dyn Song>>;
// #[derive(Debug, Serialize, Deserialize, Clone)]
// pub enum SongType {
// Youtube,
// Spotify,
// Soundcloud,
// }
// impl TryFrom<url::Url> for SongType {
// type Error = anyhow::Error;
// fn try_from(url: url::Url) -> std::prelude::v1::Result<Self, Self::Error> {
// let Some(host) = url.host_str() else {
// bail!("{url} does not have a host");
// };
// match host {
// "youtube.com" | "youtu.be" => Ok(Self::Youtube),
// "open.spotify.com" => Ok(Self::Spotify),
// "SOUNDCLOUD" => Ok(Self::Soundcloud), // TODO: Fix this
// _ => bail!("Unknwon host {url}")
// }
// }
// }
// #[allow(non_camel_case_types)]
// #[derive(Debug, Serialize, Deserialize, Clone)]
// pub enum Format {
// m4a,
// aac,
// flac,
// mp3,
// vaw,
// }
// impl TryFrom<String> for Format {
// type Error = anyhow::Error;
// fn try_from(value: String) -> std::prelude::v1::Result<Self, Self::Error> {
// match value.as_str() {
// "m4a" => Ok(Self::m4a),
// "aac" => Ok(Self::aac),
// "flac" => Ok(Self::flac),
// "mp3" => Ok(Self::mp3),
// "vaw" => Ok(Self::vaw),
// v => bail!("Unknown format {v}")
// }
// }
// }
// impl Display for Format {
// fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
// match self {
// Format::m4a => write!(f, "m4a")?,
// Format::aac => write!(f, "aac")?,
// Format::flac => write!(f, "flac")?,
// Format::mp3 => write!(f, "mp3")?,
// Format::vaw => write!(f, "vaw")?,
// }
// Ok(())
// }
// }
// pub trait Song: Debug + serde_traitobject::Serialize + serde_traitobject::Deserialize{
// fn get_url(&self) -> Result<url::Url>;
// fn get_url_str(&self) -> &String;
// fn get_url_str_mut(&mut self) -> &mut String;
// fn get_type(&self) -> &SongType;
// fn get_type_mut(&mut self) -> &mut SongType;
// }
// pub trait Manifest: Debug + Clone + serde_traitobject::Serialize + serde_traitobject::Deserialize{
// fn get_format(&self) -> Result<Format>;
// fn add_song(&mut self, genre: GenreName, name: SongName, song: &dyn Song) -> Option<Box<dyn Song>>;
// fn get_song(&self, genre: GenreName, name: SongName) -> Option<&Box<dyn Song>>;
// fn get_song_mut(&mut self, genre: GenreName, name: SongName) -> Option<&mut Box<dyn Song>>;
// fn add_genre(&mut self, genre: GenreName);
// fn get_genre(&self, genre: GenreName) -> Option<&Genre>;
// fn get_genre_mut(&mut self, genre: GenreName) -> Option<&mut Genre>;
// fn get_genres(&self) -> &HashMap<GenreName, Genre>;
// fn get_genres_mut(&mut self) -> &mut HashMap<GenreName, Genre>;
// fn load(&mut self, p: PathBuf);
// fn save(&self, p: Option<&PathBuf>);
// }

View File

@ -0,0 +1,72 @@
pub mod song;
use std::{collections::HashMap, path::PathBuf, str::FromStr};
use anyhow::Result;
use serde::{Deserialize, Serialize};
use self::song::SongV1;
use super::{Format, Genre, GenreName, Manifest, Song, SongName};
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ManifestV1 {
version: usize,
format: String,
genres: HashMap<GenreName, HashMap<SongName, Box<SongV1>>>
}
impl Manifest for ManifestV1 {
fn get_format(&self) -> Result<super::Format> {
Ok(Format::try_from(self.format.clone())?)
}
fn add_song(&mut self, genre: GenreName, name: SongName, song: &dyn Song) -> Option<Box<dyn Song>> {
let song: SongV1 = song.into();
self.get_genre_mut(genre)?
.insert(name, Box::new(song))
}
fn get_song(&self, genre: GenreName, name: SongName) -> Option<&Box<dyn Song>> {
self.get_genre(genre)?.get(&name)
}
fn get_song_mut(&mut self, genre: GenreName, name: SongName) -> Option<&mut Box<dyn Song>> {
self.get_genre_mut(genre)?.get_mut(&name)
}
fn add_genre(&mut self, name: GenreName) {
self.genres.insert(name, Default::default());
}
fn get_genre(&self, genre: GenreName) -> Option<&super::Genre> {
unsafe {
std::mem::transmute(self.genres.get(&genre))
}
}
fn get_genre_mut(&mut self, genre: GenreName) -> Option<&mut super::Genre> {
unsafe {
std::mem::transmute(self.genres.get_mut(&genre))
}
}
fn get_genres(&self) -> &HashMap<GenreName, super::Genre> {
&self.genres
}
fn get_genres_mut(&mut self) -> &mut HashMap<GenreName, super::Genre> {
todo!()
}
fn load(&mut self, p: PathBuf) {
todo!()
}
fn save(&self, p: Option<&PathBuf>) {
todo!()
}
}

View File

@ -0,0 +1,57 @@
use std::str::FromStr;
use serde::{Deserialize, Serialize};
use anyhow::Result;
use crate::manifest::{Song, SongType};
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct SongV1 {
url: String,
typ: SongType
}
impl SongV1 {
pub fn from_str(url: String) -> Result<Self> {
Self::from_url(url::Url::from_str(&url)?)
}
pub fn from_url(url: url::Url) -> Result<Self> {
let typ = SongType::try_from(url.clone())?;
Ok(Self {
url: url.to_string(),
typ,
})
}
}
impl From<&dyn Song> for SongV1 {
fn from(value: &dyn Song) -> Self {
Self {
url: value.get_url_str().clone(),
typ: value.get_type().clone(),
}
}
}
impl Song for SongV1 {
fn get_url(&self) -> anyhow::Result<url::Url> {
Ok(url::Url::from_str(&self.url)?)
}
fn get_url_str(&self) -> &String {
&self.url
}
fn get_url_str_mut(&mut self) -> &mut String {
&mut self.url
}
fn get_type(&self) -> &crate::manifest::SongType {
&self.typ
}
fn get_type_mut(&mut self) -> &mut crate::manifest::SongType {
&mut self.typ
}
}

View File

@ -2,7 +2,7 @@ use std::{collections::HashMap, io::Write};
pub fn simple_prompt(p: &str) -> String {
pub(crate) fn simple_prompt(p: &str) -> String {
print!("{c}prompt{r}: {p} > ",
c=anstyle::AnsiColor::Cyan.render_fg(),
@ -19,7 +19,7 @@ pub fn simple_prompt(p: &str) -> String {
}
#[allow(dead_code)]
pub fn prompt_with_list(p: &str, options: &[&str]) -> usize {
pub(crate) fn prompt_with_list(p: &str, options: &[&str]) -> usize {
println!("{c}prompt{r}: {p}",
c=anstyle::AnsiColor::Cyan.render_fg(),
r=anstyle::Reset.render()
@ -47,7 +47,7 @@ pub fn prompt_with_list(p: &str, options: &[&str]) -> usize {
}
}
pub fn prompt_with_list_or_str(p: &str, options: &[String]) -> String {
pub(crate) fn prompt_with_list_or_str(p: &str, options: &[String]) -> String {
println!("{c}prompt{r}: {p} (select with number or input text)",
c=anstyle::AnsiColor::Cyan.render_fg(),
r=anstyle::Reset.render()
@ -77,7 +77,7 @@ pub fn prompt_with_list_or_str(p: &str, options: &[String]) -> String {
#[allow(dead_code)]
pub fn prompt_with_map(p: &str, options: HashMap<&str, &str>) -> String {
pub(crate) fn prompt_with_map(p: &str, options: HashMap<&str, &str>) -> String {
println!("{c}prompt{r}: {p}",
c=anstyle::AnsiColor::Cyan.render_fg(),
r=anstyle::Reset.render()

View File

@ -2,10 +2,25 @@ use std::path::PathBuf;
use crate::constants;
pub(crate) fn escape_song_name(s: String) -> String {
s
.replace(".", "")
.replace("'", "")
}
pub(crate) fn is_supported_host(url: url::Url) -> bool {
let host = url.host_str();
if host.is_none() {
return false;
}
match host.unwrap() {
"youtube.com" | "youtu.be" |
"open.spotify.com" => true,
_ => false
}
}
pub fn is_program_in_path(program: &str) -> Option<PathBuf> {
pub(crate) fn is_program_in_path(program: &str) -> Option<PathBuf> {
if let Ok(path) = std::env::var("PATH") {
for p in path.split(constants::PATH_VAR_SEP) {
let exec_path = PathBuf::from(p).join(program).with_extension(constants::EXEC_EXT);
@ -18,7 +33,7 @@ pub fn is_program_in_path(program: &str) -> Option<PathBuf> {
}
#[cfg(target_family="unix")]
pub fn isatty() -> bool {
pub(crate) fn isatty() -> bool {
use std::{ffi::c_int, os::fd::AsRawFd};
unsafe {
let fd = std::io::stdin().as_raw_fd();
@ -27,7 +42,7 @@ pub fn isatty() -> bool {
}
#[cfg(target_family="windows")]
pub fn isatty() -> bool {
pub(crate) fn isatty() -> bool {
unsafe {
use windows::Win32::System::Console;
use Console::{CONSOLE_MODE, STD_OUTPUT_HANDLE};