feat: create database if not exists

Author: PPsteven

.leetcode.example.json

@@ -5,6 +5,11 @@
     "api_key": "sk-xxx",
     "model": "gpt-3.5-turbo",
     "prompt": "您是一个算法专家，请基于下面的算法题目，给出该算法的思路和复杂度, 使用 {{ .TextLang }} 回答\nSETP1. 给出算法的归类，如递归，栈\nSETP2. 若是存在暴力解法，给出思路和复杂度\nSETP3. 给出最优解法和复杂度\nSETP4. 代码实现，使用 {{ .Lang }} 语言，代码带注释和测试样例。\n\n{{ .Problem }}"
+  },
+  "notion": {
+    "token": "secret_nDvoJWo1g2TWUqYxqnArow4w3qo2NX",
+    "database_id": "3fb81c2bf55e4fa18c66a5b0b61fa403",
+    "page_id": "59194b5733c444eca8e7c996a13db505"
   }
 }
 

cmd/sync/main.go

@@ -19,7 +19,7 @@ func Run(lc *leetcode.Leetcode, isNotion bool) {
 
 	if isNotion {
 		nc := notion.NewNotion(lc.Config.Notion.Token).
-			WithConfig("", lc.Config.Notion.DatabaseID)
+			WithConfig(lc.Config.Notion.PageID, lc.Config.Notion.DatabaseID)
 
 		err := nc.Init()
 		if err != nil {

internal/config/config.go

@@ -14,6 +14,7 @@ type GptCfg struct {
 type NotionCfg struct {
 	Token      string `json:"token" mapstructure:"token"`
 	DatabaseID string `json:"database_id" mapstructure:"database_id"`
+	PageID     string `json:"page_id" mapstructure:"page_id"`
 }
 
 type Config struct {

internal/notion/notion.go

@@ -87,7 +87,7 @@ type PageData struct {
 
 func ParsePage(page *notionapi.Page) (PageUID, *PageData) {
 	var pageUID PageUID
-	pageID := GetPageID(page.ID)
+	pageID := GetStandardID(page.ID)
 	data := make(map[string]string)
 	for name, property := range page.Properties {
 		data[name] = ParseProperty(property)
@@ -165,6 +165,22 @@ func (n *Notion) WithConfig(pageID, databaseID string) *Notion {
 }
 
 func (n *Notion) Init() error {
+	if n.DatabaseID == "" && n.PageID == "" {
+		return fmt.Errorf("both database_id and page_id are empty")
+	}
+	// 1. 未填写database_id则创建数据库
+	if n.DatabaseID == "" && n.PageID != "" {
+		db, err := n.CreateDB()
+		if err != nil {
+			return fmt.Errorf("create database failed: %v", err)
+		}
+		n.DatabaseID = notionapi.DatabaseID(GetStandardID(db.ID))
+
+		log.Printf("Create Database: %d, Please add the database_id in the config file\n", n.DatabaseID)
+		log.Printf("Visited Link: %s", fmt.Sprintf("https://www.notion.so/%s", n.DatabaseID))
+	}
+
+	// 2. 创建完成database_id则添加记录
 	if n.DatabaseID != "" {
 		records, err := n.Query()
 		if err != nil {
@@ -237,10 +253,36 @@ func (n *Notion) InsertOrUpdate(record *Record) error {
 	return n.Insert(record)
 }
 
+var EmptySelect = notionapi.Select{Options: make([]notionapi.Option, 0)}
+
+func (n *Notion) CreateDB() (db *notionapi.Database, err error) {
+	ctx := context.Background()
+
+	dbReq := &notionapi.DatabaseCreateRequest{
+		Parent: notionapi.Parent{
+			Type:   "page_id",
+			PageID: n.PageID,
+		},
+		Title: []notionapi.RichText{{Text: &notionapi.Text{Content: "Leetcode"}}},
+		Properties: notionapi.PropertyConfigs{
+			"Link":       &notionapi.URLPropertyConfig{Type: "url"},
+			"Tags":       &notionapi.MultiSelectPropertyConfig{Type: "multi_select", MultiSelect: EmptySelect},
+			"Solved":     &notionapi.SelectPropertyConfig{Type: "select", Select: EmptySelect},
+			"Difficulty": &notionapi.SelectPropertyConfig{Type: "select", Select: EmptySelect},
+			"ID":         &notionapi.RichTextPropertyConfig{Type: "rich_text"},
+			"_id":        &notionapi.RichTextPropertyConfig{Type: "rich_text"},
+			"Name":       &notionapi.TitlePropertyConfig{Type: "title"},
+		},
+		IsInline: true, // show database inline in the parent page
+	}
+
+	return n.client.Database.Create(ctx, dbReq)
+}
+
 func GetPageSig(v map[string]string) string {
 	return fmt.Sprintf("%s_%s_%s_%s_%s_%s", v["Difficulty"], v["ID"], v["Link"], v["Name"], v["Solved"], v["Tags"])
 }
 
-func GetPageID(objectID notionapi.ObjectID) string {
-	return strings.ReplaceAll(objectID.String(), "-", "")
+func GetStandardID(stringer fmt.Stringer) string {
+	return strings.ReplaceAll(stringer.String(), "-", "")
 }

internal/notion/notion_test.go

@@ -45,13 +45,20 @@ func TestNotion(t *testing.T) {
 		err := notion.Update("465b111ca3e74113a0fc382aa1d3dfa6", MetaToRecord(m))
 		assert.NoError(t, err)
 	})
+
 	t.Run("InsertOrUpdate", func(t *testing.T) {
 		_ = notion.Init()
 
 		m.Title = "test"
 		err := notion.InsertOrUpdate(MetaToRecord(m))
 		assert.NoError(t, err)
 	})
+
+	t.Run("CreateDB", func(t *testing.T) {
+		db, err := notion.CreateDB()
+		assert.NoError(t, err)
+		t.Log(db)
+	})
 }
 
 func MetaToRecord(e *meta.Meta) *Record {