Compare commits
37 Commits
v0.0.4-rc3
...
tron_ws
| Author | SHA1 | Date | |
|---|---|---|---|
| 7d0217b618 | |||
| 097c716714 | |||
| f7c5f678c4 | |||
| 3e88d7a40c | |||
| 18dcb96822 | |||
| 4750d63db3 | |||
| 7af7306729 | |||
| ff16db907a | |||
| 96c262a7dd | |||
| 8e96cc0058 | |||
| 9c003fbead | |||
| aff1055743 | |||
| f6a8fb47df | |||
| d46bc7793e | |||
| 9d01443f3e | |||
| 70f8ed4ea0 | |||
| 8cd816c35e | |||
| 2f6cd34610 | |||
| 1e2282ba2e | |||
| 786c5e8175 | |||
| 5e68bf8309 | |||
| 5168beda00 | |||
| 5e4d5dfae4 | |||
| 32ca778735 | |||
| 730ff983f8 | |||
| 93b32d38d9 | |||
| 3f071e6c01 | |||
| a7963b335b | |||
| d7fa90112d | |||
| 5a607aa347 | |||
| b91d8b819c | |||
| f87a131d7f | |||
| 258d239831 | |||
| 5a960d3eec | |||
| 2d673493cf | |||
| 537168008c | |||
| 99c4e5e072 |
@@ -1,26 +1,24 @@
|
|||||||
FROM golang:alpine AS builder
|
FROM golang:alpine AS builder
|
||||||
|
|
||||||
RUN apk add --no-cache --update git build-base
|
RUN apk add --no-cache --update git build-base
|
||||||
ENV CGO_ENABLED=1
|
ENV CGO_ENABLED=0
|
||||||
ARG VERSION=0.0.0-dev
|
|
||||||
ARG COMMIT=none
|
|
||||||
ARG DATE=unknown
|
|
||||||
WORKDIR /app
|
|
||||||
COPY ./src/go.mod ./src/go.sum ./
|
|
||||||
RUN go mod download
|
|
||||||
|
|
||||||
COPY ./src .
|
WORKDIR /app
|
||||||
RUN go build -ldflags "-s -w -X github.com/assimon/luuu/config.BuildVersion=${VERSION} -X github.com/assimon/luuu/config.BuildCommit=${COMMIT} -X github.com/assimon/luuu/config.BuildDate=${DATE}" \
|
|
||||||
-o epusdt .
|
RUN git clone https://github.com/GMwalletApp/epusdt.git .
|
||||||
|
|
||||||
|
WORKDIR /app/src
|
||||||
|
RUN go mod download
|
||||||
|
RUN go build -o /app/epusdt .
|
||||||
|
|
||||||
FROM alpine:latest AS runner
|
FROM alpine:latest AS runner
|
||||||
ENV TZ=Asia/Shanghai
|
ENV TZ=Asia/Shanghai
|
||||||
RUN apk --no-cache add ca-certificates tzdata
|
RUN apk --no-cache add ca-certificates tzdata
|
||||||
|
|
||||||
WORKDIR /app
|
WORKDIR /app
|
||||||
COPY --from=builder /app/static /app/static
|
COPY --from=builder /app/src/static /app/static
|
||||||
COPY --from=builder /app/static /static
|
COPY --from=builder /app/src/static /static
|
||||||
COPY --from=builder /app/epusdt .
|
COPY --from=builder /app/epusdt .
|
||||||
VOLUME /app/conf
|
|
||||||
|
|
||||||
ENTRYPOINT ["./epusdt" ,"http","start"]
|
VOLUME /app/conf
|
||||||
|
ENTRYPOINT ["./epusdt", "http", "start"]
|
||||||
|
|||||||
@@ -1,71 +1,169 @@
|
|||||||
## Epusdt (Easy Payment Usdt)
|
# Epusdt — Easy Payment USDT
|
||||||
|
|
||||||
<p align="center">
|
<p align="center">
|
||||||
<img src="wiki/img/usdtlogo.png">
|
<img src="wiki/img/usdtlogo.png" alt="Epusdt Logo - Multi-chain Crypto Payment Gateway" width="120">
|
||||||
</p>
|
|
||||||
<p align="center">
|
|
||||||
<a href="https://www.gnu.org/licenses/gpl-3.0.html"><img src="https://img.shields.io/badge/license-GPLV3-blue" alt="license GPLV3"></a>
|
|
||||||
<a href="https://golang.org"><img src="https://img.shields.io/badge/Golang-1.16-red" alt="Go version 1.16"></a>
|
|
||||||
<a href="https://echo.labstack.com"><img src="https://img.shields.io/badge/Echo Framework-v4-blue" alt="Echo Framework v4"></a>
|
|
||||||
<a href="https://github.com/tucnak/telebot"><img src="https://img.shields.io/badge/Telebot Framework-v3-lightgrey" alt="Telebot Framework-v3"></a>
|
|
||||||
<a href="https://github.com/assimon/epusdt/releases/tag/v0.0.1"><img src="https://img.shields.io/badge/version-v0.0.1-green" alt="version v0.0.1"></a>
|
|
||||||
</p>
|
</p>
|
||||||
|
|
||||||
|
<p align="center">
|
||||||
|
<strong>开源多链多币种 Crypto 支付网关 · 实际采用率 Top 1</strong>
|
||||||
|
</p>
|
||||||
|
|
||||||
## 项目简介
|
<p align="center">
|
||||||
`Epusdt`(全称:Easy Payment Usdt)是一个由`Go语言`编写的私有化部署`Usdt`支付中间件(`Trc20网络`)
|
<a href="https://epusdt.com"><img src="https://img.shields.io/badge/官网文档-epusdt.com-blue?style=for-the-badge" alt="Official Docs"></a>
|
||||||
站长或开发者可通过`Epusdt`提供的`http api`集成至您的任何系统,无需过多的配置,仅仅依赖`mysql`和`redis`
|
<a href="https://t.me/epusdt"><img src="https://img.shields.io/badge/Telegram-频道-26A5E4?style=for-the-badge&logo=telegram&logoColor=white" alt="Telegram Channel"></a>
|
||||||
可实现USDT的在线支付和消息回调,这一切在优雅和顷刻间完成!🎉
|
<a href="https://t.me/epusdt_group"><img src="https://img.shields.io/badge/Telegram-交流群-26A5E4?style=for-the-badge&logo=telegram&logoColor=white" alt="Telegram Group"></a>
|
||||||
私有化搭建使得无需额外的手续费和签约费用,Usdt代币直接进入您的钱包💰
|
</p>
|
||||||
`Epusdt` 遵守 [GPLv3](https://www.gnu.org/licenses/gpl-3.0.html) 开源协议!
|
|
||||||
|
|
||||||
## 项目特点
|
<p align="center">
|
||||||
- 支持私有化部署,无需担心钱包被篡改和吞单😁
|
<a href="https://github.com/assimon/epusdt/stargazers"><img src="https://img.shields.io/github/stars/assimon/epusdt?style=flat-square&color=f5c542" alt="GitHub Stars 3000+"></a>
|
||||||
- `Go语言`跨平台实现,支持x86和arm芯片架构的win/linux设备
|
<a href="https://www.gnu.org/licenses/gpl-3.0.html"><img src="https://img.shields.io/badge/License-GPLv3-blue?style=flat-square" alt="GPLv3 License"></a>
|
||||||
- 多钱包地址轮询,提高订单并发率
|
<a href="https://golang.org"><img src="https://img.shields.io/badge/Go-1.16+-00ADD8?style=flat-square&logo=go&logoColor=white" alt="Go 1.16+"></a>
|
||||||
- 异步队列响应,优雅及高性能
|
<a href="https://github.com/assimon/epusdt/releases"><img src="https://img.shields.io/github/v/release/assimon/epusdt?style=flat-square&color=green" alt="Latest Release"></a>
|
||||||
- 无需额外环境配置,仅运行一个编译后二进制文件即可使用
|
</p>
|
||||||
- 支持`http api`,其他系统亦可接入
|
|
||||||
- `Telegram`机器人接入,便捷使用和支付消息快速通知
|
---
|
||||||
|
|
||||||
|
## 🌍 What is Epusdt?
|
||||||
|
|
||||||
|
**Epusdt** (Easy Payment USDT) is a self-hosted **multi-chain, multi-token crypto payment gateway** built with Go. It has evolved from a TRC20-only solution into a comprehensive **multi-chain receiving platform**, enabling any website or application to accept crypto payments across multiple blockchain networks and token types. No third-party fees, no custodial risk — funds go directly into your wallet.
|
||||||
|
|
||||||
|
> **⭐ GitHub Star 3000+** · **🔌 已支持站点解决方案 10+** · **🏆 Crypto 支付工具实际采用率 Top 1**
|
||||||
|
|
||||||
|
Deploy it privately, integrate via HTTP API, and start receiving **crypto payments** in minutes. That's it. 🎉
|
||||||
|
|
||||||
|
### 🔗 已支持网络与代币
|
||||||
|
|
||||||
|
| 网络 | 代币 |
|
||||||
|
|------|------|
|
||||||
|
| **TRC20** (Tron) | USDT、TRX|
|
||||||
|
| **ERC20** (Ethereum) | USDT、USDC、ETH |
|
||||||
|
| **BEP20** (BSC) | USDT、USDC、BNB |
|
||||||
|
| **Polygon** | USDT、USDC |
|
||||||
|
| **更多** | 持续扩展中… |
|
||||||
|
|
||||||
|
> 💡 具体支持的链与代币以 [最新版本](https://github.com/assimon/epusdt/releases) 及 [官方文档](https://epusdt.com) 为准。
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 🔌 广泛兼容,即插即用
|
||||||
|
|
||||||
|
无论你运营的是哪类系统,Epusdt 均可基于现有接口方案,**无需重构业务逻辑**,快速接入,立即获得 Crypto 收款能力,低成本扩展全球支付场景:
|
||||||
|
|
||||||
|
| 领域 | 已支持系统 |
|
||||||
|
|------|-----------|
|
||||||
|
| **AI 分发** | [OneAPI](https://github.com/songquanpeng/one-api)、[NewAPI](https://github.com/QuantumNous/new-api) |
|
||||||
|
| **发卡系统** | [独角数卡(Dujiaoka)](https://dujiao-next.com/)、[异次元发卡](https://github.com/lizhipay/acg-faka) |
|
||||||
|
| **代理面板** | [V2Board](https://github.com/v2board/v2board)、[XBoard](https://github.com/cedar2025/Xboard)、[xiaoV2board](https://github.com/wyx2685/v2board/)、[SSPanel](https://github.com/anankke/sspanel-uim) |
|
||||||
|
| **建站生态** | [WordPress](https://wordpress.com/)、[WHMCS](https://www.whmcs.com/) |
|
||||||
|
| **Epay兼容** | 兼容各类支持Epay易支付接口的平台 |
|
||||||
|
| **更多** | 简易HTTP API 10分钟内接入 |
|
||||||
|
|
||||||
|
👉 查看更多集成列表与插件:[plugins/](plugins/)
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## ✨ 核心特性
|
||||||
|
|
||||||
|
- **多链多币种** — 支持 TRC20、ERC20、BEP20、Polygon 等主流网络,不再局限于单一链
|
||||||
|
- **私有化部署** — 无需担心钱包被篡改、吞单,资金完全自主掌控
|
||||||
|
- **零依赖运行** — 单个二进制文件即可启动,低并发场景无需安装 MySQL + Redis,一键部署零成本维护
|
||||||
|
- **跨平台** — 支持 x86 / ARM 架构的 Windows / Linux / Mac 设备
|
||||||
|
- **多钱包轮询** — 自动轮换收款地址,提高订单并发处理能力
|
||||||
|
- **异步队列** — 高性能消息回调,优雅处理高并发场景
|
||||||
|
- **HTTP API** — 标准化接口,任何语言 / 框架均可10min内集成
|
||||||
|
- **Telegram Bot** — 实时支付通知,快捷管理与监控
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 📖 文档与教程
|
||||||
|
|
||||||
|
完整文档请访问 👉 **[epusdt.com](https://epusdt.com)**
|
||||||
|
|
||||||
|
快速入门:
|
||||||
|
|
||||||
|
| 教程 | 说明 |
|
||||||
|
|------|------|
|
||||||
|
| [Docker 部署](https://epusdt.com/guide/installation/docker) | 推荐方式,一键启动 |
|
||||||
|
| [宝塔面板部署](https://epusdt.com/guide/installation/aapanel) | 适合宝塔用户 |
|
||||||
|
| [手动部署](https://epusdt.com/guide/installation/manual.html) | 完全手动控制 |
|
||||||
|
| [开发者 API 文档](https://epusdt.com/zh/guide/integration/gmpay.html) | 接口集成指南 |
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 🏗️ 项目结构
|
||||||
|
|
||||||
## 项目结构
|
|
||||||
```
|
```
|
||||||
Epusdt
|
Epusdt
|
||||||
├── plugins ---> (已集成的插件库,例如dujiaoka)
|
├── plugins/ → 已集成的系统插件(独角数卡等)
|
||||||
├── src ---> (项目核心目录)
|
├── src/ → 项目核心代码
|
||||||
├── sdk ---> (接入SDK)
|
├── sdk/ → 接入 SDK
|
||||||
├── sql ---> (安装sql文件或更新sql文件)
|
├── sql/ → 数据库安装 / 升级脚本
|
||||||
└── wiki ---> (知识库)
|
└── wiki/ → 文档与知识库
|
||||||
```
|
```
|
||||||
|
|
||||||
## 教程:
|
---
|
||||||
- 宝塔运行`epusdt`教程👉🏻[宝塔运行epusdt](wiki/BT_RUN.md)
|
|
||||||
- 不好意思我有洁癖,手动运行`epusdt`教程👉🏻[手动运行epusdt](wiki/manual_RUN.md)
|
|
||||||
- 开发者接入`epusdt`文档👉🏻[开发者接入epusdt](wiki/API.md)
|
|
||||||
- HTML+PHP极速运行`epusdt`教程👉🏻[使用PHPAPI-for-epusdt极速接入epusdt](https://github.com/BlueSkyXN/PHPAPI-for-epusdt)
|
|
||||||
|
|
||||||
## 已适配系统插件
|
## 🔧 实现原理
|
||||||
- 独角数卡[插件地址](plugins/dujiaoka)
|
|
||||||
|
|
||||||
## 加入交流/意见反馈
|
Epusdt 通过监听多条区块链网络(TRC20、ERC20、BEP20、Polygon 等)的 API 或RPC节点,实时捕获钱包地址的代币入账事件,利用**金额差异**与**时效性**精确匹配交易归属:
|
||||||
- `Epusdt`频道[https://t.me/epusdt](https://t.me/epusdt)
|
|
||||||
- `Epusdt`交流群组[https://t.me/epusdt_group](https://t.me/epusdt_group)
|
|
||||||
|
|
||||||
## 设计实现
|
|
||||||
`Epusdt`的实现方式与其他项目原理类似,都是通过监听`trc20`网络的api或节点,
|
|
||||||
监听钱包地址`usdt`代币入账事件,通过`金额差异`和`时效性`来判定交易归属信息,
|
|
||||||
可参考下方`流程图`
|
|
||||||
```
|
```
|
||||||
简单的原理:
|
工作流程:
|
||||||
1.客户需要支付20.05usdt
|
1. 客户发起支付,需支付 20.05 USDT
|
||||||
2.服务器有一个hash表存储钱包地址对应的待支付金额 例如:address_1 : 20.05
|
2. 系统在哈希表中查找可用的 钱包地址 + 金额 组合
|
||||||
3.发起支付的时候,我们可以判定钱包address_1的20.05金额是否被占用,如果没有被占用那么可以直接返回这个钱包地址和金额给客户,告知客户需按规定金额20.05准确支付,少一分都不行。且将钱包地址和金额 address_1:20.05锁起来,有效期10分钟。
|
3. 若 address_1:20.05 未被占用 → 锁定该组合(有效期 10 分钟),返回给客户
|
||||||
4.如果订单并发下,又有一个20.05元需要支付,但是在第3步的时候上一个客户已经锁定了该金额,还在等待支付中...,那么我们将待支付金额加上0.0001,再次尝试判断address_1:20.0501金额是否被占用?如果没有则重复第三步,如果还是被占用就继续累加尝试,直到加了100次后都失败
|
4. 若已被占用 → 自动累加 0.0001 尝试下一个金额组合(最多 100 次)
|
||||||
5.新开一个线程去监听所有钱包的USDT入账事件,网上有公开的api或rpc节点。如果发现有入账金额与待支付的金额相等。则判断该笔订单支付成功!
|
5. 后台线程持续监听所有钱包的入账事件,金额匹配则确认支付成功
|
||||||
```
|
```
|
||||||
### 流程图:
|
|
||||||

|
|
||||||
|
|
||||||
|

|
||||||
|
|
||||||
## 声明
|
---
|
||||||
本项目仅供学习与技术交流使用,用户在使用过程中需自行遵守所在地法律法规。由于涉及加密资产及资金安全,用户应自行审查相关代码与风险。加密资产属于高风险新兴资产(包括稳定币),其价值可能波动甚至归零,GMwallet不对任何资产或使用结果作出保证。本内容不构成任何投资、税务、法律或金融建议,仅供教育用途,相关决策请咨询专业人士。
|
|
||||||
|
|
||||||
|
## 💬 社区与支持
|
||||||
|
|
||||||
|
**遇到问题?** 请优先在 GitHub 提交 [Issue](https://github.com/assimon/epusdt/issues),我们会**优先处理** Issue 中的反馈。
|
||||||
|
|
||||||
|
加入 Telegram 社区,获取最新开发动态,参与需求调研,与超过5000名活跃用户交流使用经验,对接商业资源:
|
||||||
|
|
||||||
|
| 渠道 | 链接 |
|
||||||
|
|------|------|
|
||||||
|
| 📢 **Epusdt 频道** | [https://t.me/epusdt](https://t.me/epusdt) |
|
||||||
|
| 💬 **Epusdt 交流群** | [https://t.me/epusdt_group](https://t.me/epusdt_group) |
|
||||||
|
| 📚 **官方文档站** | [https://epusdt.com](https://epusdt.com) |
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## Star History
|
||||||
|
|
||||||
|
<a href="https://www.star-history.com/?type=date&repos=gmwalletapp%2Fepusdt">
|
||||||
|
<picture>
|
||||||
|
<source media="(prefers-color-scheme: dark)" srcset="https://api.star-history.com/chart?repos=gmwalletapp/epusdt&type=date&theme=dark&legend=top-left" />
|
||||||
|
<source media="(prefers-color-scheme: light)" srcset="https://api.star-history.com/chart?repos=gmwalletapp/epusdt&type=date&legend=top-left" />
|
||||||
|
<img alt="Star History Chart" src="https://api.star-history.com/chart?repos=gmwalletapp/epusdt&type=date&legend=top-left" />
|
||||||
|
</picture>
|
||||||
|
</a>
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 📜 开源协议
|
||||||
|
|
||||||
|
Epusdt 遵守 [GPLv3](https://www.gnu.org/licenses/gpl-3.0.html) 开源协议。
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## ⚠️ 免责声明
|
||||||
|
|
||||||
|
本项目仅供学习与技术交流使用,用户在使用过程中需自行遵守所在地法律法规。由于涉及加密资产及资金安全,用户应自行审查相关代码与风险。加密资产属于高风险新兴资产(包括稳定币),其价值可能波动甚至归零,GMwallet 不对任何资产或使用结果作出保证。本内容不构成任何投资、税务、法律或金融建议,仅供教育用途,相关决策请咨询专业人士。
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
<p align="center">
|
||||||
|
<sub>
|
||||||
|
<b>Keywords:</b> USDT Payment Gateway · Crypto Payment · Multi-chain Payment · TRC20 Payment · ERC20 Payment · BEP20 Payment ·
|
||||||
|
Self-hosted Crypto Gateway · OneAPI Payment · NewAPI Payment · 独角数卡支付 · 异次元发卡支付方式 ·
|
||||||
|
V2Board Payment · XBoard Payment · SSPanel 支付接口 ·
|
||||||
|
WordPress Crypto Payment · WHMCS USDT Payment · Polygon USDT ·
|
||||||
|
Epusdt · Easy Payment USDT · Open Source Payment Gateway · 多链收款
|
||||||
|
</sub>
|
||||||
|
</p>
|
||||||
|
|||||||
@@ -1,7 +1,6 @@
|
|||||||
version: "3"
|
|
||||||
services:
|
services:
|
||||||
epusdt:
|
epusdt:
|
||||||
image: annona/epusdt:alpine
|
image: gmwallet/epusdt:alpine
|
||||||
restart: always
|
restart: always
|
||||||
build:
|
build:
|
||||||
context: .
|
context: .
|
||||||
@@ -9,4 +8,4 @@ services:
|
|||||||
volumes:
|
volumes:
|
||||||
- ./env:/app/.env
|
- ./env:/app/.env
|
||||||
ports:
|
ports:
|
||||||
- "8088:8000"
|
- "8000:8000"
|
||||||
|
|||||||
@@ -61,3 +61,11 @@ order_notice_max_retry=0
|
|||||||
forced_usdt_rate=
|
forced_usdt_rate=
|
||||||
api_rate_url=
|
api_rate_url=
|
||||||
tron_grid_api_key=
|
tron_grid_api_key=
|
||||||
|
solana_rpc_url=
|
||||||
|
|
||||||
|
ethereum_ws_url=wss://ethereum.publicnode.com
|
||||||
|
|
||||||
|
|
||||||
|
# epay
|
||||||
|
epay_pid=
|
||||||
|
epay_key=
|
||||||
|
|||||||
@@ -1,2 +1,3 @@
|
|||||||
.DS_Store
|
.DS_Store
|
||||||
dist/
|
dist/
|
||||||
|
*.log
|
||||||
@@ -328,3 +328,23 @@ func GetCallbackRetryBaseDuration() time.Duration {
|
|||||||
}
|
}
|
||||||
return time.Duration(seconds) * time.Second
|
return time.Duration(seconds) * time.Second
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func GetSolanaRpcUrl() string {
|
||||||
|
rpcUrl := viper.GetString("solana_rpc_url")
|
||||||
|
if rpcUrl == "" {
|
||||||
|
return "https://api.mainnet-beta.solana.com"
|
||||||
|
}
|
||||||
|
return rpcUrl
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetEthereumWsUrl() string {
|
||||||
|
return strings.TrimSpace(viper.GetString("ethereum_ws_url"))
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetEpayPid() int {
|
||||||
|
return viper.GetInt("epay_pid")
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetEpayKey() string {
|
||||||
|
return viper.GetString("epay_key")
|
||||||
|
}
|
||||||
|
|||||||
@@ -67,8 +67,27 @@ func TestResolveConfigFilePathUsesCurrentDirectoryByDefault(t *testing.T) {
|
|||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("resolve config path: %v", err)
|
t.Fatalf("resolve config path: %v", err)
|
||||||
}
|
}
|
||||||
if got != configPath {
|
|
||||||
t.Fatalf("config path = %s, want %s", got, configPath)
|
gotAbs, err := filepath.Abs(got)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("abs got: %v", err)
|
||||||
|
}
|
||||||
|
wantAbs, err := filepath.Abs(configPath)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("abs want: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
gotReal, err := filepath.EvalSymlinks(gotAbs)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("eval symlinks got: %v", err)
|
||||||
|
}
|
||||||
|
wantReal, err := filepath.EvalSymlinks(wantAbs)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("eval symlinks want: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if gotReal != wantReal {
|
||||||
|
t.Fatalf("config path = %s, want %s", gotReal, wantReal)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -1,6 +1,10 @@
|
|||||||
package comm
|
package comm
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
|
||||||
"github.com/assimon/luuu/model/request"
|
"github.com/assimon/luuu/model/request"
|
||||||
"github.com/assimon/luuu/model/service"
|
"github.com/assimon/luuu/model/service"
|
||||||
"github.com/assimon/luuu/util/constant"
|
"github.com/assimon/luuu/util/constant"
|
||||||
@@ -22,3 +26,53 @@ func (c *BaseCommController) CreateTransaction(ctx echo.Context) (err error) {
|
|||||||
}
|
}
|
||||||
return c.SucJson(ctx, resp)
|
return c.SucJson(ctx, resp)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// SwitchNetwork 切换支付网络,创建或返回子订单
|
||||||
|
func (c *BaseCommController) SwitchNetwork(ctx echo.Context) (err error) {
|
||||||
|
req := new(request.SwitchNetworkRequest)
|
||||||
|
if err = ctx.Bind(req); err != nil {
|
||||||
|
return c.FailJson(ctx, constant.ParamsMarshalErr)
|
||||||
|
}
|
||||||
|
if err = c.ValidateStruct(ctx, req); err != nil {
|
||||||
|
return c.FailJson(ctx, err)
|
||||||
|
}
|
||||||
|
resp, err := service.SwitchNetwork(req)
|
||||||
|
if err != nil {
|
||||||
|
return c.FailJson(ctx, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
jsonBytes, err := json.MarshalIndent(resp, "", " ")
|
||||||
|
if err != nil {
|
||||||
|
return c.FailJson(ctx, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Printf("switch network response: \n%s", string(jsonBytes))
|
||||||
|
|
||||||
|
return c.SucJson(ctx, resp)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *BaseCommController) CreateTransactionAndRedirect(ctx echo.Context) (err error) {
|
||||||
|
req := new(request.CreateTransactionRequest)
|
||||||
|
if err = ctx.Bind(req); err != nil {
|
||||||
|
log.Println("bind request error:", err)
|
||||||
|
return c.FailJson(ctx, constant.ParamsMarshalErr)
|
||||||
|
}
|
||||||
|
if err = c.ValidateStruct(ctx, req); err != nil {
|
||||||
|
log.Println("validate request error:", err)
|
||||||
|
return c.FailJson(ctx, err)
|
||||||
|
}
|
||||||
|
resp, err := service.CreateTransaction(req)
|
||||||
|
if err != nil {
|
||||||
|
log.Println("create transaction error:", err)
|
||||||
|
return c.FailJson(ctx, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Printf("create transaction response: %+v\n", resp)
|
||||||
|
|
||||||
|
tradeID := resp.TradeId
|
||||||
|
|
||||||
|
ctx.Redirect(302, "/pay/checkout-counter/"+tradeID)
|
||||||
|
|
||||||
|
return nil
|
||||||
|
|
||||||
|
}
|
||||||
|
|||||||
@@ -1,13 +1,16 @@
|
|||||||
package comm
|
package comm
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"html/template"
|
||||||
|
"net/http"
|
||||||
|
"path/filepath"
|
||||||
|
|
||||||
"github.com/assimon/luuu/config"
|
"github.com/assimon/luuu/config"
|
||||||
"github.com/assimon/luuu/model/response"
|
"github.com/assimon/luuu/model/response"
|
||||||
"github.com/assimon/luuu/model/service"
|
"github.com/assimon/luuu/model/service"
|
||||||
"github.com/labstack/echo/v4"
|
"github.com/labstack/echo/v4"
|
||||||
"html/template"
|
|
||||||
"net/http"
|
|
||||||
"path/filepath"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// CheckoutCounter 收银台
|
// CheckoutCounter 收银台
|
||||||
@@ -15,12 +18,27 @@ func (c *BaseCommController) CheckoutCounter(ctx echo.Context) (err error) {
|
|||||||
tradeId := ctx.Param("trade_id")
|
tradeId := ctx.Param("trade_id")
|
||||||
resp, err := service.GetCheckoutCounterByTradeId(tradeId)
|
resp, err := service.GetCheckoutCounterByTradeId(tradeId)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
if err == service.ErrOrder {
|
||||||
|
tmpl, err := template.ParseFiles(filepath.Join(config.StaticFilePath, "index.html"))
|
||||||
|
if err != nil {
|
||||||
|
return ctx.String(http.StatusOK, err.Error())
|
||||||
|
}
|
||||||
|
emptyResp := response.CheckoutCounterResponse{}
|
||||||
|
return tmpl.Execute(ctx.Response(), emptyResp)
|
||||||
|
}
|
||||||
return ctx.String(http.StatusOK, err.Error())
|
return ctx.String(http.StatusOK, err.Error())
|
||||||
}
|
}
|
||||||
tmpl, err := template.ParseFiles(filepath.Join(config.StaticFilePath, "index.html"))
|
tmpl, err := template.ParseFiles(filepath.Join(config.StaticFilePath, "index.html"))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return ctx.String(http.StatusOK, err.Error())
|
return ctx.String(http.StatusOK, err.Error())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
jsonByte, err := json.MarshalIndent(resp, "", " ")
|
||||||
|
if err != nil {
|
||||||
|
return ctx.String(http.StatusOK, err.Error())
|
||||||
|
}
|
||||||
|
fmt.Printf("%v\n", string(jsonByte))
|
||||||
|
|
||||||
return tmpl.Execute(ctx.Response(), resp)
|
return tmpl.Execute(ctx.Response(), resp)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,147 @@
|
|||||||
|
package comm
|
||||||
|
|
||||||
|
import (
|
||||||
|
"sort"
|
||||||
|
"strconv"
|
||||||
|
|
||||||
|
"github.com/assimon/luuu/model/data"
|
||||||
|
"github.com/assimon/luuu/model/response"
|
||||||
|
"github.com/assimon/luuu/util/constant"
|
||||||
|
"github.com/labstack/echo/v4"
|
||||||
|
)
|
||||||
|
|
||||||
|
type addSupportedAssetRequest struct {
|
||||||
|
Network string `json:"network" validate:"required"`
|
||||||
|
Token string `json:"token" validate:"required"`
|
||||||
|
Status int64 `json:"status" validate:"required|in:1,2"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type updateSupportedAssetRequest struct {
|
||||||
|
Network string `json:"network" validate:"required"`
|
||||||
|
Token string `json:"token" validate:"required"`
|
||||||
|
Status int64 `json:"status" validate:"required|in:1,2"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetSupportedAssets 对外公开可用链与 token 列表(无需鉴权,仅返回已启用项)。
|
||||||
|
func (c *BaseCommController) GetSupportedAssets(ctx echo.Context) error {
|
||||||
|
list, err := data.ListEnabledSupportedAssets()
|
||||||
|
if err != nil {
|
||||||
|
return c.FailJson(ctx, err)
|
||||||
|
}
|
||||||
|
wallets, err := data.GetAvailableWalletAddress()
|
||||||
|
if err != nil {
|
||||||
|
return c.FailJson(ctx, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
networkSet := make(map[string]struct{})
|
||||||
|
for _, w := range wallets {
|
||||||
|
networkSet[w.Network] = struct{}{}
|
||||||
|
}
|
||||||
|
|
||||||
|
grouped := make(map[string][]string)
|
||||||
|
for _, item := range list {
|
||||||
|
if _, ok := networkSet[item.Network]; !ok {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
grouped[item.Network] = append(grouped[item.Network], item.Token)
|
||||||
|
}
|
||||||
|
|
||||||
|
networks := make([]string, 0, len(grouped))
|
||||||
|
for network := range grouped {
|
||||||
|
networks = append(networks, network)
|
||||||
|
}
|
||||||
|
sort.Strings(networks)
|
||||||
|
|
||||||
|
supports := make([]response.NetworkTokenSupport, 0, len(networks))
|
||||||
|
for _, network := range networks {
|
||||||
|
tokens := grouped[network]
|
||||||
|
sort.Strings(tokens)
|
||||||
|
supports = append(supports, response.NetworkTokenSupport{
|
||||||
|
Network: network,
|
||||||
|
Tokens: tokens,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
return c.SucJson(ctx, response.SupportedAssetsResponse{Supports: supports})
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListSupportedAssetRecords 查询支持项明细(无需鉴权)。
|
||||||
|
func (c *BaseCommController) ListSupportedAssetRecords(ctx echo.Context) error {
|
||||||
|
network := ctx.QueryParam("network")
|
||||||
|
list, err := data.ListSupportedAssets(network)
|
||||||
|
if err != nil {
|
||||||
|
return c.FailJson(ctx, err)
|
||||||
|
}
|
||||||
|
return c.SucJson(ctx, list)
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetSupportedAsset 查询单条支持项(无需鉴权)。
|
||||||
|
func (c *BaseCommController) GetSupportedAsset(ctx echo.Context) error {
|
||||||
|
id, err := strconv.ParseUint(ctx.Param("id"), 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
return c.FailJson(ctx, constant.ParamsMarshalErr)
|
||||||
|
}
|
||||||
|
asset, err := data.GetSupportedAssetByID(id)
|
||||||
|
if err != nil {
|
||||||
|
return c.FailJson(ctx, err)
|
||||||
|
}
|
||||||
|
if asset.ID <= 0 {
|
||||||
|
return c.FailJson(ctx, constant.SupportedAssetNotFound)
|
||||||
|
}
|
||||||
|
return c.SucJson(ctx, asset)
|
||||||
|
}
|
||||||
|
|
||||||
|
// AddSupportedAsset 新增支持项(鉴权)。
|
||||||
|
func (c *BaseCommController) AddSupportedAsset(ctx echo.Context) error {
|
||||||
|
req := new(addSupportedAssetRequest)
|
||||||
|
if err := ctx.Bind(req); err != nil {
|
||||||
|
return c.FailJson(ctx, constant.ParamsMarshalErr)
|
||||||
|
}
|
||||||
|
if err := c.ValidateStruct(ctx, req); err != nil {
|
||||||
|
return c.FailJson(ctx, err)
|
||||||
|
}
|
||||||
|
asset, err := data.AddSupportedAsset(req.Network, req.Token, req.Status)
|
||||||
|
if err != nil {
|
||||||
|
return c.FailJson(ctx, err)
|
||||||
|
}
|
||||||
|
return c.SucJson(ctx, asset)
|
||||||
|
}
|
||||||
|
|
||||||
|
// UpdateSupportedAsset 修改支持项(鉴权)。
|
||||||
|
func (c *BaseCommController) UpdateSupportedAsset(ctx echo.Context) error {
|
||||||
|
id, err := strconv.ParseUint(ctx.Param("id"), 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
return c.FailJson(ctx, constant.ParamsMarshalErr)
|
||||||
|
}
|
||||||
|
req := new(updateSupportedAssetRequest)
|
||||||
|
if err := ctx.Bind(req); err != nil {
|
||||||
|
return c.FailJson(ctx, constant.ParamsMarshalErr)
|
||||||
|
}
|
||||||
|
if err := c.ValidateStruct(ctx, req); err != nil {
|
||||||
|
return c.FailJson(ctx, err)
|
||||||
|
}
|
||||||
|
asset, err := data.UpdateSupportedAsset(id, req.Network, req.Token, req.Status)
|
||||||
|
if err != nil {
|
||||||
|
return c.FailJson(ctx, err)
|
||||||
|
}
|
||||||
|
return c.SucJson(ctx, asset)
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeleteSupportedAsset 删除支持项(鉴权)。
|
||||||
|
func (c *BaseCommController) DeleteSupportedAsset(ctx echo.Context) error {
|
||||||
|
id, err := strconv.ParseUint(ctx.Param("id"), 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
return c.FailJson(ctx, constant.ParamsMarshalErr)
|
||||||
|
}
|
||||||
|
asset, err := data.GetSupportedAssetByID(id)
|
||||||
|
if err != nil {
|
||||||
|
return c.FailJson(ctx, err)
|
||||||
|
}
|
||||||
|
if asset.ID <= 0 {
|
||||||
|
return c.FailJson(ctx, constant.SupportedAssetNotFound)
|
||||||
|
}
|
||||||
|
if err := data.DeleteSupportedAssetByID(id); err != nil {
|
||||||
|
return c.FailJson(ctx, err)
|
||||||
|
}
|
||||||
|
return c.SucJson(ctx, nil)
|
||||||
|
}
|
||||||
@@ -0,0 +1,98 @@
|
|||||||
|
package comm
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strconv"
|
||||||
|
|
||||||
|
"github.com/assimon/luuu/model/data"
|
||||||
|
"github.com/assimon/luuu/model/mdb"
|
||||||
|
"github.com/assimon/luuu/util/constant"
|
||||||
|
"github.com/labstack/echo/v4"
|
||||||
|
)
|
||||||
|
|
||||||
|
type addWalletRequest struct {
|
||||||
|
Network string `json:"network" validate:"required"`
|
||||||
|
Address string `json:"address" validate:"required"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type changeStatusRequest struct {
|
||||||
|
Status int `json:"status" validate:"required|in:1,2"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// AddWallet 添加钱包地址
|
||||||
|
func (c *BaseCommController) AddWallet(ctx echo.Context) error {
|
||||||
|
req := new(addWalletRequest)
|
||||||
|
if err := ctx.Bind(req); err != nil {
|
||||||
|
return c.FailJson(ctx, constant.ParamsMarshalErr)
|
||||||
|
}
|
||||||
|
if err := c.ValidateStruct(ctx, req); err != nil {
|
||||||
|
return c.FailJson(ctx, err)
|
||||||
|
}
|
||||||
|
wallet, err := data.AddWalletAddressWithNetwork(req.Network, req.Address)
|
||||||
|
if err != nil {
|
||||||
|
return c.FailJson(ctx, err)
|
||||||
|
}
|
||||||
|
return c.SucJson(ctx, wallet)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListWallets 获取钱包列表
|
||||||
|
func (c *BaseCommController) ListWallets(ctx echo.Context) error {
|
||||||
|
network := ctx.QueryParam("network")
|
||||||
|
var wallets []mdb.WalletAddress
|
||||||
|
var err error
|
||||||
|
if network != "" {
|
||||||
|
wallets, err = data.GetAllWalletAddressByNetwork(network)
|
||||||
|
} else {
|
||||||
|
wallets, err = data.GetAllWalletAddress()
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return c.FailJson(ctx, err)
|
||||||
|
}
|
||||||
|
return c.SucJson(ctx, wallets)
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetWallet 获取单个钱包
|
||||||
|
func (c *BaseCommController) GetWallet(ctx echo.Context) error {
|
||||||
|
id, err := strconv.ParseUint(ctx.Param("id"), 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
return c.FailJson(ctx, constant.ParamsMarshalErr)
|
||||||
|
}
|
||||||
|
wallet, err := data.GetWalletAddressById(id)
|
||||||
|
if err != nil {
|
||||||
|
return c.FailJson(ctx, err)
|
||||||
|
}
|
||||||
|
if wallet.ID <= 0 {
|
||||||
|
return c.FailJson(ctx, constant.NotAvailableWalletAddress)
|
||||||
|
}
|
||||||
|
return c.SucJson(ctx, wallet)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ChangeWalletStatus 启用/禁用钱包
|
||||||
|
func (c *BaseCommController) ChangeWalletStatus(ctx echo.Context) error {
|
||||||
|
id, err := strconv.ParseUint(ctx.Param("id"), 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
return c.FailJson(ctx, constant.ParamsMarshalErr)
|
||||||
|
}
|
||||||
|
req := new(changeStatusRequest)
|
||||||
|
if err := ctx.Bind(req); err != nil {
|
||||||
|
return c.FailJson(ctx, constant.ParamsMarshalErr)
|
||||||
|
}
|
||||||
|
if err := c.ValidateStruct(ctx, req); err != nil {
|
||||||
|
return c.FailJson(ctx, err)
|
||||||
|
}
|
||||||
|
if err := data.ChangeWalletAddressStatus(id, req.Status); err != nil {
|
||||||
|
return c.FailJson(ctx, err)
|
||||||
|
}
|
||||||
|
return c.SucJson(ctx, nil)
|
||||||
|
}
|
||||||
|
|
||||||
|
// DeleteWallet 删除钱包
|
||||||
|
func (c *BaseCommController) DeleteWallet(ctx echo.Context) error {
|
||||||
|
id, err := strconv.ParseUint(ctx.Param("id"), 10, 64)
|
||||||
|
if err != nil {
|
||||||
|
return c.FailJson(ctx, constant.ParamsMarshalErr)
|
||||||
|
}
|
||||||
|
if err := data.DeleteWalletAddressById(id); err != nil {
|
||||||
|
return c.FailJson(ctx, err)
|
||||||
|
}
|
||||||
|
return c.SucJson(ctx, nil)
|
||||||
|
}
|
||||||
@@ -5,6 +5,8 @@ go 1.25.0
|
|||||||
require (
|
require (
|
||||||
github.com/btcsuite/btcutil v1.0.2
|
github.com/btcsuite/btcutil v1.0.2
|
||||||
github.com/dromara/carbon/v2 v2.6.15
|
github.com/dromara/carbon/v2 v2.6.15
|
||||||
|
github.com/ethereum/go-ethereum v1.15.11
|
||||||
|
github.com/gagliardetto/solana-go v1.16.0
|
||||||
github.com/go-resty/resty/v2 v2.11.0
|
github.com/go-resty/resty/v2 v2.11.0
|
||||||
github.com/gookit/color v1.5.0
|
github.com/gookit/color v1.5.0
|
||||||
github.com/gookit/goutil v0.4.6
|
github.com/gookit/goutil v0.4.6
|
||||||
@@ -17,10 +19,10 @@ require (
|
|||||||
github.com/satori/go.uuid v1.2.0
|
github.com/satori/go.uuid v1.2.0
|
||||||
github.com/shengdoushi/base58 v1.0.0
|
github.com/shengdoushi/base58 v1.0.0
|
||||||
github.com/shopspring/decimal v1.3.1
|
github.com/shopspring/decimal v1.3.1
|
||||||
github.com/spf13/cobra v1.2.1
|
github.com/spf13/cobra v1.9.1
|
||||||
github.com/spf13/viper v1.9.0
|
github.com/spf13/viper v1.20.1
|
||||||
github.com/tidwall/gjson v1.18.0
|
github.com/tidwall/gjson v1.18.0
|
||||||
go.uber.org/zap v1.17.0
|
go.uber.org/zap v1.27.0
|
||||||
gopkg.in/telebot.v3 v3.0.0
|
gopkg.in/telebot.v3 v3.0.0
|
||||||
gorm.io/driver/mysql v1.5.1
|
gorm.io/driver/mysql v1.5.1
|
||||||
gorm.io/driver/postgres v1.5.2
|
gorm.io/driver/postgres v1.5.2
|
||||||
@@ -29,54 +31,85 @@ require (
|
|||||||
)
|
)
|
||||||
|
|
||||||
require (
|
require (
|
||||||
|
filippo.io/edwards25519 v1.1.0 // indirect
|
||||||
|
github.com/BurntSushi/toml v1.6.0 // indirect
|
||||||
|
github.com/Microsoft/go-winio v0.6.2 // indirect
|
||||||
|
github.com/StackExchange/wmi v1.2.1 // indirect
|
||||||
|
github.com/bits-and-blooms/bitset v1.20.0 // indirect
|
||||||
|
github.com/blendle/zapdriver v1.3.1 // indirect
|
||||||
|
github.com/consensys/bavard v0.1.27 // indirect
|
||||||
|
github.com/consensys/gnark-crypto v0.16.0 // indirect
|
||||||
|
github.com/crate-crypto/go-eth-kzg v1.3.0 // indirect
|
||||||
|
github.com/crate-crypto/go-ipa v0.0.0-20240724233137-53bbb0ceb27a // indirect
|
||||||
|
github.com/davecgh/go-spew v1.1.1 // indirect
|
||||||
|
github.com/deckarep/golang-set/v2 v2.6.0 // indirect
|
||||||
|
github.com/decred/dcrd/dcrec/secp256k1/v4 v4.0.1 // indirect
|
||||||
github.com/dustin/go-humanize v1.0.1 // indirect
|
github.com/dustin/go-humanize v1.0.1 // indirect
|
||||||
github.com/fsnotify/fsnotify v1.5.1 // indirect
|
github.com/ethereum/c-kzg-4844/v2 v2.1.0 // indirect
|
||||||
|
github.com/ethereum/go-verkle v0.2.2 // indirect
|
||||||
|
github.com/fatih/color v1.18.0 // indirect
|
||||||
|
github.com/fsnotify/fsnotify v1.9.0 // indirect
|
||||||
|
github.com/gagliardetto/binary v0.8.0 // indirect
|
||||||
|
github.com/gagliardetto/treeout v0.1.4 // indirect
|
||||||
|
github.com/go-ole/go-ole v1.3.0 // indirect
|
||||||
github.com/go-sql-driver/mysql v1.7.0 // indirect
|
github.com/go-sql-driver/mysql v1.7.0 // indirect
|
||||||
|
github.com/go-viper/mapstructure/v2 v2.2.1 // indirect
|
||||||
github.com/golang-jwt/jwt v3.2.2+incompatible // indirect
|
github.com/golang-jwt/jwt v3.2.2+incompatible // indirect
|
||||||
github.com/google/go-cmp v0.7.0 // indirect
|
|
||||||
github.com/google/uuid v1.6.0 // indirect
|
github.com/google/uuid v1.6.0 // indirect
|
||||||
github.com/gookit/filter v1.1.2 // indirect
|
github.com/gookit/filter v1.1.2 // indirect
|
||||||
github.com/hashicorp/hcl v1.0.0 // indirect
|
github.com/gorilla/websocket v1.5.3 // indirect
|
||||||
github.com/inconshreveable/mousetrap v1.0.0 // indirect
|
github.com/holiman/uint256 v1.3.2 // indirect
|
||||||
|
github.com/inconshreveable/mousetrap v1.1.0 // indirect
|
||||||
github.com/jackc/pgpassfile v1.0.0 // indirect
|
github.com/jackc/pgpassfile v1.0.0 // indirect
|
||||||
github.com/jackc/pgservicefile v0.0.0-20221227161230-091c0ba34f0a // indirect
|
github.com/jackc/pgservicefile v0.0.0-20221227161230-091c0ba34f0a // indirect
|
||||||
github.com/jackc/pgx/v5 v5.3.1 // indirect
|
github.com/jackc/pgx/v5 v5.3.1 // indirect
|
||||||
github.com/jinzhu/inflection v1.0.0 // indirect
|
github.com/jinzhu/inflection v1.0.0 // indirect
|
||||||
github.com/jinzhu/now v1.1.5 // indirect
|
github.com/jinzhu/now v1.1.5 // indirect
|
||||||
|
github.com/klauspost/compress v1.18.0 // indirect
|
||||||
github.com/labstack/gommon v0.3.0 // indirect
|
github.com/labstack/gommon v0.3.0 // indirect
|
||||||
github.com/magiconair/properties v1.8.5 // indirect
|
github.com/logrusorgru/aurora v2.0.3+incompatible // indirect
|
||||||
github.com/mattn/go-colorable v0.1.12 // indirect
|
github.com/mattn/go-colorable v0.1.14 // indirect
|
||||||
github.com/mattn/go-isatty v0.0.20 // indirect
|
github.com/mattn/go-isatty v0.0.20 // indirect
|
||||||
github.com/mattn/go-sqlite3 v1.14.22 // indirect
|
github.com/mattn/go-sqlite3 v1.14.22 // indirect
|
||||||
github.com/mitchellh/go-homedir v1.1.0 // indirect
|
github.com/mitchellh/go-homedir v1.1.0 // indirect
|
||||||
github.com/mitchellh/mapstructure v1.4.2 // indirect
|
github.com/mitchellh/go-testing-interface v1.14.1 // indirect
|
||||||
|
github.com/mmcloughlin/addchain v0.4.0 // indirect
|
||||||
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
|
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
|
||||||
github.com/modern-go/reflect2 v1.0.2 // indirect
|
github.com/modern-go/reflect2 v1.0.2 // indirect
|
||||||
|
github.com/mostynb/zstdpool-freelist v0.0.0-20201229113212-927304c0c3b1 // indirect
|
||||||
|
github.com/mr-tron/base58 v1.2.0 // indirect
|
||||||
github.com/ncruces/go-strftime v1.0.0 // indirect
|
github.com/ncruces/go-strftime v1.0.0 // indirect
|
||||||
github.com/pelletier/go-toml v1.9.4 // indirect
|
github.com/pelletier/go-toml/v2 v2.2.4 // indirect
|
||||||
github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec // indirect
|
github.com/remyoudompheng/bigfft v0.0.0-20230129092748-24d4a6f8daec // indirect
|
||||||
github.com/spf13/afero v1.6.0 // indirect
|
github.com/sagikazarmark/locafero v0.9.0 // indirect
|
||||||
|
github.com/shirou/gopsutil v3.21.4-0.20210419000835-c7a38de76ee5+incompatible // indirect
|
||||||
|
github.com/sourcegraph/conc v0.3.0 // indirect
|
||||||
|
github.com/spf13/afero v1.14.0 // indirect
|
||||||
github.com/spf13/cast v1.10.0 // indirect
|
github.com/spf13/cast v1.10.0 // indirect
|
||||||
github.com/spf13/jwalterweatherman v1.1.0 // indirect
|
github.com/spf13/pflag v1.0.6 // indirect
|
||||||
github.com/spf13/pflag v1.0.5 // indirect
|
github.com/streamingfast/logging v0.0.0-20250404134358-92b15d2fbd2e // indirect
|
||||||
github.com/subosito/gotenv v1.2.0 // indirect
|
github.com/subosito/gotenv v1.6.0 // indirect
|
||||||
|
github.com/supranational/blst v0.3.14 // indirect
|
||||||
github.com/tidwall/match v1.2.0 // indirect
|
github.com/tidwall/match v1.2.0 // indirect
|
||||||
github.com/tidwall/pretty v1.2.1 // indirect
|
github.com/tidwall/pretty v1.2.1 // indirect
|
||||||
|
github.com/tklauser/go-sysconf v0.3.12 // indirect
|
||||||
|
github.com/tklauser/numcpus v0.6.1 // indirect
|
||||||
github.com/valyala/bytebufferpool v1.0.0 // indirect
|
github.com/valyala/bytebufferpool v1.0.0 // indirect
|
||||||
github.com/valyala/fasttemplate v1.2.1 // indirect
|
github.com/valyala/fasttemplate v1.2.1 // indirect
|
||||||
github.com/xo/terminfo v0.0.0-20210125001918-ca9a967f8778 // indirect
|
github.com/xo/terminfo v0.0.0-20210125001918-ca9a967f8778 // indirect
|
||||||
go.uber.org/atomic v1.7.0 // indirect
|
go.mongodb.org/mongo-driver v1.17.3 // indirect
|
||||||
go.uber.org/multierr v1.6.0 // indirect
|
go.uber.org/multierr v1.11.0 // indirect
|
||||||
golang.org/x/crypto v0.14.0 // indirect
|
golang.org/x/crypto v0.47.0 // indirect
|
||||||
golang.org/x/net v0.17.0 // indirect
|
golang.org/x/net v0.49.0 // indirect
|
||||||
|
golang.org/x/sync v0.19.0 // indirect
|
||||||
golang.org/x/sys v0.42.0 // indirect
|
golang.org/x/sys v0.42.0 // indirect
|
||||||
golang.org/x/text v0.20.0 // indirect
|
golang.org/x/term v0.39.0 // indirect
|
||||||
|
golang.org/x/text v0.33.0 // indirect
|
||||||
golang.org/x/time v0.14.0 // indirect
|
golang.org/x/time v0.14.0 // indirect
|
||||||
gopkg.in/ini.v1 v1.63.2 // indirect
|
gopkg.in/yaml.v3 v3.0.1 // indirect
|
||||||
gopkg.in/natefinch/lumberjack.v2 v2.0.0 // indirect
|
|
||||||
gopkg.in/yaml.v2 v2.4.0 // indirect
|
|
||||||
modernc.org/libc v1.70.0 // indirect
|
modernc.org/libc v1.70.0 // indirect
|
||||||
modernc.org/mathutil v1.7.1 // indirect
|
modernc.org/mathutil v1.7.1 // indirect
|
||||||
modernc.org/memory v1.11.0 // indirect
|
modernc.org/memory v1.11.0 // indirect
|
||||||
modernc.org/sqlite v1.47.0 // indirect
|
modernc.org/sqlite v1.47.0 // indirect
|
||||||
|
rsc.io/tmplfunc v0.0.3 // indirect
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -2,12 +2,13 @@ package middleware
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"bytes"
|
"bytes"
|
||||||
|
"io/ioutil"
|
||||||
|
|
||||||
"github.com/assimon/luuu/config"
|
"github.com/assimon/luuu/config"
|
||||||
"github.com/assimon/luuu/util/constant"
|
"github.com/assimon/luuu/util/constant"
|
||||||
"github.com/assimon/luuu/util/json"
|
"github.com/assimon/luuu/util/json"
|
||||||
"github.com/assimon/luuu/util/sign"
|
"github.com/assimon/luuu/util/sign"
|
||||||
"github.com/labstack/echo/v4"
|
"github.com/labstack/echo/v4"
|
||||||
"io/ioutil"
|
|
||||||
)
|
)
|
||||||
|
|
||||||
func CheckApiSign() echo.MiddlewareFunc {
|
func CheckApiSign() echo.MiddlewareFunc {
|
||||||
|
|||||||
@@ -0,0 +1,24 @@
|
|||||||
|
package middleware
|
||||||
|
|
||||||
|
import (
|
||||||
|
"github.com/assimon/luuu/config"
|
||||||
|
"github.com/assimon/luuu/util/constant"
|
||||||
|
"github.com/labstack/echo/v4"
|
||||||
|
)
|
||||||
|
|
||||||
|
// CheckApiToken validates the Authorization header against the configured api_auth_token.
|
||||||
|
// Use this for management APIs (GET/POST) where body-based signature is not practical.
|
||||||
|
func CheckApiToken() echo.MiddlewareFunc {
|
||||||
|
return func(next echo.HandlerFunc) echo.HandlerFunc {
|
||||||
|
return func(ctx echo.Context) error {
|
||||||
|
token := ctx.Request().Header.Get("Authorization")
|
||||||
|
if token == "" {
|
||||||
|
token = ctx.QueryParam("api_token")
|
||||||
|
}
|
||||||
|
if token == "" || token != config.GetApiAuthToken() {
|
||||||
|
return constant.SignatureErr
|
||||||
|
}
|
||||||
|
return next(ctx)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -22,5 +22,39 @@ func MdbTableInit() {
|
|||||||
// panic(err)
|
// panic(err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
if err := Mdb.AutoMigrate(&mdb.SupportedAsset{}); err != nil {
|
||||||
|
color.Red.Printf("[store_db] AutoMigrate DB(SupportedAsset),err=%s\n", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
seedSupportedAssets()
|
||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func seedSupportedAssets() {
|
||||||
|
var cnt int64
|
||||||
|
if err := Mdb.Model(&mdb.SupportedAsset{}).Count(&cnt).Error; err != nil {
|
||||||
|
color.Red.Printf("[store_db] Count SupportedAsset,err=%s\n", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if cnt > 0 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defaults := []mdb.SupportedAsset{
|
||||||
|
{Network: mdb.NetworkTron, Token: "TRX", Status: mdb.TokenStatusEnable},
|
||||||
|
{Network: mdb.NetworkTron, Token: "USDT", Status: mdb.TokenStatusEnable},
|
||||||
|
{Network: mdb.NetworkSolana, Token: "SOL", Status: mdb.TokenStatusEnable},
|
||||||
|
{Network: mdb.NetworkSolana, Token: "USDT", Status: mdb.TokenStatusEnable},
|
||||||
|
{Network: mdb.NetworkSolana, Token: "USDC", Status: mdb.TokenStatusEnable},
|
||||||
|
{Network: mdb.NetworkEthereum, Token: "USDT", Status: mdb.TokenStatusEnable},
|
||||||
|
{Network: mdb.NetworkEthereum, Token: "USDC", Status: mdb.TokenStatusEnable},
|
||||||
|
{Network: mdb.NetworkBsc, Token: "USDT", Status: mdb.TokenStatusEnable},
|
||||||
|
{Network: mdb.NetworkBsc, Token: "USDC", Status: mdb.TokenStatusEnable},
|
||||||
|
{Network: mdb.NetworkPolygon, Token: "USDT", Status: mdb.TokenStatusEnable},
|
||||||
|
{Network: mdb.NetworkPolygon, Token: "USDC", Status: mdb.TokenStatusEnable},
|
||||||
|
{Network: mdb.NetworkPlasma, Token: "USDT", Status: mdb.TokenStatusEnable},
|
||||||
|
}
|
||||||
|
if err := Mdb.Create(&defaults).Error; err != nil {
|
||||||
|
color.Red.Printf("[store_db] Seed SupportedAsset,err=%s\n", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -41,6 +41,9 @@ func RuntimeInit() error {
|
|||||||
if err = RuntimeDB.Exec("DROP INDEX IF EXISTS transaction_lock_token_amount_uindex").Error; err != nil {
|
if err = RuntimeDB.Exec("DROP INDEX IF EXISTS transaction_lock_token_amount_uindex").Error; err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
if err = RuntimeDB.Exec("DROP INDEX IF EXISTS transaction_lock_address_token_amount_uindex").Error; err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
if err = RuntimeDB.AutoMigrate(&mdb.TransactionLock{}); err != nil {
|
if err = RuntimeDB.AutoMigrate(&mdb.TransactionLock{}); err != nil {
|
||||||
color.Red.Printf("[runtime_db] sqlite migrate DB(TransactionLock),err=%s\n", err)
|
color.Red.Printf("[runtime_db] sqlite migrate DB(TransactionLock),err=%s\n", err)
|
||||||
return err
|
return err
|
||||||
|
|||||||
@@ -108,11 +108,103 @@ func UpdateOrderIsExpirationById(id uint64, expirationCutoff time.Time) (bool, e
|
|||||||
return result.RowsAffected > 0, result.Error
|
return result.RowsAffected > 0, result.Error
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetTradeIdByWalletAddressAndAmountAndToken resolves the reserved trade id by address, token and amount.
|
// CountActiveSubOrders counts sub-orders with status=WaitPay under a parent.
|
||||||
func GetTradeIdByWalletAddressAndAmountAndToken(address string, token string, amount float64) (string, error) {
|
func CountActiveSubOrders(parentTradeId string) (int64, error) {
|
||||||
|
var count int64
|
||||||
|
err := dao.Mdb.Model(&mdb.Orders{}).
|
||||||
|
Where("parent_trade_id = ?", parentTradeId).
|
||||||
|
Where("status = ?", mdb.StatusWaitPay).
|
||||||
|
Count(&count).Error
|
||||||
|
return count, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetSubOrderByTokenNetwork finds an existing active sub-order matching token+network under a parent.
|
||||||
|
func GetSubOrderByTokenNetwork(parentTradeId string, token string, network string) (*mdb.Orders, error) {
|
||||||
|
order := new(mdb.Orders)
|
||||||
|
err := dao.Mdb.Model(order).
|
||||||
|
Where("parent_trade_id = ?", parentTradeId).
|
||||||
|
Where("token = ?", token).
|
||||||
|
Where("network = ?", network).
|
||||||
|
Where("status = ?", mdb.StatusWaitPay).
|
||||||
|
Limit(1).
|
||||||
|
Find(order).Error
|
||||||
|
return order, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetSiblingSubOrders returns active sub-orders under the same parent, excluding the given trade_id.
|
||||||
|
func GetSiblingSubOrders(parentTradeId string, excludeTradeId string) ([]mdb.Orders, error) {
|
||||||
|
var orders []mdb.Orders
|
||||||
|
err := dao.Mdb.Model(&mdb.Orders{}).
|
||||||
|
Where("parent_trade_id = ?", parentTradeId).
|
||||||
|
Where("trade_id != ?", excludeTradeId).
|
||||||
|
Where("status = ?", mdb.StatusWaitPay).
|
||||||
|
Find(&orders).Error
|
||||||
|
return orders, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// MarkParentOrderSuccess updates the parent order with the sub-order's payment details.
|
||||||
|
// Token and network are NOT overwritten — the parent keeps its original values.
|
||||||
|
func MarkParentOrderSuccess(parentTradeId string, sub *mdb.Orders) (bool, error) {
|
||||||
|
result := dao.Mdb.Model(&mdb.Orders{}).
|
||||||
|
Where("trade_id = ?", parentTradeId).
|
||||||
|
Where("status = ?", mdb.StatusWaitPay).
|
||||||
|
Updates(map[string]interface{}{
|
||||||
|
"status": mdb.StatusPaySuccess,
|
||||||
|
"block_transaction_id": sub.BlockTransactionId,
|
||||||
|
"callback_confirm": mdb.CallBackConfirmNo,
|
||||||
|
"actual_amount": sub.ActualAmount,
|
||||||
|
"receive_address": sub.ReceiveAddress,
|
||||||
|
})
|
||||||
|
return result.RowsAffected > 0, result.Error
|
||||||
|
}
|
||||||
|
|
||||||
|
// MarkOrderSelected sets is_selected=true for the given trade_id.
|
||||||
|
func MarkOrderSelected(tradeId string) error {
|
||||||
|
return dao.Mdb.Model(&mdb.Orders{}).
|
||||||
|
Where("trade_id = ?", tradeId).
|
||||||
|
Update("is_selected", true).Error
|
||||||
|
}
|
||||||
|
|
||||||
|
// RefreshOrderExpiration resets created_at to now so the expiration timer restarts.
|
||||||
|
// Called on the parent order when a sub-order is created or returned.
|
||||||
|
func RefreshOrderExpiration(tradeId string) error {
|
||||||
|
return dao.Mdb.Model(&mdb.Orders{}).
|
||||||
|
Where("trade_id = ?", tradeId).
|
||||||
|
Update("created_at", time.Now()).Error
|
||||||
|
}
|
||||||
|
|
||||||
|
// ResetCallbackConfirmOk sets callback_confirm back to Ok.
|
||||||
|
// Prevents the callback worker from retrying a sub-order with an empty notify_url.
|
||||||
|
func ResetCallbackConfirmOk(tradeId string) error {
|
||||||
|
return dao.Mdb.Model(&mdb.Orders{}).
|
||||||
|
Where("trade_id = ?", tradeId).
|
||||||
|
Update("callback_confirm", mdb.CallBackConfirmOk).Error
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetActiveSubOrders returns all active sub-orders under a parent.
|
||||||
|
func GetActiveSubOrders(parentTradeId string) ([]mdb.Orders, error) {
|
||||||
|
var orders []mdb.Orders
|
||||||
|
err := dao.Mdb.Model(&mdb.Orders{}).
|
||||||
|
Where("parent_trade_id = ?", parentTradeId).
|
||||||
|
Where("status = ?", mdb.StatusWaitPay).
|
||||||
|
Find(&orders).Error
|
||||||
|
return orders, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// ExpireOrderByTradeId marks a single order as expired if still waiting.
|
||||||
|
func ExpireOrderByTradeId(tradeId string) error {
|
||||||
|
return dao.Mdb.Model(&mdb.Orders{}).
|
||||||
|
Where("trade_id = ?", tradeId).
|
||||||
|
Where("status = ?", mdb.StatusWaitPay).
|
||||||
|
Update("status", mdb.StatusExpired).Error
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetTradeIdByWalletAddressAndAmountAndToken resolves the reserved trade id by network, address, token and amount.
|
||||||
|
func GetTradeIdByWalletAddressAndAmountAndToken(network string, address string, token string, amount float64) (string, error) {
|
||||||
scaledAmount, _ := normalizeLockAmount(amount)
|
scaledAmount, _ := normalizeLockAmount(amount)
|
||||||
var lock mdb.TransactionLock
|
var lock mdb.TransactionLock
|
||||||
err := dao.RuntimeDB.Model(&mdb.TransactionLock{}).
|
err := dao.RuntimeDB.Model(&mdb.TransactionLock{}).
|
||||||
|
Where("network = ?", network).
|
||||||
Where("address = ?", address).
|
Where("address = ?", address).
|
||||||
Where("token = ?", normalizeLockToken(token)).
|
Where("token = ?", normalizeLockToken(token)).
|
||||||
Where("amount_scaled = ?", scaledAmount).
|
Where("amount_scaled = ?", scaledAmount).
|
||||||
@@ -128,12 +220,13 @@ func GetTradeIdByWalletAddressAndAmountAndToken(address string, token string, am
|
|||||||
return lock.TradeId, nil
|
return lock.TradeId, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// LockTransaction reserves an address+token+amount pair in sqlite until expiration.
|
// LockTransaction reserves a network+address+token+amount pair in sqlite until expiration.
|
||||||
func LockTransaction(address, token, tradeID string, amount float64, expirationTime time.Duration) error {
|
func LockTransaction(network, address, token, tradeID string, amount float64, expirationTime time.Duration) error {
|
||||||
scaledAmount, amountText := normalizeLockAmount(amount)
|
scaledAmount, amountText := normalizeLockAmount(amount)
|
||||||
normalizedToken := normalizeLockToken(token)
|
normalizedToken := normalizeLockToken(token)
|
||||||
now := time.Now()
|
now := time.Now()
|
||||||
lock := &mdb.TransactionLock{
|
lock := &mdb.TransactionLock{
|
||||||
|
Network: network,
|
||||||
Address: address,
|
Address: address,
|
||||||
Token: normalizedToken,
|
Token: normalizedToken,
|
||||||
AmountScaled: scaledAmount,
|
AmountScaled: scaledAmount,
|
||||||
@@ -143,7 +236,8 @@ func LockTransaction(address, token, tradeID string, amount float64, expirationT
|
|||||||
}
|
}
|
||||||
|
|
||||||
return dao.RuntimeDB.Transaction(func(tx *gorm.DB) error {
|
return dao.RuntimeDB.Transaction(func(tx *gorm.DB) error {
|
||||||
if err := tx.Where("address = ?", address).
|
if err := tx.Where("network = ?", network).
|
||||||
|
Where("address = ?", address).
|
||||||
Where("token = ?", normalizedToken).
|
Where("token = ?", normalizedToken).
|
||||||
Where("amount_scaled = ?", scaledAmount).
|
Where("amount_scaled = ?", scaledAmount).
|
||||||
Where("expires_at <= ?", now).
|
Where("expires_at <= ?", now).
|
||||||
@@ -165,10 +259,11 @@ func LockTransaction(address, token, tradeID string, amount float64, expirationT
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
// UnLockTransaction releases the reservation for address+token+amount.
|
// UnLockTransaction releases the reservation for network+address+token+amount.
|
||||||
func UnLockTransaction(address string, token string, amount float64) error {
|
func UnLockTransaction(network string, address string, token string, amount float64) error {
|
||||||
scaledAmount, _ := normalizeLockAmount(amount)
|
scaledAmount, _ := normalizeLockAmount(amount)
|
||||||
return dao.RuntimeDB.
|
return dao.RuntimeDB.
|
||||||
|
Where("network = ?", network).
|
||||||
Where("address = ?", address).
|
Where("address = ?", address).
|
||||||
Where("token = ?", normalizeLockToken(token)).
|
Where("token = ?", normalizeLockToken(token)).
|
||||||
Where("amount_scaled = ?", scaledAmount).
|
Where("amount_scaled = ?", scaledAmount).
|
||||||
|
|||||||
@@ -0,0 +1,123 @@
|
|||||||
|
package data
|
||||||
|
|
||||||
|
import (
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/assimon/luuu/model/dao"
|
||||||
|
"github.com/assimon/luuu/model/mdb"
|
||||||
|
"github.com/assimon/luuu/util/constant"
|
||||||
|
)
|
||||||
|
|
||||||
|
func normalizeNetworkToken(network, token string) (string, string) {
|
||||||
|
return strings.ToLower(strings.TrimSpace(network)), strings.ToUpper(strings.TrimSpace(token))
|
||||||
|
}
|
||||||
|
|
||||||
|
func AddSupportedAsset(network, token string, status int64) (*mdb.SupportedAsset, error) {
|
||||||
|
network, token = normalizeNetworkToken(network, token)
|
||||||
|
exist, err := GetSupportedAssetByNetworkAndToken(network, token)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if exist.ID > 0 {
|
||||||
|
return nil, constant.SupportedAssetAlreadyExists
|
||||||
|
}
|
||||||
|
|
||||||
|
deleted, err := getSupportedAssetByNetworkAndTokenUnscoped(network, token)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if deleted.ID > 0 && deleted.DeletedAt.Valid {
|
||||||
|
deleted.Status = status
|
||||||
|
deleted.DeletedAt.Valid = false
|
||||||
|
if err := dao.Mdb.Unscoped().Save(deleted).Error; err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return deleted, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
asset := &mdb.SupportedAsset{
|
||||||
|
Network: network,
|
||||||
|
Token: token,
|
||||||
|
Status: status,
|
||||||
|
}
|
||||||
|
err = dao.Mdb.Create(asset).Error
|
||||||
|
return asset, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func getSupportedAssetByNetworkAndTokenUnscoped(network, token string) (*mdb.SupportedAsset, error) {
|
||||||
|
network, token = normalizeNetworkToken(network, token)
|
||||||
|
asset := new(mdb.SupportedAsset)
|
||||||
|
err := dao.Mdb.Unscoped().Model(asset).
|
||||||
|
Where("network = ?", network).
|
||||||
|
Where("token = ?", token).
|
||||||
|
Limit(1).Find(asset).Error
|
||||||
|
return asset, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetSupportedAssetByNetworkAndToken(network, token string) (*mdb.SupportedAsset, error) {
|
||||||
|
network, token = normalizeNetworkToken(network, token)
|
||||||
|
asset := new(mdb.SupportedAsset)
|
||||||
|
err := dao.Mdb.Model(asset).
|
||||||
|
Where("network = ?", network).
|
||||||
|
Where("token = ?", token).
|
||||||
|
Limit(1).Find(asset).Error
|
||||||
|
return asset, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func GetSupportedAssetByID(id uint64) (*mdb.SupportedAsset, error) {
|
||||||
|
asset := new(mdb.SupportedAsset)
|
||||||
|
err := dao.Mdb.Model(asset).Limit(1).Find(asset, id).Error
|
||||||
|
return asset, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func UpdateSupportedAsset(id uint64, network, token string, status int64) (*mdb.SupportedAsset, error) {
|
||||||
|
network, token = normalizeNetworkToken(network, token)
|
||||||
|
|
||||||
|
asset, err := GetSupportedAssetByID(id)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if asset.ID <= 0 {
|
||||||
|
return nil, constant.SupportedAssetNotFound
|
||||||
|
}
|
||||||
|
|
||||||
|
exist, err := GetSupportedAssetByNetworkAndToken(network, token)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if exist.ID > 0 && exist.ID != id {
|
||||||
|
return nil, constant.SupportedAssetAlreadyExists
|
||||||
|
}
|
||||||
|
|
||||||
|
asset.Network = network
|
||||||
|
asset.Token = token
|
||||||
|
asset.Status = status
|
||||||
|
if err := dao.Mdb.Save(asset).Error; err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return asset, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func DeleteSupportedAssetByID(id uint64) error {
|
||||||
|
return dao.Mdb.Where("id = ?", id).Delete(&mdb.SupportedAsset{}).Error
|
||||||
|
}
|
||||||
|
|
||||||
|
func ListSupportedAssets(network string) ([]mdb.SupportedAsset, error) {
|
||||||
|
network = strings.ToLower(strings.TrimSpace(network))
|
||||||
|
var list []mdb.SupportedAsset
|
||||||
|
tx := dao.Mdb.Model(&mdb.SupportedAsset{})
|
||||||
|
if network != "" {
|
||||||
|
tx = tx.Where("network = ?", network)
|
||||||
|
}
|
||||||
|
err := tx.Order("network asc, token asc").Find(&list).Error
|
||||||
|
return list, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func ListEnabledSupportedAssets() ([]mdb.SupportedAsset, error) {
|
||||||
|
var list []mdb.SupportedAsset
|
||||||
|
err := dao.Mdb.Model(&mdb.SupportedAsset{}).
|
||||||
|
Where("status = ?", mdb.TokenStatusEnable).
|
||||||
|
Order("network asc, token asc").
|
||||||
|
Find(&list).Error
|
||||||
|
return list, err
|
||||||
|
}
|
||||||
@@ -1,14 +1,38 @@
|
|||||||
package data
|
package data
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"strings"
|
||||||
|
|
||||||
"github.com/assimon/luuu/model/dao"
|
"github.com/assimon/luuu/model/dao"
|
||||||
"github.com/assimon/luuu/model/mdb"
|
"github.com/assimon/luuu/model/mdb"
|
||||||
"github.com/assimon/luuu/util/constant"
|
"github.com/assimon/luuu/util/constant"
|
||||||
)
|
)
|
||||||
|
|
||||||
// AddWalletAddress 创建钱包
|
// AddWalletAddress 创建钱包 (默认 tron 网络,用于 Telegram 添加)
|
||||||
func AddWalletAddress(address string) (*mdb.WalletAddress, error) {
|
func AddWalletAddress(address string) (*mdb.WalletAddress, error) {
|
||||||
exist, err := GetWalletAddressByToken(address)
|
return AddWalletAddressWithNetwork(mdb.NetworkTron, address)
|
||||||
|
}
|
||||||
|
|
||||||
|
// isEVMNetwork 判断是否是 EVM 网络
|
||||||
|
func isEVMNetwork(network string) bool {
|
||||||
|
switch network {
|
||||||
|
case mdb.NetworkEthereum, mdb.NetworkBsc, mdb.NetworkPolygon, mdb.NetworkPlasma:
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// AddWalletAddressWithNetwork 创建指定网络的钱包地址
|
||||||
|
func AddWalletAddressWithNetwork(network, address string) (*mdb.WalletAddress, error) {
|
||||||
|
network = strings.ToLower(strings.TrimSpace(network))
|
||||||
|
address = strings.TrimSpace(address)
|
||||||
|
|
||||||
|
// evm 网络地址统一小写,tron 和 solana 保持原样
|
||||||
|
if isEVMNetwork(network) {
|
||||||
|
address = strings.ToLower(address)
|
||||||
|
}
|
||||||
|
|
||||||
|
exist, err := GetWalletAddressByNetworkAndAddress(network, address)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@@ -16,6 +40,7 @@ func AddWalletAddress(address string) (*mdb.WalletAddress, error) {
|
|||||||
return nil, constant.WalletAddressAlreadyExists
|
return nil, constant.WalletAddressAlreadyExists
|
||||||
}
|
}
|
||||||
walletAddress := &mdb.WalletAddress{
|
walletAddress := &mdb.WalletAddress{
|
||||||
|
Network: network,
|
||||||
Address: address,
|
Address: address,
|
||||||
Status: mdb.TokenStatusEnable,
|
Status: mdb.TokenStatusEnable,
|
||||||
}
|
}
|
||||||
@@ -23,7 +48,17 @@ func AddWalletAddress(address string) (*mdb.WalletAddress, error) {
|
|||||||
return walletAddress, err
|
return walletAddress, err
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetWalletAddressByToken 通过钱包地址获取address
|
// GetWalletAddressByNetworkAndAddress 通过网络和地址查询
|
||||||
|
func GetWalletAddressByNetworkAndAddress(network, address string) (*mdb.WalletAddress, error) {
|
||||||
|
walletAddress := new(mdb.WalletAddress)
|
||||||
|
err := dao.Mdb.Model(walletAddress).
|
||||||
|
Where("network = ?", network).
|
||||||
|
Where("address = ?", address).
|
||||||
|
Limit(1).Find(walletAddress).Error
|
||||||
|
return walletAddress, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// GetWalletAddressByToken 通过钱包地址获取address (兼容旧接口)
|
||||||
func GetWalletAddressByToken(address string) (*mdb.WalletAddress, error) {
|
func GetWalletAddressByToken(address string) (*mdb.WalletAddress, error) {
|
||||||
walletAddress := new(mdb.WalletAddress)
|
walletAddress := new(mdb.WalletAddress)
|
||||||
err := dao.Mdb.Model(walletAddress).Limit(1).Find(walletAddress, "address = ?", address).Error
|
err := dao.Mdb.Model(walletAddress).Limit(1).Find(walletAddress, "address = ?", address).Error
|
||||||
@@ -50,6 +85,16 @@ func GetAvailableWalletAddress() ([]mdb.WalletAddress, error) {
|
|||||||
return WalletAddressList, err
|
return WalletAddressList, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// GetAvailableWalletAddressByNetwork 获得指定网络的所有可用钱包地址
|
||||||
|
func GetAvailableWalletAddressByNetwork(network string) ([]mdb.WalletAddress, error) {
|
||||||
|
var list []mdb.WalletAddress
|
||||||
|
err := dao.Mdb.Model(list).
|
||||||
|
Where("status = ?", mdb.TokenStatusEnable).
|
||||||
|
Where("network = ?", network).
|
||||||
|
Find(&list).Error
|
||||||
|
return list, err
|
||||||
|
}
|
||||||
|
|
||||||
// GetAllWalletAddress 获得所有钱包地址
|
// GetAllWalletAddress 获得所有钱包地址
|
||||||
func GetAllWalletAddress() ([]mdb.WalletAddress, error) {
|
func GetAllWalletAddress() ([]mdb.WalletAddress, error) {
|
||||||
var WalletAddressList []mdb.WalletAddress
|
var WalletAddressList []mdb.WalletAddress
|
||||||
@@ -57,6 +102,13 @@ func GetAllWalletAddress() ([]mdb.WalletAddress, error) {
|
|||||||
return WalletAddressList, err
|
return WalletAddressList, err
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// GetAllWalletAddressByNetwork 获得指定网络的所有钱包地址
|
||||||
|
func GetAllWalletAddressByNetwork(network string) ([]mdb.WalletAddress, error) {
|
||||||
|
var list []mdb.WalletAddress
|
||||||
|
err := dao.Mdb.Model(list).Where("network = ?", network).Find(&list).Error
|
||||||
|
return list, err
|
||||||
|
}
|
||||||
|
|
||||||
// ChangeWalletAddressStatus 启用禁用钱包
|
// ChangeWalletAddressStatus 启用禁用钱包
|
||||||
func ChangeWalletAddressStatus(id uint64, status int) error {
|
func ChangeWalletAddressStatus(id uint64, status int) error {
|
||||||
err := dao.Mdb.Model(&mdb.WalletAddress{}).Where("id = ?", id).Update("status", status).Error
|
err := dao.Mdb.Model(&mdb.WalletAddress{}).Where("id = ?", id).Update("status", status).Error
|
||||||
|
|||||||
@@ -8,20 +8,29 @@ const (
|
|||||||
CallBackConfirmNo = 2
|
CallBackConfirmNo = 2
|
||||||
)
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
PaymentTypeEpay = "Epay"
|
||||||
|
)
|
||||||
|
|
||||||
type Orders struct {
|
type Orders struct {
|
||||||
TradeId string `gorm:"column:trade_id;uniqueIndex:orders_trade_id_uindex" json:"trade_id"`
|
TradeId string `gorm:"column:trade_id;uniqueIndex:orders_trade_id_uindex" json:"trade_id"`
|
||||||
OrderId string `gorm:"column:order_id;uniqueIndex:orders_order_id_uindex" json:"order_id"`
|
OrderId string `gorm:"column:order_id;uniqueIndex:orders_order_id_uindex" json:"order_id"` // the order id is generated by client, and will notify client when order is paid
|
||||||
|
ParentTradeId string `gorm:"column:parent_trade_id;index:idx_orders_parent_trade_id;default:''" json:"parent_trade_id"`
|
||||||
BlockTransactionId string `gorm:"index:orders_block_transaction_id_index;column:block_transaction_id" json:"block_transaction_id"`
|
BlockTransactionId string `gorm:"index:orders_block_transaction_id_index;column:block_transaction_id" json:"block_transaction_id"`
|
||||||
Amount float64 `gorm:"column:amount" json:"amount"`
|
Amount float64 `gorm:"column:amount" json:"amount"`
|
||||||
Currency string `gorm:"column:currency" json:"currency"`
|
Currency string `gorm:"column:currency" json:"currency"`
|
||||||
ActualAmount float64 `gorm:"column:actual_amount" json:"actual_amount"`
|
ActualAmount float64 `gorm:"column:actual_amount" json:"actual_amount"`
|
||||||
ReceiveAddress string `gorm:"column:receive_address" json:"receive_address"`
|
ReceiveAddress string `gorm:"column:receive_address" json:"receive_address"`
|
||||||
Token string `gorm:"column:token" json:"token"`
|
Token string `gorm:"column:token" json:"token"`
|
||||||
|
Network string `gorm:"column:network" json:"network"`
|
||||||
Status int `gorm:"column:status;default:1" json:"status"`
|
Status int `gorm:"column:status;default:1" json:"status"`
|
||||||
NotifyUrl string `gorm:"column:notify_url" json:"notify_url"`
|
NotifyUrl string `gorm:"column:notify_url" json:"notify_url"`
|
||||||
RedirectUrl string `gorm:"column:redirect_url" json:"redirect_url"`
|
RedirectUrl string `gorm:"column:redirect_url" json:"redirect_url"`
|
||||||
|
Name string `gorm:"column:name" json:"name"`
|
||||||
CallbackNum int `gorm:"column:callback_num;default:0" json:"callback_num"`
|
CallbackNum int `gorm:"column:callback_num;default:0" json:"callback_num"`
|
||||||
CallBackConfirm int `gorm:"column:callback_confirm;default:2" json:"callback_confirm"`
|
CallBackConfirm int `gorm:"column:callback_confirm;default:2" json:"callback_confirm"`
|
||||||
|
IsSelected bool `gorm:"column:is_selected;default:false" json:"is_selected"`
|
||||||
|
PaymentType string `gorm:"column:payment_type" json:"payment_type"`
|
||||||
BaseModel
|
BaseModel
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -0,0 +1,12 @@
|
|||||||
|
package mdb
|
||||||
|
|
||||||
|
type SupportedAsset struct {
|
||||||
|
Network string `gorm:"column:network;uniqueIndex:supported_asset_network_token_uindex,priority:1" json:"network"`
|
||||||
|
Token string `gorm:"column:token;uniqueIndex:supported_asset_network_token_uindex,priority:2" json:"token"`
|
||||||
|
Status int64 `gorm:"column:status;default:1" json:"status"`
|
||||||
|
BaseModel
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *SupportedAsset) TableName() string {
|
||||||
|
return "supported_asset"
|
||||||
|
}
|
||||||
@@ -4,9 +4,10 @@ import "time"
|
|||||||
|
|
||||||
type TransactionLock struct {
|
type TransactionLock struct {
|
||||||
ID uint64 `gorm:"column:id;primaryKey;autoIncrement" json:"id"`
|
ID uint64 `gorm:"column:id;primaryKey;autoIncrement" json:"id"`
|
||||||
Address string `gorm:"column:address;uniqueIndex:transaction_lock_address_token_amount_uindex,priority:1" json:"address"`
|
Network string `gorm:"column:network;uniqueIndex:transaction_lock_network_address_token_amount_uindex,priority:1" json:"network"`
|
||||||
Token string `gorm:"column:token;uniqueIndex:transaction_lock_address_token_amount_uindex,priority:2" json:"token"`
|
Address string `gorm:"column:address;uniqueIndex:transaction_lock_network_address_token_amount_uindex,priority:2" json:"address"`
|
||||||
AmountScaled int64 `gorm:"column:amount_scaled;uniqueIndex:transaction_lock_address_token_amount_uindex,priority:3" json:"amount_scaled"`
|
Token string `gorm:"column:token;uniqueIndex:transaction_lock_network_address_token_amount_uindex,priority:3" json:"token"`
|
||||||
|
AmountScaled int64 `gorm:"column:amount_scaled;uniqueIndex:transaction_lock_network_address_token_amount_uindex,priority:4" json:"amount_scaled"`
|
||||||
AmountText string `gorm:"column:amount_text" json:"amount_text"`
|
AmountText string `gorm:"column:amount_text" json:"amount_text"`
|
||||||
TradeId string `gorm:"column:trade_id;index:transaction_lock_trade_id_index" json:"trade_id"`
|
TradeId string `gorm:"column:trade_id;index:transaction_lock_trade_id_index" json:"trade_id"`
|
||||||
ExpiresAt time.Time `gorm:"column:expires_at;index:transaction_lock_expires_at_index" json:"expires_at"`
|
ExpiresAt time.Time `gorm:"column:expires_at;index:transaction_lock_expires_at_index" json:"expires_at"`
|
||||||
|
|||||||
@@ -5,8 +5,18 @@ const (
|
|||||||
TokenStatusDisable = 2
|
TokenStatusDisable = 2
|
||||||
)
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
NetworkTron = "tron"
|
||||||
|
NetworkSolana = "solana"
|
||||||
|
NetworkEthereum = "ethereum"
|
||||||
|
NetworkBsc = "bsc"
|
||||||
|
NetworkPolygon = "polygon"
|
||||||
|
NetworkPlasma = "plasma"
|
||||||
|
)
|
||||||
|
|
||||||
type WalletAddress struct {
|
type WalletAddress struct {
|
||||||
Address string `gorm:"column:address;uniqueIndex:wallet_address_address_uindex" json:"address"`
|
Network string `gorm:"column:network;uniqueIndex:wallet_address_network_address_uindex" json:"network"`
|
||||||
|
Address string `gorm:"column:address;uniqueIndex:wallet_address_network_address_uindex" json:"address"`
|
||||||
Status int64 `gorm:"column:status;default:1" json:"status"`
|
Status int64 `gorm:"column:status;default:1" json:"status"`
|
||||||
BaseModel
|
BaseModel
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -5,12 +5,15 @@ import "github.com/gookit/validate"
|
|||||||
// CreateTransactionRequest 创建交易请求
|
// CreateTransactionRequest 创建交易请求
|
||||||
type CreateTransactionRequest struct {
|
type CreateTransactionRequest struct {
|
||||||
OrderId string `json:"order_id" validate:"required|maxLen:32"`
|
OrderId string `json:"order_id" validate:"required|maxLen:32"`
|
||||||
Currency string `json:"currency" validate:"required"`
|
Currency string `json:"currency" validate:"required"` // 法币 如:cny
|
||||||
Token string `json:"token" validate:"required"`
|
Token string `json:"token" validate:"required"` // 币种 如:usdt
|
||||||
|
Network string `json:"network" validate:"required"` // 网络 如:TRON
|
||||||
Amount float64 `json:"amount" validate:"required|isFloat|gt:0.01"`
|
Amount float64 `json:"amount" validate:"required|isFloat|gt:0.01"`
|
||||||
NotifyUrl string `json:"notify_url" validate:"required"`
|
NotifyUrl string `json:"notify_url" validate:"required"`
|
||||||
Signature string `json:"signature" validate:"required"`
|
Signature string `json:"signature" validate:"required"`
|
||||||
RedirectUrl string `json:"redirect_url"`
|
RedirectUrl string `json:"redirect_url"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
PaymentType string `json:"payment_type"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func (r CreateTransactionRequest) Translates() map[string]string {
|
func (r CreateTransactionRequest) Translates() map[string]string {
|
||||||
@@ -18,6 +21,7 @@ func (r CreateTransactionRequest) Translates() map[string]string {
|
|||||||
"OrderId": "订单号",
|
"OrderId": "订单号",
|
||||||
"Currency": "货币",
|
"Currency": "货币",
|
||||||
"Token": "币种",
|
"Token": "币种",
|
||||||
|
"Network": "网络",
|
||||||
"Amount": "支付金额",
|
"Amount": "支付金额",
|
||||||
"NotifyUrl": "异步回调网址",
|
"NotifyUrl": "异步回调网址",
|
||||||
"Signature": "签名",
|
"Signature": "签名",
|
||||||
@@ -29,7 +33,23 @@ type OrderProcessingRequest struct {
|
|||||||
ReceiveAddress string
|
ReceiveAddress string
|
||||||
Currency string
|
Currency string
|
||||||
Token string
|
Token string
|
||||||
|
Network string
|
||||||
Amount float64
|
Amount float64
|
||||||
TradeId string
|
TradeId string
|
||||||
BlockTransactionId string
|
BlockTransactionId string
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// SwitchNetworkRequest 切换支付网络
|
||||||
|
type SwitchNetworkRequest struct {
|
||||||
|
TradeId string `json:"trade_id" validate:"required"`
|
||||||
|
Token string `json:"token" validate:"required"`
|
||||||
|
Network string `json:"network" validate:"required"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r SwitchNetworkRequest) Translates() map[string]string {
|
||||||
|
return validate.MS{
|
||||||
|
"TradeId": "订单号",
|
||||||
|
"Token": "币种",
|
||||||
|
"Network": "网络",
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -25,3 +25,16 @@ type OrderNotifyResponse struct {
|
|||||||
Signature string `json:"signature"` // 签名
|
Signature string `json:"signature"` // 签名
|
||||||
Status int `json:"status"` // 1:等待支付,2:支付成功,3:已过期
|
Status int `json:"status"` // 1:等待支付,2:支付成功,3:已过期
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// OrderNotifyResponseEpay epay订单异步回调结构体
|
||||||
|
type OrderNotifyResponseEpay struct {
|
||||||
|
PID int `json:"pid"` // 商户ID
|
||||||
|
TradeNo string `json:"trade_no"` // 平台订单号
|
||||||
|
OutTradeNo string `json:"out_trade_no"` // 商户订单号
|
||||||
|
Type string `json:"type"` // 订单类型
|
||||||
|
Name string `json:"name"` // 商品名称
|
||||||
|
Money string `json:"money"` // 订单金额,保留4位小数
|
||||||
|
Sign string `json:"sign"` // 签名
|
||||||
|
SignType string `json:"sign_type"` // 签名类型 // MD5
|
||||||
|
TradeStatus string `json:"trade_status"` // 订单状态 // only has "TRADE_SUCCESS" for now
|
||||||
|
}
|
||||||
|
|||||||
@@ -2,11 +2,16 @@ package response
|
|||||||
|
|
||||||
type CheckoutCounterResponse struct {
|
type CheckoutCounterResponse struct {
|
||||||
TradeId string `json:"trade_id"` // epusdt订单号
|
TradeId string `json:"trade_id"` // epusdt订单号
|
||||||
ActualAmount float64 `json:"actual_amount"` // 订单实际需要支付的金额,保留4位小数
|
Amount float64 `json:"amount"` // 订单金额,保留4位小数 法币金额
|
||||||
ReceiveAddress string `json:"receive_address"` // 收款钱包地址
|
ActualAmount float64 `json:"actual_amount"` // 订单实际需要支付的金额,保留4位小数 加密货币金额
|
||||||
Token string `json:"token"` // 所属币种 TRX USDT......
|
Token string `json:"token"` // 所属币种 TRX USDT......
|
||||||
|
Currency string `json:"currency"` // 法币币种 CNY USD ...
|
||||||
|
ReceiveAddress string `json:"receive_address"` // 收款钱包地址
|
||||||
|
Network string `json:"network"` // 网络 TRON ETH ...
|
||||||
ExpirationTime int64 `json:"expiration_time"` // 过期时间 时间戳
|
ExpirationTime int64 `json:"expiration_time"` // 过期时间 时间戳
|
||||||
RedirectUrl string `json:"redirect_url"`
|
RedirectUrl string `json:"redirect_url"`
|
||||||
|
CreatedAt int64 `json:"created_at"` // 订单创建时间 时间戳
|
||||||
|
IsSelected bool `json:"is_selected"`
|
||||||
}
|
}
|
||||||
|
|
||||||
type CheckStatusResponse struct {
|
type CheckStatusResponse struct {
|
||||||
|
|||||||
@@ -0,0 +1,10 @@
|
|||||||
|
package response
|
||||||
|
|
||||||
|
type NetworkTokenSupport struct {
|
||||||
|
Network string `json:"network"`
|
||||||
|
Tokens []string `json:"tokens"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type SupportedAssetsResponse struct {
|
||||||
|
Supports []NetworkTokenSupport `json:"supports"`
|
||||||
|
}
|
||||||
@@ -28,8 +28,10 @@ const (
|
|||||||
IncrementalMaximumNumber = 100
|
IncrementalMaximumNumber = 100
|
||||||
)
|
)
|
||||||
|
|
||||||
var gCreateTransactionLock sync.Mutex
|
var (
|
||||||
var gOrderProcessingLock sync.Mutex
|
gCreateTransactionLock sync.Mutex
|
||||||
|
gOrderProcessingLock sync.Mutex
|
||||||
|
)
|
||||||
|
|
||||||
// CreateTransaction creates a new payment order.
|
// CreateTransaction creates a new payment order.
|
||||||
func CreateTransaction(req *request.CreateTransactionRequest) (*response.CreateTransactionResponse, error) {
|
func CreateTransaction(req *request.CreateTransactionRequest) (*response.CreateTransactionResponse, error) {
|
||||||
@@ -38,6 +40,7 @@ func CreateTransaction(req *request.CreateTransactionRequest) (*response.CreateT
|
|||||||
|
|
||||||
token := strings.ToUpper(strings.TrimSpace(req.Token))
|
token := strings.ToUpper(strings.TrimSpace(req.Token))
|
||||||
currency := strings.ToUpper(strings.TrimSpace(req.Currency))
|
currency := strings.ToUpper(strings.TrimSpace(req.Currency))
|
||||||
|
network := strings.ToLower(strings.TrimSpace(req.Network))
|
||||||
payAmount := math.MustParsePrecFloat64(req.Amount, 2)
|
payAmount := math.MustParsePrecFloat64(req.Amount, 2)
|
||||||
rate := config.GetRateForCoin(strings.ToLower(token), strings.ToLower(currency))
|
rate := config.GetRateForCoin(strings.ToLower(token), strings.ToLower(currency))
|
||||||
if rate <= 0 {
|
if rate <= 0 {
|
||||||
@@ -61,7 +64,7 @@ func CreateTransaction(req *request.CreateTransactionRequest) (*response.CreateT
|
|||||||
return nil, constant.OrderAlreadyExists
|
return nil, constant.OrderAlreadyExists
|
||||||
}
|
}
|
||||||
|
|
||||||
walletAddress, err := data.GetAvailableWalletAddress()
|
walletAddress, err := data.GetAvailableWalletAddressByNetwork(network)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@@ -71,7 +74,7 @@ func CreateTransaction(req *request.CreateTransactionRequest) (*response.CreateT
|
|||||||
|
|
||||||
tradeID := GenerateCode()
|
tradeID := GenerateCode()
|
||||||
amount := math.MustParsePrecFloat64(decimalTokenAmount.InexactFloat64(), 2)
|
amount := math.MustParsePrecFloat64(decimalTokenAmount.InexactFloat64(), 2)
|
||||||
availableAddress, availableAmount, err := ReserveAvailableWalletAndAmount(tradeID, token, amount, walletAddress)
|
availableAddress, availableAmount, err := ReserveAvailableWalletAndAmount(tradeID, network, token, amount, walletAddress)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
@@ -88,9 +91,12 @@ func CreateTransaction(req *request.CreateTransactionRequest) (*response.CreateT
|
|||||||
ActualAmount: availableAmount,
|
ActualAmount: availableAmount,
|
||||||
ReceiveAddress: availableAddress,
|
ReceiveAddress: availableAddress,
|
||||||
Token: token,
|
Token: token,
|
||||||
|
Network: network,
|
||||||
Status: mdb.StatusWaitPay,
|
Status: mdb.StatusWaitPay,
|
||||||
NotifyUrl: req.NotifyUrl,
|
NotifyUrl: req.NotifyUrl,
|
||||||
RedirectUrl: req.RedirectUrl,
|
RedirectUrl: req.RedirectUrl,
|
||||||
|
Name: req.Name,
|
||||||
|
PaymentType: req.PaymentType,
|
||||||
}
|
}
|
||||||
if err = data.CreateOrderWithTransaction(tx, order); err != nil {
|
if err = data.CreateOrderWithTransaction(tx, order); err != nil {
|
||||||
tx.Rollback()
|
tx.Rollback()
|
||||||
@@ -148,20 +154,83 @@ func OrderProcessing(req *request.OrderProcessingRequest) error {
|
|||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|
||||||
if err = data.UnLockTransaction(req.ReceiveAddress, req.Token, req.Amount); err != nil {
|
if err = data.UnLockTransaction(req.Network, req.ReceiveAddress, req.Token, req.Amount); err != nil {
|
||||||
log.Sugar.Warnf("[order] unlock transaction after pay success failed, trade_id=%s, err=%v", req.TradeId, err)
|
log.Sugar.Warnf("[order] unlock transaction after pay success failed, trade_id=%s, err=%v", req.TradeId, err)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Load order to check parent-child relationship
|
||||||
|
order, err := data.GetOrderInfoByTradeId(req.TradeId)
|
||||||
|
if err != nil {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Parent order paid directly: expire all sub-orders and release their locks
|
||||||
|
if order.ParentTradeId == "" {
|
||||||
|
subs, subErr := data.GetActiveSubOrders(order.TradeId)
|
||||||
|
if subErr != nil {
|
||||||
|
log.Sugar.Errorf("[order] get sub-orders for parent failed, trade_id=%s, err=%v", order.TradeId, subErr)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
for _, sub := range subs {
|
||||||
|
if err = data.ExpireOrderByTradeId(sub.TradeId); err != nil {
|
||||||
|
log.Sugar.Warnf("[order] expire sub-order failed, trade_id=%s, err=%v", sub.TradeId, err)
|
||||||
|
}
|
||||||
|
if err = data.UnLockTransaction(sub.Network, sub.ReceiveAddress, sub.Token, sub.ActualAmount); err != nil {
|
||||||
|
log.Sugar.Warnf("[order] unlock sub-order transaction failed, trade_id=%s, err=%v", sub.TradeId, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Sub-order should not trigger its own callback (notify_url is empty).
|
||||||
|
// OrderSuccessWithTransaction unconditionally sets callback_confirm=No,
|
||||||
|
// reset it here to prevent the callback worker from retrying an empty URL.
|
||||||
|
if err = data.ResetCallbackConfirmOk(order.TradeId); err != nil {
|
||||||
|
log.Sugar.Warnf("[order] reset sub-order callback_confirm failed, trade_id=%s, err=%v", order.TradeId, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
parent, err := data.GetOrderInfoByTradeId(order.ParentTradeId)
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Errorf("[order] load parent order failed, parent_trade_id=%s, err=%v", order.ParentTradeId, err)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Mark parent as paid with sub-order's payment details
|
||||||
|
if _, err = data.MarkParentOrderSuccess(parent.TradeId, order); err != nil {
|
||||||
|
log.Sugar.Errorf("[order] mark parent success failed, parent_trade_id=%s, err=%v", parent.TradeId, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Release parent's own wallet lock
|
||||||
|
if err = data.UnLockTransaction(parent.Network, parent.ReceiveAddress, parent.Token, parent.ActualAmount); err != nil {
|
||||||
|
log.Sugar.Warnf("[order] unlock parent transaction failed, parent_trade_id=%s, err=%v", parent.TradeId, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Expire sibling sub-orders and release their locks
|
||||||
|
siblings, err := data.GetSiblingSubOrders(parent.TradeId, order.TradeId)
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Errorf("[order] get sibling sub-orders failed, parent_trade_id=%s, err=%v", parent.TradeId, err)
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
for _, sib := range siblings {
|
||||||
|
if err = data.ExpireOrderByTradeId(sib.TradeId); err != nil {
|
||||||
|
log.Sugar.Warnf("[order] expire sibling failed, trade_id=%s, err=%v", sib.TradeId, err)
|
||||||
|
}
|
||||||
|
if err = data.UnLockTransaction(sib.Network, sib.ReceiveAddress, sib.Token, sib.ActualAmount); err != nil {
|
||||||
|
log.Sugar.Warnf("[order] unlock sibling transaction failed, trade_id=%s, err=%v", sib.TradeId, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
// ReserveAvailableWalletAndAmount finds and locks an address+token+amount pair.
|
// ReserveAvailableWalletAndAmount finds and locks a network+address+token+amount pair.
|
||||||
func ReserveAvailableWalletAndAmount(tradeID string, token string, amount float64, walletAddress []mdb.WalletAddress) (string, float64, error) {
|
func ReserveAvailableWalletAndAmount(tradeID string, network string, token string, amount float64, walletAddress []mdb.WalletAddress) (string, float64, error) {
|
||||||
availableAddress := ""
|
availableAddress := ""
|
||||||
availableAmount := amount
|
availableAmount := amount
|
||||||
|
|
||||||
tryLockWalletFunc := func(targetAmount float64) (string, error) {
|
tryLockWalletFunc := func(targetAmount float64) (string, error) {
|
||||||
for _, address := range walletAddress {
|
for _, address := range walletAddress {
|
||||||
err := data.LockTransaction(address.Address, token, tradeID, targetAmount, config.GetOrderExpirationTimeDuration())
|
err := data.LockTransaction(network, address.Address, token, tradeID, targetAmount, config.GetOrderExpirationTimeDuration())
|
||||||
if err == nil {
|
if err == nil {
|
||||||
return address.Address, nil
|
return address.Address, nil
|
||||||
}
|
}
|
||||||
@@ -208,3 +277,141 @@ func GetOrderInfoByTradeId(tradeId string) (*mdb.Orders, error) {
|
|||||||
}
|
}
|
||||||
return order, nil
|
return order, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
const MaxSubOrders = 2
|
||||||
|
|
||||||
|
// SwitchNetwork creates or returns an existing sub-order for a different token+network.
|
||||||
|
func SwitchNetwork(req *request.SwitchNetworkRequest) (*response.CheckoutCounterResponse, error) {
|
||||||
|
gCreateTransactionLock.Lock()
|
||||||
|
defer gCreateTransactionLock.Unlock()
|
||||||
|
|
||||||
|
token := strings.ToUpper(strings.TrimSpace(req.Token))
|
||||||
|
network := strings.ToLower(strings.TrimSpace(req.Network))
|
||||||
|
|
||||||
|
// 1. Load parent order
|
||||||
|
parent, err := data.GetOrderInfoByTradeId(req.TradeId)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if parent.ID <= 0 {
|
||||||
|
return nil, constant.OrderNotExists
|
||||||
|
}
|
||||||
|
if parent.ParentTradeId != "" {
|
||||||
|
return nil, constant.CannotSwitchSubOrder
|
||||||
|
}
|
||||||
|
if parent.Status != mdb.StatusWaitPay {
|
||||||
|
return nil, constant.OrderNotWaitPay
|
||||||
|
}
|
||||||
|
|
||||||
|
// 2. Same token+network as parent → mark selected and return
|
||||||
|
if strings.EqualFold(parent.Token, token) && strings.EqualFold(parent.Network, network) {
|
||||||
|
_ = data.MarkOrderSelected(parent.TradeId)
|
||||||
|
parent.IsSelected = true
|
||||||
|
return buildCheckoutResponse(parent), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// 3. Existing active sub-order for this token+network → return it
|
||||||
|
existing, err := data.GetSubOrderByTokenNetwork(parent.TradeId, token, network)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if existing.ID > 0 {
|
||||||
|
_ = data.MarkOrderSelected(parent.TradeId)
|
||||||
|
_ = data.MarkOrderSelected(existing.TradeId)
|
||||||
|
_ = data.RefreshOrderExpiration(parent.TradeId)
|
||||||
|
existing.IsSelected = true
|
||||||
|
return buildCheckoutResponse(existing), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// 4. Check sub-order limit
|
||||||
|
count, err := data.CountActiveSubOrders(parent.TradeId)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if count >= MaxSubOrders {
|
||||||
|
return nil, constant.SubOrderLimitExceeded
|
||||||
|
}
|
||||||
|
|
||||||
|
// 5. Calculate amount for the new network
|
||||||
|
rate := config.GetRateForCoin(strings.ToLower(token), strings.ToLower(parent.Currency))
|
||||||
|
if rate <= 0 {
|
||||||
|
return nil, constant.RateAmountErr
|
||||||
|
}
|
||||||
|
decimalPayAmount := decimal.NewFromFloat(parent.Amount)
|
||||||
|
decimalTokenAmount := decimalPayAmount.Mul(decimal.NewFromFloat(rate))
|
||||||
|
if decimalTokenAmount.Cmp(decimal.NewFromFloat(UsdtMinimumPaymentAmount)) == -1 {
|
||||||
|
return nil, constant.PayAmountErr
|
||||||
|
}
|
||||||
|
|
||||||
|
// 6. Find and lock wallet
|
||||||
|
walletAddress, err := data.GetAvailableWalletAddressByNetwork(network)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if len(walletAddress) <= 0 {
|
||||||
|
return nil, constant.NotAvailableWalletAddress
|
||||||
|
}
|
||||||
|
|
||||||
|
subTradeID := GenerateCode()
|
||||||
|
amount := math.MustParsePrecFloat64(decimalTokenAmount.InexactFloat64(), 2)
|
||||||
|
availableAddress, availableAmount, err := ReserveAvailableWalletAndAmount(subTradeID, network, token, amount, walletAddress)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if availableAddress == "" {
|
||||||
|
return nil, constant.NotAvailableAmountErr
|
||||||
|
}
|
||||||
|
|
||||||
|
// 7. Create sub-order
|
||||||
|
tx := dao.Mdb.Begin()
|
||||||
|
subOrder := &mdb.Orders{
|
||||||
|
TradeId: subTradeID,
|
||||||
|
OrderId: subTradeID, // sub-order uses its own trade_id as order_id (unique constraint)
|
||||||
|
ParentTradeId: parent.TradeId,
|
||||||
|
Amount: parent.Amount,
|
||||||
|
Currency: parent.Currency,
|
||||||
|
ActualAmount: availableAmount,
|
||||||
|
ReceiveAddress: availableAddress,
|
||||||
|
Token: token,
|
||||||
|
Network: network,
|
||||||
|
Status: mdb.StatusWaitPay,
|
||||||
|
IsSelected: true,
|
||||||
|
NotifyUrl: "",
|
||||||
|
RedirectUrl: parent.RedirectUrl,
|
||||||
|
Name: parent.Name,
|
||||||
|
CallBackConfirm: mdb.CallBackConfirmOk, // don't trigger callback on sub-order
|
||||||
|
PaymentType: parent.PaymentType,
|
||||||
|
}
|
||||||
|
if err = data.CreateOrderWithTransaction(tx, subOrder); err != nil {
|
||||||
|
tx.Rollback()
|
||||||
|
_ = data.UnLockTransactionByTradeId(subTradeID)
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if err = tx.Commit().Error; err != nil {
|
||||||
|
tx.Rollback()
|
||||||
|
_ = data.UnLockTransactionByTradeId(subTradeID)
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// Mark parent as selected and refresh its expiration to match the sub-order
|
||||||
|
_ = data.MarkOrderSelected(parent.TradeId)
|
||||||
|
_ = data.RefreshOrderExpiration(parent.TradeId)
|
||||||
|
|
||||||
|
return buildCheckoutResponse(subOrder), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func buildCheckoutResponse(order *mdb.Orders) *response.CheckoutCounterResponse {
|
||||||
|
return &response.CheckoutCounterResponse{
|
||||||
|
TradeId: order.TradeId,
|
||||||
|
Amount: order.Amount,
|
||||||
|
ActualAmount: order.ActualAmount,
|
||||||
|
Token: order.Token,
|
||||||
|
Currency: order.Currency,
|
||||||
|
ReceiveAddress: order.ReceiveAddress,
|
||||||
|
Network: order.Network,
|
||||||
|
ExpirationTime: order.CreatedAt.AddMinutes(config.GetOrderExpirationTime()).TimestampMilli(),
|
||||||
|
RedirectUrl: order.RedirectUrl,
|
||||||
|
CreatedAt: order.CreatedAt.TimestampMilli(),
|
||||||
|
IsSelected: order.IsSelected,
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -18,6 +18,7 @@ func newCreateTransactionRequest(orderID string, amount float64) *request.Create
|
|||||||
OrderId: orderID,
|
OrderId: orderID,
|
||||||
Currency: "CNY",
|
Currency: "CNY",
|
||||||
Token: "USDT",
|
Token: "USDT",
|
||||||
|
Network: "tron",
|
||||||
Amount: amount,
|
Amount: amount,
|
||||||
NotifyUrl: "https://merchant.example/callback",
|
NotifyUrl: "https://merchant.example/callback",
|
||||||
}
|
}
|
||||||
@@ -53,7 +54,7 @@ func TestCreateTransactionAssignsIncrementedAmountsAndLocks(t *testing.T) {
|
|||||||
t.Fatalf("unexpected tokens: %s, %s", resp1.Token, resp2.Token)
|
t.Fatalf("unexpected tokens: %s, %s", resp1.Token, resp2.Token)
|
||||||
}
|
}
|
||||||
|
|
||||||
tradeID1, err := data.GetTradeIdByWalletAddressAndAmountAndToken(resp1.ReceiveAddress, resp1.Token, resp1.ActualAmount)
|
tradeID1, err := data.GetTradeIdByWalletAddressAndAmountAndToken("tron", resp1.ReceiveAddress, resp1.Token, resp1.ActualAmount)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("get first runtime lock: %v", err)
|
t.Fatalf("get first runtime lock: %v", err)
|
||||||
}
|
}
|
||||||
@@ -61,7 +62,7 @@ func TestCreateTransactionAssignsIncrementedAmountsAndLocks(t *testing.T) {
|
|||||||
t.Fatalf("first runtime lock = %s, want %s", tradeID1, resp1.TradeId)
|
t.Fatalf("first runtime lock = %s, want %s", tradeID1, resp1.TradeId)
|
||||||
}
|
}
|
||||||
|
|
||||||
tradeID2, err := data.GetTradeIdByWalletAddressAndAmountAndToken(resp2.ReceiveAddress, resp2.Token, resp2.ActualAmount)
|
tradeID2, err := data.GetTradeIdByWalletAddressAndAmountAndToken("tron", resp2.ReceiveAddress, resp2.Token, resp2.ActualAmount)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("get second runtime lock: %v", err)
|
t.Fatalf("get second runtime lock: %v", err)
|
||||||
}
|
}
|
||||||
@@ -86,6 +87,7 @@ func TestOrderProcessingMarksPaidAndReleasesLock(t *testing.T) {
|
|||||||
err = OrderProcessing(&request.OrderProcessingRequest{
|
err = OrderProcessing(&request.OrderProcessingRequest{
|
||||||
ReceiveAddress: resp.ReceiveAddress,
|
ReceiveAddress: resp.ReceiveAddress,
|
||||||
Token: resp.Token,
|
Token: resp.Token,
|
||||||
|
Network: "tron",
|
||||||
TradeId: resp.TradeId,
|
TradeId: resp.TradeId,
|
||||||
Amount: resp.ActualAmount,
|
Amount: resp.ActualAmount,
|
||||||
BlockTransactionId: "block_1",
|
BlockTransactionId: "block_1",
|
||||||
@@ -108,7 +110,7 @@ func TestOrderProcessingMarksPaidAndReleasesLock(t *testing.T) {
|
|||||||
t.Fatalf("block transaction id = %s, want block_1", order.BlockTransactionId)
|
t.Fatalf("block transaction id = %s, want block_1", order.BlockTransactionId)
|
||||||
}
|
}
|
||||||
|
|
||||||
tradeID, err := data.GetTradeIdByWalletAddressAndAmountAndToken(resp.ReceiveAddress, resp.Token, resp.ActualAmount)
|
tradeID, err := data.GetTradeIdByWalletAddressAndAmountAndToken("tron", resp.ReceiveAddress, resp.Token, resp.ActualAmount)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("get runtime lock after processing: %v", err)
|
t.Fatalf("get runtime lock after processing: %v", err)
|
||||||
}
|
}
|
||||||
@@ -133,6 +135,7 @@ func TestOrderProcessingRejectsDuplicateBlockForSameOrder(t *testing.T) {
|
|||||||
req := &request.OrderProcessingRequest{
|
req := &request.OrderProcessingRequest{
|
||||||
ReceiveAddress: resp.ReceiveAddress,
|
ReceiveAddress: resp.ReceiveAddress,
|
||||||
Token: resp.Token,
|
Token: resp.Token,
|
||||||
|
Network: "tron",
|
||||||
TradeId: resp.TradeId,
|
TradeId: resp.TradeId,
|
||||||
Amount: resp.ActualAmount,
|
Amount: resp.ActualAmount,
|
||||||
BlockTransactionId: "block_1",
|
BlockTransactionId: "block_1",
|
||||||
@@ -180,6 +183,7 @@ func TestOrderProcessingDoesNotReviveExpiredOrder(t *testing.T) {
|
|||||||
err = OrderProcessing(&request.OrderProcessingRequest{
|
err = OrderProcessing(&request.OrderProcessingRequest{
|
||||||
ReceiveAddress: resp.ReceiveAddress,
|
ReceiveAddress: resp.ReceiveAddress,
|
||||||
Token: resp.Token,
|
Token: resp.Token,
|
||||||
|
Network: "tron",
|
||||||
TradeId: resp.TradeId,
|
TradeId: resp.TradeId,
|
||||||
Amount: resp.ActualAmount,
|
Amount: resp.ActualAmount,
|
||||||
BlockTransactionId: "block_expired",
|
BlockTransactionId: "block_expired",
|
||||||
@@ -239,6 +243,7 @@ func TestOrderProcessingOnlyOneOrderClaimsABlockTransaction(t *testing.T) {
|
|||||||
errs <- OrderProcessing(&request.OrderProcessingRequest{
|
errs <- OrderProcessing(&request.OrderProcessingRequest{
|
||||||
ReceiveAddress: address,
|
ReceiveAddress: address,
|
||||||
Token: token,
|
Token: token,
|
||||||
|
Network: "tron",
|
||||||
TradeId: tradeID,
|
TradeId: tradeID,
|
||||||
Amount: amount,
|
Amount: amount,
|
||||||
BlockTransactionId: "shared_block",
|
BlockTransactionId: "shared_block",
|
||||||
|
|||||||
@@ -9,6 +9,8 @@ import (
|
|||||||
"github.com/assimon/luuu/model/response"
|
"github.com/assimon/luuu/model/response"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
var ErrOrder = errors.New("不存在待支付订单或已过期")
|
||||||
|
|
||||||
// GetCheckoutCounterByTradeId returns checkout info for a pending order.
|
// GetCheckoutCounterByTradeId returns checkout info for a pending order.
|
||||||
func GetCheckoutCounterByTradeId(tradeId string) (*response.CheckoutCounterResponse, error) {
|
func GetCheckoutCounterByTradeId(tradeId string) (*response.CheckoutCounterResponse, error) {
|
||||||
orderInfo, err := data.GetOrderInfoByTradeId(tradeId)
|
orderInfo, err := data.GetOrderInfoByTradeId(tradeId)
|
||||||
@@ -16,16 +18,21 @@ func GetCheckoutCounterByTradeId(tradeId string) (*response.CheckoutCounterRespo
|
|||||||
return nil, err
|
return nil, err
|
||||||
}
|
}
|
||||||
if orderInfo.ID <= 0 || orderInfo.Status != mdb.StatusWaitPay {
|
if orderInfo.ID <= 0 || orderInfo.Status != mdb.StatusWaitPay {
|
||||||
return nil, errors.New("不存在待支付订单或已过期")
|
return nil, ErrOrder
|
||||||
}
|
}
|
||||||
|
|
||||||
resp := &response.CheckoutCounterResponse{
|
resp := &response.CheckoutCounterResponse{
|
||||||
TradeId: orderInfo.TradeId,
|
TradeId: orderInfo.TradeId,
|
||||||
|
Amount: orderInfo.Amount,
|
||||||
ActualAmount: orderInfo.ActualAmount,
|
ActualAmount: orderInfo.ActualAmount,
|
||||||
ReceiveAddress: orderInfo.ReceiveAddress,
|
|
||||||
Token: orderInfo.Token,
|
Token: orderInfo.Token,
|
||||||
|
Currency: orderInfo.Currency,
|
||||||
|
ReceiveAddress: orderInfo.ReceiveAddress,
|
||||||
|
Network: orderInfo.Network,
|
||||||
ExpirationTime: orderInfo.CreatedAt.AddMinutes(config.GetOrderExpirationTime()).TimestampMilli(),
|
ExpirationTime: orderInfo.CreatedAt.AddMinutes(config.GetOrderExpirationTime()).TimestampMilli(),
|
||||||
RedirectUrl: orderInfo.RedirectUrl,
|
RedirectUrl: orderInfo.RedirectUrl,
|
||||||
|
CreatedAt: orderInfo.CreatedAt.TimestampMilli(),
|
||||||
|
IsSelected: orderInfo.IsSelected,
|
||||||
}
|
}
|
||||||
return resp, nil
|
return resp, nil
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,609 @@
|
|||||||
|
package service
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"math/big"
|
||||||
|
"sort"
|
||||||
|
"strings"
|
||||||
|
"sync"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/assimon/luuu/config"
|
||||||
|
"github.com/assimon/luuu/model/data"
|
||||||
|
"github.com/assimon/luuu/model/mdb"
|
||||||
|
"github.com/assimon/luuu/model/request"
|
||||||
|
"github.com/assimon/luuu/util/constant"
|
||||||
|
"github.com/assimon/luuu/util/log"
|
||||||
|
"github.com/assimon/luuu/util/math"
|
||||||
|
"github.com/gagliardetto/solana-go"
|
||||||
|
"github.com/go-resty/resty/v2"
|
||||||
|
"github.com/shopspring/decimal"
|
||||||
|
"github.com/tidwall/gjson"
|
||||||
|
)
|
||||||
|
|
||||||
|
// gProcessedSignatures 已处理签名缓存,避免重复调用 getTransaction
|
||||||
|
var gProcessedSignatures sync.Map // sig -> unix timestamp
|
||||||
|
|
||||||
|
type TransferInfo struct {
|
||||||
|
Source string // Source address (for SOL) or source ATA (for SPL tokens)
|
||||||
|
Destination string // Destination address (for SOL) or destination ATA (for SPL tokens)
|
||||||
|
Mint string // Token mint (e.g. USDT mint, USDC mint, or "SOL" for native transfers)
|
||||||
|
Amount float64 // Human-readable amount (adjusted for decimals)
|
||||||
|
RawAmount uint64 // Raw amount from the transaction (before adjusting for decimals)
|
||||||
|
Decimals *int // Optional decimals from the transaction, if available
|
||||||
|
BlockTime int64 // Block time of the transfer unit is seconds since epoch
|
||||||
|
}
|
||||||
|
|
||||||
|
// SolCallBack 扫描指定钱包地址的 Solana 链上交易,匹配待支付订单并确认收款。
|
||||||
|
func SolCallBack(address string, wg *sync.WaitGroup) {
|
||||||
|
defer wg.Done()
|
||||||
|
defer func() {
|
||||||
|
if err := recover(); err != nil {
|
||||||
|
log.Sugar.Errorf("[SOL][%s] panic recovered: %v", address, err)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
// Clean up old entries from processed cache (older than 1 hour)
|
||||||
|
cleanupCutoff := time.Now().Add(-1 * time.Hour).Unix()
|
||||||
|
gProcessedSignatures.Range(func(key, value interface{}) bool {
|
||||||
|
if ts, ok := value.(int64); ok && ts < cleanupCutoff {
|
||||||
|
gProcessedSignatures.Delete(key)
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
})
|
||||||
|
|
||||||
|
limit := 1000
|
||||||
|
|
||||||
|
// 查询钱包地址 + USDT ATA + USDC ATA 三个地址的签名
|
||||||
|
queryAddrs := []string{address}
|
||||||
|
if usdtAta, err := FindATAAddress(address, USDT_Mint); err == nil {
|
||||||
|
queryAddrs = append(queryAddrs, usdtAta)
|
||||||
|
} else {
|
||||||
|
log.Sugar.Errorf("[SOL][%s] failed to derive USDT ATA: %v", address, err)
|
||||||
|
}
|
||||||
|
if usdcAta, err := FindATAAddress(address, USDC_Mint); err == nil {
|
||||||
|
queryAddrs = append(queryAddrs, usdcAta)
|
||||||
|
} else {
|
||||||
|
log.Sugar.Errorf("[SOL][%s] failed to derive USDC ATA: %v", address, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 拉取签名并去重
|
||||||
|
seen := make(map[string]bool)
|
||||||
|
var result []solSignatureResult
|
||||||
|
for _, queryAddr := range queryAddrs {
|
||||||
|
respBody, err := SolGetSignaturesForAddress(queryAddr, limit, "", "")
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Errorf("[SOL][%s] SolGetSignaturesForAddress(%s) failed: %v", address, queryAddr, err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
resultBody := gjson.GetBytes(respBody, "result")
|
||||||
|
if !resultBody.Exists() || !resultBody.IsArray() {
|
||||||
|
log.Sugar.Errorf("[SOL][%s] unexpected response format for %s: %s", address, queryAddr, string(respBody))
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
var batch []solSignatureResult
|
||||||
|
err = json.Unmarshal([]byte(resultBody.Raw), &batch)
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Errorf("[SOL][%s] failed to unmarshal signatures for %s: %v", address, queryAddr, err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, sig := range batch {
|
||||||
|
if !seen[sig.Signature] {
|
||||||
|
seen[sig.Signature] = true
|
||||||
|
result = append(result, sig)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if len(result) == 0 {
|
||||||
|
log.Sugar.Debugf("[SOL][%s] no transaction signatures found", address)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
// 按 blockTime 降序排列
|
||||||
|
sort.Slice(result, func(i, j int) bool {
|
||||||
|
if result[i].BlockTime == nil {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
if result[j].BlockTime == nil {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
return *result[i].BlockTime > *result[j].BlockTime
|
||||||
|
})
|
||||||
|
|
||||||
|
// 时间截止线:订单过期时间 + 5 分钟
|
||||||
|
cutoffTime := time.Now().Add(-config.GetOrderExpirationTimeDuration() - 5*time.Minute).Unix()
|
||||||
|
|
||||||
|
log.Sugar.Debugf("[SOL][%s] fetched %d unique signatures from %d addresses, cutoff=%d",
|
||||||
|
address, len(result), len(queryAddrs), cutoffTime)
|
||||||
|
|
||||||
|
// Process each transaction signature
|
||||||
|
for sigIdx, txSig := range result {
|
||||||
|
sig := txSig.Signature
|
||||||
|
|
||||||
|
// Skip failed transactions
|
||||||
|
if txSig.Err != nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
// 超过截止时间的旧签名不再处理
|
||||||
|
if txSig.BlockTime != nil && *txSig.BlockTime < cutoffTime {
|
||||||
|
log.Sugar.Debugf("[SOL][%s] [%d/%d] sig=%s blockTime=%d before cutoff=%d, stopping scan",
|
||||||
|
address, sigIdx+1, len(result), sig, *txSig.BlockTime, cutoffTime)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
|
||||||
|
// 跳过已处理的签名
|
||||||
|
if _, ok := gProcessedSignatures.Load(sig); ok {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Sugar.Debugf("[SOL][%s] [%d/%d] processing sig=%s slot=%d", address, sigIdx+1, len(result), sig, txSig.Slot)
|
||||||
|
|
||||||
|
txData, err := SolGetTransaction(sig)
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Debugf("[SOL][%s] sig=%s fetch failed: %v", address, sig, err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
instructions := gjson.GetBytes(txData, "result.transaction.message.instructions").Array()
|
||||||
|
log.Sugar.Debugf("[SOL][%s] sig=%s has %d instructions", address, sig, len(instructions))
|
||||||
|
|
||||||
|
for instrIdx, instruction := range instructions {
|
||||||
|
programID := instruction.Get("programId").String()
|
||||||
|
parsedType := instruction.Get("parsed.type").String()
|
||||||
|
log.Sugar.Debugf("[SOL][%s] sig=%s instr[%d] programId=%s parsedType=%s", address, sig, instrIdx, programID, parsedType)
|
||||||
|
|
||||||
|
transferInfo, err := ParseTransferInfoFromInstruction(instruction, txData)
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Debugf("[SOL][%s] sig=%s instr[%d] parse error: %v", address, sig, instrIdx, err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if transferInfo == nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Sugar.Debugf("[SOL][%s] sig=%s instr[%d] transfer: src=%s dst=%s mint=%s rawAmount=%d amount=%.6f blockTime=%d",
|
||||||
|
address, sig, instrIdx,
|
||||||
|
transferInfo.Source, transferInfo.Destination, transferInfo.Mint,
|
||||||
|
transferInfo.RawAmount, transferInfo.Amount, transferInfo.BlockTime)
|
||||||
|
|
||||||
|
if !isTransferToAddress(transferInfo, address) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
token, amount := getTokenTypeAndAmount(transferInfo)
|
||||||
|
if token == "" || amount <= 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Sugar.Infof("[SOL][%s] sig=%s instr[%d] incoming transfer confirmed: token=%s amount=%.2f -> querying transaction_lock network=solana address=%s token=%s amount=%.2f",
|
||||||
|
address, sig, instrIdx, token, amount, address, token, amount)
|
||||||
|
|
||||||
|
tradeID, err := data.GetTradeIdByWalletAddressAndAmountAndToken(mdb.NetworkSolana, address, token, amount)
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Errorf("[SOL][%s] sig=%s query transaction_lock failed: %v", address, sig, err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if tradeID == "" {
|
||||||
|
log.Sugar.Infof("[SOL][%s] sig=%s no active transaction_lock matched: network=solana address=%s token=%s amount=%.2f (no order or expired)",
|
||||||
|
address, sig, address, token, amount)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
log.Sugar.Infof("[SOL][%s] transaction_lock matched: trade_id=%s sig=%s token=%s amount=%.2f",
|
||||||
|
address, tradeID, sig, token, amount)
|
||||||
|
|
||||||
|
order, err := data.GetOrderInfoByTradeId(tradeID)
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Errorf("[SOL][%s] sig=%s load order failed for trade_id=%s: %v", address, sig, tradeID, err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
log.Sugar.Infof("[SOL][%s] order loaded: trade_id=%s order_id=%s status=%d created_at_ms=%d",
|
||||||
|
address, tradeID, order.OrderId, order.Status, order.CreatedAt.TimestampMilli())
|
||||||
|
|
||||||
|
// blockTime 秒 → 毫秒,与订单创建时间对齐
|
||||||
|
blockTimestamp := transferInfo.BlockTime * 1000
|
||||||
|
createTime := order.CreatedAt.TimestampMilli()
|
||||||
|
log.Sugar.Infof("[SOL][%s] time check: sig=%s block_time_ms=%d order_created_ms=%d diff_ms=%d",
|
||||||
|
address, sig, blockTimestamp, createTime, blockTimestamp-createTime)
|
||||||
|
if blockTimestamp < createTime {
|
||||||
|
log.Sugar.Warnf("[SOL][%s] sig=%s skipped: block_time_ms=%d is %d ms before order created_ms=%d (transaction predates the order)",
|
||||||
|
address, sig, blockTimestamp, createTime-blockTimestamp, createTime)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
req := &request.OrderProcessingRequest{
|
||||||
|
ReceiveAddress: address,
|
||||||
|
Token: token,
|
||||||
|
Network: mdb.NetworkSolana,
|
||||||
|
TradeId: tradeID,
|
||||||
|
Amount: amount,
|
||||||
|
BlockTransactionId: sig,
|
||||||
|
}
|
||||||
|
log.Sugar.Infof("[SOL][%s] calling OrderProcessing: trade_id=%s sig=%s token=%s amount=%.2f",
|
||||||
|
address, tradeID, sig, token, amount)
|
||||||
|
err = OrderProcessing(req)
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, constant.OrderBlockAlreadyProcess) || errors.Is(err, constant.OrderStatusConflict) {
|
||||||
|
log.Sugar.Infof("[SOL][%s] sig=%s already resolved: trade_id=%s reason=%v", address, sig, tradeID, err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
log.Sugar.Errorf("[SOL][%s] sig=%s OrderProcessing failed for trade_id=%s: %v", address, sig, tradeID, err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Sugar.Infof("[SOL][%s] order marked paid: trade_id=%s sig=%s token=%s amount=%.2f, sending telegram notification",
|
||||||
|
address, tradeID, sig, token, amount)
|
||||||
|
sendPaymentNotification(order)
|
||||||
|
log.Sugar.Infof("[SOL][%s] payment fully processed: trade_id=%s sig=%s", address, tradeID, sig)
|
||||||
|
}
|
||||||
|
|
||||||
|
// 标记已处理
|
||||||
|
gProcessedSignatures.Store(sig, time.Now().Unix())
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// solSignatureResult getSignaturesForAddress 返回结构
|
||||||
|
type solSignatureResult struct {
|
||||||
|
Signature string `json:"signature"`
|
||||||
|
Slot uint64 `json:"slot"`
|
||||||
|
Err interface{} `json:"err"`
|
||||||
|
BlockTime *int64 `json:"blockTime"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// SolRetryClient 发送 Solana JSON-RPC 请求,自动重试
|
||||||
|
func SolRetryClient(method string, params []interface{}) ([]byte, error) {
|
||||||
|
client := resty.New()
|
||||||
|
client.SetRetryCount(5)
|
||||||
|
client.SetRetryWaitTime(2 * time.Second)
|
||||||
|
client.SetRetryMaxWaitTime(10 * time.Second)
|
||||||
|
client.AddRetryCondition(func(r *resty.Response, err error) bool {
|
||||||
|
if err != nil {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
if r.StatusCode() >= 429 || r.StatusCode() >= 500 {
|
||||||
|
return true
|
||||||
|
}
|
||||||
|
return false
|
||||||
|
})
|
||||||
|
|
||||||
|
rpcUrl := config.GetSolanaRpcUrl()
|
||||||
|
|
||||||
|
resp, err := client.R().
|
||||||
|
SetHeader("Content-Type", "application/json").
|
||||||
|
SetBody(map[string]interface{}{
|
||||||
|
"jsonrpc": "2.0",
|
||||||
|
"id": 1,
|
||||||
|
"method": method,
|
||||||
|
"params": params,
|
||||||
|
}).
|
||||||
|
Post(rpcUrl)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
respBody := resp.Body()
|
||||||
|
|
||||||
|
return respBody, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func SolGetSignaturesForAddress(address string, limit int, untilSig string, beforeSig string) ([]byte, error) {
|
||||||
|
opts := map[string]interface{}{
|
||||||
|
"commitment": "finalized",
|
||||||
|
"limit": limit,
|
||||||
|
}
|
||||||
|
if untilSig != "" {
|
||||||
|
opts["until"] = untilSig
|
||||||
|
}
|
||||||
|
if beforeSig != "" {
|
||||||
|
opts["before"] = beforeSig
|
||||||
|
}
|
||||||
|
|
||||||
|
bodyData, err := SolRetryClient("getSignaturesForAddress",
|
||||||
|
[]interface{}{address, opts})
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
var result map[string]interface{}
|
||||||
|
err = json.Unmarshal(bodyData, &result)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
_, ok := result["result"].([]interface{})
|
||||||
|
if !ok {
|
||||||
|
return nil, fmt.Errorf("unexpected response format: %v", result)
|
||||||
|
}
|
||||||
|
|
||||||
|
return bodyData, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func SolGetTransaction(sig string) ([]byte, error) {
|
||||||
|
txData, err := SolRetryClient("getTransaction", []interface{}{
|
||||||
|
sig,
|
||||||
|
map[string]interface{}{
|
||||||
|
"encoding": "jsonParsed",
|
||||||
|
"commitment": "confirmed",
|
||||||
|
"maxSupportedTransactionVersion": 0, // suport
|
||||||
|
},
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Errorf("SolRetryClient failed: %v", err)
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
errField := gjson.GetBytes(txData, "result.meta.err")
|
||||||
|
if errField.Exists() && errField.Type != gjson.Null {
|
||||||
|
log.Sugar.Warnf("Transaction failed: %v", errField.String())
|
||||||
|
return nil, fmt.Errorf("transaction failed: %s", errField.String())
|
||||||
|
}
|
||||||
|
|
||||||
|
return txData, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// isTransferToAddress 判断转账目标是否为指定钱包地址
|
||||||
|
func isTransferToAddress(transfer *TransferInfo, targetAddress string) bool {
|
||||||
|
// Native SOL transfer - check destination directly
|
||||||
|
if transfer.Mint == "SOL" {
|
||||||
|
return strings.EqualFold(transfer.Destination, targetAddress)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Skip Transfer instruction without mint info (use TransferChecked instead)
|
||||||
|
if transfer.Mint == "" {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
// SPL Token transfer - check if destination ATA matches
|
||||||
|
return MatchAtaAddress(targetAddress, transfer.Mint, transfer.Destination)
|
||||||
|
}
|
||||||
|
|
||||||
|
// getTokenTypeAndAmount 根据 mint 识别代币类型并计算可读金额
|
||||||
|
func getTokenTypeAndAmount(transfer *TransferInfo) (string, float64) {
|
||||||
|
mint := transfer.Mint
|
||||||
|
|
||||||
|
// Native SOL
|
||||||
|
if mint == "SOL" {
|
||||||
|
return "SOL", transfer.Amount
|
||||||
|
}
|
||||||
|
|
||||||
|
// SPL Tokens
|
||||||
|
switch mint {
|
||||||
|
case USDT_Mint:
|
||||||
|
decimals := USDT_Decimals
|
||||||
|
if transfer.Decimals != nil {
|
||||||
|
decimals = int(*transfer.Decimals)
|
||||||
|
}
|
||||||
|
return "USDT", ADJustAmount(transfer.RawAmount, decimals)
|
||||||
|
|
||||||
|
case USDC_Mint:
|
||||||
|
decimals := USDC_Decimals
|
||||||
|
if transfer.Decimals != nil {
|
||||||
|
decimals = int(*transfer.Decimals)
|
||||||
|
}
|
||||||
|
return "USDC", ADJustAmount(transfer.RawAmount, decimals)
|
||||||
|
|
||||||
|
default:
|
||||||
|
// Unsupported token
|
||||||
|
return "", 0
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
const (
|
||||||
|
// Mint token
|
||||||
|
USDT_Mint = "Es9vMFrzaCERmJfrF4H2FYD4KCoNkY11McCe8BenwNYB"
|
||||||
|
USDC_Mint = "EPjFWdd5AufqSSqeM2qN1xzybapC8G4wEGGkZwyTDt1v"
|
||||||
|
|
||||||
|
USDT_Decimals = 6
|
||||||
|
USDC_Decimals = 6
|
||||||
|
SOL_Decimals = 9
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
// SPL Token instructions
|
||||||
|
InstructionTransfer = 3
|
||||||
|
InstructionTransferChecked = 12
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
// System Program
|
||||||
|
SystemProgramID = "11111111111111111111111111111111"
|
||||||
|
InstructionSystemTransfer = 2
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
TokenProgramID = "TokenkegQfeZyiNwAJbNbGKPFXCWuBvf9Ss623VQ5DA"
|
||||||
|
Token2022ProgramID = "TokenzQdBNbLqP5VEhdkAS6EPFLC1PHnBqCXEpPxuEb"
|
||||||
|
)
|
||||||
|
|
||||||
|
// ADJustAmount 将链上原始金额转为可读金额(除以 10^decimals,保留 2 位小数)
|
||||||
|
func ADJustAmount(amount uint64, decimals int) float64 {
|
||||||
|
if amount == 0 {
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
decimalAmount := decimal.NewFromBigInt(new(big.Int).SetUint64(amount), 0)
|
||||||
|
// 10^decimals
|
||||||
|
decimalDivisor := decimal.New(1, int32(decimals))
|
||||||
|
adjustedAmount := decimalAmount.Div(decimalDivisor)
|
||||||
|
// Round to 2 decimal places
|
||||||
|
return math.MustParsePrecFloat64(adjustedAmount.InexactFloat64(), 2)
|
||||||
|
}
|
||||||
|
|
||||||
|
func MatchUsdtAtaAddress(address string, ataTo string) bool {
|
||||||
|
ata, err := FindATAAddress(address, USDT_Mint)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("FindATAAddress failed: %v\n", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
return strings.EqualFold(ata, ataTo)
|
||||||
|
}
|
||||||
|
|
||||||
|
func MatchUsdcAtaAddress(address string, ataTo string) bool {
|
||||||
|
ata, err := FindATAAddress(address, USDC_Mint)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("FindATAAddress failed: %v\n", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
return strings.EqualFold(ata, ataTo)
|
||||||
|
}
|
||||||
|
|
||||||
|
func MatchAtaAddress(address string, mint string, ataTo string) bool {
|
||||||
|
ata, err := FindATAAddress(address, mint)
|
||||||
|
if err != nil {
|
||||||
|
fmt.Printf("FindATAAddress failed: %v\n", err)
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
|
||||||
|
return strings.EqualFold(ata, ataTo)
|
||||||
|
}
|
||||||
|
|
||||||
|
func FindATAAddress(owner, mint string) (string, error) {
|
||||||
|
ownerPubKey, err := solana.PublicKeyFromBase58(owner)
|
||||||
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("invalid owner public key: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
mintPubKey, err := solana.PublicKeyFromBase58(mint)
|
||||||
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("invalid mint public key: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
ata, _, err := solana.FindAssociatedTokenAddress(ownerPubKey, mintPubKey)
|
||||||
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("find associated token address failed: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return ata.String(), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ParseTransferInfoFromInstruction 从单条指令中解析转账信息,非转账指令返回 nil
|
||||||
|
func ParseTransferInfoFromInstruction(instruction gjson.Result, txData []byte) (*TransferInfo, error) {
|
||||||
|
programID := instruction.Get("programId").String()
|
||||||
|
parsedType := instruction.Get("parsed.type").String()
|
||||||
|
|
||||||
|
if programID == SystemProgramID && parsedType == "transfer" {
|
||||||
|
return parseSystemTransfer(instruction, txData)
|
||||||
|
}
|
||||||
|
|
||||||
|
if programID == TokenProgramID || programID == Token2022ProgramID {
|
||||||
|
switch parsedType {
|
||||||
|
case "transfer":
|
||||||
|
return parseSplTransfer(instruction, txData)
|
||||||
|
case "transferChecked":
|
||||||
|
return parseSplTransferChecked(instruction, txData)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Skip non-transfer instructions (ComputeBudget, AToken create, etc.)
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func parseSystemTransfer(instruction gjson.Result, txData []byte) (*TransferInfo, error) {
|
||||||
|
info := instruction.Get("parsed.info")
|
||||||
|
source := info.Get("source").String()
|
||||||
|
destination := info.Get("destination").String()
|
||||||
|
lamports := info.Get("lamports").Uint()
|
||||||
|
blockTime := gjson.GetBytes(txData, "result.blockTime").Int()
|
||||||
|
|
||||||
|
return &TransferInfo{
|
||||||
|
Source: source,
|
||||||
|
Destination: destination,
|
||||||
|
Mint: "SOL",
|
||||||
|
Amount: ADJustAmount(lamports, SOL_Decimals),
|
||||||
|
RawAmount: lamports,
|
||||||
|
BlockTime: blockTime,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// parseSplTransfer 解析 SPL Token "transfer" 指令,mint 从 postTokenBalances 中查找
|
||||||
|
func parseSplTransfer(instruction gjson.Result, txData []byte) (*TransferInfo, error) {
|
||||||
|
info := instruction.Get("parsed.info")
|
||||||
|
source := info.Get("source").String()
|
||||||
|
destination := info.Get("destination").String()
|
||||||
|
amountStr := info.Get("amount").String()
|
||||||
|
blockTime := gjson.GetBytes(txData, "result.blockTime").Int()
|
||||||
|
|
||||||
|
rawAmount, ok := new(big.Int).SetString(amountStr, 10)
|
||||||
|
if !ok {
|
||||||
|
return nil, fmt.Errorf("invalid amount: %s", amountStr)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Look up mint and decimals from postTokenBalances using the destination ATA
|
||||||
|
mint, decimals, found := findMintFromTokenBalances(destination, txData)
|
||||||
|
if !found {
|
||||||
|
// Try source ATA as fallback
|
||||||
|
mint, decimals, found = findMintFromTokenBalances(source, txData)
|
||||||
|
}
|
||||||
|
if !found {
|
||||||
|
return nil, fmt.Errorf("could not determine mint for transfer: source=%s dest=%s", source, destination)
|
||||||
|
}
|
||||||
|
|
||||||
|
d := decimals
|
||||||
|
return &TransferInfo{
|
||||||
|
Source: source,
|
||||||
|
Destination: destination,
|
||||||
|
Mint: mint,
|
||||||
|
Amount: ADJustAmount(rawAmount.Uint64(), decimals),
|
||||||
|
RawAmount: rawAmount.Uint64(),
|
||||||
|
Decimals: &d,
|
||||||
|
BlockTime: blockTime,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// parseSplTransferChecked 解析 SPL Token "transferChecked" 指令
|
||||||
|
func parseSplTransferChecked(instruction gjson.Result, txData []byte) (*TransferInfo, error) {
|
||||||
|
info := instruction.Get("parsed.info")
|
||||||
|
source := info.Get("source").String()
|
||||||
|
destination := info.Get("destination").String()
|
||||||
|
mint := info.Get("mint").String()
|
||||||
|
amountStr := info.Get("tokenAmount.amount").String()
|
||||||
|
decimals := int(info.Get("tokenAmount.decimals").Int())
|
||||||
|
blockTime := gjson.GetBytes(txData, "result.blockTime").Int()
|
||||||
|
|
||||||
|
rawAmount, ok := new(big.Int).SetString(amountStr, 10)
|
||||||
|
if !ok {
|
||||||
|
return nil, fmt.Errorf("invalid amount: %s", amountStr)
|
||||||
|
}
|
||||||
|
|
||||||
|
return &TransferInfo{
|
||||||
|
Source: source,
|
||||||
|
Destination: destination,
|
||||||
|
Mint: mint,
|
||||||
|
Amount: ADJustAmount(rawAmount.Uint64(), decimals),
|
||||||
|
RawAmount: rawAmount.Uint64(),
|
||||||
|
Decimals: &decimals,
|
||||||
|
BlockTime: blockTime,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// findMintFromTokenBalances 从 postTokenBalances 中查找 ATA 对应的 mint 和 decimals
|
||||||
|
func findMintFromTokenBalances(ataAddress string, txData []byte) (string, int, bool) {
|
||||||
|
accountKeys := gjson.GetBytes(txData, "result.transaction.message.accountKeys").Array()
|
||||||
|
accountIndex := -1
|
||||||
|
for i, key := range accountKeys {
|
||||||
|
if key.Get("pubkey").String() == ataAddress {
|
||||||
|
accountIndex = i
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if accountIndex == -1 {
|
||||||
|
return "", 0, false
|
||||||
|
}
|
||||||
|
|
||||||
|
balances := gjson.GetBytes(txData, "result.meta.postTokenBalances").Array()
|
||||||
|
for _, balance := range balances {
|
||||||
|
if int(balance.Get("accountIndex").Int()) == accountIndex {
|
||||||
|
mint := balance.Get("mint").String()
|
||||||
|
decimals := int(balance.Get("uiTokenAmount.decimals").Int())
|
||||||
|
return mint, decimals, true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return "", 0, false
|
||||||
|
}
|
||||||
@@ -0,0 +1,349 @@
|
|||||||
|
package service
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/tidwall/gjson"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestSolClientHealthy(t *testing.T) {
|
||||||
|
bodyData, err := SolRetryClient("getHealth", nil)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("SolRetryClient failed: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var result map[string]interface{}
|
||||||
|
err = json.Unmarshal(bodyData, &result)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to unmarshal response: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
status, ok := result["result"].(string)
|
||||||
|
if !ok {
|
||||||
|
t.Fatalf("Unexpected response format: %v", result)
|
||||||
|
}
|
||||||
|
|
||||||
|
t.Logf("RPC Health Status: %s", status)
|
||||||
|
|
||||||
|
if status != "ok" {
|
||||||
|
t.Errorf("Expected health status 'ok', got '%s'", status)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSolClientGetSignaturesForAddress(t *testing.T) {
|
||||||
|
// Example wallet address (replace with actual test address)
|
||||||
|
address := "2uFTf9TZ8gd7Kg6hkb79TxfaeNpaAgpJ8uVHguv2Yweu"
|
||||||
|
|
||||||
|
bodyData, err := SolRetryClient("getSignaturesForAddress", []interface{}{address, map[string]interface{}{"commitment": "finalized", "limit": 100}})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("SolRetryClient failed: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
var result map[string]interface{}
|
||||||
|
err = json.Unmarshal(bodyData, &result)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to unmarshal response: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
signatures, ok := result["result"].([]interface{})
|
||||||
|
if !ok {
|
||||||
|
t.Fatalf("Unexpected response format: %v", result)
|
||||||
|
}
|
||||||
|
|
||||||
|
t.Logf("Found %d signatures for address %s", len(signatures), address)
|
||||||
|
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSolClientGetTransaction(t *testing.T) {
|
||||||
|
// Example transaction signature (replace with actual test signature)
|
||||||
|
sig := "2aEoNykk4ZJ27C3y7EDJiQUc7GFnnsMe7ofFzB73swGL8kTxSBFCnwzWw3jzr3BND7k8hx15fZHUUAbG1XemNFe5"
|
||||||
|
|
||||||
|
txData, err := SolRetryClient("getTransaction", []interface{}{sig, map[string]interface{}{"encoding": "jsonParsed", "commitment": "finalized"}})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("SolRetryClient failed: %v", err)
|
||||||
|
}
|
||||||
|
fmt.Printf("%v\n", string(txData))
|
||||||
|
|
||||||
|
var result map[string]interface{}
|
||||||
|
err = json.Unmarshal(txData, &result)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("Failed to unmarshal response: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
txInfo, ok := result["result"].(map[string]interface{})
|
||||||
|
if !ok {
|
||||||
|
t.Fatalf("Unexpected response format: %v", result)
|
||||||
|
}
|
||||||
|
|
||||||
|
t.Logf("Transaction Info for signature %s: %v", sig, txInfo)
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestFindATAAddress(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
owner string
|
||||||
|
mint string
|
||||||
|
want string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "RAY token ATA",
|
||||||
|
owner: "2uFTf9TZ8gd7Kg6hkb79TxfaeNpaAgpJ8uVHguv2Yweu",
|
||||||
|
mint: "4k3Dyjzvzp8eMZWUXbBCjEvwSkkk59S5iCNLY3QrkX6R",
|
||||||
|
want: "GgmJrwuP946uV8qAwsnXxzYrJqEwW6eGnsVnQZFS5rp4",
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
ata, err := FindATAAddress(tt.owner, tt.mint)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("FindATAAddress failed: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
t.Logf("Owner: %s", tt.owner)
|
||||||
|
t.Logf("Mint: %s", tt.mint)
|
||||||
|
t.Logf("ATA: %s", ata)
|
||||||
|
|
||||||
|
if tt.want != "" && ata != tt.want {
|
||||||
|
t.Errorf("Expected ATA %s, got %s", tt.want, ata)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMatchATAAddress(t *testing.T) {
|
||||||
|
owner := "2uFTf9TZ8gd7Kg6hkb79TxfaeNpaAgpJ8uVHguv2Yweu"
|
||||||
|
mint := "4k3Dyjzvzp8eMZWUXbBCjEvwSkkk59S5iCNLY3QrkX6R" // ray token
|
||||||
|
expectedATA := "GgmJrwuP946uV8qAwsnXxzYrJqEwW6eGnsVnQZFS5rp4"
|
||||||
|
|
||||||
|
ok := MatchAtaAddress(owner, mint, expectedATA)
|
||||||
|
t.Logf("Owner: %s", owner)
|
||||||
|
t.Logf("Mint: %s", mint)
|
||||||
|
t.Logf("Expected ATA: %s", expectedATA)
|
||||||
|
t.Logf("Match result: %v", ok)
|
||||||
|
|
||||||
|
if !ok {
|
||||||
|
t.Error("Expected ATA to match, but it didn't")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMatchUsdtAtaAddress(t *testing.T) {
|
||||||
|
// Example wallet address (replace with actual test address)
|
||||||
|
owner := "2uFTf9TZ8gd7Kg6hkb79TxfaeNpaAgpJ8uVHguv2Yweu"
|
||||||
|
|
||||||
|
ata, err := FindATAAddress(owner, USDT_Mint)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("FindATAAddress failed: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
t.Logf("Owner: %s", owner)
|
||||||
|
t.Logf("USDT Mint: %s", USDT_Mint)
|
||||||
|
t.Logf("USDT ATA: %s", ata)
|
||||||
|
|
||||||
|
ok := MatchUsdtAtaAddress(owner, ata)
|
||||||
|
if !ok {
|
||||||
|
t.Error("Expected USDT ATA to match")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestMatchUsdcAtaAddress(t *testing.T) {
|
||||||
|
// Example wallet address (replace with actual test address)
|
||||||
|
owner := "2uFTf9TZ8gd7Kg6hkb79TxfaeNpaAgpJ8uVHguv2Yweu"
|
||||||
|
|
||||||
|
ata, err := FindATAAddress(owner, USDC_Mint)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("FindATAAddress failed: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
t.Logf("Owner: %s", owner)
|
||||||
|
t.Logf("USDC Mint: %s", USDC_Mint)
|
||||||
|
t.Logf("USDC ATA: %s", ata)
|
||||||
|
|
||||||
|
ok := MatchUsdcAtaAddress(owner, ata)
|
||||||
|
if !ok {
|
||||||
|
t.Error("Expected USDC ATA to match")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAdjustAmount(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
amount uint64
|
||||||
|
decimals int
|
||||||
|
want float64
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "USDT amount (6 decimals)",
|
||||||
|
amount: 123456789,
|
||||||
|
decimals: 6,
|
||||||
|
want: 123.46,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "USDC amount (6 decimals)",
|
||||||
|
amount: 1000000,
|
||||||
|
decimals: 6,
|
||||||
|
want: 1.0,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "SOL amount (9 decimals)",
|
||||||
|
amount: 1000000000,
|
||||||
|
decimals: 9,
|
||||||
|
want: 1.0,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Zero amount",
|
||||||
|
amount: 0,
|
||||||
|
decimals: 6,
|
||||||
|
want: 0,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Small amount",
|
||||||
|
amount: 1,
|
||||||
|
decimals: 6,
|
||||||
|
want: 0.0,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, tt := range tests {
|
||||||
|
t.Run(tt.name, func(t *testing.T) {
|
||||||
|
adjusted := ADJustAmount(tt.amount, tt.decimals)
|
||||||
|
t.Logf("Raw amount: %d, Decimals: %d, Adjusted: %.2f", tt.amount, tt.decimals, adjusted)
|
||||||
|
|
||||||
|
if adjusted != tt.want {
|
||||||
|
t.Errorf("Expected %.2f, got %.2f", tt.want, adjusted)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestParseTransferInfoFromInstruction_SplTransfer(t *testing.T) {
|
||||||
|
if testing.Short() {
|
||||||
|
t.Skip("Skipping integration test")
|
||||||
|
}
|
||||||
|
|
||||||
|
// SPL Token "transfer" (no mint in instruction, must look up from postTokenBalances)
|
||||||
|
sig := "3tZTwLrvmiZ59h4UzyMHPd7DPux7t9eXZgkUvEfquaoSuERrPSRNzWuSHKQM2fbiCWFDGNqoLpu2kLZnfoegVpqN"
|
||||||
|
txData, err := SolGetTransaction(sig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("SolGetTransaction failed: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
instructions := gjson.GetBytes(txData, "result.transaction.message.instructions").Array()
|
||||||
|
var found bool
|
||||||
|
for _, inst := range instructions {
|
||||||
|
info, err := ParseTransferInfoFromInstruction(inst, txData)
|
||||||
|
if err != nil {
|
||||||
|
t.Logf("parse error (ok to skip): %v", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if info == nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
found = true
|
||||||
|
t.Logf("SPL transfer: source=%s dest=%s mint=%s amount=%.6f raw=%d blockTime=%d",
|
||||||
|
info.Source, info.Destination, info.Mint, info.Amount, info.RawAmount, info.BlockTime)
|
||||||
|
|
||||||
|
if info.Mint == "" {
|
||||||
|
t.Error("Expected mint to be resolved from postTokenBalances")
|
||||||
|
}
|
||||||
|
if info.RawAmount != 50000 {
|
||||||
|
t.Errorf("Expected raw amount 50000, got %d", info.RawAmount)
|
||||||
|
}
|
||||||
|
if info.BlockTime == 0 {
|
||||||
|
t.Error("Expected non-zero blockTime")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !found {
|
||||||
|
t.Error("No transfer instruction found in transaction")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestParseTransferInfoFromInstruction_TransferChecked(t *testing.T) {
|
||||||
|
if testing.Short() {
|
||||||
|
t.Skip("Skipping integration test")
|
||||||
|
}
|
||||||
|
|
||||||
|
// SPL Token "transferChecked" (has mint and tokenAmount in instruction)
|
||||||
|
sig := "2aEoNykk4ZJ27C3y7EDJiQUc7GFnnsMe7ofFzB73swGL8kTxSBFCnwzWw3jzr3BND7k8hx15fZHUUAbG1XemNFe5"
|
||||||
|
txData, err := SolGetTransaction(sig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("SolGetTransaction failed: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
instructions := gjson.GetBytes(txData, "result.transaction.message.instructions").Array()
|
||||||
|
var found bool
|
||||||
|
for _, inst := range instructions {
|
||||||
|
info, err := ParseTransferInfoFromInstruction(inst, txData)
|
||||||
|
if err != nil {
|
||||||
|
t.Logf("parse error (ok to skip): %v", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if info == nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
found = true
|
||||||
|
t.Logf("TransferChecked: source=%s dest=%s mint=%s amount=%.6f raw=%d blockTime=%d",
|
||||||
|
info.Source, info.Destination, info.Mint, info.Amount, info.RawAmount, info.BlockTime)
|
||||||
|
|
||||||
|
if info.Mint != USDT_Mint {
|
||||||
|
t.Errorf("Expected USDT mint %s, got %s", USDT_Mint, info.Mint)
|
||||||
|
}
|
||||||
|
if info.RawAmount != 300000 {
|
||||||
|
t.Errorf("Expected raw amount 300000, got %d", info.RawAmount)
|
||||||
|
}
|
||||||
|
if info.Amount != 0.3 {
|
||||||
|
t.Errorf("Expected amount 0.3, got %f", info.Amount)
|
||||||
|
}
|
||||||
|
if info.BlockTime == 0 {
|
||||||
|
t.Error("Expected non-zero blockTime")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if !found {
|
||||||
|
t.Error("No transferChecked instruction found in transaction")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestParseTransferInfoFromInstruction_SystemTransfer(t *testing.T) {
|
||||||
|
if testing.Short() {
|
||||||
|
t.Skip("Skipping integration test")
|
||||||
|
}
|
||||||
|
|
||||||
|
// System program SOL transfer
|
||||||
|
sig := "5pNMonUBvLVpxXTmyd5CGVBs49W6781g2ACnrCXhbmtz58KENYA7HSqu6hQkQweg3qQboRd8WAscphNAtiq9UtZZ"
|
||||||
|
txData, err := SolGetTransaction(sig)
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("SolGetTransaction failed: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
instructions := gjson.GetBytes(txData, "result.transaction.message.instructions").Array()
|
||||||
|
transferCount := 0
|
||||||
|
for _, inst := range instructions {
|
||||||
|
info, err := ParseTransferInfoFromInstruction(inst, txData)
|
||||||
|
if err != nil {
|
||||||
|
t.Logf("parse error (ok to skip): %v", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
if info == nil {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
transferCount++
|
||||||
|
t.Logf("System transfer #%d: source=%s dest=%s mint=%s amount=%.9f raw=%d blockTime=%d",
|
||||||
|
transferCount, info.Source, info.Destination, info.Mint, info.Amount, info.RawAmount, info.BlockTime)
|
||||||
|
|
||||||
|
if info.Mint != "SOL" {
|
||||||
|
t.Errorf("Expected mint SOL, got %s", info.Mint)
|
||||||
|
}
|
||||||
|
if info.RawAmount == 0 {
|
||||||
|
t.Error("Expected non-zero raw amount")
|
||||||
|
}
|
||||||
|
if info.BlockTime == 0 {
|
||||||
|
t.Error("Expected non-zero blockTime")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if transferCount == 0 {
|
||||||
|
t.Error("No system transfer instruction found")
|
||||||
|
}
|
||||||
|
t.Logf("Found %d system transfers", transferCount)
|
||||||
|
}
|
||||||
@@ -4,6 +4,7 @@ import (
|
|||||||
"encoding/hex"
|
"encoding/hex"
|
||||||
"errors"
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"math/big"
|
||||||
"net/http"
|
"net/http"
|
||||||
"strings"
|
"strings"
|
||||||
"sync"
|
"sync"
|
||||||
@@ -19,6 +20,7 @@ import (
|
|||||||
"github.com/assimon/luuu/util/log"
|
"github.com/assimon/luuu/util/log"
|
||||||
"github.com/assimon/luuu/util/math"
|
"github.com/assimon/luuu/util/math"
|
||||||
"github.com/dromara/carbon/v2"
|
"github.com/dromara/carbon/v2"
|
||||||
|
"github.com/ethereum/go-ethereum/common"
|
||||||
"github.com/gookit/goutil/stdutil"
|
"github.com/gookit/goutil/stdutil"
|
||||||
"github.com/shopspring/decimal"
|
"github.com/shopspring/decimal"
|
||||||
"github.com/tidwall/gjson"
|
"github.com/tidwall/gjson"
|
||||||
@@ -110,7 +112,7 @@ func checkTrxTransfers(address string, wg *sync.WaitGroup) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
txID := transfer.Get("txID").String()
|
txID := transfer.Get("txID").String()
|
||||||
tradeID, err := data.GetTradeIdByWalletAddressAndAmountAndToken(address, "TRX", amount)
|
tradeID, err := data.GetTradeIdByWalletAddressAndAmountAndToken(mdb.NetworkTron, address, "TRX", amount)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
panic(err)
|
panic(err)
|
||||||
}
|
}
|
||||||
@@ -134,6 +136,7 @@ func checkTrxTransfers(address string, wg *sync.WaitGroup) {
|
|||||||
req := &request.OrderProcessingRequest{
|
req := &request.OrderProcessingRequest{
|
||||||
ReceiveAddress: address,
|
ReceiveAddress: address,
|
||||||
Token: "TRX",
|
Token: "TRX",
|
||||||
|
Network: mdb.NetworkTron,
|
||||||
TradeId: tradeID,
|
TradeId: tradeID,
|
||||||
Amount: amount,
|
Amount: amount,
|
||||||
BlockTransactionId: txID,
|
BlockTransactionId: txID,
|
||||||
@@ -212,7 +215,7 @@ func checkTrc20Transfers(address string, wg *sync.WaitGroup) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
txID := transfer.Get("transaction_id").String()
|
txID := transfer.Get("transaction_id").String()
|
||||||
tradeID, err := data.GetTradeIdByWalletAddressAndAmountAndToken(address, "USDT", amount)
|
tradeID, err := data.GetTradeIdByWalletAddressAndAmountAndToken(mdb.NetworkTron, address, "USDT", amount)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
panic(err)
|
panic(err)
|
||||||
}
|
}
|
||||||
@@ -236,6 +239,7 @@ func checkTrc20Transfers(address string, wg *sync.WaitGroup) {
|
|||||||
req := &request.OrderProcessingRequest{
|
req := &request.OrderProcessingRequest{
|
||||||
ReceiveAddress: address,
|
ReceiveAddress: address,
|
||||||
Token: "USDT",
|
Token: "USDT",
|
||||||
|
Network: mdb.NetworkTron,
|
||||||
TradeId: tradeID,
|
TradeId: tradeID,
|
||||||
Amount: amount,
|
Amount: amount,
|
||||||
BlockTransactionId: txID,
|
BlockTransactionId: txID,
|
||||||
@@ -254,6 +258,196 @@ func checkTrc20Transfers(address string, wg *sync.WaitGroup) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func evmChainLogLabel(chainNetwork string) string {
|
||||||
|
switch chainNetwork {
|
||||||
|
case mdb.NetworkEthereum:
|
||||||
|
return "ETH"
|
||||||
|
case mdb.NetworkBsc:
|
||||||
|
return "BSC"
|
||||||
|
case mdb.NetworkPolygon:
|
||||||
|
return "POLYGON"
|
||||||
|
case mdb.NetworkPlasma:
|
||||||
|
return "PLASMA"
|
||||||
|
default:
|
||||||
|
return "EVM"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TryProcessEvmERC20Transfer 处理各 EVM 链上 USDT/USDC(及 Polygon USDC.e)的 Transfer 入账(合约与 network 需一致)。
|
||||||
|
func TryProcessEvmERC20Transfer(chainNetwork string, contract common.Address, toAddr common.Address, rawValue *big.Int, txHash string, blockTsMs int64) {
|
||||||
|
defer func() {
|
||||||
|
if err := recover(); err != nil {
|
||||||
|
log.Sugar.Errorf("[%s-WS] TryProcessEvmERC20Transfer panic: %v", evmChainLogLabel(chainNetwork), err)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
var usdt, usdc common.Address
|
||||||
|
var polygonUsdcE common.Address
|
||||||
|
switch chainNetwork {
|
||||||
|
case mdb.NetworkEthereum:
|
||||||
|
usdt = common.HexToAddress("0xdAC17F958D2ee523a2206206994597C13D831ec7")
|
||||||
|
usdc = common.HexToAddress("0xA0b86991c6218b36c1d19d4a2e9eb0ce3606eb48")
|
||||||
|
case mdb.NetworkBsc:
|
||||||
|
usdt = common.HexToAddress("0x55d398326f99059fF775485246999027B3197955")
|
||||||
|
usdc = common.HexToAddress("0x8AC76a51cc950d9822D68b83fE1Ad97B32Cd580d")
|
||||||
|
case mdb.NetworkPolygon:
|
||||||
|
usdt = common.HexToAddress("0xc2132D05D31c914a87C6611C10748AEb04B58e8F")
|
||||||
|
usdc = common.HexToAddress("0x3c499c542cEF5E3811e1192ce70d8cC03d5c3359")
|
||||||
|
polygonUsdcE = common.HexToAddress("0x2791Bca1f2de4661ED88A30C99A7a9449Aa84174")
|
||||||
|
case mdb.NetworkPlasma:
|
||||||
|
// USDT0(官方),6 decimals;链上暂无与 ETH 同级的 Circle USDC 部署,仅匹配 USDT 订单
|
||||||
|
usdt = common.HexToAddress("0xB8CE59FC3717ada4C02eaDF9682A9e934F625ebb")
|
||||||
|
default:
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
var tokenSym string
|
||||||
|
switch {
|
||||||
|
case contract == usdt:
|
||||||
|
tokenSym = "USDT"
|
||||||
|
case contract == usdc || (polygonUsdcE != (common.Address{}) && contract == polygonUsdcE):
|
||||||
|
tokenSym = "USDC"
|
||||||
|
default:
|
||||||
|
net := evmChainLogLabel(chainNetwork)
|
||||||
|
log.Sugar.Warnf("[%s-WS] skip unsupported contract %s", net, contract.Hex())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
net := evmChainLogLabel(chainNetwork)
|
||||||
|
walletAddr := strings.ToLower(toAddr.Hex())
|
||||||
|
if rawValue == nil || rawValue.Sign() <= 0 {
|
||||||
|
log.Sugar.Infof("[%s-%s][%s] skip non-positive or nil amount", net, tokenSym, walletAddr)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
decimalQuant := decimal.NewFromBigInt(rawValue, 0)
|
||||||
|
amount := math.MustParsePrecFloat64(decimalQuant.Div(decimal.NewFromInt(1_000_000)).InexactFloat64(), 2)
|
||||||
|
if amount <= 0 {
|
||||||
|
log.Sugar.Warnf("[%s-%s][%s] skip non-positive amount %.2f", net, tokenSym, walletAddr, amount)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Sugar.Debugf("[%s-%s][%s] processing transfer hash=%s amount=%.2f", net, tokenSym, walletAddr, txHash, amount)
|
||||||
|
|
||||||
|
tradeID, err := data.GetTradeIdByWalletAddressAndAmountAndToken(chainNetwork, walletAddr, tokenSym, amount)
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Warnf("[%s-%s][%s] lock lookup: %v", net, tokenSym, walletAddr, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if tradeID == "" {
|
||||||
|
log.Sugar.Warnf("[%s-%s][%s] skip unmatched tx hash=%s amount=%.2f", net, tokenSym, walletAddr, txHash, amount)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
order, err := data.GetOrderInfoByTradeId(tradeID)
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Warnf("[%s-%s][%s] load order: %v", net, tokenSym, walletAddr, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if strings.ToLower(strings.TrimSpace(order.Network)) != chainNetwork {
|
||||||
|
log.Sugar.Warnf("[%s-%s][%s] skip trade_id=%s network=%q", net, tokenSym, walletAddr, tradeID, order.Network)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if strings.ToUpper(strings.TrimSpace(order.Token)) != tokenSym {
|
||||||
|
log.Sugar.Warnf("[%s-%s][%s] skip trade_id=%s token mismatch order=%s", net, tokenSym, walletAddr, tradeID, order.Token)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
req := &request.OrderProcessingRequest{
|
||||||
|
ReceiveAddress: walletAddr,
|
||||||
|
Token: tokenSym,
|
||||||
|
Network: chainNetwork,
|
||||||
|
TradeId: tradeID,
|
||||||
|
Amount: amount,
|
||||||
|
BlockTransactionId: txHash,
|
||||||
|
}
|
||||||
|
err = OrderProcessing(req)
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, constant.OrderBlockAlreadyProcess) || errors.Is(err, constant.OrderStatusConflict) {
|
||||||
|
log.Sugar.Infof("[%s-%s][%s] skip resolved trade_id=%s hash=%s err=%v", net, tokenSym, walletAddr, tradeID, txHash, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
log.Sugar.Errorf("[%s-%s][%s] OrderProcessing: %v", net, tokenSym, walletAddr, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
sendPaymentNotification(order)
|
||||||
|
log.Sugar.Infof("[%s-%s][%s] payment processed trade_id=%s hash=%s", net, tokenSym, walletAddr, tradeID, txHash)
|
||||||
|
}
|
||||||
|
|
||||||
|
// TryProcessTronTRC20Transfer 处理 Tron 链上 TRC20-USDT 的 Transfer 入账。
|
||||||
|
func TryProcessTronTRC20Transfer(toAddr string, rawValue *big.Int, txHash string, blockTsMs int64) {
|
||||||
|
defer func() {
|
||||||
|
if err := recover(); err != nil {
|
||||||
|
log.Sugar.Errorf("[TRON-WS] TryProcessTronTRC20Transfer panic: %v", err)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
|
||||||
|
walletAddr := strings.TrimSpace(toAddr)
|
||||||
|
if walletAddr == "" {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if rawValue == nil || rawValue.Sign() <= 0 {
|
||||||
|
log.Sugar.Infof("[TRON-USDT][%s] skip non-positive or nil amount", walletAddr)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
decimalQuant := decimal.NewFromBigInt(rawValue, 0)
|
||||||
|
amount := math.MustParsePrecFloat64(decimalQuant.Div(decimal.NewFromInt(1_000_000)).InexactFloat64(), 2)
|
||||||
|
if amount <= 0 {
|
||||||
|
log.Sugar.Warnf("[TRON-USDT][%s] skip non-positive amount %.2f", walletAddr, amount)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
tradeID, err := data.GetTradeIdByWalletAddressAndAmountAndToken(mdb.NetworkTron, walletAddr, "USDT", amount)
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Warnf("[TRON-USDT][%s] lock lookup: %v", walletAddr, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if tradeID == "" {
|
||||||
|
log.Sugar.Debugf("[TRON-USDT][%s] skip unmatched tx hash=%s amount=%.2f", walletAddr, txHash, amount)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
order, err := data.GetOrderInfoByTradeId(tradeID)
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Warnf("[TRON-USDT][%s] load order: %v", walletAddr, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if blockTsMs > 0 && blockTsMs < order.CreatedAt.TimestampMilli() {
|
||||||
|
log.Sugar.Warnf("[TRON-USDT][%s] skip tx %s because block time %d is before order create time %d", walletAddr, txHash, blockTsMs, order.CreatedAt.TimestampMilli())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if strings.ToLower(strings.TrimSpace(order.Network)) != mdb.NetworkTron {
|
||||||
|
log.Sugar.Warnf("[TRON-USDT][%s] skip trade_id=%s network=%q", walletAddr, tradeID, order.Network)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if strings.ToUpper(strings.TrimSpace(order.Token)) != "USDT" {
|
||||||
|
log.Sugar.Warnf("[TRON-USDT][%s] skip trade_id=%s token mismatch order=%s", walletAddr, tradeID, order.Token)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
req := &request.OrderProcessingRequest{
|
||||||
|
ReceiveAddress: walletAddr,
|
||||||
|
Token: "USDT",
|
||||||
|
Network: mdb.NetworkTron,
|
||||||
|
TradeId: tradeID,
|
||||||
|
Amount: amount,
|
||||||
|
BlockTransactionId: txHash,
|
||||||
|
}
|
||||||
|
err = OrderProcessing(req)
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, constant.OrderBlockAlreadyProcess) || errors.Is(err, constant.OrderStatusConflict) {
|
||||||
|
log.Sugar.Infof("[TRON-USDT][%s] skip resolved trade_id=%s hash=%s err=%v", walletAddr, tradeID, txHash, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
log.Sugar.Errorf("[TRON-USDT][%s] OrderProcessing: %v", walletAddr, err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
sendPaymentNotification(order)
|
||||||
|
log.Sugar.Infof("[TRON-USDT][%s] payment processed trade_id=%s hash=%s", walletAddr, tradeID, txHash)
|
||||||
|
}
|
||||||
|
|
||||||
func sendPaymentNotification(order *mdb.Orders) {
|
func sendPaymentNotification(order *mdb.Orders) {
|
||||||
msg := fmt.Sprintf(
|
msg := fmt.Sprintf(
|
||||||
"🎉 <b>收款成功通知</b>\n\n"+
|
"🎉 <b>收款成功通知</b>\n\n"+
|
||||||
@@ -263,6 +457,7 @@ func sendPaymentNotification(order *mdb.Orders) {
|
|||||||
"📋 <b>订单信息</b>\n"+
|
"📋 <b>订单信息</b>\n"+
|
||||||
"├ 交易号:<code>%s</code>\n"+
|
"├ 交易号:<code>%s</code>\n"+
|
||||||
"├ 订单号:<code>%s</code>\n"+
|
"├ 订单号:<code>%s</code>\n"+
|
||||||
|
"├ 网络:<code>%s</code>\n"+
|
||||||
"└ 钱包地址:<code>%s</code>\n\n"+
|
"└ 钱包地址:<code>%s</code>\n\n"+
|
||||||
"⏰ <b>时间信息</b>\n"+
|
"⏰ <b>时间信息</b>\n"+
|
||||||
"├ 创建时间:%s\n"+
|
"├ 创建时间:%s\n"+
|
||||||
@@ -273,9 +468,32 @@ func sendPaymentNotification(order *mdb.Orders) {
|
|||||||
strings.ToUpper(order.Token),
|
strings.ToUpper(order.Token),
|
||||||
order.TradeId,
|
order.TradeId,
|
||||||
order.OrderId,
|
order.OrderId,
|
||||||
|
networkDisplay(order.Network),
|
||||||
order.ReceiveAddress,
|
order.ReceiveAddress,
|
||||||
order.CreatedAt.ToDateTimeString(),
|
order.CreatedAt.ToDateTimeString(),
|
||||||
carbon.Now().ToDateTimeString(),
|
carbon.Now().ToDateTimeString(),
|
||||||
)
|
)
|
||||||
telegram.SendToBot(msg)
|
telegram.SendToBot(msg)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func networkDisplay(n string) string {
|
||||||
|
switch strings.ToLower(strings.TrimSpace(n)) {
|
||||||
|
case mdb.NetworkTron:
|
||||||
|
return "Tron"
|
||||||
|
case mdb.NetworkSolana:
|
||||||
|
return "Solana"
|
||||||
|
case mdb.NetworkEthereum:
|
||||||
|
return "Ethereum"
|
||||||
|
case mdb.NetworkBsc:
|
||||||
|
return "BSC"
|
||||||
|
case mdb.NetworkPolygon:
|
||||||
|
return "Polygon"
|
||||||
|
case mdb.NetworkPlasma:
|
||||||
|
return "Plasma"
|
||||||
|
default:
|
||||||
|
if n == "" {
|
||||||
|
return "Tron"
|
||||||
|
}
|
||||||
|
return strings.ToUpper(n)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -2,7 +2,10 @@ package mq
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
"net/url"
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
@@ -23,6 +26,7 @@ const sqliteBusyRetryAttempts = 3
|
|||||||
type expirableOrder struct {
|
type expirableOrder struct {
|
||||||
ID uint64 `gorm:"column:id"`
|
ID uint64 `gorm:"column:id"`
|
||||||
TradeId string `gorm:"column:trade_id"`
|
TradeId string `gorm:"column:trade_id"`
|
||||||
|
Network string `gorm:"column:network"`
|
||||||
ReceiveAddress string `gorm:"column:receive_address"`
|
ReceiveAddress string `gorm:"column:receive_address"`
|
||||||
Token string `gorm:"column:token"`
|
Token string `gorm:"column:token"`
|
||||||
ActualAmount float64 `gorm:"column:actual_amount"`
|
ActualAmount float64 `gorm:"column:actual_amount"`
|
||||||
@@ -65,7 +69,7 @@ func processExpiredOrders() {
|
|||||||
var orders []expirableOrder
|
var orders []expirableOrder
|
||||||
err := withSQLiteBusyRetry(func() error {
|
err := withSQLiteBusyRetry(func() error {
|
||||||
return dao.Mdb.Model(&mdb.Orders{}).
|
return dao.Mdb.Model(&mdb.Orders{}).
|
||||||
Select("id", "trade_id", "receive_address", "token", "actual_amount").
|
Select("id", "trade_id", "network", "receive_address", "token", "actual_amount").
|
||||||
Where("status = ?", mdb.StatusWaitPay).
|
Where("status = ?", mdb.StatusWaitPay).
|
||||||
Where("created_at <= ?", expirationCutoff).
|
Where("created_at <= ?", expirationCutoff).
|
||||||
Order("id asc").
|
Order("id asc").
|
||||||
@@ -89,7 +93,7 @@ func processExpiredOrders() {
|
|||||||
if !expired {
|
if !expired {
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
if err = data.UnLockTransaction(order.ReceiveAddress, order.Token, order.ActualAmount); err != nil {
|
if err = data.UnLockTransaction(order.Network, order.ReceiveAddress, order.Token, order.ActualAmount); err != nil {
|
||||||
log.Sugar.Warnf("[mq] release expired transaction lock failed, trade_id=%s, err=%v", order.TradeId, err)
|
log.Sugar.Warnf("[mq] release expired transaction lock failed, trade_id=%s, err=%v", order.TradeId, err)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -161,6 +165,54 @@ func processCallback(tradeID string) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func sendOrderCallback(order *mdb.Orders) error {
|
func sendOrderCallback(order *mdb.Orders) error {
|
||||||
|
|
||||||
|
switch order.PaymentType {
|
||||||
|
case mdb.PaymentTypeEpay:
|
||||||
|
// 构造 EPay 标准回调参数
|
||||||
|
notifyData := response.OrderNotifyResponseEpay{
|
||||||
|
PID: config.GetEpayPid(),
|
||||||
|
TradeNo: order.TradeId, // epusdt 订单号作为 EPay 平台订单号
|
||||||
|
OutTradeNo: order.OrderId, // 注意:EPay 回调要求商户订单号使用 out_trade_no 参数
|
||||||
|
|
||||||
|
Type: "alipay",
|
||||||
|
Name: order.Name,
|
||||||
|
Money: fmt.Sprintf("%.4f", order.Amount),
|
||||||
|
TradeStatus: "TRADE_SUCCESS",
|
||||||
|
}
|
||||||
|
|
||||||
|
signstr2, err := sign.Get(notifyData, config.GetEpayKey())
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
// 使用 form-encoded POST(EPay 标准协议格式)
|
||||||
|
formData := url.Values{
|
||||||
|
"pid": {fmt.Sprintf("%d", notifyData.PID)},
|
||||||
|
"trade_no": {notifyData.TradeNo},
|
||||||
|
"out_trade_no": {notifyData.OutTradeNo},
|
||||||
|
"type": {notifyData.Type},
|
||||||
|
"name": {notifyData.Name},
|
||||||
|
"money": {notifyData.Money},
|
||||||
|
"trade_status": {notifyData.TradeStatus},
|
||||||
|
"sign": {signstr2},
|
||||||
|
"sign_type": {"MD5"},
|
||||||
|
}
|
||||||
|
|
||||||
|
resp, err := http.PostForm(order.NotifyUrl, formData)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
responseBody, err := io.ReadAll(resp.Body)
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
fmt.Printf("notify_url response status: %d, body: %s\n", resp.StatusCode, string(responseBody))
|
||||||
|
|
||||||
|
default:
|
||||||
|
|
||||||
client := http_client.GetHttpClient()
|
client := http_client.GetHttpClient()
|
||||||
orderResp := response.OrderNotifyResponse{
|
orderResp := response.OrderNotifyResponse{
|
||||||
TradeId: order.TradeId,
|
TradeId: order.TradeId,
|
||||||
@@ -191,6 +243,8 @@ func sendOrderCallback(order *mdb.Orders) error {
|
|||||||
if string(resp.Body()) != "ok" {
|
if string(resp.Body()) != "ok" {
|
||||||
return errors.New("not ok")
|
return errors.New("not ok")
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -27,6 +27,7 @@ func TestProcessExpiredOrdersExpiresWaitingOrdersAndReleasesLocks(t *testing.T)
|
|||||||
ActualAmount: 1,
|
ActualAmount: 1,
|
||||||
ReceiveAddress: "wallet_1",
|
ReceiveAddress: "wallet_1",
|
||||||
Token: "USDT",
|
Token: "USDT",
|
||||||
|
Network: "tron",
|
||||||
Status: mdb.StatusWaitPay,
|
Status: mdb.StatusWaitPay,
|
||||||
NotifyUrl: "https://merchant.example/callback",
|
NotifyUrl: "https://merchant.example/callback",
|
||||||
}
|
}
|
||||||
@@ -36,7 +37,7 @@ func TestProcessExpiredOrdersExpiresWaitingOrdersAndReleasesLocks(t *testing.T)
|
|||||||
if err := dao.Mdb.Model(order).UpdateColumn("created_at", time.Now().Add(-20*time.Minute)).Error; err != nil {
|
if err := dao.Mdb.Model(order).UpdateColumn("created_at", time.Now().Add(-20*time.Minute)).Error; err != nil {
|
||||||
t.Fatalf("age expired order: %v", err)
|
t.Fatalf("age expired order: %v", err)
|
||||||
}
|
}
|
||||||
if err := data.LockTransaction(order.ReceiveAddress, order.Token, order.TradeId, order.ActualAmount, time.Hour); err != nil {
|
if err := data.LockTransaction("tron", order.ReceiveAddress, order.Token, order.TradeId, order.ActualAmount, time.Hour); err != nil {
|
||||||
t.Fatalf("lock expired order: %v", err)
|
t.Fatalf("lock expired order: %v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -48,13 +49,14 @@ func TestProcessExpiredOrdersExpiresWaitingOrdersAndReleasesLocks(t *testing.T)
|
|||||||
ActualAmount: 1.01,
|
ActualAmount: 1.01,
|
||||||
ReceiveAddress: "wallet_1",
|
ReceiveAddress: "wallet_1",
|
||||||
Token: "USDT",
|
Token: "USDT",
|
||||||
|
Network: "tron",
|
||||||
Status: mdb.StatusWaitPay,
|
Status: mdb.StatusWaitPay,
|
||||||
NotifyUrl: "https://merchant.example/callback",
|
NotifyUrl: "https://merchant.example/callback",
|
||||||
}
|
}
|
||||||
if err := dao.Mdb.Create(recentOrder).Error; err != nil {
|
if err := dao.Mdb.Create(recentOrder).Error; err != nil {
|
||||||
t.Fatalf("create recent order: %v", err)
|
t.Fatalf("create recent order: %v", err)
|
||||||
}
|
}
|
||||||
if err := data.LockTransaction(recentOrder.ReceiveAddress, recentOrder.Token, recentOrder.TradeId, recentOrder.ActualAmount, time.Hour); err != nil {
|
if err := data.LockTransaction("tron", recentOrder.ReceiveAddress, recentOrder.Token, recentOrder.TradeId, recentOrder.ActualAmount, time.Hour); err != nil {
|
||||||
t.Fatalf("lock recent order: %v", err)
|
t.Fatalf("lock recent order: %v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -67,7 +69,7 @@ func TestProcessExpiredOrdersExpiresWaitingOrdersAndReleasesLocks(t *testing.T)
|
|||||||
if expired.Status != mdb.StatusExpired {
|
if expired.Status != mdb.StatusExpired {
|
||||||
t.Fatalf("expired order status = %d, want %d", expired.Status, mdb.StatusExpired)
|
t.Fatalf("expired order status = %d, want %d", expired.Status, mdb.StatusExpired)
|
||||||
}
|
}
|
||||||
lockTradeID, err := data.GetTradeIdByWalletAddressAndAmountAndToken(order.ReceiveAddress, order.Token, order.ActualAmount)
|
lockTradeID, err := data.GetTradeIdByWalletAddressAndAmountAndToken("tron", order.ReceiveAddress, order.Token, order.ActualAmount)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("expired order lock lookup: %v", err)
|
t.Fatalf("expired order lock lookup: %v", err)
|
||||||
}
|
}
|
||||||
@@ -82,7 +84,7 @@ func TestProcessExpiredOrdersExpiresWaitingOrdersAndReleasesLocks(t *testing.T)
|
|||||||
if recent.Status != mdb.StatusWaitPay {
|
if recent.Status != mdb.StatusWaitPay {
|
||||||
t.Fatalf("recent order status = %d, want %d", recent.Status, mdb.StatusWaitPay)
|
t.Fatalf("recent order status = %d, want %d", recent.Status, mdb.StatusWaitPay)
|
||||||
}
|
}
|
||||||
lockTradeID, err = data.GetTradeIdByWalletAddressAndAmountAndToken(recentOrder.ReceiveAddress, recentOrder.Token, recentOrder.ActualAmount)
|
lockTradeID, err = data.GetTradeIdByWalletAddressAndAmountAndToken("tron", recentOrder.ReceiveAddress, recentOrder.Token, recentOrder.ActualAmount)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("recent order lock lookup: %v", err)
|
t.Fatalf("recent order lock lookup: %v", err)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,10 +1,19 @@
|
|||||||
package route
|
package route
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
"strconv"
|
||||||
|
|
||||||
|
"github.com/assimon/luuu/config"
|
||||||
"github.com/assimon/luuu/controller/comm"
|
"github.com/assimon/luuu/controller/comm"
|
||||||
"github.com/assimon/luuu/middleware"
|
"github.com/assimon/luuu/middleware"
|
||||||
|
"github.com/assimon/luuu/model/mdb"
|
||||||
|
"github.com/assimon/luuu/util/constant"
|
||||||
|
"github.com/assimon/luuu/util/sign"
|
||||||
"github.com/labstack/echo/v4"
|
"github.com/labstack/echo/v4"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -13,16 +22,154 @@ func RegisterRoute(e *echo.Echo) {
|
|||||||
e.Any("/", func(c echo.Context) error {
|
e.Any("/", func(c echo.Context) error {
|
||||||
return c.String(http.StatusOK, "hello epusdt, https://github.com/GMwalletApp/epusdt")
|
return c.String(http.StatusOK, "hello epusdt, https://github.com/GMwalletApp/epusdt")
|
||||||
})
|
})
|
||||||
// ==== 支付相关=====
|
|
||||||
payRoute := e.Group("/pay")
|
|
||||||
// 收银台
|
|
||||||
payRoute.GET("/checkout-counter/:trade_id", comm.Ctrl.CheckoutCounter)
|
|
||||||
// 状态检测
|
|
||||||
payRoute.GET("/check-status/:trade_id", comm.Ctrl.CheckStatus)
|
|
||||||
|
|
||||||
apiV1Route := e.Group("/api/v1")
|
payRoute := e.Group("/pay")
|
||||||
// ====订单相关====
|
payRoute.GET("/checkout-counter/:trade_id", comm.Ctrl.CheckoutCounter)
|
||||||
orderRoute := apiV1Route.Group("/order", middleware.CheckApiSign())
|
payRoute.GET("/check-status/:trade_id", comm.Ctrl.CheckStatus)
|
||||||
// 创建订单
|
payRoute.POST("/switch-network", comm.Ctrl.SwitchNetwork)
|
||||||
orderRoute.POST("/create-transaction", comm.Ctrl.CreateTransaction)
|
|
||||||
|
// payment routes
|
||||||
|
paymentRoute := e.Group("/payments")
|
||||||
|
|
||||||
|
// for epusdt
|
||||||
|
epusdtV1 := paymentRoute.Group("/epusdt/v1")
|
||||||
|
epusdtV1.POST("/order/create-transaction", func(ctx echo.Context) error {
|
||||||
|
// add default token and currency for old plugin
|
||||||
|
|
||||||
|
body := make(map[string]interface{})
|
||||||
|
if err := ctx.Bind(&body); err != nil {
|
||||||
|
return comm.Ctrl.FailJson(ctx, err)
|
||||||
|
}
|
||||||
|
if _, ok := body["token"]; !ok {
|
||||||
|
body["token"] = "usdt"
|
||||||
|
}
|
||||||
|
if _, ok := body["currency"]; !ok {
|
||||||
|
body["currency"] = "cny"
|
||||||
|
}
|
||||||
|
if _, ok := body["network"]; !ok {
|
||||||
|
body["network"] = "tron"
|
||||||
|
}
|
||||||
|
ctx.Set("request_body", body)
|
||||||
|
|
||||||
|
jsonBytes, err := json.Marshal(body)
|
||||||
|
if err != nil {
|
||||||
|
return comm.Ctrl.FailJson(ctx, err)
|
||||||
|
}
|
||||||
|
ctx.Request().Body = io.NopCloser(bytes.NewBuffer(jsonBytes))
|
||||||
|
ctx.Request().ContentLength = int64(len(jsonBytes))
|
||||||
|
|
||||||
|
return comm.Ctrl.CreateTransaction(ctx)
|
||||||
|
}, middleware.CheckApiSign())
|
||||||
|
|
||||||
|
// gmpay v1 routes
|
||||||
|
gmpayV1 := paymentRoute.Group("/gmpay/v1")
|
||||||
|
gmpayV1.POST("/order/create-transaction", comm.Ctrl.CreateTransaction, middleware.CheckApiSign())
|
||||||
|
gmpayV1.GET("/supported-assets", comm.Ctrl.GetSupportedAssets)
|
||||||
|
gmpayV1.GET("/supported-assets/records", comm.Ctrl.ListSupportedAssetRecords)
|
||||||
|
gmpayV1.GET("/supported-assets/:id", comm.Ctrl.GetSupportedAsset)
|
||||||
|
gmpayV1.POST("/supported-assets/add", comm.Ctrl.AddSupportedAsset, middleware.CheckApiToken())
|
||||||
|
gmpayV1.POST("/supported-assets/:id/update", comm.Ctrl.UpdateSupportedAsset, middleware.CheckApiToken())
|
||||||
|
gmpayV1.POST("/supported-assets/:id/delete", comm.Ctrl.DeleteSupportedAsset, middleware.CheckApiToken())
|
||||||
|
|
||||||
|
// wallet management routes
|
||||||
|
walletV1 := gmpayV1.Group("/wallet", middleware.CheckApiToken())
|
||||||
|
walletV1.POST("/add", comm.Ctrl.AddWallet)
|
||||||
|
walletV1.GET("/list", comm.Ctrl.ListWallets)
|
||||||
|
walletV1.GET("/:id", comm.Ctrl.GetWallet)
|
||||||
|
walletV1.POST("/:id/status", comm.Ctrl.ChangeWalletStatus)
|
||||||
|
walletV1.POST("/:id/delete", comm.Ctrl.DeleteWallet)
|
||||||
|
|
||||||
|
// epay v1 routes
|
||||||
|
epayV1 := paymentRoute.Group("/epay/v1")
|
||||||
|
epayV1.Match([]string{http.MethodPost, http.MethodGet}, "/order/create-transaction/submit.php", func(ctx echo.Context) error {
|
||||||
|
params := make(map[string]interface{})
|
||||||
|
copyParams := func(values map[string][]string) {
|
||||||
|
for k, v := range values {
|
||||||
|
if len(v) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
params[k] = v[0]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
copyParams(ctx.QueryParams())
|
||||||
|
|
||||||
|
formParams, err := ctx.FormParams()
|
||||||
|
if err != nil && ctx.Request().Method == http.MethodPost {
|
||||||
|
return comm.Ctrl.FailJson(ctx, fmt.Errorf("invalid epay form params: %w", err))
|
||||||
|
}
|
||||||
|
if err == nil {
|
||||||
|
copyParams(formParams)
|
||||||
|
}
|
||||||
|
|
||||||
|
getString := func(m map[string]interface{}, key string) string {
|
||||||
|
v, ok := m[key]
|
||||||
|
if !ok {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
s, ok := v.(string)
|
||||||
|
if !ok {
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
return s
|
||||||
|
}
|
||||||
|
|
||||||
|
signstr := getString(params, "sign")
|
||||||
|
if signstr == "" {
|
||||||
|
return constant.SignatureErr
|
||||||
|
}
|
||||||
|
|
||||||
|
delete(params, "sign")
|
||||||
|
delete(params, "sign_type")
|
||||||
|
|
||||||
|
// we need to add pid to params for signature verification
|
||||||
|
params["pid"] = config.GetEpayPid()
|
||||||
|
|
||||||
|
checkSignature, err := sign.Get(params, config.GetApiAuthToken())
|
||||||
|
if err != nil {
|
||||||
|
return constant.SignatureErr
|
||||||
|
}
|
||||||
|
if checkSignature != signstr {
|
||||||
|
return constant.SignatureErr
|
||||||
|
}
|
||||||
|
|
||||||
|
money := getString(params, "money")
|
||||||
|
name := getString(params, "name")
|
||||||
|
notifyURL := getString(params, "notify_url")
|
||||||
|
outTradeNo := getString(params, "out_trade_no")
|
||||||
|
returnURL := getString(params, "return_url")
|
||||||
|
|
||||||
|
amountFloat, err := strconv.ParseFloat(money, 64)
|
||||||
|
if err != nil {
|
||||||
|
return comm.Ctrl.FailJson(ctx, fmt.Errorf("invalid money value: %s", money))
|
||||||
|
}
|
||||||
|
|
||||||
|
body := map[string]interface{}{
|
||||||
|
"token": "usdt",
|
||||||
|
"currency": "cny",
|
||||||
|
"network": "tron",
|
||||||
|
"amount": amountFloat,
|
||||||
|
"notify_url": notifyURL,
|
||||||
|
"order_id": outTradeNo,
|
||||||
|
"redirect_url": returnURL,
|
||||||
|
"signature": signstr,
|
||||||
|
"name": name,
|
||||||
|
"payment_type": mdb.PaymentTypeEpay,
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx.Set("request_body", body)
|
||||||
|
|
||||||
|
jsonBytes, err := json.Marshal(body)
|
||||||
|
if err != nil {
|
||||||
|
return comm.Ctrl.FailJson(ctx, err)
|
||||||
|
}
|
||||||
|
|
||||||
|
ctx.Request().Body = io.NopCloser(bytes.NewBuffer(jsonBytes))
|
||||||
|
ctx.Request().ContentLength = int64(len(jsonBytes))
|
||||||
|
ctx.Request().Method = http.MethodPost
|
||||||
|
ctx.Request().Header.Set("Content-Type", "application/json")
|
||||||
|
|
||||||
|
return comm.Ctrl.CreateTransactionAndRedirect(ctx)
|
||||||
|
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,604 @@
|
|||||||
|
package route
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"net/http"
|
||||||
|
"net/http/httptest"
|
||||||
|
"net/url"
|
||||||
|
"os"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"github.com/assimon/luuu/model/dao"
|
||||||
|
"github.com/assimon/luuu/model/mdb"
|
||||||
|
"github.com/assimon/luuu/util/log"
|
||||||
|
"github.com/assimon/luuu/util/sign"
|
||||||
|
"github.com/labstack/echo/v4"
|
||||||
|
"github.com/spf13/viper"
|
||||||
|
)
|
||||||
|
|
||||||
|
const testAPIToken = "test-secret-token"
|
||||||
|
|
||||||
|
func setupTestEnv(t *testing.T) *echo.Echo {
|
||||||
|
t.Helper()
|
||||||
|
|
||||||
|
tmpDir := t.TempDir()
|
||||||
|
|
||||||
|
// minimal viper config
|
||||||
|
viper.Reset()
|
||||||
|
viper.Set("db_type", "sqlite")
|
||||||
|
viper.Set("api_auth_token", testAPIToken)
|
||||||
|
viper.Set("epay_pid", 1)
|
||||||
|
viper.Set("app_uri", "http://localhost:8080")
|
||||||
|
viper.Set("order_expiration_time", 10)
|
||||||
|
viper.Set("api_rate_url", "")
|
||||||
|
viper.Set("forced_usdt_rate", 7.0)
|
||||||
|
viper.Set("runtime_root_path", tmpDir)
|
||||||
|
viper.Set("log_save_path", tmpDir)
|
||||||
|
viper.Set("sqlite_database_filename", tmpDir+"/test.db")
|
||||||
|
viper.Set("runtime_sqlite_filename", tmpDir+"/runtime.db")
|
||||||
|
|
||||||
|
log.Init()
|
||||||
|
|
||||||
|
// init config paths
|
||||||
|
os.Setenv("EPUSDT_CONFIG", tmpDir)
|
||||||
|
defer os.Unsetenv("EPUSDT_CONFIG")
|
||||||
|
|
||||||
|
// init DB
|
||||||
|
if err := dao.DBInit(); err != nil {
|
||||||
|
t.Fatalf("DBInit: %v", err)
|
||||||
|
}
|
||||||
|
if err := dao.RuntimeInit(); err != nil {
|
||||||
|
t.Fatalf("RuntimeInit: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
// ensure tables exist (MdbTableInit uses sync.Once, so migrate directly)
|
||||||
|
dao.Mdb.AutoMigrate(&mdb.Orders{}, &mdb.WalletAddress{}, &mdb.SupportedAsset{})
|
||||||
|
|
||||||
|
// seed wallet addresses
|
||||||
|
dao.Mdb.Create(&mdb.WalletAddress{Network: mdb.NetworkTron, Address: "TTestTronAddress001", Status: mdb.TokenStatusEnable})
|
||||||
|
dao.Mdb.Create(&mdb.WalletAddress{Network: mdb.NetworkSolana, Address: "SolTestAddress001", Status: mdb.TokenStatusEnable})
|
||||||
|
// seed supported assets if empty
|
||||||
|
var supportCnt int64
|
||||||
|
dao.Mdb.Model(&mdb.SupportedAsset{}).Count(&supportCnt)
|
||||||
|
if supportCnt == 0 {
|
||||||
|
dao.Mdb.Create(&[]mdb.SupportedAsset{
|
||||||
|
{Network: mdb.NetworkTron, Token: "TRX", Status: mdb.TokenStatusEnable},
|
||||||
|
{Network: mdb.NetworkTron, Token: "USDT", Status: mdb.TokenStatusEnable},
|
||||||
|
{Network: mdb.NetworkSolana, Token: "SOL", Status: mdb.TokenStatusEnable},
|
||||||
|
{Network: mdb.NetworkSolana, Token: "USDT", Status: mdb.TokenStatusEnable},
|
||||||
|
{Network: mdb.NetworkSolana, Token: "USDC", Status: mdb.TokenStatusEnable},
|
||||||
|
{Network: mdb.NetworkEthereum, Token: "USDT", Status: mdb.TokenStatusEnable},
|
||||||
|
{Network: mdb.NetworkEthereum, Token: "USDC", Status: mdb.TokenStatusEnable},
|
||||||
|
{Network: mdb.NetworkBsc, Token: "USDT", Status: mdb.TokenStatusEnable},
|
||||||
|
{Network: mdb.NetworkBsc, Token: "USDC", Status: mdb.TokenStatusEnable},
|
||||||
|
{Network: mdb.NetworkPolygon, Token: "USDT", Status: mdb.TokenStatusEnable},
|
||||||
|
{Network: mdb.NetworkPolygon, Token: "USDC", Status: mdb.TokenStatusEnable},
|
||||||
|
{Network: mdb.NetworkPlasma, Token: "USDT", Status: mdb.TokenStatusEnable},
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
e := echo.New()
|
||||||
|
RegisterRoute(e)
|
||||||
|
return e
|
||||||
|
}
|
||||||
|
|
||||||
|
func signBody(body map[string]interface{}) map[string]interface{} {
|
||||||
|
sig, _ := sign.Get(body, testAPIToken)
|
||||||
|
body["signature"] = sig
|
||||||
|
return body
|
||||||
|
}
|
||||||
|
|
||||||
|
func doPost(e *echo.Echo, path string, body map[string]interface{}) *httptest.ResponseRecorder {
|
||||||
|
jsonBytes, _ := json.Marshal(body)
|
||||||
|
req := httptest.NewRequest(http.MethodPost, path, strings.NewReader(string(jsonBytes)))
|
||||||
|
req.Header.Set(echo.HeaderContentType, echo.MIMEApplicationJSON)
|
||||||
|
rec := httptest.NewRecorder()
|
||||||
|
e.ServeHTTP(rec, req)
|
||||||
|
return rec
|
||||||
|
}
|
||||||
|
|
||||||
|
func signEpayValues(values url.Values) url.Values {
|
||||||
|
signParams := make(map[string]interface{})
|
||||||
|
for key, items := range values {
|
||||||
|
if key == "sign" || key == "sign_type" || len(items) == 0 {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
signParams[key] = items[0]
|
||||||
|
}
|
||||||
|
sig, _ := sign.Get(signParams, testAPIToken)
|
||||||
|
values.Set("sign", sig)
|
||||||
|
values.Set("sign_type", "MD5")
|
||||||
|
return values
|
||||||
|
}
|
||||||
|
|
||||||
|
func doFormPost(e *echo.Echo, path string, values url.Values) *httptest.ResponseRecorder {
|
||||||
|
req := httptest.NewRequest(http.MethodPost, path, strings.NewReader(values.Encode()))
|
||||||
|
req.Header.Set(echo.HeaderContentType, echo.MIMEApplicationForm)
|
||||||
|
rec := httptest.NewRecorder()
|
||||||
|
e.ServeHTTP(rec, req)
|
||||||
|
return rec
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestCreateOrderEpusdtDefaultTron tests the epusdt compatibility route defaults to tron network.
|
||||||
|
func TestCreateOrderEpusdtDefaultTron(t *testing.T) {
|
||||||
|
e := setupTestEnv(t)
|
||||||
|
|
||||||
|
body := signBody(map[string]interface{}{
|
||||||
|
"order_id": "test-tron-001",
|
||||||
|
"amount": 1.00,
|
||||||
|
"notify_url": "http://localhost/notify",
|
||||||
|
})
|
||||||
|
|
||||||
|
rec := doPost(e, "/payments/epusdt/v1/order/create-transaction", body)
|
||||||
|
t.Logf("Status: %d, Body: %s", rec.Code, rec.Body.String())
|
||||||
|
|
||||||
|
if rec.Code != http.StatusOK {
|
||||||
|
t.Fatalf("expected 200, got %d", rec.Code)
|
||||||
|
}
|
||||||
|
|
||||||
|
var resp map[string]interface{}
|
||||||
|
if err := json.Unmarshal(rec.Body.Bytes(), &resp); err != nil {
|
||||||
|
t.Fatalf("unmarshal: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
data, ok := resp["data"].(map[string]interface{})
|
||||||
|
if !ok {
|
||||||
|
t.Fatalf("expected data in response, got: %v", resp)
|
||||||
|
}
|
||||||
|
|
||||||
|
if data["trade_id"] == nil || data["trade_id"] == "" {
|
||||||
|
t.Error("expected trade_id in response")
|
||||||
|
}
|
||||||
|
if data["receive_address"] != "TTestTronAddress001" {
|
||||||
|
t.Errorf("expected tron address, got: %v", data["receive_address"])
|
||||||
|
}
|
||||||
|
t.Logf("Order created: trade_id=%v address=%v amount=%v", data["trade_id"], data["receive_address"], data["actual_amount"])
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestCreateOrderGmpayV1Solana tests the gmpay route with solana network.
|
||||||
|
func TestCreateOrderGmpayV1Solana(t *testing.T) {
|
||||||
|
e := setupTestEnv(t)
|
||||||
|
|
||||||
|
body := signBody(map[string]interface{}{
|
||||||
|
"order_id": "test-sol-001",
|
||||||
|
"amount": 1.00,
|
||||||
|
"token": "usdt",
|
||||||
|
"currency": "cny",
|
||||||
|
"network": "solana",
|
||||||
|
"notify_url": "http://localhost/notify",
|
||||||
|
})
|
||||||
|
|
||||||
|
rec := doPost(e, "/payments/gmpay/v1/order/create-transaction", body)
|
||||||
|
t.Logf("Status: %d, Body: %s", rec.Code, rec.Body.String())
|
||||||
|
|
||||||
|
if rec.Code != http.StatusOK {
|
||||||
|
t.Fatalf("expected 200, got %d", rec.Code)
|
||||||
|
}
|
||||||
|
|
||||||
|
var resp map[string]interface{}
|
||||||
|
if err := json.Unmarshal(rec.Body.Bytes(), &resp); err != nil {
|
||||||
|
t.Fatalf("unmarshal: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
data, ok := resp["data"].(map[string]interface{})
|
||||||
|
if !ok {
|
||||||
|
t.Fatalf("expected data in response, got: %v", resp)
|
||||||
|
}
|
||||||
|
|
||||||
|
if data["trade_id"] == nil || data["trade_id"] == "" {
|
||||||
|
t.Error("expected trade_id in response")
|
||||||
|
}
|
||||||
|
if data["receive_address"] != "SolTestAddress001" {
|
||||||
|
t.Errorf("expected solana address, got: %v", data["receive_address"])
|
||||||
|
}
|
||||||
|
t.Logf("Order created: trade_id=%v address=%v amount=%v", data["trade_id"], data["receive_address"], data["actual_amount"])
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestCreateOrderGmpayV1SolNative tests creating an order for native SOL token.
|
||||||
|
func TestCreateOrderGmpayV1SolNative(t *testing.T) {
|
||||||
|
e := setupTestEnv(t)
|
||||||
|
|
||||||
|
body := signBody(map[string]interface{}{
|
||||||
|
"order_id": "test-sol-native-001",
|
||||||
|
"amount": 0.05,
|
||||||
|
"token": "sol",
|
||||||
|
"currency": "usd",
|
||||||
|
"network": "solana",
|
||||||
|
"notify_url": "http://localhost/notify",
|
||||||
|
})
|
||||||
|
|
||||||
|
rec := doPost(e, "/payments/gmpay/v1/order/create-transaction", body)
|
||||||
|
t.Logf("Status: %d, Body: %s", rec.Code, rec.Body.String())
|
||||||
|
|
||||||
|
var resp map[string]interface{}
|
||||||
|
json.Unmarshal(rec.Body.Bytes(), &resp)
|
||||||
|
t.Logf("Response: %v", resp)
|
||||||
|
|
||||||
|
// This may fail if rate API is not configured, which is expected in test
|
||||||
|
// The important thing is the route accepts the request with network=solana token=sol
|
||||||
|
if rec.Code != http.StatusOK {
|
||||||
|
t.Logf("Note: non-200 may be expected if rate API is not configured for SOL")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func doGet(e *echo.Echo, path string) *httptest.ResponseRecorder {
|
||||||
|
req := httptest.NewRequest(http.MethodGet, path, nil)
|
||||||
|
req.Header.Set("Authorization", testAPIToken)
|
||||||
|
rec := httptest.NewRecorder()
|
||||||
|
e.ServeHTTP(rec, req)
|
||||||
|
return rec
|
||||||
|
}
|
||||||
|
|
||||||
|
func doPostWithToken(e *echo.Echo, path string, body map[string]interface{}) *httptest.ResponseRecorder {
|
||||||
|
jsonBytes, _ := json.Marshal(body)
|
||||||
|
req := httptest.NewRequest(http.MethodPost, path, strings.NewReader(string(jsonBytes)))
|
||||||
|
req.Header.Set(echo.HeaderContentType, echo.MIMEApplicationJSON)
|
||||||
|
req.Header.Set("Authorization", testAPIToken)
|
||||||
|
rec := httptest.NewRecorder()
|
||||||
|
e.ServeHTTP(rec, req)
|
||||||
|
return rec
|
||||||
|
}
|
||||||
|
|
||||||
|
func parseResp(t *testing.T, rec *httptest.ResponseRecorder) map[string]interface{} {
|
||||||
|
t.Helper()
|
||||||
|
var resp map[string]interface{}
|
||||||
|
if err := json.Unmarshal(rec.Body.Bytes(), &resp); err != nil {
|
||||||
|
t.Fatalf("unmarshal: %v", err)
|
||||||
|
}
|
||||||
|
return resp
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestGetSupportedAssetsPublic(t *testing.T) {
|
||||||
|
e := setupTestEnv(t)
|
||||||
|
|
||||||
|
req := httptest.NewRequest(http.MethodGet, "/payments/gmpay/v1/supported-assets", nil)
|
||||||
|
rec := httptest.NewRecorder()
|
||||||
|
e.ServeHTTP(rec, req)
|
||||||
|
|
||||||
|
if rec.Code != http.StatusOK {
|
||||||
|
t.Fatalf("expected 200, got %d, body=%s", rec.Code, rec.Body.String())
|
||||||
|
}
|
||||||
|
|
||||||
|
resp := parseResp(t, rec)
|
||||||
|
if resp["status_code"].(float64) != 200 {
|
||||||
|
t.Fatalf("expected status_code=200, got %v", resp["status_code"])
|
||||||
|
}
|
||||||
|
|
||||||
|
data, ok := resp["data"].(map[string]interface{})
|
||||||
|
if !ok {
|
||||||
|
t.Fatalf("expected object data, got %T", resp["data"])
|
||||||
|
}
|
||||||
|
supports, ok := data["supports"].([]interface{})
|
||||||
|
if !ok {
|
||||||
|
t.Fatalf("expected supports array, got %T", data["supports"])
|
||||||
|
}
|
||||||
|
if len(supports) < 2 {
|
||||||
|
t.Fatalf("expected >= 2 network supports, got %d", len(supports))
|
||||||
|
}
|
||||||
|
|
||||||
|
seen := map[string]bool{}
|
||||||
|
for _, item := range supports {
|
||||||
|
row := item.(map[string]interface{})
|
||||||
|
network := row["network"].(string)
|
||||||
|
seen[network] = true
|
||||||
|
}
|
||||||
|
for _, n := range []string{"tron", "solana"} {
|
||||||
|
if !seen[n] {
|
||||||
|
t.Fatalf("missing network support: %s", n)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestSupportedAssetCRUD(t *testing.T) {
|
||||||
|
e := setupTestEnv(t)
|
||||||
|
|
||||||
|
// public query list (no auth)
|
||||||
|
req := httptest.NewRequest(http.MethodGet, "/payments/gmpay/v1/supported-assets/records", nil)
|
||||||
|
rec := httptest.NewRecorder()
|
||||||
|
e.ServeHTTP(rec, req)
|
||||||
|
resp := parseResp(t, rec)
|
||||||
|
if resp["status_code"].(float64) != 200 {
|
||||||
|
t.Fatalf("public list failed: %v", resp)
|
||||||
|
}
|
||||||
|
|
||||||
|
// create requires auth
|
||||||
|
rec = doPostWithToken(e, "/payments/gmpay/v1/supported-assets/add", map[string]interface{}{
|
||||||
|
"network": "arb",
|
||||||
|
"token": "usdt",
|
||||||
|
"status": 1,
|
||||||
|
})
|
||||||
|
resp = parseResp(t, rec)
|
||||||
|
if resp["status_code"].(float64) != 200 {
|
||||||
|
t.Fatalf("add supported asset failed: %v", resp)
|
||||||
|
}
|
||||||
|
created := resp["data"].(map[string]interface{})
|
||||||
|
assetID := fmt.Sprintf("%.0f", created["id"].(float64))
|
||||||
|
|
||||||
|
// get by id is public
|
||||||
|
req = httptest.NewRequest(http.MethodGet, "/payments/gmpay/v1/supported-assets/"+assetID, nil)
|
||||||
|
rec = httptest.NewRecorder()
|
||||||
|
e.ServeHTTP(rec, req)
|
||||||
|
resp = parseResp(t, rec)
|
||||||
|
if resp["status_code"].(float64) != 200 {
|
||||||
|
t.Fatalf("public get by id failed: %v", resp)
|
||||||
|
}
|
||||||
|
|
||||||
|
// update requires auth
|
||||||
|
rec = doPostWithToken(e, "/payments/gmpay/v1/supported-assets/"+assetID+"/update", map[string]interface{}{
|
||||||
|
"network": "arbitrum",
|
||||||
|
"token": "usdc",
|
||||||
|
"status": 1,
|
||||||
|
})
|
||||||
|
resp = parseResp(t, rec)
|
||||||
|
if resp["status_code"].(float64) != 200 {
|
||||||
|
t.Fatalf("update supported asset failed: %v", resp)
|
||||||
|
}
|
||||||
|
updated := resp["data"].(map[string]interface{})
|
||||||
|
if updated["network"] != "arbitrum" || updated["token"] != "USDC" {
|
||||||
|
t.Fatalf("unexpected updated data: %v", updated)
|
||||||
|
}
|
||||||
|
|
||||||
|
// delete requires auth
|
||||||
|
rec = doPostWithToken(e, "/payments/gmpay/v1/supported-assets/"+assetID+"/delete", nil)
|
||||||
|
resp = parseResp(t, rec)
|
||||||
|
if resp["status_code"].(float64) != 200 {
|
||||||
|
t.Fatalf("delete supported asset failed: %v", resp)
|
||||||
|
}
|
||||||
|
|
||||||
|
// recreate after delete should restore soft-deleted row, not unique-conflict
|
||||||
|
rec = doPostWithToken(e, "/payments/gmpay/v1/supported-assets/add", map[string]interface{}{
|
||||||
|
"network": "arbitrum",
|
||||||
|
"token": "usdc",
|
||||||
|
"status": 1,
|
||||||
|
})
|
||||||
|
resp = parseResp(t, rec)
|
||||||
|
if resp["status_code"].(float64) != 200 {
|
||||||
|
t.Fatalf("recreate after delete failed: %v", resp)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestWalletAddAndList tests adding wallets via API and listing them.
|
||||||
|
func TestWalletAddAndList(t *testing.T) {
|
||||||
|
e := setupTestEnv(t)
|
||||||
|
|
||||||
|
// Add a solana wallet
|
||||||
|
rec := doPostWithToken(e, "/payments/gmpay/v1/wallet/add", map[string]interface{}{
|
||||||
|
"network": "solana",
|
||||||
|
"address": "NewSolWallet001",
|
||||||
|
})
|
||||||
|
t.Logf("Add: %s", rec.Body.String())
|
||||||
|
resp := parseResp(t, rec)
|
||||||
|
if resp["status_code"].(float64) != 200 {
|
||||||
|
t.Fatalf("add wallet failed: %v", resp)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Add a tron wallet
|
||||||
|
rec = doPostWithToken(e, "/payments/gmpay/v1/wallet/add", map[string]interface{}{
|
||||||
|
"network": "tron",
|
||||||
|
"address": "NewTronWallet001",
|
||||||
|
})
|
||||||
|
resp = parseResp(t, rec)
|
||||||
|
if resp["status_code"].(float64) != 200 {
|
||||||
|
t.Fatalf("add tron wallet failed: %v", resp)
|
||||||
|
}
|
||||||
|
|
||||||
|
// List all wallets
|
||||||
|
rec = doGet(e, "/payments/gmpay/v1/wallet/list")
|
||||||
|
resp = parseResp(t, rec)
|
||||||
|
wallets := resp["data"].([]interface{})
|
||||||
|
// 2 seeded + 2 added = 4
|
||||||
|
if len(wallets) != 4 {
|
||||||
|
t.Fatalf("expected 4 wallets, got %d: %v", len(wallets), wallets)
|
||||||
|
}
|
||||||
|
|
||||||
|
// List by network
|
||||||
|
rec = doGet(e, "/payments/gmpay/v1/wallet/list?network=solana")
|
||||||
|
resp = parseResp(t, rec)
|
||||||
|
wallets = resp["data"].([]interface{})
|
||||||
|
if len(wallets) != 2 {
|
||||||
|
t.Fatalf("expected 2 solana wallets, got %d", len(wallets))
|
||||||
|
}
|
||||||
|
|
||||||
|
rec = doGet(e, "/payments/gmpay/v1/wallet/list?network=tron")
|
||||||
|
resp = parseResp(t, rec)
|
||||||
|
wallets = resp["data"].([]interface{})
|
||||||
|
if len(wallets) != 2 {
|
||||||
|
t.Fatalf("expected 2 tron wallets, got %d", len(wallets))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestWalletDuplicateRejected tests that adding the same network+address twice fails.
|
||||||
|
func TestWalletDuplicateRejected(t *testing.T) {
|
||||||
|
e := setupTestEnv(t)
|
||||||
|
|
||||||
|
body := map[string]interface{}{"network": "solana", "address": "DupWallet001"}
|
||||||
|
rec := doPostWithToken(e, "/payments/gmpay/v1/wallet/add", body)
|
||||||
|
resp := parseResp(t, rec)
|
||||||
|
if resp["status_code"].(float64) != 200 {
|
||||||
|
t.Fatalf("first add failed: %v", resp)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Same network+address should fail
|
||||||
|
rec = doPostWithToken(e, "/payments/gmpay/v1/wallet/add", body)
|
||||||
|
resp = parseResp(t, rec)
|
||||||
|
if resp["status_code"].(float64) == 200 {
|
||||||
|
t.Fatal("expected duplicate to be rejected")
|
||||||
|
}
|
||||||
|
t.Logf("Duplicate rejected: %v", resp["message"])
|
||||||
|
|
||||||
|
// Same address, different network should succeed
|
||||||
|
rec = doPostWithToken(e, "/payments/gmpay/v1/wallet/add", map[string]interface{}{
|
||||||
|
"network": "tron",
|
||||||
|
"address": "DupWallet001",
|
||||||
|
})
|
||||||
|
resp = parseResp(t, rec)
|
||||||
|
if resp["status_code"].(float64) != 200 {
|
||||||
|
t.Fatalf("same address on different network should succeed: %v", resp)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestWalletStatusAndDelete tests enable/disable/delete operations.
|
||||||
|
func TestWalletStatusAndDelete(t *testing.T) {
|
||||||
|
e := setupTestEnv(t)
|
||||||
|
|
||||||
|
// Add a wallet
|
||||||
|
rec := doPostWithToken(e, "/payments/gmpay/v1/wallet/add", map[string]interface{}{
|
||||||
|
"network": "solana",
|
||||||
|
"address": "StatusTestWallet",
|
||||||
|
})
|
||||||
|
resp := parseResp(t, rec)
|
||||||
|
wallet := resp["data"].(map[string]interface{})
|
||||||
|
walletID := fmt.Sprintf("%.0f", wallet["id"].(float64))
|
||||||
|
|
||||||
|
// Get wallet
|
||||||
|
rec = doGet(e, "/payments/gmpay/v1/wallet/"+walletID)
|
||||||
|
resp = parseResp(t, rec)
|
||||||
|
if resp["status_code"].(float64) != 200 {
|
||||||
|
t.Fatalf("get wallet failed: %v", resp)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Disable wallet
|
||||||
|
rec = doPostWithToken(e, "/payments/gmpay/v1/wallet/"+walletID+"/status", map[string]interface{}{
|
||||||
|
"status": 2,
|
||||||
|
})
|
||||||
|
resp = parseResp(t, rec)
|
||||||
|
if resp["status_code"].(float64) != 200 {
|
||||||
|
t.Fatalf("disable wallet failed: %v", resp)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify disabled — should not appear in available list
|
||||||
|
rec = doGet(e, "/payments/gmpay/v1/wallet/list?network=solana")
|
||||||
|
resp = parseResp(t, rec)
|
||||||
|
wallets := resp["data"].([]interface{})
|
||||||
|
for _, w := range wallets {
|
||||||
|
wm := w.(map[string]interface{})
|
||||||
|
if wm["address"] == "StatusTestWallet" && wm["status"].(float64) != 2 {
|
||||||
|
t.Error("wallet should be disabled")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Delete wallet
|
||||||
|
rec = doPostWithToken(e, "/payments/gmpay/v1/wallet/"+walletID+"/delete", nil)
|
||||||
|
resp = parseResp(t, rec)
|
||||||
|
if resp["status_code"].(float64) != 200 {
|
||||||
|
t.Fatalf("delete wallet failed: %v", resp)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Verify deleted
|
||||||
|
rec = doGet(e, "/payments/gmpay/v1/wallet/"+walletID)
|
||||||
|
resp = parseResp(t, rec)
|
||||||
|
// Should return not found
|
||||||
|
if resp["status_code"].(float64) == 200 {
|
||||||
|
data := resp["data"].(map[string]interface{})
|
||||||
|
if data["id"].(float64) > 0 {
|
||||||
|
t.Error("wallet should be deleted")
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestWalletAuthRequired tests that wallet APIs require auth token.
|
||||||
|
func TestWalletAuthRequired(t *testing.T) {
|
||||||
|
e := setupTestEnv(t)
|
||||||
|
|
||||||
|
// No auth header — should not return success
|
||||||
|
req := httptest.NewRequest(http.MethodGet, "/payments/gmpay/v1/wallet/list", nil)
|
||||||
|
rec := httptest.NewRecorder()
|
||||||
|
e.ServeHTTP(rec, req)
|
||||||
|
|
||||||
|
// The response should indicate auth failure (not 200 success)
|
||||||
|
var resp map[string]interface{}
|
||||||
|
if err := json.Unmarshal(rec.Body.Bytes(), &resp); err != nil {
|
||||||
|
// echo may return plain text error
|
||||||
|
if rec.Code == http.StatusOK {
|
||||||
|
t.Error("expected auth failure without token")
|
||||||
|
}
|
||||||
|
t.Logf("Auth rejected (non-JSON): status=%d body=%s", rec.Code, rec.Body.String())
|
||||||
|
return
|
||||||
|
}
|
||||||
|
statusCode, _ := resp["status_code"].(float64)
|
||||||
|
if statusCode == 200 {
|
||||||
|
t.Error("expected auth failure without token")
|
||||||
|
}
|
||||||
|
t.Logf("Auth rejected: %v", resp)
|
||||||
|
}
|
||||||
|
|
||||||
|
// TestCreateOrderNetworkIsolation verifies tron and solana wallets don't mix.
|
||||||
|
func TestCreateOrderNetworkIsolation(t *testing.T) {
|
||||||
|
e := setupTestEnv(t)
|
||||||
|
|
||||||
|
// Try to create a solana order — should get solana address, not tron
|
||||||
|
body := signBody(map[string]interface{}{
|
||||||
|
"order_id": fmt.Sprintf("test-isolation-%d", 1),
|
||||||
|
"amount": 1.00,
|
||||||
|
"token": "usdt",
|
||||||
|
"currency": "cny",
|
||||||
|
"network": "solana",
|
||||||
|
"notify_url": "http://localhost/notify",
|
||||||
|
})
|
||||||
|
rec := doPost(e, "/payments/gmpay/v1/order/create-transaction", body)
|
||||||
|
|
||||||
|
var resp map[string]interface{}
|
||||||
|
json.Unmarshal(rec.Body.Bytes(), &resp)
|
||||||
|
|
||||||
|
data, ok := resp["data"].(map[string]interface{})
|
||||||
|
if !ok {
|
||||||
|
t.Fatalf("expected data, got: %v", resp)
|
||||||
|
}
|
||||||
|
if data["receive_address"] == "TTestTronAddress001" {
|
||||||
|
t.Error("solana order should NOT get a tron address")
|
||||||
|
}
|
||||||
|
if data["receive_address"] != "SolTestAddress001" {
|
||||||
|
t.Errorf("expected SolTestAddress001, got %v", data["receive_address"])
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestEpaySubmitPhpGetCompatible(t *testing.T) {
|
||||||
|
e := setupTestEnv(t)
|
||||||
|
|
||||||
|
values := signEpayValues(url.Values{
|
||||||
|
"pid": {"1"},
|
||||||
|
"name": {"epay-get-001"},
|
||||||
|
"type": {"alipay"},
|
||||||
|
"money": {"1.00"},
|
||||||
|
"out_trade_no": {"epay-get-001"},
|
||||||
|
"notify_url": {"http://localhost/notify"},
|
||||||
|
"return_url": {"http://localhost/return"},
|
||||||
|
})
|
||||||
|
|
||||||
|
req := httptest.NewRequest(http.MethodGet, "/payments/epay/v1/order/create-transaction/submit.php?"+values.Encode(), nil)
|
||||||
|
rec := httptest.NewRecorder()
|
||||||
|
e.ServeHTTP(rec, req)
|
||||||
|
|
||||||
|
if rec.Code != http.StatusFound {
|
||||||
|
t.Fatalf("expected 302, got %d body=%s", rec.Code, rec.Body.String())
|
||||||
|
}
|
||||||
|
if !strings.HasPrefix(rec.Header().Get("Location"), "/pay/checkout-counter/") {
|
||||||
|
t.Fatalf("expected checkout redirect, got %q", rec.Header().Get("Location"))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestEpaySubmitPhpPostFormCompatible(t *testing.T) {
|
||||||
|
e := setupTestEnv(t)
|
||||||
|
|
||||||
|
values := signEpayValues(url.Values{
|
||||||
|
"pid": {"1"},
|
||||||
|
"name": {"epay-post-001"},
|
||||||
|
"type": {"alipay"},
|
||||||
|
"money": {"1.00"},
|
||||||
|
"out_trade_no": {"epay-post-001"},
|
||||||
|
"notify_url": {"http://localhost/notify"},
|
||||||
|
"return_url": {"http://localhost/return"},
|
||||||
|
"sitename": {"example-shop"},
|
||||||
|
})
|
||||||
|
|
||||||
|
rec := doFormPost(e, "/payments/epay/v1/order/create-transaction/submit.php", values)
|
||||||
|
|
||||||
|
if rec.Code != http.StatusFound {
|
||||||
|
t.Fatalf("expected 302, got %d body=%s", rec.Code, rec.Body.String())
|
||||||
|
}
|
||||||
|
if !strings.HasPrefix(rec.Header().Get("Location"), "/pay/checkout-counter/") {
|
||||||
|
t.Fatalf("expected checkout redirect, got %q", rec.Header().Get("Location"))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
After Width: | Height: | Size: 2.0 KiB |
|
After Width: | Height: | Size: 597 B |
|
After Width: | Height: | Size: 43 KiB |
|
After Width: | Height: | Size: 2.0 KiB |
|
After Width: | Height: | Size: 1.7 KiB |
|
After Width: | Height: | Size: 952 B |
|
After Width: | Height: | Size: 1.7 KiB |
|
After Width: | Height: | Size: 1.5 KiB |
|
After Width: | Height: | Size: 1.5 KiB |
|
After Width: | Height: | Size: 1.9 KiB |
|
After Width: | Height: | Size: 1.3 KiB |
|
After Width: | Height: | Size: 1.0 KiB |
@@ -1,44 +0,0 @@
|
|||||||
html, body, div, span, applet, object, iframe,
|
|
||||||
h1, h2, h3, h4, h5, h6, p, blockquote, pre,
|
|
||||||
a, abbr, acronym, address, big, cite, code,
|
|
||||||
del, dfn, em, img, ins, kbd, q, s, samp,
|
|
||||||
small, strike, strong, sub, sup, tt, var,
|
|
||||||
b, u, i, center,
|
|
||||||
dl, dt, dd, ol, ul, li,
|
|
||||||
fieldset, form, label, legend,
|
|
||||||
table, caption, tbody, tfoot, thead, tr, th, td,
|
|
||||||
article, aside, canvas, details, embed,
|
|
||||||
figure, figcaption, footer, header, hgroup,
|
|
||||||
menu, nav, output, ruby, section, summary,
|
|
||||||
time, mark, audio, video {
|
|
||||||
margin: 0;
|
|
||||||
padding: 0;
|
|
||||||
border: 0;
|
|
||||||
font-size: 100%;
|
|
||||||
font: inherit;
|
|
||||||
vertical-align: baseline;
|
|
||||||
}
|
|
||||||
/* HTML5 display-role reset for older browsers */
|
|
||||||
article, aside, details, figcaption, figure,
|
|
||||||
footer, header, hgroup, menu, nav, section {
|
|
||||||
display: block;
|
|
||||||
}
|
|
||||||
body {
|
|
||||||
line-height: 1;
|
|
||||||
background-color: #fafbfc;
|
|
||||||
}
|
|
||||||
ol, ul {
|
|
||||||
list-style: none;
|
|
||||||
}
|
|
||||||
blockquote, q {
|
|
||||||
quotes: none;
|
|
||||||
}
|
|
||||||
blockquote:before, blockquote:after,
|
|
||||||
q:before, q:after {
|
|
||||||
content: '';
|
|
||||||
content: none;
|
|
||||||
}
|
|
||||||
table {
|
|
||||||
border-collapse: collapse;
|
|
||||||
border-spacing: 0;
|
|
||||||
}
|
|
||||||
@@ -7,14 +7,20 @@ import (
|
|||||||
|
|
||||||
func Start() {
|
func Start() {
|
||||||
log.Sugar.Info("[task] Starting task scheduler...")
|
log.Sugar.Info("[task] Starting task scheduler...")
|
||||||
|
go StartEthereumWebSocketListener()
|
||||||
|
go StartBscWebSocketListener()
|
||||||
|
go StartPolygonWebSocketListener()
|
||||||
|
go StartPlasmaWebSocketListener()
|
||||||
|
go StartTronTrc20WebSocketListener()
|
||||||
|
|
||||||
c := cron.New()
|
c := cron.New()
|
||||||
// trc20钱包监听
|
// solana钱包监听
|
||||||
_, err := c.AddJob("@every 5s", ListenTrc20Job{})
|
_, err := c.AddJob("@every 5s", ListenSolJob{})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Sugar.Errorf("[task] Failed to add ListenTrc20Job: %v", err)
|
log.Sugar.Errorf("[task] Failed to add ListenSolJob: %v", err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
log.Sugar.Info("[task] ListenTrc20Job scheduled successfully (@every 5s)")
|
log.Sugar.Info("[task] ListenSolJob scheduled successfully (@every 5s)")
|
||||||
c.Start()
|
c.Start()
|
||||||
log.Sugar.Info("[task] Task scheduler started")
|
log.Sugar.Info("[task] Task scheduler started")
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,112 @@
|
|||||||
|
package task
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"math/big"
|
||||||
|
"strings"
|
||||||
|
"sync/atomic"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/assimon/luuu/model/data"
|
||||||
|
"github.com/assimon/luuu/model/mdb"
|
||||||
|
"github.com/assimon/luuu/model/service"
|
||||||
|
"github.com/assimon/luuu/util/log"
|
||||||
|
|
||||||
|
"github.com/ethereum/go-ethereum"
|
||||||
|
"github.com/ethereum/go-ethereum/common"
|
||||||
|
"github.com/ethereum/go-ethereum/core/types"
|
||||||
|
"github.com/ethereum/go-ethereum/ethclient"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
// USDT / USDC 合约地址(BSC 主网)
|
||||||
|
bscUsdtContract = common.HexToAddress("0x55d398326f99059fF775485246999027B3197955")
|
||||||
|
bscUsdcContract = common.HexToAddress("0x8AC76a51cc950d9822D68b83fE1Ad97B32Cd580d")
|
||||||
|
)
|
||||||
|
|
||||||
|
type bscRecipientSnapshot struct {
|
||||||
|
addrs map[string]struct{}
|
||||||
|
}
|
||||||
|
|
||||||
|
var bscWatchedRecipients atomic.Pointer[bscRecipientSnapshot]
|
||||||
|
|
||||||
|
func StartBscWebSocketListener() {
|
||||||
|
wallets, err := data.GetAvailableWalletAddressByNetwork(mdb.NetworkBsc)
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Errorf("[BSC-WS] Failed to get wallet addresses: %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
storeBscRecipientsFromWallets(wallets)
|
||||||
|
go func() {
|
||||||
|
ticker := time.NewTicker(5 * time.Second)
|
||||||
|
defer ticker.Stop()
|
||||||
|
for range ticker.C {
|
||||||
|
w, err := data.GetAvailableWalletAddressByNetwork(mdb.NetworkBsc)
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Warnf("[BSC-WS] refresh wallet addresses: %v", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
storeBscRecipientsFromWallets(w)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
wsURL := "wss://bsc.drpc.org"
|
||||||
|
query := ethereum.FilterQuery{
|
||||||
|
Addresses: []common.Address{
|
||||||
|
bscUsdtContract,
|
||||||
|
bscUsdcContract,
|
||||||
|
},
|
||||||
|
Topics: [][]common.Hash{},
|
||||||
|
}
|
||||||
|
|
||||||
|
runWsLogListener("[BSC-WS]", wsURL, query, func(client *ethclient.Client, vLog types.Log) {
|
||||||
|
if len(vLog.Topics) < 3 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
event := vLog.Topics[0].String()
|
||||||
|
if event != transferEventHash.String() {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
amount := new(big.Int).SetBytes(vLog.Data)
|
||||||
|
|
||||||
|
toAddr := common.HexToAddress(vLog.Topics[2].Hex())
|
||||||
|
|
||||||
|
if !isWatchedBscRecipient(toAddr) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
var blockTsMs int64
|
||||||
|
header, err := client.HeaderByNumber(context.Background(), big.NewInt(int64(vLog.BlockNumber)))
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Warnf("[BSC-WS] HeaderByNumber block=%d: %v, using local time", vLog.BlockNumber, err)
|
||||||
|
blockTsMs = time.Now().UnixMilli()
|
||||||
|
} else {
|
||||||
|
blockTsMs = int64(header.Time) * 1000
|
||||||
|
}
|
||||||
|
|
||||||
|
service.TryProcessEvmERC20Transfer(mdb.NetworkBsc, vLog.Address, toAddr, amount, vLog.TxHash.Hex(), blockTsMs)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func storeBscRecipientsFromWallets(wallets []mdb.WalletAddress) int {
|
||||||
|
m := make(map[string]struct{})
|
||||||
|
for _, w := range wallets {
|
||||||
|
a := strings.TrimSpace(w.Address)
|
||||||
|
if !common.IsHexAddress(a) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
m[strings.ToLower(common.HexToAddress(a).Hex())] = struct{}{}
|
||||||
|
}
|
||||||
|
bscWatchedRecipients.Store(&bscRecipientSnapshot{addrs: m})
|
||||||
|
return len(m)
|
||||||
|
}
|
||||||
|
|
||||||
|
func isWatchedBscRecipient(to common.Address) bool {
|
||||||
|
snap := bscWatchedRecipients.Load()
|
||||||
|
if snap == nil || len(snap.addrs) == 0 {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
_, ok := snap.addrs[strings.ToLower(to.Hex())]
|
||||||
|
return ok
|
||||||
|
}
|
||||||
@@ -0,0 +1,132 @@
|
|||||||
|
package task
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"math/big"
|
||||||
|
"strings"
|
||||||
|
"sync/atomic"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/assimon/luuu/model/data"
|
||||||
|
"github.com/assimon/luuu/model/mdb"
|
||||||
|
"github.com/assimon/luuu/model/service"
|
||||||
|
"github.com/assimon/luuu/util/log"
|
||||||
|
|
||||||
|
"github.com/ethereum/go-ethereum"
|
||||||
|
"github.com/ethereum/go-ethereum/common"
|
||||||
|
"github.com/ethereum/go-ethereum/core/types"
|
||||||
|
"github.com/ethereum/go-ethereum/ethclient"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
|
||||||
|
// USDT / USDC 合约地址(ETH 主网)
|
||||||
|
usdtContract = common.HexToAddress("0xdAC17F958D2ee523a2206206994597C13D831ec7")
|
||||||
|
usdcContract = common.HexToAddress("0xA0b86991c6218b36c1d19d4a2e9eb0ce3606eb48")
|
||||||
|
|
||||||
|
// Transfer 事件签名
|
||||||
|
transferEventHash = common.HexToHash("0xddf252ad1be2c89b69c2b068fc378daa952ba7f163c4a11628f55a4df523b3ef")
|
||||||
|
)
|
||||||
|
|
||||||
|
type ethRecipientSnapshot struct {
|
||||||
|
addrs map[string]struct{}
|
||||||
|
}
|
||||||
|
|
||||||
|
var ethWatchedRecipients atomic.Pointer[ethRecipientSnapshot]
|
||||||
|
|
||||||
|
func StartEthereumWebSocketListener() {
|
||||||
|
wallets, err := data.GetAvailableWalletAddressByNetwork(mdb.NetworkEthereum)
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Errorf("Failed to get wallet addresses: %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
StoreEthRecipientsFromWallets(wallets)
|
||||||
|
go func() {
|
||||||
|
ticker := time.NewTicker(5 * time.Second)
|
||||||
|
defer ticker.Stop()
|
||||||
|
for range ticker.C {
|
||||||
|
w, err := data.GetAvailableWalletAddressByNetwork(mdb.NetworkEthereum)
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Warnf("[ETH-WS] refresh wallet addresses: %v", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
StoreEthRecipientsFromWallets(w)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
wsURL := "wss://ethereum.publicnode.com"
|
||||||
|
query := ethereum.FilterQuery{
|
||||||
|
Addresses: []common.Address{
|
||||||
|
usdtContract,
|
||||||
|
usdcContract,
|
||||||
|
},
|
||||||
|
Topics: [][]common.Hash{},
|
||||||
|
}
|
||||||
|
|
||||||
|
runWsLogListener("[ETH-WS]", wsURL, query, func(client *ethclient.Client, vLog types.Log) {
|
||||||
|
if len(vLog.Topics) < 3 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
event := vLog.Topics[0].String()
|
||||||
|
if event != transferEventHash.String() {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
amount := new(big.Int).SetBytes(vLog.Data)
|
||||||
|
|
||||||
|
toAddr := common.HexToAddress(vLog.Topics[2].Hex())
|
||||||
|
|
||||||
|
if !isWatchedEthRecipient(toAddr) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
var blockTsMs int64
|
||||||
|
header, err := client.HeaderByNumber(context.Background(), big.NewInt(int64(vLog.BlockNumber)))
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Warnf("[ETH-WS] HeaderByNumber block=%d: %v, using local time", vLog.BlockNumber, err)
|
||||||
|
blockTsMs = time.Now().UnixMilli()
|
||||||
|
} else {
|
||||||
|
blockTsMs = int64(header.Time) * 1000
|
||||||
|
}
|
||||||
|
|
||||||
|
service.TryProcessEvmERC20Transfer(mdb.NetworkEthereum, vLog.Address, toAddr, amount, vLog.TxHash.Hex(), blockTsMs)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func StoreEthRecipientsFromWallets(wallets []mdb.WalletAddress) int {
|
||||||
|
m := make(map[string]struct{})
|
||||||
|
for _, w := range wallets {
|
||||||
|
a := strings.TrimSpace(w.Address)
|
||||||
|
if !common.IsHexAddress(a) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
m[strings.ToLower(common.HexToAddress(a).Hex())] = struct{}{}
|
||||||
|
}
|
||||||
|
ethWatchedRecipients.Store(ðRecipientSnapshot{addrs: m})
|
||||||
|
return len(m)
|
||||||
|
}
|
||||||
|
|
||||||
|
func isWatchedEthRecipient(to common.Address) bool {
|
||||||
|
snap := ethWatchedRecipients.Load()
|
||||||
|
if snap == nil || len(snap.addrs) == 0 {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
_, ok := snap.addrs[strings.ToLower(to.Hex())]
|
||||||
|
return ok
|
||||||
|
}
|
||||||
|
|
||||||
|
func formatAmount(amount *big.Int, decimals int) string {
|
||||||
|
f := new(big.Float).SetInt(amount)
|
||||||
|
divisor := new(big.Float).SetFloat64(float64Pow(10, decimals))
|
||||||
|
result := new(big.Float).Quo(f, divisor)
|
||||||
|
|
||||||
|
return result.Text('f', 6)
|
||||||
|
}
|
||||||
|
|
||||||
|
func float64Pow(a, b int) float64 {
|
||||||
|
result := 1.0
|
||||||
|
for i := 0; i < b; i++ {
|
||||||
|
result *= float64(a)
|
||||||
|
}
|
||||||
|
return result
|
||||||
|
}
|
||||||
@@ -0,0 +1,81 @@
|
|||||||
|
package task
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/assimon/luuu/util/log"
|
||||||
|
|
||||||
|
"github.com/ethereum/go-ethereum"
|
||||||
|
"github.com/ethereum/go-ethereum/core/types"
|
||||||
|
"github.com/ethereum/go-ethereum/ethclient"
|
||||||
|
)
|
||||||
|
|
||||||
|
func runWsLogListener(logPrefix, wsURL string, query ethereum.FilterQuery, handleLog func(*ethclient.Client, types.Log)) {
|
||||||
|
const (
|
||||||
|
minBackoff = 2 * time.Second
|
||||||
|
maxBackoff = 60 * time.Second
|
||||||
|
rejoinWait = 3 * time.Second
|
||||||
|
)
|
||||||
|
failWait := minBackoff
|
||||||
|
|
||||||
|
for {
|
||||||
|
client, err := ethclient.Dial(wsURL)
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Warnf("%s dial: %v, retry in %s", logPrefix, err, failWait)
|
||||||
|
time.Sleep(failWait)
|
||||||
|
failWait = nextBackoff(failWait, maxBackoff)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
|
logsCh := make(chan types.Log)
|
||||||
|
sub, err := client.SubscribeFilterLogs(context.Background(), query, logsCh)
|
||||||
|
if err != nil {
|
||||||
|
client.Close()
|
||||||
|
log.Sugar.Warnf("%s subscribe: %v, retry in %s", logPrefix, err, failWait)
|
||||||
|
time.Sleep(failWait)
|
||||||
|
failWait = nextBackoff(failWait, maxBackoff)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
failWait = minBackoff
|
||||||
|
|
||||||
|
log.Sugar.Infof("%s connected, subscribed to USDT/USDC Transfer logs", logPrefix)
|
||||||
|
|
||||||
|
recvLoop(client, sub, logsCh, logPrefix, handleLog)
|
||||||
|
|
||||||
|
time.Sleep(rejoinWait)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func recvLoop(client *ethclient.Client, sub ethereum.Subscription, logsCh <-chan types.Log, logPrefix string, handleLog func(*ethclient.Client, types.Log)) {
|
||||||
|
defer func() {
|
||||||
|
sub.Unsubscribe()
|
||||||
|
client.Close()
|
||||||
|
}()
|
||||||
|
|
||||||
|
for {
|
||||||
|
select {
|
||||||
|
case err := <-sub.Err():
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Warnf("%s subscription error: %v, reconnecting", logPrefix, err)
|
||||||
|
} else {
|
||||||
|
log.Sugar.Warnf("%s subscription closed, reconnecting", logPrefix)
|
||||||
|
}
|
||||||
|
return
|
||||||
|
case vLog, ok := <-logsCh:
|
||||||
|
if !ok {
|
||||||
|
log.Sugar.Warnf("%s log channel closed, reconnecting", logPrefix)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
handleLog(client, vLog)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func nextBackoff(cur, max time.Duration) time.Duration {
|
||||||
|
n := cur * 2
|
||||||
|
if n > max {
|
||||||
|
return max
|
||||||
|
}
|
||||||
|
return n
|
||||||
|
}
|
||||||
@@ -0,0 +1,108 @@
|
|||||||
|
package task
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"math/big"
|
||||||
|
"strings"
|
||||||
|
"sync/atomic"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/assimon/luuu/model/data"
|
||||||
|
"github.com/assimon/luuu/model/mdb"
|
||||||
|
"github.com/assimon/luuu/model/service"
|
||||||
|
"github.com/assimon/luuu/util/log"
|
||||||
|
|
||||||
|
"github.com/ethereum/go-ethereum"
|
||||||
|
"github.com/ethereum/go-ethereum/common"
|
||||||
|
"github.com/ethereum/go-ethereum/core/types"
|
||||||
|
"github.com/ethereum/go-ethereum/ethclient"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
plasmaUsdt0Contract = common.HexToAddress("0xB8CE59FC3717ada4C02eaDF9682A9e934F625ebb")
|
||||||
|
)
|
||||||
|
|
||||||
|
type plasmaRecipientSnapshot struct {
|
||||||
|
addrs map[string]struct{}
|
||||||
|
}
|
||||||
|
|
||||||
|
var plasmaWatchedRecipients atomic.Pointer[plasmaRecipientSnapshot]
|
||||||
|
|
||||||
|
func StartPlasmaWebSocketListener() {
|
||||||
|
wallets, err := data.GetAvailableWalletAddressByNetwork(mdb.NetworkPlasma)
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Errorf("[PLASMA-WS] Failed to get wallet addresses: %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
storePlasmaRecipientsFromWallets(wallets)
|
||||||
|
go func() {
|
||||||
|
ticker := time.NewTicker(5 * time.Second)
|
||||||
|
defer ticker.Stop()
|
||||||
|
for range ticker.C {
|
||||||
|
w, err := data.GetAvailableWalletAddressByNetwork(mdb.NetworkPlasma)
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Warnf("[PLASMA-WS] refresh wallet addresses: %v", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
storePlasmaRecipientsFromWallets(w)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
// 文档提供 https://rpc.plasma.to;常见实现同主机 wss
|
||||||
|
wsURL := "wss://rpc.plasma.to"
|
||||||
|
query := ethereum.FilterQuery{
|
||||||
|
Addresses: []common.Address{plasmaUsdt0Contract},
|
||||||
|
Topics: [][]common.Hash{},
|
||||||
|
}
|
||||||
|
|
||||||
|
runWsLogListener("[PLASMA-WS]", wsURL, query, func(client *ethclient.Client, vLog types.Log) {
|
||||||
|
if len(vLog.Topics) < 3 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
event := vLog.Topics[0].String()
|
||||||
|
if event != transferEventHash.String() {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
amount := new(big.Int).SetBytes(vLog.Data)
|
||||||
|
|
||||||
|
toAddr := common.HexToAddress(vLog.Topics[2].Hex())
|
||||||
|
|
||||||
|
if !isWatchedPlasmaRecipient(toAddr) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
var blockTsMs int64
|
||||||
|
header, err := client.HeaderByNumber(context.Background(), big.NewInt(int64(vLog.BlockNumber)))
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Warnf("[PLASMA-WS] HeaderByNumber block=%d: %v, using local time", vLog.BlockNumber, err)
|
||||||
|
blockTsMs = time.Now().UnixMilli()
|
||||||
|
} else {
|
||||||
|
blockTsMs = int64(header.Time) * 1000
|
||||||
|
}
|
||||||
|
|
||||||
|
service.TryProcessEvmERC20Transfer(mdb.NetworkPlasma, vLog.Address, toAddr, amount, vLog.TxHash.Hex(), blockTsMs)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func storePlasmaRecipientsFromWallets(wallets []mdb.WalletAddress) int {
|
||||||
|
m := make(map[string]struct{})
|
||||||
|
for _, w := range wallets {
|
||||||
|
a := strings.TrimSpace(w.Address)
|
||||||
|
if !common.IsHexAddress(a) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
m[strings.ToLower(common.HexToAddress(a).Hex())] = struct{}{}
|
||||||
|
}
|
||||||
|
plasmaWatchedRecipients.Store(&plasmaRecipientSnapshot{addrs: m})
|
||||||
|
return len(m)
|
||||||
|
}
|
||||||
|
|
||||||
|
func isWatchedPlasmaRecipient(to common.Address) bool {
|
||||||
|
snap := plasmaWatchedRecipients.Load()
|
||||||
|
if snap == nil || len(snap.addrs) == 0 {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
_, ok := snap.addrs[strings.ToLower(to.Hex())]
|
||||||
|
return ok
|
||||||
|
}
|
||||||
@@ -0,0 +1,114 @@
|
|||||||
|
package task
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"math/big"
|
||||||
|
"strings"
|
||||||
|
"sync/atomic"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/assimon/luuu/model/data"
|
||||||
|
"github.com/assimon/luuu/model/mdb"
|
||||||
|
"github.com/assimon/luuu/model/service"
|
||||||
|
"github.com/assimon/luuu/util/log"
|
||||||
|
|
||||||
|
"github.com/ethereum/go-ethereum"
|
||||||
|
"github.com/ethereum/go-ethereum/common"
|
||||||
|
"github.com/ethereum/go-ethereum/core/types"
|
||||||
|
"github.com/ethereum/go-ethereum/ethclient"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
// Polygon PoS 主网 USDT / USDC(原生)/ USDC.e(桥接)
|
||||||
|
polygonUsdtContract = common.HexToAddress("0xc2132D05D31c914a87C6611C10748AEb04B58e8F")
|
||||||
|
polygonUsdcContract = common.HexToAddress("0x3c499c542cEF5E3811e1192ce70d8cC03d5c3359")
|
||||||
|
polygonUsdcEContract = common.HexToAddress("0x2791Bca1f2de4661ED88A30C99A7a9449Aa84174")
|
||||||
|
)
|
||||||
|
|
||||||
|
type polygonRecipientSnapshot struct {
|
||||||
|
addrs map[string]struct{}
|
||||||
|
}
|
||||||
|
|
||||||
|
var polygonWatchedRecipients atomic.Pointer[polygonRecipientSnapshot]
|
||||||
|
|
||||||
|
func StartPolygonWebSocketListener() {
|
||||||
|
wallets, err := data.GetAvailableWalletAddressByNetwork(mdb.NetworkPolygon)
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Errorf("[POLYGON-WS] Failed to get wallet addresses: %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
storePolygonRecipientsFromWallets(wallets)
|
||||||
|
go func() {
|
||||||
|
ticker := time.NewTicker(5 * time.Second)
|
||||||
|
defer ticker.Stop()
|
||||||
|
for range ticker.C {
|
||||||
|
w, err := data.GetAvailableWalletAddressByNetwork(mdb.NetworkPolygon)
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Warnf("[POLYGON-WS] refresh wallet addresses: %v", err)
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
storePolygonRecipientsFromWallets(w)
|
||||||
|
}
|
||||||
|
}()
|
||||||
|
wsURL := "wss://polygon-bor-rpc.publicnode.com"
|
||||||
|
query := ethereum.FilterQuery{
|
||||||
|
Addresses: []common.Address{
|
||||||
|
polygonUsdtContract,
|
||||||
|
polygonUsdcContract,
|
||||||
|
polygonUsdcEContract,
|
||||||
|
},
|
||||||
|
Topics: [][]common.Hash{},
|
||||||
|
}
|
||||||
|
|
||||||
|
runWsLogListener("[POLYGON-WS]", wsURL, query, func(client *ethclient.Client, vLog types.Log) {
|
||||||
|
if len(vLog.Topics) < 3 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
event := vLog.Topics[0].String()
|
||||||
|
if event != transferEventHash.String() {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
amount := new(big.Int).SetBytes(vLog.Data)
|
||||||
|
|
||||||
|
toAddr := common.HexToAddress(vLog.Topics[2].Hex())
|
||||||
|
|
||||||
|
if !isWatchedPolygonRecipient(toAddr) {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
var blockTsMs int64
|
||||||
|
header, err := client.HeaderByNumber(context.Background(), big.NewInt(int64(vLog.BlockNumber)))
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Warnf("[POLYGON-WS] HeaderByNumber block=%d: %v, using local time", vLog.BlockNumber, err)
|
||||||
|
blockTsMs = time.Now().UnixMilli()
|
||||||
|
} else {
|
||||||
|
blockTsMs = int64(header.Time) * 1000
|
||||||
|
}
|
||||||
|
|
||||||
|
service.TryProcessEvmERC20Transfer(mdb.NetworkPolygon, vLog.Address, toAddr, amount, vLog.TxHash.Hex(), blockTsMs)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func storePolygonRecipientsFromWallets(wallets []mdb.WalletAddress) int {
|
||||||
|
m := make(map[string]struct{})
|
||||||
|
for _, w := range wallets {
|
||||||
|
a := strings.TrimSpace(w.Address)
|
||||||
|
if !common.IsHexAddress(a) {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
m[strings.ToLower(common.HexToAddress(a).Hex())] = struct{}{}
|
||||||
|
}
|
||||||
|
polygonWatchedRecipients.Store(&polygonRecipientSnapshot{addrs: m})
|
||||||
|
return len(m)
|
||||||
|
}
|
||||||
|
|
||||||
|
func isWatchedPolygonRecipient(to common.Address) bool {
|
||||||
|
snap := polygonWatchedRecipients.Load()
|
||||||
|
if snap == nil || len(snap.addrs) == 0 {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
_, ok := snap.addrs[strings.ToLower(to.Hex())]
|
||||||
|
return ok
|
||||||
|
}
|
||||||
@@ -0,0 +1,38 @@
|
|||||||
|
package task
|
||||||
|
|
||||||
|
import (
|
||||||
|
"sync"
|
||||||
|
|
||||||
|
"github.com/assimon/luuu/model/data"
|
||||||
|
"github.com/assimon/luuu/model/mdb"
|
||||||
|
"github.com/assimon/luuu/model/service"
|
||||||
|
"github.com/assimon/luuu/util/log"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ListenSolJob struct{}
|
||||||
|
|
||||||
|
var gListenSolJobLock sync.Mutex
|
||||||
|
|
||||||
|
func (r ListenSolJob) Run() {
|
||||||
|
gListenSolJobLock.Lock()
|
||||||
|
defer gListenSolJobLock.Unlock()
|
||||||
|
log.Sugar.Debug("[ListenSolJob] Job triggered")
|
||||||
|
walletAddress, err := data.GetAvailableWalletAddressByNetwork(mdb.NetworkSolana)
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Errorf("[ListenSolJob] Failed to get wallet addresses: %v", err)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if len(walletAddress) <= 0 {
|
||||||
|
log.Sugar.Debug("[ListenSolJob] No available wallet addresses")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
log.Sugar.Infof("[ListenSolJob] Found %d wallet addresses to monitor", len(walletAddress))
|
||||||
|
var wg sync.WaitGroup
|
||||||
|
for _, address := range walletAddress {
|
||||||
|
log.Sugar.Infof("[ListenSolJob] Listening to address: %s", address.Address)
|
||||||
|
wg.Add(1)
|
||||||
|
go service.SolCallBack(address.Address, &wg)
|
||||||
|
}
|
||||||
|
wg.Wait()
|
||||||
|
log.Sugar.Debug("[ListenSolJob] Job completed")
|
||||||
|
}
|
||||||
@@ -1,39 +1,116 @@
|
|||||||
package task
|
package task
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"sync"
|
"context"
|
||||||
|
"math/big"
|
||||||
|
"strings"
|
||||||
|
"sync/atomic"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
tron "github.com/assimon/luuu/crypto"
|
||||||
"github.com/assimon/luuu/model/data"
|
"github.com/assimon/luuu/model/data"
|
||||||
|
"github.com/assimon/luuu/model/mdb"
|
||||||
"github.com/assimon/luuu/model/service"
|
"github.com/assimon/luuu/model/service"
|
||||||
"github.com/assimon/luuu/util/log"
|
"github.com/assimon/luuu/util/log"
|
||||||
|
"github.com/ethereum/go-ethereum"
|
||||||
|
"github.com/ethereum/go-ethereum/common"
|
||||||
|
"github.com/ethereum/go-ethereum/core/types"
|
||||||
|
"github.com/ethereum/go-ethereum/ethclient"
|
||||||
)
|
)
|
||||||
|
|
||||||
type ListenTrc20Job struct {
|
const tronUsdtContractBase58 = "TR7NHqjeKQxGTCi8q8ZY4pL8otSzgjLj6t"
|
||||||
|
|
||||||
|
// TronGrid 的 JSON-RPC 使用 20-byte hex 地址(去掉 Tron 主网 0x41 前缀)。
|
||||||
|
var tronUsdtTrc20Contract = common.HexToAddress("0xa614f803b6fd780986a42c78ec9c7f77e6ded13c")
|
||||||
|
|
||||||
|
type tronRecipientSnapshot struct {
|
||||||
|
addrs map[string]struct{}
|
||||||
}
|
}
|
||||||
|
|
||||||
var gListenTrc20JobLock sync.Mutex
|
var tronWatchedRecipients atomic.Pointer[tronRecipientSnapshot]
|
||||||
|
|
||||||
func (r ListenTrc20Job) Run() {
|
func StartTronTrc20WebSocketListener() {
|
||||||
gListenTrc20JobLock.Lock()
|
log.Sugar.Infof("[TRON-WS] listening contract base58=%s hex=%s", tronUsdtContractBase58, tronUsdtTrc20Contract.Hex())
|
||||||
defer gListenTrc20JobLock.Unlock()
|
|
||||||
log.Sugar.Debug("[ListenTrc20Job] Job triggered")
|
wallets, err := data.GetAvailableWalletAddressByNetwork(mdb.NetworkTron)
|
||||||
walletAddress, err := data.GetAvailableWalletAddress()
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Sugar.Errorf("[ListenTrc20Job] Failed to get wallet addresses: %v", err)
|
log.Sugar.Errorf("[TRON-WS] Failed to get wallet addresses: %v", err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
if len(walletAddress) <= 0 {
|
storeTronRecipientsFromWallets(wallets)
|
||||||
log.Sugar.Debug("[ListenTrc20Job] No available wallet addresses")
|
go func() {
|
||||||
return
|
ticker := time.NewTicker(5 * time.Second)
|
||||||
|
defer ticker.Stop()
|
||||||
|
for range ticker.C {
|
||||||
|
w, err := data.GetAvailableWalletAddressByNetwork(mdb.NetworkTron)
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Warnf("[TRON-WS] refresh wallet addresses: %v", err)
|
||||||
|
continue
|
||||||
}
|
}
|
||||||
log.Sugar.Infof("[ListenTrc20Job] Found %d wallet addresses to monitor", len(walletAddress))
|
storeTronRecipientsFromWallets(w)
|
||||||
var wg sync.WaitGroup
|
}
|
||||||
for _, address := range walletAddress {
|
}()
|
||||||
log.Sugar.Infof("[ListenTrc20Job] Listening to address: %s", address.Address)
|
|
||||||
|
|
||||||
wg.Add(1)
|
wsURL := "wss://api.trongrid.io/jsonrpc"
|
||||||
go service.Trc20CallBack(address.Address, &wg)
|
query := ethereum.FilterQuery{
|
||||||
|
Addresses: []common.Address{tronUsdtTrc20Contract},
|
||||||
|
Topics: [][]common.Hash{},
|
||||||
}
|
}
|
||||||
wg.Wait()
|
runWsLogListener("[TRON-WS]", wsURL, query, func(client *ethclient.Client, vLog types.Log) {
|
||||||
log.Sugar.Debug("[ListenTrc20Job] Job completed")
|
if len(vLog.Topics) < 3 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if vLog.Topics[0].String() != transferEventHash.String() {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
toAddr := tronHexToBase58(common.HexToAddress(vLog.Topics[2].Hex()))
|
||||||
|
//if !isWatchedTronRecipient(toAddr) {
|
||||||
|
// return
|
||||||
|
//}
|
||||||
|
|
||||||
|
amount := new(big.Int).SetBytes(vLog.Data)
|
||||||
|
if amount.Sign() <= 0 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Sugar.Infof("[TRON-WS] Detected TRC20 transfer to %s amount=%s tx=%s", toAddr, amount.String(), vLog.TxHash.Hex())
|
||||||
|
var blockTsMs int64
|
||||||
|
header, err := client.HeaderByNumber(context.Background(), big.NewInt(int64(vLog.BlockNumber)))
|
||||||
|
if err != nil {
|
||||||
|
log.Sugar.Warnf("[TRON-WS] HeaderByNumber block=%d: %v, using local time", vLog.BlockNumber, err)
|
||||||
|
blockTsMs = time.Now().UnixMilli()
|
||||||
|
} else {
|
||||||
|
blockTsMs = int64(header.Time) * 1000
|
||||||
|
}
|
||||||
|
|
||||||
|
service.TryProcessTronTRC20Transfer(toAddr, amount, vLog.TxHash.Hex(), blockTsMs)
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
func storeTronRecipientsFromWallets(wallets []mdb.WalletAddress) int {
|
||||||
|
m := make(map[string]struct{})
|
||||||
|
for _, w := range wallets {
|
||||||
|
addr := strings.TrimSpace(w.Address)
|
||||||
|
if addr == "" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
m[addr] = struct{}{}
|
||||||
|
}
|
||||||
|
tronWatchedRecipients.Store(&tronRecipientSnapshot{addrs: m})
|
||||||
|
return len(m)
|
||||||
|
}
|
||||||
|
|
||||||
|
func isWatchedTronRecipient(to string) bool {
|
||||||
|
snap := tronWatchedRecipients.Load()
|
||||||
|
if snap == nil || len(snap.addrs) == 0 {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
_, ok := snap.addrs[strings.TrimSpace(to)]
|
||||||
|
return ok
|
||||||
|
}
|
||||||
|
|
||||||
|
func tronHexToBase58(addr common.Address) string {
|
||||||
|
raw := append([]byte{tron.PrefixMainnet}, addr.Bytes()...)
|
||||||
|
return tron.EncodeCheck(raw)
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -2,6 +2,8 @@ package telegram
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"sort"
|
||||||
|
"strings"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
@@ -13,12 +15,14 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
const (
|
const (
|
||||||
ReplayAddWallet = "请发给我一个合法的钱包地址"
|
ReplaySelectNetwork = "请选择要添加的钱包网络"
|
||||||
|
ReplayAddWallet = "请发送 %s 网络收款地址"
|
||||||
pendingWalletAddressTTL = 5 * time.Minute
|
pendingWalletAddressTTL = 5 * time.Minute
|
||||||
)
|
)
|
||||||
|
|
||||||
type pendingWalletAddressState struct {
|
type pendingWalletAddressState struct {
|
||||||
RequestedAt time.Time
|
RequestedAt time.Time
|
||||||
|
Network string
|
||||||
}
|
}
|
||||||
|
|
||||||
var pendingWalletAddressUsers sync.Map
|
var pendingWalletAddressUsers sync.Map
|
||||||
@@ -35,7 +39,7 @@ func OnTextMessageHandle(c tb.Context) error {
|
|||||||
senderID = sender.ID
|
senderID = sender.ID
|
||||||
}
|
}
|
||||||
|
|
||||||
isReplyFlow := msg.ReplyTo != nil && msg.ReplyTo.Text == ReplayAddWallet
|
isReplyFlow := msg.ReplyTo != nil && strings.HasPrefix(msg.ReplyTo.Text, "请发送 ")
|
||||||
isPendingFlow := isWalletAddressPending(senderID)
|
isPendingFlow := isWalletAddressPending(senderID)
|
||||||
if !isReplyFlow && !isPendingFlow {
|
if !isReplyFlow && !isPendingFlow {
|
||||||
return nil
|
return nil
|
||||||
@@ -45,19 +49,26 @@ func OnTextMessageHandle(c tb.Context) error {
|
|||||||
defer bots.Delete(msg.ReplyTo)
|
defer bots.Delete(msg.ReplyTo)
|
||||||
}
|
}
|
||||||
|
|
||||||
msgText := msg.Text
|
msgText := strings.TrimSpace(msg.Text)
|
||||||
if !isValidTronAddress(msgText) {
|
state, _ := getPendingWalletAddressState(senderID)
|
||||||
_ = c.Send(fmt.Sprintf("钱包 [%s] 添加失败:不是合法的 TRON 地址", msgText))
|
if state.Network == "" {
|
||||||
|
_ = c.Send("请先选择网络,再发送地址。")
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
_, err := data.AddWalletAddress(msgText)
|
var err error
|
||||||
|
if !isValidAddressByNetwork(state.Network, msgText) {
|
||||||
|
_ = c.Send(fmt.Sprintf("钱包 [%s] 添加失败:不是合法的 %s 地址", msgText, strings.ToUpper(state.Network)))
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
storeAddress := normalizeWalletAddressByNetwork(state.Network, msgText)
|
||||||
|
_, err = data.AddWalletAddressWithNetwork(state.Network, storeAddress)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return c.Send(err.Error())
|
return c.Send(err.Error())
|
||||||
}
|
}
|
||||||
pendingWalletAddressUsers.Delete(senderID)
|
pendingWalletAddressUsers.Delete(senderID)
|
||||||
|
|
||||||
_ = c.Send(fmt.Sprintf("钱包 [%s] 添加成功", msgText))
|
_ = c.Send(fmt.Sprintf("钱包 [%s] 添加成功(%s)", storeAddress, strings.ToUpper(state.Network)))
|
||||||
return WalletList(c)
|
return WalletList(c)
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -73,10 +84,14 @@ func WalletList(c tb.Context) error {
|
|||||||
if wallet.Status == mdb.TokenStatusDisable {
|
if wallet.Status == mdb.TokenStatusDisable {
|
||||||
status = "已禁用🚫"
|
status = "已禁用🚫"
|
||||||
}
|
}
|
||||||
|
net := wallet.Network
|
||||||
|
if net == "" {
|
||||||
|
net = mdb.NetworkTron
|
||||||
|
}
|
||||||
|
|
||||||
btnInfo := tb.InlineButton{
|
btnInfo := tb.InlineButton{
|
||||||
Unique: wallet.Address,
|
Unique: wallet.Address,
|
||||||
Text: fmt.Sprintf("%s [%s]", wallet.Address, status),
|
Text: fmt.Sprintf("[%s] %s [%s]", net, wallet.Address, status),
|
||||||
Data: strutil.MustString(wallet.ID),
|
Data: strutil.MustString(wallet.ID),
|
||||||
}
|
}
|
||||||
bots.Handle(&btnInfo, WalletInfo)
|
bots.Handle(&btnInfo, WalletInfo)
|
||||||
@@ -85,11 +100,26 @@ func WalletList(c tb.Context) error {
|
|||||||
|
|
||||||
addBtn := tb.InlineButton{Text: "添加钱包地址", Unique: "AddWallet"}
|
addBtn := tb.InlineButton{Text: "添加钱包地址", Unique: "AddWallet"}
|
||||||
bots.Handle(&addBtn, func(c tb.Context) error {
|
bots.Handle(&addBtn, func(c tb.Context) error {
|
||||||
if sender := c.Sender(); sender != nil {
|
chains, err := getEnabledSupportedNetworks()
|
||||||
pendingWalletAddressUsers.Store(sender.ID, pendingWalletAddressState{RequestedAt: time.Now()})
|
if err != nil {
|
||||||
|
return c.Send("读取支持链失败:" + err.Error())
|
||||||
}
|
}
|
||||||
return c.Send(ReplayAddWallet, &tb.ReplyMarkup{
|
if len(chains) == 0 {
|
||||||
ForceReply: true,
|
return c.Send("当前没有可用链,请先在后台配置 supported-assets。")
|
||||||
|
}
|
||||||
|
|
||||||
|
rows := make([][]tb.InlineButton, 0, len(chains))
|
||||||
|
for _, network := range chains {
|
||||||
|
btn := tb.InlineButton{
|
||||||
|
Text: strings.ToUpper(network),
|
||||||
|
Unique: "SelectWalletNetwork",
|
||||||
|
Data: network,
|
||||||
|
}
|
||||||
|
bots.Handle(&btn, SelectWalletNetwork)
|
||||||
|
rows = append(rows, []tb.InlineButton{btn})
|
||||||
|
}
|
||||||
|
return c.EditOrSend(ReplaySelectNetwork, &tb.ReplyMarkup{
|
||||||
|
InlineKeyboard: rows,
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
btnList = append(btnList, []tb.InlineButton{addBtn})
|
btnList = append(btnList, []tb.InlineButton{addBtn})
|
||||||
@@ -99,6 +129,22 @@ func WalletList(c tb.Context) error {
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func SelectWalletNetwork(c tb.Context) error {
|
||||||
|
network := strings.ToLower(strings.TrimSpace(c.Data()))
|
||||||
|
if network == "" {
|
||||||
|
return c.Send("请选择有效网络")
|
||||||
|
}
|
||||||
|
if sender := c.Sender(); sender != nil {
|
||||||
|
pendingWalletAddressUsers.Store(sender.ID, pendingWalletAddressState{
|
||||||
|
RequestedAt: time.Now(),
|
||||||
|
Network: network,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
return c.Send(fmt.Sprintf(ReplayAddWallet, strings.ToUpper(network)), &tb.ReplyMarkup{
|
||||||
|
ForceReply: true,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
func WalletInfo(c tb.Context) error {
|
func WalletInfo(c tb.Context) error {
|
||||||
id := mathutil.MustUint(c.Data())
|
id := mathutil.MustUint(c.Data())
|
||||||
tokenInfo, err := data.GetWalletAddressById(id)
|
tokenInfo, err := data.GetWalletAddressById(id)
|
||||||
@@ -131,7 +177,12 @@ func WalletInfo(c tb.Context) error {
|
|||||||
bots.Handle(&delBtn, DelWallet)
|
bots.Handle(&delBtn, DelWallet)
|
||||||
bots.Handle(&backBtn, WalletList)
|
bots.Handle(&backBtn, WalletList)
|
||||||
|
|
||||||
return c.EditOrReply(tokenInfo.Address, &tb.ReplyMarkup{InlineKeyboard: [][]tb.InlineButton{
|
net := tokenInfo.Network
|
||||||
|
if net == "" {
|
||||||
|
net = mdb.NetworkTron
|
||||||
|
}
|
||||||
|
detail := fmt.Sprintf("网络:%s\n地址:%s", net, tokenInfo.Address)
|
||||||
|
return c.EditOrReply(detail, &tb.ReplyMarkup{InlineKeyboard: [][]tb.InlineButton{
|
||||||
{
|
{
|
||||||
enableBtn,
|
enableBtn,
|
||||||
disableBtn,
|
disableBtn,
|
||||||
@@ -195,3 +246,33 @@ func isWalletAddressPending(userID int64) bool {
|
|||||||
}
|
}
|
||||||
return true
|
return true
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func getPendingWalletAddressState(userID int64) (pendingWalletAddressState, bool) {
|
||||||
|
v, ok := pendingWalletAddressUsers.Load(userID)
|
||||||
|
if !ok {
|
||||||
|
return pendingWalletAddressState{}, false
|
||||||
|
}
|
||||||
|
state, ok := v.(pendingWalletAddressState)
|
||||||
|
return state, ok
|
||||||
|
}
|
||||||
|
|
||||||
|
func getEnabledSupportedNetworks() ([]string, error) {
|
||||||
|
assets, err := data.ListEnabledSupportedAssets()
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
set := make(map[string]struct{})
|
||||||
|
for _, a := range assets {
|
||||||
|
n := strings.ToLower(strings.TrimSpace(a.Network))
|
||||||
|
if n == "" {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
set[n] = struct{}{}
|
||||||
|
}
|
||||||
|
networks := make([]string, 0, len(set))
|
||||||
|
for n := range set {
|
||||||
|
networks = append(networks, n)
|
||||||
|
}
|
||||||
|
sort.Strings(networks)
|
||||||
|
return networks, nil
|
||||||
|
}
|
||||||
|
|||||||
@@ -2,10 +2,24 @@ package telegram
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"crypto/sha256"
|
"crypto/sha256"
|
||||||
|
"encoding/hex"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/assimon/luuu/model/mdb"
|
||||||
"github.com/btcsuite/btcutil/base58"
|
"github.com/btcsuite/btcutil/base58"
|
||||||
|
"github.com/gagliardetto/solana-go"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// isValidEthereumAddress 校验 0x + 20 字节十六进制(主网收款)。
|
||||||
|
func isValidEthereumAddress(addr string) bool {
|
||||||
|
addr = strings.TrimSpace(addr)
|
||||||
|
if len(addr) != 42 || !strings.HasPrefix(addr, "0x") {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
_, err := hex.DecodeString(addr[2:])
|
||||||
|
return err == nil
|
||||||
|
}
|
||||||
|
|
||||||
// isValidTronAddress 校验 Tron Base58Check 地址是否合法
|
// isValidTronAddress 校验 Tron Base58Check 地址是否合法
|
||||||
func isValidTronAddress(addr string) bool {
|
func isValidTronAddress(addr string) bool {
|
||||||
// 基本过滤
|
// 基本过滤
|
||||||
@@ -32,3 +46,35 @@ func isValidTronAddress(addr string) bool {
|
|||||||
|
|
||||||
return string(checksum) == string(hash2[:4])
|
return string(checksum) == string(hash2[:4])
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func isValidAddressByNetwork(network, addr string) bool {
|
||||||
|
switch strings.ToLower(strings.TrimSpace(network)) {
|
||||||
|
case mdb.NetworkTron:
|
||||||
|
return isValidTronAddress(addr)
|
||||||
|
case mdb.NetworkSolana:
|
||||||
|
return isValidSolanaAddress(addr)
|
||||||
|
default:
|
||||||
|
// 其余 EVM 链统一使用 0x 地址校验
|
||||||
|
return isValidEthereumAddress(addr)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func normalizeWalletAddressByNetwork(network, addr string) string {
|
||||||
|
addr = strings.TrimSpace(addr)
|
||||||
|
switch strings.ToLower(strings.TrimSpace(network)) {
|
||||||
|
case mdb.NetworkTron, mdb.NetworkSolana:
|
||||||
|
return addr
|
||||||
|
default:
|
||||||
|
return strings.ToLower(addr)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// isValidSolanaAddress 校验 Solana Base58 地址是否合法(32 字节公钥)。
|
||||||
|
func isValidSolanaAddress(addr string) bool {
|
||||||
|
addr = strings.TrimSpace(addr)
|
||||||
|
if len(addr) < 32 || len(addr) > 44 {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
_, err := solana.PublicKeyFromBase58(addr)
|
||||||
|
return err == nil
|
||||||
|
}
|
||||||
|
|||||||
@@ -13,6 +13,11 @@ var Errno = map[int]string{
|
|||||||
10008: "order does not exist",
|
10008: "order does not exist",
|
||||||
10009: "failed to parse request params",
|
10009: "failed to parse request params",
|
||||||
10010: "order status already changed",
|
10010: "order status already changed",
|
||||||
|
10011: "exceeded maximum sub-order limit",
|
||||||
|
10012: "cannot switch network on a sub-order",
|
||||||
|
10013: "order is not awaiting payment",
|
||||||
|
10014: "supported asset already exists",
|
||||||
|
10015: "supported asset not found",
|
||||||
}
|
}
|
||||||
|
|
||||||
var (
|
var (
|
||||||
@@ -28,6 +33,11 @@ var (
|
|||||||
OrderNotExists = Err(10008)
|
OrderNotExists = Err(10008)
|
||||||
ParamsMarshalErr = Err(10009)
|
ParamsMarshalErr = Err(10009)
|
||||||
OrderStatusConflict = Err(10010)
|
OrderStatusConflict = Err(10010)
|
||||||
|
SubOrderLimitExceeded = Err(10011)
|
||||||
|
CannotSwitchSubOrder = Err(10012)
|
||||||
|
OrderNotWaitPay = Err(10013)
|
||||||
|
SupportedAssetAlreadyExists = Err(10014)
|
||||||
|
SupportedAssetNotFound = Err(10015)
|
||||||
)
|
)
|
||||||
|
|
||||||
type RspError struct {
|
type RspError struct {
|
||||||
|
|||||||
@@ -0,0 +1,103 @@
|
|||||||
|
# API 变更文档
|
||||||
|
|
||||||
|
## 概述
|
||||||
|
|
||||||
|
## 后台配置说明(重要)
|
||||||
|
|
||||||
|
### 对于 dujiaoka 用户
|
||||||
|
|
||||||
|
**只需要修改一个地方**:在 dujiaoka 后台支付插件配置中,将 API 地址前缀从 `/api` 改为 `/payments/epusdt` 即可。
|
||||||
|
|
||||||
|
**示例**:
|
||||||
|
|
||||||
|
```
|
||||||
|
旧配置:https://your-domain.com/api/v1/order/create-transaction
|
||||||
|
新配置:https://your-domain.com/payments/epusdt/v1/order/create-transaction
|
||||||
|
```
|
||||||
|
|
||||||
|
**就这么简单!** 其他配置(密钥、回调地址等)完全不需要修改。
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 路由变更对照表
|
||||||
|
|
||||||
|
### API 路由
|
||||||
|
|
||||||
|
| 原路由 | 新路由 | 说明 |
|
||||||
|
|--------|--------|------|
|
||||||
|
| `POST /api/v1/order/create-transaction` | `POST /payments/epusdt/v1/order/create-transaction` | 创建交易订单 |
|
||||||
|
|
||||||
|
|
||||||
|
## 新增配置项说明
|
||||||
|
|
||||||
|
本次更新新增了以下配置项(位于 `src/.env.example` 文件):
|
||||||
|
|
||||||
|
### 1. `api_rate_url` - 汇率接口 URL
|
||||||
|
|
||||||
|
用于获取实时汇率的 API 地址。系统会根据此接口动态获取不同币种的汇率。
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 汇率接口url
|
||||||
|
api_rate_url=https://your-rate-api.com/
|
||||||
|
```
|
||||||
|
|
||||||
|
**API 格式要求**:
|
||||||
|
|
||||||
|
系统会请求 `{api_rate_url}/{currency}.json`,例如:
|
||||||
|
- `https://your-rate-api.com/cny.json`
|
||||||
|
- `https://your-rate-api.com/usd.json`
|
||||||
|
|
||||||
|
**返回格式示例**:
|
||||||
|
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"cny": {
|
||||||
|
"usdt": 0.1389,
|
||||||
|
"trx": 0.0123
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
其中 `0.1389` 表示 1 CNY = 0.1389 USDT(即 1 USDT ≈ 7.2 CNY)
|
||||||
|
|
||||||
|
**说明**:
|
||||||
|
- 支持自建汇率 API,只需按照上述格式返回数据即可
|
||||||
|
|
||||||
|
### 2. `tron_grid_api_key` - TRON Grid API Key
|
||||||
|
|
||||||
|
TRON Grid API 密钥,用于提高 API 请求限制和稳定性。
|
||||||
|
|
||||||
|
```bash
|
||||||
|
tron_grid_api_key=
|
||||||
|
```
|
||||||
|
|
||||||
|
**如何获取 API Key**:
|
||||||
|
|
||||||
|
1. 访问 [https://www.trongrid.io/](https://www.trongrid.io/)
|
||||||
|
2. 注册账号并登录
|
||||||
|
3. 在控制台创建 API Key
|
||||||
|
4. 将 API Key 填入配置文件
|
||||||
|
|
||||||
|
**为什么需要 API Key**:
|
||||||
|
|
||||||
|
- ✅ **提高请求限制**:免费账号有更高的 API 调用配额
|
||||||
|
- ✅ **更好的稳定性**:避免公共 API 的限流问题
|
||||||
|
- ✅ **支持更多功能**:为后续支持 TRX 等其他代币做准备
|
||||||
|
|
||||||
|
### 配置示例
|
||||||
|
|
||||||
|
`.env` 配置示例:
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 订单过期时间(分钟)
|
||||||
|
order_expiration_time=15
|
||||||
|
|
||||||
|
# 订单回调失败最大重试次数
|
||||||
|
order_notice_max_retry=0
|
||||||
|
|
||||||
|
# 汇率接口url(动态获取汇率)
|
||||||
|
api_rate_url=
|
||||||
|
|
||||||
|
# TRON Grid API Key(推荐配置)
|
||||||
|
tron_grid_api_key=your-api-key-here
|
||||||
|
```
|
||||||
@@ -0,0 +1,104 @@
|
|||||||
|
1. 创建和进去这个目录
|
||||||
|
```shell
|
||||||
|
mkdir epusdt && cd epusdt
|
||||||
|
```
|
||||||
|
|
||||||
|
2. 把配置文件放进去这个目录,只需要改
|
||||||
|
|
||||||
|
app_url
|
||||||
|
|
||||||
|
tron_grid_api_key
|
||||||
|
|
||||||
|
api_rate_url
|
||||||
|
|
||||||
|
```shell
|
||||||
|
cat <<EOF > env
|
||||||
|
app_name=epusdt
|
||||||
|
app_uri=https://dujiaoka.com
|
||||||
|
log_level=info
|
||||||
|
http_access_log=false
|
||||||
|
sql_debug=false
|
||||||
|
http_listen=:8000
|
||||||
|
|
||||||
|
static_path=/static
|
||||||
|
runtime_root_path=/runtime
|
||||||
|
|
||||||
|
log_save_path=/logs
|
||||||
|
log_max_size=32
|
||||||
|
log_max_age=7
|
||||||
|
max_backups=3
|
||||||
|
|
||||||
|
# supported values: postgres,mysql,sqlite
|
||||||
|
db_type=sqlite
|
||||||
|
|
||||||
|
# sqlite primary database config
|
||||||
|
sqlite_database_filename=
|
||||||
|
sqlite_table_prefix=
|
||||||
|
|
||||||
|
# postgres config
|
||||||
|
postgres_host=127.0.0.1
|
||||||
|
postgres_port=3306
|
||||||
|
postgres_user=mysql_user
|
||||||
|
postgres_passwd=mysql_password
|
||||||
|
postgres_database=database_name
|
||||||
|
postgres_table_prefix=
|
||||||
|
postgres_max_idle_conns=10
|
||||||
|
postgres_max_open_conns=100
|
||||||
|
postgres_max_life_time=6
|
||||||
|
|
||||||
|
# mysql config
|
||||||
|
mysql_host=127.0.0.1
|
||||||
|
mysql_port=3306
|
||||||
|
mysql_user=mysql_user
|
||||||
|
mysql_passwd=mysql_password
|
||||||
|
mysql_database=database_name
|
||||||
|
mysql_table_prefix=
|
||||||
|
mysql_max_idle_conns=10
|
||||||
|
mysql_max_open_conns=100
|
||||||
|
mysql_max_life_time=6
|
||||||
|
|
||||||
|
# sqlite runtime store config
|
||||||
|
runtime_sqlite_filename=epusdt-runtime.db
|
||||||
|
|
||||||
|
# background scheduler config
|
||||||
|
queue_concurrency=10
|
||||||
|
queue_poll_interval_ms=1000
|
||||||
|
callback_retry_base_seconds=5
|
||||||
|
|
||||||
|
tg_bot_token=
|
||||||
|
tg_proxy=
|
||||||
|
tg_manage=
|
||||||
|
|
||||||
|
api_auth_token=
|
||||||
|
|
||||||
|
order_expiration_time=10
|
||||||
|
order_notice_max_retry=0
|
||||||
|
forced_usdt_rate=
|
||||||
|
api_rate_url=https://cdn.jsdelivr.net/npm/@fawazahmed0/currency-api@latest/v1/currencies/
|
||||||
|
tron_grid_api_key=
|
||||||
|
EOF
|
||||||
|
```
|
||||||
|
3. docker compose 创建
|
||||||
|
```shell
|
||||||
|
cat <<EOF > docker-compose.yaml
|
||||||
|
services:
|
||||||
|
epusdt:
|
||||||
|
image: gmwallet/epusdt:alpine
|
||||||
|
restart: always
|
||||||
|
build:
|
||||||
|
context: .
|
||||||
|
dockerfile: Dockerfile
|
||||||
|
volumes:
|
||||||
|
- ./env:/app/.env
|
||||||
|
ports:
|
||||||
|
- "8000:8000"
|
||||||
|
EOF
|
||||||
|
```
|
||||||
|
4. 运行
|
||||||
|
```shell
|
||||||
|
docker compose up -d
|
||||||
|
```
|
||||||
|
5. 配置独角兽后台
|
||||||
|
|
||||||
|
商户密钥: http://your_domain/payments/epusdt/v1/order/create-transaction
|
||||||
|
|
||||||