Skip to content

Commit

Permalink
add an upload example
Browse files Browse the repository at this point in the history
  • Loading branch information
jbr committed Jun 26, 2020
1 parent 1797c0e commit 7ea9006
Showing 1 changed file with 49 additions and 0 deletions.
49 changes: 49 additions & 0 deletions examples/upload.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,49 @@
use async_std::{fs::OpenOptions, io};
use tempfile::TempDir;
use tide::prelude::*;
use tide::{Body, Request, Response, StatusCode};

#[async_std::main]
async fn main() -> Result<(), std::io::Error> {
tide::log::start();
let mut app = tide::with_state(tempfile::tempdir()?);

// To test this example:
// $ cargo run --example upload
// $ curl -T ./README.md locahost:8080 # this writes the file to a temp directory
// $ curl localhost:8080/README.md # this reads the file from the same temp directory

app.at(":file")
.put(|req: Request<TempDir>| async move {
let path: String = req.param("file")?;
let fs_path = req.state().path().join(path);

let file = OpenOptions::new()
.create(true)
.write(true)
.open(&fs_path)
.await?;

let bytes_written = io::copy(req, file).await?;

tide::log::info!("file written", {
bytes: bytes_written,
path: fs_path.canonicalize()?.to_str()
});

Ok(json!({ "bytes": bytes_written }))
})
.get(|req: Request<TempDir>| async move {
let path: String = req.param("file")?;
let fs_path = req.state().path().join(path);

if let Ok(body) = Body::from_file(fs_path).await {
Ok(body.into())
} else {
Ok(Response::new(StatusCode::NotFound))
}
});

app.listen("127.0.0.1:8080").await?;
Ok(())
}

0 comments on commit 7ea9006

Please sign in to comment.