ficdown/Ficdown.Parser/Engine/BlockHandler.cs

namespace Ficdown.Parser.Engine
{
    using System;
    using System.Collections.Generic;
    using System.Linq;
    using System.Text.RegularExpressions;
    using Model.Parser;
    using Model.Story;
    using Action = Model.Story.Action;

    public class BlockHandler : IBlockHandler
    {
        public IEnumerable<Block> ExtractBlocks(IEnumerable<string> lines)
        {
            var blocks = new List<Block>();
            Block currentBlock = null;
            foreach (var line in lines)
            {
                var match = Regex.Match(line, @"^(?<level>#{1,3})\s+(?<name>[^#].*)$");
                if (match.Success)
                {
                    if (currentBlock != null) blocks.Add(currentBlock);
                    currentBlock = new Block()
                    {
                        Type = (BlockType) match.Groups["level"].Length,
                        Name = match.Groups["name"].Value,
                        Lines = new List<string>()
                    };
                }
                else
                {
                    if (currentBlock != null) currentBlock.Lines.Add(line);
                }
            }
            if (currentBlock != null) blocks.Add(currentBlock);
            return blocks;
        }

        public Story ParseBlocks(IEnumerable<Block> blocks)
        {
            // get the story
            var storyBlock = blocks.Single(b => b.Type == BlockType.Story);
            var storyName = RegexLib.Anchors.Match(storyBlock.Name);

            string storyTarget;
            try
            {
                Utilities.ParseHref(storyName.Groups["href"].Value, out storyTarget);
            }
            catch (FormatException)
            {
                throw new FormatException(string.Format("Story href should only have target: {0}",
                    storyName.Groups["href"].Value));
            }

            if (!storyName.Success)
                throw new FormatException("Story name must link to the first scene.");

            var story = new Story
            {
                Name = storyName.Groups["text"].Value,
                Description = string.Join("\n", storyBlock.Lines).Trim(),
                Scenes = new Dictionary<string, IList<Scene>>(),
                States = new Dictionary<string, IList<Action>>()
            };

            var scenes = blocks.Where(b => b.Type == BlockType.Scene).Select(BlockToScene);
            foreach (var scene in scenes)
            {
                var key = Utilities.NormalizeString(scene.Name);
                if (!story.Scenes.ContainsKey(key)) story.Scenes.Add(key, new List<Scene>());
                story.Scenes[key].Add(scene);
            }

            if (!story.Scenes.ContainsKey(storyTarget))
                throw new FormatException(string.Format("Story targets non-existent scene: {0}", storyTarget));
            story.FirstScene = storyTarget;

            return story;
        }


        private Scene BlockToScene(Block block)
        {
            var scene = new Scene
            {
                Description = string.Join("\n", block.Lines).Trim()
            };

            var sceneName = RegexLib.Anchors.Match(block.Name);
            if (sceneName.Success)
            {
                scene.Name = sceneName.Groups["text"].Value.Trim();
                IList<string> conditions;
                try
                {
                    Utilities.ParseHref(sceneName.Groups["href"].Value, out conditions);
                }
                catch (FormatException)
                {
                    throw new FormatException(string.Format("Scene href should only have conditions: {0}", block.Name));
                }
                scene.Conditions = conditions;
            }
            else scene.Name = block.Name;

            return scene;
        }
    }
}