Mercurial > lbo > hg > async-google-apis
view drive_example/src/main.rs @ 45:0f8e74d57c90
Follow redirects for downloads.
author | Lewin Bormann <lbo@spheniscida.de> |
---|---|
date | Mon, 19 Oct 2020 13:59:20 +0200 |
parents | a3d203a72475 |
children | c6034ecb5716 |
line wrap: on
line source
//! List your Google Drive root folder, or upload a file there. //! //! When run with no arguments, a very detailed listing of all objects in your root folder is //! printed. //! //! When you specify a file name as command line argument, the given file is uploaded to your //! Google Drive. mod drive_v3_types; use drive_v3_types as drive; use async_google_apis_common as common; use std::rc::Rc; use std::fs; use std::path::Path; /// Create a new HTTPS client. fn https_client() -> common::TlsClient { let conn = hyper_rustls::HttpsConnector::new(); let cl = hyper::Client::builder().build(conn); cl } /// Upload a local file `f` to your drive. async fn upload_file(mut cl: drive::FilesService, f: &Path) -> anyhow::Result<()> { cl.set_scopes(&["https://www.googleapis.com/auth/drive.file"]); let data = hyper::body::Bytes::from(fs::read(&f)?); let mut params = drive::FilesCreateParams::default(); params.include_permissions_for_view = Some("published".to_string()); println!("{:?}", params); // Upload data using the upload version of create(). We obtain a `File` object. let resp = cl.create_upload(¶ms, data).await?; println!("{:?}", resp); // Copy ID from response. let file_id = resp.id.unwrap(); let fname = f.file_name().unwrap().to_str().unwrap(); // Rename file to the file name on our computer. let mut params = drive::FilesUpdateParams::default(); println!("{:?}", params); params.file_id = file_id.clone(); params.include_permissions_for_view = Some("published".to_string()); let mut file = drive::File::default(); file.name = Some(fname.to_string()); let update_resp = cl.update(¶ms, &file).await; println!("{:?}", update_resp); // Now get the file and check that it is correct. let mut params = drive::FilesGetParams::default(); params.file_id = file_id.clone(); let get_file = cl.get(¶ms).await?; println!("{:?}", get_file); assert!(get_file.name == Some(fname.to_string())); Ok(()) } #[tokio::main] async fn main() { let https = https_client(); // Put your client secret in the working directory! let sec = yup_oauth2::read_application_secret("client_secret.json") .await .expect("client secret couldn't be read."); let auth = yup_oauth2::InstalledFlowAuthenticator::builder( sec, yup_oauth2::InstalledFlowReturnMethod::HTTPRedirect, ) .persist_tokens_to_disk("tokencache.json") .hyper_client(https.clone()) .build() .await .expect("InstalledFlowAuthenticator failed to build"); let scopes = vec![drive::DriveScopes::DriveFile]; let mut cl = drive::FilesService::new(https, Rc::new(auth)); cl.set_scopes(&scopes); let arg = std::env::args().skip(1).next(); if let Some(fp) = arg { upload_file(cl, Path::new(&fp)) .await .expect("Upload failed :("); } else { // By default, list root directory. let mut p = drive::FilesListParams::default(); p.q = Some("'root' in parents".to_string()); let resp = cl.list(&p).await.expect("listing your Drive failed!"); if let Some(files) = resp.files { for f in files { println!( "{} => {:?}", f.name.as_ref().unwrap_or(&"???".to_string()), f ); } } } }