/*
 * meli - jmap module.
 *
 * Copyright 2019 Manos Pitsidianakis
 *
 * This file is part of meli.
 *
 * meli is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * meli 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 General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with meli. If not, see <http://www.gnu.org/licenses/>.
 */

use std::sync::Arc;

use indexmap::IndexMap;
use serde_json::Value;
use url::Url;

use crate::jmap::{
    identity::IdentityObject,
    methods::{u64_zero, RequestUrlTemplate},
    objects::{Account, Id, Object, State},
    protocol::JmapMailCapability,
};

#[derive(Clone, Debug, Deserialize, Serialize)]
#[serde(rename_all = "camelCase")]
pub struct Session {
    pub capabilities: IndexMap<String, CapabilitiesObject>,
    pub accounts: IndexMap<Id<Account>, Account>,
    pub primary_accounts: IndexMap<String, Id<Account>>,
    #[serde(skip)]
    pub identities: IndexMap<Id<IdentityObject>, IdentityObject>,
    pub username: String,
    pub api_url: Arc<Url>,
    pub download_url: Arc<RequestUrlTemplate>,
    pub upload_url: Arc<RequestUrlTemplate>,
    pub event_source_url: Arc<RequestUrlTemplate>,
    pub state: State<Session>,
    #[serde(flatten)]
    pub extra_properties: IndexMap<String, Value>,
}

impl Object for Session {
    const NAME: &'static str = crate::identify!(Session);
}

impl Session {
    /// Return the first identity.
    pub fn mail_identity_id(&self) -> Option<Id<IdentityObject>> {
        self.identities.keys().next().cloned()
    }

    /// Return the account ID corresponding to the [`JmapMailCapability`]
    /// capability.
    pub fn mail_account_id(&self) -> Id<Account> {
        self.primary_accounts[JmapMailCapability::uri()].clone()
    }
}

#[derive(Clone, Debug, Default, Deserialize, Serialize)]
#[serde(rename_all = "camelCase")]
pub struct CapabilitiesObject {
    #[serde(default, skip_serializing_if = "u64_zero")]
    pub max_size_upload: u64,
    #[serde(default, skip_serializing_if = "u64_zero")]
    pub max_concurrent_upload: u64,
    #[serde(default, skip_serializing_if = "u64_zero")]
    pub max_size_request: u64,
    #[serde(default, skip_serializing_if = "u64_zero")]
    pub max_concurrent_requests: u64,
    #[serde(default, skip_serializing_if = "u64_zero")]
    pub max_calls_in_request: u64,
    #[serde(default, skip_serializing_if = "u64_zero")]
    pub max_objects_in_get: u64,
    #[serde(default, skip_serializing_if = "u64_zero")]
    pub max_objects_in_set: u64,
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub collation_algorithms: Vec<String>,
    #[serde(flatten, skip_serializing_if = "IndexMap::is_empty")]
    pub extra_properties: IndexMap<String, Value>,
}
