server_twilio.go 3.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110
  1. package server
  2. import (
  3. "bytes"
  4. "encoding/xml"
  5. "fmt"
  6. "heckel.io/ntfy/log"
  7. "heckel.io/ntfy/util"
  8. "io"
  9. "net/http"
  10. "net/url"
  11. "strings"
  12. )
  13. const (
  14. twilioMessageEndpoint = "Messages.json"
  15. twilioMessageFooterFormat = "This message was sent by %s via %s"
  16. twilioCallEndpoint = "Calls.json"
  17. twilioCallFormat = `
  18. <Response>
  19. <Pause length="1"/>
  20. <Say>You have a message from notify on topic %s. Message:</Say>
  21. <Pause length="1"/>
  22. <Say>%s</Say>
  23. <Pause length="1"/>
  24. <Say>End message.</Say>
  25. <Pause length="1"/>
  26. <Say>%s</Say>
  27. <Pause length="1"/>
  28. </Response>`
  29. )
  30. func (s *Server) sendSMS(v *visitor, r *http.Request, m *message, to string) {
  31. body := fmt.Sprintf("%s\n\n--\n%s", m.Message, s.messageFooter(m))
  32. data := url.Values{}
  33. data.Set("From", s.config.TwilioFromNumber)
  34. data.Set("To", to)
  35. data.Set("Body", body)
  36. s.performTwilioRequest(v, r, m, twilioMessageEndpoint, to, body, data)
  37. }
  38. func (s *Server) callPhone(v *visitor, r *http.Request, m *message, to string) {
  39. body := fmt.Sprintf(twilioCallFormat, xmlEscapeText(m.Topic), xmlEscapeText(m.Message), xmlEscapeText(s.messageFooter(m)))
  40. data := url.Values{}
  41. data.Set("From", s.config.TwilioFromNumber)
  42. data.Set("To", to)
  43. data.Set("Twiml", body)
  44. s.performTwilioRequest(v, r, m, twilioCallEndpoint, to, body, data)
  45. }
  46. func (s *Server) performTwilioRequest(v *visitor, r *http.Request, m *message, endpoint, to, body string, data url.Values) {
  47. logContext := log.Context{
  48. "twilio_from": s.config.TwilioFromNumber,
  49. "twilio_to": to,
  50. }
  51. ev := logvrm(v, r, m).Tag(tagTwilio).Fields(logContext)
  52. if ev.IsTrace() {
  53. ev.Field("twilio_body", body).Trace("Sending Twilio request")
  54. } else if ev.IsDebug() {
  55. ev.Debug("Sending Twilio request")
  56. }
  57. response, err := s.performTwilioRequestInternal(endpoint, data)
  58. if err != nil {
  59. ev.
  60. Field("twilio_body", body).
  61. Field("twilio_response", response).
  62. Err(err).
  63. Warn("Error sending Twilio request")
  64. return
  65. }
  66. if ev.IsTrace() {
  67. ev.Field("twilio_response", response).Trace("Received successful Twilio response")
  68. } else if ev.IsDebug() {
  69. ev.Debug("Received successful Twilio response")
  70. }
  71. }
  72. func (s *Server) performTwilioRequestInternal(endpoint string, data url.Values) (string, error) {
  73. requestURL := fmt.Sprintf("%s/2010-04-01/Accounts/%s/%s", s.config.TwilioBaseURL, s.config.TwilioAccount, endpoint)
  74. req, err := http.NewRequest(http.MethodPost, requestURL, strings.NewReader(data.Encode()))
  75. if err != nil {
  76. return "", err
  77. }
  78. req.Header.Set("Authorization", util.BasicAuth(s.config.TwilioAccount, s.config.TwilioAuthToken))
  79. req.Header.Add("Content-Type", "application/x-www-form-urlencoded")
  80. resp, err := http.DefaultClient.Do(req)
  81. if err != nil {
  82. return "", err
  83. }
  84. response, err := io.ReadAll(resp.Body)
  85. if err != nil {
  86. return "", err
  87. }
  88. return string(response), nil
  89. }
  90. func (s *Server) messageFooter(m *message) string {
  91. topicURL := s.config.BaseURL + "/" + m.Topic
  92. sender := m.Sender.String()
  93. if m.User != "" {
  94. sender = fmt.Sprintf("%s (%s)", m.User, m.Sender)
  95. }
  96. return fmt.Sprintf(twilioMessageFooterFormat, sender, util.ShortTopicURL(topicURL))
  97. }
  98. func xmlEscapeText(text string) string {
  99. var buf bytes.Buffer
  100. _ = xml.EscapeText(&buf, []byte(text))
  101. return buf.String()
  102. }