Difficult-Rocket/Difficult_Rocket/command/line.py

#  -------------------------------
#  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")