added file name to json output

gopherguides · Feb 12, 2024 · 328ccd8 · 328ccd8
1 parent 7991606
commit 328ccd8
Show file tree

Hide file tree

Showing 4 changed files with 11 additions and 2 deletions.
diff --git a/cmd/hype/cli/testdata/encode/json/success/execute-file.json b/cmd/hype/cli/testdata/encode/json/success/execute-file.json
@@ -92,5 +92,6 @@
   "section_id": 1,
   "snippets": {},
   "title": "JSON Encoding",
-  "type": "*hype.Document"
+  "type": "*hype.Document",
+  "filename": "module.md"
 }
diff --git a/cmd/hype/cli/testdata/encode/json/success/parse-file.json b/cmd/hype/cli/testdata/encode/json/success/parse-file.json
@@ -92,5 +92,6 @@
   "section_id": 1,
   "snippets": {},
   "title": "JSON Encoding",
-  "type": "*hype.Document"
+  "type": "*hype.Document",
+  "filename": "module.md"
 }
diff --git a/document.go b/document.go
@@ -23,6 +23,8 @@ type Document struct {
 	SectionID int      `json:"section_id,omitempty"`
 	Snippets  Snippets `json:"snippets,omitempty"`
 	Title     string   `json:"title,omitempty"`
+	ID        string   `json:"id,omitempty"`
+	Filename  string   `json:"filename,omitempty"`
 }
 
 func (doc *Document) MarshalJSON() ([]byte, error) {
@@ -38,6 +40,8 @@ func (doc *Document) MarshalJSON() ([]byte, error) {
 		Snippets  Snippets `json:"snippets,omitempty"`
 		Title     string   `json:"title,omitempty"`
 		Type      string   `json:"type"`
+		ID        string   `json:"id,omitempty"`
+		Filename  string   `json:"filename,omitempty"`
 	}{
 		Type:      fmt.Sprintf("%T", doc),
 		Parser:    doc.Parser,
@@ -46,6 +50,8 @@ func (doc *Document) MarshalJSON() ([]byte, error) {
 		Snippets:  doc.Snippets,
 		Title:     doc.Title,
 		Nodes:     doc.Nodes,
+		ID:        doc.ID,
+		Filename:  doc.Filename,
 	}
 
 	return json.Marshal(x)

diff --git a/parser.go b/parser.go
@@ -422,6 +422,7 @@ func (p *Parser) newDoc() *Document {
 		Root:      p.Root,
 		SectionID: p.Section,
 		Snippets:  p.Snippets,
+		Filename:  p.fileName,
 	}
 
 	return doc