smsgate.go

package main

import (
	"encoding/xml"
	"errors"
	"fmt"
	"sync/atomic"
	"time"

	"github.com/Sirupsen/logrus"
	"github.com/mdigger/mxsms2/sms"
	"github.com/mdigger/mxsms2/zabbix"
)

// SMSTemplates описывает шаблоны сообщений.
type SMSTemplates struct {
	NoPhone   string `yaml:"noPhone,omitempty"` // не начинается с телефонного номера
	Incorrect string `yaml:",omitempty"`        // содержит некорректный номер
	Accepted  string `yaml:",omitempty"`        // отправлено
	Delivered string `yaml:",omitempty"`        // доставлено
	Error     string `yaml:",omitempty"`        // ошибка отправки или доставки
	Incoming  string `yaml:",omitempty"`        // входящее
}

// SMSGate описывает конфигурацию для отправки SMS.
type SMSGate struct {
	SMPP      *sms.SMPP    // SMPP-соединение
	Responses SMSTemplates `yaml:"messageTemplates"` // список шаблонов ответов
	MYSQL     string       `yaml:"mySqlLog"`         // инициализация подключения к логу
	Zabbix    *zabbix.Log  `yaml:"zabbix"`
	counter   uint32       // счетчик отправленных сообщений
	history   History      // история отправленных сообщений
}

func (s *SMSGate) Connect() {
	s.SMPP.Connect() // устанавливаем соединение с SMPP серверами
	go func() {
		for msg := range s.SMPP.Receive {
			// s.Logger.Debugln("Received:", msg)
			switch msg := msg.(type) {
			case sms.Received: // входящая SMS
				s.Receive(msg) // обрабатываем входящее сообщение
			}
		}
	}()
}

func (s *SMSGate) Close() {
	s.SMPP.Close() // останавливаем соединение с SMPP
}

func (s *SMSGate) Send(mxName, jid string, msgID int64, to, msg string) (err error) {
	from := s.history.GetFrom(config.MX[mxName].From, to, jid) // получаем лучший исходящий номер
	if from == "" {
		return errors.New("from phone is empty")
	}
	if to == "" {
		return errors.New("to phone is empty")
	}
	phoneType := int64(11 - len(from))
	smsMessage := &sms.SendMessage{From: from, To: to, Text: msg}
	if err = s.SMPP.Send(smsMessage); err != nil { // отправляем СМС
		zabbixLog.Send("gw.smsc.error", err.Error())
		sglogDB.Insert(mxName, from, to, msg, false, phoneType, msgID, 0)
		return err
	}
	sglogDB.Insert(mxName, from, to, msg, false, phoneType, msgID, 1)
	s.history.Add(mxName, jid, from, to) // добавляем информацию о связи телефонов в историю
	return nil
}

// Receive обрабатывает входящие сообщения
func (s *SMSGate) Receive(msg sms.Received) {
	incoming := s.Responses.Incoming
	if incoming == "" {
		incoming = "%s: %s"
	}
	// избавляемся от плюсов
	if msg.From[0] == '+' {
		msg.From = msg.From[1:]
	}
	if msg.To[0] == '+' {
		msg.To = msg.To[1:]
	}
	phoneType := int64(11 - len(msg.From))
	mxName, jid := s.history.Get(msg.To, msg.From)
	sglogDB.Insert(mxName, msg.From, msg.To, msg.Text, true, phoneType, 0, 2)
	if mxName == "" || jid == "" { // в истории не найдено подходящего пользователя, кому это адресовано
		for name, mx := range config.MX { // перебираем все настройки MX-серверов
			for _, from := range mx.From { // перебираем все их телефоны
				if msg.To == from { // нашли подходящий номер
					mxName = name
					jid = mx.DefaultJID
					if jid == "" { // не сконфигурирована доставка нераспознанных сообщений
						return
					}
					goto next // нашли подходящее
				}
			}
		}
		// номер, на который пришло сообщение, не указан ни для одного сервера
		llog.WithFields(logrus.Fields{
			"from": msg.From,
			"to":   msg.To,
		}).Warnf("SPAM detected: %q", msg.Text)
		return
	}
next:
	mx := config.MX[mxName]
	if mx == nil {
		return
	}
	for mx.handler == nil {
		llog.Debug("MX Handler not initialised... Waiting...")
		time.Sleep(time.Second)
	}
	mx.client.Send(mx.handler.getMessage(
		jid, incoming, msg.From, msg.Text))
	mx.Logger.WithFields(logrus.Fields{
		"jid":  jid,
		"from": msg.From,
		"to":   msg.To,
	}).Info("SMS incoming")
	// проверяем на спам
	for _, from := range mx.From {
		if msg.To == from {
			return // это не спам - прислано нам
		}
	}
	return
}

// getMessage возвращает сформированную команду для отправки подтверждающего сообщения
// на основе текста шаблона. Если текст шаблона пустой, то сообщение не отправляется
func (s *SMSGate) getMessage(to, tmpl string, items ...interface{}) *sendMessage {
	if tmpl == "" || to == "" {
		return nil // тема письма или адрес получателя не определены
	}
	return &sendMessage{
		To:    to,
		MsgID: atomic.AddUint32(&s.counter, 1),
		Body:  fmt.Sprintf(tmpl, items...),
	}
}

// sendMessage описывает структуру исходящего сообщения.
type sendMessage struct {
	XMLName xml.Name `xml:"message"`
	To      string   `xml:"to,attr"`            // уникальный идентификатор получателя
	MsgID   uint32   `xml:"msgId,attr"`         // идентификатор сообщения
	Ext     string   `xml:"ext,attr,omitempty"` // внутренний номер получателя
	Body    string   `xml:",chardata"`          // текст сообщения
}