BUG-7833: Fix identityref codecs
[yangtools.git] / yang / yang-data-util / src / main / java / org / opendaylight / yangtools / yang / data / util / codec / QNameCodecUtil.java
1 /*
2  * Copyright (c) 2017 Pantheon Technologies, s.r.o. 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.codec;
9
10 import com.google.common.annotations.Beta;
11 import com.google.common.base.Preconditions;
12 import com.google.common.base.Splitter;
13 import java.util.Iterator;
14 import java.util.function.Function;
15 import org.opendaylight.yangtools.yang.common.QName;
16 import org.opendaylight.yangtools.yang.common.QNameModule;
17
18 /**
19  * Utility methods for parsing and writing QNames.
20  *
21  * @author Robert Varga
22  */
23 @Beta
24 public final class QNameCodecUtil {
25     private static final Splitter COLON_SPLITTER = Splitter.on(':').trimResults();
26
27     public static QName decodeQName(final String str, final Function<String, QNameModule> prefixToModule) {
28         final Iterator<String> it = COLON_SPLITTER.split(str).iterator();
29
30         // Empty string
31         final String identifier;
32         final String prefix;
33         if (it.hasNext()) {
34             final String first = it.next();
35             if (it.hasNext()) {
36                 // It is "prefix:value"
37                 prefix = first;
38                 identifier = it.next();
39                 Preconditions.checkArgument(!it.hasNext(), "Malformed QName '" + str + "'");
40             } else {
41                 prefix = "";
42                 identifier = first;
43             }
44         } else {
45             prefix = "";
46             identifier = "";
47         }
48
49         final QNameModule module = prefixToModule.apply(prefix);
50         Preconditions.checkArgument(module != null, "Cannot resolve prefix '%s' from %s", prefix, str);
51         return QName.create(module, identifier);
52     }
53
54     public static String encodeQName(final QName qname, final Function<QNameModule, String> moduleToPrefix) {
55         final String prefix = moduleToPrefix.apply(qname.getModule());
56         Preconditions.checkArgument(prefix != null, "Cannot allocated prefix for %s", qname);
57         return prefix.isEmpty() ? qname.getLocalName() : prefix + ":" + qname.getLocalName();
58     }
59 }