love/ilovetv
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

A lot of refactoring

Browse Source 
Made the program a lot more configurable and also cleaned up the parser
struct
This commit is contained in:
Love
2023-03-01 21:59:52 +01:00
parent 86f85011ec
commit 36b79fd01f
6 changed files with 248 additions and 150 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
src/lib.rs
View File

@ -1,47 +1,14 @@
mod m3u8;
mod parser;
use std::{
fs,
io::{stdin, stdout, Stdin, StdoutLock, Write},
process,
};
use std::io::{stdin, stdout, Stdin, StdoutLock, Write};
pub use m3u8::M3u8;
pub use parser::Parser;
mod config;
pub use config::Configuration;
mod downloader;
use directories::ProjectDirs;
pub use downloader::download_with_progress;
pub fn setup() -> String {
let project_dirs = ProjectDirs::from("com", "billenius", "iptvnator_rs").unwrap();
let config_dir = project_dirs.config_dir();
let ilovetv_config_file = config_dir.join("iptv_url.txt");
if ilovetv_config_file.exists() {
return fs::read_to_string(&ilovetv_config_file).expect("Failed to read iptv_url");
}
let mut readline = Readline::new();
println!("Hello, I would need an url to your iptv/m3u/m3u8 stream");
let url = loop {
let url = readline.input("enter url: ");
let yn = readline.input("Are you sure? (Y/n) ");
if yn.trim().to_lowercase() != "n" {
break url.trim().to_string();
}
};
let _ = fs::create_dir_all(config_dir);
if let Err(e) = fs::write(ilovetv_config_file, &url) {
eprintln!("{:?}", e);
process::exit(-1);
}
url
}
pub struct Readline<'a> {
stdout: StdoutLock<'a>,
stdin: Stdin,
@ -66,7 +33,7 @@ impl<'a> Readline<'a> {
/**
* I know that this isn't considered true rusty code, but the places it's used in is
* safe. For this I'll leave the funciton as unsafe, so to better see it's uses.
* safe. For this I'll leave the funciton as unsafe, so to better see it's uses.
* This solution makes the uses BLAZINGLY FAST which moreover is the most rusty you can get.
*/
pub unsafe fn get_mut_ref<T>(reference: &T) -> &mut T {