Sulkta-OSS/rustypipe
Archived
2
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

feat: add build_with_client method to RustyPipeBuilder

Browse source 
fix: create data dir if it doesn't exist
This commit is contained in:
ThetaDev 2023-08-04 15:14:55 +02:00
parent 43ed52daf9
commit 43ef8d15c4
2 changed files with 7 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
cli/src/main.rs
View file

@ -393,7 +393,7 @@ async fn main() {
let mut storage_dir = dirs::data_dir().expect("no data dir"); let mut storage_dir = dirs::data_dir().expect("no data dir");
storage_dir.push("rustypipe"); storage_dir.push("rustypipe");
_ = std::fs::create_dir(&storage_dir); std::fs::create_dir_all(&storage_dir).expect("could not create data dir");
let mut rp = RustyPipe::builder().storage_dir(storage_dir); let mut rp = RustyPipe::builder().storage_dir(storage_dir);
if cli.report { if cli.report {

7
src/client/mod.rs
View file

@ -458,7 +458,12 @@ impl RustyPipeBuilder {
/// Return a new, configured RustyPipe instance. /// Return a new, configured RustyPipe instance.
pub fn build(self) -> Result<RustyPipe, Error> { pub fn build(self) -> Result<RustyPipe, Error> {
let mut client_builder = ClientBuilder::new() self.build_with_client(ClientBuilder::new())
}
/// Return a new, configured RustyPipe instance using a Reqwest client builder.
pub fn build_with_client(self, mut client_builder: ClientBuilder) -> Result<RustyPipe, Error> {
client_builder = client_builder
.user_agent(self.user_agent.unwrap_or_else(|| DEFAULT_UA.to_owned())) .user_agent(self.user_agent.unwrap_or_else(|| DEFAULT_UA.to_owned()))
.gzip(true) .gzip(true)
.brotli(true) .brotli(true)