xapi-rs 0.1.22

// SPDX-License-Identifier: GPL-3.0-or-later

#![doc = include_str!("../../../doc/EXT_VERBS.md")]

use crate::{
    DataError, MyError, MyLanguageTag, Validate, Verb, config,
    db::{
        Aggregates,
        schema::TVerb,
        verb::{
            ext_compute_aggregates, ext_find_by_iri, ext_find_by_rid, ext_find_some, ext_update,
            insert_verb,
        },
    },
    eval_preconditions,
    lrs::{DB, Headers, User, etag_from_str, no_content, resources::WithETag},
};
use core::fmt;
use iri_string::types::IriStr;
use rocket::{
    Request, Responder, State,
    form::FromForm,
    get,
    http::{Header, Status, hyper::header},
    patch, post, put,
    request::{FromRequest, Outcome},
    routes,
};
use serde::{Deserialize, Serialize};
use sqlx::PgPool;
use std::str::FromStr;
use tracing::{debug, error, info, warn};

const DEFAULT_START_RID: i32 = 0;
const DEFAULT_COUNT: i32 = 50;
const DEFAULT_ASC: bool = true;

#[derive(Debug, Serialize)]
pub(crate) struct VerbExt {
    pub(crate) rid: i32,
    pub(crate) verb: Verb,
}

/// Simplified Verb representation.
#[derive(Debug, Deserialize, Serialize)]
pub struct VerbUI {
    pub(crate) rid: i32,
    pub(crate) iri: String,
    pub(crate) display: String,
}

impl VerbUI {
    pub(crate) fn from(v: TVerb, language: &MyLanguageTag) -> Self {
        VerbUI {
            rid: v.id,
            iri: v.iri,
            display: match v.display {
                Some(x) => String::from(x.0.get(language).unwrap_or("")),
                None => String::from(""),
            },
        }
    }

    /// Return the row identifier.
    pub fn rid(&self) -> i32 {
        self.rid
    }

    /// Return the IRI as a `&str`.
    pub fn iri_as_str(&self) -> &str {
        &self.iri
    }

    /// Return the `display` as a `&str`.
    pub fn display(&self) -> &str {
        &self.display
    }
}

/// Rocket Responder that returns an OK HTTP Status w/ a JSON string of either
/// a _Verb_ or a _VerbExt_ along with ETag, and Content-Type HTTP headers.
#[derive(Responder)]
#[response(status = 200, content_type = "json")]
struct ETaggedResource {
    inner: String,
    etag: Header<'static>,
}

pub(crate) struct QueryParams<'a> {
    pub(crate) language: &'a str,
    pub(crate) start: i32,
    pub(crate) count: i32,
    pub(crate) asc: bool,
}

/// A structure grouping the GET multi request parameters.
#[rocket::async_trait]
impl<'r> FromRequest<'r> for QueryParams<'r> {
    type Error = MyError;

    async fn from_request(req: &'r Request<'_>) -> Outcome<Self, Self::Error> {
        let language = match qp::<&str>(req, "language", &config().default_language) {
            Ok(x) => x,
            Err(x) => return Outcome::Error((Status::BadRequest, x)),
        };
        // ensure `language` is a valid language tag...
        match MyLanguageTag::from_str(language) {
            Ok(_) => (),
            Err(x) => {
                error!("This ({}) is NOT a valid language tag: {}", language, x);
                return Outcome::Error((Status::BadRequest, MyError::Data(x)));
            }
        }

        let start = match qp::<i32>(req, "start", DEFAULT_START_RID) {
            Ok(x) => x,
            Err(x) => return Outcome::Error((Status::BadRequest, x)),
        };
        // must be >= 0
        if start < 0 {
            let msg = format!("Start ({start}) MUST be greater than or equal to 0");
            error!("Failed: {}", msg);
            return Outcome::Error((Status::BadRequest, MyError::Runtime(msg.into())));
        }

        let count = match qp::<i32>(req, "count", DEFAULT_COUNT) {
            Ok(x) => x,
            Err(x) => return Outcome::Error((Status::BadRequest, x)),
        };
        // must be in [10..=100]
        if !(10..=100).contains(&count) {
            let msg = format!("Count ({count}) MUST be w/in [10..101]");
            error!("Failed: {}", msg);
            return Outcome::Error((Status::BadRequest, MyError::Runtime(msg.into())));
        }

        let asc = match qp::<bool>(req, "asc", DEFAULT_ASC) {
            Ok(x) => x,
            Err(x) => return Outcome::Error((Status::BadRequest, x)),
        };

        Outcome::Success(QueryParams {
            language,
            start,
            count,
            asc,
        })
    }
}

/// Generic function to assign a value to an expected query parameter. If the
/// named parameter is missing, a provided default value will be used instead.
fn qp<'r, T: FromForm<'r>>(
    req: &'r Request<'_>,
    name: &str,
    default_value: T,
) -> Result<T, MyError> {
    match req.query_value::<T>(name) {
        Some(Ok(x)) => Ok(x),
        Some(Err(x)) => {
            let msg = format!("Failed parsing query parameter '{name}': {x}");
            error!("{}", msg);
            Err(MyError::Runtime(msg.into()))
        }
        None => {
            info!("Missing query parameter '{}'. Use default value", name);
            Ok(default_value)
        }
    }
}

impl fmt::Display for QueryParams<'_> {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        write!(
            f,
            "[language={}, start={}, count={}, asc? {}]",
            self.language, self.start, self.count, self.asc
        )
    }
}

#[doc(hidden)]
pub fn routes() -> Vec<rocket::Route> {
    routes![
        post,
        put,
        put_rid,
        patch,
        patch_rid,
        get_iri,
        get_rid,
        get_aggregates,
        get_some
    ]
}

fn parse_verb(s: &str) -> Result<Verb, MyError> {
    if s.is_empty() {
        return Err(MyError::HTTP {
            status: Status::BadRequest,
            info: "Body must NOT be empty".into(),
        });
    }

    let v = serde_json::from_str::<Verb>(s)
        .map_err(|x| MyError::Data(DataError::JSON(x)).with_status(Status::BadRequest))?;
    debug!("v = {}", v);
    // a valid Verb may have a None or empty `display` language map.  this is
    // not acceptable here...
    if v.is_valid() {
        match v.display_as_map() {
            Some(x) => {
                if x.is_empty() {
                    Err(MyError::HTTP {
                        status: Status::BadRequest,
                        info: "Verb's 'display' language map MUST not be empty".into(),
                    })
                } else {
                    Ok(v)
                }
            }
            None => Err(MyError::HTTP {
                status: Status::BadRequest,
                info: "Verb's 'display' language map MUST not be null".into(),
            }),
        }
    } else {
        Err(MyError::HTTP {
            status: Status::BadRequest,
            info: "Verb is invalid".into(),
        })
    }
}

/// Create new Verb resource.
#[post("/", data = "<body>")]
async fn post(body: &str, db: &State<DB>, user: User) -> Result<WithETag, MyError> {
    debug!("----- post ----- {}", user);
    user.can_use_verbs()?;

    let new_verb = parse_verb(body)?;
    let conn = db.pool();
    let rid = insert_verb(conn, &new_verb)
        .await
        .map_err(|x| x.with_status(Status::BadRequest))?;
    info!("Created Verb at #{}", rid);
    let etag = etag_from_str(body);
    Ok(WithETag {
        inner: Status::Ok,
        etag: Header::new(header::ETAG.as_str(), etag.to_string()),
    })
}

/// Update existing Verb replacing its `display` field.
#[put("/", data = "<body>")]
async fn put(c: Headers, body: &str, db: &State<DB>, user: User) -> Result<WithETag, MyError> {
    debug!("----- put ----- {}", user);
    user.can_use_verbs()?;

    let new_verb = parse_verb(body)?;
    let conn = db.pool();
    // must already exist...
    let x = ext_find_by_iri(conn, new_verb.id_as_str())
        .await
        .map_err(|x| x.with_status(Status::NotFound))?;
    update_it(c, conn, x.rid, x.verb, new_verb).await
}

#[put("/<rid>", data = "<body>")]
async fn put_rid(
    c: Headers,
    rid: i32,
    body: &str,
    db: &State<DB>,
    user: User,
) -> Result<WithETag, MyError> {
    debug!("----- put_rid ----- {}", user);
    user.can_use_verbs()?;

    let new_verb = parse_verb(body)?;
    let conn = db.pool();
    // must already exist...
    let old_verb = ext_find_by_rid(conn, rid)
        .await
        .map_err(|x| x.with_status(Status::NotFound))?;
    update_it(c, conn, rid, old_verb, new_verb).await
}

async fn update_it(
    c: Headers,
    conn: &PgPool,
    rid: i32,
    old_verb: Verb,
    new_verb: Verb,
) -> Result<WithETag, MyError> {
    // only update if pre-conditions exist + pass...
    if c.has_no_conditionals() {
        Err(MyError::HTTP {
            status: Status::Conflict,
            info: "Update existing Verb w/ no pre-conditions is NOT allowed".into(),
        })
    } else {
        debug!("old_verb = {}", old_verb);
        let x = serde_json::to_string(&old_verb).map_err(|x| MyError::Data(DataError::JSON(x)))?;
        let etag = etag_from_str(&x);
        debug!("etag (old) = {}", etag);
        match eval_preconditions!(&etag, c) {
            s if s != Status::Ok => Err(MyError::HTTP {
                status: s,
                info: "Failed pre-condition(s)".into(),
            }),
            _ => {
                if new_verb == old_verb {
                    info!("Old + new Verbs are identical. Pass");
                    Ok(no_content(&etag))
                } else {
                    ext_update(conn, rid, &new_verb).await?;
                    // etag changed.  recompute...
                    let x = serde_json::to_string(&new_verb)
                        .map_err(|x| MyError::Data(DataError::JSON(x)))?;
                    let etag = etag_from_str(&x);
                    debug!("etag (new) = {}", etag);
                    Ok(no_content(&etag))
                }
            }
        }
    }
}

/// Update existing Verb merging `display` fields.
#[patch("/", data = "<body>")]
async fn patch(c: Headers, body: &str, db: &State<DB>, user: User) -> Result<WithETag, MyError> {
    debug!("----- patch ----- {}", user);
    user.can_use_verbs()?;

    let new_verb = parse_verb(body)?;
    let conn = db.pool();
    // must already exist...
    let x = ext_find_by_iri(conn, new_verb.id_as_str())
        .await
        .map_err(|x| x.with_status(Status::NotFound))?;
    patch_it(c, conn, x.rid, x.verb, new_verb).await
}

/// Update existing Verb merging `display` fields.
#[patch("/<rid>", data = "<body>")]
async fn patch_rid(
    c: Headers,
    rid: i32,
    body: &str,
    db: &State<DB>,
    user: User,
) -> Result<WithETag, MyError> {
    debug!("----- patch_rid ----- {}", user);
    user.can_use_verbs()?;

    let new_verb = parse_verb(body)?;
    let conn = db.pool();
    // must already exist...
    let old_verb = ext_find_by_rid(conn, rid)
        .await
        .map_err(|x| x.with_status(Status::NotFound))?;
    patch_it(c, conn, rid, old_verb, new_verb).await
}

async fn patch_it(
    c: Headers,
    conn: &PgPool,
    rid: i32,
    mut old_verb: Verb,
    new_verb: Verb,
) -> Result<WithETag, MyError> {
    // proceed if pre-conditions exist + pass...
    if c.has_no_conditionals() {
        Err(MyError::HTTP {
            status: Status::Conflict,
            info: "Patching existing Verb w/ no pre-conditions is NOT allowed".into(),
        })
    } else {
        debug!("old_verb = {}", old_verb);
        let x = serde_json::to_string(&old_verb).map_err(|x| MyError::Data(DataError::JSON(x)))?;
        let etag = etag_from_str(&x);
        debug!("etag (old) = {}", etag);
        match eval_preconditions!(&etag, c) {
            s if s != Status::Ok => Err(MyError::HTTP {
                status: s,
                info: "Failed pre-condition(s)".into(),
            }),
            _ => {
                if new_verb == old_verb {
                    info!("Old + new Verbs are identical. Pass");
                    Ok(no_content(&etag))
                } else if !old_verb.extend(new_verb) {
                    info!("Old + merged versions are identical. Pass");
                    Ok(no_content(&etag))
                } else {
                    debug!("patched_verb = {}", old_verb);
                    ext_update(conn, rid, &old_verb).await?;
                    let x = serde_json::to_string(&old_verb)
                        .map_err(|x| MyError::Data(DataError::JSON(x)))?;
                    let etag = etag_from_str(&x);
                    debug!("etag (new) = {}", etag);
                    Ok(no_content(&etag))
                }
            }
        }
    }
}

#[get("/?<iri>")]
async fn get_iri(iri: &str, db: &State<DB>, user: User) -> Result<ETaggedResource, MyError> {
    debug!("----- get_iri ----- {}", user);
    user.can_use_verbs()?;

    let iri = if IriStr::new(iri).is_err() {
        warn!(
            "This <{}> is not a valid IRI. Assume it's an alias + continue",
            iri
        );
        let iri2 = format!("http://adlnet.gov/expapi/verbs/{iri}");
        // is it valid now?
        if IriStr::new(&iri2).is_err() {
            return Err(MyError::HTTP {
                status: Status::BadRequest,
                info: format!("Input <{iri}> is not a valid IRI nor an alias of one").into(),
            });
        } else {
            iri2
        }
    } else {
        iri.to_owned()
    };

    let x = ext_find_by_iri(db.pool(), &iri)
        .await
        .map_err(|x| x.with_status(Status::NotFound))?;
    tag_n_bag_it::<Verb>(x.verb)
}

#[get("/<rid>")]
async fn get_rid(rid: i32, db: &State<DB>, user: User) -> Result<ETaggedResource, MyError> {
    debug!("----- get_rid ----- {}", user);
    user.can_use_verbs()?;

    let x = ext_find_by_rid(db.pool(), rid)
        .await
        .map_err(|x| x.with_status(Status::NotFound))?;
    tag_n_bag_it::<Verb>(x)
}

#[get("/aggregates")]
async fn get_aggregates(db: &State<DB>, user: User) -> Result<ETaggedResource, MyError> {
    debug!("----- get_aggregates ----- {}", user);
    user.can_use_verbs()?;

    let x = ext_compute_aggregates(db.pool()).await?;
    tag_n_bag_it::<Aggregates>(x)
}

#[get("/")]
async fn get_some(
    q: QueryParams<'_>,
    db: &State<DB>,
    user: User,
) -> Result<ETaggedResource, MyError> {
    debug!("----- get_some ----- {}", user);
    user.can_use_verbs()?;

    debug!("q = {}", q);
    let x = ext_find_some(db.pool(), q).await?;
    tag_n_bag_it::<Vec<VerbUI>>(x)
}

fn tag_n_bag_it<T: Serialize>(resource: T) -> Result<ETaggedResource, MyError> {
    let json = serde_json::to_string(&resource).map_err(|x| MyError::Data(DataError::JSON(x)))?;
    debug!("json = {}", json);
    let etag = etag_from_str(&json);
    debug!("etag = {}", etag);

    Ok(ETaggedResource {
        inner: json,
        etag: Header::new(header::ETAG.as_str(), etag.to_string()),
    })
}