You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
102 lines
3.0 KiB
102 lines
3.0 KiB
/* |
|
TU - Time's Up! |
|
Copyright (C) 2020 Julio Biason |
|
|
|
This program is free software: you can redistribute it and/or modify |
|
it under the terms of the GNU Affero General Public License as published by |
|
the Free Software Foundation, either version 3 of the License, or |
|
(at your option) any later version. |
|
|
|
This program is distributed in the hope that it will be useful, |
|
but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the |
|
GNU Affero General Public License for more details. |
|
|
|
You should have received a copy of the GNU Affero General Public License |
|
along with this program. If not, see <https://www.gnu.org/licenses/>. |
|
*/ |
|
|
|
use clap::crate_authors; |
|
use clap::crate_description; |
|
use clap::crate_name; |
|
use clap::crate_version; |
|
use clap::App; |
|
use clap::Arg; |
|
use clap::ArgMatches; |
|
use clap::SubCommand; |
|
|
|
use crate::date::Date; |
|
use crate::date_errors::DateError; |
|
use crate::datetime::DateTime; |
|
|
|
type Description = String; |
|
|
|
pub enum ParseError { |
|
InvalidDate, |
|
UnknownOption, |
|
} |
|
|
|
impl From<DateError> for ParseError { |
|
fn from(_: DateError) -> ParseError { |
|
ParseError::InvalidDate |
|
} |
|
} |
|
|
|
#[derive(Debug)] |
|
pub enum Action { |
|
List, |
|
Add(Description, Date), |
|
AddWithTime(Description, DateTime), |
|
} |
|
|
|
pub fn parse() -> Result<Action, ParseError> { |
|
let params = App::new(crate_name!()) |
|
.version(crate_version!()) |
|
.author(crate_authors!()) |
|
.about(crate_description!()) |
|
.subcommand( |
|
SubCommand::with_name("add") |
|
.about("Add a new event") |
|
.arg( |
|
Arg::with_name("date") |
|
.required(true) |
|
.takes_value(true) |
|
.help("Date for the event, in YYYY-MM-DD format"), |
|
) |
|
.arg( |
|
Arg::with_name("description") |
|
.required(true) |
|
.takes_value(true) |
|
.help("Event description"), |
|
) |
|
.arg( |
|
Arg::with_name("time") |
|
.short("t") |
|
.long("time") |
|
.takes_value(true) |
|
.required(false) |
|
.help("Time for the event"), |
|
), |
|
); |
|
let matches = params.get_matches(); |
|
|
|
match matches.subcommand() { |
|
("", _) => Ok(Action::List), |
|
("add", Some(arguments)) => parse_add(arguments), |
|
(_, _) => Err(ParseError::UnknownOption), |
|
} |
|
} |
|
|
|
fn parse_add(arguments: &ArgMatches) -> Result<Action, ParseError> { |
|
let description = arguments.value_of("description").unwrap(); |
|
let date = arguments.value_of("date").unwrap(); |
|
|
|
if let Some(time) = arguments.value_of("time") { |
|
Ok(Action::AddWithTime( |
|
description.into(), |
|
DateTime::try_from(date, time)?, |
|
)) |
|
} else { |
|
Ok(Action::Add(description.into(), Date::try_from(date)?)) |
|
} |
|
}
|
|
|