添加 auto_order.go
This commit is contained in:
432
auto_order.go
Normal file
432
auto_order.go
Normal file
@@ -0,0 +1,432 @@
|
|||||||
|
// auto_order.go
|
||||||
|
package main
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"log"
|
||||||
|
"math/rand"
|
||||||
|
"net/http"
|
||||||
|
"os"
|
||||||
|
"strconv"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/playwright-community/playwright-go"
|
||||||
|
)
|
||||||
|
|
||||||
|
type OrderConfig struct {
|
||||||
|
URL string `json:"url"` // 商品页 URL
|
||||||
|
Cookies []Cookie `json:"cookies"` // 登录 Cookie
|
||||||
|
MaxPrice float64 `json:"max_price"` // 最高价格
|
||||||
|
TargetLocation string `json:"target_location"` // 目标机房（如 "LA"）
|
||||||
|
StockKeyword string `json:"stock_keyword"` // 缺货关键字（如 "Out of Stock"）
|
||||||
|
Timeout int `json:"timeout"` // 超时时间（秒）
|
||||||
|
}
|
||||||
|
|
||||||
|
type Cookie struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
Value string `json:"value"`
|
||||||
|
Domain string `json:"domain"`
|
||||||
|
Path string `json:"path"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type OrderResult struct {
|
||||||
|
Success bool `json:"success"`
|
||||||
|
Message string `json:"message"`
|
||||||
|
OrderID string `json:"order_id"`
|
||||||
|
Price string `json:"price"`
|
||||||
|
Location string `json:"location"`
|
||||||
|
Screenshot string `json:"screenshot"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
// 从命令行参数或配置文件读取
|
||||||
|
configFile := "order_config.json"
|
||||||
|
if len(os.Args) > 1 {
|
||||||
|
configFile = os.Args[1]
|
||||||
|
}
|
||||||
|
|
||||||
|
config, err := loadConfig(configFile)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal("加载配置失败:", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
result := executeOrder(config)
|
||||||
|
|
||||||
|
// 输出结果
|
||||||
|
output, _ := json.MarshalIndent(result, "", " ")
|
||||||
|
fmt.Println(string(output))
|
||||||
|
|
||||||
|
// 保存结果到文件
|
||||||
|
os.WriteFile("order_result.json", output, 0644)
|
||||||
|
|
||||||
|
if !result.Success {
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func loadConfig(filename string) (*OrderConfig, error) {
|
||||||
|
data, err := os.ReadFile(filename)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
var config OrderConfig
|
||||||
|
if err := json.Unmarshal(data, &config); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// 设置默认值
|
||||||
|
if config.Timeout == 0 {
|
||||||
|
config.Timeout = 60
|
||||||
|
}
|
||||||
|
if config.StockKeyword == "" {
|
||||||
|
config.StockKeyword = "Out of Stock"
|
||||||
|
}
|
||||||
|
|
||||||
|
return &config, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func executeOrder(config *OrderConfig) *OrderResult {
|
||||||
|
result := &OrderResult{Success: false}
|
||||||
|
|
||||||
|
// 启动 Playwright
|
||||||
|
pw, err := playwright.Run()
|
||||||
|
if err != nil {
|
||||||
|
result.Message = fmt.Sprintf("启动 Playwright 失败: %v", err)
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
defer pw.Stop()
|
||||||
|
|
||||||
|
// 启动浏览器（使用隐身模式 + 反检测）
|
||||||
|
browser, err := pw.Chromium.Launch(playwright.BrowserTypeLaunchOptions{
|
||||||
|
Headless: playwright.Bool(true),
|
||||||
|
Args: []string{
|
||||||
|
"--disable-blink-features=AutomationControlled",
|
||||||
|
"--disable-dev-shm-usage",
|
||||||
|
"--no-sandbox",
|
||||||
|
},
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
result.Message = fmt.Sprintf("启动浏览器失败: %v", err)
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
defer browser.Close()
|
||||||
|
|
||||||
|
// 创建上下文
|
||||||
|
context, err := browser.NewContext(playwright.BrowserNewContextOptions{
|
||||||
|
UserAgent: playwright.String(randomUserAgent()),
|
||||||
|
Viewport: &playwright.Size{
|
||||||
|
Width: 1920,
|
||||||
|
Height: 1080,
|
||||||
|
},
|
||||||
|
Locale: playwright.String("en-US"),
|
||||||
|
TimezoneId: playwright.String("America/New_York"),
|
||||||
|
JavaScriptEnabled: playwright.Bool(true),
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
result.Message = fmt.Sprintf("创建上下文失败: %v", err)
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
defer context.Close()
|
||||||
|
|
||||||
|
// 注入 Cookie
|
||||||
|
for _, c := range config.Cookies {
|
||||||
|
context.AddCookies(playwright.BrowserContextAddCookiesOptionsCookies{
|
||||||
|
Name: playwright.String(c.Name),
|
||||||
|
Value: playwright.String(c.Value),
|
||||||
|
Domain: playwright.String(c.Domain),
|
||||||
|
Path: playwright.String(c.Path),
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
// 创建页面
|
||||||
|
page, err := context.NewPage()
|
||||||
|
if err != nil {
|
||||||
|
result.Message = fmt.Sprintf("创建页面失败: %v", err)
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
|
// 注入反检测脚本
|
||||||
|
page.AddInitScript(playwright.Script{
|
||||||
|
Content: playwright.String(`
|
||||||
|
Object.defineProperty(navigator, 'webdriver', {get: () => undefined});
|
||||||
|
window.chrome = {runtime: {}};
|
||||||
|
`),
|
||||||
|
})
|
||||||
|
|
||||||
|
// 设置超时
|
||||||
|
page.SetDefaultTimeout(float64(config.Timeout * 1000))
|
||||||
|
|
||||||
|
// 执行下单流程
|
||||||
|
return processOrder(page, config)
|
||||||
|
}
|
||||||
|
|
||||||
|
func processOrder(page playwright.Page, config *OrderConfig) *OrderResult {
|
||||||
|
result := &OrderResult{Success: false}
|
||||||
|
|
||||||
|
log.Println("步骤 1: 访问商品页...")
|
||||||
|
if _, err := page.Goto(config.URL, playwright.PageGotoOptions{
|
||||||
|
WaitUntil: playwright.WaitUntilStateNetworkidle,
|
||||||
|
}); err != nil {
|
||||||
|
result.Message = fmt.Sprintf("访问失败: %v", err)
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
|
// 随机等待（模拟人类行为）
|
||||||
|
randomSleep(1000, 3000)
|
||||||
|
|
||||||
|
log.Println("步骤 2: 检查库存...")
|
||||||
|
outOfStockCount, _ := page.Locator(fmt.Sprintf("text=/%s/i", config.StockKeyword)).Count()
|
||||||
|
if outOfStockCount > 0 {
|
||||||
|
result.Message = "商品无货"
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Println("步骤 3: 检查价格...")
|
||||||
|
priceText := ""
|
||||||
|
priceSelectors := []string{
|
||||||
|
".price", ".product-price", "[class*='price']",
|
||||||
|
"span:has-text('$')", "div:has-text('$')",
|
||||||
|
}
|
||||||
|
for _, selector := range priceSelectors {
|
||||||
|
if text, err := page.Locator(selector).First().InnerText(); err == nil {
|
||||||
|
priceText = text
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if priceText != "" {
|
||||||
|
price := extractPrice(priceText)
|
||||||
|
result.Price = fmt.Sprintf("$%.2f", price)
|
||||||
|
|
||||||
|
if config.MaxPrice > 0 && price > config.MaxPrice {
|
||||||
|
result.Message = fmt.Sprintf("价格超出预算: $%.2f > $%.2f", price, config.MaxPrice)
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
log.Printf("价格: $%.2f ✓", price)
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Println("步骤 4: 检查机房位置...")
|
||||||
|
if config.TargetLocation != "" {
|
||||||
|
locationText := ""
|
||||||
|
locationSelectors := []string{
|
||||||
|
".location", ".datacenter", "[class*='location']",
|
||||||
|
"text=/Los Angeles/i", "text=/LA/i",
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, selector := range locationSelectors {
|
||||||
|
if text, err := page.Locator(selector).First().InnerText(); err == nil {
|
||||||
|
locationText = text
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
result.Location = locationText
|
||||||
|
|
||||||
|
if locationText != "" && !strings.Contains(strings.ToLower(locationText), strings.ToLower(config.TargetLocation)) {
|
||||||
|
result.Message = fmt.Sprintf("机房不匹配: %s (需要: %s)", locationText, config.TargetLocation)
|
||||||
|
|
||||||
|
// 截图保存（用于调试）
|
||||||
|
screenshotPath := fmt.Sprintf("wrong_location_%d.png", time.Now().Unix())
|
||||||
|
page.Screenshot(playwright.PageScreenshotOptions{
|
||||||
|
Path: playwright.String(screenshotPath),
|
||||||
|
FullPage: playwright.Bool(true),
|
||||||
|
})
|
||||||
|
result.Screenshot = screenshotPath
|
||||||
|
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
log.Printf("机房: %s ✓", locationText)
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Println("步骤 5: 点击购买按钮...")
|
||||||
|
randomSleep(500, 1500)
|
||||||
|
|
||||||
|
orderButtonSelectors := []string{
|
||||||
|
"button:has-text('Order Now')",
|
||||||
|
"button:has-text('Add to Cart')",
|
||||||
|
"a:has-text('Order Now')",
|
||||||
|
".order-button",
|
||||||
|
"#order-button",
|
||||||
|
}
|
||||||
|
|
||||||
|
clicked := false
|
||||||
|
for _, selector := range orderButtonSelectors {
|
||||||
|
if err := page.Locator(selector).First().Click(); err == nil {
|
||||||
|
clicked = true
|
||||||
|
log.Printf("点击按钮: %s", selector)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if !clicked {
|
||||||
|
result.Message = "未找到购买按钮"
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
|
// 等待页面跳转
|
||||||
|
randomSleep(2000, 4000)
|
||||||
|
|
||||||
|
log.Println("步骤 6: 等待购物车/结账页...")
|
||||||
|
cartURLPatterns := []string{"**/cart**", "**/checkout**", "**/order**"}
|
||||||
|
|
||||||
|
currentURL := page.URL()
|
||||||
|
isCartPage := false
|
||||||
|
for _, pattern := range cartURLPatterns {
|
||||||
|
if strings.Contains(currentURL, strings.ReplaceAll(pattern, "**", "")) {
|
||||||
|
isCartPage = true
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if !isCartPage {
|
||||||
|
// 尝试等待 URL 变化
|
||||||
|
time.Sleep(3 * time.Second)
|
||||||
|
currentURL = page.URL()
|
||||||
|
log.Printf("当前页面: %s", currentURL)
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Println("步骤 7: 确认订单（不支付）...")
|
||||||
|
|
||||||
|
// 寻找"确认订单"按钮（但不是"支付"按钮）
|
||||||
|
confirmButtonSelectors := []string{
|
||||||
|
"button:has-text('Confirm Order')",
|
||||||
|
"button:has-text('Place Order')",
|
||||||
|
"button:has-text('Complete Order'):not(:has-text('Pay'))",
|
||||||
|
".btn-confirm",
|
||||||
|
}
|
||||||
|
|
||||||
|
confirmed := false
|
||||||
|
for _, selector := range confirmButtonSelectors {
|
||||||
|
if count, _ := page.Locator(selector).Count(); count > 0 {
|
||||||
|
// 检查按钮文本，确保不包含 "Pay"
|
||||||
|
btnText, _ := page.Locator(selector).First().InnerText()
|
||||||
|
if !strings.Contains(strings.ToLower(btnText), "pay") {
|
||||||
|
randomSleep(1000, 2000)
|
||||||
|
if err := page.Locator(selector).First().Click(); err == nil {
|
||||||
|
confirmed = true
|
||||||
|
log.Printf("点击确认按钮: %s", selector)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if !confirmed {
|
||||||
|
log.Println("未找到确认按钮，可能已经自动确认或需要手动处理")
|
||||||
|
}
|
||||||
|
|
||||||
|
// 等待订单完成
|
||||||
|
randomSleep(3000, 5000)
|
||||||
|
|
||||||
|
log.Println("步骤 8: 提取订单信息...")
|
||||||
|
|
||||||
|
// 尝试提取订单号
|
||||||
|
orderIDSelectors := []string{
|
||||||
|
".order-id", ".invoice-id", "[class*='order']",
|
||||||
|
"text=/Order ID/i", "text=/Invoice #/i",
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, selector := range orderIDSelectors {
|
||||||
|
if text, err := page.Locator(selector).First().InnerText(); err == nil {
|
||||||
|
result.OrderID = extractOrderID(text)
|
||||||
|
if result.OrderID != "" {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// 从 URL 提取订单号
|
||||||
|
if result.OrderID == "" {
|
||||||
|
currentURL := page.URL()
|
||||||
|
if id := extractOrderIDFromURL(currentURL); id != "" {
|
||||||
|
result.OrderID = id
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
log.Println("步骤 9: 截图保存...")
|
||||||
|
screenshotPath := fmt.Sprintf("order_success_%d.png", time.Now().Unix())
|
||||||
|
page.Screenshot(playwright.PageScreenshotOptions{
|
||||||
|
Path: playwright.String(screenshotPath),
|
||||||
|
FullPage: playwright.Bool(true),
|
||||||
|
})
|
||||||
|
result.Screenshot = screenshotPath
|
||||||
|
|
||||||
|
// 成功
|
||||||
|
result.Success = true
|
||||||
|
result.Message = "下单成功（待支付）"
|
||||||
|
|
||||||
|
if result.OrderID != "" {
|
||||||
|
log.Printf("✅ 订单号: %s", result.OrderID)
|
||||||
|
} else {
|
||||||
|
log.Println("⚠️ 未能提取订单号，请查看截图")
|
||||||
|
}
|
||||||
|
|
||||||
|
return result
|
||||||
|
}
|
||||||
|
|
||||||
|
// === 辅助函数 ===
|
||||||
|
|
||||||
|
func randomUserAgent() string {
|
||||||
|
agents := []string{
|
||||||
|
"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
|
||||||
|
"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/119.0.0.0 Safari/537.36",
|
||||||
|
"Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
|
||||||
|
"Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/120.0.0.0 Safari/537.36",
|
||||||
|
}
|
||||||
|
return agents[rand.Intn(len(agents))]
|
||||||
|
}
|
||||||
|
|
||||||
|
func randomSleep(minMs, maxMs int) {
|
||||||
|
duration := minMs + rand.Intn(maxMs-minMs)
|
||||||
|
time.Sleep(time.Duration(duration) * time.Millisecond)
|
||||||
|
}
|
||||||
|
|
||||||
|
func extractPrice(text string) float64 {
|
||||||
|
// 提取价格：$15.60 或 15.60
|
||||||
|
text = strings.ReplaceAll(text, "$", "")
|
||||||
|
text = strings.ReplaceAll(text, ",", "")
|
||||||
|
text = strings.TrimSpace(text)
|
||||||
|
|
||||||
|
price, _ := strconv.ParseFloat(text, 64)
|
||||||
|
return price
|
||||||
|
}
|
||||||
|
|
||||||
|
func extractOrderID(text string) string {
|
||||||
|
// 提取订单号：Order ID: 12345 或 #12345
|
||||||
|
text = strings.TrimSpace(text)
|
||||||
|
parts := strings.Fields(text)
|
||||||
|
|
||||||
|
for _, part := range parts {
|
||||||
|
part = strings.Trim(part, "#:")
|
||||||
|
if len(part) > 3 && len(part) < 20 {
|
||||||
|
// 检查是否全是数字或字母数字组合
|
||||||
|
if isAlphanumeric(part) {
|
||||||
|
return part
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func extractOrderIDFromURL(url string) string {
|
||||||
|
// 从 URL 提取订单号：/invoice/12345 或 /order/12345
|
||||||
|
parts := strings.Split(url, "/")
|
||||||
|
for i, part := range parts {
|
||||||
|
if (part == "invoice" || part == "order") && i+1 < len(parts) {
|
||||||
|
return parts[i+1]
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
func isAlphanumeric(s string) bool {
|
||||||
|
for _, r := range s {
|
||||||
|
if !((r >= 'a' && r <= 'z') || (r >= 'A' && r <= 'Z') || (r >= '0' && r <= '9')) {
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user