Add db notion bidirectional syncing

src/util/notion/notion.go

@@ -3,6 +3,8 @@ package notion
 import (
 	"context"
 	"errors"
+	"fmt"
+	"log/slog"
 
 	"github.com/jomei/notionapi"
 )
@@ -17,23 +19,85 @@ func GetClient() *notionapi.Client {
 	return client
 }
 
-func GetTableRows(tableId string, numberOfRows int, startFromId string) ([]notionapi.Block, error) {
+func GetTableRows(tableId string, numberOfRows int, startFromId string) ([]notionapi.TableRowBlock, error) {
 	if client == nil {
 		return nil, errors.New("notion client not initialized")
 	}
-	block, err := client.Block.GetChildren(context.Background(), notionapi.BlockID(tableId), &notionapi.Pagination{
-		StartCursor: notionapi.Cursor(startFromId),
-		PageSize:    numberOfRows,
-	})
-	if err != nil {
-		return nil, err
+	var rows []notionapi.TableRowBlock
+	var nextNumberToGet int
+	if numberOfRows > 100 {
+		nextNumberToGet = 100
+	} else {
+		nextNumberToGet = numberOfRows
 	}
-	return block.Results, nil
+	for numberOfRows > 0 {
+		block, err := client.Block.GetChildren(context.Background(), notionapi.BlockID(tableId), &notionapi.Pagination{
+			StartCursor: notionapi.Cursor(startFromId),
+			PageSize:    nextNumberToGet,
+		})
+		if err != nil {
+			return nil, err
+		}
+		for _, block := range block.Results {
+			if block.GetType().String() == "table_row" {
+				tableRow, ok := block.(*notionapi.TableRowBlock)
+				if !ok {
+					slog.Error("Notion.GetTableRows Failed to cast block to table row")
+					return nil, errors.New("Notion.GetTableRows failed to cast block to table row")
+				}
+				rows = append(rows, *tableRow)
+			} else {
+				slog.Error(fmt.Sprintf("Block ID %s is not a table row", block.GetID()))
+				return nil, errors.New("Notion.GetAllTableRows block ID is not a table row")
+			}
+		}
+		numberOfRows -= nextNumberToGet
+		if numberOfRows > 100 {
+			nextNumberToGet = 100
+		} else {
+			nextNumberToGet = numberOfRows
+		}
+	}
+	return rows, nil
 }
 
-func WriteTableRow(content []string, tableId string, after string) error {
+func GetAllTableRows(tableId string) ([]notionapi.TableRowBlock, error) {
 	if client == nil {
-		return errors.New("notion client not initialized")
+		return nil, errors.New("notion client not initialized")
+	}
+	rows := []notionapi.TableRowBlock{}
+	nextCursor := ""
+	hasMore := true
+	for hasMore {
+		blockChildren, err := client.Block.GetChildren(context.Background(), notionapi.BlockID(tableId), &notionapi.Pagination{
+			StartCursor: notionapi.Cursor(nextCursor),
+			PageSize:    100,
+		})
+		if err != nil {
+			return nil, err
+		}
+		for _, block := range blockChildren.Results {
+			if block.GetType().String() == "table_row" {
+				tableRow, ok := block.(*notionapi.TableRowBlock)
+				if !ok {
+					slog.Error("Notion.GetAllTableRows Failed to cast block to table row")
+					return nil, errors.New("Notion.GetAllTableRows failed to cast block to table row")
+				}
+				rows = append(rows, *tableRow)
+			} else {
+				slog.Error(fmt.Sprintf("Block ID %s is not a table row", block.GetID()))
+				return nil, errors.New("Notion.GetAllTableRows block ID is not a table row")
+			}
+		}
+		nextCursor = blockChildren.NextCursor
+		hasMore = blockChildren.HasMore
+	}
+	return rows, nil
+}
+
+func WriteTableRow(content []string, tableId string, after string) (string, error) {
+	if client == nil {
+		return "", errors.New("notion client not initialized")
 	}
 	rich := [][]notionapi.RichText{}
 	for _, c := range content {
@@ -56,10 +120,10 @@ func WriteTableRow(content []string, tableId string, after string) error {
 		},
 	}
 
-	_, err := client.Block.AppendChildren(context.Background(), notionapi.BlockID(tableId), &notionapi.AppendBlockChildrenRequest{
+	res, err := client.Block.AppendChildren(context.Background(), notionapi.BlockID(tableId), &notionapi.AppendBlockChildrenRequest{
 		After:    notionapi.BlockID(after),
 		Children: []notionapi.Block{tableRow},
 	})
 
-	return err
+	return res.Results[0].GetID().String(), err
 }