mirror of
https://github.com/bettercap/bettercap
synced 2025-07-16 10:03:39 -07:00
fix: no more explicit call to that ugly Updated method from proxy js plugins
This commit is contained in:
parent
3c2932514a
commit
adc2dde170
7 changed files with 26 additions and 16 deletions
|
@ -1,6 +1,7 @@
|
|||
package modules
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
"strings"
|
||||
|
@ -16,8 +17,8 @@ type JSResponse struct {
|
|||
Headers string
|
||||
Body string
|
||||
|
||||
wasUpdated bool
|
||||
resp *http.Response
|
||||
refHash string
|
||||
resp *http.Response
|
||||
}
|
||||
|
||||
func NewJSResponse(res *http.Response) *JSResponse {
|
||||
|
@ -33,16 +34,31 @@ func NewJSResponse(res *http.Response) *JSResponse {
|
|||
}
|
||||
}
|
||||
|
||||
return &JSResponse{
|
||||
resp := &JSResponse{
|
||||
Status: res.StatusCode,
|
||||
ContentType: cType,
|
||||
Headers: headers,
|
||||
resp: res,
|
||||
}
|
||||
resp.UpdateHash()
|
||||
|
||||
return resp
|
||||
}
|
||||
|
||||
func (j *JSResponse) Updated() {
|
||||
j.wasUpdated = true
|
||||
func (j *JSResponse) NewHash() string {
|
||||
return fmt.Sprintf("%d.%s.%s.%s", j.Status, j.ContentType, j.Headers, j.Body)
|
||||
}
|
||||
|
||||
func (j *JSResponse) UpdateHash() {
|
||||
j.refHash = j.NewHash()
|
||||
}
|
||||
|
||||
func (j *JSResponse) WasModified() bool {
|
||||
newHash := j.NewHash()
|
||||
if newHash != j.refHash {
|
||||
return true
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
func (j *JSResponse) ToResponse(req *http.Request) (resp *http.Response) {
|
||||
|
@ -71,6 +87,7 @@ func (j *JSResponse) ReadBody() string {
|
|||
}
|
||||
|
||||
j.Body = string(raw)
|
||||
j.UpdateHash()
|
||||
|
||||
return j.Body
|
||||
}
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue