create assets

This commit is contained in:
Quaternions 2024-04-24 21:31:53 -07:00
parent c084219534
commit 5ceb118ba1

View File

@ -22,6 +22,7 @@ struct Cli{
enum Commands{ enum Commands{
DownloadHistory(DownloadHistorySubcommand), DownloadHistory(DownloadHistorySubcommand),
Download(DownloadSubcommand), Download(DownloadSubcommand),
Create(CreateSubcommand),
Upload(UploadSubcommand), Upload(UploadSubcommand),
Compile(CompileSubcommand), Compile(CompileSubcommand),
Decompile(DecompileSubcommand), Decompile(DecompileSubcommand),
@ -58,6 +59,19 @@ struct DownloadSubcommand{
asset_ids:Vec<AssetID>, asset_ids:Vec<AssetID>,
} }
#[derive(Args)] #[derive(Args)]
struct CreateSubcommand{
#[arg(long)]
cookie_type:CookieType,
#[arg(long)]
cookie:String,
#[arg(long)]
model_name:String,
#[arg(long)]
input_file:PathBuf,
#[arg(long)]
group:Option<u64>,
}
#[derive(Args)]
struct UploadSubcommand{ struct UploadSubcommand{
#[arg(long)] #[arg(long)]
asset_id:AssetID, asset_id:AssetID,
@ -196,6 +210,12 @@ async fn main()->AResult<()>{
}).collect() }).collect()
).await ).await
}, },
Commands::Create(subcommand)=>create(
Cookie::from_type(subcommand.cookie_type,subcommand.cookie).await?.0,
subcommand.group,
subcommand.input_file,
subcommand.model_name,
).await,
Commands::Upload(subcommand)=>upload_list( Commands::Upload(subcommand)=>upload_list(
Cookie::from_type(subcommand.cookie_type,subcommand.cookie).await?.0, Cookie::from_type(subcommand.cookie_type,subcommand.cookie).await?.0,
subcommand.group, subcommand.group,
@ -264,6 +284,51 @@ fn maybe_gzip_decode<R:Read>(input:R)->AResult<ReaderType<R>>{
} }
} }
async fn create(cookie:String,group:Option<u64>,file:PathBuf,model_name:String)->AResult<()>{
let client=reqwest::Client::new();
let client=&client;
let cookie=cookie.as_str();
let group=&group;
let mut url=reqwest::Url::parse("https://data.roblox.com/Data/Upload.ashx?json=1&type=Model&genreTypeId=1&description&ispublic=False&allowComments=False")?;
//url borrow scope
{
let mut query=url.query_pairs_mut();//borrow here
//archaic roblox api uses 0 for new asset
query.append_pair("assetid","0");
query.append_pair("name",model_name.as_str());
match group{
Some(group_id)=>{query.append_pair("groupId",group_id.to_string().as_str());},
None=>(),
}
}
let body=tokio::fs::read(file).await?;
let mut resp=client.post(url.clone())
.header("Cookie",cookie)
.body(body.clone())
.send().await?;
//This is called a CSRF challenge apparently
if resp.status()==reqwest::StatusCode::FORBIDDEN{
if let Some(csrf_token)=resp.headers().get("X-CSRF-Token"){
resp=client.post(url)
.header("X-CSRF-Token",csrf_token)
.header("Cookie",cookie)
.body(body)
.send().await?;
}else{
Err(anyhow::Error::msg("Roblox returned 403 with no CSRF"))?;
}
}
let body=match resp.status(){
reqwest::StatusCode::OK=>Ok(resp.bytes().await?),
other=>Err(anyhow::Error::msg(other)),
};
println!("response.body={:?}",body?);
Ok(())
}
async fn upload_list(cookie:String,group:Option<u64>,asset_id_file_map:AssetIDFileMap)->AResult<()>{ async fn upload_list(cookie:String,group:Option<u64>,asset_id_file_map:AssetIDFileMap)->AResult<()>{
let client=reqwest::Client::new(); let client=reqwest::Client::new();
//this is calling map on the vec because the closure produces an iterator of futures //this is calling map on the vec because the closure produces an iterator of futures