use clap::Args;
use serde::{Serialize, Deserialize};
use colored::*;

use crate::types::Status;

#[derive(Serialize)]
#[serde(rename_all = "camelCase")]
pub struct Travel {
    token: String,
    #[serde(flatten)]
    action: Action,
}

#[derive(Serialize, Debug)]
#[serde(rename_all = "camelCase")]
#[serde(tag = "action")]
pub enum Action {
    #[serde(rename = "checkin")]
    #[serde(rename_all = "camelCase")]
    CheckIn {
        token: String,
        train: TrainRef,
        from_station: String,
        #[serde(skip_serializing_if = "Option::is_none")]
        to_station: Option<String>,
        #[serde(skip_serializing_if = "Option::is_none")]
        comment: Option<String>,
    },
    #[serde(rename = "checkout")]
    #[serde(rename_all = "camelCase")]
    CheckOut {
        to_station: String,
        force: bool,
        #[serde(skip_serializing_if = "Option::is_none")]
        comment: Option<String>,
        token: String
    },
    Undo {token: String},
}

#[derive(Args, Serialize, Debug)]
pub struct TrainRef {
    #[clap(name = "TRAIN TYPE")]
    #[serde(rename = "type")]
    pub _type: String,
    #[clap(name = "NUMBER")]
    pub no: String,
}

#[derive(Deserialize, Debug)]
pub struct Response {
    success: Option<bool>,
    deprecated: bool,
    status: Status,
    error: Option<String>
}

impl std::fmt::Display for Response {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match &self.error {
            Some(msg) => write!(f, "{}", msg.red()),
            None => write!(f, "{}\n\n{}", "Success!".green(), self.status)
        }
    }
}