package com.rumtel.wep.modules.security.dao;

import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Types;
import java.util.List;

import javax.sql.DataSource;

import org.springframework.jdbc.core.SqlParameter;
import org.springframework.jdbc.object.MappingSqlQuery;

import com.rumtel.wep.modules.security.domain.Members;

public class ManagerDAO {
	private DataSource dataSource;

	private String table;

	private ManagerSingleQuery queryMemberByUserNameandPassword;

	private ManagerQuery queryRolesByUserNameAndPassword;

	public void setDataSource(DataSource dataSource) {
		this.dataSource = dataSource;
		// -------------------------------------------------queryMemberByUserNameAndPassword-------------------------
		String queryMemberByUserNameAndPasswordSQL = "select * from " + table
				+ " where username=? and password=?";
		queryMemberByUserNameandPassword = new ManagerSingleQuery(dataSource,
				queryMemberByUserNameAndPasswordSQL);
		queryMemberByUserNameandPassword.declareParameter(new SqlParameter(
				"username", Types.VARCHAR));
		queryMemberByUserNameandPassword.declareParameter(new SqlParameter(
				"password", Types.VARCHAR));
		queryMemberByUserNameandPassword.compile();
		// ------------------------------------------------queryRolesByUserNameAndPassword---------------------------
		String queryRolesByUserNameAndPasswordSQL = "select * from " + table
				+ " where username=? and password=?";
		queryRolesByUserNameAndPassword = new ManagerQuery(dataSource,
				queryRolesByUserNameAndPasswordSQL);
		queryRolesByUserNameAndPassword.declareParameter(new SqlParameter(
				"username", Types.VARCHAR));
		queryRolesByUserNameAndPassword.declareParameter(new SqlParameter(
				"password", Types.VARCHAR));
		queryRolesByUserNameAndPassword.compile();
	}

	// ------------------------------------------------------queryMemberByUserNameAndPassword-----------------------
	public boolean queryMemberByUserNameandPassword(String username,
			String password) {
		List mList = queryMemberByUserNameandPassword.execute(new Object[] {
				username, password });
		return mList.size() > 0 ? true : false;
	}

	// -----------------------------------------------------queryRolesByUserNameAndPassword-------------------------
	public Members queryRolesByUserNameAndPassword(String username,
			String password) {
		List mList = queryRolesByUserNameAndPassword.execute(new Object[] {
				username, password });
		return mList.size() > 0 ? (Members) mList.get(0) : null;
	}

	public class ManagerQuery extends MappingSqlQuery {
		public ManagerQuery(DataSource ds, String sql) {
			super(ds, sql);
		}

		protected Object mapRow(ResultSet rs, int rowNumber)
				throws SQLException {
			Members meber = new Members();
			meber.setId(rs.getLong("id"));
			meber.setUsername(rs.getString("username"));
			meber.setPassword(rs.getString("password"));
			meber.setTruename(rs.getString("truename"));
			meber.setState(rs.getInt("state"));
			meber.setRegisteredAt(rs.getString("registered_at"));
			meber.setRoleId(rs.getLong("role_id"));
			return meber;
		}
	}

	public class ManagerSingleQuery extends MappingSqlQuery {
		public ManagerSingleQuery(DataSource ds, String sql) {
			super(ds, sql);
		}

		protected Object mapRow(ResultSet rs, int rowNumber)
				throws SQLException {
			String results = rs.getString(1);
			return results;
		}
	}

	public void setTable(String table) {
		this.table = table;
	}
}