Add support for the table of contents block type

mikehutility · mikehutility · commit 3af7b984fe8b · 2021-11-06T12:00:33.000-04:00
diff --git a/Src/Notion.Client/Api/Blocks/RequestParams/BlocksUpdateParameters/UpdateBlocks/TableOfContentsUpdateBlock.cs b/Src/Notion.Client/Api/Blocks/RequestParams/BlocksUpdateParameters/UpdateBlocks/TableOfContentsUpdateBlock.cs
@@ -0,0 +1,21 @@
+﻿using Newtonsoft.Json;
+
+namespace Notion.Client
+{
+    public class TableOfContentsUpdateBlock : IUpdateBlock
+    {
+        public bool Archived { get; set; }
+
+        [JsonProperty("table_of_contents")]
+        public Data TableOfContents { get; set; }
+
+        public class Data
+        {
+        }
+
+        public TableOfContentsUpdateBlock()
+        {
+            TableOfContents = new Data();
+        }
+    }
+}
diff --git a/Src/Notion.Client/Models/Blocks/Block.cs b/Src/Notion.Client/Models/Blocks/Block.cs
@@ -22,6 +22,7 @@ namespace Notion.Client
     [JsonSubtypes.KnownSubType(typeof(NumberedListItemBlock), BlockType.NumberedListItem)]
     [JsonSubtypes.KnownSubType(typeof(ParagraphBlock), BlockType.Paragraph)]
     [JsonSubtypes.KnownSubType(typeof(PDFBlock), BlockType.PDF)]
+    [JsonSubtypes.KnownSubType(typeof(TableOfContentsBlock), BlockType.TableOfContents)]
     [JsonSubtypes.KnownSubType(typeof(ToDoBlock), BlockType.ToDo)]
     [JsonSubtypes.KnownSubType(typeof(ToggleBlock), BlockType.Toggle)]
     [JsonSubtypes.KnownSubType(typeof(VideoBlock), BlockType.Video)]
diff --git a/Src/Notion.Client/Models/Blocks/BlockType.cs b/Src/Notion.Client/Models/Blocks/BlockType.cs
@@ -67,6 +67,9 @@ public enum BlockType
         [EnumMember(Value = "audio")]
         Audio,
 
+        [EnumMember(Value = "table_of_contents")]
+        TableOfContents,
+
         [EnumMember(Value = "unsupported")]
         Unsupported
     }
diff --git a/Src/Notion.Client/Models/Blocks/TableOfContentsBlock.cs b/Src/Notion.Client/Models/Blocks/TableOfContentsBlock.cs
@@ -0,0 +1,16 @@
+﻿using Newtonsoft.Json;
+
+namespace Notion.Client
+{
+    public class TableOfContentsBlock : Block
+    {
+        public override BlockType Type => BlockType.TableOfContents;
+
+        [JsonProperty("table_of_contents")]
+        public Data TableOfContents { get; set; }
+
+        public class Data
+        {
+        }
+    }
+}
diff --git a/Test/Notion.IntegrationTests/IBlocksClientTests.cs b/Test/Notion.IntegrationTests/IBlocksClientTests.cs
@@ -43,12 +43,16 @@ public async Task AppendChildrenAsync_AppendsBlocksGivenBlocks()
                         new DividerBlock
                         {
                             Divider = new DividerBlock.Data()
+                        },
+                        new TableOfContentsBlock
+                        {
+                            TableOfContents = new TableOfContentsBlock.Data()
                         }
                     }
                 }
             );
 
-            blocks.Results.Should().HaveCount(2);
+            blocks.Results.Should().HaveCount(3);
 
             // cleanup
             await _client.Pages.UpdateAsync(page.Id, new PagesUpdateParameters
@@ -133,12 +137,16 @@ public async Task DeleteAsync_DeleteBlockWithGivenId()
                         new DividerBlock
                         {
                             Divider = new DividerBlock.Data()
-                        }
+                        },
+                        new TableOfContentsBlock
+                        {
+                            TableOfContents = new TableOfContentsBlock.Data()
+                        },
                     }
                 }
             );
 
-            blocks.Results.Should().HaveCount(1);
+            blocks.Results.Should().HaveCount(2);
 
             // cleanup
             await _client.Pages.UpdateAsync(page.Id, new PagesUpdateParameters
@@ -291,6 +299,18 @@ private static IEnumerable<object[]> BlockData()
                             .Audio.Should().BeOfType<ExternalFile>().Subject
                             .External.Url.Should().Be("https://www.soundhelix.com/examples/mp3/SoundHelix-Song-3.mp3");
                     })
+                },
+                new object[]
+                {
+                    new TableOfContentsBlock {
+                        TableOfContents = new TableOfContentsBlock.Data()
+                    },
+                    new TableOfContentsUpdateBlock(),
+                    new Action<Block>((block) =>
+                    {
+                        Assert.NotNull(block);
+                        Assert.IsType<TableOfContentsBlock>(block);
+                    })
                 }
             };
         }

Original file line number	Diff line number	Diff line change
`@@ -67,6 +67,9 @@ public enum BlockType`
`67`	`67`	`[EnumMember(Value = "audio")]`
`68`	`68`	`Audio,`
`69`	`69`
	`70`	`+ [EnumMember(Value = "table_of_contents")]`
	`71`	`+ TableOfContents,`
	`72`	`+`
`70`	`73`	`[EnumMember(Value = "unsupported")]`
`71`	`74`	`Unsupported`
`72`	`75`	`}`