350 lines
11 KiB
Python
350 lines
11 KiB
Python
# -------------------------------
|
|
# Difficult Rocket
|
|
# Copyright © 2020-2023 by shenjackyuanjie 3695888@qq.com
|
|
# All rights reserved
|
|
# -------------------------------
|
|
|
|
"""
|
|
writen by shenjackyuanjie
|
|
mail: 3695888@qq.com
|
|
github: @shenjackyuanjie
|
|
gitee: @shenjackyuanjie
|
|
"""
|
|
|
|
# system function
|
|
import time
|
|
|
|
from typing import Union, Optional, Tuple
|
|
from decimal import Decimal
|
|
|
|
# from pyglet
|
|
import pyglet
|
|
from pyglet.text import Label
|
|
from pyglet.window import key
|
|
from pyglet.gui import widgets
|
|
from pyglet.graphics import Batch, Group
|
|
|
|
# from DR
|
|
from Difficult_Rocket.utils import translate
|
|
from Difficult_Rocket.command.api import CommandText
|
|
from Difficult_Rocket.utils.thread import new_thread
|
|
|
|
|
|
class CommandLineTextEntry(widgets.TextEntry):
|
|
"""
|
|
基于 Text Entry 重写的 Command Line
|
|
"""
|
|
|
|
def __init__(
|
|
self,
|
|
x: int,
|
|
y: int,
|
|
width: int,
|
|
color: Optional[Tuple[int, int, int, int]] = (255, 255, 255, 255),
|
|
text_color: Optional[Tuple[int, int, int, int]] = (0, 0, 0, 255),
|
|
caret_color: Optional[Tuple[int, int, int, int]] = (0, 0, 0),
|
|
batch: Optional[Batch] = None,
|
|
group: Optional[Group] = None,
|
|
):
|
|
super().__init__(
|
|
x=x,
|
|
y=y,
|
|
width=width,
|
|
color=color,
|
|
text_color=text_color,
|
|
caret_color=caret_color,
|
|
batch=batch,
|
|
group=group,
|
|
text="",
|
|
)
|
|
...
|
|
|
|
|
|
class CommandLine(widgets.WidgetBase):
|
|
"""
|
|
command line show
|
|
"""
|
|
|
|
def __init__(
|
|
self,
|
|
x: int,
|
|
y: int,
|
|
width: int,
|
|
height: int,
|
|
length: int,
|
|
batch: Batch,
|
|
group: Group = None,
|
|
command_text: str = "/",
|
|
font_size: int = 20,
|
|
):
|
|
super().__init__(x, y, width, height)
|
|
|
|
# normal values
|
|
self.length = length
|
|
self._command_list = ["" for _ in range(length)]
|
|
self._command_text = command_text
|
|
self._text_position = 0
|
|
self._command_view = 0
|
|
self._value = 0
|
|
self._text = ""
|
|
self.command_split = 25
|
|
self.command_distance = 20
|
|
|
|
# group
|
|
self._user_group = group
|
|
bg_group = Group(order=0, parent=group)
|
|
fg_group = Group(order=1, parent=group)
|
|
|
|
# hidden value
|
|
self._text = ""
|
|
self._line = Label(
|
|
x=x,
|
|
y=y,
|
|
batch=batch,
|
|
text=self.text,
|
|
color=(100, 255, 255, 255),
|
|
anchor_x="left",
|
|
anchor_y="bottom",
|
|
font_size=font_size,
|
|
font_name=translate.微软等宽,
|
|
group=fg_group,
|
|
)
|
|
self._label = [
|
|
Label(
|
|
x=x + 10,
|
|
y=y + self.command_distance + (line * self.command_split),
|
|
batch=batch,
|
|
text="a",
|
|
anchor_x="left",
|
|
anchor_y="bottom",
|
|
font_size=font_size - 3,
|
|
font_name=translate.鸿蒙简体,
|
|
group=bg_group,
|
|
)
|
|
for line in range(length)
|
|
]
|
|
# Rectangular outline with 5-pixel pad:
|
|
color = (100, 100, 100, 100)
|
|
self._pad = p = 5
|
|
self._outline = pyglet.shapes.Rectangle(
|
|
x=x - p,
|
|
y=y - p,
|
|
width=width + p,
|
|
height=height + p,
|
|
color=color[:3],
|
|
batch=batch,
|
|
group=fg_group,
|
|
)
|
|
self._outline.opacity = color[3]
|
|
|
|
self.editing = False
|
|
|
|
def _update_position(self):
|
|
pass
|
|
|
|
def update_groups(self, order):
|
|
self._line.group = Group(order=order + 1, parent=self._user_group)
|
|
for label in self._label:
|
|
label.group = Group(order=order + 1, parent=self._user_group)
|
|
self._outline.group = Group(order=order + 2, parent=self._user_group)
|
|
|
|
"""
|
|
values
|
|
"""
|
|
|
|
@property
|
|
def value(self):
|
|
return self.text
|
|
|
|
@property
|
|
def text(self):
|
|
return self._text
|
|
|
|
@text.setter
|
|
def text(self, value):
|
|
assert isinstance(value, str), "CommandLine's text must be string!"
|
|
self._text = value
|
|
self._line.text = value
|
|
|
|
@property
|
|
def command_view(self):
|
|
return self._command_view
|
|
|
|
@command_view.setter
|
|
def command_view(self, value):
|
|
"""
|
|
value:
|
|
-1 -> 将整个列表添加一个数据
|
|
如果长度超过length就删掉多余的
|
|
将视角移动到最下面，刷新显示列表
|
|
0 ~ (self.length-1) -> 切换视角到对应的行数
|
|
实际上还有一个限制
|
|
"""
|
|
assert isinstance(value, int), "Command View must be integer"
|
|
assert (
|
|
-self.length < value < self.length
|
|
), f"Command View must be bigger than {-self.length} and smaller than {self.length}"
|
|
if value == -1: # flush command list
|
|
self._label.insert(0, self._label[-1])
|
|
self._label.pop(-1)
|
|
for line in range(self.length):
|
|
self._label[line].y = (
|
|
self.y + self.command_distance + (line * self.command_split)
|
|
)
|
|
self._label[0].text = self.text
|
|
self.text = ""
|
|
self._command_view = 0
|
|
self._text_position = 0
|
|
elif value == self._command_view: # not doing anything
|
|
pass
|
|
elif value > self._command_view: # move upwards
|
|
pass
|
|
else: # move downwards
|
|
pass
|
|
# self._command_view = value
|
|
|
|
@property
|
|
def editing(self):
|
|
return self._editing
|
|
|
|
@editing.setter
|
|
def editing(self, value):
|
|
assert isinstance(value, bool), "Command editing must be bool!"
|
|
self._editing = value
|
|
self._line.visible = value
|
|
self._outline.visible = value
|
|
for label in self._label:
|
|
label.visible = value
|
|
|
|
@new_thread("command wait", daemon=True, log_thread=False)
|
|
def wait(self, wait: Union[float, int] = 0):
|
|
this = self._label[0]
|
|
self._label[0].visible = True
|
|
time.sleep(wait)
|
|
if self._label[0].visible and not self.editing:
|
|
while (
|
|
(self._label[0].opacity >= 30)
|
|
and self._label[0].visible
|
|
and (self._label[0] is this)
|
|
and not self.editing
|
|
):
|
|
# (label 的透明度不是 0) and (label 还在显示) and (label 还是载入线程时候的那个label) and (现在不在输入新行)
|
|
self._label[0].opacity -= 2
|
|
time.sleep(0.01)
|
|
if self._label[0] is this: # 如果结束的时候label还是这个label
|
|
self._label[0].opacity = 255
|
|
else: # 如果不是就赶快找回来！
|
|
now = self._label.index(this)
|
|
self._label[now].opacity = 255
|
|
if not self.editing: # 如果不在编辑再隐藏
|
|
self._label[0].visible = False
|
|
|
|
@property
|
|
def command_list(self):
|
|
return self._command_list
|
|
|
|
"""
|
|
events
|
|
"""
|
|
|
|
def on_text(self, text):
|
|
# 这里的大部分东西都会在最近被重写
|
|
# TODO 重写成基于新的 InputBox 的解析
|
|
if self.editing:
|
|
if text in ("\r", "\n"): # goto a new line
|
|
if not self.text:
|
|
pass
|
|
elif self.text[0] == self._command_text:
|
|
self.dispatch_event("on_command", CommandText(self.text[1:]))
|
|
else:
|
|
self.dispatch_event("on_message", CommandText(self.text))
|
|
# on_message 和 on_command 可能会覆盖 self.text 需要再次判定
|
|
if self.text:
|
|
self.command_view = -1
|
|
self.editing = False
|
|
self.wait()
|
|
else:
|
|
self.text = f"{self.text[:self._text_position]}{text}{self.text[self._text_position:]}" # 插入字符（简单粗暴）
|
|
self._text_position += 1
|
|
elif text == "t": # open message line
|
|
self.editing = True
|
|
elif text == "/": # open command line
|
|
self.editing = True
|
|
self.text = "/"
|
|
self._text_position = 1
|
|
|
|
def on_text_motion(self, motion):
|
|
if self.editing:
|
|
# edit motion
|
|
if motion == key.MOTION_DELETE: # 确保不越界
|
|
self.text = f"{self.text[:self._text_position]}{self.text[self._text_position + 1:]}" # 简单粗暴的删除
|
|
elif (
|
|
motion == key.MOTION_BACKSPACE and self._text_position >= 1
|
|
): # 确保不越界
|
|
self.text = f"{self.text[:self._text_position - 1]}{self.text[self._text_position:]}" # 简单粗暴的删除
|
|
self._text_position -= 1 # 记得切换光标位置
|
|
|
|
# move motion
|
|
elif motion == key.MOTION_LEFT and self._text_position >= 0: # 确保不越界
|
|
self._text_position -= 1
|
|
elif motion == key.MOTION_RIGHT and self._text_position <= len(
|
|
self.text
|
|
): # 确保不越界
|
|
self._text_position += 1
|
|
elif motion in (
|
|
key.MOTION_BEGINNING_OF_LINE,
|
|
key.MOTION_BEGINNING_OF_FILE,
|
|
key.MOTION_PREVIOUS_PAGE,
|
|
):
|
|
self._text_position = 0
|
|
elif motion in (
|
|
key.MOTION_END_OF_LINE,
|
|
key.MOTION_END_OF_FILE,
|
|
key.MOTION_NEXT_PAGE,
|
|
):
|
|
self._text_position = len(self.text)
|
|
|
|
# view move motion
|
|
elif motion == key.MOTION_DOWN:
|
|
if self.command_view != -1:
|
|
self.command_view -= 1
|
|
else:
|
|
pass
|
|
|
|
def on_text_motion_select(self, motion):
|
|
if self.editing:
|
|
pass
|
|
|
|
def on_mouse_drag(self, x, y, dx, dy, buttons, modifiers):
|
|
if self.editing:
|
|
pass
|
|
|
|
def on_mouse_press(self, x, y, buttons, modifiers):
|
|
if self.editing:
|
|
pass
|
|
|
|
"""
|
|
custom event
|
|
"""
|
|
|
|
def on_command(self, command: text):
|
|
if self.editing:
|
|
return
|
|
"""give command to it"""
|
|
|
|
def on_message(self, message: text):
|
|
if self.editing:
|
|
return
|
|
"""give message to it"""
|
|
|
|
def push_line(self, line: Union[str, int, float, Decimal], block_line: bool = False):
|
|
_text = self.text
|
|
self.text = str(line)
|
|
self.command_view = -1
|
|
if not block_line:
|
|
self.text = _text
|
|
|
|
|
|
CommandLine.register_event_type("on_command")
|
|
CommandLine.register_event_type("on_message")
|