87 lines
1.8 KiB
Go
87 lines
1.8 KiB
Go
// Copyright (C) 2019 Storj Labs, Inc.
|
|
// See LICENSE for copying information
|
|
|
|
package simulate
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
"regexp"
|
|
"strings"
|
|
|
|
"github.com/zeebo/errs"
|
|
monkit "gopkg.in/spacemonkeygo/monkit.v2"
|
|
|
|
"storj.io/storj/internal/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{}
|
|
|
|
// FromAddress return empty mail address
|
|
func (clicker *LinkClicker) FromAddress() post.Address {
|
|
return post.Address{}
|
|
}
|
|
|
|
// SendEmail click all links from email html parts
|
|
func (clicker *LinkClicker) SendEmail(ctx context.Context, msg *post.Message) (err error) {
|
|
defer mon.Task()(&ctx)(&err)
|
|
|
|
// dirty way to find links without pulling in a html dependency
|
|
regx := regexp.MustCompile(`href="([^\s])+"`)
|
|
// collect all links
|
|
var links []string
|
|
for _, part := range msg.Parts {
|
|
tags := findLinkTags(part.Content)
|
|
for _, tag := range tags {
|
|
href := regx.FindString(tag)
|
|
if href == "" {
|
|
continue
|
|
}
|
|
|
|
links = append(links, href[len(`href="`):len(href)-1])
|
|
}
|
|
}
|
|
// click all links
|
|
var sendError error
|
|
for _, link := range links {
|
|
response, err := http.Get(link)
|
|
if err != nil {
|
|
continue
|
|
}
|
|
sendError = errs.Combine(sendError, err, response.Body.Close())
|
|
}
|
|
|
|
return sendError
|
|
}
|
|
|
|
func findLinkTags(body string) []string {
|
|
var tags []string
|
|
Loop:
|
|
for {
|
|
stTag := strings.Index(body, "<a")
|
|
if stTag < 0 {
|
|
break Loop
|
|
}
|
|
|
|
stripped := body[stTag:]
|
|
endTag := strings.Index(stripped, "</a>")
|
|
if endTag < 0 {
|
|
break Loop
|
|
}
|
|
|
|
offset := endTag + len("</a>") + 1
|
|
body = stripped[offset:]
|
|
tags = append(tags, stripped[:offset])
|
|
}
|
|
return tags
|
|
}
|