package service import ( butil "enterprise/base/util" "enterprise/common/config" "enterprise/common/dao" "enterprise/common/global" "enterprise/common/model" "fmt" log "github.com/sirupsen/logrus" "github.com/smbrave/goutil" "gitlab.batiao8.com/open/gosdk/qyweixin" "gitlab.batiao8.com/open/gosdk/wechat/message" "strings" ) var ( SpStatusCreated = 1 SpStatusPassed = 2 SpStatusRefused = 3 SpStatusCanceled = 4 ) type Approve struct { approveClient *qyweixin.AppApprove } func (b *Approve) Reply(msg message.MixMessage) *message.Reply { if b.approveClient == nil { cfg := config.GetConfig() b.approveClient = qyweixin.NewAppApprove(&qyweixin.AppConfig{ Corpid: cfg.QyWeixin.Corpid, Secret: cfg.QyWeixin.ApproveSecret, Agent: cfg.QyWeixin.ApproveAgent, }) } go b.handle(&msg) return &message.Reply{message.MsgTypeText, message.NewText("")} } func (a *Approve) handle(msg *message.MixMessage) { msgType := msg.MsgType if msgType == message.MsgTypeEvent { event := strings.ToUpper(string(msg.Event)) if event == message.EventClick { a.handleClick(msg) } else if event == "SYS_APPROVAL_CHANGE" { a.handleApprovalChange(msg) } } else if msgType == message.MsgTypeText { a.handleText(msg) } log.Infof(goutil.EncodeJSONIndent(msg)) } func (a *Approve) handleApprovalChange(msg *message.MixMessage) { spStatus := msg.ApprovalInfo.SpStatus spNo := msg.ApprovalInfo.SpNo templateId := msg.ApprovalInfo.TemplateId if spStatus != SpStatusPassed { return } detail, err := a.approveClient.GetDetail(spNo) if err != nil { log.Errorf("get spn detail error :%s", err.Error()) return } if templateId == model.ApprovalTpIdRefund { a.handleRefund(detail) } else if templateId == model.ApprovalTpIdVacation { a.handleVacation(detail) } else if templateId == model.ApprovalTpIdCheckin { a.handleCheckin(detail) } } func (a *Approve) handleVacation(detail *qyweixin.ApproveDetail) { newData := new(model.ApprovalVacation) newData.From(detail) dbDao := dao.NewApprovalVacationDao() old, err := dbDao.GetBySpNo(detail.SpNo) if err != nil { log.Errorf("db error :%s", err.Error()) return } if old != nil { newData.Id = old.Id newData.CreateTime = old.CreateTime err = dbDao.Update(newData) } else { _, err = dbDao.Create(newData) } if err != nil { log.Errorf("db error :%s", err.Error()) return } } func (a *Approve) handleCheckin(detail *qyweixin.ApproveDetail) { newData := new(model.ApprovalCheckin) newData.From(detail) dbDao := dao.NewApprovalCheckinDao() old, err := dbDao.GetBySpNo(detail.SpNo) if err != nil { log.Errorf("db error :%s", err.Error()) return } if old != nil { newData.Id = old.Id newData.CreateTime = old.CreateTime err = dbDao.Update(newData) } else { _, err = dbDao.Create(newData) } if err != nil { log.Errorf("db error :%s", err.Error()) return } } func (a *Approve) handleRefund(detail *qyweixin.ApproveDetail) { newData := new(model.ApprovalRefund) newData.From(detail) dbDao := dao.NewApprovalRefundDao() old, err := dbDao.GetBySpNo(detail.SpNo) if err != nil { log.Errorf("db error :%s", err.Error()) return } if old != nil { newData.Id = old.Id newData.CreateTime = old.CreateTime err = dbDao.Update(newData) } else { _, err = dbDao.Create(newData) } if err != nil { log.Errorf("db error :%s", err.Error()) return } // 支付费用 openid, err := a.approveClient.GetOpenid(newData.Username) if err != nil { log.Errorf("GetOpenid error :%s", err.Error()) return } var req qyweixin.PayReq req.BillNo = fmt.Sprintf("BX%s%s", newData.SpNo, butil.CutTail(newData.Username, 12)) req.Title = fmt.Sprintf("【%s】报销", newData.RefundType) req.Openid = openid req.TotalAmount = int64(100 * newData.RefundAmount) qyPay := qyweixin.NewAppPay(&qyweixin.PayConfig{ Corpid: config.GetConfig().QyWeixin.Corpid, Secret: config.GetConfig().QyWeixin.PaySecret, Agent: config.GetConfig().QyWeixin.PayAgent, SerialNumber: config.GetConfig().WxPay.PaySerialNumber, ApiKey: config.GetConfig().WxPay.PayApiKeyV2, MchId: config.GetConfig().WxPay.PayMchId, CertPem: config.GetConfig().WxPay.PayCertPem, KeyPem: config.GetConfig().WxPay.PayKeyPem, }) if err = qyPay.PayMoney(&req); err != nil { log.Errorf("pay error :%s", err.Error()) return } message := make([]string, 0) message = append(message, fmt.Sprintf("【红包发放】[%s]", newData.RefundType)) message = append(message, fmt.Sprintf("发放金额:%s", fmt.Sprintf("%.2f", newData.RefundAmount))) message = append(message, fmt.Sprintf("员工名称:%s", newData.Username)) message = append(message, fmt.Sprintf("费用说明:%s", newData.RefundRemark)) if err := global.SendMessage([]string{"jiangyong"}, strings.Join(message, "\n")); err != nil { log.Errorf("send message error :%s", err.Error()) } newData.Status = model.ApprovalRefundStatusPayed if err := dao.NewApprovalRefundDao().Update(newData); err != nil { log.Errorf("db error :%s", err.Error()) } } func (a *Approve) handleText(msg *message.MixMessage) { } func (a *Approve) handleClick(msg *message.MixMessage) { } func (a *Approve) sendText(message string) { }