path: root/src/abook.rs

// zotapi - Rust wrapper for the Zot API as implemented by Hubzilla
// Copyright (C) 2018  Harald Eilertsen <haraldei@anduin.net>
//
// This program 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.
//
// 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 General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program.  If not, see <https://www.gnu.org/licenses/>.

use crate::{client::Client, error::Error, XChan, ZotAPI};
use serde::Deserialize;
use std::convert::TryFrom;
use zotapi_derive::ZotAPI;

#[derive(Debug, Default, Deserialize, ZotAPI)]
pub struct Abook {
    #[serde(rename = "abook_id")]
    pub id: u32,

    #[serde(rename = "abook_account")]
    pub account: u32,

    #[serde(rename = "abook_channel")]
    pub channel: u32,

    #[serde(rename = "abook_xchan")]
    pub xchan_id: String,

    #[serde(skip)]
    pub xchan: XChan,

    #[serde(rename = "abook_my_perms")]
    pub my_perms: u16,

    #[serde(rename = "abook_their_perms")]
    pub their_perms: u16,

    // Closeness is a number between 0 and 99, but is somehow
    // returned as a string.
    #[serde(rename = "abook_closeness")]
    pub closeness: String,

    #[serde(rename = "abook_created")]
    pub created: String,

    #[serde(rename = "abook_updated")]
    pub updated: String,

    #[serde(rename = "abook_connected")]
    pub connected: String,

    #[serde(rename = "abook_dob")]
    pub dob: String,

    #[serde(rename = "abook_flags")]
    pub flags: u16,         // No longer used

    #[serde(rename = "abook_profile")]
    pub profile: String,

    #[serde(rename = "abook_blocked")]
    pub blocked: u8,

    #[serde(rename = "abook_ignored")]
    pub ignored: u8,

    #[serde(rename = "abook_hidden")]
    pub hidden: u8,

    #[serde(rename = "abook_archived")]
    pub archived: u8,

    #[serde(rename = "abook_pending")]
    pub pending: u8,

    #[serde(rename = "abook_unconnected")]
    pub unconnected: u8,  // Currently unused

    #[serde(rename = "abook_self")]
    pub myself: u8,       // original name `self`

    #[serde(rename = "abook_feed")]
    pub feed: u8,

    #[serde(rename = "abook_not_here")]
    pub not_here: u8,

    #[serde(rename = "abook_incl")]
    pub incl: String,

    #[serde(rename = "abook_excl")]
    pub excl: String,

    #[serde(rename = "abook_instance")]
    pub instance: String,

    #[serde(rename = "abook_role")]
    pub role: String,
}


impl<'a> TryFrom<&'a str> for Abook {
    type Error = Error;

    fn try_from(s: &'a str) -> Result<Self, Self::Error> {
        Ok(serde_json::from_str(s)?)
    }
}

#[derive(Debug, Default)]
pub struct AbookRequest {
    abook_id: Option<u32>,
}

impl AbookRequest {
    /// Limit to ABConfigs for a given contact
    ///
    /// `abook` is the abook id of the given contact
    pub fn with_abook_id(mut self, abook: u32) -> AbookRequest {
        self.abook_id = Some(abook);
        self
    }

    pub async fn fetch_raw(&self, client: &Client) -> Result<String, Error> {
        let mut req = client.get("abook");

        if let Some(id) = self.abook_id {
            req = req.query(&[("abook_id", id.to_string())]);
        }

        Ok(req.send().await?
            .text().await?)
    }

    pub async fn fetch(&self, client: &Client) -> Result<Vec<Abook>, Error> {
        let json: serde_json::Value = serde_json::from_str(&self.fetch_raw(&client).await?)?;
        let arr = json.as_array().ok_or("Abook: Not an array")?;
        let res: Vec<Abook> = arr.iter()
            .map(|v| {
                let mut abook: Abook = serde_json::from_value(v.clone()).unwrap();
                abook.xchan = serde_json::from_value(v.clone()).unwrap();
                abook
            })
            .collect::<_>();

        Ok(res)
    }
}