//----------------------------------------------------------------
// Copyright (c) Microsoft Corporation.  All rights reserved.
//----------------------------------------------------------------

namespace System.ServiceModel.Activities
{
    using System.Collections.Generic;
    using System.Diagnostics.CodeAnalysis;
    using System.Runtime;
    using System.Workflow.Activities;

    class WorkflowRuntimeServicesExtensionProvider
    {
        Dictionary<Type, object> services;

        public WorkflowRuntimeServicesExtensionProvider()
        {
            this.services = new Dictionary<Type, object>();
        }

        public void AddService(object service)
        {
            Fx.Assert(service != null, "This should have been checked by our caller.");

            this.services.Add(service.GetType(), service);
        }

        public void RemoveService(object service)
        {
            Fx.Assert(service != null, "This should have been checked by our caller.");

            this.services.Remove(service.GetType());
        }

        [SuppressMessage(FxCop.Category.Security, FxCop.Rule.AptcaMethodsShouldOnlyCallAptcaMethods, Justification = "APTCA related issues are not relevant because this code path is not supported in partial trust.")]
        public object GetService(Type serviceType)
        {
            object service;
            if (!this.services.TryGetValue(serviceType, out service))
            {
                object dataExchangeService;
                if (this.services.TryGetValue(typeof(ExternalDataExchangeService), out dataExchangeService))
                {
                    Fx.Assert(dataExchangeService is ExternalDataExchangeService, "Something went wrong with our housekeeping.");

                    service = ((ExternalDataExchangeService)dataExchangeService).GetService(serviceType);
                }
            }

            return service;
        }

        public T GetService<T>()
        {
            return (T)GetService(typeof(T));
        }

        internal void PopulateExtensions(WorkflowServiceHost host, string baseUri)
        {
            Fx.Assert(host != null, "WorkflowServiceHost parameter was null");

            foreach (object service in this.services.Values)
            {
                host.WorkflowExtensions.Add(service);

                ExternalDataExchangeService dataExchangeService = service as ExternalDataExchangeService;
                if (dataExchangeService != null)
                {
                    dataExchangeService.SetEnqueueMessageWrapper(new WorkflowClientDeliverMessageWrapper(baseUri));

                    foreach (object innerService in dataExchangeService.GetAllServices())
                    {
                        host.WorkflowExtensions.Add(innerService);
                    }
                }
            }
        }
    }
}