fixes #11

hjhee · Apr 27, 2020 · f66d916 · f66d916
1 parent 3878d6d
commit f66d916
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 1 deletion.
diff --git a/parse.go b/parse.go
@@ -55,6 +55,8 @@ func homepageParser(done <-chan struct{}, page *HTMLPage, pc *PageChannel, tmMap
 		}
 		log.Printf("[homepage] Title: %s", title)
 		tf.Title = title
+		// issue #11 add url to page content
+		tf.Url = page.URL.String()
 
 		var pageNum int64
 		if s := doc.Find("span.red").Eq(1); s.Text() == "" {

diff --git a/template.go b/template.go
@@ -89,9 +89,10 @@ func renderHTML(done <-chan struct{}, tempc <-chan *TemplateField, tmpl *templat
 					err = writeOutput(filename, func(w *bufio.Writer) error {
 						if err := tmpl.Execute(w, struct {
 							Title    string
+							Url string
 							Comments []*OutputField
 							Lzls     map[uint64]*LzlComment
-						}{Title: t.Title, Comments: t.Comments, Lzls: t.Lzls.Map}); err != nil {
+						}{Title: t.Title, Url: t.Url, Comments: t.Comments, Lzls: t.Lzls.Map}); err != nil {
 							return fmt.Errorf("error executing template %s: %v", filename, err)
 						}
 						return nil

diff --git a/template/template1.html b/template/template1.html
@@ -21,6 +21,9 @@
 </style>
 </head>
 <body>
+<h1>{{.Title}}</h1>
+<div><a href="{{.Url}}">{{.Url}}</a></div>
+<hr />
 {{range .Comments}}
 <div>
     <div>

diff --git a/type.go b/type.go
@@ -187,6 +187,7 @@ func (lzl *LzlMap) IsExist(k uint64) bool {
 // TemplateField stores all necessary information to render a HTML page
 type TemplateField struct {
 	Title     string
+	Url 	string
 	ThreadID  uint64
 	Comments  []*OutputField
 	pagesLeft int64