mod admin;
mod common;
mod english;
mod spanish;

use swanling::prelude::*;

use crate::admin::*;
use crate::english::*;
use crate::spanish::*;

/// Defines the actual load test. Each task set simulates a type of user.
///  - Anonymous English user: loads the English version of all pages
///  - Anonymous Spanish user: loads the Spanish version of all pages
fn main() -> Result<(), SwanlingError> {
    let _swanling_metrics = SwanlingAttack::initialize()?
        .register_taskset(
            taskset!("Anonymous English user")
                .set_weight(40)?
                .set_wait_time(0, 3)?
                .register_task(task!(front_page_en).set_name("anon /").set_weight(2)?)
                .register_task(task!(basic_page_en).set_name("anon /en/basicpage"))
                .register_task(task!(article_listing_en).set_name("anon /en/articles/"))
                .register_task(
                    task!(article_en)
                        .set_name("anon /en/articles/%")
                        .set_weight(2)?,
                )
                .register_task(task!(recipe_listing_en).set_name("anon /en/recipes/"))
                .register_task(
                    task!(recipe_en)
                        .set_name("anon /en/recipes/%")
                        .set_weight(4)?,
                )
                .register_task(task!(page_by_nid).set_name("anon /node/%nid"))
                .register_task(
                    task!(term_listing_en)
                        .set_name("anon /en term")
                        .set_weight(2)?,
                )
                .register_task(task!(search_en).set_name("anon /en/search"))
                .register_task(task!(anonymous_contact_form_en).set_name("anon /en/contact")),
        )
        .register_taskset(
            taskset!("Anonymous Spanish user")
                .set_weight(9)?
                .set_wait_time(0, 3)?
                .register_task(task!(front_page_es).set_name("anon /es/").set_weight(2)?)
                .register_task(task!(basic_page_es).set_name("anon /es/basicpage"))
                .register_task(task!(article_listing_es).set_name("anon /es/articles/"))
                .register_task(
                    task!(article_es)
                        .set_name("anon /es/articles/%")
                        .set_weight(2)?,
                )
                .register_task(task!(recipe_listing_es).set_name("anon /es/recipes/"))
                .register_task(
                    task!(recipe_es)
                        .set_name("anon /es/recipes/%")
                        .set_weight(4)?,
                )
                .register_task(
                    task!(term_listing_es)
                        .set_name("anon /es term")
                        .set_weight(2)?,
                )
                .register_task(task!(search_es).set_name("anon /es/search"))
                .register_task(task!(anonymous_contact_form_es).set_name("anon /es/contact")),
        )
        .register_taskset(
            taskset!("Admin user")
                .set_weight(1)?
                .set_wait_time(3, 10)?
                .register_task(task!(log_in).set_on_start().set_name("auth /en/user/login"))
                .register_task(task!(front_page_en).set_name("auth /").set_weight(2)?)
                .register_task(task!(article_listing_en).set_name("auth /en/articles/"))
                .register_task(
                    task!(edit_article)
                        .set_name("auth /en/node/%/edit")
                        .set_weight(2)?,
                ),
        )
        .set_default(SwanlingDefault::Host, "https://drupal-9.ddev.site/")?
        .execute()?
        .print();

    Ok(())
}