265aea10c59f8f858fb036b5fb97216ccc404a35
[controller.git] / opendaylight / config / yang-jmx-generator-plugin / src / main / java / org / opendaylight / controller / config / yangjmxgenerator / plugin / util / StringUtil.java
1 package org.opendaylight.controller.config.yangjmxgenerator.plugin.util;
2
3 import com.google.common.base.Joiner;
4 import com.google.common.base.Optional;
5 import com.google.common.base.Splitter;
6 import com.google.common.base.Strings;
7 import org.apache.commons.io.IOUtils;
8 import org.apache.commons.lang3.StringUtils;
9 import org.opendaylight.controller.config.yangjmxgenerator.plugin.java.FullyQualifiedName;
10 import org.slf4j.Logger;
11 import org.slf4j.LoggerFactory;
12
13 import java.io.IOException;
14 import java.io.InputStream;
15 import java.util.List;
16 import java.util.regex.Pattern;
17
18 public class StringUtil {
19     private static final Logger logger = LoggerFactory.getLogger(StringUtils.class);
20
21     /**
22      * @param list   of strings to be joined by ','
23      * @param prefix e.g. 'extends' or 'implements'
24      */
25     public static String prefixAndJoin(List<FullyQualifiedName> list, String prefix) {
26         if (list.isEmpty()) {
27             return "";
28         }
29         Joiner joiner = Joiner.on(",");
30         return " " + prefix + " " + joiner.join(list);
31     }
32
33     public static String addAsterixAtEachLineStart(String input) {
34         String s = Pattern.compile("^", Pattern.MULTILINE).matcher(input).replaceAll("* ");
35         // remove trailing spaces
36         s = Pattern.compile("\\s+$", Pattern.MULTILINE).matcher(s).replaceAll("");
37         s = ensureEndsWithSingleNewLine(s);
38         return s;
39     }
40
41     private static String ensureEndsWithSingleNewLine(String s) {
42         // .split Only trailing empty strings are skipped.
43         String[] split = s.split("\n");
44         s = Joiner.on("\n").join(split);
45         s = s + "\n";
46         return s;
47     }
48
49     public static String writeComment(String input, boolean isJavadoc) {
50         StringBuilder content = new StringBuilder();
51         content.append("/*");
52         if (isJavadoc) {
53             content.append("*");
54         }
55         content.append("\n");
56
57         content.append(addAsterixAtEachLineStart(input));
58         content.append("*/\n");
59         return content.toString();
60     }
61
62
63     public static Optional<String> loadCopyright() {
64         try (InputStream in = StringUtil.class.getResourceAsStream("/copyright.txt")) {
65             if (in != null) {
66                 return Optional.of(IOUtils.toString(in));
67             }
68         } catch (IOException e) {
69             logger.warn("Cannot load copyright.txt", e);
70         }
71         return Optional.absent();
72     }
73
74     public static String formatJavaSource(String input) {
75         Iterable<String> split = Splitter.on("\n").trimResults().split(input);
76
77         int basicIndent = 4;
78         StringBuilder sb = new StringBuilder();
79         int intends = 0, empty = 0;
80         for (String line : split) {
81             intends -= StringUtils.countMatches(line, "}");
82             if (intends < 0) {
83                 intends = 0;
84             }
85             if (line.isEmpty() == false) {
86                 sb.append(Strings.repeat(" ", basicIndent * intends));
87                 sb.append(line);
88                 sb.append("\n");
89                 empty = 0;
90             } else {
91                 empty++; // one empty line is allowed
92                 if (empty < 2) {
93                     sb.append("\n");
94                 }
95             }
96             intends += StringUtils.countMatches(line, "{");
97         }
98         return ensureEndsWithSingleNewLine(sb.toString());
99     }
100
101 }