/*

  * Copyright 2009 The Kuali Foundation

  *

  * Licensed under the Educational Community License, Version 2.0 (the "License");

  * you may not use this file except in compliance with the License.

  * You may        obtain a copy of the License at

  *

  *         http://www.osedu.org/licenses/ECL-2.0

  *

  * Unless required by applicable law or agreed to in writing, software

  * distributed under the License is distributed on an "AS IS" BASIS,

  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

  * See the License for the specific language governing permissions and

  * limitations under the License.

  */

 package org.kuali.student.contract.writer;

 import java.io.ByteArrayOutputStream;

 import java.io.File;

 import java.io.FileNotFoundException;

 import java.io.FileOutputStream;

 import java.io.PrintStream;

 import java.util.Set;

 import java.util.TreeSet;

 import java.util.regex.Matcher;

 import java.util.regex.Pattern;

 import org.kuali.student.contract.exception.DictionaryExecutionException;

 /**

  *

  * @author nwright

  */

 public abstract class JavaClassWriter extends XmlWriter {

     private String rootDirectory;

     private String packageName;

     private String className;

     private String fileName;

     private String directory;

     private ByteArrayOutputStream body;

     private Set<String> imports;

     public JavaClassWriter(String rootDirectory, String packageName,

             String className) {

         super();

         this.body = new ByteArrayOutputStream(1000);

         this.setOut(new PrintStream(body));

         this.setIndent(0);

         this.rootDirectory = rootDirectory;

         this.packageName = packageName;

         this.className = className;

         this.fileName =

                 new JavaClassFileNameBuilder(rootDirectory, packageName, className).build();

         this.directory =

                 new JavaClassFileNameBuilder(rootDirectory, packageName, className).buildDirectory();

         this.imports = new TreeSet();

     }

     public ByteArrayOutputStream getBody() {

         return body;

     }

     public String getClassName() {

         return className;

     }

     public String getDirectory() {

         return directory;

     }

     public String getFileName() {

         return fileName;

     }

     public String getPackageName() {

         return packageName;

     }

     public String getRootDirectory() {

         return rootDirectory;

     }

     public void importsAdd(String pack) {

         this.imports.add(pack);

     }

     public void writeHeader() {

         indentPrintln("/*");

         indentPrintln(" * Copyright 2011 The Kuali Foundation");

         indentPrintln(" *");

         indentPrintln(" * Licensed under the Educational Community License, Version 2.0 (the \"License\");");

         indentPrintln(" * you may not use this file except in compliance with the License.");

         indentPrintln(" * You may        obtain a copy of the License at");

         indentPrintln(" *");

         indentPrintln(" *         http://www.osedu.org/licenses/ECL-2.0");

         indentPrintln(" *");

         indentPrintln(" * Unless required by applicable law or agreed to in writing, software");

         indentPrintln(" * distributed under the License is distributed on an \"AS IS\" BASIS,");

         indentPrintln(" * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.");

         indentPrintln(" * See the License for the specific language governing permissions and");

         indentPrintln(" * limitations under the License.");

         indentPrintln(" */");

         indentPrintln("package " + packageName + ";");

         indentPrintln("");

     }

     public void writeImports() {

         if (imports.size() == 0) {

             return;

         }

         for (String imprt : imports) {

             // exclude imports from same package

             if (imprt.startsWith(packageName)) {

                 // don't exclude imports for same package that are including nested classes

                 if (!imprt.substring(packageName.length() + 1).contains(".")) {

                     continue;

                 }

             }

             indentPrintln("import " + imprt + ";");

         }

         indentPrintln("");

     }

     public void writeJavaClassAndImportsOutToFile() {

         File dir = new File(this.directory);

         //System.out.println ("Writing java class: " + fileName + " to " + dir.getAbsolutePath ());

         if (!dir.exists()) {

             if (!dir.mkdirs()) {

                 throw new DictionaryExecutionException("Could not create directory "

                         + this.directory);

             }

         }

         try {

             PrintStream out = new PrintStream(new FileOutputStream(fileName, false));

             this.setOut(out);

         } catch (FileNotFoundException ex) {

             throw new DictionaryExecutionException(ex);

         }

         writeHeader();

         indentPrintln("");

         writeImports();

         indentPrintln("");

         indentPrintln(body.toString());

     }

     public void openBrace() {

         indentPrintln("{");

         incrementIndent();

     }

     public void closeBrace() {

         decrementIndent();

         indentPrintln("}");

     }

     public void indentPrintWrappedComment(String str) {

         Pattern pattern = Pattern.compile(".{0,79}(?:\\S(?:-| |$)|$)");

         Matcher m = pattern.matcher(str);

         while (m.find()) {

             // suppresss blank lines

             if (m.group().equals("")) {

                 continue;

             }

             indentPrint("* ");

             println(m.group());

         }

     }

 }