2023-01-17 18:05:15 +01:00
|
|
|
mod m3u8;
|
|
|
|
mod parser;
|
|
|
|
use std::{
|
|
|
|
fs,
|
|
|
|
io::{stdin, stdout, Write},
|
|
|
|
process,
|
|
|
|
};
|
|
|
|
|
|
|
|
pub use m3u8::M3u8;
|
|
|
|
pub use parser::Parser;
|
|
|
|
mod config;
|
2023-01-18 21:23:04 +01:00
|
|
|
mod downloader;
|
2023-01-17 18:36:28 +01:00
|
|
|
use directories::ProjectDirs;
|
2023-01-19 01:22:29 +01:00
|
|
|
pub use downloader::download_with_progress;
|
2023-01-17 18:05:15 +01:00
|
|
|
|
|
|
|
pub fn setup() -> String {
|
2023-01-17 18:36:28 +01:00
|
|
|
let project_dirs = ProjectDirs::from("com", "billenius", "iptvnator_rs").unwrap();
|
2023-01-17 18:05:15 +01:00
|
|
|
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 stdout = stdout().lock();
|
|
|
|
let stdin = stdin();
|
2023-01-17 20:59:19 +01:00
|
|
|
|
|
|
|
println!("Hello, I would need an url to your iptv/m3u/m3u8 stream");
|
|
|
|
let url = loop {
|
|
|
|
print!("enter url: ");
|
|
|
|
stdout.flush().unwrap();
|
|
|
|
let mut url = String::new();
|
|
|
|
let _ = stdin.read_line(&mut url);
|
|
|
|
|
|
|
|
print!("Are you sure? (Y/n) ");
|
|
|
|
stdout.flush().unwrap();
|
|
|
|
let mut yn = String::new();
|
|
|
|
let _ = stdin.read_line(&mut yn);
|
|
|
|
|
|
|
|
if yn.to_lowercase() != "n" {
|
|
|
|
break url.trim().to_string();
|
|
|
|
}
|
|
|
|
};
|
2023-01-17 18:05:15 +01:00
|
|
|
|
|
|
|
let _ = fs::create_dir_all(config_dir);
|
2023-01-17 20:59:19 +01:00
|
|
|
if let Err(e) = fs::write(ilovetv_config_file, &url) {
|
2023-01-17 18:05:15 +01:00
|
|
|
eprintln!("{:?}", e);
|
|
|
|
process::exit(-1);
|
|
|
|
}
|
|
|
|
|
2023-01-17 20:59:19 +01:00
|
|
|
url
|
2023-01-17 18:05:15 +01:00
|
|
|
}
|