51ebc564d9
Updates the password reset page to use the new theme. Adds new endpoint '/api/v0/auth/reset-password' for password reset. Additionally, updates the link-clicking mail simulator to only click links with a specified attribute. Otherwise, the password reset cancellation link would be clicked before the password reset link could be accessed, rendering testing impossible. Change-Id: I8fde74ef7ad980880a7bf6558e3b9ed31509a393
97 lines
2.5 KiB
Go
97 lines
2.5 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information
|
|
|
|
package simulate
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"strings"
|
|
|
|
"github.com/spacemonkeygo/monkit/v3"
|
|
"github.com/zeebo/errs"
|
|
"golang.org/x/net/html"
|
|
|
|
"storj.io/storj/private/post"
|
|
"storj.io/storj/satellite/mailservice"
|
|
)
|
|
|
|
var mon = monkit.Package()
|
|
|
|
var _ mailservice.Sender = (*LinkClicker)(nil)
|
|
|
|
// LinkClicker is mailservice.Sender that click all links from html msg parts.
|
|
//
|
|
// architecture: Service
|
|
type LinkClicker struct {
|
|
// MarkerAttribute specifies the attribute every anchor element must have in order to be clicked.
|
|
// This prevents the link clicker from clicking links that it should not (such as the password reset cancellation link).
|
|
// Leaving this field empty will make it click every link.
|
|
MarkerAttribute string
|
|
}
|
|
|
|
// NewDefaultLinkClicker returns a LinkClicker with the default marker attribute.
|
|
func NewDefaultLinkClicker() *LinkClicker {
|
|
return &LinkClicker{MarkerAttribute: "data-simulate"}
|
|
}
|
|
|
|
// FromAddress return empty mail address.
|
|
func (clicker *LinkClicker) FromAddress() post.Address {
|
|
return post.Address{}
|
|
}
|
|
|
|
// SendEmail click all links belonging to properly attributed anchors from email html parts.
|
|
func (clicker *LinkClicker) SendEmail(ctx context.Context, msg *post.Message) (err error) {
|
|
defer mon.Task()(&ctx)(&err)
|
|
|
|
var body string
|
|
for _, part := range msg.Parts {
|
|
body += part.Content
|
|
}
|
|
|
|
// click all links
|
|
var sendError error
|
|
for _, link := range clicker.FindLinks(body) {
|
|
req, err := http.NewRequestWithContext(ctx, "GET", link, nil)
|
|
if err != nil {
|
|
continue
|
|
}
|
|
response, err := http.DefaultClient.Do(req)
|
|
if err != nil {
|
|
continue
|
|
}
|
|
sendError = errs.Combine(sendError, err, response.Body.Close())
|
|
}
|
|
|
|
return sendError
|
|
}
|
|
|
|
// FindLinks returns a list of all links belonging to properly attributed anchors in the HTML body.
|
|
func (clicker *LinkClicker) FindLinks(body string) (links []string) {
|
|
tokens := html.NewTokenizer(strings.NewReader(body))
|
|
Loop:
|
|
for {
|
|
switch tokens.Next() {
|
|
case html.ErrorToken:
|
|
break Loop
|
|
case html.StartTagToken:
|
|
token := tokens.Token()
|
|
if strings.ToLower(token.Data) == "a" {
|
|
simulate := clicker.MarkerAttribute == ""
|
|
var href string
|
|
for _, attr := range token.Attr {
|
|
if strings.ToLower(attr.Key) == "href" {
|
|
href = attr.Val
|
|
} else if !simulate && attr.Key == clicker.MarkerAttribute {
|
|
simulate = true
|
|
}
|
|
}
|
|
if simulate && href != "" {
|
|
links = append(links, href)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return links
|
|
}
|