franta-hg@12: package cz.frantovo.sql.vyuka.dto;
franta-hg@12: 
franta-hg@12: import cz.frantovo.sql.vyuka.Html;
franta-hg@12: import java.util.ArrayList;
franta-hg@12: import java.util.Collection;
franta-hg@12: 
franta-hg@12: /**
franta-hg@12:  * Tabulka, která je výsledkem SQL dotazu.
franta-hg@12:  * @author fiki
franta-hg@12:  */
franta-hg@12: public class Tabulka implements HtmlObjekt {
franta-hg@12: 
franta-hg@12:     private String[] zahlavi;
franta-hg@12:     private Collection<Object[]> hodnoty = new ArrayList<Object[]>();
franta-hg@12: 
franta-hg@12:     public String getHtml() {
franta-hg@12: 
franta-hg@12:         if (getZahlavi() == null || getHodnoty() == null || getZahlavi().length < 1) {
franta-hg@12:             return "<p>Chybná tabulka</p>";
franta-hg@12:         } else {
franta-hg@12: 
franta-hg@12:             StringBuffer html = new StringBuffer();
franta-hg@12: 
franta-hg@12:             html.append("<table>");
franta-hg@12: 
franta-hg@12: 
franta-hg@12:             html.append("<thead title=\"Chceš setřídit výsledek podle nějakého sloupce? Co takhle ORDER BY sloupec.\">");
franta-hg@12:             html.append("<tr>");
franta-hg@12:             for (String z : getZahlavi()) {
franta-hg@12:                 html.append("<td>" + Html.escapuj(z) + "</td>");
franta-hg@12:             }
franta-hg@12:             html.append("</tr>");
franta-hg@12:             html.append("</thead>");
franta-hg@12: 
franta-hg@12: 
franta-hg@12:             html.append("<tbody>");
franta-hg@12:             for (Object[] hh : getHodnoty()) {
franta-hg@13:                 html.append("<tr>");
franta-hg@12:                 for (Object h : hh) {
franta-hg@13:                     html.append(formatujBunku(h));
franta-hg@12:                 }
franta-hg@13:                 html.append("</tr>");
franta-hg@12:             }
franta-hg@12:             html.append("</tbody>");
franta-hg@12: 
franta-hg@12:             html.append("</table>");
franta-hg@12: 
franta-hg@12: 
franta-hg@12:             return html.toString();
franta-hg@12:         }
franta-hg@12:     }
franta-hg@12: 
franta-hg@13:     private String formatujBunku(Object o) {
franta-hg@13:         /** TODO: podporovat i jiné typy */
franta-hg@12:         if (o instanceof Integer) {
franta-hg@12:             return "<td class=\"cislo\">" + Html.escapuj(String.valueOf(o)) + "</td>";
franta-hg@12:         } else {
franta-hg@12:             return "<td>" + Html.escapuj(String.valueOf(o)) + "</td>";
franta-hg@12:         }
franta-hg@12:     }
franta-hg@12: 
franta-hg@12:     public String[] getZahlavi() {
franta-hg@12:         return zahlavi;
franta-hg@12:     }
franta-hg@12: 
franta-hg@12:     public void setZahlavi(String[] zahlavi) {
franta-hg@12:         this.zahlavi = zahlavi;
franta-hg@12:     }
franta-hg@12: 
franta-hg@12:     public Collection<Object[]> getHodnoty() {
franta-hg@12:         return hodnoty;
franta-hg@12:     }
franta-hg@12: }