available Blocks restrictable with BlockItemTypeAttribute for Content…

core/Piranha.AttributeBuilder/ContentTypeBuilder.cs

@@ -312,7 +312,7 @@ private ContentType GetContentType(Type type)
                     throw new ArgumentException($"[{ type.Name }] Id and Title is mandatory for content types.");
                 }
 
-                return new ContentType
+                var contentType= new ContentType
                 {
                     Id = attr.Id,
                     CLRType = type.GetTypeInfo().AssemblyQualifiedName,
@@ -326,6 +326,14 @@ private ContentType GetContentType(Type type)
                     CustomEditors = GetEditors(type),
                     Regions = GetRegions(type)
                 };
+                // Add block types
+                var blockTypes = type.GetCustomAttributes<BlockItemTypeAttribute>();
+                foreach (var blockType in blockTypes)
+                {
+                    contentType.BlockItemTypes.Add(blockType.Type.FullName);
+                }
+
+                return contentType;
             }
             return null;
         }

core/Piranha.Manager/Controllers/ContentApiController.cs

@@ -68,6 +68,20 @@ public BlockListModel GetBlockTypesForPost(string postType, string parentType =
             return _contentType.GetPostBlockTypes(postType, parentType);
         }
 
+        /// <summary>
+        /// Gets the currently available block types for the
+        /// specified content type.
+        /// </summary>
+        /// <param name="contentType">The content type id</param>
+        /// <param name="parentType">The optional parent group type</param>
+        /// <returns>The block list model</returns>
+        [Route("blocktypes/content/{contentType}/{parentType?}")]
+        [HttpGet]
+        public BlockListModel GetBlockTypesForContentType(string contentType, string parentType = null)
+        {
+            return _contentType.GetContentTypeBlockTypes(contentType, parentType);
+        }
+
         /// <summary>
         /// Gets the currently available block types.
         /// </summary>

core/Piranha.Manager/Services/ContentTypeService.cs

@@ -85,6 +85,32 @@ public BlockListModel GetPostBlockTypes(string postType, string parentType = nul
             return model;
         }
 
+        /// <summary>
+        /// Gets the currently available block types for the
+        /// specified content type.
+        /// </summary>
+        /// <param name="contentType">The content type id</param>
+        /// <param name="parentType">The optional parent group type</param>
+        /// <returns>The block list model</returns>
+        public BlockListModel GetContentTypeBlockTypes(string contentType, string parentType = null)
+        {
+            var type = App.ContentTypes.GetById(contentType);
+            var model = GetBlockTypes(parentType);
+
+            if (type != null && type.BlockItemTypes.Count > 0)
+            {
+                // First remove all block types that are not allowed
+                foreach (var category in model.Categories)
+                {
+                    category.Items = category.Items.Where(i => type.BlockItemTypes.Contains(i.Type)).ToList();
+                }
+
+                // Secondly remove all empty categories
+                model.Categories = model.Categories.Where(c => c.Items.Count > 0).ToList();
+            }
+            return model;
+        }
+
         /// <summary>
         /// Gets the currently available block types.
         /// </summary>