elseware/src/bin/main.rs

use std::net::Ipv4Addr;
use std::time::SystemTime;
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::ShipServerState;
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::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,
                created_at: chrono::Utc::now(),
                flags: 0,
            };
            let fake_user = entity_gateway.create_user(fake_user).await.unwrap();
            entity_gateway.create_user_settings(NewUserSettingsEntity::new(fake_user.id)).await;
            let mut character = NewCharacterEntity::new(fake_user.id);
            character.name = format!("Test Char {}", i*2);
            entity_gateway.create_character(character).await;
            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();

            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,
                        slot: 0,
                        equipped: false,
                    }
                }).await;
            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,
                        slot: 1,
                        equipped: false,
                    }
                }).await;
            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,
                        slot: 2,
                        equipped: true,
                    }
                }).await;
            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,
                        slot: 3,
                        equipped: true,
                    }
                }).await;
            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,
                        slot: 4,
                        equipped: true,
                    }
                }).await;
            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;
            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,
                        slot: 5,
                        equipped: true,
                    }
                }
            ).await;
            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,
                        slot: 6,
                        equipped: true,
                    }
                }
            ).await;
            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,
                        }
                    ),
                    location: ItemLocation::Inventory {
                        character_id: character.id,
                        slot: 7,
                        equipped: true,
                    }
                }
            ).await;
            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,
                        slot: 8,
                        equipped: true,
                    }
                }
            ).await;
            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,
                        slot: 9,
                        equipped: true,
                    }
                }
            ).await;
            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,
                        slot: 10,
                        equipped: true,
                    }
                }
            ).await;
            entity_gateway.create_item(
                NewItemEntity {
                    item: ItemDetail::Mag(
                        item::mag::Mag::baby_mag(5)
                    ),
                    location: ItemLocation::Inventory {
                        character_id: character.id,
                        slot: 11,
                        equipped: true,
                    }
                }
            ).await;
        }

        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 = 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 = 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 = CharacterServerState::new(thread_entity_gateway);
        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 = ShipServerStateBuilder::new()
            .name("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 = ShipServerStateBuilder::new()
            .name("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 = ShipServerStateBuilder::new()
            .name("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;
    });
}