extern crate fediwiki;

mod common;

use crate::common::{
    create_article, edit_article, edit_article_with_conflict, follow_instance, get_article,
    get_query, post, TestData, TEST_ARTICLE_DEFAULT_TEXT,
};
use common::get;
use fediwiki::api::{
    ApiConflict, EditArticleData, ForkArticleData, ResolveObject, SearchArticleData,
};
use fediwiki::error::MyResult;
use fediwiki::federation::objects::article::DbArticle;
use fediwiki::federation::objects::edit::ApubEdit;
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();

    // create article
    let title = "Manu_Chao".to_string();
    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.ap_id).await?;
    assert_eq!(title, get_res.title);
    assert_eq!(TEST_ARTICLE_DEFAULT_TEXT, get_res.text);
    assert!(get_res.local);

    // error on article which wasnt federated
    let not_found = get_article(data.hostname_beta, &create_res.ap_id).await;
    assert!(not_found.is_err());

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

    let search_form = SearchArticleData {
        title: title.clone(),
    };
    let search_res: Vec<DbArticle> =
        get_query(data.hostname_alpha, "search", Some(search_form)).await?;
    assert_eq!(1, search_res.len());
    assert_eq!(edit_res, search_res[0]);

    data.stop()
}

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

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

    let create_res = create_article(data.hostname_alpha, title.clone()).await;
    assert!(create_res.is_err());

    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!(1, 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\n".to_string(),
        previous_version: create_res.latest_version,
        resolve_conflict_id: None,
    };
    edit_article(data.hostname_alpha, &edit_form).await?;

    // article is not yet on beta
    let get_res = get_article(data.hostname_beta, &create_res.ap_id).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.ap_id).await?;
    assert_eq!(create_res.ap_id, get_res.ap_id);
    assert_eq!(title, get_res.title);
    assert_eq!(2, 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.ap_id).await?;
    assert_eq!(create_res.title, get_res.title);
    assert_eq!(1, 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(),
        previous_version: get_res.latest_version,
        resolve_conflict_id: None,
    };
    let edit_res = edit_article(data.hostname_beta, &edit_form).await?;
    assert_eq!(edit_res.text, edit_form.new_text);
    assert_eq!(edit_res.edits.len(), 2);
    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.ap_id).await?;
    assert_eq!(edit_res.title, get_res.title);
    assert_eq!(edit_res.edits.len(), 2);
    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, &create_res.ap_id).await?;
    assert_eq!(create_res.title, get_res.title);
    assert_eq!(1, get_res.edits.len());
    assert!(!get_res.local);

    let get_res = get_article(data.hostname_gamma, &create_res.ap_id).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.clone(),
        new_text: "Lorem Ipsum 2".to_string(),
        previous_version: get_res.latest_version,
        resolve_conflict_id: None,
    };
    let edit_res = edit_article(data.hostname_alpha, &edit_form).await?;
    assert_eq!(edit_form.new_text, edit_res.text);
    assert_eq!(2, edit_res.edits.len());
    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, &create_res.ap_id).await?;
    assert_eq!(edit_res.title, get_res.title);
    assert_eq!(edit_res.edits.len(), 2);
    assert_eq!(edit_res.text, get_res.text);

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

    data.stop()
}

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

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

    // one user edits article
    let edit_form = EditArticleData {
        ap_id: create_res.ap_id.clone(),
        new_text: "Lorem Ipsum\n".to_string(),
        previous_version: create_res.latest_version.clone(),
        resolve_conflict_id: None,
    };
    let edit_res = edit_article(data.hostname_alpha, &edit_form).await?;
    assert_eq!(edit_res.text, edit_form.new_text);
    assert_eq!(2, edit_res.edits.len());

    // another user edits article, without being aware of previous edit
    let edit_form = EditArticleData {
        ap_id: create_res.ap_id.clone(),
        new_text: "Ipsum Lorem\n".to_string(),
        previous_version: create_res.latest_version,
        resolve_conflict_id: None,
    };
    let edit_res = edit_article_with_conflict(data.hostname_alpha, &edit_form)
        .await?
        .unwrap();
    assert_eq!("<<<<<<< ours\nIpsum Lorem\n||||||| original\nsome\nexample\ntext\n=======\nLorem Ipsum\n>>>>>>> theirs\n", edit_res.three_way_merge);

    let conflicts: Vec<ApiConflict> =
        get_query(data.hostname_alpha, "edit_conflicts", None::<()>).await?;
    assert_eq!(1, conflicts.len());
    assert_eq!(conflicts[0], edit_res);

    let edit_form = EditArticleData {
        ap_id: create_res.ap_id.clone(),
        new_text: "Lorem Ipsum and Ipsum Lorem\n".to_string(),
        previous_version: edit_res.previous_version,
        resolve_conflict_id: Some(edit_res.id),
    };
    let edit_res = edit_article(data.hostname_alpha, &edit_form).await?;
    assert_eq!(edit_form.new_text, edit_res.text);

    let conflicts: Vec<ApiConflict> =
        get_query(data.hostname_alpha, "edit_conflicts", None::<()>).await?;
    assert_eq!(0, conflicts.len());

    data.stop()
}

#[tokio::test]
#[serial]
async fn test_federated_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\n".to_string(),
        previous_version: create_res.latest_version.clone(),
        resolve_conflict_id: None,
    };
    let edit_res = edit_article(data.hostname_alpha, &edit_form).await?;
    assert_eq!(edit_res.text, edit_form.new_text);
    assert_eq!(2, edit_res.edits.len());
    assert!(!edit_res.local);
    assert!(edit_res.edits[1]
        .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.clone(),
        new_text: "aaaa\n".to_string(),
        previous_version: create_res.latest_version,
        resolve_conflict_id: None,
    };
    let edit_res = edit_article(data.hostname_gamma, &edit_form).await?;
    assert_ne!(edit_form.new_text, edit_res.text);
    assert_eq!(2, edit_res.edits.len());
    assert!(!edit_res.local);

    let conflicts: Vec<ApiConflict> =
        get_query(data.hostname_gamma, "edit_conflicts", None::<()>).await?;
    assert_eq!(1, conflicts.len());

    // resolve the conflict
    let edit_form = EditArticleData {
        ap_id: create_res.ap_id,
        new_text: "aaaa\n".to_string(),
        previous_version: conflicts[0].previous_version.clone(),
        resolve_conflict_id: Some(conflicts[0].id),
    };
    let edit_res = edit_article(data.hostname_gamma, &edit_form).await?;
    assert_eq!(edit_form.new_text, edit_res.text);
    assert_eq!(3, edit_res.edits.len());

    let conflicts: Vec<ApubEdit> =
        get_query(data.hostname_gamma, "edit_conflicts", None::<()>).await?;
    assert_eq!(0, conflicts.len());

    data.stop()
}

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

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

    // one user edits article
    let edit_form = EditArticleData {
        ap_id: create_res.ap_id.clone(),
        new_text: "my\nexample\ntext\n".to_string(),
        previous_version: create_res.latest_version.clone(),
        resolve_conflict_id: None,
    };
    let edit_res = edit_article(data.hostname_alpha, &edit_form).await?;
    assert_eq!(edit_res.text, edit_form.new_text);
    assert_eq!(2, edit_res.edits.len());

    // another user edits article, without being aware of previous edit
    let edit_form = EditArticleData {
        ap_id: create_res.ap_id.clone(),
        new_text: "some\nexample\narticle\n".to_string(),
        previous_version: create_res.latest_version,
        resolve_conflict_id: None,
    };
    let edit_res = edit_article(data.hostname_alpha, &edit_form).await?;
    let conflicts: Vec<ApiConflict> =
        get_query(data.hostname_alpha, "edit_conflicts", None::<()>).await?;
    assert_eq!(0, conflicts.len());
    assert_eq!(3, edit_res.edits.len());
    assert_eq!("my\nexample\narticle\n", edit_res.text);

    data.stop()
}

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

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

    // fetch on beta
    let resolve_object = ResolveObject {
        id: create_res.ap_id.into_inner(),
    };
    let resolved_article =
        get_query::<DbArticle, _>(data.hostname_beta, "resolve_article", Some(resolve_object))
            .await?;
    assert_eq!(create_res.edits.len(), resolved_article.edits.len());

    // fork the article to local instance
    let fork_form = ForkArticleData {
        ap_id: resolved_article.ap_id.clone(),
    };
    let fork_res: DbArticle = post(data.hostname_beta, "article/fork", &fork_form).await?;
    assert_eq!(resolved_article.title, fork_res.title);
    assert_eq!(resolved_article.text, fork_res.text);
    assert_eq!(resolved_article.edits, fork_res.edits);
    assert_eq!(resolved_article.latest_version, fork_res.latest_version);
    assert_ne!(resolved_article.ap_id, fork_res.ap_id);
    assert!(fork_res.local);

    let beta_instance: DbInstance = get(data.hostname_beta, "instance").await?;
    assert_eq!(fork_res.instance, beta_instance.ap_id);

    // now search returns two articles for this title (original and forked)
    let search_form = SearchArticleData {
        title: title.clone(),
    };
    let search_res: Vec<DbArticle> =
        get_query(data.hostname_beta, "search", Some(search_form)).await?;
    assert_eq!(2, search_res.len());

    data.stop()
}