iOSAI/Module/FlaskService.py

import json
import os
import socket
import threading
import warnings
from queue import Queue
from typing import Any, Dict
from Utils.AiUtils import AiUtils
from Utils.LogManager import LogManager
from Utils.Requester import Requester

import tidevice
import wda
from flask import Flask, request
from flask_cors import CORS
from Entity.ResultData import ResultData
from Utils.ControlUtils import ControlUtils
from Utils.ThreadManager import ThreadManager
from script.ScriptManager import ScriptManager
from Entity.Variables import anchorList, addModelToAnchorList

app = Flask(__name__)
CORS(app)

listData = []
dataQueue = Queue()


def start_socket_listener():
    port = int(os.getenv('FLASK_COMM_PORT', 0))
    LogManager.info(f"Received port from environment: {port}")
    print(f"Received port from environment: {port}")
    if port <= 0:
        LogManager.info("未获取到通信端口，跳过Socket监听")
        print("未获取到通信端口，跳过Socket监听")
        return
    try:
        with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
            # 设置端口复用，避免端口被占用时无法绑定
            s.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)

            # 尝试绑定端口
            try:
                s.bind(('127.0.0.1', port))
                print(f"[INFO] Socket successfully bound to port {port}")
                LogManager.info(f"[INFO] Socket successfully bound to port {port}")
            except Exception as bind_error:
                print(f"[ERROR]端口绑定失败: {bind_error}")
                LogManager.info(f"[ERROR]端口绑定失败: {bind_error}")
                return

            # 开始监听
            s.listen()
            LogManager.info(f"[INFO] Socket listener started on port {port}, waiting for connections...")
            print(f"[INFO] Socket listener started on port {port}, waiting for connections...")
            while True:
                try:
                    LogManager.info(f"[INFO] Waiting for a new connection on port {port}...")
                    print(f"[INFO] Waiting for a new connection on port {port}...")
                    conn, addr = s.accept()
                    LogManager.info(f"[INFO] Connection accepted from: {addr}")
                    print(f"[INFO] Connection accepted from: {addr}")

                    raw_data = conn.recv(1024).decode('utf-8').strip()
                    LogManager.info(f"[INFO] Raw data received: {raw_data}")
                    print(f"[INFO] Raw data received: {raw_data}")

                    data = json.loads(raw_data)
                    LogManager.info(f"[INFO] Parsed data: {data}")
                    print(f"[INFO] Parsed data: {data}")
                    dataQueue.put(data)
                except Exception as conn_error:
                    LogManager.error(f"[ERROR]连接处理失败: {conn_error}")
                    print(f"[ERROR]连接处理失败: {conn_error}")
    except Exception as e:
        LogManager.error(f"[ERROR]Socket服务启动失败: {e}")
        print(f"[ERROR]Socket服务启动失败: {e}")


# 在独立线程中启动Socket服务
listener_thread = threading.Thread(target=start_socket_listener, daemon=True)
listener_thread.start()


@app.route('/passToken', methods=['POST'])
def passToken():
    try:
        data = request.get_json()
        token = data['token']
        Requester.requestPrologue(token)
        return ResultData(data="").toJson()
    except Exception as e:
        print(e)
    return ResultData(data="").toJson()


# 获取设备列表
@app.route('/deviceList', methods=['GET'])
def deviceList():
    try:
        while not dataQueue.empty():
            obj = dataQueue.get()
            type = obj["type"]
            if type == 1:
                listData.append(obj)
            else:
                for data in listData:
                    if data.get("deviceId") == obj.get("deviceId") and data.get("screenPort") == obj.get("screenPort"):
                        listData.remove(data)
        return ResultData(data=listData).toJson()
    except Exception as e:
        print(e)
        LogManager.error("获取设备列表失败：", e)
        return ResultData(data=[]).toJson()


# 获取设备应用列表
@app.route('/deviceAppList', methods=['POST'])
def deviceAppList():
    param = request.get_json()
    udid = param["udid"]
    apps = ControlUtils.getDeviceAppList(udid)
    return ResultData(data=apps).toJson()


# 打开指定app
@app.route('/launchApp', methods=['POST'])
def launchApp():
    body = request.get_json()
    udid = body.get("udid")
    bundleId = body.get("bundleId")
    t = tidevice.Device(udid)
    t.app_start(bundleId)
    return ResultData(data="").toJson()


# 回到首页
@app.route('/toHome', methods=['POST'])
def toHome():
    body = request.get_json()
    udid = body.get("udid")
    client = wda.USBClient(udid)
    client.home()
    return ResultData(data="").toJson()


# 点击事件
@app.route('/tapAction', methods=['POST'])
def tapAction():
    body = request.get_json()
    udid = body.get("udid")
    client = wda.USBClient(udid)
    session = client.session()
    session.appium_settings({"snapshotMaxDepth": 0})
    x = body.get("x")
    y = body.get("y")
    session.tap(x, y)
    return ResultData(data="").toJson()


# 拖拽事件
@app.route('/swipeAction', methods=['POST'])
def swipeAction():
    body = request.get_json()
    udid = body.get("udid")
    direction = body.get("direction")
    client = wda.USBClient(udid)
    session = client.session()
    session.appium_settings({"snapshotMaxDepth": 0})

    if direction == 1:
        session.swipe_up()
    elif direction == 2:
        session.swipe_left()
    elif direction == 3:
        session.swipe_down()
    else:
        session.swipe_right()
    return ResultData(data="").toJson()


# 长按事件
@app.route('/longPressAction', methods=['POST'])
def longPressAction():
    body = request.get_json()
    udid = body.get("udid")
    x = body.get("x")
    y = body.get("y")
    client = wda.USBClient(udid)
    session = client.session()
    session.appium_settings({"snapshotMaxDepth": 5})
    session.tap_hold(x, y, 1.0)
    return ResultData(data="").toJson()


# 养号
@app.route('/growAccount', methods=['POST'])
def growAccount():
    body = request.get_json()
    udid = body.get("udid")

    manager = ScriptManager()
    event = threading.Event()
    # 启动脚本
    thread = threading.Thread(target=manager.growAccount, args=(udid, event))
    thread.start()
    # 添加到线程管理
    ThreadManager.add(udid, thread, event)
    return ResultData(data="").toJson()


# 观看直播
@app.route("/watchLiveForGrowth", methods=['POST'])
def watchLiveForGrowth():
    body = request.get_json()
    udid = body.get("udid")
    manager = ScriptManager()
    event = threading.Event()
    thread = threading.Thread(target=manager.watchLiveForGrowth, args=(udid, event))
    thread.start()
    # 添加到线程管理
    ThreadManager.add(udid, thread, event)
    return ResultData(data="").toJson()


# 停止脚本
@app.route("/stopScript", methods=['POST'])
def stopScript():
    body = request.get_json()
    udid = body.get("udid")
    code, msg = ThreadManager.stop(udid)
    return ResultData(code=code, data="", msg=msg).toJson()


# 传递主播数据(关注主播打招呼)
@app.route('/passAnchorData', methods=['POST'])
def passAnchorData():
    data: Dict[str, Any] = request.get_json()
    # 设备列表

    print("接收的数据", data)

    idList = data.get("deviceList", [])
    # 主播列表
    acList = data.get("anchorList", [])
    # 是否需要回复
    needReply = data.get("needReply", True)
    # 添加主播数据
    addModelToAnchorList(acList)
    # 启动线程，执行脚本
    for udid in idList:
        manager = ScriptManager()
        event = threading.Event()
        # 启动脚本
        # thread = threading.Thread(target=manager.greetNewFollowers, args=(udid, needReply, event))
        thread = threading.Thread(target=manager.safe_greetNewFollowers, args=(udid, needReply, event))
        thread.start()
        # 添加到线程管理
        ThreadManager.add(udid, thread, event)
    return ResultData(data="").toJson()


# 添加临时数据
@app.route("/addTempAnchorData", methods=['POST'])
def addTempAnchorData():
    data = request.get_json()
    addModelToAnchorList(data)
    return ResultData(data="").toJson()


# 获取当前屏幕上的聊天信息
@app.route("/getChatTextInfo", methods=['POST'])
def getChatTextInfo():
    data = request.get_json()
    udid = data.get("udid")
    client = wda.USBClient(udid)
    session = client.session()
    xml = session.source()
    try:
        result = AiUtils.extract_messages_from_xml(xml)
        print(result)
        return ResultData(data=result).toJson()
    except Exception as e:
        data = [
            {
                'type': 'msg',
                'dir': 'in',
                'text': '当前页面无法获取聊天记录，请在tiktok聊天页面进行获取！！！'
            },
            {
                'type': 'msg',
                'dir': 'in',
                'text': 'Unable to retrieve chat messages on the current screen. Please navigate to the TikTok chat page and try again!!!'
            }
        ]
        return ResultData(data=data, msg="解析失败").toJson()


# 监控消息
@app.route("/replyMessages", methods=['POST'])
def monitorMessages():
    body = request.get_json()
    udid = body.get("udid")
    manager = ScriptManager()
    event = threading.Event()
    thread = threading.Thread(target=manager.replyMessages, args=(udid, event))
    thread.start()
    # 添加到线程管理
    ThreadManager.add(udid, thread, event)
    return ResultData(data="").toJson()



@app.route("/upLoadLogFile", methods=['POST'])
def upLoadLogFile():
    ok = LogManager.upload_all_logs(
        server_url="http://47.79.98.113:8101/api/log/upload",
        extra_data={"project": "TikTokAuto", "env": "dev"}
    )
    if ok:
        return ResultData(data="日志上传成功").toJson()
    else:
        return ResultData(data="", msg="日志上传失败").toJson()



if __name__ == '__main__':
    app.run("0.0.0.0", port=5000, debug=True, use_reloader=False)