85 lines
2.5 KiB
Rust
85 lines
2.5 KiB
Rust
use crate::rbx_util::{get_mapinfo,read_dom,MapInfo,ReadDomError,GetMapInfoError,ParseGameIDError};
|
|
|
|
#[allow(dead_code)]
|
|
#[derive(Debug)]
|
|
pub enum Error{
|
|
ModelVersionsPage(rbx_asset::cookie::PageError),
|
|
EmptyVersionsPage,
|
|
CreatorTypeMustBeUser(rbx_asset::cookie::CreatorType),
|
|
ModelDetails(rbx_asset::cookie::GetError),
|
|
ModelInfoDownload(rbx_asset::cookie::GetAssetV2Error),
|
|
ModelFileDownload(rbx_asset::cookie::GetError),
|
|
NoLocations,
|
|
ModelFileDecode(ReadDomError),
|
|
GetMapInfo(GetMapInfoError),
|
|
ParseGameID(ParseGameIDError),
|
|
}
|
|
impl std::fmt::Display for Error{
|
|
fn fmt(&self,f:&mut std::fmt::Formatter<'_>)->std::fmt::Result{
|
|
write!(f,"{self:?}")
|
|
}
|
|
}
|
|
impl std::error::Error for Error{}
|
|
|
|
#[allow(nonstandard_style)]
|
|
pub struct CreateRequest{
|
|
pub ModelID:u64,
|
|
}
|
|
#[allow(nonstandard_style)]
|
|
pub struct CreateResult{
|
|
pub AssetOwner:i64,
|
|
pub DisplayName:String,
|
|
pub Creator:String,
|
|
pub GameID:i32,
|
|
pub AssetVersion:u64,
|
|
}
|
|
impl crate::message_handler::MessageHandler{
|
|
pub async fn create_inner(&self,create_info:CreateRequest)->Result<CreateResult,Error>{
|
|
// discover asset creator
|
|
let creator_fut=async{
|
|
self.cookie_context.get_asset_details(
|
|
rbx_asset::cookie::GetAssetDetailsRequest{asset_id:create_info.ModelID}
|
|
).await.map_err(Error::ModelDetails)
|
|
};
|
|
|
|
// download the map model
|
|
let asset_fut=async{
|
|
let asset_info=self.cookie_context.get_asset_v2(rbx_asset::cookie::GetAssetRequest{
|
|
asset_id:create_info.ModelID,
|
|
version:None,
|
|
}).await.map_err(Error::ModelInfoDownload)?;
|
|
|
|
let location=asset_info.info.locations.first().ok_or(Error::NoLocations)?;
|
|
let data=self.cookie_context.get_asset_v2_download(location).await.map_err(Error::ModelFileDownload)?;
|
|
|
|
Ok((asset_info.version,data))
|
|
};
|
|
|
|
let (details,(asset_version,model_data))=tokio::try_join!(creator_fut,asset_fut)?;
|
|
|
|
if details.Creator.CreatorType!=rbx_asset::cookie::CreatorType::User{
|
|
return Err(Error::CreatorTypeMustBeUser(details.Creator.CreatorType));
|
|
}
|
|
|
|
// decode dom (slow!)
|
|
let dom=read_dom(&mut std::io::Cursor::new(model_data)).map_err(Error::ModelFileDecode)?;
|
|
|
|
// parse create fields out of asset
|
|
let MapInfo{
|
|
display_name,
|
|
creator,
|
|
game_id,
|
|
}=get_mapinfo(&dom).map_err(Error::GetMapInfo)?;
|
|
|
|
let game_id=game_id.map_err(Error::ParseGameID)?;
|
|
|
|
Ok(CreateResult{
|
|
AssetOwner:details.Creator.Id as i64,
|
|
DisplayName:display_name.unwrap_or_default().to_owned(),
|
|
Creator:creator.unwrap_or_default().to_owned(),
|
|
GameID:game_id as i32,
|
|
AssetVersion:asset_version,
|
|
})
|
|
}
|
|
}
|