fix: Fix local node source being ignored

This commit is contained in:
sprites0
2025-07-02 10:45:31 +00:00
parent d90ddbbc65
commit ff6744c6d4

View File

@ -22,6 +22,12 @@ pub struct BlockSourceArgs {
impl BlockSourceArgs { impl BlockSourceArgs {
pub async fn parse(&self) -> eyre::Result<BlockSourceConfig> { pub async fn parse(&self) -> eyre::Result<BlockSourceConfig> {
let config = self.create_base_config().await?;
let config = self.apply_node_source_config(config);
Ok(config)
}
async fn create_base_config(&self) -> eyre::Result<BlockSourceConfig> {
if self.s3 { if self.s3 {
return Ok(BlockSourceConfig::s3_default().await); return Ok(BlockSourceConfig::s3_default().await);
} }
@ -32,17 +38,19 @@ impl BlockSourceArgs {
)); ));
}; };
let mut config = if let Some(bucket) = value.strip_prefix("s3://") { if let Some(bucket) = value.strip_prefix("s3://") {
BlockSourceConfig::s3(bucket.to_string()).await Ok(BlockSourceConfig::s3(bucket.to_string()).await)
} else { } else {
BlockSourceConfig::local(value.to_string()) Ok(BlockSourceConfig::local(value.to_string()))
}; }
if let Some(block_source_from_node) = self.block_source_from_node.as_ref() {
config = config.with_block_source_from_node(block_source_from_node.to_string());
} }
Ok(config) fn apply_node_source_config(&self, config: BlockSourceConfig) -> BlockSourceConfig {
let Some(block_source_from_node) = self.block_source_from_node.as_ref() else {
return config;
};
config.with_block_source_from_node(block_source_from_node.to_string())
} }
} }