From c0f0089f7a2fa2f41483991ebb6038dfa5f40009 Mon Sep 17 00:00:00 2001
From: benjobs <benjobs@qq.com>
Date: Thu, 26 May 2022 10:50:25 +0800
Subject: [PATCH] [KYUUBI #2743] colorfully kyuubi logo support

We can make the logo printed by Kyuubi on the terminal colorful and vibrant.

### Why are the changes needed?

- When TTY is available, print the colorfully logo, else print default

### _How was this patch tested?_
- [ ] Add some test cases that check the changes thoroughly including negative and positive cases if possible

- [ ] Add screenshots for manual tests if appropriate

- [ ] [Run test](https://kyuubi.apache.org/docs/latest/develop_tools/testing.html#running-tests) locally before make a pull request

Closes #2744 from wolfboys/master.

Closes #2743

2205e79c [benjobs] [KYUUBI #2743] colorfully kyuubi logo support

Authored-by: benjobs <benjobs@qq.com>
Signed-off-by: Cheng Pan <chengpan@apache.org>
---
 bin/kyuubi      |  2 +-
 bin/kyuubi-logo | 65 ++++++++++++++++++++++++++++++++++++++++++-------
 2 files changed, 57 insertions(+), 10 deletions(-)
 mode change 100644 => 100755 bin/kyuubi-logo

diff --git a/bin/kyuubi b/bin/kyuubi
index 0f1789e2a..082129e7a 100755
--- a/bin/kyuubi
+++ b/bin/kyuubi
@@ -36,7 +36,7 @@ if [[ "$@" = *--help ]] || [[ "$@" = *-h ]]; then
 fi
 
 function kyuubi_logo() {
-  cat ${KYUUBI_HOME}/bin/kyuubi-logo | grep -v "^#"
+  source ${KYUUBI_HOME}/bin/kyuubi-logo
 }
 
 function kyuubi_rotate_log() {
diff --git a/bin/kyuubi-logo b/bin/kyuubi-logo
old mode 100644
new mode 100755
index 431498cdc..15a45a4bb
--- a/bin/kyuubi-logo
+++ b/bin/kyuubi-logo
@@ -1,3 +1,4 @@
+#!/usr/bin/env bash
 #
 # Licensed to the Apache Software Foundation (ASF) under one or more
 # contributor license agreements.  See the NOTICE file distributed with
@@ -14,12 +15,58 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-  __  __                           __
- /\ \/\ \                         /\ \      __
- \ \ \/'/'  __  __  __  __  __  __\ \ \____/\_\
-  \ \ , <  /\ \/\ \/\ \/\ \/\ \/\ \\ \ '__`\/\ \
-   \ \ \\`\\ \ \_\ \ \ \_\ \ \ \_\ \\ \ \L\ \ \ \
-    \ \_\ \_\/`____ \ \____/\ \____/ \ \_,__/\ \_\
-     \/_/\/_/`/___/> \/___/  \/___/   \/___/  \/_/
-                /\___/
-                \/__/
+
+# Bugzilla 37848: When no TTY is available, don't output to console
+have_tty=0
+# shellcheck disable=SC2006
+if [[ "`tty`" != "not a tty" ]]; then
+    have_tty=1
+fi
+
+# Bugzilla 37848: When no TTY is available, don't output to console
+have_tty=0
+# shellcheck disable=SC2006
+if [[ "`tty`" != "not a tty" ]]; then
+    have_tty=1
+fi
+
+ # Only use colors if connected to a terminal
+if [[ ${have_tty} -eq 1 ]]; then
+  RAINBOW="
+    $(printf '\033[38;5;202m')
+    $(printf '\033[38;5;226m')
+    $(printf '\033[38;5;082m')
+    $(printf '\033[38;5;082m')
+    $(printf '\033[38;5;163m')
+    $(printf '\033[38;5;093m')
+  "
+  GREEN=$(printf '\033[32m')
+  YELLOW=$(printf '\033[33m')
+  BLUE=$(printf '\033[34m')
+  BLUE=$(printf '\033[34m')
+  RESET=$(printf '\033[0m')
+else
+  RAINBOW=""
+  GREEN=""
+  YELLOW=""
+  BLUE=""
+  BLUE=""
+  RESET=""
+fi
+
+print_logo() {
+  printf '\n\n'
+  printf '%s            __  __%s         %s        %s      %s     __    %s                      %s\n' $RAINBOW $RESET
+  printf '%s           /` \/` \%s         %s        %s      %s   /` \    %s   __                %s\n' $RAINBOW $RESET
+  printf '%s           \ \ \/ /%s    __  __%s  __  __%s   __ __%s\ \ \____%s /\_\               %s\n' $RAINBOW $RESET
+  printf '%s            \ \ , <  %s /` \/` \%s/` \/` \%s/` \/` \%s\ \ \__` %s\/` \              %s\n' $RAINBOW $RESET
+  printf '%s             \ \ \ \`\%s\ \ \_\ \%s\ \_\  \%s\ \_\  \%s\ \ \L\  %s\ \ \             %s\n' $RAINBOW $RESET
+  printf '%s              \ \_\ \_%s \/`____ \%s \____/ %s\ \____`%s\ \_,__/ %s\ \_\            %s\n' $RAINBOW $RESET
+  printf '%s               \/_/\/_/%s `/___/> \%s/___/   %s\/___/  %s\/___/   %s\/_/            %s\n' $RAINBOW $RESET
+  printf '%s                        %s   /\___/ %s        %s         %s           %s            %s\n' $RAINBOW $RESET
+  printf '%s                         %s  \/__\/  %s        %s         %s           %s           %s\n' $RAINBOW $RESET
+  printf '\n\n'
+}
+
+print_logo
+