misc/xgen/src/main/java/org/onap/aaf/misc/xgen/XGen.java

   1 /**\r
   2  * ============LICENSE_START====================================================\r
   3  * org.onap.aaf\r
   4  * ===========================================================================\r
   5  * Copyright (c) 2018 AT&T Intellectual Property. All rights reserved.\r
   6  * ===========================================================================\r
   7  * Licensed under the Apache License, Version 2.0 (the "License");\r
   8  * you may not use this file except in compliance with the License.\r
   9  * You may obtain a copy of the License at\r
  10  * \r
  11  *      http://www.apache.org/licenses/LICENSE-2.0\r
  12  * \r
  13  * Unless required by applicable law or agreed to in writing, software\r
  14  * distributed under the License is distributed on an "AS IS" BASIS,\r
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.\r
  16  * See the License for the specific language governing permissions and\r
  17  * limitations under the License.\r
  18  * ============LICENSE_END====================================================\r
  19  *\r
  20  */\r
  21 \r
  22 package org.onap.aaf.misc.xgen;\r
  23 \r
  24 import java.io.PrintWriter;\r
  25 import java.io.Writer;\r
  26 import java.util.Stack;\r
  27 \r
  28 import org.onap.aaf.misc.env.util.IndentPrintWriter;\r
  29 import org.onap.aaf.misc.env.util.StringBuilderWriter;\r
  30 \r
  31 public class XGen<RT extends XGen<RT>> {\r
  32 \r
  33         private static final int COMMENT_COLUMN = 40;\r
  34         private StringBuilder backSB = new StringBuilder();\r
  35         private Stack<Back> backStack = new Stack<Back>();\r
  36         \r
  37         protected XGen(Writer w) {\r
  38                 forward = new IndentPrintWriter(w);\r
  39         }\r
  40 \r
  41         public int pushBack(Back b) {\r
  42                 int rv = backStack.size();\r
  43                 backStack.push(b);\r
  44                 return rv;\r
  45         }\r
  46 \r
  47         public boolean pretty = false;\r
  48         protected IndentPrintWriter forward;\r
  49 \r
  50         public IndentPrintWriter getWriter() {\r
  51                 return forward;\r
  52         }\r
  53 \r
  54         protected PrintWriter back = new PrintWriter(\r
  55                                 new StringBuilderWriter(backSB));\r
  56 \r
  57         @SuppressWarnings("unchecked")\r
  58         public RT pretty() {\r
  59                 pretty = true;\r
  60                 return (RT) this;\r
  61         }\r
  62 \r
  63         protected void prettyln(PrintWriter pw) {\r
  64                 if(pretty)pw.println();\r
  65         }\r
  66 \r
  67         public RT leaf(Mark mark, String tag, String ... args) {\r
  68                 mark.spot = backStack.size();\r
  69                 return leaf(tag, args);\r
  70         }\r
  71 \r
  72         @SuppressWarnings("unchecked")\r
  73         public RT leaf(String tag, String ... attrs) {\r
  74                 forward.append('<');\r
  75                 forward.append(tag);\r
  76                 addAttrs(attrs);\r
  77                 forward.append('>');\r
  78                 back.append("</");\r
  79                 back.append(tag);\r
  80                 back.append('>');\r
  81                 backStack.push(new Back(backSB.toString(), false, true));\r
  82                 backSB.setLength(0);\r
  83                 return (RT)this;\r
  84         }\r
  85 \r
  86         public RT incr(String tag, String ... args) {\r
  87                 return incr(null, tag, false, args);\r
  88         }\r
  89 \r
  90         public RT incr(String tag, boolean oneLine, String ... args) {\r
  91                 return incr(null, tag, oneLine, args);\r
  92         }\r
  93 \r
  94         public RT incr(Mark mark) {\r
  95                 return incr(mark,mark.comment, false, new String[0]);\r
  96         }\r
  97 \r
  98         public RT incr(Mark mark, String tag, String ... attrs) {\r
  99                 return incr(mark, tag, false, attrs);\r
 100         }\r
 101 \r
 102         @SuppressWarnings("unchecked")\r
 103         public RT incr(Mark mark, String tag, boolean oneLine, String ... attrs) {\r
 104                 forward.append('<');\r
 105                 forward.append(tag);\r
 106                 addAttrs(attrs);\r
 107                 forward.append('>');\r
 108                 \r
 109                 back.append("</");\r
 110                 back.append(tag);\r
 111                 back.append('>');\r
 112         \r
 113                 if(pretty) {\r
 114                         if(mark!=null && mark.comment!=null) {\r
 115                                 int fi = forward.getIndent()*IndentPrintWriter.INDENT_MULTIPLIER;\r
 116                                 for(int i = fi+backSB.length();i<=COMMENT_COLUMN;++i) {\r
 117                                         back.append(' ');\r
 118                                 }\r
 119                                 back.append("<!-- end ");\r
 120                                 back.append(mark.comment);\r
 121                                 back.append(" -->");\r
 122                                 \r
 123                                 forward.toCol(COMMENT_COLUMN);\r
 124                                 forward.append("<!-- begin ");\r
 125                                 forward.append(mark.comment);\r
 126                                 forward.append(" -->");\r
 127                         }\r
 128                         forward.inc();\r
 129                         if(!oneLine) {\r
 130                                 forward.println();\r
 131                         }\r
 132                         back.println();\r
 133                 }\r
 134                 if(mark!=null)mark.spot = backStack.size();\r
 135                 backStack.push(new Back(backSB.toString(),true, false));\r
 136                 backSB.setLength(0);\r
 137                 return (RT)this;\r
 138         }\r
 139 \r
 140         @SuppressWarnings("unchecked")\r
 141         public RT tagOnly(String tag, String ... attrs) {\r
 142                 forward.append('<');\r
 143                 forward.append(tag);\r
 144                 addAttrs(attrs);\r
 145                 forward.append(" />");\r
 146                 if(pretty) {\r
 147                         forward.println();\r
 148                 }\r
 149                 return (RT)this;\r
 150         }\r
 151 \r
 152         @SuppressWarnings("unchecked")\r
 153         public RT text(String txt) {\r
 154                 forward.append(txt);\r
 155                 return (RT)this;\r
 156         }\r
 157         \r
 158         @SuppressWarnings("unchecked")\r
 159         public RT xml(String txt) {\r
 160                 for(int i=0; i<txt.length();++i) {\r
 161                         char c = txt.charAt(i);\r
 162                         switch(c) {\r
 163                                 case '<':\r
 164                                         forward.append("&lt;");\r
 165                                         break;\r
 166                                 case '>':\r
 167                                         forward.append("&gt;");\r
 168                                         break;\r
 169                                 case '&':\r
 170                                         forward.append("&amp;");\r
 171                                         break;\r
 172                                 default:\r
 173                                         forward.append(c);\r
 174                         }\r
 175                 }\r
 176                 return (RT)this;\r
 177         }\r
 178 \r
 179 \r
 180         @SuppressWarnings("unchecked")\r
 181         public RT textCR(int tabs, String txt) {\r
 182                 for(int i=0;i<tabs;++i) {\r
 183                         forward.append("  ");\r
 184                 }\r
 185                 forward.append(txt);\r
 186                 if(pretty)forward.println();\r
 187                 return (RT)this;\r
 188         }\r
 189 \r
 190         @SuppressWarnings("unchecked")\r
 191         public RT value() {\r
 192                 Mark mark = new Mark();\r
 193                 mark.spot = backStack.size()-1;\r
 194                 end(mark);\r
 195                 return (RT)this;\r
 196         }\r
 197 \r
 198         @SuppressWarnings("unchecked")\r
 199         public RT value(String txt) {\r
 200                 forward.append(txt);\r
 201                 Mark mark = new Mark();\r
 202                 mark.spot = backStack.size()-1;\r
 203                 end(mark);\r
 204                 return (RT)this;\r
 205         }\r
 206 \r
 207         @SuppressWarnings("unchecked")\r
 208         public RT value(String txt, int levels) {\r
 209                 forward.append(txt);\r
 210                 Mark mark = new Mark();\r
 211                 mark.spot = backStack.size()-levels;\r
 212                 end(mark);\r
 213                 return (RT)this;\r
 214         }\r
 215 \r
 216         @SuppressWarnings("unchecked")\r
 217         public RT end(Mark mark) {\r
 218                 int size = backStack.size();\r
 219                 Back c;\r
 220                 boolean println = false;\r
 221                 for(int i=mark==null?0:mark.spot;i<size;++i) {\r
 222                         c = backStack.pop();\r
 223                         if(c.dec)forward.dec();\r
 224                         forward.append(c.str);\r
 225                         println = c.cr;\r
 226                 }\r
 227                 if(pretty && println) {\r
 228                         forward.println();\r
 229                 }\r
 230                 return (RT)this;\r
 231         }\r
 232 \r
 233         public RT end() {\r
 234                 Mark mark = new Mark();\r
 235                 mark.spot=backStack.size()-1;\r
 236                 if(mark.spot<0)mark.spot=0;\r
 237                 return end(mark);\r
 238         }\r
 239 \r
 240         public RT end(int i) {\r
 241                 Mark mark = new Mark();\r
 242                 mark.spot=backStack.size()-i;\r
 243                 if(mark.spot<0)mark.spot=0;\r
 244                 return end(mark);\r
 245         }\r
 246 \r
 247         public void endAll() {\r
 248                 end(new Mark());\r
 249                 forward.flush();\r
 250         }\r
 251 \r
 252         protected void addAttrs(String[] attrs) {\r
 253                 if(attrs!=null) {\r
 254                         for(String attr : attrs) {\r
 255                                 if(attr!=null && attr.length()>0) {\r
 256                                         forward.append(' ');\r
 257                                         String[] split = attr.split("=",2);\r
 258                                         switch(split.length) {\r
 259                                                 case 0:\r
 260                                                         break;\r
 261                                                 case 1:\r
 262                                                         forward.append(split[0]);\r
 263 //                                                      forward.append("=\"\"");\r
 264                                                         break;\r
 265                                                 default:\r
 266                                                         forward.append(split[0]);\r
 267                                                         forward.append("=\"");\r
 268                                                         forward.append(split[1]);\r
 269                                                         forward.append('"');\r
 270                                                         break;\r
 271                                         }\r
 272                                 }\r
 273                         }\r
 274                 }\r
 275         }\r
 276 \r
 277         @SuppressWarnings("unchecked")\r
 278         public RT comment(String string) {\r
 279                 if(pretty) {\r
 280                         forward.print("<!--  ");\r
 281                         forward.print(string);\r
 282                         forward.println("  -->");\r
 283                 }\r
 284                 return (RT)this;\r
 285         }\r
 286 \r
 287         public void setIndent(int indent) {\r
 288                 forward.setIndent(indent);\r
 289                 forward.toIndent();\r
 290         }\r
 291 \r
 292         public int getIndent() {\r
 293                 return forward.getIndent();\r
 294         }\r
 295 \r
 296 }\r