package br.mia.test.model.dao;

 import java.io.IOException;

 import java.sql.Connection;

 import java.sql.PreparedStatement;

 import java.sql.ResultSet;

 import java.sql.SQLException;

 import java.util.ArrayList;

 import br.mia.test.seguranca.Vulnerabilidade;

 import br.mia.test.seguranca.VulnerabilidadeValor;

 public class DAOSecurity {

         public void updateIDVulnerabilidadeValue(int id1, int id2)

                         throws InstantiationException, IllegalAccessException,

                         ClassNotFoundException, SQLException, IOException {

                 Connection con = Singleton.getConnection();

                 PreparedStatement st = con

                                 .prepareStatement("update vulnerabilidades set id=0 where id=?");

                 st.setInt(1, id1);

                 st.executeUpdate();

                 st = con

                                 .prepareStatement("update vulnerabilidades set id=? where id=?");

                 st.setInt(1, id1);

                 st.setInt(2, id2);

                 st.executeUpdate();

                 st = con

                                 .prepareStatement("update vulnerabilidades set id=? where id=0");

                 st.setInt(1, id2);

                 st.executeUpdate();

         }

         public void delete(Vulnerabilidade vulnerabilidade)

                         throws InstantiationException, IllegalAccessException,

                         ClassNotFoundException, SQLException, IOException {

                 Connection con = Singleton.getConnection();

                 PreparedStatement st = con

                                 .prepareStatement("delete from tipovulnerabilidade where id=?");

                 st.setInt(1, vulnerabilidade.getId());

                 st.executeUpdate();

         }

         public void deleteValor(VulnerabilidadeValor vulnerabilidadeValor)

                         throws InstantiationException, IllegalAccessException,

                         ClassNotFoundException, SQLException, IOException {

                 Connection con = Singleton.getConnection();

                 PreparedStatement st = con

                                 .prepareStatement("delete from vulnerabilidades where id=?");

                 st.setInt(1, vulnerabilidadeValor.getId());

                 st.executeUpdate();

         }

         public Vulnerabilidade update(Vulnerabilidade vulnerabilidade)

                         throws SQLException, InstantiationException,

                         IllegalAccessException, ClassNotFoundException,

                         IOException {

                 Connection con = Singleton.getConnection();

                 if (vulnerabilidade.getId() == 0) {

                         vulnerabilidade = this.insert(vulnerabilidade);

                 } else {

                         PreparedStatement st = con

                                         .prepareStatement("update tipovulnerabilidade  set descricao=?,card=? where id=?");

                         st.setString(1, vulnerabilidade.getName());

                         st.setString(2, vulnerabilidade.getCard());

                         st.setInt(3, vulnerabilidade.getId());

                         st.executeUpdate();

                 }

                 return vulnerabilidade;

         }

         public VulnerabilidadeValor updateValor(

                         VulnerabilidadeValor vulnerabilidadeValor) throws SQLException,

                         InstantiationException, IllegalAccessException,

                         ClassNotFoundException, IOException {

                 Connection con = Singleton.getConnection();

                 if (vulnerabilidadeValor.getId() != 0) {

                         PreparedStatement st = con

                                         .prepareStatement("update vulnerabilidades  set descricao=?,observacao=? where id=?");

                         st.setString(1, vulnerabilidadeValor.getValor().toString());

                         st.setString(2, vulnerabilidadeValor.getObservacao());

                         st.setInt(3, vulnerabilidadeValor.getId());

                         st.executeUpdate();

                 }

                 return vulnerabilidadeValor;

         }

         public ArrayList<Vulnerabilidade> selectAll() throws InstantiationException,

                         IllegalAccessException, ClassNotFoundException, SQLException,

                         IOException {

                 Connection con = Singleton.getConnection();

                 ArrayList<Vulnerabilidade> arrays = new ArrayList<Vulnerabilidade>();

                 PreparedStatement st1 = con

                                 .prepareStatement("Select descricao,card,id from tipovulnerabilidade ");

                 ResultSet rs1 = st1.executeQuery();

                 while (rs1.next()) {

                         String descricao = rs1.getString(1);

                         String card = rs1.getString(2);

                         int id = rs1.getInt(3);

                         Vulnerabilidade vulnerabilidade = new Vulnerabilidade();

                         vulnerabilidade.setName(descricao);

                         vulnerabilidade.setCard(card);

                         vulnerabilidade.setId(id);

                         ArrayList<VulnerabilidadeValor> valores = this.selectAllValores(vulnerabilidade);

                         vulnerabilidade.getValues().addAll(valores);

                         arrays.add(vulnerabilidade);

                 }

                 return arrays;

         }

         public ArrayList<VulnerabilidadeValor> selectAllValores(Vulnerabilidade vulnerabilidade)

                         throws InstantiationException, IllegalAccessException,

                         ClassNotFoundException, SQLException, IOException {

                 Connection con = Singleton.getConnection();

                 ArrayList<VulnerabilidadeValor> arrays = new ArrayList<VulnerabilidadeValor>();

                 PreparedStatement st1 = con

                                 .prepareStatement("Select descricao,observacao,id from vulnerabilidades where idtipo=?");

                 st1.setInt(1, vulnerabilidade.getId());

                 ResultSet rs1 = st1.executeQuery();

                 while (rs1.next()) {

                         String descricao = rs1.getString(1);

                         String observacao = rs1.getString(2);

                         int id = rs1.getInt(3);

                         VulnerabilidadeValor vulnerabilidadeValor = new VulnerabilidadeValor();

                         vulnerabilidadeValor.setValor(descricao);

                         vulnerabilidadeValor.setObservacao(observacao);

                         vulnerabilidadeValor.setId(id);

                         arrays.add(vulnerabilidadeValor);

                 }

                 return arrays;

         }

         public VulnerabilidadeValor insertValor(Vulnerabilidade vulnerabilidade,

                         VulnerabilidadeValor vulnerabilidadeValor) throws SQLException,

                         InstantiationException, IllegalAccessException,

                         ClassNotFoundException, IOException {

                 Connection con = Singleton.getConnection();

                 if (vulnerabilidadeValor.getId() == 0) {

                         PreparedStatement st = con

                                         .prepareStatement("Insert into vulnerabilidades(descricao,observacao,idtipo) values (?,?,?)");

                         st.setString(1, vulnerabilidadeValor.getValor().toString());

                         st.setString(2, vulnerabilidadeValor.getObservacao());

                         st.setInt(3, vulnerabilidade.getId());

                         st.executeUpdate();

                         st = con

                                         .prepareStatement("select id from vulnerabilidades where descricao=? and observacao=? and idtipo=?");

                         st.setString(1, vulnerabilidadeValor.getValor().toString());

                         st.setString(2, vulnerabilidadeValor.getObservacao());

                         st.setInt(3, vulnerabilidade.getId());

                         ResultSet rs = st.executeQuery();

                         rs.next();

                         int id = rs.getInt(1);

                         vulnerabilidadeValor.setId(id);

                 }

                 return vulnerabilidadeValor;

         }

         public Vulnerabilidade insert(Vulnerabilidade vulnerabilidade)

                         throws SQLException, InstantiationException,

                         IllegalAccessException, ClassNotFoundException,

                         IOException {

                 Connection con = Singleton.getConnection();

                 if (vulnerabilidade.getId() == 0) {

                         PreparedStatement st = con

                                         .prepareStatement("Insert into tipovulnerabilidade(descricao,card) values (?,?)");

                         st.setString(1, vulnerabilidade.getName());

                         st.setString(2, vulnerabilidade.getCard());

                         st.executeUpdate();

                         st = con

                                         .prepareStatement("select id from tipovulnerabilidade where descricao=? and card=?");

                         st.setString(1, vulnerabilidade.getName());

                         st.setString(2, vulnerabilidade.getCard());

                         ResultSet rs = st.executeQuery();

                         rs.next();

                         int id = rs.getInt(1);

                         vulnerabilidade.setId(id);

                 }

                 return vulnerabilidade;

         }

 }