chroe: rebuild memo references

This commit is contained in:
johnnyjoy 2024-12-26 20:04:36 +08:00
parent 3b0be442de
commit 541e07af8c

View File

@ -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
}