src/com/interrupt/bob/generator/InterfaceGenerator.java

   1 package com.interrupt.bob.generator;
   2
   3 import org.apache.log4j.Logger;
   4 import org.apache.velocity.app.Velocity;
   5 import org.apache.velocity.VelocityContext;
   6 import org.apache.velocity.Template;
   7 import org.apache.velocity.exception.ParseErrorException;
   8 import org.apache.velocity.exception.ResourceNotFoundException;
   9 import org.xml.sax.Attributes;
  10 import java.io.StringWriter;
  11 import java.io.FileWriter;
  12 import java.io.IOException;
  13
  14
  15 import org.apache.velocity.runtime.resource.loader.ClasspathResourceLoader;
  16
  17
  18 import com.interrupt.bob.base.Bob;
  19 import com.interrupt.bob.base.IBob;
  20 import com.interrupt.bob.util.StringUtil;
  21 import com.interrupt.bob.util.Templates;
  22
  23 public class InterfaceGenerator implements IGenerator {
  24
  25
  26         private Logger logger = Logger.getLogger(InterfaceGenerator.class);
  27
  28         private static InterfaceGenerator instance = null;
  29
  30     private InterfaceGenerator() {
  31     }
  32
  33     public static IGenerator getInstance() {
  34         if(instance == null) {
  35             instance = new InterfaceGenerator();
  36         }
  37         return instance;
  38     }
  39
  40     public void mergeTemplate(VelocityContext context, String tplName, StringBuilder sb) {
  41
  42                 Template tpl = null;
  43                 java.io.InputStream istream = null;
  44                 try {
  45
  46
  47                         // Option 1
  48                         ClassLoader cl = ClassLoader.getSystemClassLoader();
  49                     istream = cl.getResourceAsStream("velocity.properties");
  50
  51
  52                         // inputstream for velocity.properties
  53                         //logger.debug( "InputStream ["+istream+"]");
  54
  55
  56                         java.util.Properties props = new java.util.Properties();
  57                         props.load(istream);
  58
  59
  60                         //props.list( System.out );
  61                         //System.exit(0);
  62
  63                         ClasspathResourceLoader cpl = new ClasspathResourceLoader();
  64                         //logger.debug( "Initialised Class ["+ cpl +"]");
  65
  66                         // System Classpath
  67                         //logger.debug( "System Classpath ["+System.getProperty("java.class.path")+"]");
  68                         //logger.debug( "Loaded Class ["+cl.loadClass( "org.apache.velocity.runtime.resource.loader.ClasspathResourceLoader" )+"]");
  69                         //System.exit(0);
  70
  71
  72                     Velocity.init(props);
  73                     tpl = Velocity.getTemplate(tplName);
  74
  75                     StringWriter swriter = new StringWriter();
  76
  77                     if(tpl != null) {
  78                                 tpl.merge(context, swriter);
  79                     }
  80                     sb.append(swriter.toString());
  81                 }
  82                 catch(ResourceNotFoundException e) {
  83                     e.printStackTrace();
  84                 }
  85                 catch(ParseErrorException e) {
  86                     e.printStackTrace();
  87                 }
  88                 catch(Exception e) {
  89                     e.printStackTrace();
  90                 }
  91                 finally {
  92
  93                         try {
  94                                 istream.close();
  95                         }
  96                         catch( IOException e ) {
  97                                 e.printStackTrace();
  98                         }
  99
 100                 }
 101
 102     }
 103
 104     public void appendClassName(IBob tagdef, String visibility, String clazz,
 105         String clazzName, StringBuilder sb) {
 106
 107         VelocityContext context = new VelocityContext();
 108         context.put("visibility", visibility);
 109         context.put("clazz", clazz);
 110         context.put("clazzName", clazzName);
 111
 112         mergeTemplate(context, Templates.ICLASSDECL, sb);
 113         Template tpl = null;
 114     }
 115
 116     public void appendAttribute(String attName, StringBuilder sb) {
 117
 118         /*VelocityContext context = new VelocityContext();
 119         context.put("visibility", "private");
 120         context.put("attName", attName);
 121         mergeTemplate(context, Templates.ATTRDECL, sb);
 122         */
 123     }
 124
 125     public void appendAttributeAccessors(String attName, StringBuilder sb) {
 126
 127         VelocityContext context = new VelocityContext();
 128         context.put("attrUpper", StringUtil.upperFirstLetter(attName));
 129         context.put("attr", attName);
 130         context.put("in", "_in");
 131         mergeTemplate(context, Templates.IGETTER, sb);
 132         mergeTemplate(context, Templates.ISETTER, sb);
 133     }
 134
 135     public void appendChild(String tagName, StringBuilder sb) {
 136
 137         /*VelocityContext context = new VelocityContext();
 138         context.put("tag", tagName);
 139         mergeTemplate(context, Templates.CHILD, sb);
 140         */
 141     }
 142
 143     public void appendAdders(IBob td, StringBuilder sb) {
 144
 145         String tagName = td.getTagName();
 146         String package_ = td.getNamespace();
 147         package_ = StringUtil.namespaceToPackage(package_);
 148
 149         VelocityContext context = new VelocityContext();
 150         context.put("package_", package_);
 151         context.put("tag", tagName);
 152         context.put("tagUpper", StringUtil.upperFirstLetter(tagName));
 153         mergeTemplate(context, Templates.IADDER, sb);
 154     }
 155
 156     public void appendFinders(IBob td, StringBuilder sb) {
 157
 158         Attributes attrs = td.getAttributes();
 159                 String tagName = td.getTagName();
 160                 String package_ = td.getNamespace();
 161                 package_ = StringUtil.namespaceToPackage(package_);
 162
 163         String currentAttr = null;
 164         VelocityContext context = new VelocityContext();
 165         for(int i = 0; i  < attrs.getLength(); i++) {
 166
 167             currentAttr = attrs.getLocalName(i);
 168
 169                     context = new VelocityContext();
 170                     context.put("package_", package_);
 171                     context.put("tag", tagName);
 172                     context.put("tagUpper", StringUtil.upperFirstLetter(tagName));
 173                     context.put("attributeUpper", StringUtil.upperFirstLetter(currentAttr));
 174
 175                     mergeTemplate(context, Templates.IFINDER, sb);
 176                 }
 177
 178     }
 179
 180
 181     public void appendListers(IBob td, StringBuilder sb) {
 182
 183         Attributes attrs = td.getAttributes();
 184                 String tagName = td.getTagName();
 185                 String package_ = td.getNamespace();
 186                 package_ = StringUtil.namespaceToPackage(package_);
 187
 188         String currentAttr = null;
 189         VelocityContext context = new VelocityContext();
 190         for(int i = 0; i  < attrs.getLength(); i++) {
 191
 192             currentAttr = attrs.getLocalName(i);
 193
 194                     context = new VelocityContext();
 195                     context.put("package_", package_);
 196                     context.put("tag", tagName);
 197                     context.put("tagUpper", StringUtil.upperFirstLetter(tagName));
 198                     context.put("attributeUpper", StringUtil.upperFirstLetter(currentAttr));
 199
 200                     mergeTemplate(context, Templates.ILISTER, sb);
 201
 202                 }
 203
 204     }
 205
 206
 207     public void appendRemovers(IBob td, StringBuilder sb) {
 208
 209         Attributes attrs = td.getAttributes();
 210         String tagName = td.getTagName();
 211         String package_ = td.getNamespace();
 212         package_ = StringUtil.namespaceToPackage(package_);
 213
 214         String currentAttr = null;
 215         StringBuilder findersBuf = new StringBuilder();
 216
 217         VelocityContext context = new VelocityContext();
 218         for(int i = 0; i  < attrs.getLength(); i++) {
 219
 220             currentAttr = attrs.getLocalName(i);
 221
 222             context = new VelocityContext();
 223             context.put("package_", package_);
 224             context.put("tag", tagName);
 225             context.put("tagUpper", StringUtil.upperFirstLetter(tagName));
 226             context.put("attributeUpper", StringUtil.upperFirstLetter(currentAttr));
 227
 228             mergeTemplate(context, Templates.IREMOVER, sb);
 229         }
 230     }
 231
 232     public void appendRemoveAll(IBob td, StringBuilder sb) {
 233
 234         String tagName = td.getTagName();
 235
 236         VelocityContext context = new VelocityContext();
 237         context.put("tagUpper", StringUtil.upperFirstLetter(tagName));
 238
 239         mergeTemplate(context, Templates.IREMOVEALL, sb);
 240     }
 241
 242     public void appendMake(IBob td, StringBuilder sb) {
 243
 244     }
 245
 246     /* close off the class' braces
 247      */
 248     public void appendClose(StringBuilder sb) {
 249         sb.append("\n");
 250         sb.append("}");
 251     }
 252
 253     public void appendPackage(String nameSpace, StringBuilder sb) {
 254
 255         String package_ = StringUtil.namespaceToPackage(nameSpace);
 256         VelocityContext context = new VelocityContext();
 257         context.put("package_", package_);
 258
 259         mergeTemplate(context, Templates.PACKAGE, sb);
 260     }
 261
 262     public static void main(String[ ] args) {
 263
 264         String classVisibility = "public";
 265         String className = "MyClass";
 266
 267
 268         StringBuilder sb = new StringBuilder();
 269
 270         //(InterfaceGenerator.getInstance()).appendClassName(classVisibility, "class", className, sb);
 271
 272         // print the closing brace that ends the class file
 273         sb.append("}");
 274
 275         FileWriter fw = null;
 276
 277         try {
 278             fw = new FileWriter("MyClass.java");
 279             fw.write(sb.toString());
 280
 281             fw.close();
 282         }
 283         catch(IOException e) {
 284             e.printStackTrace(System.out);
 285         }
 286
 287     }
 288 }