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

use std::collections::HashSet;
use std::convert::AsRef;
use std::hash::Hash;
use std::io;

use lber::structures::{Tag, Enumerated, Sequence, Set, OctetString};
use lber::common::TagClass;

use futures::{future, Future};
use tokio_service::Service;

use ldap::{Ldap, LdapOp, next_req_controls};
use result::LdapResult;

/// Possible sub-operations for the Modify operation.
#[derive(Clone, Debug, PartialEq)]
pub enum Mod<S: AsRef<str> + Eq + Hash> {
    /// Add an attribute, with at least one value.
    Add(S, HashSet<S>),
    /// Delete the entire attribute, or the given values of an attribute.
    Delete(S, HashSet<S>),
    /// Replace an existing attribute, setting its values to those in the set, or delete it if no values are given.
    Replace(S, HashSet<S>),
}

impl Ldap {
    /// See [`LdapConn::modify()`](struct.LdapConn.html#method.modify).
    pub fn modify<S: AsRef<str> + Eq + Hash>(&self, dn: &str, mods: Vec<Mod<S>>) ->
            Box<Future<Item=LdapResult, Error=io::Error>> {
        let mut any_add_empty = false;
        let req = Tag::Sequence(Sequence {
            id: 6,
            class: TagClass::Application,
            inner: vec![
                Tag::OctetString(OctetString {
                    inner: Vec::from(dn.as_bytes()),
                    .. Default::default()
                }),
                Tag::Sequence(Sequence {
                    inner: mods.into_iter().map(|m| {
                        let mut is_add = false;
                        let (num, attr, set) = match m {
                            Mod::Add(attr, set) => { is_add = true; (0, attr, set) },
                            Mod::Delete(attr, set) => (1, attr, set),
                            Mod::Replace(attr, set) => (2, attr, set),
                        };
                        if set.is_empty() && is_add {
                            any_add_empty = true;
                        }
                        let op = Tag::Enumerated(Enumerated {
                            inner: num,
                            .. Default::default()
                        });
                        let part_attr = Tag::Sequence(Sequence {
                            inner: vec![
                                Tag::OctetString(OctetString {
                                    inner: Vec::from(attr.as_ref()),
                                    .. Default::default()
                                }),
                                Tag::Set(Set {
                                    inner: set.into_iter().map(|val| {
                                        Tag::OctetString(OctetString {
                                            inner: Vec::from(val.as_ref()),
                                            .. Default::default()
                                        })
                                    }).collect(),
                                    .. Default::default()
                                })
                            ],
                            .. Default::default()
                        });
                        Tag::Sequence(Sequence {
                            inner: vec![op, part_attr],
                            .. Default::default()
                        })
                    }).collect(),
                    .. Default::default()
                })
            ]
        });
        if any_add_empty {
            return Box::new(future::err(io::Error::new(io::ErrorKind::Other, "empty value set for Add")));
        }

        let fut = self.call(LdapOp::Single(req, next_req_controls(self)))
            .and_then(|response| {
                let (mut result, controls) = (LdapResult::from(response.0), response.1);
                result.ctrls = controls;
                Ok(result)
            });

        Box::new(fut)
    }
}