25 lines
470 B
Go
25 lines
470 B
Go
package middleware
|
|
|
|
import (
|
|
"bytes"
|
|
"github.com/PuerkitoBio/goquery"
|
|
"github.com/geziyor/geziyor/client"
|
|
"log"
|
|
)
|
|
|
|
// ParseHTML parses response if response is HTML
|
|
type ParseHTML struct {
|
|
ParseHTMLDisabled bool
|
|
}
|
|
|
|
func (p *ParseHTML) ProcessResponse(r *client.Response) {
|
|
if !p.ParseHTMLDisabled && r.IsHTML() {
|
|
doc, err := goquery.NewDocumentFromReader(bytes.NewReader(r.Body))
|
|
if err != nil {
|
|
log.Println(err.Error())
|
|
return
|
|
}
|
|
r.HTMLDoc = doc
|
|
}
|
|
}
|