crrc/internal/app/ami/index.go

package ami

import (
	"net"
	"pbx-api-gin/internal/app/ami/model"
	"pbx-api-gin/internal/app/mysql"
	alstatus "pbx-api-gin/internal/app/stc/sendstatus"
	"pbx-api-gin/internal/pkg/configs"
	"pbx-api-gin/pkg/lfshook"
	"pbx-api-gin/pkg/utils"

	"github.com/sirupsen/logrus"
	"github.com/tqcenglish/amigo-go"
	"github.com/tqcenglish/amigo-go/pkg"
)

var AminInstance *amigo.Amigo

func StartAMI(connectOKCallBack func(), handleEvents []func(event map[string]string), conn net.Conn) {
	lfshook.NewLogger().Info("Start AMI")
	settings := &amigo.Settings{
		Host:     configs.ConfigGlobal.AsteriskAMIHost,
		Port:     configs.ConfigGlobal.AsteriskAMIPort,
		Username: configs.ConfigGlobal.AsteriskAMIUser,
		Password: configs.ConfigGlobal.AsteriskAMISecret,
		LogLevel: logrus.ErrorLevel}
	lfshook.NewLogger().Infof("ami setting: %+v", settings)
	AminInstance = amigo.New(settings, lfshook.NewLogger())
	AminInstance.EventOn(func(payload ...interface{}) {
		// lfshook.NewLogger().Infof("ami event on %+v", payload[0])
		event := payload[0].(map[string]string)
		handleAMI(event, conn)

		for _, handle := range handleEvents {
			go handle(event)
		}

	})
	AminInstance.ConnectOn(func(payload ...interface{}) {
		lfshook.NewLogger().Infof("ami connect on %+v", payload[0])
		if payload[0] == pkg.Connect_OK {

			connectOKCallBack()
		} else {
			lfshook.NewLogger().Errorf("ami connect failure %+v", payload)
		}
	})
	AminInstance.Connect()
}

func handleAMI(event map[string]string, conn net.Conn) {
	switch event["Event"] {
	case "DialBegin":

		if utils.IsPAIU(event["CallerIDNum"]) && utils.IsICP(event["DestCallerIDNum"]) {

			alstatus.AlarmStatus(event["CallerIDNum"], "dial", conn) // Alarm dial ICP start
		}

	case "DialEnd":

		if utils.IsPAIU(event["CallerIDNum"]) {
			alstatus.AlarmStatus(event["CallerIDNum"], "Idle", conn) // Alarm dial end
		}
	case "QueueCallerJoin":

		if utils.IsPAIU(event["CallerIDNum"]) {
			alstatus.AlarmStatus(event["CallerIDNum"], "queue", conn) // Alarm join the queue
		}
	case "ExtensionStatus":

		if event["StatusText"] == "Idle" || event["StatusText"] == "Unavailable" {
			status := &model.Extension{
				Extension: event["Exten"],
				Status:    event["StatusText"],
			}
			_, err := mysql.DBOrmInstance.Where("exten = ?", status.Extension).Cols("status").Update(status)
			if err != nil {
				lfshook.NewLogger().Infof("update extension status err : %+v", err.Error())
			}
			alstatus.AlarmStatus(event["Exten"], event["StatusText"], conn) // Alarm idle + unavailable
		}
	case "BridgeEnter":

		if utils.IsPAIU(event["CallerIDNum"]) {
			alstatus.AlarmStatus(event["CallerIDNum"], "connect", conn) // Alarm connected
		}
	}
}

func Connected() bool {
	if AminInstance != nil {
		return AminInstance.Connected()
	}
	return false
}