bonkmaykr/ojrc
1
0
Fork 0
mirror of https://git.ngram.ca/OpenJam/rc-servers synced 2026-08-23 23:08:52 +00:00
Code Activity

Add minimal robot factory functionality and arc archive support for #6

Browse Source
This commit is contained in:
NG (Graham)
2025-05-17 23:06:07 -04:00
parent 3f0c30f6c3
commit 3f1438c068
34 changed files with 869 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

122
rc_factory/src/arc/adapter.rs Normal file
View File

@@ -0,0 +1,122 @@
use sea_orm::{ColumnTrait, EntityTrait, PaginatorTrait, QueryFilter, QueryOrder};
pub struct ArcAdapter {
orm: sea_orm::DatabaseConnection,
ignore_expiry: bool,
}
impl ArcAdapter {
pub async fn init(uri: &str, show_expired: bool) -> Result<Self, sea_orm::DbErr>{
log::debug!("Connecting to Archive of RoboCraft (ARC) vehicle factory database URI: {}", uri);
let db = sea_orm::Database::connect(uri).await?;
Ok(Self {
orm: db,
ignore_expiry: show_expired
})
}
fn default_query(&self) -> sea_orm::Select<super::entities::robot_metadata::Entity> {
super::entities::robot_metadata::Entity::find()
.order_by_desc(super::entities::robot_metadata::Column::RentCount)
}
}
#[async_trait::async_trait]
impl crate::VehicleFactoryAdapter for ArcAdapter {
async fn vehicle(&self, id: u32) -> Result<Option<crate::VehicleInfo>, Box<dyn std::error::Error>> {
log::debug!("Get vehicle id {}", id);
let cubes = super::entities::robot_cubes::Entity::find_by_id(id).one(&self.orm).await?;
if let Some(cubes) = cubes {
use base64::Engine;
Ok(Some(crate::VehicleInfo {
id: id as _,
cube_data: base64::prelude::BASE64_STANDARD.decode(cubes.cube_data.as_bytes()).unwrap_or_default(),
colour_data: base64::prelude::BASE64_STANDARD.decode(cubes.colour_data.as_bytes()).unwrap_or_default(),
}))
} else {
Ok(None)
}
}
async fn list(&self, query: libfj::robocraft::ListQuery) -> Result<Vec<crate::VehicleQueryInfo>, Box<dyn std::error::Error>> {
log::debug!("Search vehicles with query {:?}", query);
let query_params = if query.default_page {
log::debug!("Default vehicle list query");
self.default_query()
} else {
let mut query_builder = super::entities::robot_metadata::Entity::find();
match query.order {
libfj::robocraft::FactoryOrderType::Suggested => { query_builder = query_builder.order_by_desc(super::entities::robot_metadata::Column::RentCount); },
libfj::robocraft::FactoryOrderType::CombatRating => { query_builder = query_builder.order_by_desc(super::entities::robot_metadata::Column::CombatRating); },
libfj::robocraft::FactoryOrderType::CosmeticRating => { query_builder = query_builder.order_by_desc(super::entities::robot_metadata::Column::CosmeticRating); },
libfj::robocraft::FactoryOrderType::Added => { query_builder = query_builder.order_by_asc(super::entities::robot_metadata::Column::AddedDate); },
libfj::robocraft::FactoryOrderType::CPU => { query_builder = query_builder.order_by_desc(super::entities::robot_metadata::Column::Cpu); },
libfj::robocraft::FactoryOrderType::MostBought => { query_builder = query_builder.order_by_desc(super::entities::robot_metadata::Column::BuyCount); },
}
if !query.text_filter.is_empty() {
if query.player_filter {
query_builder = query_builder.filter(
sea_orm::sea_query::Condition::any()
.add(super::entities::robot_metadata::Column::AddedBy.like(query.text_filter.clone()))
.add(super::entities::robot_metadata::Column::AddedByDisplayName.like(query.text_filter))
);
} else {
query_builder = query_builder.filter(
sea_orm::sea_query::Condition::any()
.add(super::entities::robot_metadata::Column::AddedBy.like(query.text_filter.clone()))
.add(super::entities::robot_metadata::Column::AddedByDisplayName.like(query.text_filter.clone()))
.add(super::entities::robot_metadata::Column::Name.like(query.text_filter.clone()))
.add(super::entities::robot_metadata::Column::Description.like(query.text_filter.clone()))
);
}
}
// movement filters not supported
// weapon filters not supported
if query.minimum_cpu > 0 {
query_builder = query_builder.filter(super::entities::robot_metadata::Column::Cpu.gte(query.minimum_cpu as u32));
}
if query.maximum_cpu < usize::MAX {
query_builder = query_builder.filter(super::entities::robot_metadata::Column::Cpu.lte(query.maximum_cpu as u32));
}
if query.buyable {
query_builder = query_builder.filter(super::entities::robot_metadata::Column::Buyable.ne(0));
}
query_builder
};
// FIXME add support for query.prepend_featured_bot
let metadata_pages = query_params.paginate(&self.orm, query.page_size as u64);
let metadatas = metadata_pages.fetch_page(query.page as u64).await?;
let mut infos = Vec::with_capacity(metadatas.len());
for meta in metadatas {
//let cube_amounts = super::entities::robot_cubes::Entity::find_by_id(meta.id).one(&self.orm).await?.map(|x| x.cube_amounts).unwrap_or_else(|| "".to_owned());
infos.push(
crate::VehicleQueryInfo {
id: meta.id as _,
name: meta.name,
description: meta.description,
thumbnail: meta.thumbnail,
added_by: meta.added_by,
added_by_display_name: meta.added_by_display_name,
added_date: crate::traits::parse_rc_date(&meta.added_date).unwrap_or_default(),
expiry_date: if self.ignore_expiry { chrono::Utc::now() + chrono::Duration::weeks(2) } else { crate::traits::parse_rc_date(&meta.expiry_date).unwrap_or_default() },
cpu: meta.cpu as _,
total_robot_ranking: meta.total_robot_ranking as _,
rent_count: meta.rent_count as _,
buy_count: meta.buy_count as _,
buyable: meta.buyable != 0,
removed_date: Default::default(),
ban_date: Default::default(),
featured: meta.featured != 0,
banner_message: Default::default(),
combat_rating: meta.combat_rating,
cosmetic_rating: meta.cosmetic_rating,
cube_amounts: Default::default(),
}
);
}
log::debug!("Search vehicles returned {} results", infos.len());
Ok(infos)
}
}