iOSAI/Module/FlaskService.py

import json
import os
import socket
import threading
from pathlib import Path
from queue import Queue
from typing import Any, Dict

from Entity.AnchorModel import AnchorModel
from Utils.AiUtils import AiUtils
from Utils.LogManager import LogManager
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, prologueList, addModelToAnchorList, removeModelFromAnchorList
import Entity.Variables as ev

app = Flask(__name__)
CORS(app)
app.config['JSON_AS_ASCII'] = False  # Flask jsonify 不转义中文/emoji
app.config['JSONIFY_MIMETYPE'] = "application/json; charset=utf-8"

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()

# 传递token，暂时用不到了
# @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():
    try:
        data: Dict[str, Any] = request.get_json()
        # 设备列表
        idList = data.get("deviceList", [])
        # 主播列表
        acList = data.get("anchorList", [])
        AiUtils.save_aclist_flat_append(acList)

        # 是否需要回复
        needReply = data.get("needReply", True)
        # 获取打招呼数据
        ev.prologueList = data.get("prologueList", [])

        # 添加主播数据
        addModelToAnchorList(acList)
        # 启动线程，执行脚本
        for udid in idList:
            manager = ScriptManager()
            event = threading.Event()
            # 启动脚本
            thread = threading.Thread(target=manager.safe_greetNewFollowers, args=(udid, needReply, event))
            thread.start()
            # 添加到线程管理
            ThreadManager.add(udid, thread, event)
        return ResultData(data="").toJson()
    except Exception as e:
        LogManager.error(e)

# 获取私信数据
@app.route("/getPrologueList", methods=['GET'])
def getPrologueList():
    import Entity.Variables as Variables
    return ResultData(data=Variables.prologueList).toJson()

# 添加临时数据
# 批量追加主播到 JSON 文件
@app.route("/addTempAnchorData", methods=['POST'])
def addTempAnchorData():
    """
    请求体支持：
    - 单个对象：{"anchorId": "xxx", "country": "CN"}
    - 对象数组：[{"anchorId": "xxx", "country": "CN"}, {"anchorId": "yyy", "country": "US"}]
    """
    data = request.get_json()
    if not data:
        return ResultData(code=400, msg="请求数据为空").toJson()
    # 追加到 JSON 文件
    AiUtils.save_aclist_flat_append(data, "log/acList.json")
    return ResultData(data="ok").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("/setLoginInfo", methods=['POST'])
def upLoadLogLogs():
    data = request.get_json()  # 解析 JSON
    token = data.get("token")
    userId = data.get("userId")
    tenantId = data.get("tenantId")
    ok = LogManager.upload_all_logs("http://47.79.98.113:8101/api/log/upload", token, userId, tenantId)
    if ok:
        return ResultData(data="日志上传成功").toJson()
    else:
        return ResultData(data="", msg="日志上传失败").toJson()

# 获取当前的主播列表数据
@app.route("/anchorList", methods=['POST'])
def queryAnchorList():
    file_path = "log/acList.json"
    data = []
    if Path(file_path).exists():
        try:
            with open(file_path, "r", encoding="utf-8") as f:
                data = json.load(f)
        except Exception as e:
            LogManager.error(f"[anchorList] 读取失败: {e}")
            data = []
    return ResultData(data=data).toJson()


# 删除主播
@app.route("/deleteAnchorWithIds", methods=['POST'])
def deleteAnchorWithIds():
    ls: list[dict] = request.get_json()  # [{"anchorId": "xxx"}, ...]
    ids = [d.get("anchorId") for d in ls if d.get("anchorId")]
    deleted = AiUtils.delete_anchors_by_ids(ids)
    return ResultData(data={"deleted": deleted}).toJson()



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