extern crate fediwiki;

mod common;

use crate::common::{
    create_article, edit_article, follow_instance, get_article, get_query, TestData,
};
use common::get;
use fediwiki::api::{EditArticleData, ResolveObject};
use fediwiki::error::MyResult;
use fediwiki::federation::objects::article::DbArticle;
use fediwiki::federation::objects::instance::DbInstance;
use serial_test::serial;
use url::Url;

#[tokio::test]
#[serial]
async fn test_create_read_and_edit_article() -> MyResult<()> {
    let data = TestData::start();

    // error on nonexistent article
    let title = "Manu_Chao".to_string();
    let not_found = get_article(data.hostname_alpha, &title).await;
    assert!(not_found.is_err());

    // create article
    let create_res = create_article(data.hostname_alpha, title.clone()).await?;
    assert_eq!(title, create_res.title);
    assert!(create_res.local);

    // now article can be read
    let get_res = get_article(data.hostname_alpha, &create_res.title).await?;
    assert_eq!(title, get_res.title);
    assert!(get_res.text.is_empty());
    assert!(get_res.local);

    // edit article
    let edit_form = EditArticleData {
        ap_id: create_res.ap_id.clone(),
        new_text: "Lorem Ipsum 2".to_string(),
    };
    let edit_res = edit_article(data.hostname_alpha, &create_res.title, &edit_form).await?;
    assert_eq!(edit_form.new_text, edit_res.text);
    assert_eq!(1, edit_res.edits.len());

    data.stop()
}

#[tokio::test]
#[serial]
async fn test_follow_instance() -> MyResult<()> {
    let data = TestData::start();

    // check initial state
    let alpha_instance: DbInstance = get(data.hostname_alpha, "instance").await?;
    assert_eq!(0, alpha_instance.follows.len());
    let beta_instance: DbInstance = get(data.hostname_beta, "instance").await?;
    assert_eq!(0, beta_instance.followers.len());

    follow_instance(data.hostname_alpha, &data.hostname_beta).await?;

    // check that follow was federated
    let beta_instance: DbInstance = get(data.hostname_beta, "instance").await?;
    assert_eq!(1, beta_instance.followers.len());

    let alpha_instance: DbInstance = get(data.hostname_alpha, "instance").await?;
    assert_eq!(1, alpha_instance.follows.len());

    data.stop()
}

#[tokio::test]
#[serial]
async fn test_synchronize_articles() -> MyResult<()> {
    let data = TestData::start();

    // create article on alpha
    let title = "Manu_Chao".to_string();
    let create_res = create_article(data.hostname_alpha, title.clone()).await?;
    assert_eq!(title, create_res.title);
    assert_eq!(0, create_res.edits.len());
    assert!(create_res.local);

    // edit the article
    let edit_form = EditArticleData {
        ap_id: create_res.ap_id.clone(),
        new_text: "Lorem Ipsum 2".to_string(),
    };
    edit_article(data.hostname_alpha, &title, &edit_form).await?;

    // article is not yet on beta
    let get_res = get_article(data.hostname_beta, &create_res.title).await;
    assert!(get_res.is_err());

    // fetch alpha instance on beta, articles are also fetched automatically
    let resolve_object = ResolveObject {
        id: Url::parse(&format!("http://{}", data.hostname_alpha))?,
    };
    get_query::<DbInstance, _>(data.hostname_beta, "resolve_instance", Some(resolve_object))
        .await?;

    // get the article and compare
    let get_res = get_article(data.hostname_beta, &create_res.title).await?;
    assert_eq!(create_res.ap_id, get_res.ap_id);
    assert_eq!(title, get_res.title);
    assert_eq!(1, get_res.edits.len());
    assert_eq!(edit_form.new_text, get_res.text);
    assert!(!get_res.local);

    data.stop()
}

#[tokio::test]
#[serial]
async fn test_edit_local_article() -> MyResult<()> {
    let data = TestData::start();

    follow_instance(data.hostname_alpha, data.hostname_beta).await?;

    // create new article
    let title = "Manu_Chao".to_string();
    let create_res = create_article(data.hostname_beta, title.clone()).await?;
    assert_eq!(title, create_res.title);
    assert!(create_res.local);

    // article should be federated to alpha
    let get_res = get_article(data.hostname_alpha, &create_res.title).await?;
    assert_eq!(create_res.title, get_res.title);
    assert_eq!(0, get_res.edits.len());
    assert!(!get_res.local);
    assert_eq!(create_res.text, get_res.text);

    // edit the article
    let edit_form = EditArticleData {
        ap_id: create_res.ap_id,
        new_text: "Lorem Ipsum 2".to_string(),
    };
    let edit_res = edit_article(data.hostname_beta, &create_res.title, &edit_form).await?;
    assert_eq!(edit_res.text, edit_form.new_text);
    assert_eq!(edit_res.edits.len(), 1);
    assert!(edit_res.edits[0]
        .id
        .to_string()
        .starts_with(&edit_res.ap_id.to_string()));

    // edit should be federated to alpha
    let get_res = get_article(data.hostname_alpha, &edit_res.title).await?;
    assert_eq!(edit_res.title, get_res.title);
    assert_eq!(edit_res.edits.len(), 1);
    assert_eq!(edit_res.text, get_res.text);

    data.stop()
}

#[tokio::test]
#[serial]
async fn test_edit_remote_article() -> MyResult<()> {
    let data = TestData::start();

    follow_instance(data.hostname_alpha, data.hostname_beta).await?;
    follow_instance(data.hostname_gamma, data.hostname_beta).await?;

    // create new article
    let title = "Manu_Chao".to_string();
    let create_res = create_article(data.hostname_beta, title.clone()).await?;
    assert_eq!(title, create_res.title);
    assert!(create_res.local);

    // article should be federated to alpha and gamma
    let get_res = get_article(data.hostname_alpha, &title).await?;
    assert_eq!(create_res.title, get_res.title);
    assert_eq!(0, get_res.edits.len());
    assert!(!get_res.local);

    let get_res = get_article(data.hostname_gamma, &title).await?;
    assert_eq!(create_res.title, get_res.title);
    assert_eq!(create_res.text, get_res.text);

    let edit_form = EditArticleData {
        ap_id: create_res.ap_id,
        new_text: "Lorem Ipsum 2".to_string(),
    };
    let edit_res = edit_article(data.hostname_alpha, &title, &edit_form).await?;
    assert_eq!(edit_res.text, edit_form.new_text);
    assert_eq!(edit_res.edits.len(), 1);
    assert!(!edit_res.local);
    assert!(edit_res.edits[0]
        .id
        .to_string()
        .starts_with(&edit_res.ap_id.to_string()));

    // edit should be federated to beta and gamma
    let get_res = get_article(data.hostname_alpha, &title).await?;
    assert_eq!(edit_res.title, get_res.title);
    assert_eq!(edit_res.edits.len(), 1);
    assert_eq!(edit_res.text, get_res.text);

    let get_res = get_article(data.hostname_gamma, &title).await?;
    assert_eq!(edit_res.title, get_res.title);
    assert_eq!(edit_res.edits.len(), 1);
    assert_eq!(edit_res.text, get_res.text);

    data.stop()
}

#[tokio::test]
#[serial]
async fn test_edit_conflict() -> MyResult<()> {
    let data = TestData::start();

    follow_instance(data.hostname_alpha, data.hostname_beta).await?;

    // create new article
    let title = "Manu_Chao".to_string();
    let create_res = create_article(data.hostname_beta, title.clone()).await?;
    assert_eq!(title, create_res.title);
    assert!(create_res.local);

    // fetch article to gamma
    let resolve_object = ResolveObject {
        id: create_res.ap_id.inner().clone(),
    };
    let resolve_res: DbArticle =
        get_query(data.hostname_gamma, "resolve_article", Some(resolve_object)).await?;
    assert_eq!(create_res.text, resolve_res.text);

    // alpha edits article
    let edit_form = EditArticleData {
        ap_id: create_res.ap_id.clone(),
        new_text: "Lorem Ipsum".to_string(),
    };
    let edit_res = edit_article(data.hostname_alpha, &create_res.title, &edit_form).await?;
    assert_eq!(edit_res.text, edit_form.new_text);
    assert_eq!(edit_res.edits.len(), 1);
    assert!(!edit_res.local);
    assert!(edit_res.edits[0]
        .id
        .to_string()
        .starts_with(&edit_res.ap_id.to_string()));

    // gamma also edits, as its not the latest version there is a conflict. local version should
    // not be updated with this conflicting version, instead user needs to handle the conflict
    let edit_form = EditArticleData {
        ap_id: create_res.ap_id,
        new_text: "aaaa".to_string(),
    };
    let edit_res = edit_article(data.hostname_gamma, &create_res.title, &edit_form).await?;
    assert_eq!(create_res.text, edit_res.text);
    assert_eq!(0, edit_res.edits.len());
    assert!(!edit_res.local);

    // TODO: need to federate the conflict as `Reject` and then resolve it

    data.stop()
}