|
|
|
@ -7,28 +7,33 @@ use log::{debug, error, info, log_enabled, warn}; |
|
|
|
|
use reqwest; |
|
|
|
|
use serde::Deserialize; |
|
|
|
|
use std::collections::HashMap; |
|
|
|
|
use std::fs; |
|
|
|
|
use std::io::{self, Write}; |
|
|
|
|
#[allow(unused)] |
|
|
|
|
use std::process::Command; |
|
|
|
|
use std::str; |
|
|
|
|
use toml; |
|
|
|
|
use url::Url; |
|
|
|
|
#[allow(unused)] |
|
|
|
|
use yaserde_derive::{YaDeserialize, YaSerialize}; |
|
|
|
|
use yaserde::de::from_str; |
|
|
|
|
|
|
|
|
|
#[derive(Deserialize)] |
|
|
|
|
#[derive(Debug, Deserialize)] |
|
|
|
|
pub struct Config { |
|
|
|
|
credentials: Credentials, |
|
|
|
|
paths: NextcloudPaths, |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
#[derive(Deserialize)] |
|
|
|
|
#[derive(Debug, Deserialize)] |
|
|
|
|
pub struct Credentials { |
|
|
|
|
username: String, |
|
|
|
|
password: String, |
|
|
|
|
password: Option<String>, |
|
|
|
|
password_script: Option<String>, |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
#[derive(Deserialize)] |
|
|
|
|
#[derive(Debug, Deserialize)] |
|
|
|
|
pub struct NextcloudPaths { |
|
|
|
|
root: Url, |
|
|
|
|
root: String, |
|
|
|
|
target: String, |
|
|
|
|
} |
|
|
|
|
|
|
|
|
@ -111,13 +116,16 @@ pub struct Collection { |
|
|
|
|
|
|
|
|
|
fn init() -> Result<Config, Box<dyn std::error::Error>> { |
|
|
|
|
let xdg_dirs = xdg::BaseDirectories::with_prefix("publicise-rs").unwrap(); |
|
|
|
|
let config_file = xdg_dirs.find_config_file("config.toml").unwrap(); |
|
|
|
|
return Error{}; |
|
|
|
|
let config_path = xdg_dirs.find_config_file("config.toml").unwrap(); |
|
|
|
|
let config_file_contents = fs::read_to_string(config_path).unwrap(); |
|
|
|
|
let config: Config = toml::from_str(&config_file_contents).unwrap(); |
|
|
|
|
return Ok(config); |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
async fn get_folder_contents(url_tail: &str, password: &str) -> Result<String, Box<dyn std::error::Error>> { |
|
|
|
|
#[allow(unused)] |
|
|
|
|
async fn get_folder_contents(url_tail: &str, config: &Config) -> Result<String, Box<dyn std::error::Error>> { |
|
|
|
|
debug!("[get_folder_contents] Entering function..."); |
|
|
|
|
let root_url = Url::parse("https://cloud.theadamcooper.com")?; |
|
|
|
|
let root_url = Url::parse(&config.paths.root)?; |
|
|
|
|
let method = reqwest::Method::from_bytes(b"PROPFIND").unwrap(); |
|
|
|
|
let client = reqwest::Client::new(); |
|
|
|
|
let url = root_url.join(url_tail)?; |
|
|
|
@ -138,7 +146,7 @@ async fn get_folder_contents(url_tail: &str, password: &str) -> Result<String, B |
|
|
|
|
); |
|
|
|
|
let response_text = client |
|
|
|
|
.request(method, url) |
|
|
|
|
.basic_auth("adam", Some(password)) |
|
|
|
|
.basic_auth("adam", config.credentials.password.as_ref()) |
|
|
|
|
.body(body) |
|
|
|
|
.send() |
|
|
|
|
.await? |
|
|
|
@ -148,6 +156,7 @@ async fn get_folder_contents(url_tail: &str, password: &str) -> Result<String, B |
|
|
|
|
Ok(response_text) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
#[allow(unused)] |
|
|
|
|
async fn publicise_it(path: &str, password: &str) -> Result<bool, Box<dyn std::error::Error>> { |
|
|
|
|
debug!("[publicise_it] Entering function..."); |
|
|
|
|
let url = "https://cloud.theadamcooper.com/ocs/v2.php/apps/files_sharing/api/v1/shares"; |
|
|
|
@ -169,7 +178,8 @@ async fn publicise_it(path: &str, password: &str) -> Result<bool, Box<dyn std::e |
|
|
|
|
Ok(true) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
async fn traverse(mut result: Multistatus, password: &str) -> Result<bool, Box<dyn std::error::Error>> { |
|
|
|
|
#[allow(unused)] |
|
|
|
|
async fn traverse(mut result: Multistatus, config: &Config) -> Result<bool, Box<dyn std::error::Error>> { |
|
|
|
|
debug!("[traverse] Entering function..."); |
|
|
|
|
// Initialize the indexed "pointer"
|
|
|
|
|
let mut current_index: usize = 0; |
|
|
|
@ -193,7 +203,7 @@ async fn traverse(mut result: Multistatus, password: &str) -> Result<bool, Box<d |
|
|
|
|
// Get the contents XML
|
|
|
|
|
let folder_contents: String = get_folder_contents( |
|
|
|
|
&result.response[current_index].href, // change to mutable borrow if necessary
|
|
|
|
|
password, |
|
|
|
|
config, |
|
|
|
|
) |
|
|
|
|
.await |
|
|
|
|
.unwrap(); |
|
|
|
@ -215,7 +225,7 @@ async fn traverse(mut result: Multistatus, password: &str) -> Result<bool, Box<d |
|
|
|
|
let index = &result.response[current_index].href.find(username_seg).unwrap_or(0); |
|
|
|
|
let new_index = index + username.len() + 2; |
|
|
|
|
let new_href = &result.response[current_index].href[new_index..]; |
|
|
|
|
publicise_it(new_href, password).await.unwrap(); |
|
|
|
|
publicise_it(new_href, config.credentials.password.as_ref().unwrap()).await.unwrap(); |
|
|
|
|
} else { |
|
|
|
|
debug!("[traverse] it's already public"); |
|
|
|
|
} |
|
|
|
@ -230,6 +240,7 @@ async fn traverse(mut result: Multistatus, password: &str) -> Result<bool, Box<d |
|
|
|
|
Ok(true) |
|
|
|
|
} |
|
|
|
|
|
|
|
|
|
#[allow(unused)] |
|
|
|
|
fn get_password() -> Result<String, std::io::Error> { |
|
|
|
|
print!("Nextcloud password: "); |
|
|
|
|
io::stdout().flush().unwrap(); |
|
|
|
@ -257,16 +268,21 @@ async fn main() -> std::io::Result<()> { |
|
|
|
|
.init(); |
|
|
|
|
println!("Publicise it!\n\n"); |
|
|
|
|
|
|
|
|
|
let config = init().unwrap(); |
|
|
|
|
let password: String = get_password().unwrap().trim().to_string(); |
|
|
|
|
let folder_contents: String = |
|
|
|
|
get_folder_contents("/remote.php/dav/files/adam/test_public", &password) |
|
|
|
|
let mut config = init().unwrap(); |
|
|
|
|
if config.credentials.password == None { |
|
|
|
|
let output = Command::new(&config.credentials.password_script.as_ref().unwrap()).output().unwrap(); |
|
|
|
|
config.credentials.password = Some(String::from_utf8(output.stdout).unwrap()); |
|
|
|
|
} |
|
|
|
|
debug!("{:?}", &config); |
|
|
|
|
let full_path = &(String::from("/remote.php/dav/files/") + &config.credentials.username + "/" + &config.paths.target); |
|
|
|
|
let folder_contents: String =
|
|
|
|
|
get_folder_contents(full_path, &config) |
|
|
|
|
.await |
|
|
|
|
.unwrap(); |
|
|
|
|
|
|
|
|
|
let result: Multistatus = from_str(&folder_contents).unwrap(); |
|
|
|
|
debug!("{:?}", result); |
|
|
|
|
|
|
|
|
|
let _ = traverse(result, &password).await.unwrap(); |
|
|
|
|
let _ = traverse(result, &config).await.unwrap(); |
|
|
|
|
Ok(()) |
|
|
|
|
} |
|
|
|
|