elseware/src/entity/gateway/postgres/postgres.rs

use std::convert::{From, TryFrom, Into, TryInto};
use serde::{Serialize, Deserialize};
use futures::future::join_all;
use futures::TryStreamExt;
//use futures::StreamExt;
use async_std::stream::StreamExt;
//use futures::StreamExt;
use libpso::character::{settings, guildcard};
use libpso::util::vec_to_array;
use crate::entity::account::*;
use crate::entity::character::*;
use crate::entity::gateway::EntityGateway;
use crate::entity::item::*;
use super::models::*;

use sqlx::postgres::PgPoolOptions;
use sqlx::Row;
use sqlx::Execute;
use postgres::{Client, NoTls};

mod embedded {
    use refinery::embed_migrations;
    embed_migrations!("src/entity/gateway/postgres/migrations");
}


#[derive(Clone)]
pub struct PostgresGateway {
    pool: sqlx::Pool<sqlx::Postgres>,
}

impl PostgresGateway {
    pub fn new(host: &str, dbname: &str, username: &str, password: &str) -> PostgresGateway {
        // the postgres dep can be removed once refinery supports sqlx
        let mut conn = Client::connect(&format!("host='{}' dbname='{}' user='{}' password='{}'", host, dbname, username, password), NoTls).unwrap();
        embedded::migrations::runner().run(&mut conn).unwrap();
        
        let pool = async_std::task::block_on(async move {
            let pool = PgPoolOptions::new()
                .max_connections(5)
                .connect(&format!("postgresql://{}:{}@{}:5432/{}", username, password, host, dbname)).await.unwrap();

            pool
        });
        
        PostgresGateway {
            pool: pool,
        }
    }

    async fn apply_item_modifications(&self, item: ItemEntity) -> ItemEntity {
        let ItemEntity {id, item, location} = item;
        
        let item = match item {
            ItemDetail::Mag(mut mag) => {
                let q = r#"select mag, modifier, item.item -> 'Tool' as feed, item2.item -> 'Tool' as cell 
                           from mag_modifier 
                           left join item on item.id = cast (modifier ->> 'FeedMag' as integer)
                           left join item as item2 on item2.id = cast (modifier ->> 'MagCell' as integer)
                           where mag = $1 order by created_at"#;
                let mag_modifiers = sqlx::query_as::<_, PgMagModifierWithParameters>(q)
                    .bind(id.0 as i32)
                    .fetch(&self.pool);

                mag_modifiers.for_each(|modifier| {
                    let PgMagModifierWithParameters {modifier, feed, cell, ..} = modifier.unwrap();
                    let modifier: mag::MagModifier = modifier.0.into();
                    match modifier {
                        mag::MagModifier::FeedMag{..} => {
                            mag.feed(feed.unwrap().tool)
                        },
                        mag::MagModifier::BankMag => {
                            mag.bank()
                        },
                        mag::MagModifier::MagCell(_) => {
                            mag.apply_mag_cell(mag::MagCell::try_from(Into::<tool::Tool>::into(cell.unwrap().0).tool).unwrap())
                        },
                        mag::MagModifier::OwnerChange(class, section_id) => {
                            mag.change_owner(class, section_id)
                        },
                    }
                }).await;
                
                ItemDetail::Mag(mag)
            },
            item @ _ => item
        };

        ItemEntity {
            id: id,
            item: item,
            location: location
        }
    }
}

#[async_trait::async_trait]
impl EntityGateway for PostgresGateway {
    async fn create_user(&mut self, user: NewUserAccountEntity) -> Option<UserAccountEntity> {
        let new_user = sqlx::query_as::<_, PgUserAccount>("insert into user_accounts (name, password) values ($1, $2) returning *;")
            .bind(user.username)
            .bind(user.password)
            .fetch_one(&self.pool).await.unwrap();
        Some(new_user.into())
    }

    async fn get_user_by_id(&self, id: UserAccountId) -> Option<UserAccountEntity> {
        let user = sqlx::query_as::<_, PgUserAccount>("select * from user_accounts where id = $1")
            .bind(id.0)
            .fetch_one(&self.pool).await.unwrap();
        Some(user.into())
    }

    async fn get_user_by_name(&self, username: String) -> Option<UserAccountEntity> {
        let user = sqlx::query_as::<_, PgUserAccount>("select * from user_accounts where name = $1")
            .bind(username)
            .fetch_one(&self.pool).await.unwrap();
        Some(user.into())
    }

    async fn save_user(&mut self, user: &UserAccountEntity) {
        sqlx::query("UPDATE user_accounts set name=$1, password=$2, banned=$3, muted=$4, flags=$5 where id=$6")
            .bind(&user.username)
            .bind(&user.password)
            .bind(&user.banned_until)
            .bind(&user.muted_until)
            .bind(&user.flags)
            .bind(&user.id.0)
            .fetch_one(&self.pool).await.unwrap();
    }

    async fn create_user_settings(&mut self, settings: NewUserSettingsEntity) -> Option<UserSettingsEntity> {
        let new_settings = sqlx::query_as::<_, PgUserSettings>("insert into user_settings (user_account, blocked_users, key_config, joystick_config, option_flags, shortcuts, symbol_chats, team_name)
                                                                values ($1, $2, $3, $4, $5, $6, $7, $8) returning *;")
            .bind(settings.user_id.0)
            .bind(settings.settings.blocked_users.to_vec().into_iter().map(|i| i.to_le_bytes().to_vec()).flatten().collect::<Vec<u8>>())
            .bind(settings.settings.key_config.to_vec())
            .bind(settings.settings.joystick_config.to_vec())
            .bind(settings.settings.option_flags as i32)
            .bind(settings.settings.shortcuts.to_vec())
            .bind(settings.settings.symbol_chats.to_vec())
            .bind(settings.settings.team_name.to_vec().into_iter().map(|i| i.to_le_bytes().to_vec()).flatten().collect::<Vec<u8>>())
            .fetch_one(&self.pool).await.unwrap();
        Some(new_settings.into())
    }

    async fn get_user_settings_by_user(&self, user: &UserAccountEntity) -> Option<UserSettingsEntity> {
        let settings = sqlx::query_as::<_, PgUserSettings>("select * from user_settings where id = $1")
            .bind(user.id.0)
            .fetch_one(&self.pool).await.unwrap();
        Some(settings.into())
    }

    async fn save_user_settings(&mut self, settings: &UserSettingsEntity) {
        sqlx::query("update user_settings set blocked_users=$1, key_config=$2, joystick_config=$3, option_flags=$4, shortcuts=$5, symbol_chats=$6, team_name=$7 where id=$8")
            .bind(settings.settings.blocked_users.to_vec().into_iter().map(|i| i.to_le_bytes().to_vec()).flatten().collect::<Vec<u8>>())
            .bind(&settings.settings.key_config.to_vec())
            .bind(&settings.settings.joystick_config.to_vec())
            .bind(&settings.settings.option_flags)
            .bind(&settings.settings.shortcuts.to_vec())
            .bind(&settings.settings.symbol_chats.to_vec())
            .bind(settings.settings.team_name.to_vec().into_iter().map(|i| i.to_le_bytes().to_vec()).flatten().collect::<Vec<u8>>())
            .bind(&settings.id.0)
            .fetch_one(&self.pool).await.unwrap();
    }

    async fn create_character(&mut self, char: NewCharacterEntity) -> Option<CharacterEntity> {
        let q = r#"insert into player_character
                  (user_account, slot, name, exp, class, section_id, costume, skin, face, head, hair, hair_r, hair_g, hair_b, prop_x, prop_y, techs, 
                   config, infoboard, guildcard, power, mind, def, evade, luck, hp, tp, tech_menu, meseta, bank_meseta)
                  values
                  ($1, $2, $3, $4, $5, $6, $7, $8, $9, $10, $11, $12, $13, $14, $15, $16, $17, $18, $19, $20, $21, $22, $23, $24, $25, $26, $27, $28, $29, $30)
                  returning *;"#;
        let character = sqlx::query_as::<_, PgCharacter>(q)
        //sqlx::query(q)
            .bind(char.user_id.0)
            .bind(char.slot as i16)
            .bind(char.name)
            .bind(char.exp as i32)
            .bind(char.char_class.to_string())
            .bind(char.section_id.to_string())
            .bind(char.appearance.costume as i16)
            .bind(char.appearance.skin as i16)
            .bind(char.appearance.face as i16)
            .bind(char.appearance.head as i16)
            .bind(char.appearance.hair as i16)
            .bind(char.appearance.hair_r as i16)
            .bind(char.appearance.hair_g as i16)
            .bind(char.appearance.hair_b as i16)
            .bind(char.appearance.prop_x)
            .bind(char.appearance.prop_y)
            .bind(&char.techs.as_bytes().to_vec())
            .bind(&char.config.as_bytes().to_vec())
            .bind(String::from_utf16_lossy(&char.info_board.board).trim_matches(char::from(0)))
            .bind(char.guildcard.description)
            .bind(char.materials.power as i16)
            .bind(char.materials.mind as i16)
            .bind(char.materials.def as i16)
            .bind(char.materials.evade as i16)
            .bind(char.materials.luck as i16)
            .bind(char.materials.hp as i16)
            .bind(char.materials.tp as i16)
            .bind(char.tech_menu.tech_menu.to_vec())
            .bind(char.meseta as i32)
            .bind(char.bank_meseta as i32)
            .fetch_one(&self.pool).await.unwrap();
        Some(character.into())
    }

    async fn get_characters_by_user(&self, user: &UserAccountEntity) -> [Option<CharacterEntity>; 4] {
        let mut stream = sqlx::query_as::<_, PgCharacter>("select * from player_character where user_account = $1 and slot < 4 order by slot")
            .bind(user.id.0)
            .fetch(&self.pool);
        let mut result = [None; 4];
        while let Some(character) = stream.try_next().await.unwrap() {
            let index = character.slot as usize;
            result[index] = Some(character.into())
        }
        
        result
    }

    async fn save_character(&mut self, char: &CharacterEntity) {
        let q = r#"update player_character set
                   user_account=$1, slot=$2, name=$3, exp=$4, class=$5, section_id=$6, costume=$7, skin=$8, face=$9, head=$10, hair=$11, hair_r=$12,
                   hair_g=$13, hair_b=$14, prop_x=$15, prop_y=$16, techs=$17, config=$18, infoboard=$19, guildcard=$20, power=$21, mind=$22, def=$23,
                   evade=$24, luck=$25, hp=$26, tp=$27, tech_menu=$28, meseta=$29, bank_meseta=$30
                   where id=$31;"#;
        sqlx::query(q)
            .bind(char.user_id.0)
            .bind(char.slot as i16)
            .bind(&char.name)
            .bind(char.exp as i32)
            .bind(char.char_class.to_string())
            .bind(char.section_id.to_string())
            .bind(char.appearance.costume as i16)
            .bind(char.appearance.skin as i16)
            .bind(char.appearance.face as i16)
            .bind(char.appearance.head as i16)
            .bind(char.appearance.hair as i16)
            .bind(char.appearance.hair_r as i16)
            .bind(char.appearance.hair_g as i16)
            .bind(char.appearance.hair_b as i16)
            .bind(char.appearance.prop_x)
            .bind(char.appearance.prop_y)
            .bind(&char.techs.as_bytes().to_vec())
            .bind(&char.config.as_bytes().to_vec())
            .bind(String::from_utf16_lossy(&char.info_board.board).trim_matches(char::from(0)))
            .bind(&char.guildcard.description)
            .bind(char.materials.power as i16)
            .bind(char.materials.mind as i16)
            .bind(char.materials.def as i16)
            .bind(char.materials.evade as i16)
            .bind(char.materials.luck as i16)
            .bind(char.materials.hp as i16)
            .bind(char.materials.tp as i16)
            .bind(char.tech_menu.tech_menu.to_vec())
            .bind(char.meseta as i32)
            .bind(char.bank_meseta as i32)
            .bind(char.id.0 as i32)
            .execute(&self.pool).await.unwrap();
    }

    async fn get_guild_card_data_by_user(&self, user: &UserAccountEntity) -> GuildCardDataEntity {
        GuildCardDataEntity {
            id: GuildCardDataId(0),
            user_id: user.id,
            guildcard: guildcard::GuildCardData::default(),
        }
    }

    async fn create_item(&mut self, item: NewItemEntity) -> Option<ItemEntity> {
        let new_item = sqlx::query_as::<_, PgItem>("insert into item (item) values ($1) returning *;")
            .bind(sqlx::types::Json(PgItemDetail::from(item.item)))
            .fetch_one(&self.pool).await.unwrap();
        let location = sqlx::query_as::<_, PgItemLocation>("insert into item_location (item, location) values ($1, $2) returning *;")
            .bind(new_item.id)
            .bind(sqlx::types::Json(PgItemLocationDetail::from(item.location)))
            .fetch_one(&self.pool).await.unwrap();
        Some(ItemEntity {
            id: ItemEntityId(new_item.id as u32),
            item: new_item.item.0.into(),
            location: location.location.0.into(),
        })
    }

    async fn change_item_location(&mut self, item_id: &ItemEntityId, item_location: ItemLocation) {
        sqlx::query("insert into item_location (item, location) values ($1, $2);")
            .bind(item_id.0)
            .bind(sqlx::types::Json(PgItemLocationDetail::from(item_location)))
            .execute(&self.pool).await.unwrap();
    }

    async fn feed_mag(&mut self, mag_item_id: &ItemEntityId, tool_item_id: &ItemEntityId) {
        sqlx::query("insert into mag_modifier (mag, modifier) values ($1, $2);")
            .bind(mag_item_id.0)
            .bind(sqlx::types::Json(PgMagModifierDetail::from(mag::MagModifier::FeedMag{food: *tool_item_id})))
            .execute(&self.pool).await.unwrap();
    }

    async fn change_mag_owner(&mut self, mag_item_id: &ItemEntityId, character: &CharacterEntity) {
        sqlx::query("insert into mag_modifier (mag, modifier) values ($1, $2);")
            .bind(mag_item_id.0)
            .bind(sqlx::types::Json(PgMagModifierDetail::from(mag::MagModifier::OwnerChange(character.char_class, character.section_id))))
            .execute(&self.pool).await.unwrap();
    }

    async fn use_mag_cell(&mut self, mag_item_id: &ItemEntityId, mag_cell_id: &ItemEntityId) {
        // consume cell?
        sqlx::query("insert into mag_modifier (mag, modifier) values ($1, $2);")
            .bind(mag_item_id.0)
            .bind(sqlx::types::Json(PgMagModifierDetail::from(mag::MagModifier::MagCell(*mag_cell_id))))
            .execute(&self.pool).await.unwrap();
    }

    async fn get_items_by_character(&self, char: &CharacterEntity) -> Vec<ItemEntity> {
        let q = r#"select * from (
                      select distinct on (item_location.item) item.id, item_location.location, item.item
                      from item_location join item on item.id = item_location.item
                      order by item_location.item, item_location.created_at desc
                   ) as i
                   where cast (location -> 'Inventory' ->> 'character_id' as integer) = 2
                      or cast (location -> 'Bank' ->> 'character_id' as integer) = 2"#;
        let items = sqlx::query_as::<_, PgItemWithLocation>(q)
            .bind(char.id.0)
            .fetch(&self.pool);
        join_all(items
            .filter_map(|item: Result<PgItemWithLocation, _>| {
                let item = item.ok()?;
                Some(ItemEntity {
                    id: ItemEntityId(item.id as u32),
                    item: item.item.0.into(),
                    location: item.location.0.into()
                })
            })
            .map(|item: ItemEntity| {
                self.apply_item_modifications(item)
            })
            .collect::<Vec<_>>()
            .await).await
    }
}