Difficult-Rocket/Difficult_Rocket/command/line.py

293 lines
10 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.new_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 type(value) is 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 type(value) is 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 type(value) is 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')