iOSAI/Module/FlaskService.py

import json
import os
import socket
import threading
import time
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
from Utils.JsonUtils import JsonUtils

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

listData = []
listLock = threading.Lock()

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:
                    conn, addr = s.accept()
                except Exception as e:
                    LogManager.error(f"[ERROR] accept 失败: {e}")
                    continue

                # 独立线程处理单条连接，避免单客户端异常拖垮监听线程
                threading.Thread(target=_handle_conn, args=(conn, addr), daemon=True).start()
            # 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}")

def _handle_conn(conn: socket.socket, addr):
    try:
        with conn:
            # 1. 循环收包直到拿到完整 JSON
            buffer = ""
            while True:
                data = conn.recv(1024)
                if not data:        # 对端关闭
                    break
                buffer += data.decode('utf-8', errors='ignore')
                # 2. 尝试切出完整 JSON（简单按行，也可按长度头、分隔符）
                while True:
                    line, sep, buffer = buffer.partition('\n')
                    if not sep:     # 没找到完整行
                        break
                    line = line.strip()
                    if not line:    # 空行跳过
                        continue
                    try:
                        obj = json.loads(line)
                    except json.JSONDecodeError as e:
                        LogManager.warning(f"[WARN] 非法 JSON 丢弃: {line[:100]}  {e}")
                        continue
                    # 3. 收到合法数据，塞进队列
                    dataQueue.put(obj)
                    LogManager.info(f"[INFO] 收到合法消息: {obj}")
    except Exception as e:
        LogManager.error(f"[ERROR] 连接处理异常: {e}")


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


# 获取设备列表
@app.route('/deviceList', methods=['GET'])
def deviceList():
    try:
        with listLock:
            # 1. 消费完队列
            while not dataQueue.empty():
                obj = dataQueue.get()
                if obj["type"] == 1:
                    # 上线：先踢掉同 deviceId 的旧记录（端口可能变）
                    listData[:] = [d for d in listData if d.get("deviceId") != obj.get("deviceId")]
                    listData.append(obj)
                else:
                    # 下线：只要同 deviceId 就删，不管端口
                    listData[:] = [d for d in listData if d.get("deviceId") != obj.get("deviceId")]

            # 2. 兜底：只保留在线
            listData[:] = [d for d in listData if d.get('type') == 1]

            return ResultData(data=listData.copy()).toJson()
    except Exception as 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, massage = ThreadManager.stop(udid)
    return ResultData(code=code, data="", massage=massage).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, massage="请求数据为空").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:

        LogManager.error(f"获取屏幕翻译出现错误：{e}", "获取屏幕翻译")

        data = [
            {
                'type': 'massage',
                'dir': 'in',
                'text': '当前页面无法获取聊天记录，请在tiktok聊天页面进行获取！！！'
            },
            {
                'type': 'massage',
                '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, massage="解析失败").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="", massage="日志上传失败").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()


@app.route("/aiConfig", methods=['POST'])
def aiConfig():
    data = request.get_json()
    agentName = data.get("agentName")
    guildName = data.get("guildName")
    contactTool = data.get("contactTool")
    contact = data.get("contact")

    dict = {
        "agentName": agentName,
        "guildName": guildName,
        "contactTool": contactTool,
        "contact": contact
    }

    JsonUtils.write_json("aiConfig", dict)
    return ResultData(data="").toJson()


# 查询主播聊天发送的最后一条信息
@app.route("/select_last_message", methods=['GET'])
def select_last_message():
    data = JsonUtils.query_all_json_items()

    return ResultData(data=data).toJson()


@app.route("/update_last_message", methods=['POST'])
def update_last_message():
    data = request.get_json()  # 解析 JSON
    sender = data.get("sender")
    udid = data.get("device")
    text = data.get("text")

    updated_count = JsonUtils.update_json_items(
        match={"sender": sender, "text": text},  # 匹配条件
        patch={"status": 1},  # 修改内容
        filename="log/last_message.json",  # 要修改的文件
        multi=False  # 只改第一条匹配的
    )
    if updated_count > 0:
        return ResultData(data=updated_count, massage="修改成功").toJson()
    return ResultData(data=updated_count, massage="修改失败").toJson()


@app.route("/delete_last_message", methods=['POST'])
def delete_last_message():
    data = request.get_json()  # 解析 JSON
    sender = data.get("sender")
    udid = data.get("device")
    text = data.get("text")

    updated_count = JsonUtils.delete_json_items(
        match={"sender": sender, "text": text},  # 匹配条件
        filename="log/last_message.json",  # 要修改的文件
        multi=False  # 只改第一条匹配的
    )
    if updated_count > 0:
        return ResultData(data=updated_count, massage="修改成功").toJson()
    return ResultData(data=updated_count, massage="修改失败").toJson()


# @app.route("/killWda", methods=['POST'])
# def killWda():
#     data = request.get_json()  # 解析 JSON
#     udid = data.get("device")
#     print(udid)
#
#
#     AiUtils.kill_wda(udid)
#     time.sleep(10)
#     AiUtils.launch_wda(udid)
#
#     return ResultData(data="", msg="WDA重新启动").toJson()


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