//------------------------------------------------------------
// Copyright (c) Microsoft Corporation.  All rights reserved.
//------------------------------------------------------------
namespace System.ServiceModel.Dispatcher
{
    using System;
    using System.ServiceModel.Description;

    [Obsolete("The WF3 types are deprecated.  Instead, please use the new WF4 types from System.Activities.*")]
    public static class DurableOperationContext
    {
        public static Guid InstanceId
        {
            get
            {
                ServiceDurableInstance durableInstance = GetInstanceContextExtension();

                return durableInstance.InstanceId;
            }
        }

        public static void AbortInstance()
        {
            ServiceDurableInstance durableInstance = GetInstanceContextExtension();

            durableInstance.AbortInstance();
        }

        public static void CompleteInstance()
        {
            ServiceDurableInstance durableInstance = GetInstanceContextExtension();

            durableInstance.MarkForCompletion();
        }

        internal static void BeginOperation()
        {
            OperationContext operationContext = OperationContext.Current;

            if (operationContext != null)
            {
                operationContext.Extensions.Add(new DurableOperationContext.IsInOperation());
            }
        }

        internal static void EndOperation()
        {
            OperationContext operationContext = OperationContext.Current;

            if (operationContext != null)
            {
                DurableOperationContext.IsInOperation isInOperation = operationContext.Extensions.Find<DurableOperationContext.IsInOperation>();

                if (isInOperation != null)
                {
                    operationContext.Extensions.Remove(isInOperation);
                }
            }
        }

        static ServiceDurableInstance GetInstanceContextExtension()
        {
            OperationContext operationContext = OperationContext.Current;

            if (operationContext == null)
            {
                throw DiagnosticUtility.ExceptionUtility.ThrowHelperError(
                    new InvalidOperationException(
                    SR2.GetString(
                    SR2.OnlyCallableFromServiceOperation,
                    typeof(DurableOperationContext).Name)));
            }

            IsInOperation isInOperation = operationContext.Extensions.Find<IsInOperation>();

            if (isInOperation == null)
            {
                throw DiagnosticUtility.ExceptionUtility.ThrowHelperError(
                    new InvalidOperationException(
                    SR2.GetString(
                    SR2.OnlyCallableWhileInOperation,
                    typeof(DurableOperationContext).Name)));
            }

            InstanceContext currentInstanceContext = operationContext.InstanceContext;

            if (currentInstanceContext == null)
            {
                throw DiagnosticUtility.ExceptionUtility.ThrowHelperError(
                    new InvalidOperationException(
                    SR2.GetString(
                    SR2.OnlyCallableFromServiceOperation,
                    typeof(DurableOperationContext).Name)));
            }

            ServiceDurableInstance durableInstance =
                currentInstanceContext.Extensions.Find<ServiceDurableInstance>();

            if (durableInstance == null)
            {
                throw DiagnosticUtility.ExceptionUtility.ThrowHelperError(
                    new InvalidOperationException(
                    SR2.GetString(
                    SR2.OnlyCallableFromDurableService,
                    typeof(DurableOperationContext).Name,
                    typeof(DurableServiceAttribute).Name)));
            }

            return durableInstance;
        }

        class IsInOperation : IExtension<OperationContext>
        {
            public void Attach(OperationContext owner)
            {
            }

            public void Detach(OperationContext owner)
            {
            }
        }
    }
}