StrafesNET/asset-tool
3
Fork 1
Code Issues Pull Requests Packages Projects Releases 4 Wiki Activity

winner

Browse Source
This commit is contained in:
Quaternions 2024-01-11 02:02:49 -08:00
parent 3ed2b71e07
commit 52b43cef2b
1 changed files with 21 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
src/main.rs
View File

@ -1,4 +1,4 @@
use std::io::{Read,Seek};
use std::io::Read;
use clap::{Args,Parser,Subcommand};
use anyhow::Result as AResult;
use futures::StreamExt;
@ -49,6 +49,7 @@ enum Commands{
Upload,
Compile,
Decompile,
DecompileHistoryIntoGit,
}
#[derive(Args)]
@ -112,6 +113,14 @@ async fn main()->AResult<()>{
write_models:!cli.no_models.unwrap_or(false),
write_scripts:!cli.no_scripts.unwrap_or(false),
}),
Commands::DecompileHistoryIntoGit=>decompile_history_into_git(DecompileHistoryConfig{
cookie:cookie.unwrap(),
asset_id:cli.asset_id.unwrap(),
output_folder:cli.output.unwrap(),
write_template:!cli.no_template.unwrap_or(false),
write_models:!cli.no_models.unwrap_or(false),
write_scripts:!cli.no_scripts.unwrap_or(false),
}).await,
}
}
@ -666,7 +675,7 @@ struct WriteCommitConfig{
write_scripts:bool,
}
fn write_commit(config:WriteCommitConfig,b:AResult<(AssetVersion,DecompiledContext)>,repo:&git2::Repository,last_commit:Option<&git2::Commit>)->AResult<git2::Oid>{
fn write_commit(config:WriteCommitConfig,b:AResult<(AssetVersion,DecompiledContext)>,repo:&git2::Repository)->AResult<()>{
let (asset_version,context)=b?;
//clean output dir
@ -676,12 +685,12 @@ fn write_commit(config:WriteCommitConfig,b:AResult<(AssetVersion,DecompiledConte
std::fs::remove_dir_all(src)?;
let mut template=config.output_folder.clone();
template.push("template.rbxlx");
std::fs::remove_file(template);
std::fs::remove_file(template)?;
}
//write files
write_files(WriteConfig{
output_folder:config.output_folder,
output_folder:config.output_folder.clone(),
write_template:config.write_template,
write_models:config.write_models,
write_scripts:config.write_scripts,
@ -696,14 +705,15 @@ fn write_commit(config:WriteCommitConfig,b:AResult<(AssetVersion,DecompiledConte
tree_index.write_tree()?
};
let tree=repo.find_tree(tree_id)?;
let commit_id=match last_commit{
Some(commit)=>repo.commit(
match repo.revwalk().unwrap().next(){
Some(Ok(commit_id))=>repo.commit(
Some("HEAD"),//update_ref
&sig,//author
&sig,//commiter
&format!("v{}", asset_version.assetVersionNumber),//message
&tree,//tree (basically files)
&[commit],//parents
&[&repo.find_commit(commit_id)?],//parents
).expect("commit failed"),
None=>repo.commit(
Some("HEAD"),//update_ref
@ -713,10 +723,11 @@ fn write_commit(config:WriteCommitConfig,b:AResult<(AssetVersion,DecompiledConte
&tree,//tree (basically files)
&[],//parents
).expect("commit failed"),
Some(Err(e))=>Err(e)?,
};
//commit
Ok(commit_id)
Ok(())
}
struct DecompileHistoryConfig{
@ -735,7 +746,6 @@ async fn decompile_history_into_git(config:DecompileHistoryConfig)->AResult<()>{
let asset_list=get_version_history(&client,&config.cookie.as_str(),config.asset_id).await?;
let repo=git2::Repository::init(config.output_folder.clone())?;
let mut commits=Vec::new();
//download all versions
let asset_id_string=config.asset_id.to_string();
@ -761,8 +771,8 @@ async fn decompile_history_into_git(config:DecompileHistoryConfig)->AResult<()>{
write_template:config.write_template,
write_models:config.write_models,
write_scripts:config.write_scripts,
},b,&repo,commits.last()){
Ok(commit_id)=>commits.push(repo.find_commit(commit_id).expect("Failed to find commit")),
},b,&repo){
Ok(())=>(),
Err(e)=>println!("download/unzip/decompile/write error: {}",e),
}
}).await;