Mirai Console UI

mirai-console/build.gradle.kts

@@ -42,5 +42,6 @@ dependencies {
     api(group = "com.alibaba", name = "fastjson", version = "1.2.62")
     api(group = "org.yaml", name = "snakeyaml", version = "1.25")
     api(group = "com.moandjiezana.toml", name = "toml4j", version = "0.7.2")
+    api(group = "com.googlecode.lanterna", name = "lanterna", version = "3.0.2")
     // classpath is not set correctly by IDE
 }
\ No newline at end of file

mirai-console/src/main/kotlin/net/mamoe/mirai/MiraiConsole.kt

@@ -17,6 +17,7 @@ import net.mamoe.mirai.plugins.withDefaultWriteSave
 import net.mamoe.mirai.api.http.MiraiHttpAPIServer
 import net.mamoe.mirai.api.http.generateSessionKey
 import net.mamoe.mirai.contact.sendMessage
+import net.mamoe.mirai.utils.MiraiLogger
 import java.io.File
 import kotlin.concurrent.thread
 
@@ -37,7 +38,7 @@ object MiraiConsole {
         get() = PluginManager
 
     var logger: MiraiConsoleLogger =
-        DefaultLogger
+        UIPushLogger()
 
     var path: String = System.getProperty("user.dir")
 
@@ -254,14 +255,11 @@ object MiraiConsole {
         }
     }
 
-    interface MiraiConsoleLogger {
-        operator fun invoke(any: Any? = null)
-    }
-
-    object DefaultLogger : MiraiConsoleLogger {
+    class UIPushLogger(override val identity: String?, override var follower: MiraiLogger?) : MiraiLogger {
         override fun invoke(any: Any?) {
+            MiraiConsoleUI.start()
             if (any != null) {
-                println("[Mirai$version $build]: " + any.toString())
+                MiraiConsoleUI.pushLog(0, "[Mirai$version $build]: $any")
             }
         }
     }
@@ -287,6 +285,7 @@ class MiraiConsoleLoader {
             Runtime.getRuntime().addShutdownHook(thread(start = false) {
                 MiraiConsole.stop()
             })
+
         }
     }
 }

mirai-console/src/main/kotlin/net/mamoe/mirai/MiraiConsoleUI.kt

+package net.mamoe.mirai
+
+import com.googlecode.lanterna.SGR
+import com.googlecode.lanterna.TerminalSize
+import com.googlecode.lanterna.TextColor
+import com.googlecode.lanterna.graphics.TextGraphics
+import com.googlecode.lanterna.input.KeyStroke
+import com.googlecode.lanterna.input.KeyType
+import com.googlecode.lanterna.terminal.DefaultTerminalFactory
+import com.googlecode.lanterna.terminal.Terminal
+import com.googlecode.lanterna.terminal.TerminalResizeListener
+import com.googlecode.lanterna.terminal.swing.SwingTerminal
+import com.googlecode.lanterna.terminal.swing.SwingTerminalFrame
+import java.lang.StringBuilder
+import java.util.*
+import kotlin.concurrent.thread
+
+
+object MiraiConsoleUI {
+
+    val log = mutableMapOf<Long, LimitLinkedQueue<String>>().also { it[0L] = LimitLinkedQueue(50) }
+
+
+    private val screens = mutableListOf(0L)
+    private var currentScreenId = 0
+
+    fun addBotScreen(uin: Long) {
+        screens.add(uin)
+        log[uin] = LimitLinkedQueue()
+    }
+
+    fun pushLog(uin: Long, str: String) {
+        log[uin]!!.push(str)
+    }
+
+    var hasStart = false
+    fun start() {
+        if (hasStart) {
+            return
+        }
+        hasStart = true
+        val defaultTerminalFactory = DefaultTerminalFactory()
+        var terminal: Terminal? = null
+        try {
+            terminal = defaultTerminalFactory.createTerminal()
+            terminal.enterPrivateMode()
+            terminal.clearScreen()
+            terminal.setCursorVisible(false)
+        } catch (e: Exception) {
+            terminal = SwingTerminalFrame("Mirai Console")
+            terminal.enterPrivateMode()
+            terminal.clearScreen()
+            terminal.setCursorVisible(false)
+        }
+        if (terminal == null) {
+            error("can not create terminal")
+        }
+
+        val textGraphics: TextGraphics = terminal.newTextGraphics()
+
+        try {
+            fun getLeftScreenId(): Int {
+                var newId = currentScreenId - 1
+                if (newId < 0) {
+                    newId = screens.size - 1
+                }
+                return newId
+            }
+
+            fun getRightScreenId(): Int {
+                var newId = 1 + currentScreenId
+                if (newId >= screens.size) {
+                    newId = 0
+                }
+                return newId
+            }
+
+            fun getScreenName(id: Int): String {
+                return when (screens[id]) {
+                    0L -> {
+                        "Console Screen"
+                    }
+                    else -> {
+                        "Bot: ${screens[id]}"
+                    }
+                }
+            }
+
+
+            var inited = false
+            fun clearRows(row: Int) {
+                textGraphics.putString(0, row, " ".repeat(terminal.terminalSize.columns))
+            }
+
+            fun drawFrame(
+                title: String
+            ) {
+                val width = terminal.terminalSize.columns
+                val height = terminal.terminalSize.rows
+                terminal.setBackgroundColor(TextColor.ANSI.DEFAULT)
+                if (!inited) {
+                    val mainTitle = "Mirai Console v0.01 Core v0.14"
+                    textGraphics.foregroundColor = TextColor.ANSI.WHITE
+                    textGraphics.backgroundColor = TextColor.ANSI.GREEN
+                    textGraphics.putString((width - mainTitle.length) / 2, 1, mainTitle, SGR.BOLD)
+                    textGraphics.foregroundColor = TextColor.ANSI.DEFAULT
+                    textGraphics.backgroundColor = TextColor.ANSI.DEFAULT
+                    textGraphics.putString(2, 3, "-".repeat(width - 4))
+                    textGraphics.putString(2, 5, "-".repeat(width - 4))
+                    textGraphics.putString(2, height - 4, "-".repeat(width - 4))
+                    textGraphics.putString(2, height - 3, "|>>>")
+                    textGraphics.putString(width - 3, height - 3, "|")
+                    textGraphics.putString(2, height - 2, "-".repeat(width - 4))
+                    inited = true
+                }
+                textGraphics.foregroundColor = TextColor.ANSI.DEFAULT
+                textGraphics.backgroundColor = TextColor.ANSI.DEFAULT
+                val leftName = getScreenName(getLeftScreenId())
+                clearRows(2)
+                textGraphics.putString((width - title.length) / 2 - "$leftName << ".length, 2, "$leftName << ")
+                textGraphics.foregroundColor = TextColor.ANSI.WHITE
+                textGraphics.backgroundColor = TextColor.ANSI.YELLOW
+                textGraphics.putString((width - title.length) / 2, 2, title, SGR.BOLD)
+                textGraphics.foregroundColor = TextColor.ANSI.DEFAULT
+                textGraphics.backgroundColor = TextColor.ANSI.DEFAULT
+                val rightName = getScreenName(getRightScreenId())
+                textGraphics.putString((width + title.length) / 2 + 1, 2, ">> $rightName")
+            }
+
+            fun drawMainFrame(
+                onlineBotCount: Number
+            ) {
+                drawFrame("Console Screen")
+                val width = terminal.terminalSize.columns
+                textGraphics.foregroundColor = TextColor.ANSI.DEFAULT
+                textGraphics.backgroundColor = TextColor.ANSI.DEFAULT
+                clearRows(4)
+                textGraphics.putString(2, 4, "|Online Bots: $onlineBotCount")
+                textGraphics.putString(
+                    width - 2 - "Powered By Mamoe Technologies|".length,
+                    4,
+                    "Powered By Mamoe Technologies|"
+                )
+            }
+
+            fun drawBotFrame(
+                qq: Long,
+                adminCount: Number
+            ) {
+                drawFrame("Bot: $qq")
+                val width = terminal.terminalSize.columns
+                textGraphics.foregroundColor = TextColor.ANSI.DEFAULT
+                textGraphics.backgroundColor = TextColor.ANSI.DEFAULT
+                clearRows(4)
+                textGraphics.putString(2, 4, "|Admins: $adminCount")
+                textGraphics.putString(width - 2 - "Add admins via commands|".length, 4, "Add admins via commands|")
+            }
+
+            fun drawLogs(values: List<String>) {
+                val width = terminal.terminalSize.columns - 6
+                val heightMin = 5
+
+                var currentHeight = terminal.terminalSize.rows - 5
+
+                for (index in heightMin until currentHeight) {
+                    clearRows(index)
+                }
+
+                values.forEach {
+                    if (currentHeight > heightMin) {
+                        var x = it
+                        while (currentHeight > heightMin) {
+                            if (x.isEmpty() || x.isBlank()) break
+                            textGraphics.foregroundColor = TextColor.ANSI.GREEN
+                            textGraphics.backgroundColor = TextColor.ANSI.DEFAULT
+                            val towrite = if (x.length > width) {
+                                x.substring(0, width).also {
+                                    x = x.substring(width)
+                                }
+                            } else {
+                                x.also {
+                                    x = ""
+                                }
+                            }
+                            textGraphics.putString(3, currentHeight, towrite, SGR.ITALIC)
+                            --currentHeight
+                        }
+                    }
+                }
+
+                if (terminal is SwingTerminalFrame) {
+                    terminal.flush()
+                }
+            }
+
+
+            var commandBuilder = StringBuilder()
+
+            fun redrawCommand() {
+                val height = terminal.terminalSize.rows
+                val width = terminal.terminalSize.columns
+                clearRows(height - 3)
+                textGraphics.foregroundColor = TextColor.ANSI.DEFAULT
+                textGraphics.putString(2, height - 3, "|>>>")
+                textGraphics.putString(width - 3, height - 3, "|")
+                textGraphics.foregroundColor = TextColor.ANSI.BLUE
+                textGraphics.putString(7, height - 3, commandBuilder.toString())
+                if (terminal is SwingTerminalFrame) {
+                    terminal.flush()
+                }
+            }
+
+            fun addCommandChar(
+                c: Char
+            ) {
+                if (commandBuilder.isEmpty() && c != '/') {
+                    addCommandChar('/')
+                }
+                textGraphics.foregroundColor = TextColor.ANSI.BLUE
+                val height = terminal.terminalSize.rows
+                commandBuilder.append(c)
+                if (terminal is SwingTerminalFrame) {
+                    redrawCommand()
+                } else {
+                    textGraphics.putString(6 + commandBuilder.length, height - 3, c.toString())
+                }
+            }
+
+            fun deleteCommandChar() {
+                if (!commandBuilder.isEmpty()) {
+                    commandBuilder = StringBuilder(commandBuilder.toString().substring(0, commandBuilder.length - 1))
+                }
+                val height = terminal.terminalSize.rows
+                if (terminal is SwingTerminalFrame) {
+                    redrawCommand()
+                } else {
+                    textGraphics.putString(7 + commandBuilder.length, height - 3, " ")
+                }
+            }
+
+
+            fun emptyCommand() {
+                commandBuilder = StringBuilder()
+                redrawCommand()
+                if (terminal is SwingTerminal) {
+                    terminal.flush()
+                }
+            }
+
+            fun update() {
+                when (screens[currentScreenId]) {
+                    0L -> {
+                        drawMainFrame(screens.size - 1)
+                    }
+                    else -> {
+                        drawBotFrame(screens[currentScreenId], 0)
+                    }
+                }
+                terminal.flush()
+
+            }
+
+            terminal.addResizeListener(TerminalResizeListener { terminal1: Terminal, newSize: TerminalSize ->
+                terminal.clearScreen()
+                inited = false
+                update()
+                redrawCommand()
+            })
+
+            update()
+
+            val charList = listOf(',', '.', '/', '@', '#', '$', '%', '^', '&', '*', '(', ')', '_', '=', '+', '!', ' ')
+            thread {
+                while (true) {
+                    var keyStroke: KeyStroke = terminal.readInput()
+
+                    when (keyStroke.keyType) {
+                        KeyType.ArrowLeft -> {
+                            currentScreenId = getLeftScreenId()
+                            update()
+                        }
+                        KeyType.ArrowRight -> {
+                            currentScreenId = getRightScreenId()
+                            update()
+                        }
+                        KeyType.Enter -> {
+                            emptyCommand()
+                        }
+                        else -> {
+                            if (keyStroke.character != null) {
+                                if (keyStroke.character.toInt() == 8) {
+                                    deleteCommandChar()
+                                }
+                                if (keyStroke.character.isLetterOrDigit() || charList.contains(keyStroke.character)) {
+                                    addCommandChar(keyStroke.character)
+                                }
+                            }
+                        }
+                    }
+                }
+            }
+
+        } catch (e: Exception) {
+            e.printStackTrace()
+        }
+    }
+}
+
+class LimitLinkedQueue<T>(
+    val limit: Int = 50
+) : LinkedList<T>() {
+    override fun push(e: T) {
+        if (size >= limit) {
+            pollLast()
+        }
+        super.push(e)
+    }
+}
\ No newline at end of file