package cz.frantovo.hrisniciSpameri.dao;

import cz.frantovo.hrisniciSpameri.dto.Cil;

import cz.frantovo.hrisniciSpameri.dto.SitovaAdresa;

import cz.frantovo.hrisniciSpameri.dto.Souhrn;

import java.sql.Connection;

import java.sql.Date;

import java.sql.PreparedStatement;

import java.sql.ResultSet;

import java.util.ArrayList;

import java.util.Collection;

import java.util.logging.Level;

/**

 *

 * @author fiki

 */

public class SouhrnDAO extends HrisniciSuperDAO {

    private enum SQL {

        DENNI_SOUHRN,

        DENNI_SOUHRN_CIL,

        SOUHRN_ZA_OBDOBI,

        SOUHRN_ZA_OBDOBI_CIL

    }

    /**

     * Získá z databáze souhrn útoků. Buď pro konkrétní den nebo pro dané období.

     * @param zacatek začátek období nebo konkrétní den

     * @param konec konec období nebo null (pokud nás zajímá konkrétní den)

     * @param cil id cíle nebo nula, pokud chceme všechny cíle

     * @return seznam útoků

     */

    public Collection<Souhrn> getSouhrn(Date zacatek, Date konec, int cil) {

        Collection<Souhrn> souhrn = new ArrayList<Souhrn>();

        Connection db = getSpojeni(null);

        if (db == null) {

            return null;

        } else {

            PreparedStatement ps = null;

            ResultSet rs = null;

            try {

                if (konec == null) {

                    if (cil == 0) {

                        ps = db.prepareStatement(getSQL(SQL.DENNI_SOUHRN));

                        ps.setDate(1, zacatek);

                    } else {

                        ps = db.prepareStatement(getSQL(SQL.DENNI_SOUHRN_CIL));

                        ps.setDate(1, zacatek);

                        ps.setInt(2, cil);

                    }

                } else {

                    if (cil == 0) {

                        ps = db.prepareStatement(getSQL(SQL.SOUHRN_ZA_OBDOBI));

                        ps.setDate(1, zacatek);

                        ps.setDate(2, konec);

                    } else {

                        ps = db.prepareStatement(getSQL(SQL.SOUHRN_ZA_OBDOBI_CIL));

                        ps.setDate(1, zacatek);

                        ps.setDate(2, konec);

                        ps.setInt(3, cil);

                    }

                }

                rs = ps.executeQuery();

                while (rs.next()) {

                    Souhrn s = new Souhrn();

                    s.setAdresa(new SitovaAdresa(rs.getString("ip"), rs.getString("domena")));

                    s.setDen(rs.getDate("den"));

                    s.setPocet(rs.getInt("pocet"));

                    Cil c = new Cil();

                    c.setId(rs.getInt("id"));

                    c.setNazev(rs.getString("nazev"));

                    c.setUrl(rs.getString("url"));

                    c.setSelekt(rs.getString("selekt"));

                    c.setDatabaze(rs.getString("databaze"));

                    s.setCil(c);

                    souhrn.add(s);

                }

            } catch (Exception e) {

                log.log(Level.SEVERE, "Chyba při získávání souhrnu z databáze", e);

            } finally {

                zavri(db, ps, rs);

            }

        }

        return souhrn;

    }

}