Commit 541e07af authored by johnnyjoy's avatar johnnyjoy

chroe: rebuild memo references

parent 3b0be442
...@@ -56,6 +56,7 @@ func RebuildMemoPayload(memo *store.Memo) error { ...@@ -56,6 +56,7 @@ func RebuildMemoPayload(memo *store.Memo) error {
memo.Payload = &storepb.MemoPayload{} memo.Payload = &storepb.MemoPayload{}
} }
tags := []string{} tags := []string{}
references := []string{}
property := &storepb.MemoPayload_Property{} property := &storepb.MemoPayload_Property{}
TraverseASTNodes(nodes, func(node ast.Node) { TraverseASTNodes(nodes, func(node ast.Node) {
switch n := node.(type) { switch n := node.(type) {
...@@ -73,9 +74,13 @@ func RebuildMemoPayload(memo *store.Memo) error { ...@@ -73,9 +74,13 @@ func RebuildMemoPayload(memo *store.Memo) error {
} }
case *ast.Code, *ast.CodeBlock: case *ast.Code, *ast.CodeBlock:
property.HasCode = true property.HasCode = true
case *ast.EmbeddedContent:
// TODO: validate references.
references = append(references, n.ResourceName)
} }
}) })
memo.Payload.Tags = tags memo.Payload.Tags = tags
memo.Payload.References = references
memo.Payload.Property = property memo.Payload.Property = property
return nil return nil
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment