Orchard/src/Orchard.Web/Modules/Orchard.Recipes/Providers/Executors/RecipesStep.cs

using System;
using System.Collections.Generic;
using System.Linq;
using NHibernate;
using Orchard.ContentManagement;
using Orchard.Data;
using Orchard.Logging;
using Orchard.Recipes.Models;
using Orchard.Recipes.Services;

namespace Orchard.Recipes.Providers.Executors {
    public class RecipesStep : RecipeExecutionStep {
        private readonly IRecipeHarvester _recipeHarvester;
        private readonly IRecipeStepQueue _recipeStepQueue;
        private readonly IRepository<RecipeStepResultRecord> _recipeStepResultRecordRepository;
        private readonly ISessionLocator _sessionLocator;

        public RecipesStep(
            IRecipeHarvester recipeHarvester, 
            IRecipeStepQueue recipeStepQueue, 
            IRepository<RecipeStepResultRecord> recipeStepResultRecordRepository, 
            ISessionLocator sessionLocator) {

            _recipeHarvester = recipeHarvester;
            _recipeStepQueue = recipeStepQueue;
            _recipeStepResultRecordRepository = recipeStepResultRecordRepository;
            _sessionLocator = sessionLocator;
        }

        public override string Name { get { return "Recipes"; } }

        /*  
         <Recipes>
          <Recipe ExtensionId="Orchard.Setup" Name="Core" />
         </Recipes>
        */
        public override void Execute(RecipeExecutionContext context) {
            var recipeElements = context.RecipeStep.Step.Elements();
            var recipesDictionary = new Dictionary<string, IDictionary<string, Recipe>>();
            var session = _sessionLocator.For(typeof(RecipeStepResultRecord));

            foreach (var recipeElement in recipeElements) {
                var extensionId = recipeElement.Attr("ExtensionId");
                var recipeName = recipeElement.Attr("Name");
                var recipes = recipesDictionary.ContainsKey(extensionId) ? recipesDictionary[extensionId] : default(IDictionary<string, Recipe>);

                if (recipes == null) {
                    recipes = recipesDictionary[extensionId] = HarvestRecipes(extensionId);
                }

                var recipe = recipes.ContainsKey(recipeName) ? recipes[recipeName] : default(Recipe);

                if (recipe == null) {
                    Logger.Error(String.Format("No recipe named {0} was found for extension {1}", recipeName, extensionId));
                    continue;
                }

                EnqueueRecipe(session, context.ExecutionId, recipe);
            }
        }

        private void EnqueueRecipe(ISession session, string executionId, Recipe recipe) {
            foreach (var recipeStep in recipe.RecipeSteps) {
                _recipeStepQueue.Enqueue(executionId, recipeStep);
                _recipeStepResultRecordRepository.Create(new RecipeStepResultRecord {
                    ExecutionId = executionId,
                    RecipeName = recipe.Name,
                    StepId = recipeStep.Id,
                    StepName = recipeStep.Name
                });
            }
        }

        private IDictionary<string, Recipe> HarvestRecipes(string extensionId) {
            return _recipeHarvester.HarvestRecipes(extensionId).ToDictionary(x => x.Name);
        }
    }
}