package cz.frantovo.sql.vyuka.dto;

import cz.frantovo.sql.vyuka.Html;
import java.util.ArrayList;
import java.util.Collection;

/**
 * Tabulka, která je výsledkem SQL dotazu.
 * @author fiki
 */
public class Tabulka implements HtmlObjekt {

    private String[] zahlavi;
    private Collection<Object[]> hodnoty = new ArrayList<Object[]>();

    public String getHtml() {

        if (getZahlavi() == null || getHodnoty() == null || getZahlavi().length < 1) {
            return "<p>Chybná tabulka</p>";
        } else {

            StringBuffer html = new StringBuffer();

            html.append("<table>");


            html.append("<thead title=\"Chceš setřídit výsledek podle nějakého sloupce? Co takhle ORDER BY sloupec.\">");
            html.append("<tr>");
            for (String z : getZahlavi()) {
                html.append("<td>" + Html.escapuj(z) + "</td>");
            }
            html.append("</tr>");
            html.append("</thead>");


            html.append("<tbody>");
            for (Object[] hh : getHodnoty()) {
                html.append("<tr>");
                for (Object h : hh) {
                    html.append(formatujBunku(h));
                }
                html.append("</tr>");
            }
            html.append("</tbody>");

            html.append("</table>");


            return html.toString();
        }
    }

    private String formatujBunku(Object o) {
        /** TODO: podporovat i jiné typy */
        if (o instanceof Integer) {
            return "<td class=\"cislo\">" + Html.escapuj(String.valueOf(o)) + "</td>";
        } else {
            return "<td>" + Html.escapuj(String.valueOf(o)) + "</td>";
        }
    }

    public String[] getZahlavi() {
        return zahlavi;
    }

    public void setZahlavi(String[] zahlavi) {
        this.zahlavi = zahlavi;
    }

    public Collection<Object[]> getHodnoty() {
        return hodnoty;
    }

    public void setHodnoty(Collection<Object[]> hodnoty) {
        this.hodnoty = hodnoty;
    }
}