Fix checkstyle in mdsal-binding2-generator-util
[mdsal.git] / binding2 / mdsal-binding2-java-api-generator / src / main / java / org / opendaylight / mdsal / binding / javav2 / java / api / generator / range_generators / LengthGenerator.java
1 /*
2  * Copyright (c) 2017 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.mdsal.binding.javav2.java.api.generator.range_generators;
9
10 import com.google.common.collect.Range;
11 import java.util.ArrayList;
12 import java.util.Collection;
13 import java.util.Set;
14 import javax.annotation.Nonnull;
15 import javax.annotation.Nullable;
16 import org.opendaylight.mdsal.binding.javav2.model.api.Type;
17 import org.opendaylight.yangtools.yang.model.api.type.LengthConstraint;
18
19 public final class LengthGenerator {
20     private LengthGenerator() {
21         throw new UnsupportedOperationException();
22     }
23
24     private static String lengthCheckerName(final String member) {
25         return "check" + member + "Length";
26     }
27
28     private static Collection<String> createExpressions(final LengthConstraint constraint) {
29         final Set<Range<Integer>> constraints = constraint.getAllowedRanges().asRanges();
30         final Collection<String> ret = new ArrayList<>(constraints.size());
31
32         for (Range<Integer> l : constraints) {
33             final StringBuilder sb = new StringBuilder("length >");
34
35             // We have to deal with restrictions being out of integer's range
36             if (l.lowerEndpoint() <= Integer.MAX_VALUE) {
37                 sb.append('=');
38             }
39             sb.append(' ').append(l.lowerEndpoint());
40
41             final int max = l.upperEndpoint();
42             if (max < Integer.MAX_VALUE) {
43                 sb.append(" && length <= ").append(max);
44             }
45
46             ret.add(sb.toString());
47         }
48
49         return ret;
50     }
51
52     private static String createLengthString(final LengthConstraint constraint) {
53         return new ArrayList<>(constraint.getAllowedRanges().asRanges()).toString();
54     }
55
56     private static String generateArrayLengthChecker(final String member, final LengthConstraint constraint) {
57         final StringBuilder sb = new StringBuilder();
58         final Collection<String> expressions = createExpressions(constraint);
59
60         sb.append("private static void ").append(lengthCheckerName(member)).append("(final byte[] value) {\n");
61
62         if (!expressions.isEmpty()) {
63             sb.append("    final int length = value.length;\n");
64
65             for (String exp : expressions) {
66                 sb.append("    if (").append(exp).append(") {\n");
67                 sb.append("        return;\n");
68                 sb.append("    }\n");
69             }
70
71             sb.append("    throw new IllegalArgumentException(String.format(\"Invalid length: %s, expected: ")
72               .append(createLengthString(constraint)).append(".\", java.util.Arrays.toString(value)));\n");
73         }
74
75         sb.append("}\n");
76
77         return sb.toString();
78     }
79
80     private static String generateStringLengthChecker(final String member, final LengthConstraint constraint) {
81         final StringBuilder sb = new StringBuilder();
82         final Collection<String> expressions = createExpressions(constraint);
83
84         sb.append("private static void ").append(lengthCheckerName(member)).append("(final String value) {\n");
85
86         if (!expressions.isEmpty()) {
87             sb.append("    final int length = value.length();\n");
88
89             for (String exp : expressions) {
90                 sb.append("    if (").append(exp).append(") {\n");
91                 sb.append("        return;\n");
92                 sb.append("    }\n");
93             }
94
95             sb.append("    throw new IllegalArgumentException(String.format(\"Invalid length: %s, expected: ")
96               .append(createLengthString(constraint)).append(".\", value));\n");
97         }
98
99         sb.append("}\n");
100
101         return sb.toString();
102     }
103
104     public static String generateLengthChecker(final String member, final Type type,
105             final LengthConstraint constraint) {
106         if (TypeUtils.getBaseYangType(type).getName().contains("[")) {
107             return generateArrayLengthChecker(member, constraint);
108         } else {
109             return generateStringLengthChecker(member, constraint);
110         }
111     }
112
113     public static String generateLengthCheckerCall(@Nullable final String member,
114             @Nonnull final String valueReference) {
115         return lengthCheckerName(member) + '(' + valueReference + ");\n";
116     }
117 }