hyperdav 0.1.2

WebDAV client
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120

use hyper::client::Client as HttpClient;
use hyper::client::RequestBuilder;
use hyper::client::response::Response;
use hyper::client::IntoUrl;
use hyper::client::Body;
use std::io::Read;
use hyper::header::{Authorization, Basic};
use super::header::{Destination, Depth};
use super::Method;
use super::response::{parse_propfind_response, PropfindResponse};
use Error;

pub struct Client {
    http_client: HttpClient,
}

impl Client {
    pub fn new() -> Self {
        Client { http_client: HttpClient::new() }
    }

    /// Get a file
    pub fn get<'a, U: IntoUrl + Clone>(&'a self, url: U) -> Result<Response, Error> {
        let res = try!(self.request(Method::Get, url).send());

        if !res.status.is_success() {
            return Err(Error::ErrorResponse(res));
        }

        Ok(res)
    }

    /// Put a file
    pub fn put<'a, U: IntoUrl + Clone>(&'a self, body: &'a mut Read, url: U) -> Result<(), Error> {
        let res = try!(self.request(Method::Put, url).body(Body::ChunkedBody(body)).send());

        if !res.status.is_success() {
            return Err(Error::ErrorResponse(res));
        }

        Ok(())
    }

    /// Create a directory
    pub fn mkdir<'a, U: IntoUrl + Clone>(&'a self, url: U) -> Result<(), Error> {
        let res = try!(self.request(Method::Mkcol, url).send());

        if !res.status.is_success() {
            return Err(Error::ErrorResponse(res));
        }

        Ok(())
    }

    /// Rename/move a directory or file
    pub fn mv<'a, U: IntoUrl + Clone>(&'a self, from: U, to: U) -> Result<(), Error> {
        let req = self.request(Method::Move, from);

        // Set destination header
        let req = if let Ok(url) = to.into_url() {
            req.header(Destination(url.to_string()))
        } else {
            req
        };

        let res = try!(req.send());

        if !res.status.is_success() {
            return Err(Error::ErrorResponse(res));
        }
        Ok(())
    }

    /// List files in a directory
    pub fn ls<'a, U: IntoUrl + Clone>(&'a self, url: U) -> Result<Vec<PropfindResponse>, Error> {
        let body = r#"<?xml version="1.0" encoding="utf-8" ?>
            <D:propfind xmlns:D="DAV:">
                <D:allprop/>
            </D:propfind>
        "#;

        let res = try!(self.request(Method::Propfind, url)
                           .header(Depth("Infinity".into()))
                           .body(body)
                           .send());

        if !res.status.is_success() {
            return Err(Error::ErrorResponse(res));
        }

        Ok(try!(parse_propfind_response(res)))
    }


    // FIXME can we somehow parse the url AND get rid of Clone?
    pub fn request<'a, U: IntoUrl + Clone>(&'a self, method: Method, url: U) -> RequestBuilder<'a> {
        let auth_header = match url.clone().into_url() {
            Ok(url) => {
                match url.password() {
                    Some(password) => {
                        Some(Authorization(Basic {
                            username: url.username().into(),
                            password: Some(password.into()),
                        }))
                    }
                    None => None,
                }
            }
            Err(_) => None,
        };

        let req = self.http_client.request(method.into(), url);

        if let Some(header) = auth_header {
            req.header(header)
        } else {
            req
        }
    }
}