use std::net::Ipv4Addr;
use log::{info};

use elseware::patch::patch::{PatchServerState, generate_patch_tree, load_config, load_motd};
use elseware::login::login::LoginServerState;
use elseware::login::character::CharacterServerState;
use elseware::ship::ship::ShipServerStateBuilder;
use elseware::entity::account::{NewUserAccountEntity, NewUserSettingsEntity};
use elseware::entity::gateway::{EntityGateway, InMemoryGateway, PostgresGateway};
use elseware::entity::character::NewCharacterEntity;
use elseware::entity::item::{NewItemEntity, ItemDetail, ItemLocation};
use elseware::common::interserver::AuthToken;

use elseware::entity::item;
use elseware::common::mainloop::*;

fn setup_logger() {
    let colors = fern::colors::ColoredLevelConfig::new()
        .error(fern::colors::Color::Red)
        .warn(fern::colors::Color::Yellow)
        .info(fern::colors::Color::Green)
        .debug(fern::colors::Color::White)
        .trace(fern::colors::Color::BrightBlack);
    let stdio = fern::Dispatch::new()
        .level(log::LevelFilter::Debug)
        .format(move |out, message, record| {
            out.finish(format_args!(
                "\x1B[{}m[{}][{}][{}] {}\x1B[0m",
                colors.get_color(&record.level()).to_fg_str(),
                chrono::Local::now().format("%Y-%m-%d %H:%M:%S"),
                record.target(),
                record.level(),
                message,
            ))
        })
        .chain(std::io::stdout());
    let fileout = fern::Dispatch::new()
        .level(log::LevelFilter::Trace)
        .chain(fern::log_file(format!("log/elseware-{}.log", chrono::Local::now().format("%Y-%m-%d_%H-%M-%S"))).unwrap());
    fern::Dispatch::new()
        .chain(stdio)
        .chain(fileout)
        .apply().unwrap();
}


fn main() {
    setup_logger();
    async_std::task::block_on(async move {
        //let mut entity_gateway = PostgresGateway::new("localhost", "elsewhere", "elsewhere", "");
        let mut entity_gateway = InMemoryGateway::new();

        for i in 0..5 {
            let fake_user = NewUserAccountEntity {
                email: format!("fake{}@email.com", i),
                username: if i == 0 { "hi".to_string() } else { format!("hi{}", i+1) },
                password: bcrypt::hash("qwer", 5).unwrap(),
                guildcard: i + 1,
                team_id: None,
                banned_until: None,
                muted_until: None,
                flags: 0,
                activated: true,
            };
            let fake_user = entity_gateway.create_user(fake_user).await.unwrap();
            entity_gateway.create_user_settings(NewUserSettingsEntity::new(fake_user.id)).await.unwrap();
            let mut character = NewCharacterEntity::new(fake_user.id);
            character.name = format!("Test Char {}", i*2);
            entity_gateway.create_character(character).await.unwrap();
            let mut character = NewCharacterEntity::new(fake_user.id);
            character.slot = 2;
            character.name = "ItemRefactor".into();
            character.exp = 80000000;
            character.meseta = 999999;
            let character = entity_gateway.create_character(character).await.unwrap();

            for _ in 0..3 {
                entity_gateway.create_item(
                    item::NewItemEntity {
                        item: item::ItemDetail::Weapon(
                            item::weapon::Weapon {
                                weapon: item::weapon::WeaponType::Vulcan,
                                grind: 0,
                                special: None,
                                attrs: [None, None, None],
                                tekked: true,
                                modifiers: Vec::new(),
                            }
                        ),
                        location: item::ItemLocation::Bank {
                            character_id: character.id,
                            name: item::BankName("".to_string())
                        }
                    }).await.unwrap();
            }

            for _ in 0..8 {
                entity_gateway.create_item(
                    NewItemEntity {
                        item: ItemDetail::Tool (
                            item::tool::Tool {
                                tool: item::tool::ToolType::Monomate,
                            }
                        ),
                        location: item::ItemLocation::Bank {
                            character_id: character.id,
                            name: item::BankName("".to_string())
                        }
                    }).await.unwrap();
            }

            let item0 = entity_gateway.create_item(
                NewItemEntity {
                    item: ItemDetail::Weapon(
                        item::weapon::Weapon {
                            weapon: item::weapon::WeaponType::Raygun,
                            grind: 5,
                            special: Some(item::weapon::WeaponSpecial::Hell),
                            attrs: [Some(item::weapon::WeaponAttribute{attr: item::weapon::Attribute::Hit, value: 40}),
                                    Some(item::weapon::WeaponAttribute{attr: item::weapon::Attribute::Dark, value: 30}),
                                    None,],
                            tekked: true,
                            modifiers: Vec::new(),
                        }
                    ),
                    location: ItemLocation::Inventory {
                        character_id: character.id,
                    }
                }).await.unwrap();
            let item1 = entity_gateway.create_item(
                NewItemEntity {
                    item: ItemDetail::Weapon(
                        item::weapon::Weapon {
                            weapon: item::weapon::WeaponType::Handgun,
                            grind: 5,
                            special: Some(item::weapon::WeaponSpecial::Charge),
                            attrs: [Some(item::weapon::WeaponAttribute{attr: item::weapon::Attribute::Hit, value: 40}),
                                    Some(item::weapon::WeaponAttribute{attr: item::weapon::Attribute::Dark, value: 30}),
                                    None,],
                            tekked: true,
                            modifiers: Vec::new(),
                        }
                    ),
                    location: ItemLocation::Inventory {
                        character_id: character.id,
                    }
                }).await.unwrap();
            let item2_w = entity_gateway.create_item(
                NewItemEntity {
                    item: ItemDetail::Weapon(
                        item::weapon::Weapon {
                            weapon: item::weapon::WeaponType::Vjaya,
                            grind: 5,
                            special: Some(item::weapon::WeaponSpecial::Charge),
                            attrs: [Some(item::weapon::WeaponAttribute{attr: item::weapon::Attribute::Hit, value: 40}),
                                    Some(item::weapon::WeaponAttribute{attr: item::weapon::Attribute::Dark, value: 100}),
                                    None,],
                            tekked: true,
                            modifiers: Vec::new(),
                        }
                    ),
                    location: ItemLocation::Inventory {
                        character_id: character.id,
                    }
                }).await.unwrap();
            let item3 = entity_gateway.create_item(
                NewItemEntity {
                    item: ItemDetail::Weapon(
                        item::weapon::Weapon {
                            weapon: item::weapon::WeaponType::Vulcan,
                            grind: 5,
                            special: Some(item::weapon::WeaponSpecial::Charge),
                            attrs: [Some(item::weapon::WeaponAttribute{attr: item::weapon::Attribute::Hit, value: 100}),
                                    Some(item::weapon::WeaponAttribute{attr: item::weapon::Attribute::Dark, value: 100}),
                                    None,],
                            tekked: true,
                            modifiers: Vec::new(),
                        }
                    ),
                    location: ItemLocation::Inventory {
                        character_id: character.id,
                    }
                }).await.unwrap();
            let item4 = entity_gateway.create_item(
                NewItemEntity {
                    item: ItemDetail::Weapon(
                        item::weapon::Weapon {
                            weapon: item::weapon::WeaponType::DarkFlow,
                            grind: 5,
                            special: Some(item::weapon::WeaponSpecial::Charge),
                            attrs: [Some(item::weapon::WeaponAttribute{attr: item::weapon::Attribute::Hit, value: 100}),
                                    Some(item::weapon::WeaponAttribute{attr: item::weapon::Attribute::Dark, value: 100}),
                                    Some(item::weapon::WeaponAttribute{attr: item::weapon::Attribute::Native, value: 100}),],
                            tekked: true,
                            modifiers: Vec::new(),
                        }
                    ),
                    location: ItemLocation::Inventory {
                        character_id: character.id,
                    }
                }).await.unwrap();

            let item5_m = entity_gateway.create_item(
                item::NewItemEntity {
                    item: item::ItemDetail::Mag(item::mag::Mag::baby_mag(0)),
                    location: item::ItemLocation::Inventory {
                        character_id: character.id,
                    }
                }).await.unwrap();

            for _ in 0..10usize {
                let fed_tool = entity_gateway.create_item(
                    NewItemEntity {
                        item: ItemDetail::Tool (
                            item::tool::Tool {
                                tool: item::tool::ToolType::Monomate,
                            }
                        ),
                        location: item::ItemLocation::FedToMag {
                            mag: item5_m.id,
                        }
                    }).await.unwrap();
                entity_gateway.feed_mag(&item5_m.id, &fed_tool.id).await.unwrap();
            }
            entity_gateway.change_mag_owner(&item5_m.id, &character).await.unwrap();

            let item6 = entity_gateway.create_item(
                item::NewItemEntity {
                    item: ItemDetail::Tool (
                        item::tool::Tool {
                            tool: item::tool::ToolType::CellOfMag502,
                        }
                    ),
                    location: item::ItemLocation::Inventory {
                        character_id: character.id,
                    }
                }).await.unwrap();
            let cell = entity_gateway.create_item(
                item::NewItemEntity {
                    item: ItemDetail::Tool (
                        item::tool::Tool {
                            tool: item::tool::ToolType::CellOfMag502,
                        }
                    ),
                    location: item::ItemLocation::Consumed,
                }).await.unwrap();
            entity_gateway.use_mag_cell(&item5_m.id, &cell.id).await.unwrap();

            entity_gateway.create_item(
                NewItemEntity {
                    item: ItemDetail::Weapon(
                        item::weapon::Weapon {
                            weapon: item::weapon::WeaponType::Autogun,
                            grind: 5,
                            special: Some(item::weapon::WeaponSpecial::Hell),
                            attrs: [Some(item::weapon::WeaponAttribute{attr: item::weapon::Attribute::Hit, value: 70}),
                                    Some(item::weapon::WeaponAttribute{attr: item::weapon::Attribute::Dark, value: 80}),
                                    None,],
                            tekked: false,
                            modifiers: Vec::new(),
                        }
                    ),
                    location: ItemLocation::Bank {
                        character_id: character.id,
                        name: item::BankName("".to_string()),
                    }
                }).await.unwrap();
            let item7_a = entity_gateway.create_item(
                NewItemEntity {
                    item: ItemDetail::Armor(
                        item::armor::Armor {
                            armor: item::armor::ArmorType::Frame,
                            dfp: 0,
                            evp: 0,
                            slots: 4,
                            modifiers: Vec::new(),
                        }
                    ),
                    location: ItemLocation::Inventory {
                        character_id: character.id,
                    }
                }
            ).await.unwrap();
            let item8_s = entity_gateway.create_item(
                NewItemEntity {
                    item: ItemDetail::Shield(
                        item::shield::Shield {
                            shield: item::shield::ShieldType::Barrier,
                            dfp: 0,
                            evp: 0,
                        }
                    ),
                    location: ItemLocation::Inventory {
                        character_id: character.id,
                    }
                }
            ).await.unwrap();
            let item9_u0 = entity_gateway.create_item(
                NewItemEntity {
                    item: ItemDetail::Unit(
                        item::unit::Unit {
                            unit: item::unit::UnitType::PriestMind,
                            modifier: Some(item::unit::UnitModifier::Minus),
                            armor_slot: 0, // TODO: get rid of this
                        }
                    ),
                    location: ItemLocation::Inventory {
                        character_id: character.id,
                    }
                }
            ).await.unwrap();
            let item10_u1 = entity_gateway.create_item(
                NewItemEntity {
                    item: ItemDetail::Unit(
                        item::unit::Unit {
                            unit: item::unit::UnitType::PriestMind,
                            modifier: Some(item::unit::UnitModifier::Minus),
                            armor_slot: 1,
                        }
                    ),
                    location: ItemLocation::Inventory {
                        character_id: character.id,
                    }
                }
            ).await.unwrap();
            let item11_u2 = entity_gateway.create_item(
                NewItemEntity {
                    item: ItemDetail::Unit(
                        item::unit::Unit {
                            unit: item::unit::UnitType::PriestMind,
                            modifier: Some(item::unit::UnitModifier::Minus),
                            armor_slot: 2,
                        }
                    ),
                    location: ItemLocation::Inventory {
                        character_id: character.id,
                    }
                }
            ).await.unwrap();
            let item12_u3 = entity_gateway.create_item(
                NewItemEntity {
                    item: ItemDetail::Unit(
                        item::unit::Unit {
                            unit: item::unit::UnitType::PriestMind,
                            modifier: Some(item::unit::UnitModifier::Minus),
                            armor_slot: 3,
                        }
                    ),
                    location: ItemLocation::Inventory {
                        character_id: character.id,
                    }
                }
            ).await.unwrap();
            let item13 = entity_gateway.create_item(
                NewItemEntity {
                    item: ItemDetail::Mag(
                        item::mag::Mag::baby_mag(5)
                    ),
                    location: ItemLocation::Inventory {
                        character_id: character.id,
                    }
                }
            ).await.unwrap();

            let equipped = item::EquippedEntity {
                weapon: Some(item2_w.id),
                armor: Some(item7_a.id),
                shield: Some(item8_s.id),
                unit: [Some(item9_u0.id), Some(item10_u1.id), Some(item11_u2.id), Some(item12_u3.id)],
                mag: Some(item5_m.id),
            };
            entity_gateway.set_character_equips(&character.id, &equipped).await.unwrap();

            let inventory = item::InventoryEntity::new(vec![item0, item1, item2_w, item3, item4, item5_m, item6, item7_a, item8_s, item9_u0, item10_u1, item11_u2, item12_u3, item13]);
            entity_gateway.set_character_inventory(&character.id, &inventory).await.unwrap();
        }

        info!("[patch] starting server");
        let patch_config = load_config();
        let patch_motd = load_motd();
        let (patch_file_tree, patch_file_lookup) = generate_patch_tree(patch_config.path.as_str());
        let patch_state = Box::new(PatchServerState::new(patch_file_tree, patch_file_lookup, patch_motd));
        let patch_loop = patch_mainloop(*patch_state, patch_config.port);

        let thread_entity_gateway = entity_gateway.clone();
        info!("[auth] starting server");
        let login_state = Box::new(LoginServerState::new(thread_entity_gateway, "127.0.0.1".parse().unwrap()));
        let login_loop = login_mainloop(*login_state, elseware::login::login::LOGIN_PORT);

        let thread_entity_gateway = entity_gateway.clone();
        info!("[character] starting server");
        let char_state = Box::new(CharacterServerState::new(thread_entity_gateway, AuthToken("".into())));
        let character_loop = character_mainloop(*char_state, elseware::login::character::CHARACTER_PORT, elseware::login::login::COMMUNICATION_PORT);

        let thread_entity_gateway = entity_gateway.clone();
        info!("[ship] starting server");
        let ship_state = Box::new(ShipServerStateBuilder::new()
            .name("US/Sona-Nyl".into())
            .ip(Ipv4Addr::new(127,0,0,1))
            .port(elseware::ship::ship::SHIP_PORT)
            .gateway(thread_entity_gateway)
            .build());
        let ship_loop = ship_mainloop(*ship_state, elseware::ship::ship::SHIP_PORT, std::net::Ipv4Addr::new(127, 0, 0, 1), elseware::login::login::COMMUNICATION_PORT);

        let thread_entity_gateway = entity_gateway.clone();
        let ship_state = Box::new(ShipServerStateBuilder::new()
            .name("EU/Dylath-Leen".into())
            .ip(Ipv4Addr::new(127,0,0,1))
            .port(elseware::ship::ship::SHIP_PORT+2000)
            .gateway(thread_entity_gateway)
            .build());
        let ship_loop2 = ship_mainloop(*ship_state, elseware::ship::ship::SHIP_PORT+2000, std::net::Ipv4Addr::new(127, 0, 0, 1), elseware::login::login::COMMUNICATION_PORT);

        let thread_entity_gateway = entity_gateway.clone();
        let ship_state = Box::new(ShipServerStateBuilder::new()
            .name("JP/Thalarion".into())
            .ip(Ipv4Addr::new(127,0,0,1))
            .port(elseware::ship::ship::SHIP_PORT+3000)
            .gateway(thread_entity_gateway)
            .build());
        let ship_loop3 = ship_mainloop(*ship_state, elseware::ship::ship::SHIP_PORT+3000, std::net::Ipv4Addr::new(127, 0, 0, 1), elseware::login::login::COMMUNICATION_PORT);

        futures::future::join_all(vec![patch_loop, login_loop, character_loop, ship_loop, ship_loop2, ship_loop3]).await;
    });
}