157 lines
4.8 KiB
Go
157 lines
4.8 KiB
Go
package libs
|
||
|
||
import (
|
||
"bytes"
|
||
"context"
|
||
"encoding/json"
|
||
"errors"
|
||
"fmt"
|
||
"io"
|
||
"net"
|
||
"net/http"
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/zhuy1228/go-mobile-uiautomator/adb"
|
||
)
|
||
|
||
// ---------- 外部接口和类型定义 ----------
|
||
|
||
// AdbDevice 定义了设备连接接口
|
||
type AdbDevice interface {
|
||
// CreateConnection 建立到设备的 TCP 连接
|
||
// network 通常为 "tcp"，port 为设备上服务监听端口
|
||
CreateConnection(network string, port int) (net.Conn, error)
|
||
}
|
||
|
||
// AdbTunnelDevice 通过 ADB 隧道直连设备（与 Python uiautomator2 完全一致）
|
||
// 每次 CreateConnection 会建立一条新的 ADB 隧道，无需 adb forward，无需端口管理
|
||
// 等同于 Python 中 AdbHTTPConnection 继承 HTTPConnection 并重写 connect() 的方案
|
||
type AdbTunnelDevice struct {
|
||
AdbAddr string // ADB 服务器地址
|
||
Serial string // 设备序列号
|
||
}
|
||
|
||
// CreateConnection 建立到设备指定端口的 ADB 隧道连接
|
||
func (d *AdbTunnelDevice) CreateConnection(network string, port int) (net.Conn, error) {
|
||
return adb.CreateTunnel(d.AdbAddr, d.Serial, port)
|
||
}
|
||
|
||
// HTTPResponse 封装 HTTP 响应数据
|
||
type HTTPResponse struct {
|
||
Content []byte // 响应体内容
|
||
Status int // HTTP 状态码
|
||
Reason string // 状态描述
|
||
}
|
||
|
||
// JSON 将响应体解析为指定的结构体
|
||
func (r *HTTPResponse) JSON(v interface{}) error {
|
||
return json.Unmarshal(r.Content, v)
|
||
}
|
||
|
||
// Text 返回响应体的文本内容
|
||
func (r *HTTPResponse) Text() string {
|
||
return string(r.Content)
|
||
}
|
||
|
||
// ---------- 自定义错误类型 ----------
|
||
|
||
var (
|
||
// ErrHTTPTimeout 表示 HTTP 请求超时
|
||
ErrHTTPTimeout = errors.New("HTTP 请求超时")
|
||
// ErrHTTPFailed 表示 HTTP 请求失败
|
||
ErrHTTPFailed = errors.New("HTTP 请求失败")
|
||
)
|
||
|
||
// ---------- HttpRequest：通过标准 HTTP 客户端发送请求 ----------
|
||
|
||
// HttpRequest 向设备端 UIAutomator2 服务发送 HTTP 请求
|
||
// 使用 Go 标准 http.Client + 自定义 ADB 隧道 Dialer，与 Python uiautomator2 完全一致
|
||
//
|
||
// 工作原理（与 Python 的 AdbHTTPConnection 等价）：
|
||
// 1. http.Transport 的 DialContext 会为每个请求建立一条新的 ADB 隧道
|
||
// 2. 标准 http.Client 在这条隧道上发送完整的 HTTP/1.1 请求
|
||
// 3. 请求完成后隧道自动关闭，无需额外清理
|
||
func HttpRequest(ctx context.Context, dev AdbDevice, devicePort int, method, path string, data map[string]interface{}, timeoutSecs float64, printRequest bool) (*HTTPResponse, error) {
|
||
// 默认超时 10 秒
|
||
if timeoutSecs <= 0 {
|
||
timeoutSecs = 10.0
|
||
}
|
||
|
||
// 调试模式：打印 curl 风格的请求信息
|
||
if printRequest {
|
||
now := time.Now().Format("15:04:05.000")
|
||
url := fmt.Sprintf("http://<adb-tunnel>:%d%s", devicePort, path)
|
||
if data != nil {
|
||
b, _ := json.Marshal(data)
|
||
fmt.Printf("# HTTP 超时=%.3f\n%s $ curl -X %s %s -d '%s'\n", timeoutSecs, now, method, url, string(b))
|
||
} else {
|
||
fmt.Printf("# HTTP 超时=%.3f\n%s $ curl -X %s %s\n", timeoutSecs, now, method, url)
|
||
}
|
||
}
|
||
|
||
// 构造 HTTP 请求体
|
||
// URL 中的 host:port 会被自定义 DialContext 忽略，实际连接通过 ADB 隧道
|
||
url := fmt.Sprintf("http://127.0.0.1:%d%s", devicePort, path)
|
||
var body io.Reader
|
||
if data != nil {
|
||
b, err := json.Marshal(data)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("JSON 编码失败: %w", err)
|
||
}
|
||
body = bytes.NewReader(b)
|
||
}
|
||
|
||
req, err := http.NewRequestWithContext(ctx, method, url, body)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("创建 HTTP 请求失败: %w", err)
|
||
}
|
||
req.Header.Set("Content-Type", "application/json; charset=utf-8")
|
||
|
||
// 核心：自定义 Transport，用 ADB 隧道替代普通 TCP 连接
|
||
// 这与 Python 中 AdbHTTPConnection.connect() 重写 self.sock 的做法完全等价
|
||
transport := &http.Transport{
|
||
DialContext: func(ctx context.Context, network, addr string) (net.Conn, error) {
|
||
return dev.CreateConnection("tcp", devicePort)
|
||
},
|
||
DisableKeepAlives: true, // 每次请求独立隧道，与 Python 行为一致
|
||
}
|
||
client := &http.Client{
|
||
Transport: transport,
|
||
Timeout: time.Duration(timeoutSecs * float64(time.Second)),
|
||
}
|
||
|
||
resp, err := client.Do(req)
|
||
if err != nil {
|
||
var netErr net.Error
|
||
if errors.As(err, &netErr) && netErr.Timeout() {
|
||
return nil, fmt.Errorf("%w: %v", ErrHTTPTimeout, err)
|
||
}
|
||
return nil, fmt.Errorf("%w: %v", ErrHTTPFailed, err)
|
||
}
|
||
defer resp.Body.Close()
|
||
|
||
// 读取响应体
|
||
content, err := io.ReadAll(resp.Body)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("读取响应体失败: %w", err)
|
||
}
|
||
|
||
if resp.StatusCode != http.StatusOK {
|
||
return nil, fmt.Errorf("HTTP 请求失败: %d %s", resp.StatusCode, resp.Status)
|
||
}
|
||
|
||
response := &HTTPResponse{
|
||
Content: content,
|
||
Status: resp.StatusCode,
|
||
Reason: resp.Status,
|
||
}
|
||
|
||
if printRequest {
|
||
now := time.Now().Format("15:04:05.000")
|
||
fmt.Printf("%s 响应 >>>\n%s\n<<< 结束\n\n", now, strings.TrimRight(response.Text(), "\n"))
|
||
}
|
||
|
||
return response, nil
|
||
}
|