diff options
Diffstat (limited to 'src/main.rs')
| -rw-r--r-- | src/main.rs | 203 |
1 files changed, 167 insertions, 36 deletions
diff --git a/src/main.rs b/src/main.rs index 9f64ca4..057588b 100644 --- a/src/main.rs +++ b/src/main.rs @@ -1,51 +1,182 @@ -extern crate image; +/* + _______ ________ __________ + ___ |__________________ _______________ __ \_______ __ ___ ___ \ + __ /| |_ ___/ ___/ __ `/_ __ \ _ \_ / / / _ \_ | / / __ / _ \ | + _ ___ | / / /__ / /_/ /_ / / / __/ /_/ // __/_ |/ / _ / , _/ / + /_/ |_/_/ \___/ \__,_/ /_/ /_/\___//_____/ \___/_____/ |_/_/|_|_/ +MIT License -use clap::{Arg, Command}; -use termimage::{Options}; +Copyright (c) 2023-2025 ArcaneDev +Permission is hereby granted, free of charge, to any person obtaining a copy +of this software and associated documentation files (the "Software"), to deal +in the Software without restriction, including without limitation the rights +to use, copy, modify, merge, publish, distribute, sublicense, and/or sell +copies of the Software, and to permit persons to whom the Software is +furnished to do so, subject to the following conditions: + +The above copyright notice and this permission notice shall be included in all +copies or substantial portions of the Software. + +THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR +IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, +FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE +AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER +LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, +OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE +SOFTWARE. + +*/ + +use std::fs::{self, File}; +use std::path::PathBuf; + +use clap::{Parser, Subcommand}; +use termimage::Options; -mod configmanager; mod parser; +mod shared; -use crate::configmanager::{Config, handle_config}; +use parser::{get_config, parse_weather, generate_config, Config}; +use shared::WeatherData; -fn main() -> Result<(), Box<dyn std::error::Error>> { - let matches = Command::new("WeatherFetch") - .version("0.1") - .author("Borisov Alexey <arcanetmodl@gmail.com>") - .about("Weather fetch with image and ASCII art support") - .arg(Arg::new("image").short('i').long("image").value_name("PATH")) - .arg(Arg::new("exclude").short('e').long("exclude").value_name("TYPE") - .value_parser(["current", "minutely", "hourly", "daily", "alerts"])) - .arg(Arg::new("help").short('h').long("help")) - .arg(Arg::new("lat").short('t').long("lat").value_name("LATITUDE")) - .arg(Arg::new("lon").short('n').long("lon").value_name("LONGITUDE")) - .get_matches(); - - if matches.contains_id("help") { - println!("Usage: ..."); - return Ok(()); - } - - if let Some(img_path) = matches.get_one::<String>("image") { - let _img = Image::from_path(img_path)?; - } +#[derive(Parser)] +#[command(name = "wfetch")] +#[command(about = "Weather fetch tool")] +struct Cli { + #[command(subcommand)] + command: Option<Commands>, +} - let config = Config::load()?; - handle_config(&config)?; +#[derive(Subcommand)] +enum Commands { + Config, + Fetch, + Clean, + Today, + Tomorrow, + RebuildCache, + CheckCfg +} +fn process_config() -> Result<(), Box<dyn std::error::Error>> { + let _cfg: Config = get_config()?; + println!("Config is valid"); + Ok(()) +} - let opts = Options::parse(); - /* - if !opts { - opts = configmanager::Config::load(); - } +fn parse_cached() -> Result<WeatherData, Box<dyn std::error::Error>> { + let home = std::env::var("HOME")?; + let cache_path = format!("{}/.cache/WeatherFetch/weather.json", home); + + if !PathBuf::from(&cache_path).exists() { + return Err("Cache file not found".into()); + } + + let cache_data = fs::read_to_string(&cache_path)?; + let weather_data: WeatherData = serde_json::from_str(&cache_data)?; + Ok(weather_data) +} - let format = ops::guess_format(&opts.image)?; - let img = ops::load_image(&opts.image, format)?; +fn print_help() -> Result<(), Box<dyn std::error::Error>> { + println!("Help command"); + println!("Usage: wfetch <command>"); + println!("Commands:"); + println!(" config - Check config"); + println!(" fetch - Fetch weather-data"); + println!(" clean - Clean cache"); + println!(" help - Print help"); + println!(" today - Print today weather"); + println!(" tomorrow - Print tomorrow weather"); + println!(" rebuild-cache - Rebuild cache"); + Ok(()) +} - */ +fn clean_cache() -> Result<(), Box<dyn std::error::Error>> { + let home = std::env::var("HOME")?; + let cache_path = format!("{}/.cache/WeatherFetch/weather.json", home); + if PathBuf::from(&cache_path).exists() { + fs::remove_file(&cache_path)?; + println!("Cache cleaned successfully"); + } else { + println!("Cache file not found"); + } + Ok(()) +} +fn rebuild_cache() -> Result<(), Box<dyn std::error::Error>> { + let rt = tokio::runtime::Runtime::new()?; + let weather_data = rt.block_on(parse_weather())?; + + let home = std::env::var("HOME")?; + let cache_dir = format!("{}/.cache/WeatherFetch", home); + fs::create_dir_all(&cache_dir)?; + + let cache_path = format!("{}/weather.json", cache_dir); + let json_data = serde_json::to_string_pretty(&weather_data)?; + fs::write(&cache_path, json_data)?; + + println!("Cache rebuilt successfully"); Ok(()) } + +fn main() -> Result<(), Box<dyn std::error::Error>> { + let cli = Cli::parse(); + + match cli.command { + Some(Commands::Config) => { + println!("Config checker command"); + let home = std::env::var("HOME")?; + let config_path = format!("{}/.config/WeatherFetch/Config.toml", home); + if File::open(&config_path).is_err() { + generate_config()?; + println!("Config generated successfully"); + } else { + process_config()?; + println!("Config already exists and is valid."); + } + Ok(()) + }, + Some(Commands::Fetch) => { + println!("Fetch weather-data command"); + let rt = tokio::runtime::Runtime::new()?; + let weather_data = rt.block_on(parse_weather())?; + let home = std::env::var("HOME")?; + println!("Weather data fetched to: {}", format!("{}/.cache/WeatherFetch", home)); + Ok(()) + }, + Some(Commands::Clean) => { + println!("Clean cache command"); + clean_cache()?; + Ok(()) + }, + Some(Commands::Today) => { + println!("Today weather command"); + Ok(()) + }, + Some(Commands::Tomorrow) => { + println!("Tomorrow weather command"); + Ok(()) + }, + Some(Commands::RebuildCache) => { + println!("Rebuild cache command"); + rebuild_cache()?; + Ok(()) + }, + Some(Commands::CheckCfg) => { + println!("Validating cfg..."); + let home = std::env::var("HOME")?; + let config_path = format!("{}/.config/WeatherFetch/Config.toml", home); + if File::open(&config_path).is_ok() { + process_config()?; + } else { + println!("Config file not found, try `wfetch config`, its will generate default cfg."); + } + Ok(()) + } + None => { + print_help() + }, + } +}
\ No newline at end of file |