import time
import win32api
import win32con
import win32gui
import pywintypes

# 定义回复消息的格式
reply_msg = "您好，我是机器人小助手。您发送了 %s,我可以为您做 %s。"

# 注册窗口过程函数
def on_message(hwnd, msg, wparam, lparam):
    # 如果收到的消息是文本消息类型，则进行自动回复处理
    if msg == win32con.WM_TEXT:
        text = win32gui.GetWindowText(hwnd)
        # 判断是否收到了用户发送的消息
        if text.lower() == "你好":
            # 回复一条默认消息
            reply_msg = reply_msg % (text, "帮助你解决问题")
            # 在屏幕上显示回复消息
            win32gui.SendMessage(hwnd, win32con.WM_SETTEXT, 0, reply_msg)
            # 等待1秒钟后弹出提示框告知用户已经收到回复
            time.sleep(1)
            win32gui.SendMessage(hwnd, win32con.WM_SETTEXT, 0, "我已经收到你的信息啦！")
            win32gui.PumpMessages()
        else:
            # 如果不是指定的消息类型，则直接返回
            return True
    else:
        return True

# 注册消息循环处理函数
def on_loop():
    # 获取当前活动窗口句柄
    hwnd = win32gui.GetForegroundWindow()
    # 进入消息循环处理状态
    win32gui.PumpMessages()
    # 每隔5秒钟检查一次是否有新的消息到达
    while True:
        # 如果有新的消息到达，则进行自动回复处理
        if win32gui.PeekMessageW(None, None, None, None, win32con.PM_REMOVE):
            win32gui.TranslateMessage(win32con.WM_COMMAND)
            win32gui.DispatchMessageW(win32con.WM_COMMAND)
        else:
            # 如果没有新的消息到达，则休眠一段时间后再进行下一次检查
            time.sleep(5)