redpacket.go 2.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101
  1. package telegram
  2. import (
  3. "app/commons/core"
  4. "encoding/json"
  5. "fmt"
  6. tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
  7. "io"
  8. "net/http"
  9. "strings"
  10. )
  11. // handleCallbackQuery 处理回调查询
  12. func handleCallbackQuery(callback *tgbotapi.CallbackQuery) {
  13. data := callback.Data
  14. // 处理抢红包
  15. if strings.HasPrefix(data, "grab_") {
  16. packetNo := strings.TrimPrefix(data, "grab_")
  17. grabRedPacket(callback, packetNo)
  18. return
  19. }
  20. // 其他回调
  21. answerCallback(callback.ID, "未知操作")
  22. }
  23. // grabRedPacket 抢红包
  24. func grabRedPacket(callback *tgbotapi.CallbackQuery, packetNo string) {
  25. // TODO: 检查用户是否已绑定
  26. // TODO: 调用 API 抢红包
  27. // 临时实现
  28. userID := callback.From.ID
  29. username := callback.From.UserName
  30. // 模拟调用 API
  31. success, amount, message := callGrabRedPacketAPI(packetNo, userID, username)
  32. if success {
  33. // 抢到红包
  34. answerCallback(callback.ID, fmt.Sprintf("🎉 恭喜抢到 %.2f VND!", amount))
  35. // 更新原消息
  36. editText := fmt.Sprintf("%s\n\n✅ @%s 抢到了 %.2f VND",
  37. callback.Message.Text, username, amount)
  38. editMessage(callback.Message.Chat.ID, callback.Message.MessageID, editText)
  39. } else {
  40. // 抢红包失败
  41. answerCallback(callback.ID, message)
  42. }
  43. core.Log.Infof("抢红包 - 用户: %s, 红包: %s, 结果: %s",
  44. username, packetNo, message)
  45. }
  46. // callGrabRedPacketAPI 调用抢红包 API(模拟)
  47. func callGrabRedPacketAPI(packetNo string, telegramID int64, username string) (bool, float64, string) {
  48. // TODO: 实现真实的 API 调用
  49. // 这里是模拟实现
  50. apiURL := "http://localhost:2001/api/v1/redpacket/grab"
  51. payload := map[string]interface{}{
  52. "packetNo": packetNo,
  53. "telegramId": telegramID,
  54. "telegramUsername": username,
  55. }
  56. jsonData, _ := json.Marshal(payload)
  57. req, err := http.NewRequest("POST", apiURL, strings.NewReader(string(jsonData)))
  58. if err != nil {
  59. return false, 0, "系统错误"
  60. }
  61. req.Header.Set("Content-Type", "application/json")
  62. // TODO: 添加 JWT Token
  63. // req.Header.Set("Authorization", "Bearer "+token)
  64. client := &http.Client{}
  65. resp, err := client.Do(req)
  66. if err != nil {
  67. return false, 0, "网络错误"
  68. }
  69. defer resp.Body.Close()
  70. body, _ := io.ReadAll(resp.Body)
  71. var result map[string]interface{}
  72. json.Unmarshal(body, &result)
  73. if resp.StatusCode == 200 {
  74. data := result["data"].(map[string]interface{})
  75. amount := data["amount"].(float64)
  76. return true, amount, "抢红包成功"
  77. }
  78. message := result["message"].(string)
  79. return false, 0, message
  80. }