mod common;

use chrono::{Days, Utc};
use common::test_context::TestContext;
use diesel::{Connection, SqliteConnection};
use schist_fakes::{
    account::make_fake_account,
    budget_update::make_fake_budget_update_builder,
    category::make_fake_category,
    transaction::make_fake_transaction_builder,
    transaction_categorisation::make_fake_transaction_categorisation_builder,
};
use schist_models::{
    account::Account,
    budget::Budget,
    budget_update::BudgetUpdate,
    category::Category,
    date_utc::DateUtc,
    transaction::Transaction,
    transaction_categorisation::TransactionCategorisation,
};
use schist_queries as queries;

fn given_categories(connection: &mut SqliteConnection) -> Vec<Category> {
    let categories = vec![make_fake_category(0), make_fake_category(1)];
    let _ = queries::categories::insert_categories(&categories, connection).unwrap();
    categories
}

fn given_transactions(connection: &mut SqliteConnection) -> Vec<Transaction> {
    let transactions = vec![
        make_fake_transaction_builder(0).quantity(1).build().unwrap(),
        make_fake_transaction_builder(1).quantity(2).build().unwrap(),
        make_fake_transaction_builder(2).quantity(4).build().unwrap(),
        make_fake_transaction_builder(3).quantity(8).build().unwrap(),
    ];
    let _ = queries::transactions::insert_transactions(&transactions, connection).unwrap();
    transactions
}

fn given_accounts(connection: &mut SqliteConnection) -> Vec<Account> {
    let accounts = vec![
        make_fake_account(0),
    ];
    let _ = queries::accounts::insert_accounts(&accounts, connection);
    accounts
}

fn given_transaction_categorisations(connection: &mut SqliteConnection) -> Vec<TransactionCategorisation> {
    let transaction_categorisations = vec![
        make_fake_transaction_categorisation_builder(0).quantity(1).transaction_id(0).category_id(0).build().unwrap(),
        make_fake_transaction_categorisation_builder(1).quantity(1).transaction_id(1).category_id(0).build().unwrap(),
        make_fake_transaction_categorisation_builder(2).quantity(1).transaction_id(1).category_id(0).build().unwrap(),
        make_fake_transaction_categorisation_builder(3).quantity(4).transaction_id(2).category_id(1).build().unwrap(),
        make_fake_transaction_categorisation_builder(4).quantity(3).transaction_id(3).category_id(1).build().unwrap(),
        make_fake_transaction_categorisation_builder(5).quantity(5).transaction_id(3).category_id(1).build().unwrap(),
    ];
    let _ = queries::transaction_categorisations::insert_transaction_categorisations(&transaction_categorisations, connection).unwrap();
    transaction_categorisations
}

fn given_budget_updates(connection: &mut SqliteConnection) -> Vec<BudgetUpdate> {
    let now = Utc::now();
    let twenty_days_ago = DateUtc::from(now.checked_sub_days(Days::new(20)).unwrap());
    let ten_days_ago = DateUtc::from(now.checked_sub_days(Days::new(10)).unwrap());
    let budget_updates = vec![
        make_fake_budget_update_builder(0).category_id(0).date(twenty_days_ago.clone()).new_budget(100).new_period(1).build().unwrap(),
        make_fake_budget_update_builder(1).category_id(0).date(ten_days_ago.clone()).new_budget(50).new_period(2).build().unwrap(),
        make_fake_budget_update_builder(2).category_id(1).date(twenty_days_ago.clone()).new_budget(90).new_period(4).build().unwrap(),
        make_fake_budget_update_builder(3).category_id(1).date(ten_days_ago.clone()).new_budget(40).new_period(8).build().unwrap(),
    ];
    let _ = queries::budget_updates::insert_budget_updates(&budget_updates, connection);
    budget_updates
}

#[test]
fn when_insert_categories_then_returns_num_categories() {
    let context = TestContext::new();
    let connection = &mut SqliteConnection::establish(&context.db_url).unwrap();
    let categories = [make_fake_category(0), make_fake_category(1)];

    let num_categories_inserted = queries::categories::insert_categories(&categories, connection).unwrap();

    assert_eq!(num_categories_inserted, 2);
}

#[test]
fn given_categories_when_get_all_categories_then_return_categories() {
    let context = TestContext::new();
    let connection = &mut SqliteConnection::establish(&context.db_url).unwrap();
    let categories = given_categories(connection);

    let categories_in_db = queries::categories::get_all_categories(connection).unwrap();

    assert_eq!(categories_in_db.len(), categories.len());
    for category in &categories {
        assert!(categories_in_db.iter().any(|category_in_db| category_in_db.eq(&category)));
    }
    for category_in_db in categories_in_db {
        assert!(categories.iter().any(|category| category.eq(&category_in_db)));
    }
}

#[test]
fn given_categories_when_delete_all_categories_then_return_num_of_categories() {
    let context = TestContext::new();
    let connection = &mut SqliteConnection::establish(&context.db_url).unwrap();
    let categories = given_categories(connection);

    let num_categories_deleted = queries::categories::delete_all_categories(connection).unwrap();

    assert_eq!(num_categories_deleted, categories.len());
}

#[test]
fn given_deleted_categories_when_get_all_categories_then_return_empty() {
    let context = TestContext::new();
    let connection = &mut SqliteConnection::establish(&context.db_url).unwrap();
    given_categories(connection);
    let _ = queries::categories::delete_all_categories(connection).unwrap();

    let categories_returned = queries::categories::get_all_categories(connection).unwrap();

    assert_eq!(categories_returned.len(), 0);
}

#[test]
fn given_categories_and_no_transactions_when_get_all_balances_then_return_all_zero() {
    let context = TestContext::new();
    let connection = &mut SqliteConnection::establish(&context.db_url).unwrap();
    given_categories(connection);

    let balances = queries::categories::get_all_category_balances(connection);

    assert!(balances.is_ok());
    let mut balances = balances.unwrap();
    balances.sort_by(|b1, b2| b1.category_id.cmp(&b2.category_id));
    assert_eq!(balances.len(), 2);
    for balance in balances {
        assert_eq!(0, balance.balance);
    }
}

#[test]
fn given_categories_and_transaction_categorisations_when_get_all_balances_then_return_balances() {
    let context = TestContext::new();
    let connection = &mut SqliteConnection::establish(&context.db_url).unwrap();
    let categories = given_categories(connection);
    let _accounts = given_accounts(connection);
    let _transactions = given_transactions(connection);
    let transaction_categorisations = given_transaction_categorisations(connection);

    let balances = queries::categories::get_all_category_balances(connection);

    assert!(balances.is_ok());
    let mut balances = balances.unwrap();
    balances.sort_by(|b1, b2| b1.category_id.cmp(&b2.category_id));
    assert_eq!(balances.len(), categories.len());
    for category in categories {
        let balance = balances.iter().find(|b| b.category_id == category.id).unwrap();
        let expected_balance: i32 = transaction_categorisations
            .iter()
            .filter(|t| t.category_id == category.id)
            .map(|t| t.quantity)
            .sum();
        assert_eq!(expected_balance, balance.balance);
    }
}

#[test]
fn given_categories_when_get_all_ids_then_return_all_ids() {
    let context = TestContext::new();
    let connection = &mut SqliteConnection::establish(&context.db_url).unwrap();
    let categories = given_categories(connection);

    let ids = queries::categories::get_all_category_ids(connection).unwrap();

    assert_eq!(categories.len(), ids.len());
    for category in &categories {
        assert!(ids.iter().any(|id| category.id == *id));
    }
    for id in ids {
        assert!(categories.iter().any(|category| category.id == id));
    }
}

#[test]
fn given_categories_and_no_budget_updates_when_get_most_recent_budget_per_category_id_then_return_none() {
    let context = TestContext::new();
    let connection = &mut SqliteConnection::establish(&context.db_url).unwrap();
    let categories = given_categories(connection);

    let most_recent_budget_per_category_id = queries::categories::get_most_recent_category_budget_per_category_id(connection);

    assert!(most_recent_budget_per_category_id.is_ok());
    let most_recent_budget_per_category_id = most_recent_budget_per_category_id.unwrap();
    assert_eq!(categories.len(), most_recent_budget_per_category_id.len());
    for category in categories {
        let most_recent_budget = most_recent_budget_per_category_id.iter().find(|b| b.category_id == category.id);
        assert!(most_recent_budget.is_some());
        let most_recent_budget = most_recent_budget.unwrap();
        assert!(most_recent_budget.budget.is_none());
    }
}

#[test]
fn given_categories_and_budget_updates_when_get_most_recent_budget_per_category_id_then_return_most_recent_budget() {
    let context = TestContext::new();
    let connection = &mut SqliteConnection::establish(&context.db_url).unwrap();
    let categories = given_categories(connection);
    let budget_updates = given_budget_updates(connection);

    let most_recent_budget_per_category_id = queries::categories::get_most_recent_category_budget_per_category_id(connection);

    assert!(most_recent_budget_per_category_id.is_ok());
    let most_recent_budget_per_category_id = most_recent_budget_per_category_id.unwrap();
    assert_eq!(categories.len(), most_recent_budget_per_category_id.len());
    for category in categories {
        let most_recent_budget_update = budget_updates
            .iter()
            .filter(|bu| bu.category_id == category.id)
            .max_by_key(|bu| bu.date.clone())
            .unwrap();

        let most_recent_budget = most_recent_budget_per_category_id.iter().find(|b| b.category_id == category.id);
        assert!(most_recent_budget.is_some());
        let most_recent_budget = most_recent_budget.unwrap();
        assert!(most_recent_budget.budget.is_some());
        let most_recent_budget = most_recent_budget.budget.unwrap();
        assert_eq!(most_recent_budget, Budget {
            quantity: most_recent_budget_update.new_budget,
            period: most_recent_budget_update.new_period,
        });
    }
}