import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Types;
import java.util.Collection;
import java.util.HashMap;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

import javax.sql.DataSource;

import org.apache.commons.lang.ArrayUtils;
import org.apache.commons.lang.StringUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.dao.DataAccessException;
import org.springframework.jdbc.core.ResultSetExtractor;
import org.springframework.jdbc.core.RowMapper;
import org.springframework.jdbc.core.SqlParameter;
import org.springframework.jdbc.core.SqlReturnResultSet;
import org.springframework.jdbc.object.StoredProcedure;
import org.springframework.stereotype.Repository;


@Repository
public class ClientRepositoryImpl {

	@Autowired
    @Qualifier("dataSource")
    protected JdbcOperations jdbcTemplate;
	
    private GetClientHeaderProcedure getClientHeaderProcedure;

    @Autowired
    public void init(DataSource dataSource) {
        this.getClientHeaderProcedure = new GetClientHeaderProcedure(dataSource);
    }

    @SuppressWarnings("unchecked")
    @Override
    public ClientHeader getClientHeader(String userId, Long clientId) {
        Map<String, Object> results = getClientHeaderProcedure.execute(userId, clientId.intValue());
        List<Client> clientList = (List<Client>) results.get(GetClientHeaderProcedure.CLIENT_RS);
        ClientHeader clientHeader = new ClientHeader();

        clientHeader.setClient(clientList.get(0));

        List<Client> relatingClientList = (List<Client>) results
                .get(GetClientHeaderProcedure.RELATING_CLIENT_RS);
        clientHeader.setRelatingClients(relatingClientList);

        List<BasicAccount> accounts = (List<BasicAccount>) results
                .get(GetClientHeaderProcedure.ACCOUNTS_RS);
        clientHeader.setAccounts(accounts);
        clientHeader.setAccountGroups((List<AccountGroup>) results
                .get(GetClientHeaderProcedure.ACCOUNT_GROUPS_RS));

        return clientHeader;
    }

	public class ClientRowMapper implements RowMapper<Client> {

    @Override
    public Client mapRow(ResultSet rs, int rowNum) throws SQLException {
    	Client client = new Client();
        client.setId(rs.getLong("ClientProfileID"));
        client.setFirstName(rs.getString("FirstName"));
        client.setLastName(rs.getString("LastName"));
        client.setTaxId(rs.getString("TaxID"));
        return client;
    }

	}
    private static class GetClientHeaderProcedure extends StoredProcedure {

        private static final String SQL = "GetClientHeaderProcedure";
        private static final String USER_ID = "UserID";
        private static final String CLIENT_PROFILE_ID = "ClientProfileID";

        public static final String CLIENT_RS = "Client";
        public static final String ACCOUNTS_RS = "Accounts";
        public static final String RELATING_CLIENT_RS = "RelatingClient";
        public static final String ACCOUNT_GROUPS_RS = "AccountGroups";

        public GetClientHeaderByClientProfileIDProcedure(DataSource dataSource) {
            super(dataSource, SQL);
            setDataSource(dataSource);

            SqlParameter param1 = new SqlParameter(USER_ID, Types.VARCHAR);
            SqlParameter param2 = new SqlParameter(CLIENT_PROFILE_ID, Types.INTEGER);

            setParameters(new SqlParameter[] { param1, param2 });

            declareParameter(new SqlReturnResultSet(CLIENT_RS, new ClientRowMapper()));
            declareParameter(new SqlReturnResultSet(ACCOUNTS_RS, new BasicAccountRowMapper()));
            declareParameter(new SqlReturnResultSet(RELATING_CLIENT_RS, new ClientRowMapper()));
            declareParameter(new SqlReturnResultSet(ACCOUNT_GROUPS_RS, new AccountGroupRowMapper()));

            compile();
        }

        public Map<String, Object> execute(String userId, int clientProfileId) {
            Map<String, Object> inputs = new LinkedHashMap<>();
            inputs.put(USER_ID, userId);
            inputs.put(CLIENT_PROFILE_ID, clientProfileId);
            return execute(inputs);
        }
    }

}