Remove unused suppression
[yangtools.git] / data / yang-data-spi / src / main / java / org / opendaylight / yangtools / yang / data / spi / node / NormalizedNodePrettyTree.java
1 /*
2  * Copyright (c) 2021 PANTHEON.tech s.r.o. 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.spi.node;
9
10 import static java.util.Objects.requireNonNull;
11
12 import com.google.common.annotations.Beta;
13 import java.util.Base64;
14 import java.util.Iterator;
15 import java.util.Locale;
16 import java.util.Optional;
17 import org.eclipse.jdt.annotation.NonNull;
18 import org.opendaylight.yangtools.concepts.Immutable;
19 import org.opendaylight.yangtools.concepts.PrettyTree;
20 import org.opendaylight.yangtools.concepts.PrettyTreeAware;
21 import org.opendaylight.yangtools.yang.common.QName;
22 import org.opendaylight.yangtools.yang.common.QNameModule;
23 import org.opendaylight.yangtools.yang.common.Revision;
24 import org.opendaylight.yangtools.yang.data.api.schema.ForeignDataNode;
25 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
26 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNodeContainer;
27 import org.opendaylight.yangtools.yang.data.api.schema.ValueNode;
28
29 @Beta
30 public final class NormalizedNodePrettyTree extends PrettyTree implements Immutable {
31     private final @NonNull NormalizedNode node;
32
33     public NormalizedNodePrettyTree(final @NonNull NormalizedNode node) {
34         this.node = requireNonNull(node);
35     }
36
37     @Override
38     public void appendTo(final StringBuilder sb, final int depth) {
39         appendNode(sb, depth, null, node);
40     }
41
42     private static void appendNode(final StringBuilder sb, final int depth, final QNameModule parentNamespace,
43             final NormalizedNode node) {
44         final String simpleName = node.contract().getSimpleName();
45         appendIndent(sb, depth);
46         sb.append(simpleName.toLowerCase(Locale.ROOT).charAt(0)).append(simpleName, 1, simpleName.length()).append(' ');
47
48         final QName qname = node.name().getNodeType();
49         final QNameModule currentNamespace = qname.getModule();
50         appendNamespace(sb, parentNamespace, currentNamespace);
51         sb.append(qname.getLocalName()).append(' ');
52
53         if (node instanceof NormalizedNodeContainer) {
54             final NormalizedNodeContainer<?> container = (NormalizedNodeContainer<?>) node;
55             sb.append("= {");
56
57             final Iterator<? extends NormalizedNode> it = container.body().iterator();
58             if (it.hasNext()) {
59                 final int childIndent = depth + 1;
60                 do {
61                     sb.append('\n');
62                     appendNode(sb, childIndent, currentNamespace, it.next());
63                 } while (it.hasNext());
64
65                 sb.append('\n');
66                 appendIndent(sb, depth);
67             }
68             sb.append('}');
69         } else if (node instanceof ValueNode) {
70             sb.append("= ");
71             final Object value = node.body();
72             if (value instanceof byte[]) {
73                 sb.append("(byte[])").append(Base64.getEncoder().encodeToString((byte[]) value));
74             } else if (value instanceof String) {
75                 appendString(sb, (String) value);
76             } else {
77                 sb.append(value);
78             }
79         } else if (node instanceof ForeignDataNode) {
80             final ForeignDataNode<?> data = (ForeignDataNode<?>) node;
81             final Object body = data.body();
82             if (body instanceof PrettyTreeAware) {
83                 sb.append("= {\n");
84                 ((PrettyTreeAware) body).prettyTree().appendTo(sb, depth + 1);
85                 appendIndent(sb, depth);
86                 sb.append('}');
87             } else {
88                 sb.append("= (").append(data.bodyObjectModel().getName()).append(')');
89             }
90         } else {
91             throw new IllegalStateException("Unhandled node " + node);
92         }
93     }
94
95     private static boolean appendNamespace(final StringBuilder sb, final QNameModule parent,
96             final QNameModule current) {
97         if (!current.equals(parent)) {
98             sb.append('(').append(current.getNamespace());
99             final Optional<Revision> rev = current.getRevision();
100             if (rev.isPresent()) {
101                 sb.append('@').append(rev.orElseThrow());
102             }
103             sb.append(')');
104             return true;
105         }
106         return false;
107     }
108
109     private static void appendString(final StringBuilder sb, final String str) {
110         // TODO: do some escaping: '\r' '\n' '"' '\\' to make things even more zazzy
111         sb.append('"').append(str).append('"');
112     }
113 }