using System;
using System.Collections.Generic;
using System.Data;
using System.Text;
using log4net;
using NHibernate.Connection;
using NHibernate.Driver;
using NHibernate.SqlCommand;
using NHibernate.SqlTypes;

namespace SocialAnimal.Persistence
{
    class NoCountDriverConnectionProvider : ConnectionProvider
    {
        private static readonly ILog log = LogManager.GetLogger(typeof(NoCountDriverConnectionProvider));

	/// <summary>
        /// Initializes a new instance of the <see cref="NoCountDriverConnectionProvider"/> class with NOCOUNT OFF specified on the connection
	/// </summary>
        public NoCountDriverConnectionProvider()
	{
	}

	/// <summary>
	/// Closes and Disposes of the <see cref="IDbConnection"/>.
	/// </summary>
	/// <param name="conn">The <see cref="IDbConnection"/> to clean up.</param>
	public override void CloseConnection(IDbConnection conn)
	{
		base.CloseConnection(conn);
		conn.Dispose();
	}

	/// <summary>
	/// Gets a new open <see cref="IDbConnection"/> through 
	/// the <see cref="NHibernate.Driver.IDriver"/>.
	/// </summary>
	/// <returns>
	/// An Open <see cref="IDbConnection"/>.
	/// </returns>
	/// <exception cref="Exception">
	/// If there is any problem creating or opening the <see cref="IDbConnection"/>.
	/// </exception>
	public override IDbConnection GetConnection()
	{
		log.Debug("Obtaining IDbConnection from Driver");
		IDbConnection conn = Driver.CreateConnection();
		conn.ConnectionString = ConnectionString;
		conn.Open();

		IDbCommand cmd = Driver.GenerateCommand(CommandType.Text, new SqlString("SET NOCOUNT OFF"), new SqlType[] {});
		using (cmd)
		{
		cmd.Connection = conn;
		cmd.ExecuteNonQuery();
		}

		return conn;
	}
    }
}