wehub-resource-sync
121 lines
6.3 KiB
Python
121 lines
6.3 KiB
Python
import webview, threading, subprocess, sys, time, os, ctypes, atexit, socket, random
|
|
|
|
WINDOW_WIDTH, WINDOW_HEIGHT, RIGHT_PADDING, TOP_PADDING = 1200, 900, 0, 50
|
|
|
|
script_dir = os.path.dirname(os.path.abspath(__file__))
|
|
frontends_dir = os.path.join(script_dir, "frontends")
|
|
|
|
def find_free_port(lo=18501, hi=18599):
|
|
ports = list(range(lo, hi+1)); random.shuffle(ports)
|
|
for p in ports:
|
|
try: s = socket.socket(); s.bind(('127.0.0.1', p)); s.close(); return p
|
|
except OSError: continue
|
|
raise RuntimeError(f'No free port in {lo}-{hi}')
|
|
|
|
def get_screen_width():
|
|
try: return ctypes.windll.user32.GetSystemMetrics(0)
|
|
except: return 1920
|
|
|
|
def start_streamlit(port):
|
|
global proc
|
|
cmd = [sys.executable, "-m", "streamlit", "run", os.path.join(frontends_dir, "stapp.py"), "--server.port", str(port), "--server.address", "localhost", "--server.headless", "true", "--client.toolbarMode", "viewer"]
|
|
proc = subprocess.Popen(cmd)
|
|
atexit.register(proc.kill)
|
|
|
|
PASTE_HOOK_JS = """if (!window._pasteHooked) { window._pasteHooked = true;
|
|
document.addEventListener('paste', e => {
|
|
const items = e.clipboardData?.items; if (!items) return;
|
|
let t = null, hasText = false;
|
|
for (const item of items) {
|
|
if (item.kind === 'string' && (item.type === 'text/plain' || item.type === 'text/html')) hasText = true;
|
|
if (item.kind === 'file') { t = item.type.startsWith('image/') ? 'image in clipboard, ' : 'file in clipboard, '; }
|
|
}
|
|
if (!t || hasText) return;
|
|
e.preventDefault(); e.stopImmediatePropagation();
|
|
const el = document.querySelector('textarea[data-testid="stChatInputTextArea"]') || document.activeElement;
|
|
if (el && (el.tagName === 'TEXTAREA' || el.tagName === 'INPUT')) {
|
|
const s = Object.getOwnPropertyDescriptor(HTMLTextAreaElement.prototype, 'value')?.set || Object.getOwnPropertyDescriptor(HTMLInputElement.prototype, 'value')?.set;
|
|
s.call(el, el.value + t); el.dispatchEvent(new Event('input', { bubbles: true }));
|
|
}
|
|
}, true);
|
|
}"""
|
|
|
|
def paste_hook_injector():
|
|
"""注入剪贴板粘贴图片钩子，等窗口加载后执行一次，之后每60秒补注入防页面刷新丢失"""
|
|
time.sleep(5) # 等 webview 加载
|
|
while True:
|
|
try: window.evaluate_js(PASTE_HOOK_JS)
|
|
except: pass
|
|
time.sleep(60)
|
|
|
|
if __name__ == '__main__':
|
|
import argparse
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('port', nargs='?', default='0');
|
|
parser.add_argument('--tg', action='store_true', help='启动 Telegram Bot');
|
|
parser.add_argument('--qq', action='store_true', help='启动 QQ Bot');
|
|
parser.add_argument('--feishu', '--fs', dest='feishu', action='store_true', help='启动 Feishu Bot');
|
|
parser.add_argument('--wechat', '--wx', dest='wechat', action='store_true', help='启动 WeChat Bot');
|
|
parser.add_argument('--wecom', action='store_true', help='启动 WeCom Bot');
|
|
parser.add_argument('--dingtalk', '--dt', dest='dingtalk', action='store_true', help='启动 DingTalk Bot');
|
|
parser.add_argument('--sched', action='store_true', help='启动计划任务调度器')
|
|
parser.add_argument('--llm_no', type=int, default=0, help='LLM编号')
|
|
args = parser.parse_args()
|
|
port = str(find_free_port()) if args.port == '0' else args.port
|
|
print(f'[Launch] Using port {port}')
|
|
threading.Thread(target=start_streamlit, args=(port,), daemon=True).start()
|
|
|
|
if args.tg:
|
|
tgproc = subprocess.Popen([sys.executable, os.path.join(frontends_dir, "tgapp.py")], creationflags=subprocess.CREATE_NO_WINDOW if os.name=='nt' else 0)
|
|
atexit.register(tgproc.kill)
|
|
print('[Launch] Telegram Bot started')
|
|
else: print('[Launch] Telegram Bot not enabled (use --tg to start)')
|
|
|
|
if args.qq:
|
|
qqproc = subprocess.Popen([sys.executable, os.path.join(frontends_dir, "qqapp.py")], creationflags=subprocess.CREATE_NO_WINDOW if os.name=='nt' else 0)
|
|
atexit.register(qqproc.kill)
|
|
print('[Launch] QQ Bot started')
|
|
else: print('[Launch] QQ Bot not enabled (use --qq to start)')
|
|
|
|
if args.feishu:
|
|
fsproc = subprocess.Popen([sys.executable, os.path.join(frontends_dir, "fsapp.py")], creationflags=subprocess.CREATE_NO_WINDOW if os.name=='nt' else 0)
|
|
atexit.register(fsproc.kill)
|
|
print('[Launch] Feishu Bot started')
|
|
else: print('[Launch] Feishu Bot not enabled (use --feishu to start)')
|
|
|
|
if args.wechat:
|
|
wxproc = subprocess.Popen([sys.executable, os.path.join(frontends_dir, 'wechatapp.py')], creationflags=subprocess.CREATE_NO_WINDOW if os.name=='nt' else 0)
|
|
atexit.register(wxproc.kill)
|
|
print('[Launch] WeChat Bot started')
|
|
else: print('[Launch] WeChat Bot not enabled (use --wechat to start)')
|
|
|
|
if args.wecom:
|
|
wcproc = subprocess.Popen([sys.executable, os.path.join(frontends_dir, "wecomapp.py")], creationflags=subprocess.CREATE_NO_WINDOW if os.name=='nt' else 0)
|
|
atexit.register(wcproc.kill)
|
|
print('[Launch] WeCom Bot started')
|
|
else: print('[Launch] WeCom Bot not enabled (use --wecom to start)')
|
|
|
|
if args.dingtalk:
|
|
dtproc = subprocess.Popen([sys.executable, os.path.join(frontends_dir, "dingtalkapp.py")], creationflags=subprocess.CREATE_NO_WINDOW if os.name=='nt' else 0)
|
|
atexit.register(dtproc.kill)
|
|
print('[Launch] DingTalk Bot started')
|
|
else: print('[Launch] DingTalk Bot not enabled (use --dingtalk to start)')
|
|
|
|
if args.sched:
|
|
scheduler_proc = subprocess.Popen([sys.executable, os.path.join(script_dir, "agentmain.py"), "--reflect", os.path.join(script_dir, "reflect", "scheduler.py"), "--llm_no", str(args.llm_no)], creationflags=subprocess.CREATE_NO_WINDOW if os.name=='nt' else 0)
|
|
atexit.register(scheduler_proc.kill)
|
|
print('[Launch] Task Scheduler started (duplicate prevented by scheduler port lock)')
|
|
else: print('[Launch] Task Scheduler not enabled (--sched)')
|
|
|
|
threading.Thread(target=paste_hook_injector, daemon=True).start()
|
|
if os.name == 'nt':
|
|
screen_width = get_screen_width()
|
|
x_pos = screen_width - WINDOW_WIDTH - RIGHT_PADDING
|
|
else: x_pos = 100
|
|
time.sleep(2)
|
|
window = webview.create_window(
|
|
title='GenericAgent', url=f'http://localhost:{port}',
|
|
width=WINDOW_WIDTH, height=WINDOW_HEIGHT, x=x_pos, y=TOP_PADDING,
|
|
resizable=True, text_select=True)
|
|
webview.start()
|