Initial commit: FunStat MCP Server Go implementation
- Telegram integration for customer statistics - MCP server implementation with rate limiting - Cache system for performance optimization - Multi-language support - RESTful API endpoints 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
你的用户名
38
.gitignore
vendored
Normal file
38
.gitignore
vendored
Normal file
@@ -0,0 +1,38 @@
|
||||
# Binaries for programs and plugins
|
||||
*.exe
|
||||
*.exe~
|
||||
*.dll
|
||||
*.so
|
||||
*.dylib
|
||||
|
||||
# Test binary, built with `go test -c`
|
||||
*.test
|
||||
|
||||
# Output of the go coverage tool
|
||||
*.out
|
||||
|
||||
# Dependency directories
|
||||
vendor/
|
||||
|
||||
# Go workspace file
|
||||
go.work
|
||||
|
||||
# Environment variables
|
||||
.env
|
||||
.env.local
|
||||
|
||||
# IDE
|
||||
.vscode/
|
||||
.idea/
|
||||
*.swp
|
||||
*.swo
|
||||
*~
|
||||
|
||||
# OS
|
||||
.DS_Store
|
||||
Thumbs.db
|
||||
|
||||
# Build output
|
||||
/bin/
|
||||
/dist/
|
||||
/build/
|
||||
108
README.md
Normal file
108
README.md
Normal file
@@ -0,0 +1,108 @@
|
||||
# Funstat MCP (Go)
|
||||
|
||||
This project is a complete Go reimplementation of the original Funstat MCP
|
||||
server. It exposes the same MCP tools wrapped around the Telegram
|
||||
`@openaiw_bot`, but with a native Go runtime and HTTP transport.
|
||||
|
||||
## Features
|
||||
|
||||
- Streamable HTTP transport (`/sse` + `/messages`) compatible with MCP clients
|
||||
such as Codex CLI, Cursor, and Claude Code.
|
||||
- Full parity with the original nine Funstat tools (search, topchat, text,
|
||||
human lookup, user info, user messages, balance, menu, start).
|
||||
- Built-in rate limiting and response caching to stay within Telegram limits.
|
||||
- Telegram session bootstrapping from a Telethon string session or persisted Go
|
||||
session storage.
|
||||
- Configurable proxy, cache TTL, and rate limit settings via environment
|
||||
variables.
|
||||
|
||||
## Requirements
|
||||
|
||||
- Go 1.22 or newer (the module was developed with Go 1.24).
|
||||
- A valid Telegram API ID and API hash.
|
||||
- A logged-in Telegram session. The Go server expects either:
|
||||
- `TELEGRAM_SESSION_STRING` (or `TELEGRAM_SESSION_STRING_FILE`) containing a
|
||||
Telethon *StringSession*, which will be converted automatically into Go
|
||||
session storage, **or**
|
||||
- `TELEGRAM_SESSION_PATH` pointing to a Go session file created by this
|
||||
service on a previous run (`~/.funstatmcp/session.json` by default).
|
||||
- Network access to Telegram (configure proxies via the `FUNSTAT_PROXY_*`
|
||||
variables if necessary).
|
||||
|
||||
## Environment Variables
|
||||
|
||||
| Variable | Description |
|
||||
|----------|-------------|
|
||||
| `TELEGRAM_API_ID` | **Required.** Telegram API ID. |
|
||||
| `TELEGRAM_API_HASH` | **Required.** Telegram API hash. |
|
||||
| `TELEGRAM_SESSION_STRING` | Base64 Telethon StringSession (optional if session file already exists). |
|
||||
| `TELEGRAM_SESSION_STRING_FILE` | Path to a file containing the StringSession. |
|
||||
| `TELEGRAM_SESSION_PATH` | Optional path for Go session storage (`.session` suffix is appended if missing). |
|
||||
| `FUNSTAT_BOT_USERNAME` | Bot username (default `@openaiw_bot`). |
|
||||
| `FUNSTAT_RATE_LIMIT_PER_SECOND` | Requests per second (default `18`). |
|
||||
| `FUNSTAT_RATE_LIMIT_WINDOW` | Duration window, e.g. `1s` (default `1s`). |
|
||||
| `FUNSTAT_CACHE_TTL` | Cache lifetime for command responses (default `1h`). |
|
||||
| `FUNSTAT_PROXY_TYPE` | Proxy type (`socks5` supported). |
|
||||
| `FUNSTAT_PROXY_HOST` | Proxy host. |
|
||||
| `FUNSTAT_PROXY_PORT` | Proxy port. |
|
||||
| `FUNSTAT_PROXY_USERNAME` / `FUNSTAT_PROXY_PASSWORD` | Optional proxy credentials. |
|
||||
| `FUNSTAT_HOST` | Bind host (default `127.0.0.1`). |
|
||||
| `FUNSTAT_PORT` | Bind port (default `8091`). |
|
||||
| `FUNSTAT_REQUIRE_SESSION` | When `true`, enables strict session ID enforcement (not yet required by the Go transport). |
|
||||
|
||||
## Running the Server
|
||||
|
||||
```bash
|
||||
export TELEGRAM_API_ID=...
|
||||
export TELEGRAM_API_HASH=...
|
||||
export TELEGRAM_SESSION_STRING=... # or TELEGRAM_SESSION_STRING_FILE
|
||||
|
||||
go run ./cmd/funstat-mcp
|
||||
```
|
||||
|
||||
The server starts on `http://127.0.0.1:8091` by default with:
|
||||
|
||||
- `GET /sse` — server-sent events stream for MCP clients.
|
||||
- `POST /messages` — JSON-RPC 2.0 endpoint for MCP requests.
|
||||
- `GET /health` — simple health probe.
|
||||
|
||||
For convenience a helper script is available:
|
||||
|
||||
```bash
|
||||
./scripts/start_server.sh
|
||||
```
|
||||
|
||||
## Tool Coverage
|
||||
|
||||
The server exposes the following MCP tools (identical to the Python version):
|
||||
|
||||
| Tool | Description |
|
||||
|------|-------------|
|
||||
| `funstat_search` | 搜索 Telegram 群组/频道。 |
|
||||
| `funstat_topchat` | 获取热门群组列表。 |
|
||||
| `funstat_text` | 按文本搜索消息。 |
|
||||
| `funstat_human` | 按姓名搜索用户。 |
|
||||
| `funstat_user_info` | 查询用户详情。 |
|
||||
| `funstat_user_messages` | 抓取用户聊天记录并自动翻页。 |
|
||||
| `funstat_balance` | 查询积分余额。 |
|
||||
| `funstat_menu` | 显示 Funstat 菜单。 |
|
||||
| `funstat_start` | Funstat 欢迎信息。 |
|
||||
|
||||
## Session Notes
|
||||
|
||||
- If you already possess the Telethon `.session` SQLite file, generate a
|
||||
StringSession first (e.g. via Telethon or `python -m telethon.sessions`)
|
||||
and provide it through `TELEGRAM_SESSION_STRING`.
|
||||
- After the first successful run the Go server stores an encrypted Go session
|
||||
at `~/.funstatmcp/session.json` (unless overridden). Subsequent runs only
|
||||
require the API ID/hash.
|
||||
|
||||
## Development
|
||||
|
||||
- Format code with `gofmt -w cmd internal`.
|
||||
- Run `go test ./...` (no automated tests are included yet).
|
||||
- Dependencies are managed via Go modules (`go mod tidy`).
|
||||
|
||||
## License
|
||||
|
||||
Follows the license of the original Funstat MCP project (MIT, if applicable).
|
||||
32
cmd/funstat-mcp/main.go
Normal file
32
cmd/funstat-mcp/main.go
Normal file
@@ -0,0 +1,32 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
"log"
|
||||
"os/signal"
|
||||
"syscall"
|
||||
|
||||
"funstatmcp/internal/app"
|
||||
"funstatmcp/internal/transport"
|
||||
)
|
||||
|
||||
func main() {
|
||||
cfg, err := app.FromEnv()
|
||||
if err != nil {
|
||||
log.Fatalf("load config: %v", err)
|
||||
}
|
||||
|
||||
application, err := app.New(cfg)
|
||||
if err != nil {
|
||||
log.Fatalf("initialize app: %v", err)
|
||||
}
|
||||
|
||||
ctx, cancel := signal.NotifyContext(context.Background(), syscall.SIGINT, syscall.SIGTERM)
|
||||
defer cancel()
|
||||
|
||||
server := transport.NewServer(application, cfg)
|
||||
|
||||
if err := server.Run(ctx); err != nil {
|
||||
log.Fatalf("server exited with error: %v", err)
|
||||
}
|
||||
}
|
||||
43
go.mod
Normal file
43
go.mod
Normal file
@@ -0,0 +1,43 @@
|
||||
module funstatmcp
|
||||
|
||||
go 1.24.3
|
||||
|
||||
require (
|
||||
github.com/gotd/td v0.132.0
|
||||
golang.org/x/net v0.46.0
|
||||
)
|
||||
|
||||
require (
|
||||
github.com/cenkalti/backoff/v4 v4.3.0 // indirect
|
||||
github.com/coder/websocket v1.8.14 // indirect
|
||||
github.com/dlclark/regexp2 v1.11.5 // indirect
|
||||
github.com/fatih/color v1.18.0 // indirect
|
||||
github.com/ghodss/yaml v1.0.0 // indirect
|
||||
github.com/go-faster/errors v0.7.1 // indirect
|
||||
github.com/go-faster/jx v1.1.0 // indirect
|
||||
github.com/go-faster/xor v1.0.0 // indirect
|
||||
github.com/go-faster/yaml v0.4.6 // indirect
|
||||
github.com/google/uuid v1.6.0 // indirect
|
||||
github.com/gotd/ige v0.2.2 // indirect
|
||||
github.com/gotd/neo v0.1.5 // indirect
|
||||
github.com/klauspost/compress v1.18.0 // indirect
|
||||
github.com/mattn/go-colorable v0.1.14 // indirect
|
||||
github.com/mattn/go-isatty v0.0.20 // indirect
|
||||
github.com/ogen-go/ogen v1.15.2 // indirect
|
||||
github.com/segmentio/asm v1.2.0 // indirect
|
||||
go.opentelemetry.io/otel v1.38.0 // indirect
|
||||
go.opentelemetry.io/otel/metric v1.38.0 // indirect
|
||||
go.opentelemetry.io/otel/trace v1.38.0 // indirect
|
||||
go.uber.org/atomic v1.11.0 // indirect
|
||||
go.uber.org/multierr v1.11.0 // indirect
|
||||
go.uber.org/zap v1.27.0 // indirect
|
||||
golang.org/x/crypto v0.43.0 // indirect
|
||||
golang.org/x/exp v0.0.0-20230725093048-515e97ebf090 // indirect
|
||||
golang.org/x/mod v0.29.0 // indirect
|
||||
golang.org/x/sync v0.17.0 // indirect
|
||||
golang.org/x/sys v0.37.0 // indirect
|
||||
golang.org/x/text v0.30.0 // indirect
|
||||
golang.org/x/tools v0.38.0 // indirect
|
||||
gopkg.in/yaml.v2 v2.4.0 // indirect
|
||||
rsc.io/qr v0.2.0 // indirect
|
||||
)
|
||||
98
go.sum
Normal file
98
go.sum
Normal file
@@ -0,0 +1,98 @@
|
||||
github.com/cenkalti/backoff/v4 v4.3.0 h1:MyRJ/UdXutAwSAT+s3wNd7MfTIcy71VQueUuFK343L8=
|
||||
github.com/cenkalti/backoff/v4 v4.3.0/go.mod h1:Y3VNntkOUPxTVeUxJ/G5vcM//AlwfmyYozVcomhLiZE=
|
||||
github.com/coder/websocket v1.8.14 h1:9L0p0iKiNOibykf283eHkKUHHrpG7f65OE3BhhO7v9g=
|
||||
github.com/coder/websocket v1.8.14/go.mod h1:NX3SzP+inril6yawo5CQXx8+fk145lPDC6pumgx0mVg=
|
||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/dlclark/regexp2 v1.11.5 h1:Q/sSnsKerHeCkc/jSTNq1oCm7KiVgUMZRDUoRu0JQZQ=
|
||||
github.com/dlclark/regexp2 v1.11.5/go.mod h1:DHkYz0B9wPfa6wondMfaivmHpzrQ3v9q8cnmRbL6yW8=
|
||||
github.com/fatih/color v1.18.0 h1:S8gINlzdQ840/4pfAwic/ZE0djQEH3wM94VfqLTZcOM=
|
||||
github.com/fatih/color v1.18.0/go.mod h1:4FelSpRwEGDpQ12mAdzqdOukCy4u8WUtOY6lkT/6HfU=
|
||||
github.com/ghodss/yaml v1.0.0 h1:wQHKEahhL6wmXdzwWG11gIVCkOv05bNOh+Rxn0yngAk=
|
||||
github.com/ghodss/yaml v1.0.0/go.mod h1:4dBDuWmgqj2HViK6kFavaiC9ZROes6MMH2rRYeMEF04=
|
||||
github.com/go-faster/errors v0.7.1 h1:MkJTnDoEdi9pDabt1dpWf7AA8/BaSYZqibYyhZ20AYg=
|
||||
github.com/go-faster/errors v0.7.1/go.mod h1:5ySTjWFiphBs07IKuiL69nxdfd5+fzh1u7FPGZP2quo=
|
||||
github.com/go-faster/jx v1.1.0 h1:ZsW3wD+snOdmTDy9eIVgQdjUpXRRV4rqW8NS3t+20bg=
|
||||
github.com/go-faster/jx v1.1.0/go.mod h1:vKDNikrKoyUmpzaJ0OkIkRQClNHFX/nF3dnTJZb3skg=
|
||||
github.com/go-faster/xor v0.3.0/go.mod h1:x5CaDY9UKErKzqfRfFZdfu+OSTfoZny3w5Ak7UxcipQ=
|
||||
github.com/go-faster/xor v1.0.0 h1:2o8vTOgErSGHP3/7XwA5ib1FTtUsNtwCoLLBjl31X38=
|
||||
github.com/go-faster/xor v1.0.0/go.mod h1:x5CaDY9UKErKzqfRfFZdfu+OSTfoZny3w5Ak7UxcipQ=
|
||||
github.com/go-faster/yaml v0.4.6 h1:lOK/EhI04gCpPgPhgt0bChS6bvw7G3WwI8xxVe0sw9I=
|
||||
github.com/go-faster/yaml v0.4.6/go.mod h1:390dRIvV4zbnO7qC9FGo6YYutc+wyyUSHBgbXL52eXk=
|
||||
github.com/go-logr/logr v1.4.3 h1:CjnDlHq8ikf6E492q6eKboGOC0T8CDaOvkHCIg8idEI=
|
||||
github.com/go-logr/logr v1.4.3/go.mod h1:9T104GzyrTigFIr8wt5mBrctHMim0Nb2HLGrmQ40KvY=
|
||||
github.com/go-logr/stdr v1.2.2 h1:hSWxHoqTgW2S2qGc0LTAI563KZ5YKYRhT3MFKZMbjag=
|
||||
github.com/go-logr/stdr v1.2.2/go.mod h1:mMo/vtBO5dYbehREoey6XUKy/eSumjCCveDpRre4VKE=
|
||||
github.com/google/go-cmp v0.7.0 h1:wk8382ETsv4JYUZwIsn6YpYiWiBsYLSJiTsyBybVuN8=
|
||||
github.com/google/go-cmp v0.7.0/go.mod h1:pXiqmnSA92OHEEa9HXL2W4E7lf9JzCmGVUdgjX3N/iU=
|
||||
github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0=
|
||||
github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
|
||||
github.com/gotd/ige v0.2.2 h1:XQ9dJZwBfDnOGSTxKXBGP4gMud3Qku2ekScRjDWWfEk=
|
||||
github.com/gotd/ige v0.2.2/go.mod h1:tuCRb+Y5Y3eNTo3ypIfNpQ4MFjrnONiL2jN2AKZXmb0=
|
||||
github.com/gotd/neo v0.1.5 h1:oj0iQfMbGClP8xI59x7fE/uHoTJD7NZH9oV1WNuPukQ=
|
||||
github.com/gotd/neo v0.1.5/go.mod h1:9A2a4bn9zL6FADufBdt7tZt+WMhvZoc5gWXihOPoiBQ=
|
||||
github.com/gotd/td v0.132.0 h1:Iqm3S2b+8kDgA9237IDXRxj7sryUpvy+4Cr50/0tpx4=
|
||||
github.com/gotd/td v0.132.0/go.mod h1:4CDGYS+rDtOqotRheGaF9MS5g6jaUewvSXqBNJnx8SQ=
|
||||
github.com/klauspost/compress v1.18.0 h1:c/Cqfb0r+Yi+JtIEq73FWXVkRonBlf0CRNYc8Zttxdo=
|
||||
github.com/klauspost/compress v1.18.0/go.mod h1:2Pp+KzxcywXVXMr50+X0Q/Lsb43OQHYWRCY2AiWywWQ=
|
||||
github.com/kr/pretty v0.2.1 h1:Fmg33tUaq4/8ym9TJN1x7sLJnHVwhP33CNkpYV/7rwI=
|
||||
github.com/kr/pretty v0.2.1/go.mod h1:ipq/a2n7PKx3OHsz4KJII5eveXtPO4qwEXGdVfWzfnI=
|
||||
github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
|
||||
github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
|
||||
github.com/mattn/go-colorable v0.1.14 h1:9A9LHSqF/7dyVVX6g0U9cwm9pG3kP9gSzcuIPHPsaIE=
|
||||
github.com/mattn/go-colorable v0.1.14/go.mod h1:6LmQG8QLFO4G5z1gPvYEzlUgJ2wF+stgPZH1UqBm1s8=
|
||||
github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
|
||||
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
||||
github.com/ogen-go/ogen v1.15.2 h1:Hy5XNcDgWur758Kf0+DTQFN8cyBOs58EjDD3NMqih54=
|
||||
github.com/ogen-go/ogen v1.15.2/go.mod h1:bS+BP2cV7+IGjOM24znBmh+PrpZvYFXA7o3BNF4Hj2E=
|
||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||
github.com/segmentio/asm v1.2.0 h1:9BQrFxC+YOHJlTlHGkTrFWf59nbL3XnCoFLTwDCI7ys=
|
||||
github.com/segmentio/asm v1.2.0/go.mod h1:BqMnlJP91P8d+4ibuonYZw9mfnzI9HfxselHZr5aAcs=
|
||||
github.com/stretchr/testify v1.11.1 h1:7s2iGBzp5EwR7/aIZr8ao5+dra3wiQyKjjFuvgVKu7U=
|
||||
github.com/stretchr/testify v1.11.1/go.mod h1:wZwfW3scLgRK+23gO65QZefKpKQRnfz6sD981Nm4B6U=
|
||||
go.opentelemetry.io/auto/sdk v1.1.0 h1:cH53jehLUN6UFLY71z+NDOiNJqDdPRaXzTel0sJySYA=
|
||||
go.opentelemetry.io/auto/sdk v1.1.0/go.mod h1:3wSPjt5PWp2RhlCcmmOial7AvC4DQqZb7a7wCow3W8A=
|
||||
go.opentelemetry.io/otel v1.38.0 h1:RkfdswUDRimDg0m2Az18RKOsnI8UDzppJAtj01/Ymk8=
|
||||
go.opentelemetry.io/otel v1.38.0/go.mod h1:zcmtmQ1+YmQM9wrNsTGV/q/uyusom3P8RxwExxkZhjM=
|
||||
go.opentelemetry.io/otel/metric v1.38.0 h1:Kl6lzIYGAh5M159u9NgiRkmoMKjvbsKtYRwgfrA6WpA=
|
||||
go.opentelemetry.io/otel/metric v1.38.0/go.mod h1:kB5n/QoRM8YwmUahxvI3bO34eVtQf2i4utNVLr9gEmI=
|
||||
go.opentelemetry.io/otel/trace v1.38.0 h1:Fxk5bKrDZJUH+AMyyIXGcFAPah0oRcT+LuNtJrmcNLE=
|
||||
go.opentelemetry.io/otel/trace v1.38.0/go.mod h1:j1P9ivuFsTceSWe1oY+EeW3sc+Pp42sO++GHkg4wwhs=
|
||||
go.uber.org/atomic v1.11.0 h1:ZvwS0R+56ePWxUNi+Atn9dWONBPp/AUETXlHW0DxSjE=
|
||||
go.uber.org/atomic v1.11.0/go.mod h1:LUxbIzbOniOlMKjJjyPfpl4v+PKK2cNJn91OQbhoJI0=
|
||||
go.uber.org/goleak v1.3.0 h1:2K3zAYmnTNqV73imy9J1T3WC+gmCePx2hEGkimedGto=
|
||||
go.uber.org/goleak v1.3.0/go.mod h1:CoHD4mav9JJNrW/WLlf7HGZPjdw8EucARQHekz1X6bE=
|
||||
go.uber.org/multierr v1.11.0 h1:blXXJkSxSSfBVBlC76pxqeO+LN3aDfLQo+309xJstO0=
|
||||
go.uber.org/multierr v1.11.0/go.mod h1:20+QtiLqy0Nd6FdQB9TLXag12DsQkrbs3htMFfDN80Y=
|
||||
go.uber.org/zap v1.27.0 h1:aJMhYGrd5QSmlpLMr2MftRKl7t8J8PTZPA732ud/XR8=
|
||||
go.uber.org/zap v1.27.0/go.mod h1:GB2qFLM7cTU87MWRP2mPIjqfIDnGu+VIO4V/SdhGo2E=
|
||||
golang.org/x/crypto v0.43.0 h1:dduJYIi3A3KOfdGOHX8AVZ/jGiyPa3IbBozJ5kNuE04=
|
||||
golang.org/x/crypto v0.43.0/go.mod h1:BFbav4mRNlXJL4wNeejLpWxB7wMbc79PdRGhWKncxR0=
|
||||
golang.org/x/exp v0.0.0-20230725093048-515e97ebf090 h1:Di6/M8l0O2lCLc6VVRWhgCiApHV8MnQurBnFSHsQtNY=
|
||||
golang.org/x/exp v0.0.0-20230725093048-515e97ebf090/go.mod h1:FXUEEKJgO7OQYeo8N01OfiKP8RXMtf6e8aTskBGqWdc=
|
||||
golang.org/x/mod v0.29.0 h1:HV8lRxZC4l2cr3Zq1LvtOsi/ThTgWnUk/y64QSs8GwA=
|
||||
golang.org/x/mod v0.29.0/go.mod h1:NyhrlYXJ2H4eJiRy/WDBO6HMqZQ6q9nk4JzS3NuCK+w=
|
||||
golang.org/x/net v0.46.0 h1:giFlY12I07fugqwPuWJi68oOnpfqFnJIJzaIIm2JVV4=
|
||||
golang.org/x/net v0.46.0/go.mod h1:Q9BGdFy1y4nkUwiLvT5qtyhAnEHgnQ/zd8PfU6nc210=
|
||||
golang.org/x/sync v0.0.0-20201207232520-09787c993a3a/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM=
|
||||
golang.org/x/sync v0.17.0 h1:l60nONMj9l5drqw6jlhIELNv9I0A4OFgRsG9k2oT9Ug=
|
||||
golang.org/x/sync v0.17.0/go.mod h1:9KTHXmSnoGruLpwFjVSX0lNNA75CykiMECbovNTZqGI=
|
||||
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
golang.org/x/sys v0.37.0 h1:fdNQudmxPjkdUTPnLn5mdQv7Zwvbvpaxqs831goi9kQ=
|
||||
golang.org/x/sys v0.37.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
|
||||
golang.org/x/text v0.30.0 h1:yznKA/E9zq54KzlzBEAWn1NXSQ8DIp/NYMy88xJjl4k=
|
||||
golang.org/x/text v0.30.0/go.mod h1:yDdHFIX9t+tORqspjENWgzaCVXgk0yYnYuSZ8UzzBVM=
|
||||
golang.org/x/tools v0.38.0 h1:Hx2Xv8hISq8Lm16jvBZ2VQf+RLmbd7wVUsALibYI/IQ=
|
||||
golang.org/x/tools v0.38.0/go.mod h1:yEsQ/d/YK8cjh0L6rZlY8tgtlKiBNTL14pGDJPJpYQs=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15 h1:YR8cESwS4TdDjEe65xsg0ogRM/Nc3DYOhEAlW+xobZo=
|
||||
gopkg.in/check.v1 v1.0.0-20190902080502-41f04d3bba15/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY=
|
||||
gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ=
|
||||
gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA=
|
||||
gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
||||
nhooyr.io/websocket v1.8.17 h1:KEVeLJkUywCKVsnLIDlD/5gtayKp8VoCkksHCGGfT9Y=
|
||||
nhooyr.io/websocket v1.8.17/go.mod h1:rN9OFWIUwuxg4fR5tELlYC04bXYowCP9GX47ivo2l+c=
|
||||
rsc.io/qr v0.2.0 h1:6vBLea5/NRMVTz8V66gipeLycZMl/+UlFmk8DvqQ6WY=
|
||||
rsc.io/qr v0.2.0/go.mod h1:IF+uZjkb9fqyeF/4tlBoynqmQxUoPfWEKh921coOuXs=
|
||||
153
internal/app/app.go
Normal file
153
internal/app/app.go
Normal file
@@ -0,0 +1,153 @@
|
||||
package app
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
tgclient "funstatmcp/internal/telegram"
|
||||
)
|
||||
|
||||
type App struct {
|
||||
cfg Config
|
||||
client *tgclient.Client
|
||||
}
|
||||
|
||||
func New(cfg Config) (*App, error) {
|
||||
client, err := tgclient.New(cfg.Telegram)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return &App{
|
||||
cfg: cfg,
|
||||
client: client,
|
||||
}, nil
|
||||
}
|
||||
|
||||
func (a *App) Close() error {
|
||||
return nil
|
||||
}
|
||||
|
||||
func (a *App) SendCommand(ctx context.Context, command string, useCache bool) (string, error) {
|
||||
return a.client.SendCommand(ctx, command, useCache)
|
||||
}
|
||||
|
||||
func (a *App) CallTool(ctx context.Context, name string, args map[string]any) (string, error) {
|
||||
switch name {
|
||||
case "funstat_search":
|
||||
query, err := requireString(args, "query")
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
return a.client.SendCommand(ctx, fmt.Sprintf("/search %s", query), true)
|
||||
|
||||
case "funstat_topchat":
|
||||
category := optionalString(args, "category")
|
||||
if category != "" {
|
||||
return a.client.SendCommand(ctx, fmt.Sprintf("/topchat %s", category), true)
|
||||
}
|
||||
return a.client.SendCommand(ctx, "/topchat", true)
|
||||
|
||||
case "funstat_text":
|
||||
text, err := requireString(args, "text")
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
return a.client.SendCommand(ctx, fmt.Sprintf("/text %s", text), true)
|
||||
|
||||
case "funstat_human":
|
||||
nameArg, err := requireString(args, "name")
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
return a.client.SendCommand(ctx, fmt.Sprintf("/human %s", nameArg), true)
|
||||
|
||||
case "funstat_user_info":
|
||||
identifier, err := requireString(args, "identifier")
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
identifier = strings.TrimSpace(identifier)
|
||||
if identifier == "" {
|
||||
return "", fmt.Errorf("identifier cannot be empty")
|
||||
}
|
||||
return a.client.SendCommand(ctx, fmt.Sprintf("/user_info %s", identifier), true)
|
||||
|
||||
case "funstat_user_messages":
|
||||
identifier, err := requireString(args, "identifier")
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
var maxPagesPtr *int
|
||||
if value, ok := args["max_pages"]; ok {
|
||||
v, err := toInt(value)
|
||||
if err != nil {
|
||||
return "", fmt.Errorf("max_pages must be an integer: %w", err)
|
||||
}
|
||||
maxPagesPtr = &v
|
||||
}
|
||||
return a.client.FetchUserMessages(ctx, identifier, maxPagesPtr)
|
||||
|
||||
case "funstat_balance":
|
||||
return a.client.SendCommand(ctx, "/balance", true)
|
||||
|
||||
case "funstat_menu":
|
||||
return a.client.SendCommand(ctx, "/menu", true)
|
||||
|
||||
case "funstat_start":
|
||||
return a.client.SendCommand(ctx, "/start", true)
|
||||
|
||||
default:
|
||||
return "", fmt.Errorf("unknown tool: %s", name)
|
||||
}
|
||||
}
|
||||
|
||||
func requireString(args map[string]any, key string) (string, error) {
|
||||
value, ok := args[key]
|
||||
if !ok {
|
||||
return "", fmt.Errorf("missing required argument: %s", key)
|
||||
}
|
||||
str, ok := value.(string)
|
||||
if !ok {
|
||||
return "", fmt.Errorf("argument %s must be a string", key)
|
||||
}
|
||||
str = strings.TrimSpace(str)
|
||||
if str == "" {
|
||||
return "", fmt.Errorf("argument %s cannot be empty", key)
|
||||
}
|
||||
return str, nil
|
||||
}
|
||||
|
||||
func optionalString(args map[string]any, key string) string {
|
||||
if value, ok := args[key]; ok {
|
||||
if str, ok := value.(string); ok {
|
||||
return strings.TrimSpace(str)
|
||||
}
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func toInt(value any) (int, error) {
|
||||
switch v := value.(type) {
|
||||
case float64:
|
||||
return int(v), nil
|
||||
case float32:
|
||||
return int(v), nil
|
||||
case int:
|
||||
return v, nil
|
||||
case int32:
|
||||
return int(v), nil
|
||||
case int64:
|
||||
return int(v), nil
|
||||
case string:
|
||||
parsed, err := strconv.Atoi(strings.TrimSpace(v))
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
return parsed, nil
|
||||
default:
|
||||
return 0, fmt.Errorf("unsupported type %T", value)
|
||||
}
|
||||
}
|
||||
158
internal/app/config.go
Normal file
158
internal/app/config.go
Normal file
@@ -0,0 +1,158 @@
|
||||
package app
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strconv"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
tgconfig "funstatmcp/internal/telegram"
|
||||
)
|
||||
|
||||
type Config struct {
|
||||
Telegram tgconfig.Config
|
||||
Host string
|
||||
Port int
|
||||
RequireSession bool
|
||||
}
|
||||
|
||||
func FromEnv() (Config, error) {
|
||||
var cfg Config
|
||||
|
||||
telegramCfg, err := loadTelegramConfig()
|
||||
if err != nil {
|
||||
return cfg, err
|
||||
}
|
||||
|
||||
cfg.Telegram = telegramCfg
|
||||
cfg.Host = getEnvDefault("FUNSTAT_HOST", "127.0.0.1")
|
||||
|
||||
portStr := getEnvDefault("FUNSTAT_PORT", "8091")
|
||||
port, err := strconv.Atoi(portStr)
|
||||
if err != nil {
|
||||
return cfg, fmt.Errorf("invalid FUNSTAT_PORT: %w", err)
|
||||
}
|
||||
cfg.Port = port
|
||||
|
||||
cfg.RequireSession = parseBool(getEnvDefault("FUNSTAT_REQUIRE_SESSION", "false"))
|
||||
|
||||
return cfg, nil
|
||||
}
|
||||
|
||||
func loadTelegramConfig() (tgconfig.Config, error) {
|
||||
var cfg tgconfig.Config
|
||||
|
||||
apiIDStr := os.Getenv("TELEGRAM_API_ID")
|
||||
if apiIDStr == "" {
|
||||
return cfg, fmt.Errorf("TELEGRAM_API_ID is required")
|
||||
}
|
||||
apiID, err := strconv.Atoi(apiIDStr)
|
||||
if err != nil {
|
||||
return cfg, fmt.Errorf("invalid TELEGRAM_API_ID: %w", err)
|
||||
}
|
||||
cfg.APIID = apiID
|
||||
|
||||
cfg.APIHash = strings.TrimSpace(os.Getenv("TELEGRAM_API_HASH"))
|
||||
if cfg.APIHash == "" {
|
||||
return cfg, fmt.Errorf("TELEGRAM_API_HASH is required")
|
||||
}
|
||||
|
||||
cfg.BotUsername = getEnvDefault("FUNSTAT_BOT_USERNAME", "@openaiw_bot")
|
||||
cfg.SessionString = strings.TrimSpace(os.Getenv("TELEGRAM_SESSION_STRING"))
|
||||
sessionStringFile := strings.TrimSpace(os.Getenv("TELEGRAM_SESSION_STRING_FILE"))
|
||||
if cfg.SessionString == "" && sessionStringFile != "" {
|
||||
data, err := os.ReadFile(expandPath(sessionStringFile))
|
||||
if err != nil {
|
||||
return cfg, fmt.Errorf("read TELEGRAM_SESSION_STRING_FILE: %w", err)
|
||||
}
|
||||
cfg.SessionString = strings.TrimSpace(string(data))
|
||||
}
|
||||
|
||||
sessionPath := strings.TrimSpace(os.Getenv("TELEGRAM_SESSION_PATH"))
|
||||
if sessionPath != "" {
|
||||
if !strings.HasSuffix(sessionPath, ".session") {
|
||||
sessionPath = sessionPath + ".session"
|
||||
}
|
||||
cfg.SessionStorage = expandPath(sessionPath)
|
||||
} else {
|
||||
cfg.SessionStorage = defaultSessionPath()
|
||||
}
|
||||
|
||||
if value := strings.TrimSpace(os.Getenv("FUNSTAT_RATE_LIMIT_PER_SECOND")); value != "" {
|
||||
parsed, err := strconv.Atoi(value)
|
||||
if err != nil {
|
||||
return cfg, fmt.Errorf("invalid FUNSTAT_RATE_LIMIT_PER_SECOND: %w", err)
|
||||
}
|
||||
cfg.RateLimit = parsed
|
||||
}
|
||||
|
||||
if value := strings.TrimSpace(os.Getenv("FUNSTAT_RATE_LIMIT_WINDOW")); value != "" {
|
||||
duration, err := time.ParseDuration(value)
|
||||
if err != nil {
|
||||
return cfg, fmt.Errorf("invalid FUNSTAT_RATE_LIMIT_WINDOW: %w", err)
|
||||
}
|
||||
cfg.RateLimitWindow = duration
|
||||
}
|
||||
|
||||
if value := strings.TrimSpace(os.Getenv("FUNSTAT_CACHE_TTL")); value != "" {
|
||||
duration, err := time.ParseDuration(value)
|
||||
if err != nil {
|
||||
return cfg, fmt.Errorf("invalid FUNSTAT_CACHE_TTL: %w", err)
|
||||
}
|
||||
cfg.CacheTTL = duration
|
||||
}
|
||||
|
||||
proxyHost := strings.TrimSpace(os.Getenv("FUNSTAT_PROXY_HOST"))
|
||||
proxyPort := strings.TrimSpace(os.Getenv("FUNSTAT_PROXY_PORT"))
|
||||
if proxyHost != "" && proxyPort != "" {
|
||||
port, err := strconv.Atoi(proxyPort)
|
||||
if err != nil {
|
||||
return cfg, fmt.Errorf("invalid FUNSTAT_PROXY_PORT: %w", err)
|
||||
}
|
||||
cfg.Proxy = &tgconfig.ProxyConfig{
|
||||
Type: getEnvDefault("FUNSTAT_PROXY_TYPE", "socks5"),
|
||||
Host: proxyHost,
|
||||
Port: port,
|
||||
Username: strings.TrimSpace(os.Getenv("FUNSTAT_PROXY_USERNAME")),
|
||||
Password: strings.TrimSpace(os.Getenv("FUNSTAT_PROXY_PASSWORD")),
|
||||
}
|
||||
}
|
||||
|
||||
return cfg, nil
|
||||
}
|
||||
|
||||
func getEnvDefault(key, fallback string) string {
|
||||
if value := strings.TrimSpace(os.Getenv(key)); value != "" {
|
||||
return value
|
||||
}
|
||||
return fallback
|
||||
}
|
||||
|
||||
func parseBool(value string) bool {
|
||||
switch strings.ToLower(strings.TrimSpace(value)) {
|
||||
case "1", "true", "yes", "on":
|
||||
return true
|
||||
default:
|
||||
return false
|
||||
}
|
||||
}
|
||||
|
||||
func defaultSessionPath() string {
|
||||
home, err := os.UserHomeDir()
|
||||
if err != nil {
|
||||
return filepath.Join(os.TempDir(), "funstatmcp", "session.json")
|
||||
}
|
||||
return filepath.Join(home, ".funstatmcp", "session.json")
|
||||
}
|
||||
|
||||
func expandPath(path string) string {
|
||||
if strings.HasPrefix(path, "~") {
|
||||
home, err := os.UserHomeDir()
|
||||
if err == nil {
|
||||
return filepath.Join(home, strings.TrimPrefix(path, "~"))
|
||||
}
|
||||
}
|
||||
return path
|
||||
}
|
||||
124
internal/app/tools.go
Normal file
124
internal/app/tools.go
Normal file
@@ -0,0 +1,124 @@
|
||||
package app
|
||||
|
||||
type ToolDefinition struct {
|
||||
Name string `json:"name"`
|
||||
Description string `json:"description"`
|
||||
InputSchema map[string]any `json:"inputSchema"`
|
||||
}
|
||||
|
||||
func ToolDefinitions() []ToolDefinition {
|
||||
return []ToolDefinition{
|
||||
{
|
||||
Name: "funstat_search",
|
||||
Description: "搜索 Telegram 群组、频道。支持关键词搜索，返回相关的群组列表",
|
||||
InputSchema: map[string]any{
|
||||
"type": "object",
|
||||
"properties": map[string]any{
|
||||
"query": map[string]any{
|
||||
"type": "string",
|
||||
"description": "搜索关键词，例如: 'python', '区块链', 'AI'",
|
||||
},
|
||||
},
|
||||
"required": []string{"query"},
|
||||
},
|
||||
},
|
||||
{
|
||||
Name: "funstat_topchat",
|
||||
Description: "获取热门群组/频道列表，按成员数或活跃度排序",
|
||||
InputSchema: map[string]any{
|
||||
"type": "object",
|
||||
"properties": map[string]any{
|
||||
"category": map[string]any{
|
||||
"type": "string",
|
||||
"description": "分类筛选（可选），例如: 'tech', 'crypto', 'news'",
|
||||
},
|
||||
},
|
||||
},
|
||||
},
|
||||
{
|
||||
Name: "funstat_text",
|
||||
Description: "通过消息文本搜索，查找包含特定文本的消息和来源群组",
|
||||
InputSchema: map[string]any{
|
||||
"type": "object",
|
||||
"properties": map[string]any{
|
||||
"text": map[string]any{
|
||||
"type": "string",
|
||||
"description": "要搜索的文本内容",
|
||||
},
|
||||
},
|
||||
"required": []string{"text"},
|
||||
},
|
||||
},
|
||||
{
|
||||
Name: "funstat_human",
|
||||
Description: "通过姓名搜索，查找包含特定用户的群组和消息",
|
||||
InputSchema: map[string]any{
|
||||
"type": "object",
|
||||
"properties": map[string]any{
|
||||
"name": map[string]any{
|
||||
"type": "string",
|
||||
"description": "用户姓名",
|
||||
},
|
||||
},
|
||||
"required": []string{"name"},
|
||||
},
|
||||
},
|
||||
{
|
||||
Name: "funstat_user_info",
|
||||
Description: "查询用户详细信息，支持通过用户名、用户ID、联系人等方式查询",
|
||||
InputSchema: map[string]any{
|
||||
"type": "object",
|
||||
"properties": map[string]any{
|
||||
"identifier": map[string]any{
|
||||
"type": "string",
|
||||
"description": "用户标识: 用户名(@username)、用户ID、或手机号",
|
||||
},
|
||||
},
|
||||
"required": []string{"identifier"},
|
||||
},
|
||||
},
|
||||
{
|
||||
Name: "funstat_user_messages",
|
||||
Description: "获取指定用户的历史消息列表，并自动翻页汇总",
|
||||
InputSchema: map[string]any{
|
||||
"type": "object",
|
||||
"properties": map[string]any{
|
||||
"identifier": map[string]any{
|
||||
"type": "string",
|
||||
"description": "用户标识: 用户名(@username) 或用户ID",
|
||||
},
|
||||
"max_pages": map[string]any{
|
||||
"type": "integer",
|
||||
"minimum": 1,
|
||||
"description": "可选，限制抓取的最大页数",
|
||||
},
|
||||
},
|
||||
"required": []string{"identifier"},
|
||||
},
|
||||
},
|
||||
{
|
||||
Name: "funstat_balance",
|
||||
Description: "查询当前账号的积分余额和使用统计",
|
||||
InputSchema: map[string]any{
|
||||
"type": "object",
|
||||
"properties": map[string]any{},
|
||||
},
|
||||
},
|
||||
{
|
||||
Name: "funstat_menu",
|
||||
Description: "显示 funstat BOT 的主菜单和所有可用功能",
|
||||
InputSchema: map[string]any{
|
||||
"type": "object",
|
||||
"properties": map[string]any{},
|
||||
},
|
||||
},
|
||||
{
|
||||
Name: "funstat_start",
|
||||
Description: "获取 funstat BOT 的欢迎信息和使用说明",
|
||||
InputSchema: map[string]any{
|
||||
"type": "object",
|
||||
"properties": map[string]any{},
|
||||
},
|
||||
},
|
||||
}
|
||||
}
|
||||
70
internal/cache/cache.go
vendored
Normal file
70
internal/cache/cache.go
vendored
Normal file
@@ -0,0 +1,70 @@
|
||||
package cache
|
||||
|
||||
import (
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
type entry struct {
|
||||
value string
|
||||
expires time.Time
|
||||
}
|
||||
|
||||
type Cache struct {
|
||||
ttl time.Duration
|
||||
mu sync.RWMutex
|
||||
values map[string]entry
|
||||
}
|
||||
|
||||
func New(ttl time.Duration) *Cache {
|
||||
if ttl <= 0 {
|
||||
ttl = time.Hour
|
||||
}
|
||||
return &Cache{
|
||||
ttl: ttl,
|
||||
values: make(map[string]entry),
|
||||
}
|
||||
}
|
||||
|
||||
func (c *Cache) Get(key string) (string, bool) {
|
||||
c.mu.RLock()
|
||||
e, ok := c.values[key]
|
||||
c.mu.RUnlock()
|
||||
if !ok {
|
||||
return "", false
|
||||
}
|
||||
if time.Now().After(e.expires) {
|
||||
c.mu.Lock()
|
||||
delete(c.values, key)
|
||||
c.mu.Unlock()
|
||||
return "", false
|
||||
}
|
||||
return e.value, true
|
||||
}
|
||||
|
||||
func (c *Cache) Set(key string, value string) {
|
||||
c.mu.Lock()
|
||||
c.values[key] = entry{
|
||||
value: value,
|
||||
expires: time.Now().Add(c.ttl),
|
||||
}
|
||||
c.mu.Unlock()
|
||||
}
|
||||
|
||||
func (c *Cache) ClearExpired() int {
|
||||
now := time.Now()
|
||||
c.mu.Lock()
|
||||
removed := 0
|
||||
for k, v := range c.values {
|
||||
if now.After(v.expires) {
|
||||
delete(c.values, k)
|
||||
removed++
|
||||
}
|
||||
}
|
||||
c.mu.Unlock()
|
||||
return removed
|
||||
}
|
||||
|
||||
func (c *Cache) TTL() time.Duration {
|
||||
return c.ttl
|
||||
}
|
||||
71
internal/ratelimit/limiter.go
Normal file
71
internal/ratelimit/limiter.go
Normal file
@@ -0,0 +1,71 @@
|
||||
package ratelimit
|
||||
|
||||
import (
|
||||
"context"
|
||||
"sync"
|
||||
"time"
|
||||
)
|
||||
|
||||
type Limiter struct {
|
||||
maxRequests int
|
||||
window time.Duration
|
||||
|
||||
mu sync.Mutex
|
||||
timestamps []time.Time
|
||||
}
|
||||
|
||||
func New(maxRequests int, window time.Duration) *Limiter {
|
||||
if maxRequests <= 0 {
|
||||
maxRequests = 1
|
||||
}
|
||||
if window <= 0 {
|
||||
window = time.Second
|
||||
}
|
||||
|
||||
return &Limiter{
|
||||
maxRequests: maxRequests,
|
||||
window: window,
|
||||
timestamps: make([]time.Time, 0, maxRequests),
|
||||
}
|
||||
}
|
||||
|
||||
func (l *Limiter) Wait(ctx context.Context) error {
|
||||
for {
|
||||
l.mu.Lock()
|
||||
now := time.Now()
|
||||
|
||||
cutoff := now.Add(-l.window)
|
||||
idx := 0
|
||||
for ; idx < len(l.timestamps); idx++ {
|
||||
if l.timestamps[idx].After(cutoff) {
|
||||
break
|
||||
}
|
||||
}
|
||||
|
||||
if idx > 0 {
|
||||
l.timestamps = append([]time.Time(nil), l.timestamps[idx:]...)
|
||||
}
|
||||
|
||||
if len(l.timestamps) < l.maxRequests {
|
||||
l.timestamps = append(l.timestamps, now)
|
||||
l.mu.Unlock()
|
||||
return nil
|
||||
}
|
||||
|
||||
waitUntil := l.timestamps[0].Add(l.window)
|
||||
waitDuration := time.Until(waitUntil)
|
||||
l.mu.Unlock()
|
||||
|
||||
if waitDuration <= 0 {
|
||||
continue
|
||||
}
|
||||
|
||||
timer := time.NewTimer(waitDuration)
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
timer.Stop()
|
||||
return ctx.Err()
|
||||
case <-timer.C:
|
||||
}
|
||||
}
|
||||
}
|
||||
71
internal/telegram/buttons.go
Normal file
71
internal/telegram/buttons.go
Normal file
@@ -0,0 +1,71 @@
|
||||
package telegram
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"strconv"
|
||||
"strings"
|
||||
"unicode"
|
||||
|
||||
"github.com/gotd/td/tg"
|
||||
)
|
||||
|
||||
func findCallbackButton(message *tg.Message, keyword string) (*tg.KeyboardButtonCallback, error) {
|
||||
markup, ok := message.ReplyMarkup.(*tg.ReplyInlineMarkup)
|
||||
if !ok || len(markup.Rows) == 0 {
|
||||
return nil, fmt.Errorf("message has no interactive buttons")
|
||||
}
|
||||
|
||||
normalizedKeyword := strings.ToLower(normalizeButtonText(keyword))
|
||||
available := make([]string, 0)
|
||||
|
||||
for _, row := range markup.Rows {
|
||||
for _, button := range row.Buttons {
|
||||
callback, ok := button.(*tg.KeyboardButtonCallback)
|
||||
if !ok {
|
||||
continue
|
||||
}
|
||||
|
||||
text := callback.Text
|
||||
normalized := strings.ToLower(normalizeButtonText(text))
|
||||
available = append(available, normalizeButtonText(text))
|
||||
|
||||
if strings.Contains(normalized, normalizedKeyword) {
|
||||
return callback, nil
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return nil, fmt.Errorf("button containing '%s' not found (available: %s)", keyword, strings.Join(available, ", "))
|
||||
}
|
||||
|
||||
func extractTotalPages(message *tg.Message) int {
|
||||
markup, ok := message.ReplyMarkup.(*tg.ReplyInlineMarkup)
|
||||
if !ok {
|
||||
return 0
|
||||
}
|
||||
|
||||
for _, row := range markup.Rows {
|
||||
for _, button := range row.Buttons {
|
||||
callback, ok := button.(*tg.KeyboardButtonCallback)
|
||||
if !ok {
|
||||
continue
|
||||
}
|
||||
|
||||
if strings.Contains(callback.Text, "⏭") {
|
||||
digits := strings.Builder{}
|
||||
for _, r := range normalizeButtonText(callback.Text) {
|
||||
if unicode.IsDigit(r) {
|
||||
digits.WriteRune(r)
|
||||
}
|
||||
}
|
||||
if digits.Len() > 0 {
|
||||
if value, err := strconv.Atoi(digits.String()); err == nil {
|
||||
return value
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return 0
|
||||
}
|
||||
497
internal/telegram/client.go
Normal file
497
internal/telegram/client.go
Normal file
@@ -0,0 +1,497 @@
|
||||
package telegram
|
||||
|
||||
import (
|
||||
"context"
|
||||
"errors"
|
||||
"fmt"
|
||||
"net"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"golang.org/x/net/proxy"
|
||||
|
||||
"github.com/gotd/td/session"
|
||||
"github.com/gotd/td/telegram"
|
||||
"github.com/gotd/td/telegram/dcs"
|
||||
"github.com/gotd/td/telegram/message"
|
||||
"github.com/gotd/td/tg"
|
||||
|
||||
"funstatmcp/internal/cache"
|
||||
"funstatmcp/internal/ratelimit"
|
||||
)
|
||||
|
||||
type Client struct {
|
||||
cfg Config
|
||||
limiter *ratelimit.Limiter
|
||||
cache *cache.Cache
|
||||
sessionPath string
|
||||
sessionOnce sync.Once
|
||||
}
|
||||
|
||||
func New(cfg Config) (*Client, error) {
|
||||
if err := cfg.Validate(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
sessionPath := cfg.SessionStorage
|
||||
if sessionPath == "" {
|
||||
home, err := os.UserHomeDir()
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("get home dir: %w", err)
|
||||
}
|
||||
sessionPath = filepath.Join(home, ".funstatmcp", "session.json")
|
||||
}
|
||||
|
||||
if err := os.MkdirAll(filepath.Dir(sessionPath), 0o700); err != nil {
|
||||
return nil, fmt.Errorf("create session directory: %w", err)
|
||||
}
|
||||
|
||||
client := &Client{
|
||||
cfg: cfg,
|
||||
limiter: ratelimit.New(cfg.RateLimitPerSecond(), cfg.RateLimitDuration()),
|
||||
cache: cache.New(cfg.CacheDuration()),
|
||||
sessionPath: sessionPath,
|
||||
}
|
||||
|
||||
if strings.TrimSpace(cfg.SessionString) != "" {
|
||||
if err := client.writeStringSession(strings.TrimSpace(cfg.SessionString)); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
|
||||
return client, nil
|
||||
}
|
||||
|
||||
func (c *Client) writeStringSession(sessionStr string) error {
|
||||
var result error
|
||||
c.sessionOnce.Do(func() {
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
defer cancel()
|
||||
|
||||
data, err := session.TelethonSession(sessionStr)
|
||||
if err != nil {
|
||||
result = fmt.Errorf("decode telethon session: %w", err)
|
||||
return
|
||||
}
|
||||
|
||||
loader := session.Loader{Storage: &session.FileStorage{Path: c.sessionPath}}
|
||||
if err := loader.Save(ctx, data); err != nil {
|
||||
result = fmt.Errorf("save session: %w", err)
|
||||
return
|
||||
}
|
||||
})
|
||||
return result
|
||||
}
|
||||
|
||||
func (c *Client) createOptions() (telegram.Options, error) {
|
||||
opts := telegram.Options{
|
||||
SessionStorage: &session.FileStorage{Path: c.sessionPath},
|
||||
NoUpdates: true,
|
||||
}
|
||||
|
||||
proxyCfg := c.cfg.Proxy
|
||||
if proxyCfg != nil && proxyCfg.Host != "" && proxyCfg.Port > 0 {
|
||||
address := fmt.Sprintf("%s:%d", proxyCfg.Host, proxyCfg.Port)
|
||||
switch strings.ToLower(proxyCfg.Type) {
|
||||
case "", "socks5", "socks":
|
||||
var auth *proxy.Auth
|
||||
if proxyCfg.Username != "" {
|
||||
auth = &proxy.Auth{User: proxyCfg.Username, Password: proxyCfg.Password}
|
||||
}
|
||||
dialer, err := proxy.SOCKS5("tcp", address, auth, proxy.Direct)
|
||||
if err != nil {
|
||||
return opts, fmt.Errorf("create SOCKS5 proxy: %w", err)
|
||||
}
|
||||
contextDialer, ok := dialer.(proxy.ContextDialer)
|
||||
if !ok {
|
||||
contextDialer = &contextDialerAdapter{Dialer: dialer}
|
||||
}
|
||||
opts.Resolver = dcs.Plain(dcs.PlainOptions{Dial: contextDialer.DialContext})
|
||||
default:
|
||||
return opts, fmt.Errorf("unsupported proxy type %q", proxyCfg.Type)
|
||||
}
|
||||
}
|
||||
|
||||
return opts, nil
|
||||
}
|
||||
|
||||
type contextDialerAdapter struct {
|
||||
Dialer proxy.Dialer
|
||||
}
|
||||
|
||||
func (a *contextDialerAdapter) DialContext(ctx context.Context, network, addr string) (net.Conn, error) {
|
||||
type dialResult struct {
|
||||
conn net.Conn
|
||||
err error
|
||||
}
|
||||
|
||||
result := make(chan dialResult, 1)
|
||||
go func() {
|
||||
conn, err := a.Dialer.Dial(network, addr)
|
||||
result <- dialResult{conn: conn, err: err}
|
||||
}()
|
||||
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
return nil, ctx.Err()
|
||||
case res := <-result:
|
||||
return res.conn, res.err
|
||||
}
|
||||
}
|
||||
|
||||
func (c *Client) withClient(ctx context.Context, fn func(ctx context.Context, api *tg.Client, sender *message.Sender) error) error {
|
||||
opts, err := c.createOptions()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
client := telegram.NewClient(c.cfg.APIID, c.cfg.APIHash, opts)
|
||||
|
||||
return client.Run(ctx, func(runCtx context.Context) error {
|
||||
if err := c.ensureAuthorized(runCtx, client); err != nil {
|
||||
return err
|
||||
}
|
||||
raw := tg.NewClient(client)
|
||||
sender := message.NewSender(raw)
|
||||
return fn(runCtx, raw, sender)
|
||||
})
|
||||
}
|
||||
|
||||
func (c *Client) withPeer(ctx context.Context, fn func(ctx context.Context, api *tg.Client, sender *message.Sender, peer tg.InputPeerClass, botID int64) error) error {
|
||||
return c.withClient(ctx, func(runCtx context.Context, api *tg.Client, sender *message.Sender) error {
|
||||
peer, botID, err := c.resolvePeer(runCtx, sender)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
return fn(runCtx, api, sender, peer, botID)
|
||||
})
|
||||
}
|
||||
|
||||
func (c *Client) ensureAuthorized(ctx context.Context, client *telegram.Client) error {
|
||||
status, err := client.Auth().Status(ctx)
|
||||
if err != nil {
|
||||
return fmt.Errorf("check auth status: %w", err)
|
||||
}
|
||||
if !status.Authorized {
|
||||
return errors.New("telegram session is not authorized; provide TELEGRAM_SESSION_STRING")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c *Client) resolvePeer(ctx context.Context, sender *message.Sender) (tg.InputPeerClass, int64, error) {
|
||||
builder := sender.Resolve(c.cfg.BotUsername)
|
||||
peer, err := builder.AsInputPeer(ctx)
|
||||
if err != nil {
|
||||
return nil, 0, fmt.Errorf("resolve bot peer: %w", err)
|
||||
}
|
||||
|
||||
inputUser, err := builder.AsInputUser(ctx)
|
||||
if err != nil {
|
||||
return nil, 0, fmt.Errorf("resolve bot user: %w", err)
|
||||
}
|
||||
|
||||
return peer, inputUser.UserID, nil
|
||||
}
|
||||
|
||||
func (c *Client) latestIncomingMessageID(ctx context.Context, api *tg.Client, peer tg.InputPeerClass, botID int64) (int, error) {
|
||||
resp, err := api.MessagesGetHistory(ctx, &tg.MessagesGetHistoryRequest{
|
||||
Peer: peer,
|
||||
Limit: 5,
|
||||
})
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
messages, err := extractMessages(resp)
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
|
||||
last := 0
|
||||
for _, msg := range messages {
|
||||
if isFromBot(msg, botID) && msg.ID > last {
|
||||
last = msg.ID
|
||||
}
|
||||
}
|
||||
return last, nil
|
||||
}
|
||||
|
||||
func (c *Client) waitForMessage(ctx context.Context, api *tg.Client, peer tg.InputPeerClass, botID int64, lastID int, timeout time.Duration) (*tg.Message, error) {
|
||||
deadline := time.Now().Add(timeout)
|
||||
for {
|
||||
if err := ctx.Err(); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
resp, err := api.MessagesGetHistory(ctx, &tg.MessagesGetHistoryRequest{
|
||||
Peer: peer,
|
||||
Limit: 5,
|
||||
})
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
messages, err := extractMessages(resp)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
for _, msg := range messages {
|
||||
if msg.ID > lastID && isFromBot(msg, botID) {
|
||||
return msg, nil
|
||||
}
|
||||
}
|
||||
|
||||
if time.Now().After(deadline) {
|
||||
return nil, fmt.Errorf("timeout waiting for bot response")
|
||||
}
|
||||
|
||||
if err := sleepWithContext(ctx, 500*time.Millisecond); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (c *Client) SendCommand(ctx context.Context, command string, useCache bool) (string, error) {
|
||||
cacheKey := fmt.Sprintf("cmd:%s", command)
|
||||
if useCache {
|
||||
if cached, ok := c.cache.Get(cacheKey); ok {
|
||||
return cached, nil
|
||||
}
|
||||
}
|
||||
|
||||
if err := c.limiter.Wait(ctx); err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
var response string
|
||||
err := c.withPeer(ctx, func(runCtx context.Context, api *tg.Client, sender *message.Sender, peer tg.InputPeerClass, botID int64) error {
|
||||
lastID, err := c.latestIncomingMessageID(runCtx, api, peer, botID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if _, err := sender.Resolve(c.cfg.BotUsername).Text(runCtx, command); err != nil {
|
||||
return fmt.Errorf("send command: %w", err)
|
||||
}
|
||||
|
||||
msg, err := c.waitForMessage(runCtx, api, peer, botID, lastID, 15*time.Second)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
response = strings.TrimSpace(msg.Message)
|
||||
return nil
|
||||
})
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
if useCache {
|
||||
c.cache.Set(cacheKey, response)
|
||||
}
|
||||
|
||||
return response, nil
|
||||
}
|
||||
|
||||
func (c *Client) SendCommandMessage(ctx context.Context, command string, timeout time.Duration) (*tg.Message, error) {
|
||||
if err := c.limiter.Wait(ctx); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
var result *tg.Message
|
||||
err := c.withPeer(ctx, func(runCtx context.Context, api *tg.Client, sender *message.Sender, peer tg.InputPeerClass, botID int64) error {
|
||||
lastID, err := c.latestIncomingMessageID(runCtx, api, peer, botID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
if _, err := sender.Resolve(c.cfg.BotUsername).Text(runCtx, command); err != nil {
|
||||
return fmt.Errorf("send command: %w", err)
|
||||
}
|
||||
|
||||
msg, err := c.waitForMessage(runCtx, api, peer, botID, lastID, timeout)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
result = msg
|
||||
return nil
|
||||
})
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return result, nil
|
||||
}
|
||||
|
||||
func (c *Client) PressButton(ctx context.Context, msg *tg.Message, keyword string) (*tg.Message, error) {
|
||||
if msg == nil {
|
||||
return nil, errors.New("message cannot be nil")
|
||||
}
|
||||
|
||||
if err := c.limiter.Wait(ctx); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
normalizedKeyword := strings.ToLower(keyword)
|
||||
|
||||
var updated *tg.Message
|
||||
err := c.withPeer(ctx, func(runCtx context.Context, api *tg.Client, sender *message.Sender, peer tg.InputPeerClass, botID int64) error {
|
||||
button, err := findCallbackButton(msg, normalizedKeyword)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
req := &tg.MessagesGetBotCallbackAnswerRequest{
|
||||
Peer: peer,
|
||||
MsgID: msg.ID,
|
||||
}
|
||||
if len(button.Data) > 0 {
|
||||
req.SetData(button.Data)
|
||||
}
|
||||
|
||||
invoke := func() error {
|
||||
_, err := api.MessagesGetBotCallbackAnswer(runCtx, req)
|
||||
return err
|
||||
}
|
||||
|
||||
if err := invoke(); err != nil {
|
||||
if wait, ok := telegram.AsFloodWait(err); ok {
|
||||
if err := sleepWithContext(runCtx, wait+time.Second); err != nil {
|
||||
return err
|
||||
}
|
||||
if err := invoke(); err != nil {
|
||||
return fmt.Errorf("callback retry failed: %w", err)
|
||||
}
|
||||
} else {
|
||||
return fmt.Errorf("press callback button: %w", err)
|
||||
}
|
||||
}
|
||||
|
||||
if err := sleepWithContext(runCtx, 1200*time.Millisecond); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
resp, err := api.MessagesGetMessages(runCtx, []tg.InputMessageClass{
|
||||
&tg.InputMessageID{ID: msg.ID},
|
||||
})
|
||||
if err != nil {
|
||||
return fmt.Errorf("fetch updated message: %w", err)
|
||||
}
|
||||
|
||||
refreshed, err := extractMessageByID(resp, msg.ID)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
updated = refreshed
|
||||
return nil
|
||||
})
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return updated, nil
|
||||
}
|
||||
|
||||
func (c *Client) FetchUserMessages(ctx context.Context, identifier string, maxPages *int) (string, error) {
|
||||
id := strings.TrimSpace(identifier)
|
||||
if id == "" {
|
||||
return "", errors.New("identifier cannot be empty")
|
||||
}
|
||||
|
||||
if !strings.HasPrefix(id, "/") {
|
||||
if !strings.HasPrefix(id, "@") && !isNumericIdentifier(id) {
|
||||
id = "@" + id
|
||||
}
|
||||
}
|
||||
|
||||
command := id
|
||||
if !strings.HasPrefix(command, "/") {
|
||||
command = fmt.Sprintf("/user_info %s", id)
|
||||
}
|
||||
|
||||
base, err := c.SendCommandMessage(ctx, command, 20*time.Second)
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
stage, err := c.PressButton(ctx, base, "messages")
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
current, err := c.PressButton(ctx, stage, "all")
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
|
||||
pages := make([]string, 0)
|
||||
seen := map[string]struct{}{}
|
||||
currentPage := 1
|
||||
totalPages := extractTotalPages(current)
|
||||
|
||||
var limit int
|
||||
if maxPages != nil {
|
||||
if *maxPages <= 0 {
|
||||
return "", errors.New("maxPages must be greater than zero")
|
||||
}
|
||||
limit = *maxPages
|
||||
}
|
||||
|
||||
for {
|
||||
text := strings.TrimSpace(current.Message)
|
||||
if text != "" {
|
||||
if _, ok := seen[text]; !ok {
|
||||
header := fmt.Sprintf("第 %d 页", currentPage)
|
||||
if totalPages > 0 {
|
||||
header = fmt.Sprintf("%s/%d", header, totalPages)
|
||||
}
|
||||
entry := strings.Join([]string{header, "", text}, "\n")
|
||||
pages = append(pages, entry)
|
||||
seen[text] = struct{}{}
|
||||
}
|
||||
}
|
||||
|
||||
if limit > 0 && currentPage >= limit {
|
||||
break
|
||||
}
|
||||
|
||||
updated, err := c.PressButton(ctx, current, "➡")
|
||||
if err != nil {
|
||||
break
|
||||
}
|
||||
|
||||
newText := strings.TrimSpace(updated.Message)
|
||||
if _, ok := seen[newText]; ok {
|
||||
break
|
||||
}
|
||||
|
||||
current = updated
|
||||
currentPage++
|
||||
}
|
||||
|
||||
if len(pages) == 0 {
|
||||
return fmt.Sprintf("未找到 %s 的消息记录。", identifier), nil
|
||||
}
|
||||
|
||||
summary := fmt.Sprintf("共收集 %d 页消息", len(pages))
|
||||
if totalPages > 0 {
|
||||
summary = fmt.Sprintf("%s（存在 %d 页）", summary, totalPages)
|
||||
}
|
||||
|
||||
result := append([]string{summary, ""}, pages...)
|
||||
return strings.Join(result, "\n\n"), nil
|
||||
}
|
||||
|
||||
func isNumericIdentifier(value string) bool {
|
||||
for _, r := range value {
|
||||
if r != '+' && (r < '0' || r > '9') {
|
||||
return false
|
||||
}
|
||||
}
|
||||
return value != ""
|
||||
}
|
||||
60
internal/telegram/config.go
Normal file
60
internal/telegram/config.go
Normal file
@@ -0,0 +1,60 @@
|
||||
package telegram
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"time"
|
||||
)
|
||||
|
||||
type ProxyConfig struct {
|
||||
Type string
|
||||
Host string
|
||||
Port int
|
||||
Username string
|
||||
Password string
|
||||
}
|
||||
|
||||
type Config struct {
|
||||
APIID int
|
||||
APIHash string
|
||||
BotUsername string
|
||||
SessionString string
|
||||
SessionStorage string
|
||||
RateLimit int
|
||||
RateLimitWindow time.Duration
|
||||
CacheTTL time.Duration
|
||||
Proxy *ProxyConfig
|
||||
}
|
||||
|
||||
func (c Config) Validate() error {
|
||||
if c.APIID == 0 {
|
||||
return fmt.Errorf("APIID must be provided")
|
||||
}
|
||||
if c.APIHash == "" {
|
||||
return fmt.Errorf("APIHash must be provided")
|
||||
}
|
||||
if c.BotUsername == "" {
|
||||
return fmt.Errorf("BotUsername must be provided")
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func (c Config) RateLimitPerSecond() int {
|
||||
if c.RateLimit <= 0 {
|
||||
return 18
|
||||
}
|
||||
return c.RateLimit
|
||||
}
|
||||
|
||||
func (c Config) RateLimitDuration() time.Duration {
|
||||
if c.RateLimitWindow <= 0 {
|
||||
return time.Second
|
||||
}
|
||||
return c.RateLimitWindow
|
||||
}
|
||||
|
||||
func (c Config) CacheDuration() time.Duration {
|
||||
if c.CacheTTL <= 0 {
|
||||
return time.Hour
|
||||
}
|
||||
return c.CacheTTL
|
||||
}
|
||||
63
internal/telegram/messages.go
Normal file
63
internal/telegram/messages.go
Normal file
@@ -0,0 +1,63 @@
|
||||
package telegram
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
|
||||
"github.com/gotd/td/tg"
|
||||
)
|
||||
|
||||
func extractMessages(resp tg.MessagesMessagesClass) ([]*tg.Message, error) {
|
||||
switch v := resp.(type) {
|
||||
case *tg.MessagesMessages:
|
||||
return filterMessages(v.Messages), nil
|
||||
case *tg.MessagesMessagesSlice:
|
||||
return filterMessages(v.Messages), nil
|
||||
case *tg.MessagesChannelMessages:
|
||||
return filterMessages(v.Messages), nil
|
||||
case *tg.MessagesMessagesNotModified:
|
||||
return nil, fmt.Errorf("messages not modified")
|
||||
default:
|
||||
return nil, fmt.Errorf("unsupported response type %T", resp)
|
||||
}
|
||||
}
|
||||
|
||||
func filterMessages(values []tg.MessageClass) []*tg.Message {
|
||||
result := make([]*tg.Message, 0, len(values))
|
||||
for _, m := range values {
|
||||
if msg, ok := m.(*tg.Message); ok {
|
||||
result = append(result, msg)
|
||||
}
|
||||
}
|
||||
return result
|
||||
}
|
||||
|
||||
func extractMessageByID(resp tg.MessagesMessagesClass, id int) (*tg.Message, error) {
|
||||
messages, err := extractMessages(resp)
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
for _, msg := range messages {
|
||||
if msg.ID == id {
|
||||
return msg, nil
|
||||
}
|
||||
}
|
||||
return nil, fmt.Errorf("message %d not found", id)
|
||||
}
|
||||
|
||||
func isFromBot(msg *tg.Message, botID int64) bool {
|
||||
if msg == nil || msg.Out {
|
||||
return false
|
||||
}
|
||||
|
||||
if peer, ok := msg.GetPeerID().(*tg.PeerUser); ok && peer.UserID == botID {
|
||||
return true
|
||||
}
|
||||
|
||||
if fromClass, ok := msg.GetFromID(); ok {
|
||||
if from, ok := fromClass.(*tg.PeerUser); ok && from.UserID == botID {
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
||||
45
internal/telegram/normalize.go
Normal file
45
internal/telegram/normalize.go
Normal file
@@ -0,0 +1,45 @@
|
||||
package telegram
|
||||
|
||||
import "strings"
|
||||
|
||||
var buttonTextTranslations = map[rune]rune{
|
||||
'ƒ': 'f',
|
||||
'Μ': 'M',
|
||||
'τ': 't',
|
||||
'ѕ': 's',
|
||||
'η': 'n',
|
||||
'Ғ': 'F',
|
||||
'α': 'a',
|
||||
'ο': 'o',
|
||||
'ᴜ': 'u',
|
||||
'о': 'o',
|
||||
'е': 'e',
|
||||
'с': 'c',
|
||||
'℮': 'e',
|
||||
'Τ': 'T',
|
||||
'ρ': 'p',
|
||||
'Δ': 'D',
|
||||
'χ': 'x',
|
||||
'β': 'b',
|
||||
'λ': 'l',
|
||||
'γ': 'y',
|
||||
'Ν': 'N',
|
||||
'μ': 'm',
|
||||
'ψ': 'y',
|
||||
'Α': 'A',
|
||||
'Ρ': 'P',
|
||||
'С': 'C',
|
||||
'ё': 'e',
|
||||
'ł': 'l',
|
||||
'Ł': 'L',
|
||||
'ց': 'g',
|
||||
}
|
||||
|
||||
func normalizeButtonText(text string) string {
|
||||
return strings.Map(func(r rune) rune {
|
||||
if mapped, ok := buttonTextTranslations[r]; ok {
|
||||
return mapped
|
||||
}
|
||||
return r
|
||||
}, text)
|
||||
}
|
||||
22
internal/telegram/util.go
Normal file
22
internal/telegram/util.go
Normal file
@@ -0,0 +1,22 @@
|
||||
package telegram
|
||||
|
||||
import (
|
||||
"context"
|
||||
"time"
|
||||
)
|
||||
|
||||
func sleepWithContext(ctx context.Context, d time.Duration) error {
|
||||
if d <= 0 {
|
||||
return nil
|
||||
}
|
||||
|
||||
timer := time.NewTimer(d)
|
||||
defer timer.Stop()
|
||||
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
return ctx.Err()
|
||||
case <-timer.C:
|
||||
return nil
|
||||
}
|
||||
}
|
||||
277
internal/transport/server.go
Normal file
277
internal/transport/server.go
Normal file
@@ -0,0 +1,277 @@
|
||||
package transport
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"context"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"log"
|
||||
"net/http"
|
||||
"strings"
|
||||
"sync"
|
||||
"time"
|
||||
|
||||
"funstatmcp/internal/app"
|
||||
)
|
||||
|
||||
type Server struct {
|
||||
app *app.App
|
||||
config app.Config
|
||||
|
||||
subscribersMu sync.Mutex
|
||||
subscribers map[int]chan []byte
|
||||
nextID int
|
||||
}
|
||||
|
||||
func NewServer(appInstance *app.App, cfg app.Config) *Server {
|
||||
return &Server{
|
||||
app: appInstance,
|
||||
config: cfg,
|
||||
subscribers: make(map[int]chan []byte),
|
||||
}
|
||||
}
|
||||
|
||||
func (s *Server) Run(ctx context.Context) error {
|
||||
mux := http.NewServeMux()
|
||||
mux.HandleFunc("/sse", s.handleSSE)
|
||||
mux.HandleFunc("/messages", s.handleMessages)
|
||||
mux.HandleFunc("/health", s.handleHealth)
|
||||
|
||||
server := &http.Server{
|
||||
Addr: fmt.Sprintf("%s:%d", s.config.Host, s.config.Port),
|
||||
Handler: s.corsMiddleware(mux),
|
||||
}
|
||||
|
||||
go func() {
|
||||
<-ctx.Done()
|
||||
shutdownCtx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
||||
defer cancel()
|
||||
if err := server.Shutdown(shutdownCtx); err != nil {
|
||||
log.Printf("HTTP server shutdown error: %v", err)
|
||||
}
|
||||
}()
|
||||
|
||||
log.Printf("Funstat MCP Go server listening on http://%s:%d", s.config.Host, s.config.Port)
|
||||
if err := server.ListenAndServe(); err != nil && err != http.ErrServerClosed {
|
||||
return err
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (s *Server) corsMiddleware(next http.Handler) http.Handler {
|
||||
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Access-Control-Allow-Origin", "*")
|
||||
w.Header().Set("Access-Control-Allow-Headers", "Content-Type, X-MCP-Session-ID")
|
||||
w.Header().Set("Access-Control-Allow-Methods", "POST, GET, OPTIONS")
|
||||
|
||||
if r.Method == http.MethodOptions {
|
||||
w.WriteHeader(http.StatusNoContent)
|
||||
return
|
||||
}
|
||||
|
||||
next.ServeHTTP(w, r)
|
||||
})
|
||||
}
|
||||
|
||||
func (s *Server) handleHealth(w http.ResponseWriter, r *http.Request) {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(http.StatusOK)
|
||||
_ = json.NewEncoder(w).Encode(map[string]any{
|
||||
"status": "ok",
|
||||
"server": "funstat-mcp-go",
|
||||
})
|
||||
}
|
||||
|
||||
func (s *Server) handleSSE(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodGet {
|
||||
w.WriteHeader(http.StatusMethodNotAllowed)
|
||||
return
|
||||
}
|
||||
|
||||
w.Header().Set("Content-Type", "text/event-stream")
|
||||
w.Header().Set("Cache-Control", "no-cache")
|
||||
w.Header().Set("Connection", "keep-alive")
|
||||
|
||||
flusher, ok := w.(http.Flusher)
|
||||
if !ok {
|
||||
http.Error(w, "streaming unsupported", http.StatusInternalServerError)
|
||||
return
|
||||
}
|
||||
|
||||
subscriber := make(chan []byte, 16)
|
||||
id := s.addSubscriber(subscriber)
|
||||
defer s.removeSubscriber(id)
|
||||
|
||||
heartbeatTicker := time.NewTicker(15 * time.Second)
|
||||
defer heartbeatTicker.Stop()
|
||||
|
||||
ctx := r.Context()
|
||||
|
||||
for {
|
||||
select {
|
||||
case <-ctx.Done():
|
||||
return
|
||||
case <-heartbeatTicker.C:
|
||||
fmt.Fprint(w, ": ping\n\n")
|
||||
flusher.Flush()
|
||||
case data := <-subscriber:
|
||||
fmt.Fprintf(w, "event: message\n")
|
||||
fmt.Fprintf(w, "data: %s\n\n", data)
|
||||
flusher.Flush()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
func (s *Server) addSubscriber(ch chan []byte) int {
|
||||
s.subscribersMu.Lock()
|
||||
defer s.subscribersMu.Unlock()
|
||||
id := s.nextID
|
||||
s.nextID++
|
||||
s.subscribers[id] = ch
|
||||
return id
|
||||
}
|
||||
|
||||
func (s *Server) removeSubscriber(id int) {
|
||||
s.subscribersMu.Lock()
|
||||
defer s.subscribersMu.Unlock()
|
||||
delete(s.subscribers, id)
|
||||
}
|
||||
|
||||
func (s *Server) broadcast(payload []byte) {
|
||||
s.subscribersMu.Lock()
|
||||
defer s.subscribersMu.Unlock()
|
||||
for _, ch := range s.subscribers {
|
||||
select {
|
||||
case ch <- payload:
|
||||
default:
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
type jsonRPCRequest struct {
|
||||
JSONRPC string `json:"jsonrpc"`
|
||||
ID any `json:"id"`
|
||||
Method string `json:"method"`
|
||||
Params json.RawMessage `json:"params"`
|
||||
}
|
||||
|
||||
type jsonRPCResponse struct {
|
||||
JSONRPC string `json:"jsonrpc"`
|
||||
ID any `json:"id"`
|
||||
Result any `json:"result,omitempty"`
|
||||
Error *jsonRPCError `json:"error,omitempty"`
|
||||
}
|
||||
|
||||
type jsonRPCError struct {
|
||||
Code int `json:"code"`
|
||||
Message string `json:"message"`
|
||||
}
|
||||
|
||||
func (s *Server) handleMessages(w http.ResponseWriter, r *http.Request) {
|
||||
if r.Method != http.MethodPost {
|
||||
w.WriteHeader(http.StatusMethodNotAllowed)
|
||||
return
|
||||
}
|
||||
|
||||
var request jsonRPCRequest
|
||||
if err := json.NewDecoder(r.Body).Decode(&request); err != nil {
|
||||
writeError(w, http.StatusBadRequest, fmt.Errorf("invalid JSON: %w", err))
|
||||
return
|
||||
}
|
||||
|
||||
response := s.handleRequest(r.Context(), request)
|
||||
|
||||
var buffer bytes.Buffer
|
||||
if err := json.NewEncoder(&buffer).Encode(response); err != nil {
|
||||
writeError(w, http.StatusInternalServerError, err)
|
||||
return
|
||||
}
|
||||
|
||||
payload := bytes.TrimSpace(buffer.Bytes())
|
||||
s.broadcast(payload)
|
||||
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.Write(payload)
|
||||
}
|
||||
|
||||
func (s *Server) handleRequest(ctx context.Context, req jsonRPCRequest) jsonRPCResponse {
|
||||
response := jsonRPCResponse{
|
||||
JSONRPC: "2.0",
|
||||
ID: req.ID,
|
||||
}
|
||||
|
||||
if strings.TrimSpace(req.JSONRPC) != "2.0" {
|
||||
response.Error = &jsonRPCError{Code: -32600, Message: "invalid jsonrpc version"}
|
||||
return response
|
||||
}
|
||||
|
||||
switch req.Method {
|
||||
case "initialize":
|
||||
response.Result = s.initializeResult()
|
||||
case "list_tools":
|
||||
response.Result = map[string]any{
|
||||
"tools": app.ToolDefinitions(),
|
||||
}
|
||||
case "call_tool":
|
||||
var payload struct {
|
||||
Name string `json:"name"`
|
||||
Arguments map[string]any `json:"arguments"`
|
||||
}
|
||||
if err := json.Unmarshal(req.Params, &payload); err != nil {
|
||||
response.Error = &jsonRPCError{Code: -32602, Message: "invalid params"}
|
||||
return response
|
||||
}
|
||||
|
||||
if payload.Arguments == nil {
|
||||
payload.Arguments = make(map[string]any)
|
||||
}
|
||||
|
||||
result, err := s.app.CallTool(ctx, payload.Name, payload.Arguments)
|
||||
if err != nil {
|
||||
response.Result = map[string]any{
|
||||
"content": []map[string]string{
|
||||
{
|
||||
"type": "text",
|
||||
"text": fmt.Sprintf("❌ 错误: %s", err.Error()),
|
||||
},
|
||||
},
|
||||
}
|
||||
return response
|
||||
}
|
||||
|
||||
response.Result = map[string]any{
|
||||
"content": []map[string]string{
|
||||
{
|
||||
"type": "text",
|
||||
"text": result,
|
||||
},
|
||||
},
|
||||
}
|
||||
default:
|
||||
response.Error = &jsonRPCError{Code: -32601, Message: "method not found"}
|
||||
}
|
||||
|
||||
return response
|
||||
}
|
||||
|
||||
func (s *Server) initializeResult() map[string]any {
|
||||
return map[string]any{
|
||||
"protocolVersion": "2025-03-26",
|
||||
"capabilities": map[string]any{
|
||||
"tools": map[string]any{},
|
||||
},
|
||||
"serverInfo": map[string]any{
|
||||
"name": "funstat-mcp-go",
|
||||
"version": "1.0.0",
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func writeError(w http.ResponseWriter, status int, err error) {
|
||||
w.Header().Set("Content-Type", "application/json")
|
||||
w.WriteHeader(status)
|
||||
_ = json.NewEncoder(w).Encode(map[string]any{
|
||||
"error": err.Error(),
|
||||
})
|
||||
}
|
||||
10
scripts/start_server.sh
Executable file
10
scripts/start_server.sh
Executable file
@@ -0,0 +1,10 @@
|
||||
#!/usr/bin/env bash
|
||||
set -euo pipefail
|
||||
|
||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||
PROJECT_ROOT="$(cd "${SCRIPT_DIR}/.." && pwd)"
|
||||
|
||||
cd "$PROJECT_ROOT"
|
||||
|
||||
echo "Starting Funstat MCP Go server..."
|
||||
go run ./cmd/funstat-mcp "$@"
|
||||
Reference in New Issue
Block a user