package liquibase.dbdoc;

 import liquibase.change.Change;

 import liquibase.database.Database;

 import liquibase.database.structure.Column;

 import liquibase.database.structure.Table;

 import liquibase.database.typeconversion.TypeConverterFactory;

 import java.io.File;

 import java.io.FileWriter;

 import java.io.IOException;

 import java.util.ArrayList;

 import java.util.Arrays;

 import java.util.List;

 public class TableWriter extends HTMLWriter {

     public TableWriter(File rootOutputDir, Database database) {

         super(new File(rootOutputDir, "tables"), database);

     }

     @Override

     protected String createTitle(Object object) {

         return "Changes affecting table \"" + object.toString() + "\"";

     }

     @Override

     protected void writeCustomHTML(FileWriter fileWriter, Object object, List<Change> changes, Database database)

             throws IOException {

         writeColumns(fileWriter, ((Table) object), database);

     }

     private void writeColumns(FileWriter fileWriter, Table table, Database database) throws IOException {

         List<List<String>> cells = new ArrayList<List<String>>();

         for (Column column : table.getColumns()) {

             cells.add(Arrays.asList(TypeConverterFactory.getInstance().findTypeConverter(database)

                     .convertToDatabaseTypeString(column, database),

                     "<A HREF=\"../columns/" + table.getName().toLowerCase() + "." + column.getName().toLowerCase()

                             + ".html" + "\">" + column.getName() + "</A>"));

             // todo: add foreign key info to columns?

         }

         writeTable("Current Columns", cells, fileWriter);

     }

 }