package dao;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.List;

import conn.DBConnection;
import domain.ArticleKind;


public class ArticleTypeDao {
    public List<ArticleKind> getArticleType(){
    	Connection conn = DBConnection.getConn();
    	Statement stmt = null;
    	ResultSet rs = null;
    	
    	List<ArticleKind> aklist = new ArrayList<ArticleKind>();
    	String sql = "SELECT ak.title FROM webpk.articlekind ak ";
    	ArticleKind ak = null;
    	
    	try{
    		stmt = conn.createStatement();
    		rs = stmt.executeQuery(sql);
    		
    		while(rs.next()){
    			ak = new ArticleKind();
    			ak.setTitle(rs.getString("title"));
    			aklist.add(ak);
    		}
    	}catch(Exception e){
    		e.printStackTrace();
    	}finally{
    		DBConnection.closeConn();
    	}
    	return aklist;
    }
    public ArticleKind getArticleTypeID(String title){
    	Connection conn = DBConnection.getConn();
    	PreparedStatement pstmt = null;
    	ResultSet rs = null;
    	
    	ArticleKind ak = new ArticleKind();
    	String sql = "SELECT ak.kindid FROM webpk.articlekind ak WHERE ak.title=?";
    	
    	try{
    		pstmt = conn.prepareStatement(sql);
    		pstmt.setString(1, title);
    		
    		rs = pstmt.executeQuery();
    		if(rs.next()){
    			ak.setKindid(rs.getInt("kindid"));
    		}
    	}catch(Exception e){
    		e.printStackTrace();
    	}finally{
    		DBConnection.closeConn();
    	}
    	return ak;
    }
    
    public int addArticleType(String title){
    	Connection conn = DBConnection.getConn();
    	PreparedStatement pstmt = null;
    	ResultSet rs = null;
    	
    	int count = 0;
    	String sql = "INSERT INTO webpk.articlekind VALUES(null,?,null,null)";
    	
    	try{
    	    pstmt = conn.prepareStatement(sql);
    	    pstmt.setString(1, title);
    	    count = pstmt.executeUpdate();
    	}catch(Exception e){
    		e.printStackTrace();    		
    	}finally{
    	    DBConnection.closeConn();
    	}
    	
    	return count;
    }
}