// Copyright (c) Microsoft Open Technologies, Inc. All rights reserved. See License.txt in the project root for license information.

namespace System.Data.Entity.Infrastructure.Interception
{
    using System.ComponentModel;
    using System.Data.Common;
    using System.Data.Entity.Core.Objects;
    using System.Data.Entity.Utilities;
    using System.Diagnostics.CodeAnalysis;

    /// <summary>
    /// Represents contextual information associated with calls to <see cref="DbTransaction"/> that don't return any results.
    /// </summary>
    public class DbTransactionInterceptionContext : MutableInterceptionContext
    {
        private DbConnection _connection;

        /// <summary>
        /// Constructs a new <see cref="DbTransactionInterceptionContext" /> with no state.
        /// </summary>
        public DbTransactionInterceptionContext()
        {
        }

        /// <summary>
        /// Creates a new <see cref="DbTransactionInterceptionContext" /> by copying immutable state from the given
        /// interception context. Also see <see cref="Clone" />
        /// </summary>
        /// <param name="copyFrom">The context from which to copy state.</param>
        public DbTransactionInterceptionContext(DbInterceptionContext copyFrom)
            : base(copyFrom)
        {
            var transactionInterceptionContext = copyFrom as DbTransactionInterceptionContext;
            if (transactionInterceptionContext != null)
            {
                _connection = transactionInterceptionContext.Connection;
            }

            Check.NotNull(copyFrom, "copyFrom");
        }

        /// <summary>
        /// The connection on which the transaction was started
        /// </summary>
        public DbConnection Connection
        {
            get { return _connection; }
        }

        /// <summary>
        /// Creates a new <see cref="DbTransactionInterceptionContext" /> that contains all the contextual information in this
        /// interception context with the addition of the given <see cref="DbConnection" />.
        /// </summary>
        /// <param name="connection">The connection on which the transaction was started.</param>
        /// <returns>A new interception context that also contains the connection on which the transaction was started.</returns>
        public DbTransactionInterceptionContext WithConnection(DbConnection connection)
        {
            Check.NotNull(connection, "connection");

            var transactionInterceptionContext = TypedClone();
            transactionInterceptionContext._connection = connection;

            return transactionInterceptionContext;
        }

        /// <summary>
        /// Creates a new <see cref="DbTransactionInterceptionContext" /> that contains all the contextual information in this
        /// interception context together with the <see cref="DbInterceptionContext.IsAsync" /> flag set to true.
        /// </summary>
        /// <returns>A new interception context associated with the async flag set.</returns>
        public new DbTransactionInterceptionContext AsAsync()
        {
            return (DbTransactionInterceptionContext)base.AsAsync();
        }

        /// <summary>
        /// Creates a new <see cref="DbTransactionInterceptionContext" /> that contains all the contextual information in this
        /// interception context with the addition of the given <see cref="ObjectContext" />.
        /// </summary>
        /// <param name="context">The context to associate.</param>
        /// <returns>A new interception context associated with the given context.</returns>
        public new DbTransactionInterceptionContext WithDbContext(DbContext context)
        {
            Check.NotNull(context, "context");

            return (DbTransactionInterceptionContext)base.WithDbContext(context);
        }

        /// <summary>
        /// Creates a new <see cref="DbTransactionInterceptionContext" /> that contains all the contextual information in this
        /// interception context with the addition of the given <see cref="ObjectContext" />.
        /// </summary>
        /// <param name="context">The context to associate.</param>
        /// <returns>A new interception context associated with the given context.</returns>
        public new DbTransactionInterceptionContext WithObjectContext(ObjectContext context)
        {
            Check.NotNull(context, "context");

            return (DbTransactionInterceptionContext)base.WithObjectContext(context);
        }

        private DbTransactionInterceptionContext TypedClone()
        {
            return (DbTransactionInterceptionContext)Clone();
        }

        /// <inheritdoc />
        protected override DbInterceptionContext Clone()
        {
            return new DbTransactionInterceptionContext(this);
        }

        /// <inheritdoc />
        [EditorBrowsable(EditorBrowsableState.Never)]
        public override string ToString()
        {
            return base.ToString();
        }

        /// <inheritdoc />
        [EditorBrowsable(EditorBrowsableState.Never)]
        public override bool Equals(object obj)
        {
            return base.Equals(obj);
        }

        /// <inheritdoc />
        [EditorBrowsable(EditorBrowsableState.Never)]
        public override int GetHashCode()
        {
            return base.GetHashCode();
        }

        /// <inheritdoc />
        [SuppressMessage("Microsoft.Design", "CA1024:UsePropertiesWhereAppropriate")]
        [EditorBrowsable(EditorBrowsableState.Never)]
        public new Type GetType()
        {
            return base.GetType();
        }
    }
}
