Browse Source

Add {{.BaseURL}} template tag to provide the root URL for static files. Fixes #1182

1205-drop-campaigns v0.7.0
Jordan Wright 2 years ago
parent
commit
15e57b6cd0
2 changed files with 58 additions and 3 deletions
  1. +11
    -3
      models/template_context.go
  2. +47
    -0
      models/template_context_test.go

+ 11
- 3
models/template_context.go View File

@ -23,6 +23,7 @@ type PhishingTemplateContext struct {
Tracker string
TrackingURL string
RId string
BaseURL string
BaseRecipient
}
@ -37,22 +38,29 @@ func NewPhishingTemplateContext(ctx TemplateContext, r BaseRecipient, rid string
if fn == "" {
fn = f.Address
}
baseURL, err := ExecuteTemplate(ctx.getBaseURL(), r)
templateURL, err := ExecuteTemplate(ctx.getBaseURL(), r)
if err != nil {
return PhishingTemplateContext{}, err
}
phishURL, _ := url.Parse(baseURL)
// For the base URL, we'll reset the the path and the query
// This will create a URL in the form of http://example.com
baseURL, _ := url.Parse(templateURL)
baseURL.Path = ""
baseURL.RawQuery = ""
phishURL, _ := url.Parse(templateURL)
q := phishURL.Query()
q.Set(RecipientParameter, rid)
phishURL.RawQuery = q.Encode()
trackingURL, _ := url.Parse(baseURL)
trackingURL, _ := url.Parse(templateURL)
trackingURL.Path = path.Join(trackingURL.Path, "/track")
trackingURL.RawQuery = q.Encode()
return PhishingTemplateContext{
BaseRecipient: r,
BaseURL: baseURL.String(),
URL: phishURL.String(),
TrackingURL: trackingURL.String(),
Tracker: "<img alt='' style='display: none' src='" + trackingURL.String() + "'/>",

+ 47
- 0
models/template_context_test.go View File

@ -0,0 +1,47 @@
package models
import (
"fmt"
check "gopkg.in/check.v1"
)
type mockTemplateContext struct {
URL string
FromAddress string
}
func (m mockTemplateContext) getFromAddress() string {
return m.FromAddress
}
func (m mockTemplateContext) getBaseURL() string {
return m.URL
}
func (s *ModelsSuite) TestNewTemplateContext(c *check.C) {
r := Result{
BaseRecipient: BaseRecipient{
FirstName: "Foo",
LastName: "Bar",
Email: "foo@bar.com",
},
RId: "1234567",
}
ctx := mockTemplateContext{
URL: "http://example.com",
FromAddress: "From Address <from@example.com>",
}
expected := PhishingTemplateContext{
URL: fmt.Sprintf("%s?rid=%s", ctx.URL, r.RId),
BaseURL: ctx.URL,
BaseRecipient: r.BaseRecipient,
TrackingURL: fmt.Sprintf("%s/track?rid=%s", ctx.URL, r.RId),
From: "From Address",
RId: r.RId,
}
expected.Tracker = "<img alt='' style='display: none' src='" + expected.TrackingURL + "'/>"
got, err := NewPhishingTemplateContext(ctx, r.BaseRecipient, r.RId)
c.Assert(err, check.Equals, nil)
c.Assert(got, check.DeepEquals, expected)
}

Loading…
Cancel
Save