🎨 支持基于文档复习闪卡 https://github.com/siyuan-note/siyuan/issues/7057

kernel/api/riff.go

@@ -27,6 +27,25 @@ import (
 	"github.com/siyuan-note/siyuan/kernel/util"
 )
 
+func getTreeRiffCards(c *gin.Context) {
+	ret := gulu.Ret.NewResult()
+	defer c.JSON(http.StatusOK, ret)
+
+	arg, ok := util.JsonArg(c, ret)
+	if !ok {
+		return
+	}
+
+	rootID := arg["rootID"].(string)
+	page := int(arg["page"].(float64))
+	blockIDs, total, pageCount := model.GetTreeFlashcards(rootID, page)
+	ret.Data = map[string]interface{}{
+		"blocks":    blockIDs,
+		"total":     total,
+		"pageCount": pageCount,
+	}
+}
+
 func getRiffCards(c *gin.Context) {
 	ret := gulu.Ret.NewResult()
 	defer c.JSON(http.StatusOK, ret)

kernel/api/router.go

@@ -314,6 +314,7 @@ func ServeAPI(ginServer *gin.Engine) {
 	ginServer.Handle("POST", "/api/riff/getTreeRiffDueCards", model.CheckAuth, getTreeRiffDueCards)
 	ginServer.Handle("POST", "/api/riff/reviewRiffCard", model.CheckAuth, reviewRiffCard)
 	ginServer.Handle("POST", "/api/riff/getRiffCards", model.CheckAuth, getRiffCards)
+	ginServer.Handle("POST", "/api/riff/getTreeRiffCards", model.CheckAuth, getTreeRiffCards)
 
 	ginServer.Handle("POST", "/api/notification/pushMsg", model.CheckAuth, pushMsg)
 	ginServer.Handle("POST", "/api/notification/pushErrMsg", model.CheckAuth, pushErrMsg)

kernel/model/flashcard.go

@@ -41,6 +41,76 @@ import (
 var Decks = map[string]*riff.Deck{}
 var deckLock = sync.Mutex{}
 
+func GetTreeFlashcards(rootID string, page int) (blocks []*Block, total, pageCount int) {
+	blocks = []*Block{}
+
+	tree, err := loadTreeByBlockID(rootID)
+	if nil != err {
+		return
+	}
+
+	treeBlockIDs := map[string]bool{}
+	ast.Walk(tree.Root, func(n *ast.Node, entering bool) ast.WalkStatus {
+		if !entering || !n.IsBlock() {
+			return ast.WalkContinue
+		}
+
+		treeBlockIDs[n.ID] = true
+		return ast.WalkContinue
+	})
+
+	var allBlockIDs []string
+	const pageSize = 20
+	deck := Decks[builtinDeckID]
+	if nil == deck {
+		return
+	}
+
+	for bID, _ := range deck.BlockCard {
+		if _, ok := treeBlockIDs[bID]; !ok {
+			continue
+		}
+
+		allBlockIDs = append(allBlockIDs, bID)
+	}
+
+	allBlockIDs = gulu.Str.RemoveDuplicatedElem(allBlockIDs)
+	sort.Strings(allBlockIDs)
+
+	start := (page - 1) * pageSize
+	end := page * pageSize
+	if start > len(allBlockIDs) {
+		start = len(allBlockIDs)
+	}
+	if end > len(allBlockIDs) {
+		end = len(allBlockIDs)
+	}
+	blockIDs := allBlockIDs[start:end]
+	total = len(allBlockIDs)
+	pageCount = int(math.Ceil(float64(total) / float64(pageSize)))
+	if 1 > len(blockIDs) {
+		blocks = []*Block{}
+		return
+	}
+
+	sqlBlocks := sql.GetBlocks(blockIDs)
+	blocks = fromSQLBlocks(&sqlBlocks, "", 36)
+	if 1 > len(blocks) {
+		blocks = []*Block{}
+		return
+	}
+
+	for i, b := range blocks {
+		if nil == b {
+			blocks[i] = &Block{
+				ID:      blockIDs[i],
+				Content: Conf.Language(180),
+			}
+		}
+	}
+	return
+}
+
 func GetFlashcards(deckID string, page int) (blocks []*Block, total, pageCount int) {
 	blocks = []*Block{}
 	var allBlockIDs []string
@@ -62,7 +132,9 @@ func GetFlashcards(deckID string, page int) (blocks []*Block, total, pageCount i
 		}
 	}
 
+	allBlockIDs = gulu.Str.RemoveDuplicatedElem(allBlockIDs)
 	sort.Strings(allBlockIDs)
+
 	start := (page - 1) * pageSize
 	end := page * pageSize
 	if start > len(allBlockIDs) {