| OLD | NEW |
| 1 // Copyright 2015 The LUCI Authors. All rights reserved. | 1 // Copyright 2015 The LUCI Authors. All rights reserved. |
| 2 // Use of this source code is governed under the Apache License, Version 2.0 | 2 // Use of this source code is governed under the Apache License, Version 2.0 |
| 3 // that can be found in the LICENSE file. | 3 // that can be found in the LICENSE file. |
| 4 | 4 |
| 5 package prod | 5 package prod |
| 6 | 6 |
| 7 import ( | 7 import ( |
| 8 gae_mail "github.com/luci/gae/service/mail" | 8 gae_mail "github.com/luci/gae/service/mail" |
| 9 "golang.org/x/net/context" | 9 "golang.org/x/net/context" |
| 10 "google.golang.org/appengine/mail" | 10 "google.golang.org/appengine/mail" |
| 11 ) | 11 ) |
| 12 | 12 |
| 13 // useMail adds a mail service implementation to context, accessible | 13 // useMail adds a mail service implementation to context, accessible |
| 14 // by "github.com/luci/gae/service/mail".Raw(c) or the exported mail service | 14 // by "github.com/luci/gae/service/mail".Raw(c) or the exported mail service |
| 15 // methods. | 15 // methods. |
| 16 func useMail(c context.Context) context.Context { | 16 func useMail(c context.Context) context.Context { |
| 17 return gae_mail.SetFactory(c, func(ci context.Context) gae_mail.RawInter
face { | 17 return gae_mail.SetFactory(c, func(ci context.Context) gae_mail.RawInter
face { |
| 18 » » return mailImpl{AEContext(ci)} | 18 » » return mailImpl{getAEContext(ci)} |
| 19 }) | 19 }) |
| 20 } | 20 } |
| 21 | 21 |
| 22 type mailImpl struct { | 22 type mailImpl struct { |
| 23 aeCtx context.Context | 23 aeCtx context.Context |
| 24 } | 24 } |
| 25 | 25 |
| 26 func (m mailImpl) Send(msg *gae_mail.Message) error { | 26 func (m mailImpl) Send(msg *gae_mail.Message) error { |
| 27 return mail.Send(m.aeCtx, msg.ToSDKMessage()) | 27 return mail.Send(m.aeCtx, msg.ToSDKMessage()) |
| 28 } | 28 } |
| 29 | 29 |
| 30 func (m mailImpl) SendToAdmins(msg *gae_mail.Message) error { | 30 func (m mailImpl) SendToAdmins(msg *gae_mail.Message) error { |
| 31 return mail.Send(m.aeCtx, msg.ToSDKMessage()) | 31 return mail.Send(m.aeCtx, msg.ToSDKMessage()) |
| 32 } | 32 } |
| 33 | 33 |
| 34 func (m mailImpl) GetTestable() gae_mail.Testable { return nil } | 34 func (m mailImpl) GetTestable() gae_mail.Testable { return nil } |
| OLD | NEW |