package cz.frantovo.hrisniciSpameri.dao;

import cz.frantovo.hrisniciSpameri.dto.Cil;

import cz.frantovo.hrisniciSpameri.dto.SitovaAdresa;

import cz.frantovo.hrisniciSpameri.dto.Soucet;

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.HashMap;

import java.util.Map;

import java.util.logging.Level;

/**

 *

 * @author fiki

 */

public class SouhrnDAO extends HrisniciSuperDAO {

    private enum SQL {

        DENNI_SOUHRN,

        DENNI_SOUHRN_CIL,

        DENNI_SOUCTY,

        DENNI_SOUCTY_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) {

                    konec = zacatek;

                }

                if (cil == 0) {

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

                    ps.setDate(1, zacatek);

                    ps.setDate(2, konec);

                } else {

                    ps = db.prepareStatement(getSQL(SQL.DENNI_SOUHRN_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;

    }

    /**

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

     * @param konec konec období nebo null (pokud nás zajímá konkrétní den – což nás tady většinou nezajímá)

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

     * @return součty útoků za jednotlivé dny, seskupené podle cílů

     */

    public Map<Cil, Collection<Soucet>> getSoucty(Date zacatek, Date konec, int cil) {

        HashMap<Cil, Collection<Soucet>> soucty = new HashMap<Cil, Collection<Soucet>>();

        Connection db = getSpojeni(null);

        if (db == null) {

            return null;

        } else {

            PreparedStatement ps = null;

            ResultSet rs = null;

            try {

                if (konec == null) {

                    konec = zacatek;

                }

                if (cil == 0) {

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

                    ps.setDate(1, zacatek);

                    ps.setDate(2, konec);

                } else {

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

                    ps.setDate(1, zacatek);

                    ps.setDate(2, konec);

                    ps.setInt(3, cil);

                }

                rs = ps.executeQuery();

                while (rs.next()) {

                    Soucet s = new Soucet();

                    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);

                    if (soucty.get(c) == null) {

                        soucty.put(c, new ArrayList<Soucet>());

                    }

                    soucty.get(c).add(s);

                }

            } catch (Exception e) {

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

            } finally {

                zavri(db, ps, rs);

            }

        }

        return soucty;

    }

}