mirror of
https://github.com/LmeSzinc/StarRailCopilot.git
synced 2024-11-26 02:11:06 +00:00
1266 lines
42 KiB
Python
1266 lines
42 KiB
Python
|
import argparse
|
|||
|
import queue
|
|||
|
import threading
|
|||
|
import time
|
|||
|
from datetime import datetime
|
|||
|
from functools import partial
|
|||
|
from typing import Dict, List, Optional
|
|||
|
|
|||
|
from pywebio import config as webconfig
|
|||
|
from pywebio.output import (
|
|||
|
Output,
|
|||
|
clear,
|
|||
|
close_popup,
|
|||
|
popup,
|
|||
|
put_button,
|
|||
|
put_buttons,
|
|||
|
put_collapse,
|
|||
|
put_column,
|
|||
|
put_error,
|
|||
|
put_html,
|
|||
|
put_link,
|
|||
|
put_loading,
|
|||
|
put_markdown,
|
|||
|
put_row,
|
|||
|
put_scope,
|
|||
|
put_table,
|
|||
|
put_text,
|
|||
|
put_warning,
|
|||
|
toast,
|
|||
|
use_scope,
|
|||
|
)
|
|||
|
from pywebio.pin import pin, pin_on_change
|
|||
|
from pywebio.session import go_app, info, local, register_thread, run_js, set_env
|
|||
|
|
|||
|
import module.webui.lang as lang
|
|||
|
from module.config.config import AzurLaneConfig, Function
|
|||
|
from module.config.utils import (
|
|||
|
alas_instance,
|
|||
|
alas_template,
|
|||
|
deep_get,
|
|||
|
deep_iter,
|
|||
|
deep_set,
|
|||
|
dict_to_kv,
|
|||
|
filepath_args,
|
|||
|
filepath_config,
|
|||
|
read_file,
|
|||
|
)
|
|||
|
from module.logger import logger
|
|||
|
from module.webui.base import Frame
|
|||
|
from module.webui.fake import (
|
|||
|
get_config_mod,
|
|||
|
load_config,
|
|||
|
)
|
|||
|
from module.webui.fastapi import asgi_app
|
|||
|
from module.webui.lang import _t, t
|
|||
|
from module.webui.pin import put_input, put_select
|
|||
|
from module.webui.process_manager import ProcessManager
|
|||
|
from module.webui.remote_access import RemoteAccess
|
|||
|
from module.webui.setting import State
|
|||
|
from module.webui.updater import updater
|
|||
|
from module.webui.utils import (
|
|||
|
Icon,
|
|||
|
Switch,
|
|||
|
TaskHandler,
|
|||
|
add_css,
|
|||
|
filepath_css,
|
|||
|
get_alas_config_listen_path,
|
|||
|
get_localstorage,
|
|||
|
get_window_visibility_state,
|
|||
|
login,
|
|||
|
parse_pin_value,
|
|||
|
raise_exception,
|
|||
|
re_fullmatch,
|
|||
|
)
|
|||
|
from module.webui.widgets import (
|
|||
|
BinarySwitchButton,
|
|||
|
RichLog,
|
|||
|
T_Output_Kwargs,
|
|||
|
put_icon_buttons,
|
|||
|
put_loading_text,
|
|||
|
put_none,
|
|||
|
put_output,
|
|||
|
)
|
|||
|
|
|||
|
task_handler = TaskHandler()
|
|||
|
|
|||
|
|
|||
|
class AlasGUI(Frame):
|
|||
|
ALAS_MENU: Dict[str, Dict[str, List[str]]]
|
|||
|
ALAS_ARGS: Dict[str, Dict[str, Dict[str, Dict[str, str]]]]
|
|||
|
theme = "default"
|
|||
|
|
|||
|
def initial(self) -> None:
|
|||
|
self.ALAS_MENU = read_file(filepath_args("menu", self.alas_mod))
|
|||
|
self.ALAS_ARGS = read_file(filepath_args("args", self.alas_mod))
|
|||
|
self._init_alas_config_watcher()
|
|||
|
|
|||
|
def __init__(self) -> None:
|
|||
|
super().__init__()
|
|||
|
# modified keys, return values of pin_wait_change()
|
|||
|
self.modified_config_queue = queue.Queue()
|
|||
|
# alas config name
|
|||
|
self.alas_name = ""
|
|||
|
self.alas_mod = "alas"
|
|||
|
self.alas_config = AzurLaneConfig("template")
|
|||
|
self.initial()
|
|||
|
|
|||
|
@use_scope("aside", clear=True)
|
|||
|
def set_aside(self) -> None:
|
|||
|
# TODO: update put_icon_buttons()
|
|||
|
put_icon_buttons(
|
|||
|
Icon.DEVELOP,
|
|||
|
buttons=[
|
|||
|
{"label": t("Gui.Aside.Home"), "value": "Home", "color": "aside"}
|
|||
|
],
|
|||
|
onclick=[self.ui_develop],
|
|||
|
),
|
|||
|
for name in alas_instance():
|
|||
|
put_icon_buttons(
|
|||
|
Icon.RUN,
|
|||
|
buttons=[{"label": name, "value": name, "color": "aside"}],
|
|||
|
onclick=self.ui_alas,
|
|||
|
)
|
|||
|
put_icon_buttons(
|
|||
|
Icon.ADD,
|
|||
|
buttons=[
|
|||
|
{"label": t("Gui.Aside.AddAlas"), "value": "AddAlas", "color": "aside"}
|
|||
|
],
|
|||
|
onclick=[self.ui_add_alas],
|
|||
|
),
|
|||
|
|
|||
|
@use_scope("header_status")
|
|||
|
def set_status(self, state: int) -> None:
|
|||
|
"""
|
|||
|
Args:
|
|||
|
state (int):
|
|||
|
1 (running)
|
|||
|
2 (not running)
|
|||
|
3 (warning, stop unexpectedly)
|
|||
|
4 (stop for update)
|
|||
|
0 (hide)
|
|||
|
-1 (*state not changed)
|
|||
|
"""
|
|||
|
if state == -1:
|
|||
|
return
|
|||
|
clear()
|
|||
|
|
|||
|
if state == 1:
|
|||
|
put_loading_text(t("Gui.Status.Running"), color="success")
|
|||
|
elif state == 2:
|
|||
|
put_loading_text(t("Gui.Status.Inactive"), color="secondary", fill=True)
|
|||
|
elif state == 3:
|
|||
|
put_loading_text(t("Gui.Status.Warning"), shape="grow", color="warning")
|
|||
|
elif state == 4:
|
|||
|
put_loading_text(t("Gui.Status.Updating"), shape="grow", color="success")
|
|||
|
|
|||
|
@classmethod
|
|||
|
def set_theme(cls, theme="default") -> None:
|
|||
|
cls.theme = theme
|
|||
|
State.deploy_config.Theme = theme
|
|||
|
State.theme = theme
|
|||
|
webconfig(theme=theme)
|
|||
|
|
|||
|
@use_scope("menu", clear=True)
|
|||
|
def alas_set_menu(self) -> None:
|
|||
|
"""
|
|||
|
Set menu
|
|||
|
"""
|
|||
|
put_buttons(
|
|||
|
[
|
|||
|
{
|
|||
|
"label": t("Gui.MenuAlas.Overview"),
|
|||
|
"value": "Overview",
|
|||
|
"color": "menu",
|
|||
|
}
|
|||
|
],
|
|||
|
onclick=[self.alas_overview],
|
|||
|
).style(f"--menu-Overview--"),
|
|||
|
|
|||
|
for key, tasks in deep_iter(self.ALAS_MENU, depth=2):
|
|||
|
# path = '.'.join(key)
|
|||
|
menu = key[1]
|
|||
|
|
|||
|
if menu == "Tool":
|
|||
|
_onclick = self.alas_daemon_overview
|
|||
|
else:
|
|||
|
_onclick = self.alas_set_group
|
|||
|
|
|||
|
task_btn_list = []
|
|||
|
for task in tasks:
|
|||
|
task_btn_list.append(
|
|||
|
put_buttons(
|
|||
|
[
|
|||
|
{
|
|||
|
"label": t(f"Task.{task}.name"),
|
|||
|
"value": task,
|
|||
|
"color": "menu",
|
|||
|
}
|
|||
|
],
|
|||
|
onclick=_onclick,
|
|||
|
).style(f"--menu-{task}--")
|
|||
|
)
|
|||
|
|
|||
|
put_collapse(title=t(f"Menu.{menu}.name"), content=task_btn_list)
|
|||
|
|
|||
|
self.alas_overview()
|
|||
|
|
|||
|
@use_scope("content", clear=True)
|
|||
|
def alas_set_group(self, task: str) -> None:
|
|||
|
"""
|
|||
|
Set arg groups from dict
|
|||
|
"""
|
|||
|
self.init_menu(name=task)
|
|||
|
self.set_title(t(f"Task.{task}.name"))
|
|||
|
|
|||
|
put_scope("_groups", [put_none(), put_scope("groups"), put_scope("navigator")])
|
|||
|
|
|||
|
task_help: str = t(f"Task.{task}.help")
|
|||
|
if task_help:
|
|||
|
put_scope(
|
|||
|
"group__info",
|
|||
|
scope="groups",
|
|||
|
content=[put_text(task_help).style("font-size: 1rem")],
|
|||
|
)
|
|||
|
|
|||
|
config = self.alas_config.read_file(self.alas_name)
|
|||
|
for group, arg_dict in deep_iter(self.ALAS_ARGS[task], depth=1):
|
|||
|
if self.set_group(group, arg_dict, config, task):
|
|||
|
self.set_navigator(group)
|
|||
|
|
|||
|
@use_scope("groups")
|
|||
|
def set_group(self, group, arg_dict, config, task):
|
|||
|
group_name = group[0]
|
|||
|
|
|||
|
output_list: List[Output] = []
|
|||
|
for arg, arg_dict in deep_iter(arg_dict, depth=1):
|
|||
|
output_kwargs: T_Output_Kwargs = arg_dict.copy()
|
|||
|
|
|||
|
# Skip hide
|
|||
|
display: Optional[str] = output_kwargs.pop("display", None)
|
|||
|
if display == "hide":
|
|||
|
continue
|
|||
|
# Disable
|
|||
|
elif display == "disabled":
|
|||
|
output_kwargs["disabled"] = True
|
|||
|
# Output type
|
|||
|
output_kwargs["widget_type"] = output_kwargs.pop("type")
|
|||
|
|
|||
|
arg_name = arg[0] # [arg_name,]
|
|||
|
# Internal pin widget name
|
|||
|
output_kwargs["name"] = f"{task}_{group_name}_{arg_name}"
|
|||
|
# Display title
|
|||
|
output_kwargs["title"] = t(f"{group_name}.{arg_name}.name")
|
|||
|
|
|||
|
# Get value from config
|
|||
|
value = deep_get(
|
|||
|
config, [task, group_name, arg_name], output_kwargs["value"]
|
|||
|
)
|
|||
|
# idk
|
|||
|
value = str(value) if isinstance(value, datetime) else value
|
|||
|
# Default value
|
|||
|
output_kwargs["value"] = value
|
|||
|
# Options
|
|||
|
output_kwargs["options"] = options = output_kwargs.pop("option", [])
|
|||
|
# Options label
|
|||
|
options_label = []
|
|||
|
for opt in options:
|
|||
|
options_label.append(t(f"{group_name}.{arg_name}.{opt}"))
|
|||
|
output_kwargs["options_label"] = options_label
|
|||
|
# Help
|
|||
|
arg_help = t(f"{group_name}.{arg_name}.help")
|
|||
|
if arg_help == "" or not arg_help:
|
|||
|
arg_help = None
|
|||
|
output_kwargs["help"] = arg_help
|
|||
|
# Invalid feedback
|
|||
|
output_kwargs["invalid_feedback"] = t("Gui.Text.InvalidFeedBack", value)
|
|||
|
|
|||
|
o = put_output(output_kwargs)
|
|||
|
if o is not None:
|
|||
|
# output will inherit current scope when created, override here
|
|||
|
o.spec["scope"] = f"#pywebio-scope-group_{group_name}"
|
|||
|
output_list.append(o)
|
|||
|
|
|||
|
if not output_list:
|
|||
|
return 0
|
|||
|
|
|||
|
with use_scope(f"group_{group_name}"):
|
|||
|
put_text(t(f"{group_name}._info.name"))
|
|||
|
group_help = t(f"{group_name}._info.help")
|
|||
|
if group_help != "":
|
|||
|
put_text(group_help)
|
|||
|
put_html('<hr class="hr-group">')
|
|||
|
for output in output_list:
|
|||
|
output.show()
|
|||
|
|
|||
|
return len(output_list)
|
|||
|
|
|||
|
@use_scope("navigator")
|
|||
|
def set_navigator(self, group):
|
|||
|
js = f"""
|
|||
|
$("#pywebio-scope-groups").scrollTop(
|
|||
|
$("#pywebio-scope-group_{group[0]}").position().top
|
|||
|
+ $("#pywebio-scope-groups").scrollTop() - 59
|
|||
|
)
|
|||
|
"""
|
|||
|
put_button(
|
|||
|
label=t(f"{group[0]}._info.name"),
|
|||
|
onclick=lambda: run_js(js),
|
|||
|
color="navigator",
|
|||
|
)
|
|||
|
|
|||
|
@use_scope("content", clear=True)
|
|||
|
def alas_overview(self) -> None:
|
|||
|
self.init_menu(name="Overview")
|
|||
|
self.set_title(t(f"Gui.MenuAlas.Overview"))
|
|||
|
|
|||
|
put_scope("overview", [put_scope("schedulers"), put_scope("logs")])
|
|||
|
|
|||
|
with use_scope("schedulers"):
|
|||
|
put_scope(
|
|||
|
"scheduler-bar",
|
|||
|
[
|
|||
|
put_text(t("Gui.Overview.Scheduler")).style(
|
|||
|
"font-size: 1.25rem; margin: auto .5rem auto;"
|
|||
|
),
|
|||
|
put_scope("scheduler_btn"),
|
|||
|
],
|
|||
|
)
|
|||
|
put_scope(
|
|||
|
"running",
|
|||
|
[
|
|||
|
put_text(t("Gui.Overview.Running")),
|
|||
|
put_html('<hr class="hr-group">'),
|
|||
|
put_scope("running_tasks"),
|
|||
|
],
|
|||
|
)
|
|||
|
put_scope(
|
|||
|
"pending",
|
|||
|
[
|
|||
|
put_text(t("Gui.Overview.Pending")),
|
|||
|
put_html('<hr class="hr-group">'),
|
|||
|
put_scope("pending_tasks"),
|
|||
|
],
|
|||
|
)
|
|||
|
put_scope(
|
|||
|
"waiting",
|
|||
|
[
|
|||
|
put_text(t("Gui.Overview.Waiting")),
|
|||
|
put_html('<hr class="hr-group">'),
|
|||
|
put_scope("waiting_tasks"),
|
|||
|
],
|
|||
|
)
|
|||
|
|
|||
|
switch_scheduler = BinarySwitchButton(
|
|||
|
label_on=t("Gui.Button.Stop"),
|
|||
|
label_off=t("Gui.Button.Start"),
|
|||
|
onclick_on=lambda: self.alas.stop(),
|
|||
|
onclick_off=lambda: self.alas.start(None, updater.event),
|
|||
|
get_state=lambda: self.alas.alive,
|
|||
|
color_on="off",
|
|||
|
color_off="on",
|
|||
|
scope="scheduler_btn",
|
|||
|
)
|
|||
|
|
|||
|
log = RichLog("log")
|
|||
|
|
|||
|
with use_scope("logs"):
|
|||
|
put_scope(
|
|||
|
"log-bar",
|
|||
|
[
|
|||
|
put_text(t("Gui.Overview.Log")).style(
|
|||
|
"font-size: 1.25rem; margin: auto .5rem auto;"
|
|||
|
),
|
|||
|
put_scope(
|
|||
|
"log-bar-btns",
|
|||
|
[
|
|||
|
put_scope("log_scroll_btn"),
|
|||
|
],
|
|||
|
),
|
|||
|
],
|
|||
|
),
|
|||
|
put_scope("log", [put_html("")])
|
|||
|
|
|||
|
log.console.width = log.get_width()
|
|||
|
|
|||
|
switch_log_scroll = BinarySwitchButton(
|
|||
|
label_on=t("Gui.Button.ScrollON"),
|
|||
|
label_off=t("Gui.Button.ScrollOFF"),
|
|||
|
onclick_on=lambda: log.set_scroll(False),
|
|||
|
onclick_off=lambda: log.set_scroll(True),
|
|||
|
get_state=lambda: log.keep_bottom,
|
|||
|
color_on="on",
|
|||
|
color_off="off",
|
|||
|
scope="log_scroll_btn",
|
|||
|
)
|
|||
|
|
|||
|
self.task_handler.add(switch_scheduler.g(), 1, True)
|
|||
|
self.task_handler.add(switch_log_scroll.g(), 1, True)
|
|||
|
self.task_handler.add(self.alas_update_overview_task, 10, True)
|
|||
|
self.task_handler.add(log.put_log(self.alas), 0.25, True)
|
|||
|
|
|||
|
def _init_alas_config_watcher(self) -> None:
|
|||
|
def put_queue(path, value):
|
|||
|
self.modified_config_queue.put({"name": path, "value": value})
|
|||
|
|
|||
|
for path in get_alas_config_listen_path(self.ALAS_ARGS):
|
|||
|
pin_on_change(
|
|||
|
name="_".join(path), onchange=partial(put_queue, ".".join(path))
|
|||
|
)
|
|||
|
logger.info("Init config watcher done.")
|
|||
|
|
|||
|
def _alas_thread_update_config(self) -> None:
|
|||
|
modified = {}
|
|||
|
while self.alive:
|
|||
|
try:
|
|||
|
d = self.modified_config_queue.get(timeout=10)
|
|||
|
config_name = self.alas_name
|
|||
|
read = self.alas_config.read_file
|
|||
|
write = self.alas_config.write_file
|
|||
|
except queue.Empty:
|
|||
|
continue
|
|||
|
modified[d["name"]] = d["value"]
|
|||
|
while True:
|
|||
|
try:
|
|||
|
d = self.modified_config_queue.get(timeout=1)
|
|||
|
modified[d["name"]] = d["value"]
|
|||
|
except queue.Empty:
|
|||
|
self._save_config(modified, config_name, read, write)
|
|||
|
modified.clear()
|
|||
|
break
|
|||
|
|
|||
|
def _save_config(
|
|||
|
self,
|
|||
|
modified: Dict[str, str],
|
|||
|
config_name: str,
|
|||
|
read=State.config_updater.read_file,
|
|||
|
write=State.config_updater.write_file,
|
|||
|
) -> None:
|
|||
|
try:
|
|||
|
valid = []
|
|||
|
invalid = []
|
|||
|
config = read(config_name)
|
|||
|
for k, v in modified.copy().items():
|
|||
|
valuetype = deep_get(self.ALAS_ARGS, k + ".valuetype")
|
|||
|
v = parse_pin_value(v, valuetype)
|
|||
|
validate = deep_get(self.ALAS_ARGS, k + ".validate")
|
|||
|
if not len(str(v)):
|
|||
|
default = deep_get(self.ALAS_ARGS, k + ".value")
|
|||
|
modified[k] = default
|
|||
|
deep_set(config, k, default)
|
|||
|
valid.append(k)
|
|||
|
pin["_".join(k.split("."))] = default
|
|||
|
|
|||
|
elif not validate or re_fullmatch(validate, v):
|
|||
|
deep_set(config, k, v)
|
|||
|
modified[k] = v
|
|||
|
valid.append(k)
|
|||
|
|
|||
|
# update Emotion Record if Emotion Value is changed
|
|||
|
if "Emotion" in k and "Value" in k:
|
|||
|
k = k.split(".")
|
|||
|
k[-1] = k[-1].replace("Value", "Record")
|
|||
|
k = ".".join(k)
|
|||
|
v = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
|||
|
modified[k] = v
|
|||
|
deep_set(config, k, v)
|
|||
|
valid.append(k)
|
|||
|
pin["_".join(k.split("."))] = v
|
|||
|
else:
|
|||
|
modified.pop(k)
|
|||
|
invalid.append(k)
|
|||
|
logger.warning(f"Invalid value {v} for key {k}, skip saving.")
|
|||
|
self.pin_remove_invalid_mark(valid)
|
|||
|
self.pin_set_invalid_mark(invalid)
|
|||
|
if modified:
|
|||
|
toast(
|
|||
|
t("Gui.Toast.ConfigSaved"),
|
|||
|
duration=1,
|
|||
|
position="right",
|
|||
|
color="success",
|
|||
|
)
|
|||
|
logger.info(
|
|||
|
f"Save config {filepath_config(config_name)}, {dict_to_kv(modified)}"
|
|||
|
)
|
|||
|
write(config_name, config)
|
|||
|
except Exception as e:
|
|||
|
logger.exception(e)
|
|||
|
|
|||
|
def alas_update_overview_task(self) -> None:
|
|||
|
if not self.visible:
|
|||
|
return
|
|||
|
self.alas_config.load()
|
|||
|
self.alas_config.get_next_task()
|
|||
|
|
|||
|
if len(self.alas_config.pending_task) >= 1:
|
|||
|
if self.alas.alive:
|
|||
|
running = self.alas_config.pending_task[:1]
|
|||
|
pending = self.alas_config.pending_task[1:]
|
|||
|
else:
|
|||
|
running = []
|
|||
|
pending = self.alas_config.pending_task[:]
|
|||
|
else:
|
|||
|
running = []
|
|||
|
pending = []
|
|||
|
waiting = self.alas_config.waiting_task
|
|||
|
|
|||
|
def put_task(func: Function):
|
|||
|
with use_scope(f"overview-task_{func.command}"):
|
|||
|
put_column(
|
|||
|
[
|
|||
|
put_text(t(f"Task.{func.command}.name")).style("--arg-title--"),
|
|||
|
put_text(str(func.next_run)).style("--arg-help--"),
|
|||
|
],
|
|||
|
size="auto auto",
|
|||
|
)
|
|||
|
put_button(
|
|||
|
label=t("Gui.Button.Setting"),
|
|||
|
onclick=lambda: self.alas_set_group(func.command),
|
|||
|
color="off",
|
|||
|
)
|
|||
|
|
|||
|
clear("running_tasks")
|
|||
|
clear("pending_tasks")
|
|||
|
clear("waiting_tasks")
|
|||
|
with use_scope("running_tasks"):
|
|||
|
if running:
|
|||
|
for task in running:
|
|||
|
put_task(task)
|
|||
|
else:
|
|||
|
put_text(t("Gui.Overview.NoTask")).style("--overview-notask-text--")
|
|||
|
with use_scope("pending_tasks"):
|
|||
|
if pending:
|
|||
|
for task in pending:
|
|||
|
put_task(task)
|
|||
|
else:
|
|||
|
put_text(t("Gui.Overview.NoTask")).style("--overview-notask-text--")
|
|||
|
with use_scope("waiting_tasks"):
|
|||
|
if waiting:
|
|||
|
for task in waiting:
|
|||
|
put_task(task)
|
|||
|
else:
|
|||
|
put_text(t("Gui.Overview.NoTask")).style("--overview-notask-text--")
|
|||
|
|
|||
|
@use_scope("content", clear=True)
|
|||
|
def alas_daemon_overview(self, task: str) -> None:
|
|||
|
self.init_menu(name=task)
|
|||
|
self.set_title(t(f"Task.{task}.name"))
|
|||
|
|
|||
|
log = RichLog("log")
|
|||
|
|
|||
|
if self.is_mobile:
|
|||
|
put_scope(
|
|||
|
"daemon-overview",
|
|||
|
[
|
|||
|
put_scope("scheduler-bar"),
|
|||
|
put_scope("groups"),
|
|||
|
put_scope("log-bar"),
|
|||
|
put_scope("log", [put_html("")]),
|
|||
|
],
|
|||
|
)
|
|||
|
else:
|
|||
|
put_scope(
|
|||
|
"daemon-overview",
|
|||
|
[
|
|||
|
put_none(),
|
|||
|
put_scope(
|
|||
|
"_daemon",
|
|||
|
[
|
|||
|
put_scope(
|
|||
|
"_daemon_upper",
|
|||
|
[put_scope("scheduler-bar"), put_scope("log-bar")],
|
|||
|
),
|
|||
|
put_scope("groups"),
|
|||
|
put_scope("log", [put_html("")]),
|
|||
|
],
|
|||
|
),
|
|||
|
put_none(),
|
|||
|
],
|
|||
|
)
|
|||
|
|
|||
|
log.console.width = log.get_width()
|
|||
|
|
|||
|
with use_scope("scheduler-bar"):
|
|||
|
put_text(t("Gui.Overview.Scheduler")).style(
|
|||
|
"font-size: 1.25rem; margin: auto .5rem auto;"
|
|||
|
)
|
|||
|
put_scope("scheduler_btn")
|
|||
|
|
|||
|
switch_scheduler = BinarySwitchButton(
|
|||
|
label_on=t("Gui.Button.Stop"),
|
|||
|
label_off=t("Gui.Button.Start"),
|
|||
|
onclick_on=lambda: self.alas.stop(),
|
|||
|
onclick_off=lambda: self.alas.start(task),
|
|||
|
get_state=lambda: self.alas.alive,
|
|||
|
color_on="off",
|
|||
|
color_off="on",
|
|||
|
scope="scheduler_btn",
|
|||
|
)
|
|||
|
|
|||
|
with use_scope("log-bar"):
|
|||
|
put_text(t("Gui.Overview.Log")).style(
|
|||
|
"font-size: 1.25rem; margin: auto .5rem auto;"
|
|||
|
)
|
|||
|
put_scope(
|
|||
|
"log-bar-btns",
|
|||
|
[
|
|||
|
put_scope("log_scroll_btn"),
|
|||
|
],
|
|||
|
)
|
|||
|
|
|||
|
switch_log_scroll = BinarySwitchButton(
|
|||
|
label_on=t("Gui.Button.ScrollON"),
|
|||
|
label_off=t("Gui.Button.ScrollOFF"),
|
|||
|
onclick_on=lambda: log.set_scroll(False),
|
|||
|
onclick_off=lambda: log.set_scroll(True),
|
|||
|
get_state=lambda: log.keep_bottom,
|
|||
|
color_on="on",
|
|||
|
color_off="off",
|
|||
|
scope="log_scroll_btn",
|
|||
|
)
|
|||
|
|
|||
|
config = self.alas_config.read_file(self.alas_name)
|
|||
|
for group, arg_dict in deep_iter(self.ALAS_ARGS[task], depth=1):
|
|||
|
if group[0] == "Storage":
|
|||
|
continue
|
|||
|
self.set_group(group, arg_dict, config, task)
|
|||
|
|
|||
|
self.task_handler.add(switch_scheduler.g(), 1, True)
|
|||
|
self.task_handler.add(switch_log_scroll.g(), 1, True)
|
|||
|
self.task_handler.add(log.put_log(self.alas), 0.25, True)
|
|||
|
|
|||
|
@use_scope("menu", clear=True)
|
|||
|
def dev_set_menu(self) -> None:
|
|||
|
self.init_menu(collapse_menu=False, name="Develop")
|
|||
|
|
|||
|
put_button(
|
|||
|
label=t("Gui.MenuDevelop.HomePage"),
|
|||
|
onclick=self.show,
|
|||
|
color="menu",
|
|||
|
).style(f"--menu-HomePage--")
|
|||
|
|
|||
|
# put_button(
|
|||
|
# label=t("Gui.MenuDevelop.Translate"),
|
|||
|
# onclick=self.dev_translate,
|
|||
|
# color="menu",
|
|||
|
# ).style(f"--menu-Translate--")
|
|||
|
|
|||
|
put_button(
|
|||
|
label=t("Gui.MenuDevelop.Update"),
|
|||
|
onclick=self.dev_update,
|
|||
|
color="menu",
|
|||
|
).style(f"--menu-Update--")
|
|||
|
|
|||
|
put_button(
|
|||
|
label=t("Gui.MenuDevelop.Remote"),
|
|||
|
onclick=self.dev_remote,
|
|||
|
color="menu",
|
|||
|
).style(f"--menu-Remote--")
|
|||
|
|
|||
|
put_button(
|
|||
|
label=t("Gui.MenuDevelop.Utils"),
|
|||
|
onclick=self.dev_utils,
|
|||
|
color="menu",
|
|||
|
).style(f"--menu-Utils--")
|
|||
|
|
|||
|
def dev_translate(self) -> None:
|
|||
|
go_app("translate", new_window=True)
|
|||
|
lang.TRANSLATE_MODE = True
|
|||
|
self.show()
|
|||
|
|
|||
|
@use_scope("content", clear=True)
|
|||
|
def dev_update(self) -> None:
|
|||
|
self.init_menu(name="Update")
|
|||
|
self.set_title(t("Gui.MenuDevelop.Update"))
|
|||
|
|
|||
|
if State.restart_event is None:
|
|||
|
put_warning(t("Gui.Update.DisabledWarn"))
|
|||
|
|
|||
|
put_row(
|
|||
|
content=[put_scope("updater_loading"), None, put_scope("updater_state")],
|
|||
|
size="auto .25rem 1fr",
|
|||
|
)
|
|||
|
|
|||
|
put_scope("updater_btn")
|
|||
|
put_scope("updater_info")
|
|||
|
|
|||
|
def update_table():
|
|||
|
with use_scope("updater_info", clear=True):
|
|||
|
local_commit = updater.get_commit(short_sha1=True)
|
|||
|
upstream_commit = updater.get_commit(
|
|||
|
f"origin/{updater.Branch}", short_sha1=True
|
|||
|
)
|
|||
|
put_table(
|
|||
|
[
|
|||
|
[t("Gui.Update.Local"), *local_commit],
|
|||
|
[t("Gui.Update.Upstream"), *upstream_commit],
|
|||
|
],
|
|||
|
header=[
|
|||
|
"",
|
|||
|
"SHA1",
|
|||
|
t("Gui.Update.Author"),
|
|||
|
t("Gui.Update.Time"),
|
|||
|
t("Gui.Update.Message"),
|
|||
|
],
|
|||
|
)
|
|||
|
with use_scope("updater_detail", clear=True):
|
|||
|
put_text(t("Gui.Update.DetailedHistory"))
|
|||
|
history = updater.get_commit(
|
|||
|
f"origin/{updater.Branch}", n=20, short_sha1=True
|
|||
|
)
|
|||
|
put_table(
|
|||
|
[commit for commit in history],
|
|||
|
header=[
|
|||
|
"SHA1",
|
|||
|
t("Gui.Update.Author"),
|
|||
|
t("Gui.Update.Time"),
|
|||
|
t("Gui.Update.Message"),
|
|||
|
],
|
|||
|
)
|
|||
|
|
|||
|
def u(state):
|
|||
|
if state == -1:
|
|||
|
return
|
|||
|
clear("updater_loading")
|
|||
|
clear("updater_state")
|
|||
|
clear("updater_btn")
|
|||
|
if state == 0:
|
|||
|
put_loading("border", "secondary", "updater_loading").style(
|
|||
|
"--loading-border-fill--"
|
|||
|
)
|
|||
|
put_text(t("Gui.Update.UpToDate"), scope="updater_state")
|
|||
|
put_button(
|
|||
|
t("Gui.Button.CheckUpdate"),
|
|||
|
onclick=updater.check_update,
|
|||
|
color="info",
|
|||
|
scope="updater_btn",
|
|||
|
)
|
|||
|
update_table()
|
|||
|
elif state == 1:
|
|||
|
put_loading("grow", "success", "updater_loading").style(
|
|||
|
"--loading-grow--"
|
|||
|
)
|
|||
|
put_text(t("Gui.Update.HaveUpdate"), scope="updater_state")
|
|||
|
put_button(
|
|||
|
t("Gui.Button.ClickToUpdate"),
|
|||
|
onclick=updater.run_update,
|
|||
|
color="success",
|
|||
|
scope="updater_btn",
|
|||
|
)
|
|||
|
update_table()
|
|||
|
elif state == "checking":
|
|||
|
put_loading("border", "primary", "updater_loading").style(
|
|||
|
"--loading-border--"
|
|||
|
)
|
|||
|
put_text(t("Gui.Update.UpdateChecking"), scope="updater_state")
|
|||
|
elif state == "failed":
|
|||
|
put_loading("grow", "danger", "updater_loading").style(
|
|||
|
"--loading-grow--"
|
|||
|
)
|
|||
|
put_text(t("Gui.Update.UpdateFailed"), scope="updater_state")
|
|||
|
put_button(
|
|||
|
t("Gui.Button.RetryUpdate"),
|
|||
|
onclick=updater.run_update,
|
|||
|
color="primary",
|
|||
|
scope="updater_btn",
|
|||
|
)
|
|||
|
elif state == "start":
|
|||
|
put_loading("border", "primary", "updater_loading").style(
|
|||
|
"--loading-border--"
|
|||
|
)
|
|||
|
put_text(t("Gui.Update.UpdateStart"), scope="updater_state")
|
|||
|
put_button(
|
|||
|
t("Gui.Button.CancelUpdate"),
|
|||
|
onclick=updater.cancel,
|
|||
|
color="danger",
|
|||
|
scope="updater_btn",
|
|||
|
)
|
|||
|
elif state == "wait":
|
|||
|
put_loading("border", "primary", "updater_loading").style(
|
|||
|
"--loading-border--"
|
|||
|
)
|
|||
|
put_text(t("Gui.Update.UpdateWait"), scope="updater_state")
|
|||
|
put_button(
|
|||
|
t("Gui.Button.CancelUpdate"),
|
|||
|
onclick=updater.cancel,
|
|||
|
color="danger",
|
|||
|
scope="updater_btn",
|
|||
|
)
|
|||
|
elif state == "run update":
|
|||
|
put_loading("border", "primary", "updater_loading").style(
|
|||
|
"--loading-border--"
|
|||
|
)
|
|||
|
put_text(t("Gui.Update.UpdateRun"), scope="updater_state")
|
|||
|
put_button(
|
|||
|
t("Gui.Button.CancelUpdate"),
|
|||
|
onclick=updater.cancel,
|
|||
|
color="danger",
|
|||
|
scope="updater_btn",
|
|||
|
disabled=True,
|
|||
|
)
|
|||
|
elif state == "reload":
|
|||
|
put_loading("grow", "success", "updater_loading").style(
|
|||
|
"--loading-grow--"
|
|||
|
)
|
|||
|
put_text(t("Gui.Update.UpdateSuccess"), scope="updater_state")
|
|||
|
update_table()
|
|||
|
elif state == "finish":
|
|||
|
put_loading("grow", "success", "updater_loading").style(
|
|||
|
"--loading-grow--"
|
|||
|
)
|
|||
|
put_text(t("Gui.Update.UpdateFinish"), scope="updater_state")
|
|||
|
update_table()
|
|||
|
elif state == "cancel":
|
|||
|
put_loading("border", "danger", "updater_loading").style(
|
|||
|
"--loading-border--"
|
|||
|
)
|
|||
|
put_text(t("Gui.Update.UpdateCancel"), scope="updater_state")
|
|||
|
put_button(
|
|||
|
t("Gui.Button.CancelUpdate"),
|
|||
|
onclick=updater.cancel,
|
|||
|
color="danger",
|
|||
|
scope="updater_btn",
|
|||
|
disabled=True,
|
|||
|
)
|
|||
|
else:
|
|||
|
put_text(
|
|||
|
"Something went wrong, please contact develops",
|
|||
|
scope="updater_state",
|
|||
|
)
|
|||
|
put_text(f"state: {state}", scope="updater_state")
|
|||
|
|
|||
|
updater_switch = Switch(
|
|||
|
status=u, get_state=lambda: updater.state, name="updater"
|
|||
|
)
|
|||
|
|
|||
|
update_table()
|
|||
|
self.task_handler.add(updater_switch.g(), delay=0.5, pending_delete=True)
|
|||
|
|
|||
|
updater.check_update()
|
|||
|
|
|||
|
@use_scope("content", clear=True)
|
|||
|
def dev_utils(self) -> None:
|
|||
|
self.init_menu(name="Utils")
|
|||
|
self.set_title(t("Gui.MenuDevelop.Utils"))
|
|||
|
put_button(label="Raise exception", onclick=raise_exception)
|
|||
|
|
|||
|
def _force_restart():
|
|||
|
if State.restart_event is not None:
|
|||
|
toast("Alas will restart in 3 seconds", duration=0, color="error")
|
|||
|
clearup()
|
|||
|
State.restart_event.set()
|
|||
|
else:
|
|||
|
toast("Reload not enabled", color="error")
|
|||
|
|
|||
|
put_button(label="Force restart", onclick=_force_restart)
|
|||
|
|
|||
|
@use_scope("content", clear=True)
|
|||
|
def dev_remote(self) -> None:
|
|||
|
self.init_menu(name="Remote")
|
|||
|
self.set_title(t("Gui.MenuDevelop.Remote"))
|
|||
|
put_row(
|
|||
|
content=[put_scope("remote_loading"), None, put_scope("remote_state")],
|
|||
|
size="auto .25rem 1fr",
|
|||
|
)
|
|||
|
put_scope("remote_info")
|
|||
|
|
|||
|
def u(state):
|
|||
|
if state == -1:
|
|||
|
return
|
|||
|
clear("remote_loading")
|
|||
|
clear("remote_state")
|
|||
|
clear("remote_info")
|
|||
|
if state in (1, 2):
|
|||
|
put_loading("grow", "success", "remote_loading").style(
|
|||
|
"--loading-grow--"
|
|||
|
)
|
|||
|
put_text(t("Gui.Remote.Running"), scope="remote_state")
|
|||
|
put_text(t("Gui.Remote.EntryPoint"), scope="remote_info")
|
|||
|
entrypoint = RemoteAccess.get_entry_point()
|
|||
|
if entrypoint:
|
|||
|
if State.electron: # Prevent click into url in electron client
|
|||
|
put_text(entrypoint, scope="remote_info").style(
|
|||
|
"text-decoration-line: underline"
|
|||
|
)
|
|||
|
else:
|
|||
|
put_link(name=entrypoint, url=entrypoint, scope="remote_info")
|
|||
|
else:
|
|||
|
put_text("Loading...", scope="remote_info")
|
|||
|
elif state in (0, 3):
|
|||
|
put_loading("border", "secondary", "remote_loading").style(
|
|||
|
"--loading-border-fill--"
|
|||
|
)
|
|||
|
if (
|
|||
|
State.deploy_config.EnableRemoteAccess
|
|||
|
and State.deploy_config.Password
|
|||
|
):
|
|||
|
put_text(t("Gui.Remote.NotRunning"), scope="remote_state")
|
|||
|
else:
|
|||
|
put_text(t("Gui.Remote.NotEnable"), scope="remote_state")
|
|||
|
put_text(t("Gui.Remote.ConfigureHint"), scope="remote_info")
|
|||
|
url = "http://app.azurlane.cloud" + (
|
|||
|
"" if State.deploy_config.Language.startswith("zh") else "/en.html"
|
|||
|
)
|
|||
|
put_html(
|
|||
|
f'<a href="{url}" target="_blank">{url}</a>', scope="remote_info"
|
|||
|
)
|
|||
|
if state == 3:
|
|||
|
put_warning(
|
|||
|
t("Gui.Remote.SSHNotInstall"),
|
|||
|
closable=False,
|
|||
|
scope="remote_info",
|
|||
|
)
|
|||
|
|
|||
|
remote_switch = Switch(
|
|||
|
status=u, get_state=RemoteAccess.get_state, name="remote"
|
|||
|
)
|
|||
|
|
|||
|
self.task_handler.add(remote_switch.g(), delay=1, pending_delete=True)
|
|||
|
|
|||
|
def ui_develop(self) -> None:
|
|||
|
if not self.is_mobile:
|
|||
|
self.show()
|
|||
|
return
|
|||
|
self.init_aside(name="Home")
|
|||
|
self.set_title(t("Gui.Aside.Home"))
|
|||
|
self.dev_set_menu()
|
|||
|
self.alas_name = ""
|
|||
|
if hasattr(self, "alas"):
|
|||
|
del self.alas
|
|||
|
self.state_switch.switch()
|
|||
|
|
|||
|
def ui_alas(self, config_name: str) -> None:
|
|||
|
if config_name == self.alas_name:
|
|||
|
self.expand_menu()
|
|||
|
return
|
|||
|
self.init_aside(name=config_name)
|
|||
|
clear("content")
|
|||
|
self.alas_name = config_name
|
|||
|
self.alas_mod = get_config_mod(config_name)
|
|||
|
self.alas = ProcessManager.get_manager(config_name)
|
|||
|
self.alas_config = load_config(config_name)
|
|||
|
self.state_switch.switch()
|
|||
|
self.initial()
|
|||
|
self.alas_set_menu()
|
|||
|
|
|||
|
def ui_add_alas(self) -> None:
|
|||
|
with popup(t("Gui.AddAlas.PopupTitle")) as s:
|
|||
|
|
|||
|
def get_unused_name():
|
|||
|
all_name = alas_instance()
|
|||
|
for i in range(2, 100):
|
|||
|
if f"alas{i}" not in all_name:
|
|||
|
return f"alas{i}"
|
|||
|
else:
|
|||
|
return ""
|
|||
|
|
|||
|
def add():
|
|||
|
name = pin["AddAlas_name"]
|
|||
|
origin = pin["AddAlas_copyfrom"]
|
|||
|
|
|||
|
if name in alas_instance():
|
|||
|
err = "Gui.AddAlas.FileExist"
|
|||
|
elif set(name) & set(".\\/:*?\"'<>|"):
|
|||
|
err = "Gui.AddAlas.InvalidChar"
|
|||
|
elif name.lower().startswith("template"):
|
|||
|
err = "Gui.AddAlas.InvalidPrefixTemplate"
|
|||
|
else:
|
|||
|
err = ""
|
|||
|
if err:
|
|||
|
clear(s)
|
|||
|
put(name, origin)
|
|||
|
put_error(t(err), scope=s)
|
|||
|
return
|
|||
|
|
|||
|
r = load_config(origin).read_file(origin)
|
|||
|
State.config_updater.write_file(name, r, get_config_mod(origin))
|
|||
|
self.set_aside()
|
|||
|
self.active_button("aside", self.alas_name)
|
|||
|
close_popup()
|
|||
|
|
|||
|
def put(name=None, origin=None):
|
|||
|
put_input(
|
|||
|
name="AddAlas_name",
|
|||
|
label=t("Gui.AddAlas.NewName"),
|
|||
|
value=name or get_unused_name(),
|
|||
|
scope=s,
|
|||
|
),
|
|||
|
put_select(
|
|||
|
name="AddAlas_copyfrom",
|
|||
|
label=t("Gui.AddAlas.CopyFrom"),
|
|||
|
options=alas_template() + alas_instance(),
|
|||
|
value=origin or "template-alas",
|
|||
|
scope=s,
|
|||
|
),
|
|||
|
put_button(label=t("Gui.AddAlas.Confirm"), onclick=add, scope=s)
|
|||
|
|
|||
|
put()
|
|||
|
|
|||
|
def show(self) -> None:
|
|||
|
self._show()
|
|||
|
self.set_aside()
|
|||
|
self.init_aside(name="Home")
|
|||
|
self.dev_set_menu()
|
|||
|
self.init_menu(name="HomePage")
|
|||
|
self.alas_name = ""
|
|||
|
if hasattr(self, "alas"):
|
|||
|
del self.alas
|
|||
|
self.set_status(0)
|
|||
|
|
|||
|
def set_language(l):
|
|||
|
lang.set_language(l)
|
|||
|
self.show()
|
|||
|
|
|||
|
def set_theme(t):
|
|||
|
self.set_theme(t)
|
|||
|
run_js("location.reload()")
|
|||
|
|
|||
|
with use_scope("content"):
|
|||
|
put_text("Select your language / 选择语言").style("text-align: center")
|
|||
|
put_buttons(
|
|||
|
[
|
|||
|
{"label": "简体中文", "value": "zh-CN"},
|
|||
|
{"label": "繁體中文", "value": "zh-TW"},
|
|||
|
{"label": "English", "value": "en-US"},
|
|||
|
{"label": "日本語", "value": "ja-JP"},
|
|||
|
],
|
|||
|
onclick=lambda l: set_language(l),
|
|||
|
).style("text-align: center")
|
|||
|
put_text("Change theme / 更改主题").style("text-align: center")
|
|||
|
put_buttons(
|
|||
|
[
|
|||
|
{"label": "Light", "value": "default", "color": "light"},
|
|||
|
{"label": "Dark", "value": "dark", "color": "dark"},
|
|||
|
],
|
|||
|
onclick=lambda t: set_theme(t),
|
|||
|
).style("text-align: center")
|
|||
|
|
|||
|
# show something
|
|||
|
put_markdown(
|
|||
|
"""
|
|||
|
Alas is a free open source software, if you paid for Alas from any channel, please refund.
|
|||
|
Alas 是一款免费开源软件,如果你在任何渠道付费购买了Alas,请退款。
|
|||
|
Project repository 项目地址:`https://github.com/LmeSzinc/AzurLaneAutoScript`
|
|||
|
"""
|
|||
|
).style("text-align: center")
|
|||
|
|
|||
|
if lang.TRANSLATE_MODE:
|
|||
|
lang.reload()
|
|||
|
|
|||
|
def _disable():
|
|||
|
lang.TRANSLATE_MODE = False
|
|||
|
self.show()
|
|||
|
|
|||
|
toast(
|
|||
|
_t("Gui.Toast.DisableTranslateMode"),
|
|||
|
duration=0,
|
|||
|
position="right",
|
|||
|
onclick=_disable,
|
|||
|
)
|
|||
|
|
|||
|
def run(self) -> None:
|
|||
|
# setup gui
|
|||
|
set_env(title="Alas", output_animation=False)
|
|||
|
add_css(filepath_css("alas"))
|
|||
|
if self.is_mobile:
|
|||
|
add_css(filepath_css("alas-mobile"))
|
|||
|
else:
|
|||
|
add_css(filepath_css("alas-pc"))
|
|||
|
|
|||
|
if self.theme == "dark":
|
|||
|
add_css(filepath_css("dark-alas"))
|
|||
|
else:
|
|||
|
add_css(filepath_css("light-alas"))
|
|||
|
|
|||
|
# Auto refresh when lost connection
|
|||
|
# [For develop] Disable by run `reload=0` in console
|
|||
|
run_js(
|
|||
|
"""
|
|||
|
reload = 1;
|
|||
|
WebIO._state.CurrentSession.on_session_close(
|
|||
|
()=>{
|
|||
|
setTimeout(
|
|||
|
()=>{
|
|||
|
if (reload == 1){
|
|||
|
location.reload();
|
|||
|
}
|
|||
|
}, 4000
|
|||
|
)
|
|||
|
}
|
|||
|
);
|
|||
|
"""
|
|||
|
)
|
|||
|
|
|||
|
aside = get_localstorage("aside")
|
|||
|
self.show()
|
|||
|
|
|||
|
# init config watcher
|
|||
|
self._init_alas_config_watcher()
|
|||
|
|
|||
|
# save config
|
|||
|
_thread_save_config = threading.Thread(target=self._alas_thread_update_config)
|
|||
|
register_thread(_thread_save_config)
|
|||
|
_thread_save_config.start()
|
|||
|
|
|||
|
visibility_state_switch = Switch(
|
|||
|
status={
|
|||
|
True: [
|
|||
|
lambda: self.__setattr__("visible", True),
|
|||
|
lambda: self.alas_update_overview_task()
|
|||
|
if self.page == "Overview"
|
|||
|
else 0,
|
|||
|
lambda: self.task_handler._task.__setattr__("delay", 15),
|
|||
|
],
|
|||
|
False: [
|
|||
|
lambda: self.__setattr__("visible", False),
|
|||
|
lambda: self.task_handler._task.__setattr__("delay", 1),
|
|||
|
],
|
|||
|
},
|
|||
|
get_state=get_window_visibility_state,
|
|||
|
name="visibility_state",
|
|||
|
)
|
|||
|
|
|||
|
self.state_switch = Switch(
|
|||
|
status=self.set_status,
|
|||
|
get_state=lambda: getattr(getattr(self, "alas", -1), "state", 0),
|
|||
|
name="state",
|
|||
|
)
|
|||
|
|
|||
|
def goto_update():
|
|||
|
self.ui_develop()
|
|||
|
self.dev_update()
|
|||
|
|
|||
|
update_switch = Switch(
|
|||
|
status={
|
|||
|
1: lambda: toast(
|
|||
|
t("Gui.Toast.ClickToUpdate"),
|
|||
|
duration=0,
|
|||
|
position="right",
|
|||
|
color="success",
|
|||
|
onclick=goto_update,
|
|||
|
)
|
|||
|
},
|
|||
|
get_state=lambda: updater.state,
|
|||
|
name="update_state",
|
|||
|
)
|
|||
|
|
|||
|
self.task_handler.add(self.state_switch.g(), 2)
|
|||
|
self.task_handler.add(visibility_state_switch.g(), 15)
|
|||
|
self.task_handler.add(update_switch.g(), 1)
|
|||
|
self.task_handler.start()
|
|||
|
|
|||
|
# Return to previous page
|
|||
|
if aside not in ["Home", None]:
|
|||
|
self.ui_alas(aside)
|
|||
|
|
|||
|
|
|||
|
def debug():
|
|||
|
"""For interactive python.
|
|||
|
$ python3
|
|||
|
>>> from module.webui.app import *
|
|||
|
>>> debug()
|
|||
|
>>>
|
|||
|
"""
|
|||
|
startup()
|
|||
|
AlasGUI().run()
|
|||
|
|
|||
|
|
|||
|
def startup():
|
|||
|
State.init()
|
|||
|
lang.reload()
|
|||
|
updater.event = State.manager.Event()
|
|||
|
if updater.delay > 0:
|
|||
|
task_handler.add(updater.check_update, updater.delay)
|
|||
|
task_handler.add(updater.schedule_update(), 86400)
|
|||
|
task_handler.start()
|
|||
|
# if State.deploy_config.DiscordRichPresence:
|
|||
|
# init_discord_rpc()
|
|||
|
# if State.deploy_config.StartOcrServer:
|
|||
|
# start_ocr_server_process(State.deploy_config.OcrServerPort)
|
|||
|
if (
|
|||
|
State.deploy_config.EnableRemoteAccess
|
|||
|
and State.deploy_config.Password is not None
|
|||
|
):
|
|||
|
task_handler.add(RemoteAccess.keep_ssh_alive(), 60)
|
|||
|
|
|||
|
|
|||
|
def clearup():
|
|||
|
"""
|
|||
|
Notice: Ensure run it before uvicorn reload app,
|
|||
|
all process will NOT EXIT after close electron app.
|
|||
|
"""
|
|||
|
logger.info("Start clearup")
|
|||
|
RemoteAccess.kill_ssh_process()
|
|||
|
# close_discord_rpc()
|
|||
|
# stop_ocr_server_process()
|
|||
|
for alas in ProcessManager._processes.values():
|
|||
|
alas.stop()
|
|||
|
State.clearup()
|
|||
|
task_handler.stop()
|
|||
|
logger.info("Alas closed.")
|
|||
|
|
|||
|
|
|||
|
def app():
|
|||
|
parser = argparse.ArgumentParser(description="Alas web service")
|
|||
|
parser.add_argument(
|
|||
|
"-k", "--key", type=str, help="Password of alas. No password by default"
|
|||
|
)
|
|||
|
parser.add_argument(
|
|||
|
"--cdn",
|
|||
|
action="store_true",
|
|||
|
help="Use jsdelivr cdn for pywebio static files (css, js). Self host cdn by default.",
|
|||
|
)
|
|||
|
parser.add_argument(
|
|||
|
"--run",
|
|||
|
nargs="+",
|
|||
|
type=str,
|
|||
|
help="Run alas by config names on startup",
|
|||
|
)
|
|||
|
args, _ = parser.parse_known_args()
|
|||
|
|
|||
|
# Apply config
|
|||
|
AlasGUI.set_theme(theme=State.deploy_config.Theme)
|
|||
|
lang.LANG = State.deploy_config.Language
|
|||
|
key = args.key or State.deploy_config.Password
|
|||
|
cdn = args.cdn if args.cdn else State.deploy_config.CDN
|
|||
|
runs = None
|
|||
|
if args.run:
|
|||
|
runs = args.run
|
|||
|
elif State.deploy_config.Run:
|
|||
|
# TODO: refactor poor_yaml_read() to support list
|
|||
|
tmp = State.deploy_config.Run.split(",")
|
|||
|
runs = [l.strip(" ['\"]") for l in tmp if len(l)]
|
|||
|
instances: List[str] = runs
|
|||
|
|
|||
|
logger.hr("Webui configs")
|
|||
|
logger.attr("Theme", State.deploy_config.Theme)
|
|||
|
logger.attr("Language", lang.LANG)
|
|||
|
logger.attr("Password", True if key else False)
|
|||
|
logger.attr("CDN", cdn)
|
|||
|
|
|||
|
def index():
|
|||
|
if key is not None and not login(key):
|
|||
|
logger.warning(f"{info.user_ip} login failed.")
|
|||
|
time.sleep(1.5)
|
|||
|
run_js("location.reload();")
|
|||
|
return
|
|||
|
gui = AlasGUI()
|
|||
|
local.gui = gui
|
|||
|
gui.run()
|
|||
|
|
|||
|
app = asgi_app(
|
|||
|
applications=[index],
|
|||
|
cdn=cdn,
|
|||
|
static_dir=None,
|
|||
|
debug=True,
|
|||
|
on_startup=[
|
|||
|
startup,
|
|||
|
lambda: ProcessManager.restart_processes(
|
|||
|
instances=instances, ev=updater.event
|
|||
|
),
|
|||
|
],
|
|||
|
on_shutdown=[clearup],
|
|||
|
)
|
|||
|
|
|||
|
return app
|