use elseware::common::serverstate::{ClientId, ServerState};
use elseware::entity::gateway::{EntityGateway, InMemoryGateway};
use elseware::common::leveltable::CharacterLevelTable;
use elseware::ship::ship::{ShipServerState, SendShipPacket, RecvShipPacket};
use elseware::ship::monster::MonsterType;
use elseware::ship::location::RoomId;

use libpso::packet::ship::*;
use libpso::packet::messages::*;

#[path = "common.rs"]
mod common;
use common::*;

#[async_std::test]
async fn test_character_gains_exp() {
    let mut entity_gateway = InMemoryGateway::default();

    let (_user1, _char1) = new_user_character(&mut entity_gateway, "a1", "a", 1).await;

    let mut ship = Box::new(ShipServerState::builder()
        .gateway(entity_gateway.clone())
        .build());
    log_in_char(&mut ship, ClientId(1), "a1", "a").await;
    join_lobby(&mut ship, ClientId(1)).await;
    create_room(&mut ship, ClientId(1), "room", "").await;

    let (enemy_id, exp) = {
        //let room = ship.blocks.0[0].rooms.get(RoomId(0)).as_ref().unwrap();
        ship.blocks.0[0].rooms.with(RoomId(0), |room| Box::pin(async move {
            let (enemy_id, map_enemy) = (0..).filter_map(|i| {
                room.maps.enemy_by_id(i).map(|enemy| {
                    (i, enemy)
                }).ok()
            }).next().unwrap();
            let map_enemy_stats = room.monster_stats.get(&map_enemy.monster).unwrap();
            (enemy_id, map_enemy_stats.exp)
        })).await.unwrap()
    };

    ship.handle(ClientId(1), RecvShipPacket::Message(Message::new(GameMessage::RequestExp(RequestExp {
        client: enemy_id as u8,
        target: 16,
        enemy_id: enemy_id as u16,
        client_id: 0,
        unused: 0,
        last_hitter: 1,
    })))).await.unwrap();

    ship.clients.with(ClientId(1), |client| Box::pin(async move {
        assert!(exp == client.character.exp);
    })).await.unwrap();
}

#[async_std::test]
async fn test_character_levels_up() {
    let mut entity_gateway = InMemoryGateway::default();

    let (_user1, mut char1) = new_user_character(&mut entity_gateway, "a1", "a", 1).await;
    char1.exp = 49;
    entity_gateway.save_character(&char1).await.unwrap();

    let mut ship = Box::new(ShipServerState::builder()
        .gateway(entity_gateway.clone())
        .build());
    log_in_char(&mut ship, ClientId(1), "a1", "a").await;
    join_lobby(&mut ship, ClientId(1)).await;
    create_room(&mut ship, ClientId(1), "room", "").await;

    let enemy_id = ship.blocks.0[0].rooms.with(RoomId(0), |room| Box::pin(async move {
        (0..).filter_map(|i| {
            room.maps.enemy_by_id(i).map(|_| {
                i
            }).ok()
        }).next().unwrap()
    })).await.unwrap();

    let levelup_pkt = ship.handle(ClientId(1), RecvShipPacket::Message(Message::new(GameMessage::RequestExp(RequestExp {
        client: enemy_id as u8,
        target: 16,
        enemy_id: enemy_id as u16,
        client_id: 0,
        unused: 0,
        last_hitter: 1,
    })))).await.unwrap();

    assert!(matches!(levelup_pkt[1].1, SendShipPacket::Message(Message {msg: GameMessage::PlayerLevelUp(PlayerLevelUp {lvl: 2, ..})})));

    let leveltable = CharacterLevelTable::default();
    ship.clients.with(ClientId(1), |client| Box::pin(async move {
        assert!(leveltable.get_level_from_exp(client.character.char_class, client.character.exp) == 2)
    })).await.unwrap();
}

#[async_std::test]
async fn test_character_levels_up_multiple_times() {
    let mut entity_gateway = InMemoryGateway::default();

    let (_user1, _char1) = new_user_character(&mut entity_gateway, "a1", "a", 1).await;

    let mut ship = Box::new(ShipServerState::builder()
        .gateway(entity_gateway.clone())
        .build());
    log_in_char(&mut ship, ClientId(1), "a1", "a").await;
    join_lobby(&mut ship, ClientId(1)).await;
    create_room(&mut ship, ClientId(1), "room", "").await;

    let (enemy_id, exp) = {
        ship.blocks.0[0].rooms.with(RoomId(0), |room| Box::pin(async move {
            let (enemy_id, map_enemy) = (0..).filter_map(|i| {
                room.maps.enemy_by_id(i).ok().and_then(|enemy| {
                    if enemy.monster == MonsterType::DarkFalz2 {
                        Some((i, enemy))
                    }
                    else {
                        None
                    }
                })
            }).next().unwrap();
            let map_enemy_stats = room.monster_stats.get(&map_enemy.monster).unwrap();
            (enemy_id, map_enemy_stats.exp)
        })).await.unwrap()
    };

    let levelup_pkt = ship.handle(ClientId(1), RecvShipPacket::Message(Message::new(GameMessage::RequestExp(RequestExp {
        client: enemy_id as u8,
        target: 16,
        enemy_id: enemy_id as u16,
        client_id: 0,
        unused: 0,
        last_hitter: 1,
    })))).await.unwrap();

    assert!(matches!(levelup_pkt[1].1, SendShipPacket::Message(Message {msg: GameMessage::PlayerLevelUp(PlayerLevelUp {lvl: 8, ..})})));

    ship.clients.with(ClientId(1), |client| Box::pin(async move {
        assert!(exp == client.character.exp);
    })).await.unwrap();
}

#[async_std::test]
async fn test_one_character_gets_full_exp_and_other_attacker_gets_partial() {
    let mut entity_gateway = InMemoryGateway::default();

    let (_user1, _char1) = new_user_character(&mut entity_gateway, "a1", "a", 1).await;
    let (_user2, _char2) = new_user_character(&mut entity_gateway, "a2", "a", 1).await;

    let mut ship = Box::new(ShipServerState::builder()
        .gateway(entity_gateway.clone())
        .build());
    log_in_char(&mut ship, ClientId(1), "a1", "a").await;
    log_in_char(&mut ship, ClientId(2), "a2", "a").await;

    join_lobby(&mut ship, ClientId(1)).await;
    join_lobby(&mut ship, ClientId(2)).await;

    create_room(&mut ship, ClientId(1), "room", "").await;
    join_room(&mut ship, ClientId(2), 0).await;

    let (enemy_id, exp) = ship.blocks.0[0].rooms.with(RoomId(0), |room| Box::pin(async move {
        let (enemy_id, map_enemy) = (0..).filter_map(|i| {
            room.maps.enemy_by_id(i).ok().and_then(|enemy| {
                if enemy.monster == MonsterType::DarkFalz2 {
                    Some((i, enemy))
                }
                else {
                    None
                }
            })
        }).next().unwrap();
        let map_enemy_stats = room.monster_stats.get(&map_enemy.monster).unwrap();
        (enemy_id, map_enemy_stats.exp)
    })).await.unwrap();

    ship.handle(ClientId(1), RecvShipPacket::Message(Message::new(GameMessage::RequestExp(RequestExp {
        client: enemy_id as u8,
        target: 16,
        enemy_id: enemy_id as u16,
        client_id: 0,
        unused: 0,
        last_hitter: 1,
    })))).await.unwrap();

    ship.handle(ClientId(2), RecvShipPacket::Message(Message::new(GameMessage::RequestExp(RequestExp {
        client: enemy_id as u8,
        target: 16,
        enemy_id: enemy_id as u16,
        client_id: 0,
        unused: 0,
        last_hitter: 0,
    })))).await.unwrap();

    ship.clients.with(ClientId(1), |client| Box::pin(async move {
        assert!(client.character.exp == exp);
    })).await.unwrap();
    ship.clients.with(ClientId(2), |client| Box::pin(async move {
        assert!(client.character.exp == (exp as f32 * 0.8) as u32);
    })).await.unwrap();
}