Merge pull request #23 from gopherguides/filename-for-doc-json

Filename for doc json
gopherguides · Feb 12, 2024 · e097409 · e097409
2 parents ba89818 + c70552d
commit e097409
Show file tree

Hide file tree

Showing 4 changed files with 8 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
@@ -93,5 +93,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
@@ -93,5 +93,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
@@ -24,6 +24,7 @@ type Document struct {
 	SectionID int      `json:"section_id,omitempty"`
 	Snippets  Snippets `json:"snippets,omitempty"`
 	Title     string   `json:"title,omitempty"`
+	Filename  string   `json:"filename,omitempty"`
 }
 
 func (doc *Document) MarshalJSON() ([]byte, error) {
@@ -40,6 +41,7 @@ func (doc *Document) MarshalJSON() ([]byte, error) {
 		Snippets  Snippets `json:"snippets,omitempty"`
 		Title     string   `json:"title,omitempty"`
 		Type      string   `json:"type"`
+		Filename  string   `json:"filename,omitempty"`
 	}{
 		Type:      fmt.Sprintf("%T", doc),
 		Parser:    doc.Parser,
@@ -49,6 +51,7 @@ func (doc *Document) MarshalJSON() ([]byte, error) {
 		Title:     doc.Title,
 		Nodes:     doc.Nodes,
 		ID:        doc.ID,
+		Filename:  doc.Filename,
 	}
 
 	return json.Marshal(x)

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