forked from StrafesNET/asset-tool
get inventory function
This commit is contained in:
parent
5ceb118ba1
commit
312c0e7cd9
78
src/main.rs
78
src/main.rs
@ -22,6 +22,7 @@ struct Cli{
|
|||||||
enum Commands{
|
enum Commands{
|
||||||
DownloadHistory(DownloadHistorySubcommand),
|
DownloadHistory(DownloadHistorySubcommand),
|
||||||
Download(DownloadSubcommand),
|
Download(DownloadSubcommand),
|
||||||
|
DownloadGroupInventoryJson(DownloadGroupInventoryJsonSubcommand),
|
||||||
Create(CreateSubcommand),
|
Create(CreateSubcommand),
|
||||||
Upload(UploadSubcommand),
|
Upload(UploadSubcommand),
|
||||||
Compile(CompileSubcommand),
|
Compile(CompileSubcommand),
|
||||||
@ -59,6 +60,17 @@ struct DownloadSubcommand{
|
|||||||
asset_ids:Vec<AssetID>,
|
asset_ids:Vec<AssetID>,
|
||||||
}
|
}
|
||||||
#[derive(Args)]
|
#[derive(Args)]
|
||||||
|
struct DownloadGroupInventoryJsonSubcommand{
|
||||||
|
#[arg(long)]
|
||||||
|
cookie_type:CookieType,
|
||||||
|
#[arg(long)]
|
||||||
|
cookie:String,
|
||||||
|
#[arg(long)]
|
||||||
|
output_folder:Option<PathBuf>,
|
||||||
|
#[arg(long)]
|
||||||
|
group:u64,
|
||||||
|
}
|
||||||
|
#[derive(Args)]
|
||||||
struct CreateSubcommand{
|
struct CreateSubcommand{
|
||||||
#[arg(long)]
|
#[arg(long)]
|
||||||
cookie_type:CookieType,
|
cookie_type:CookieType,
|
||||||
@ -187,6 +199,26 @@ struct AssetVersion{
|
|||||||
isPublished:bool,
|
isPublished:bool,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[derive(serde::Deserialize)]
|
||||||
|
#[allow(nonstandard_style,dead_code)]
|
||||||
|
struct InventoryPage{
|
||||||
|
totalResults:u64,//up to 50
|
||||||
|
filteredKeyword:Option<String>,//""
|
||||||
|
searchDebugInfo:Option<String>,//null
|
||||||
|
spellCheckerResult:Option<String>,//null
|
||||||
|
queryFacets:Option<String>,//null
|
||||||
|
imageSearchStatus:Option<String>,//null
|
||||||
|
previousPageCursor:Option<String>,
|
||||||
|
nextPageCursor:Option<String>,
|
||||||
|
data:Vec<InventoryItem>,
|
||||||
|
}
|
||||||
|
#[derive(serde::Deserialize,serde::Serialize)]
|
||||||
|
#[allow(nonstandard_style,dead_code)]
|
||||||
|
struct InventoryItem{
|
||||||
|
id:u64,
|
||||||
|
name:String,
|
||||||
|
}
|
||||||
|
|
||||||
#[tokio::main]
|
#[tokio::main]
|
||||||
async fn main()->AResult<()>{
|
async fn main()->AResult<()>{
|
||||||
let cli=Cli::parse();
|
let cli=Cli::parse();
|
||||||
@ -210,6 +242,11 @@ async fn main()->AResult<()>{
|
|||||||
}).collect()
|
}).collect()
|
||||||
).await
|
).await
|
||||||
},
|
},
|
||||||
|
Commands::DownloadGroupInventoryJson(subcommand)=>download_group_inventory_json(
|
||||||
|
Cookie::from_type(subcommand.cookie_type,subcommand.cookie).await?.0,
|
||||||
|
subcommand.group,
|
||||||
|
subcommand.output_folder.unwrap_or_else(||std::env::current_dir().unwrap()),
|
||||||
|
).await,
|
||||||
Commands::Create(subcommand)=>create(
|
Commands::Create(subcommand)=>create(
|
||||||
Cookie::from_type(subcommand.cookie_type,subcommand.cookie).await?.0,
|
Cookie::from_type(subcommand.cookie_type,subcommand.cookie).await?.0,
|
||||||
subcommand.group,
|
subcommand.group,
|
||||||
@ -426,6 +463,47 @@ async fn download_list(cookie:String,asset_id_file_map:AssetIDFileMap)->AResult<
|
|||||||
}).await;
|
}).await;
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
async fn download_inventory_page(client:&reqwest::Client,cookie:&str,group:u64,cursor:Option<String>)->AResult<InventoryPage>{
|
||||||
|
let mut url=reqwest::Url::parse(format!("https://apis.roblox.com/toolbox-service/v1/creations/group/{}/10?limit=50",group).as_str())?;
|
||||||
|
//url borrow scope
|
||||||
|
{
|
||||||
|
let mut query=url.query_pairs_mut();//borrow here
|
||||||
|
match cursor.as_deref(){
|
||||||
|
Some(next_page)=>{query.append_pair("cursor",next_page);}
|
||||||
|
None=>(),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
println!("page url={}",url);
|
||||||
|
let resp=client.get(url)
|
||||||
|
.header("Cookie",cookie)
|
||||||
|
.send().await?;
|
||||||
|
Ok(resp.json::<InventoryPage>().await?)
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn get_inventory_pages(client:&reqwest::Client,cookie:&str,group:u64)->AResult<Vec<InventoryItem>>{
|
||||||
|
let mut cursor:Option<String>=None;
|
||||||
|
let mut asset_list=Vec::new();
|
||||||
|
loop{
|
||||||
|
let mut page=download_inventory_page(client,cookie,group,cursor).await?;
|
||||||
|
asset_list.append(&mut page.data);
|
||||||
|
if page.nextPageCursor.is_none(){
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
cursor=page.nextPageCursor;
|
||||||
|
}
|
||||||
|
Ok(asset_list)
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn download_group_inventory_json(cookie:String,group:u64,output_folder:PathBuf)->AResult<()>{
|
||||||
|
let client=reqwest::Client::new();
|
||||||
|
let item_list=get_inventory_pages(&client,cookie.as_str(),group).await?;
|
||||||
|
|
||||||
|
let mut path=output_folder.clone();
|
||||||
|
path.set_file_name("versions.json");
|
||||||
|
tokio::fs::write(path,serde_json::to_string(&item_list)?).await?;
|
||||||
|
|
||||||
|
Ok(())
|
||||||
|
}
|
||||||
|
|
||||||
async fn download_page(client:&reqwest::Client,cookie:&str,asset_id:AssetID,cursor:Option<String>)->AResult<VersionPage>{
|
async fn download_page(client:&reqwest::Client,cookie:&str,asset_id:AssetID,cursor:Option<String>)->AResult<VersionPage>{
|
||||||
let mut url=reqwest::Url::parse(format!("https://develop.roblox.com/v1/assets/{}/saved-versions",asset_id).as_str())?;
|
let mut url=reqwest::Url::parse(format!("https://develop.roblox.com/v1/assets/{}/saved-versions",asset_id).as_str())?;
|
||||||
|
Loading…
Reference in New Issue
Block a user