299 lines
9.9 KiB
Python
299 lines
9.9 KiB
Python
|
import os
|
|||
|
|
import time
|
|||
|
|
import re
|
|||
|
|
import logging
|
|||
|
|
from config import CLIENT_LOG_PATH, CHAT_MESSAGES, TRADE_MESSAGES_CONFIG
|
|||
|
|
from message_sender import MessageSender
|
|||
|
|
import threading
|
|||
|
|
import signal
|
|||
|
|
import pystray
|
|||
|
|
from PIL import Image, ImageDraw
|
|||
|
|
import tempfile
|
|||
|
|
|
|||
|
|
# 配置日志
|
|||
|
|
def setup_logging():
|
|||
|
|
log_file = "poe_forwarder.log"
|
|||
|
|
logging.basicConfig(
|
|||
|
|
level=logging.INFO,
|
|||
|
|
format='%(asctime)s - %(levelname)s - %(message)s',
|
|||
|
|
handlers=[
|
|||
|
|
logging.FileHandler(log_file, encoding='utf-8'),
|
|||
|
|
logging.StreamHandler()
|
|||
|
|
]
|
|||
|
|
)
|
|||
|
|
return logging.getLogger(__name__)
|
|||
|
|
|
|||
|
|
def create_icon():
|
|||
|
|
"""创建一个更美观的图标"""
|
|||
|
|
width = 64
|
|||
|
|
height = 64
|
|||
|
|
|
|||
|
|
# 创建新图像，使用RGBA支持透明
|
|||
|
|
image = Image.new('RGBA', (width, height), (0, 0, 0, 0))
|
|||
|
|
dc = ImageDraw.Draw(image)
|
|||
|
|
|
|||
|
|
# 定义颜色
|
|||
|
|
primary_color = (65, 105, 225) # 皇家蓝
|
|||
|
|
secondary_color = (30, 144, 255) # 道奇蓝
|
|||
|
|
|
|||
|
|
# 绘制主圆形
|
|||
|
|
dc.ellipse([4, 4, width-4, height-4], fill=primary_color)
|
|||
|
|
|
|||
|
|
# 绘制内部装饰
|
|||
|
|
# 绘制字母"P"
|
|||
|
|
font_size = 32
|
|||
|
|
try:
|
|||
|
|
from PIL import ImageFont
|
|||
|
|
font = ImageFont.truetype("arial.ttf", font_size)
|
|||
|
|
except:
|
|||
|
|
font = None
|
|||
|
|
|
|||
|
|
text = "P"
|
|||
|
|
text_bbox = dc.textbbox((0, 0), text, font=font)
|
|||
|
|
text_width = text_bbox[2] - text_bbox[0]
|
|||
|
|
text_height = text_bbox[3] - text_bbox[1]
|
|||
|
|
|
|||
|
|
x = (width - text_width) // 2
|
|||
|
|
y = (height - text_height) // 2
|
|||
|
|
|
|||
|
|
dc.text((x, y), text, fill='white', font=font)
|
|||
|
|
|
|||
|
|
return image
|
|||
|
|
|
|||
|
|
class TrayIcon:
|
|||
|
|
def __init__(self, logger):
|
|||
|
|
self.logger = logger
|
|||
|
|
self.running = True
|
|||
|
|
self.icon = None
|
|||
|
|
self.create_icon()
|
|||
|
|
|
|||
|
|
def create_icon(self):
|
|||
|
|
image = create_icon()
|
|||
|
|
menu = (
|
|||
|
|
pystray.MenuItem(
|
|||
|
|
"POE消息转发器",
|
|||
|
|
lambda: None,
|
|||
|
|
enabled=False
|
|||
|
|
),
|
|||
|
|
pystray.Menu.SEPARATOR,
|
|||
|
|
pystray.MenuItem(
|
|||
|
|
"✓ 正在运行",
|
|||
|
|
lambda: None,
|
|||
|
|
enabled=False
|
|||
|
|
),
|
|||
|
|
pystray.MenuItem(
|
|||
|
|
"📝 查看日志",
|
|||
|
|
self.open_log
|
|||
|
|
),
|
|||
|
|
pystray.Menu.SEPARATOR,
|
|||
|
|
pystray.MenuItem(
|
|||
|
|
"⚙ 打开配置",
|
|||
|
|
self.open_config
|
|||
|
|
),
|
|||
|
|
pystray.MenuItem(
|
|||
|
|
"❌ 退出",
|
|||
|
|
self.stop_application
|
|||
|
|
)
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
self.icon = pystray.Icon(
|
|||
|
|
"poe_forwarder",
|
|||
|
|
image,
|
|||
|
|
"POE消息转发器",
|
|||
|
|
menu
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
def open_log(self):
|
|||
|
|
"""打开日志文件"""
|
|||
|
|
try:
|
|||
|
|
os.startfile("poe_forwarder.log")
|
|||
|
|
except Exception as e:
|
|||
|
|
self.logger.error(f"打开日志文件失败: {str(e)}")
|
|||
|
|
|
|||
|
|
def open_config(self):
|
|||
|
|
"""打开配置文件"""
|
|||
|
|
try:
|
|||
|
|
os.startfile("config.json")
|
|||
|
|
except Exception as e:
|
|||
|
|
self.logger.error(f"打开配置文件失败: {str(e)}")
|
|||
|
|
|
|||
|
|
def stop_application(self):
|
|||
|
|
"""停止应用"""
|
|||
|
|
self.logger.info("从托盘菜单退出程序")
|
|||
|
|
self.running = False
|
|||
|
|
self.icon.stop()
|
|||
|
|
|
|||
|
|
def run(self):
|
|||
|
|
"""运行托盘图标"""
|
|||
|
|
self.icon.run()
|
|||
|
|
|
|||
|
|
class TradeNotificationHandler:
|
|||
|
|
def __init__(self, message_sender, logger):
|
|||
|
|
self.sender = message_sender
|
|||
|
|
self.last_position = 0
|
|||
|
|
self.logger = logger
|
|||
|
|
|
|||
|
|
def check_new_trades(self):
|
|||
|
|
try:
|
|||
|
|
with open(CLIENT_LOG_PATH, 'r', encoding='utf-8', errors='ignore') as f:
|
|||
|
|
# 获取当前文件大小
|
|||
|
|
f.seek(0, 2)
|
|||
|
|
current_position = f.tell()
|
|||
|
|
|
|||
|
|
# 如果文件大小变小了，说明文件被重置
|
|||
|
|
if current_position < self.last_position:
|
|||
|
|
self.logger.info(f"检测到文件重置")
|
|||
|
|
self.last_position = 0 # 重置到文件开始
|
|||
|
|
|
|||
|
|
# 如果有新内容
|
|||
|
|
if current_position > self.last_position:
|
|||
|
|
# 只读取新增的内容
|
|||
|
|
f.seek(self.last_position)
|
|||
|
|
new_content = f.read()
|
|||
|
|
|
|||
|
|
# 更新位置（在处理内容之前）
|
|||
|
|
self.last_position = current_position
|
|||
|
|
|
|||
|
|
# 按行处理新内容
|
|||
|
|
lines = new_content.splitlines()
|
|||
|
|
for line in lines:
|
|||
|
|
if '@來自' in line or '@From' in line:
|
|||
|
|
self.logger.info(f"发现新消息: {line[:100]}...")
|
|||
|
|
self.process_trade_message(line)
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
self.logger.error(f"读取日志文件时发生错误: {str(e)}")
|
|||
|
|
|
|||
|
|
def process_trade_message(self, message):
|
|||
|
|
self.logger.info("\n开始处理消息...")
|
|||
|
|
|
|||
|
|
# 首先判断消息类型
|
|||
|
|
if '我想購買' in message or 'would like to buy' in message:
|
|||
|
|
self.logger.info("检测到交易消息")
|
|||
|
|
trade_patterns = [
|
|||
|
|
# 繁体中文格式
|
|||
|
|
{
|
|||
|
|
'pattern': r'.*\[INFO Client \d+\] @來自 ([^:]+): [^，]*，我想購買 ([^標]+)標價 ([^在]+)在 [^(]*\(倉庫頁 "([^"]+)"; 位置: ([^)]+)\)',
|
|||
|
|
'format': lambda m: {
|
|||
|
|
'buyer': m.group(1),
|
|||
|
|
'item': m.group(2).strip(),
|
|||
|
|
'price': m.group(3).strip(),
|
|||
|
|
'tab': m.group(4),
|
|||
|
|
'position': m.group(5)
|
|||
|
|
}
|
|||
|
|
},
|
|||
|
|
# 英文格式
|
|||
|
|
{
|
|||
|
|
'pattern': r'.*\[INFO Client \d+\] @From ([^:]+): Hi, I would like to buy your ([^listed]+)listed for ([^in]+)in [^(]*\(stash tab "([^"]+)"; position: ([^)]+)\)',
|
|||
|
|
'format': lambda m: {
|
|||
|
|
'buyer': m.group(1),
|
|||
|
|
'item': m.group(2).strip(),
|
|||
|
|
'price': m.group(3).strip(),
|
|||
|
|
'tab': m.group(4),
|
|||
|
|
'position': m.group(5)
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
for pattern_info in trade_patterns:
|
|||
|
|
match = re.match(pattern_info['pattern'], message)
|
|||
|
|
if match:
|
|||
|
|
self.logger.info("匹配到交易消息模式")
|
|||
|
|
trade_info = pattern_info['format'](match)
|
|||
|
|
self.logger.info(f"解析的交易信息: {trade_info}")
|
|||
|
|
|
|||
|
|
# 发送交易通知
|
|||
|
|
if TRADE_MESSAGES_CONFIG['message_forward'] and self.sender:
|
|||
|
|
message_content = (
|
|||
|
|
f"买家： {trade_info['buyer']}\n"
|
|||
|
|
f"商品： {trade_info['item']}\n"
|
|||
|
|
f"价格： {trade_info['price']}\n"
|
|||
|
|
f"仓库： {trade_info['tab']}\n"
|
|||
|
|
f"位置： {trade_info['position']}"
|
|||
|
|
)
|
|||
|
|
self.sender.send_message(
|
|||
|
|
content=message_content,
|
|||
|
|
title="POE 交易通知"
|
|||
|
|
)
|
|||
|
|
break
|
|||
|
|
else:
|
|||
|
|
self.logger.info("检测到私聊消息")
|
|||
|
|
# 处理私聊消息
|
|||
|
|
chat_patterns = [
|
|||
|
|
# 繁体中文格式
|
|||
|
|
r'.*\[INFO Client \d+\] @來自 ([^:]+): (.*)',
|
|||
|
|
# 英文格式
|
|||
|
|
r'.*\[INFO Client \d+\] @From ([^:]+): (.*)'
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
for pattern in chat_patterns:
|
|||
|
|
match = re.match(pattern, message)
|
|||
|
|
if match:
|
|||
|
|
sender = match.group(1)
|
|||
|
|
content = match.group(2)
|
|||
|
|
|
|||
|
|
# 发送私聊消息
|
|||
|
|
if CHAT_MESSAGES['message_forward']:
|
|||
|
|
message_content = (
|
|||
|
|
f"发送者：{sender}\n"
|
|||
|
|
f"内容：{content}\n"
|
|||
|
|
f"时间：{time.strftime('%H:%M:%S')}"
|
|||
|
|
)
|
|||
|
|
self.sender.send_message(
|
|||
|
|
content=message_content,
|
|||
|
|
title="POE 私聊消息"
|
|||
|
|
)
|
|||
|
|
break
|
|||
|
|
|
|||
|
|
def monitor_file(handler, is_running):
|
|||
|
|
handler.logger.info("文件监控已启动")
|
|||
|
|
try:
|
|||
|
|
while is_running():
|
|||
|
|
handler.check_new_trades()
|
|||
|
|
time.sleep(0.1)
|
|||
|
|
except Exception as e:
|
|||
|
|
handler.logger.error(f"监控线程异常: {str(e)}")
|
|||
|
|
handler.logger.info("监控线程已停止")
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
logger = setup_logging()
|
|||
|
|
logger.info("程序启动...")
|
|||
|
|
|
|||
|
|
# 创建托盘图标
|
|||
|
|
tray = TrayIcon(logger)
|
|||
|
|
|
|||
|
|
logger.info("初始化消息发送器...")
|
|||
|
|
message_sender = MessageSender()
|
|||
|
|
|
|||
|
|
logger.info(f"开始监控日志文件: {CLIENT_LOG_PATH}")
|
|||
|
|
handler = TradeNotificationHandler(message_sender, logger)
|
|||
|
|
|
|||
|
|
# 确保文件存在
|
|||
|
|
if not os.path.exists(CLIENT_LOG_PATH):
|
|||
|
|
logger.error(f"错误：找不到日志文件 {CLIENT_LOG_PATH}")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
with open(CLIENT_LOG_PATH, 'r', encoding='utf-8', errors='ignore') as f:
|
|||
|
|
f.seek(0, 2)
|
|||
|
|
handler.last_position = f.tell()
|
|||
|
|
logger.info(f"初始文件位置: {handler.last_position}")
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"读取文件位置时发生错误: {str(e)}")
|
|||
|
|
return
|
|||
|
|
|
|||
|
|
# 在新线程中运行文件监控
|
|||
|
|
monitor_thread = threading.Thread(
|
|||
|
|
target=monitor_file,
|
|||
|
|
args=(handler, lambda: tray.running)
|
|||
|
|
)
|
|||
|
|
monitor_thread.daemon = True
|
|||
|
|
monitor_thread.start()
|
|||
|
|
|
|||
|
|
# 运行托盘图标（这会阻塞主线程）
|
|||
|
|
tray.run()
|
|||
|
|
|
|||
|
|
logger.info("程序已退出")
|
|||
|
|
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
main()
|