mirror of https://github.com/OpenIdentityPlatform/OpenDJ.git

Violette Roche-Montane
09.26.2012 04ff552c42f3200dd71a58f94e9d57436b67c4f3
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
121
122
123
124
125
126
127
128
129
130
131
132
133
134
/*
 * CDDL HEADER START
 *
 * The contents of this file are subject to the terms of the
 * Common Development and Distribution License, Version 1.0 only
 * (the "License").  You may not use this file except in compliance
 * with the License.
 *
 * You can obtain a copy of the license at legal-notices/CDDLv1_0.txt
 * or http://forgerock.org/license/CDDLv1.0.html.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 *
 * When distributing Covered Code, include this CDDL HEADER in each
 * file and include the License file at legal-notices/CDDLv1_0.txt.
 * If applicable, add the following below this CDDL HEADER, with the
 * fields enclosed by brackets "[]" replaced with your own identifying
 * information:
 *      Portions Copyright [yyyy] [name of copyright owner]
 *
 * CDDL HEADER END
 *
 *
 *      Copyright 2009 Sun Microsystems, Inc.
 *      Portions copyright 2011 ForgeRock AS
 */
 
package org.forgerock.opendj.ldif;
 
import java.util.HashSet;
import java.util.LinkedList;
import java.util.List;
import java.util.Set;
 
import org.forgerock.opendj.ldap.AttributeDescription;
import org.forgerock.opendj.ldap.DN;
import org.forgerock.opendj.ldap.Entry;
import org.forgerock.opendj.ldap.Matcher;
import org.forgerock.opendj.ldap.schema.AttributeType;
 
/**
 * Common LDIF reader/writer functionality.
 */
abstract class AbstractLDIFStream {
 
    final Set<AttributeDescription> excludeAttributes = new HashSet<AttributeDescription>();
 
    boolean excludeOperationalAttributes = false;
 
    boolean excludeUserAttributes = false;
 
    final Set<AttributeDescription> includeAttributes = new HashSet<AttributeDescription>();
 
    final Set<DN> includeBranches = new HashSet<DN>();
 
    final Set<DN> excludeBranches = new HashSet<DN>();
 
    final List<Matcher> includeFilters = new LinkedList<Matcher>();
 
    final List<Matcher> excludeFilters = new LinkedList<Matcher>();
 
    /**
     * Creates a new abstract LDIF stream.
     */
    AbstractLDIFStream() {
        // Nothing to do.
    }
 
    final boolean isAttributeExcluded(final AttributeDescription attributeDescription) {
        if (!excludeAttributes.isEmpty() && excludeAttributes.contains(attributeDescription)) {
            return true;
        }
 
        // Let explicit include override more general exclude.
        if (!includeAttributes.isEmpty()) {
            return !includeAttributes.contains(attributeDescription);
        }
 
        final AttributeType type = attributeDescription.getAttributeType();
 
        if (excludeOperationalAttributes && type.isOperational()) {
            return true;
        }
 
        if (excludeUserAttributes && !type.isOperational()) {
            return true;
        }
 
        return false;
    }
 
    final boolean isBranchExcluded(final DN dn) {
        if (!excludeBranches.isEmpty()) {
            for (final DN excludeBranch : excludeBranches) {
                if (excludeBranch.isSuperiorOrEqualTo(dn)) {
                    return true;
                }
            }
        }
 
        if (!includeBranches.isEmpty()) {
            for (final DN includeBranch : includeBranches) {
                if (includeBranch.isSuperiorOrEqualTo(dn)) {
                    return false;
                }
            }
            return true;
        }
 
        return false;
    }
 
    final boolean isEntryExcluded(final Entry entry) {
        if (!excludeFilters.isEmpty()) {
            for (final Matcher excludeFilter : excludeFilters) {
                if (excludeFilter.matches(entry).toBoolean()) {
                    return true;
                }
            }
        }
 
        if (!includeFilters.isEmpty()) {
            for (final Matcher includeFilter : includeFilters) {
                if (includeFilter.matches(entry).toBoolean()) {
                    return false;
                }
            }
            return true;
        }
 
        return false;
    }
 
}