Populate data/ hierarchy
[yangtools.git] / data / yang-data-util / src / main / java / org / opendaylight / yangtools / yang / data / util / AbstractNamespaceCodec.java
1 /*
2  * Copyright (c) 2014 Cisco Systems, Inc. and others.  All rights reserved.
3  *
4  * This program and the accompanying materials are made available under the
5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
6  * and is available at http://www.eclipse.org/legal/epl-v10.html
7  */
8 package org.opendaylight.yangtools.yang.data.util;
9
10 import static com.google.common.base.Preconditions.checkArgument;
11
12 import com.google.common.base.Splitter;
13 import java.util.Iterator;
14 import org.eclipse.jdt.annotation.NonNull;
15 import org.eclipse.jdt.annotation.Nullable;
16 import org.opendaylight.yangtools.concepts.AbstractIllegalArgumentCodec;
17 import org.opendaylight.yangtools.yang.common.QName;
18 import org.opendaylight.yangtools.yang.common.QNameModule;
19 import org.opendaylight.yangtools.yang.common.XMLNamespace;
20
21 abstract class AbstractNamespaceCodec<T> extends AbstractIllegalArgumentCodec<String, T> {
22     private static final Splitter COLON_SPLITTER = Splitter.on(':');
23
24     /**
25      * Return string prefix for a particular namespace, allocating a new one if necessary.
26      *
27      * @param namespace Namespace to map
28      * @return Allocated unique prefix, or null if the prefix cannot be mapped.
29      */
30     protected abstract @Nullable String prefixForNamespace(@NonNull XMLNamespace namespace);
31
32     /**
33      * Create a QName for a prefix and local name.
34      *
35      * @param prefix Prefix for namespace
36      * @param localName local name
37      * @return QName
38      * @throws IllegalArgumentException if the prefix cannot be resolved
39      */
40     protected abstract @Nullable QName createQName(@NonNull String prefix, @NonNull String localName);
41
42     private static String getIdAndPrefixAsStr(final String pathPart) {
43         int predicateStartIndex = pathPart.indexOf('[');
44         return predicateStartIndex == -1 ? pathPart : pathPart.substring(0, predicateStartIndex);
45     }
46
47     protected final @NonNull StringBuilder appendQName(final StringBuilder sb, final QName qname) {
48         final String prefix = prefixForNamespace(qname.getNamespace());
49         checkArgument(prefix != null, "Failed to map QName %s to prefix", qname);
50         return sb.append(prefix).append(':').append(qname.getLocalName());
51     }
52
53     /**
54      * Append a QName, potentially taking into account last QNameModule encountered in the serialized path.
55      *
56      * @param sb target StringBuilder
57      * @param qname QName to append
58      * @param lastModule last QNameModule encountered, may be null
59      * @return target StringBuilder
60      */
61     protected StringBuilder appendQName(final StringBuilder sb, final QName qname,
62             final @Nullable QNameModule lastModule) {
63         // Covers XML and uncompressed JSON codec
64         return appendQName(sb, qname);
65     }
66
67     protected final QName parseQName(final String str) {
68         final String xPathPartTrimmed = getIdAndPrefixAsStr(str).trim();
69         final Iterator<String> it = COLON_SPLITTER.split(xPathPartTrimmed).iterator();
70
71         // Empty string
72         if (!it.hasNext()) {
73             return null;
74         }
75
76
77         final String first = it.next().trim();
78         if (first.isEmpty()) {
79             return null;
80         }
81
82         final String identifier;
83         final String prefix;
84         if (it.hasNext()) {
85             // It is "prefix:value"
86             prefix = first;
87             identifier = it.next().trim();
88         } else {
89             prefix = "";
90             identifier = first;
91         }
92         if (identifier.isEmpty()) {
93             return null;
94         }
95
96         return createQName(prefix, identifier);
97     }
98 }