feat: Add support for all block types

README.md

@@ -36,19 +36,55 @@ You can interact with the tool using the built binary:
 ./notioncli [command]
 ```
 
-Here are the available commands:
+### Task Commands (To-Do focused)
+
+- `list`: List all to-do tasks on the Notion page.
+- `add <task>`: Add a new to-do task to the Notion page.
+- `check <number>`: Mark a task as complete.
+- `uncheck <number>`: Mark a task as incomplete.
+- `delete <number>`: Delete a to-do task from the Notion page.
+
+### Block Commands (All block types)
+
+The `blocks` subcommand allows you to work with all Notion block types:
+
+```bash
+# List all blocks
+notioncli blocks list
+
+# List only specific block types
+notioncli blocks list --type heading_1
+
+# Add different block types
+notioncli blocks add "Hello world"                    # paragraph (default)
+notioncli blocks add "Section Title" -t heading_1     # heading
+notioncli blocks add "Buy milk" -t to_do              # to-do item
+notioncli blocks add "Important note" -t callout      # callout
+notioncli blocks add "" -t divider                    # divider
+
+# Delete any block by index
+notioncli blocks delete 5
+```
+
+**Supported block types:**
+- `paragraph` - Regular text
+- `heading_1`, `heading_2`, `heading_3` - Headings
+- `bulleted_list_item`, `numbered_list_item` - List items
+- `to_do` - Checkbox items
+- `toggle` - Collapsible content
+- `quote` - Block quotes
+- `callout` - Highlighted callouts
+- `divider` - Horizontal dividers
+- `code` - Code blocks
+
+### Other Commands
 
-- `list`: List all tasks on the Notion page.
-- `add`: Add a new task to the Notion page.
-- `check`: Mark a task as complete.
-- `uncheck`: Mark a task as incomplete.
-- `delete`: Delete a task from the Notion page.
 - `completion`: Generate the autocompletion script for your shell
 - `help`: Show help information.
 
 ## Known Limitations
 
-Currently, the tool only supports a single Notion page at a time and is focused on the ToDo use case.
+Currently, the tool only supports a single Notion page at a time.
 
 ## Testing
 

cmd/blocks.go

@@ -0,0 +1,178 @@
+// This code is licensed under the Apache License, Version 2.0 (the "License").
+// You may not use this file except in compliance with the License.
+// You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
+
+package cmd
+
+import (
+	"fmt"
+	"sort"
+	"strconv"
+	"strings"
+
+	"github.com/fatih/color"
+	"github.com/spf13/cobra"
+	"notioncli/utils"
+)
+
+var blockType string
+
+// blocksCmd represents the blocks command
+var blocksCmd = &cobra.Command{
+	Use:   "blocks",
+	Short: "Manage all block types on a Notion page",
+	Long: `The blocks command allows you to work with all Notion block types,
+not just to-do items. Supported block types:
+
+  paragraph, heading_1, heading_2, heading_3,
+  bulleted_list_item, numbered_list_item, to_do,
+  toggle, quote, callout, divider, code
+
+Examples:
+  notioncli blocks list              # List all blocks
+  notioncli blocks list --type to_do # List only to-do blocks
+  notioncli blocks add "Hello"       # Add a paragraph
+  notioncli blocks add "Title" -t heading_1  # Add a heading
+  notioncli blocks delete 5          # Delete block at index 5`,
+}
+
+// blocksListCmd lists all blocks
+var blocksListCmd = &cobra.Command{
+	Use:   "list",
+	Short: "List all blocks on the page",
+	Long:  `List all blocks on the Notion page with their type and content.`,
+	Run: func(cmd *cobra.Command, args []string) {
+		notionAPIKey, pageID := utils.SetAPIConfig()
+		localTimezone, err := utils.GetLocalTimeZone()
+		if err != nil {
+			color.Red("Error getting timezone: %v", err)
+			return
+		}
+
+		formatted, typeCounts, err := utils.FormatAllBlocks(
+			notionAPIKey,
+			pageID,
+			localTimezone,
+			blockType,
+		)
+		if err != nil {
+			color.Red("Error: %v", err)
+			return
+		}
+
+		if len(formatted) == 0 {
+			if blockType != "" {
+				color.Yellow("No blocks of type '%s' found.", blockType)
+			} else {
+				color.Yellow("No blocks found on this page.")
+			}
+			return
+		}
+
+		fmt.Println()
+		for _, line := range formatted {
+			fmt.Println(line)
+		}
+		fmt.Println()
+
+		// Print summary
+		var summary []string
+		for t, count := range typeCounts {
+			summary = append(summary, fmt.Sprintf("%d %s", count, t))
+		}
+		sort.Strings(summary)
+
+		total := len(formatted)
+		color.Cyan("  %d blocks: %s\n", total, strings.Join(summary, ", "))
+	},
+}
+
+// blocksAddCmd adds a new block
+var blocksAddCmd = &cobra.Command{
+	Use:   "add [text]",
+	Short: "Add a new block to the page",
+	Long: `Add a new block of the specified type. Default type is 'paragraph'.
+
+Supported types:
+  paragraph, heading_1, heading_2, heading_3,
+  bulleted_list_item, numbered_list_item, to_do,
+  toggle, quote, callout, divider, code
+
+Examples:
+  notioncli blocks add "Hello world"           # Add paragraph
+  notioncli blocks add "Section Title" -t heading_1
+  notioncli blocks add "" -t divider           # Add divider (no text needed)
+  notioncli blocks add "Buy milk" -t to_do`,
+	Args: cobra.MinimumNArgs(1),
+	Run: func(cmd *cobra.Command, args []string) {
+		text := args[0]
+
+		// Validate block type
+		if blockType == "" {
+			blockType = "paragraph"
+		}
+
+		if !utils.IsValidBlockType(blockType) {
+			color.Red("Error: unsupported block type '%s'", blockType)
+			color.Yellow("Supported types: %s", strings.Join(utils.GetSupportedBlockTypeNames(), ", "))
+			return
+		}
+
+		notionAPIKey, pageID := utils.SetAPIConfig()
+
+		err := utils.AddBlock(notionAPIKey, pageID, blockType, text)
+		if err != nil {
+			color.Red("Error adding block: %v", err)
+			return
+		}
+
+		icon := utils.SupportedBlockTypes[blockType].Icon
+		if blockType == "divider" {
+			color.Green("Added %s divider", icon)
+		} else {
+			color.Green("Added %s %s: %s", icon, blockType, text)
+		}
+	},
+}
+
+// blocksDeleteCmd deletes a block by index
+var blocksDeleteCmd = &cobra.Command{
+	Use:   "delete [number]",
+	Short: "Delete a block by its index number",
+	Long: `Delete any block from the page by its index number.
+Use 'notioncli blocks list' to see block numbers.
+
+Example:
+  notioncli blocks delete 3`,
+	Args: cobra.ExactArgs(1),
+	Run: func(cmd *cobra.Command, args []string) {
+		order, err := strconv.Atoi(args[0])
+		if err != nil {
+			color.Red("Error: '%s' is not a valid number", args[0])
+			return
+		}
+
+		notionAPIKey, pageID := utils.SetAPIConfig()
+
+		err = utils.DeleteBlock(notionAPIKey, pageID, order)
+		if err != nil {
+			color.Red("Error deleting block: %v", err)
+			return
+		}
+
+		color.Green("Deleted block %d", order)
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(blocksCmd)
+	blocksCmd.AddCommand(blocksListCmd)
+	blocksCmd.AddCommand(blocksAddCmd)
+	blocksCmd.AddCommand(blocksDeleteCmd)
+
+	// Flags for list command
+	blocksListCmd.Flags().StringVarP(&blockType, "type", "t", "", "Filter by block type")
+
+	// Flags for add command
+	blocksAddCmd.Flags().StringVarP(&blockType, "type", "t", "paragraph", "Block type to add")
+}