forked from StrafesNET/asset-tool
fix override file duplication
This commit is contained in:
parent
fc4cebc862
commit
d5b8c10264
70
src/main.rs
70
src/main.rs
@ -1111,10 +1111,16 @@ impl std::fmt::Display for QueryResolveError{
|
|||||||
}
|
}
|
||||||
impl std::error::Error for QueryResolveError{}
|
impl std::error::Error for QueryResolveError{}
|
||||||
|
|
||||||
async fn get_file_async(mut path:std::path::PathBuf,file_name:impl AsRef<std::path::Path>)->Result<tokio::fs::File,QueryResolveError>{
|
struct FileWithName{
|
||||||
|
file:tokio::fs::File,
|
||||||
|
name:String,
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn get_file_async(mut path:std::path::PathBuf,file_name:impl AsRef<std::path::Path>)->Result<FileWithName,QueryResolveError>{
|
||||||
|
let name=file_name.as_ref().to_str().unwrap().to_owned();
|
||||||
path.push(file_name);
|
path.push(file_name);
|
||||||
match tokio::fs::File::open(path).await{
|
match tokio::fs::File::open(path).await{
|
||||||
Ok(file)=>Ok(file),
|
Ok(file)=>Ok(FileWithName{file,name}),
|
||||||
Err(e)=>match e.kind(){
|
Err(e)=>match e.kind(){
|
||||||
std::io::ErrorKind::NotFound=>Err(QueryResolveError::NotFound),
|
std::io::ErrorKind::NotFound=>Err(QueryResolveError::NotFound),
|
||||||
_=>Err(QueryResolveError::IO(e)),
|
_=>Err(QueryResolveError::IO(e)),
|
||||||
@ -1125,7 +1131,7 @@ type QueryHintResult=Result<FileHint,QueryResolveError>;
|
|||||||
trait Query{
|
trait Query{
|
||||||
async fn resolve(self)->QueryHintResult;
|
async fn resolve(self)->QueryHintResult;
|
||||||
}
|
}
|
||||||
type QueryHandle=tokio::task::JoinHandle<Result<tokio::fs::File,QueryResolveError>>;
|
type QueryHandle=tokio::task::JoinHandle<Result<FileWithName,QueryResolveError>>;
|
||||||
struct QuerySingle{
|
struct QuerySingle{
|
||||||
script:QueryHandle,
|
script:QueryHandle,
|
||||||
}
|
}
|
||||||
@ -1281,14 +1287,15 @@ fn extract_script_overrides(mut source:String)->AResult<ScriptWithOverrides>{
|
|||||||
Ok(ScriptWithOverrides{overrides,source:source.split_off(count)})
|
Ok(ScriptWithOverrides{overrides,source:source.split_off(count)})
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn script_node(search_name:&str,mut file:tokio::fs::File,hint:ScriptHint)->AResult<CompileNode>{
|
async fn script_node(search_name:&str,mut file:FileWithName,hint:ScriptHint)->AResult<CompileNode>{
|
||||||
//read entire file
|
//read entire file
|
||||||
let mut buf=String::new();
|
let mut buf=String::new();
|
||||||
file.read_to_string(&mut buf).await?;
|
file.file.read_to_string(&mut buf).await?;
|
||||||
//regex script according to Properties lines at the top
|
//regex script according to Properties lines at the top
|
||||||
let script_with_overrides=extract_script_overrides(buf)?;
|
let script_with_overrides=extract_script_overrides(buf)?;
|
||||||
//script
|
//script
|
||||||
Ok(CompileNode{
|
Ok(CompileNode{
|
||||||
|
blacklist:Some(file.name),
|
||||||
name:script_with_overrides.overrides.name.unwrap_or_else(||search_name.to_owned()),
|
name:script_with_overrides.overrides.name.unwrap_or_else(||search_name.to_owned()),
|
||||||
class:match (script_with_overrides.overrides.class.as_deref(),hint){
|
class:match (script_with_overrides.overrides.class.as_deref(),hint){
|
||||||
(Some("ModuleScript"),_)
|
(Some("ModuleScript"),_)
|
||||||
@ -1302,13 +1309,14 @@ async fn script_node(search_name:&str,mut file:tokio::fs::File,hint:ScriptHint)-
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
async fn model_node(search_name:&str,mut file:tokio::fs::File)->AResult<CompileNode>{
|
async fn model_node(search_name:&str,mut file:FileWithName)->AResult<CompileNode>{
|
||||||
//read entire file
|
//read entire file
|
||||||
let mut buf=Vec::new();
|
let mut buf=Vec::new();
|
||||||
file.read_to_end(&mut buf).await?;
|
file.file.read_to_end(&mut buf).await?;
|
||||||
//model
|
//model
|
||||||
Ok(CompileNode{
|
Ok(CompileNode{
|
||||||
name:search_name.to_owned(),
|
blacklist:Some(file.name),
|
||||||
|
name:search_name.to_owned(),//wrong but gets overwritten by internal model name
|
||||||
class:CompileClass::Model(buf),
|
class:CompileClass::Model(buf),
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
@ -1340,6 +1348,7 @@ async fn locate_override_file(entry:&tokio::fs::DirEntry,style:Option<DecompileS
|
|||||||
//neither
|
//neither
|
||||||
(Err(QueryResolveError::NotFound),Err(QueryResolveError::NotFound))=>CompileNode{
|
(Err(QueryResolveError::NotFound),Err(QueryResolveError::NotFound))=>CompileNode{
|
||||||
name:search_name.to_owned(),
|
name:search_name.to_owned(),
|
||||||
|
blacklist:None,
|
||||||
class:CompileClass::Folder,
|
class:CompileClass::Folder,
|
||||||
},
|
},
|
||||||
//other error
|
//other error
|
||||||
@ -1399,8 +1408,8 @@ async fn discern_file(entry:&tokio::fs::DirEntry,style:Option<DecompileStyle>)->
|
|||||||
file_name.truncate(file_name.len()-ext_len);
|
file_name.truncate(file_name.len()-ext_len);
|
||||||
let file=tokio::fs::File::open(entry.path()).await?;
|
let file=tokio::fs::File::open(entry.path()).await?;
|
||||||
Ok(match file_discernment{
|
Ok(match file_discernment{
|
||||||
FileDiscernment::Model=>model_node(file_name.as_str(),file).await?,
|
FileDiscernment::Model=>model_node(file_name.as_str(),FileWithName{file,name:file_name.clone()}).await?,
|
||||||
FileDiscernment::Script(hint)=>script_node(file_name.as_str(),file,hint).await?,
|
FileDiscernment::Script(hint)=>script_node(file_name.as_str(),FileWithName{file,name:file_name.clone()},hint).await?,
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1411,7 +1420,7 @@ enum ScriptHint{
|
|||||||
ModuleScript,
|
ModuleScript,
|
||||||
}
|
}
|
||||||
struct FileHint{
|
struct FileHint{
|
||||||
file:tokio::fs::File,
|
file:FileWithName,
|
||||||
hint:ScriptHint,
|
hint:ScriptHint,
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1430,11 +1439,12 @@ enum CompileClass{
|
|||||||
|
|
||||||
struct CompileNode{
|
struct CompileNode{
|
||||||
name:String,
|
name:String,
|
||||||
|
blacklist:Option<String>,
|
||||||
class:CompileClass,
|
class:CompileClass,
|
||||||
}
|
}
|
||||||
|
|
||||||
enum CompileStackInstruction{
|
enum CompileStackInstruction{
|
||||||
TraverseReferent(rbx_dom_weak::types::Ref),
|
TraverseReferent(rbx_dom_weak::types::Ref,Option<String>),
|
||||||
PopFolder,
|
PopFolder,
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -1469,10 +1479,10 @@ async fn compile(config:CompileConfig)->AResult<()>{
|
|||||||
//add in scripts and models
|
//add in scripts and models
|
||||||
let mut folder=config.input_folder.clone();
|
let mut folder=config.input_folder.clone();
|
||||||
folder.push("src");
|
folder.push("src");
|
||||||
let mut stack:Vec<CompileStackInstruction>=dom.root().children().into_iter().map(|&referent|CompileStackInstruction::TraverseReferent(referent)).collect();
|
let mut stack:Vec<CompileStackInstruction>=dom.root().children().into_iter().map(|&referent|CompileStackInstruction::TraverseReferent(referent,None)).collect();
|
||||||
while let Some(instruction)=stack.pop(){
|
while let Some(instruction)=stack.pop(){
|
||||||
match instruction{
|
match instruction{
|
||||||
CompileStackInstruction::TraverseReferent(item_ref)=>{
|
CompileStackInstruction::TraverseReferent(item_ref,blacklist)=>{
|
||||||
let sans={
|
let sans={
|
||||||
let item=dom.get_by_ref(item_ref).ok_or(anyhow::Error::msg("null child ref"))?;
|
let item=dom.get_by_ref(item_ref).ok_or(anyhow::Error::msg("null child ref"))?;
|
||||||
sanitize(item.name.as_str()).to_string()
|
sanitize(item.name.as_str()).to_string()
|
||||||
@ -1482,6 +1492,9 @@ async fn compile(config:CompileConfig)->AResult<()>{
|
|||||||
//check if a folder exists with item.name
|
//check if a folder exists with item.name
|
||||||
if let Ok(mut dir)=tokio::fs::read_dir(folder.as_path()).await{
|
if let Ok(mut dir)=tokio::fs::read_dir(folder.as_path()).await{
|
||||||
let mut exist_names=std::collections::HashSet::new();
|
let mut exist_names=std::collections::HashSet::new();
|
||||||
|
if let Some(dont)=blacklist{
|
||||||
|
exist_names.insert(dont);
|
||||||
|
}
|
||||||
{
|
{
|
||||||
let item=dom.get_by_ref(item_ref).ok_or(anyhow::Error::msg("null child ref"))?;
|
let item=dom.get_by_ref(item_ref).ok_or(anyhow::Error::msg("null child ref"))?;
|
||||||
for &child_ref in item.children(){
|
for &child_ref in item.children(){
|
||||||
@ -1508,28 +1521,37 @@ async fn compile(config:CompileConfig)->AResult<()>{
|
|||||||
//discern that bad boy
|
//discern that bad boy
|
||||||
Ok::<_,anyhow::Error>(
|
Ok::<_,anyhow::Error>(
|
||||||
//prepare data structure
|
//prepare data structure
|
||||||
match compile_class.class{
|
(compile_class.blacklist,match compile_class.class{
|
||||||
CompileClass::Folder=>PreparedData::Builder(rbx_dom_weak::InstanceBuilder::new("Folder").with_name(compile_class.name.as_str())),
|
CompileClass::Folder=>PreparedData::Builder(rbx_dom_weak::InstanceBuilder::new("Folder").with_name(compile_class.name.as_str())),
|
||||||
CompileClass::Script(source)=>PreparedData::Builder(script_builder("Script",compile_class.name.as_str(),source)),
|
CompileClass::Script(source)=>PreparedData::Builder(script_builder("Script",compile_class.name.as_str(),source)),
|
||||||
CompileClass::LocalScript(source)=>PreparedData::Builder(script_builder("LocalScript",compile_class.name.as_str(),source)),
|
CompileClass::LocalScript(source)=>PreparedData::Builder(script_builder("LocalScript",compile_class.name.as_str(),source)),
|
||||||
CompileClass::ModuleScript(source)=>PreparedData::Builder(script_builder("ModuleScript",compile_class.name.as_str(),source)),
|
CompileClass::ModuleScript(source)=>PreparedData::Builder(script_builder("ModuleScript",compile_class.name.as_str(),source)),
|
||||||
CompileClass::Model(buf)=>PreparedData::Model(rbx_xml::from_reader_default(std::io::Cursor::new(buf))?),
|
CompileClass::Model(buf)=>PreparedData::Model(rbx_xml::from_reader_default(std::io::Cursor::new(buf))?),
|
||||||
}
|
})
|
||||||
)
|
)
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
//this is only able to begin after dir iterator is exhausted
|
//push existing dom children objects onto stack
|
||||||
|
{
|
||||||
|
let item=dom.get_by_ref(item_ref).ok_or(anyhow::Error::msg("null child ref"))?;
|
||||||
|
stack.extend(item.children().into_iter().map(|&referent|CompileStackInstruction::TraverseReferent(referent,None)))
|
||||||
|
};
|
||||||
//push child objects onto dom
|
//push child objects onto dom
|
||||||
|
//this is only able to begin after dir iterator is exhausted
|
||||||
while let Some(goober)=join_set.join_next().await{
|
while let Some(goober)=join_set.join_next().await{
|
||||||
match goober??{
|
let (blacklist,data)=goober??;
|
||||||
PreparedData::Model(mut model_dom)=>model_dom.transfer(model_dom.root().children()[0],&mut dom,item_ref),
|
let referent=match data{
|
||||||
PreparedData::Builder(script)=>{dom.insert(item_ref,script);},
|
PreparedData::Model(mut model_dom)=>{
|
||||||
}
|
let referent=model_dom.root().children()[0];
|
||||||
|
model_dom.transfer(referent,&mut dom,item_ref);
|
||||||
|
referent
|
||||||
|
},
|
||||||
|
PreparedData::Builder(script)=>dom.insert(item_ref,script),
|
||||||
|
};
|
||||||
|
//new children need to be traversed
|
||||||
|
stack.push(CompileStackInstruction::TraverseReferent(referent,blacklist));
|
||||||
}
|
}
|
||||||
//push dom children objects onto stack
|
|
||||||
let item=dom.get_by_ref(item_ref).ok_or(anyhow::Error::msg("null child ref"))?;
|
|
||||||
stack.extend(item.children().into_iter().map(|&referent|CompileStackInstruction::TraverseReferent(referent)));
|
|
||||||
}
|
}
|
||||||
},
|
},
|
||||||
CompileStackInstruction::PopFolder=>assert!(folder.pop(),"pop folder bad"),
|
CompileStackInstruction::PopFolder=>assert!(folder.pop(),"pop folder bad"),
|
||||||
|
Loading…
Reference in New Issue
Block a user