crrc/internal/app/ami/action/index.go

package action

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

const pacu1 = "2111"
const pacu2 = "2121"
const pacu3 = "2131"
const pacu4 = "2141"
const pacu5 = "2151"
const pacu6 = "2161"
const pacu7 = "2171"
const pacu8 = "2181"

//type Operation func(src, dst string, whisper, bargein bool)

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

			switch string(event["CallerIDNum"][2]) { // connect the pacu to pad acording to the pad number
			case "1":
				ChanSpy(pacu1, event["CallerIDNum"], false, true) //connect pacu to pad
			case "2":
				ChanSpy(pacu2, event["CallerIDNum"], false, true) //connect pacu to pad
			case "3":
				ChanSpy(pacu3, event["CallerIDNum"], false, true) //connect pacu to pad
			case "4":
				ChanSpy(pacu4, event["CallerIDNum"], false, true) //connect pacu to pad
			case "5":
				ChanSpy(pacu5, event["CallerIDNum"], false, true) //connect pacu to pad
			case "6":
				ChanSpy(pacu6, event["CallerIDNum"], false, true) //connect pacu to pad
			case "7":
				ChanSpy(pacu7, event["CallerIDNum"], false, true) //connect pacu to pad
			case "8":
				ChanSpy(pacu8, event["CallerIDNum"], false, true) //connect pacu to pad
			}
		}

	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 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 Connected() bool {
	if AminInstance != nil {
		return AminInstance.Connected()
	}
	return false
}