main.py

#!/usr/bin/env python3
"""
SmartMoney Tracker - 聪明钱监控工具
追踪鲸鱼地址，大额转账自动提醒
"""

import os
import json
import time
import argparse
import requests
from datetime import datetime
from dotenv import load_dotenv

# 加载支付集成
from skillpay_integration import check_payment_status, SkillPayVerifier

# 加载环境变量
load_dotenv()

# 配置文件路径
CONFIG_FILE = "wallets.json"
HISTORY_FILE = "history.json"

# API Keys
ETHERSCAN_API_KEY = os.getenv("ETHERSCAN_API_KEY", "")
TELEGRAM_BOT_TOKEN = os.getenv("TELEGRAM_BOT_TOKEN", "")
TELEGRAM_CHAT_ID = os.getenv("TELEGRAM_CHAT_ID", "")
MERCHANT_ADDRESS = os.getenv("MERCHANT_ADDRESS", "0xd49536C908fa29c103501ed17b568628b16C8cE2")  # BNB收款地址

def load_wallets():
    """加载监控的钱包列表"""
    if os.path.exists(CONFIG_FILE):
        with open(CONFIG_FILE, 'r') as f:
            return json.load(f)
    return {}

def save_wallets(wallets):
    """保存钱包列表"""
    with open(CONFIG_FILE, 'w') as f:
        json.dump(wallets, f, indent=2)

def add_wallet(address, name, threshold, chain="eth", user_id="default"):
    """添加监控地址"""
    wallets = load_wallets()
    
    # 检查支付状态
    payment_status = check_payment_status(user_id)
    
    # 免费版限制：最多3个地址
    if len(wallets) >= 3 and not payment_status["paid"]:
        print("❌ 免费版最多监控 3 个地址")
        print(f"💳 {payment_status['message']}")
        print(f"\n付款地址：{MERCHANT_ADDRESS}")
        print("付款后请联系客服激活")
        return False
    
    wallets[address] = {
        "name": name,
        "threshold": threshold,
        "chain": chain,
        "added_at": datetime.now().isoformat(),
        "last_checked": 0,
        "user_id": user_id
    }
    save_wallets(wallets)
    
    plan = "付费版" if payment_status["paid"] else "免费版"
    print(f"✅ 已添加监控: {name} ({address})")
    print(f"   阈值: ${threshold:,.2f}")
    print(f"   套餐: {plan}")
    return True

def get_transactions(address, chain="eth"):
    """从 Etherscan 获取交易记录"""
    if chain == "eth":
        base_url = "https://api.etherscan.io/api"
    elif chain == "bsc":
        base_url = "https://api.bscscan.com/api"
    else:
        return []
    
    if not ETHERSCAN_API_KEY:
        print("❌ 请设置 ETHERSCAN_API_KEY")
        return []
    
    params = {
        "module": "account",
        "action": "txlist",
        "address": address,
        "startblock": 0,
        "endblock": 99999999,
        "sort": "desc",
        "apikey": ETHERSCAN_API_KEY
    }
    
    try:
        response = requests.get(base_url, params=params, timeout=10)
        data = response.json()
        if data["status"] == "1":
            return data["result"]
    except Exception as e:
        print(f"❌ 获取交易失败: {e}")
    
    return []

def send_telegram_alert(wallet_name, address, tx, value_usd):
    """发送 Telegram 提醒"""
    if not TELEGRAM_BOT_TOKEN or not TELEGRAM_CHAT_ID:
        print("⚠️ 未配置 Telegram，跳过推送")
        return
    
    message = f"""🐋 *聪明钱监控提醒*

💼 钱包: {wallet_name}
📍 地址: `{address[:6]}...{address[-4:]}`
💰 金额: ${value_usd:,.2f}
🔄 类型: {"转入" if tx.get('to', '').lower() == address.lower() else "转出"}
⏰ 时间: {datetime.now().strftime('%Y-%m-%d %H:%M:%S')}

🔗 [查看交易](https://etherscan.io/tx/{tx['hash']})
"""
    
    url = f"https://api.telegram.org/bot{TELEGRAM_BOT_TOKEN}/sendMessage"
    payload = {
        "chat_id": TELEGRAM_CHAT_ID,
        "text": message,
        "parse_mode": "Markdown",
        "disable_web_page_preview": True
    }
    
    try:
        requests.post(url, json=payload, timeout=5)
        print(f"📱 Telegram 提醒已发送")
    except Exception as e:
        print(f"❌ Telegram 发送失败: {e}")

def get_eth_price():
    """获取 ETH 当前价格"""
    try:
        response = requests.get(
            "https://api.coingecko.com/api/v3/simple/price?ids=ethereum&vs_currencies=usd",
            timeout=5
        )
        data = response.json()
        return data["ethereum"]["usd"]
    except:
        return 3000  # 默认值

def monitor_wallets():
    """监控所有钱包"""
    wallets = load_wallets()
    if not wallets:
        print("❌ 没有监控的钱包，先用 `add` 命令添加")
        return
    
    eth_price = get_eth_price()
    print(f"💰 当前 ETH 价格: ${eth_price:,.2f}")
    print(f"👀 正在监控 {len(wallets)} 个地址...")
    print("-" * 50)
    
    for address, config in wallets.items():
        print(f"\n🔍 检查: {config['name']}")
        
        transactions = get_transactions(address, config.get('chain', 'eth'))
        threshold = config.get('threshold', 10000)
        last_checked = config.get('last_checked', 0)
        
        alert_count = 0
        for tx in transactions[:5]:  # 只检查最近5笔
            tx_time = int(tx.get('timeStamp', 0))
            
            # 跳过已检查的
            if tx_time <= last_checked:
                continue
            
            # 计算金额
            value_eth = int(tx.get('value', 0)) / 1e18
            value_usd = value_eth * eth_price
            
            if value_usd >= threshold:
                print(f"🚨 大额交易 detected: ${value_usd:,.2f}")
                send_telegram_alert(config['name'], address, tx, value_usd)
                alert_count += 1
        
        # 更新最后检查时间
        if transactions:
            config['last_checked'] = int(time.time())
        
        if alert_count == 0:
            print(f"   ✅ 无异常交易")
    
    save_wallets(wallets)

def show_history(address=None):
    """显示历史记录"""
    wallets = load_wallets()
    
    if address:
        if address in wallets:
            config = wallets[address]
            print(f"\n📊 {config['name']} ({address})")
            print(f"   阈值: ${config['threshold']:,.2f}")
            print(f"   添加时间: {config.get('added_at', 'Unknown')}")
        else:
            print("❌ 地址未找到")
    else:
        print(f"\n📊 共监控 {len(wallets)} 个地址:")
        for addr, config in wallets.items():
            print(f"\n   {config['name']}")
            print(f"   地址: {addr}")
            print(f"   阈值: ${config['threshold']:,.2f}")

def main():
    parser = argparse.ArgumentParser(description="SmartMoney Tracker - 聪明钱监控")
    subparsers = parser.add_subparsers(dest='command', help='可用命令')
    
    # add 命令
    add_parser = subparsers.add_parser('add', help='添加监控地址')
    add_parser.add_argument('address', help='钱包地址')
    add_parser.add_argument('--name', required=True, help='钱包名称')
    add_parser.add_argument('--threshold', type=float, default=10000, help='提醒阈值 (USD)')
    add_parser.add_argument('--chain', default='eth', choices=['eth', 'bsc'], help='区块链')
    add_parser.add_argument('--user-id', default='default', help='用户ID（用于支付验证）')
    
    # monitor 命令
    monitor_parser = subparsers.add_parser('monitor', help='开始监控')
    monitor_parser.add_argument('--interval', type=int, default=300, help='检查间隔 (秒)')
    monitor_parser.add_argument('--user-id', default='default', help='用户ID（用于支付验证）')
    
    # history 命令
    history_parser = subparsers.add_parser('history', help='查看历史')
    history_parser.add_argument('--address', help='指定地址')
    
    # status 命令 - 查看支付状态
    status_parser = subparsers.add_parser('status', help='查看支付状态')
    status_parser.add_argument('--user-id', default='default', help='用户ID')
    status_parser.add_argument('--tx-hash', help='交易哈希（用于手动验证支付）')
    status_parser.add_argument('--user-address', help='用户付款地址')
    
    args = parser.parse_args()
    
    if args.command == 'add':
        add_wallet(args.address, args.name, args.threshold, args.chain, args.user_id)
    
    elif args.command == 'monitor':
        print("🚀 SmartMoney Tracker 启动")
        print("按 Ctrl+C 停止\n")
        
        # 付费版检查
        payment_status = check_payment_status(args.user_id)
        if payment_status["paid"]:
            print(f"✅ {payment_status['message']}")
            # 付费用户使用更短的检查间隔
            if args.interval > 60:
                args.interval = 60  # 付费版最少60秒
        else:
            print(f"ℹ️ {payment_status['message']}")
            print(f"\n付款地址：{MERCHANT_ADDRESS}")
            # 免费版强制5分钟间隔
            if args.interval < 300:
                args.interval = 300
                print("⚠️ 免费版检查间隔已调整为300秒")
        
        try:
            while True:
                monitor_wallets()
                print(f"\n⏰ {datetime.now().strftime('%H:%M:%S')} 等待下次检查...")
                time.sleep(args.interval)
        except KeyboardInterrupt:
            print("\n\n👋 已停止监控")
    
    elif args.command == 'history':
        show_history(args.address)
    
    elif args.command == 'status':
        if args.tx_hash and args.user_address:
            # 手动记录支付
            verifier = SkillPayVerifier(MERCHANT_ADDRESS)
            success = verifier.record_payment(
                args.user_id, 
                args.tx_hash, 
                0.01,  # 假设付款 0.01 BNB/USDC
                args.user_address
            )
            if success:
                print(f"✅ 支付已记录！已为用户 {args.user_id} 激活付费版")
            else:
                print("❌ 支付记录失败")
        else:
            # 查看支付状态
            status = check_payment_status(args.user_id)
            print(f"\n用户ID: {args.user_id}")
            print(f"状态: {status['plan']}")
            print(f"消息: {status['message']}")
            if status['expiry']:
                print(f"有效期至: {status['expiry'][:10]}")
    
    else:
        parser.print_help()

if __name__ == '__main__':
    main()