From 541e07af8c7754844b43356bf38010fe8b6951c5 Mon Sep 17 00:00:00 2001
From: johnnyjoy <yourselfhosted@gmail.com>
Date: Thu, 26 Dec 2024 20:04:36 +0800
Subject: [PATCH] chroe: rebuild memo references

---
 server/runner/memopayload/runner.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/server/runner/memopayload/runner.go b/server/runner/memopayload/runner.go
index a746e648..66756878 100644
--- a/server/runner/memopayload/runner.go
+++ b/server/runner/memopayload/runner.go
@@ -56,6 +56,7 @@ func RebuildMemoPayload(memo *store.Memo) error {
 		memo.Payload = &storepb.MemoPayload{}
 	}
 	tags := []string{}
+	references := []string{}
 	property := &storepb.MemoPayload_Property{}
 	TraverseASTNodes(nodes, func(node ast.Node) {
 		switch n := node.(type) {
@@ -73,9 +74,13 @@ func RebuildMemoPayload(memo *store.Memo) error {
 			}
 		case *ast.Code, *ast.CodeBlock:
 			property.HasCode = true
+		case *ast.EmbeddedContent:
+			// TODO: validate references.
+			references = append(references, n.ResourceName)
 		}
 	})
 	memo.Payload.Tags = tags
+	memo.Payload.References = references
 	memo.Payload.Property = property
 	return nil
 }