From 2019f16e562df22dd481eae7f894d8fdb89e9ac8 Mon Sep 17 00:00:00 2001
From: "xiongdian.me" <xiongdian.me@bytedance.com>
Date: Tue, 9 May 2023 12:47:52 +0800
Subject: [PATCH] riot: add . for .ibd and non for proxied

---
 scripts/riot | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/scripts/riot b/scripts/riot
index 941fb73..702a9c2 100755
--- a/scripts/riot
+++ b/scripts/riot
@@ -13,8 +13,11 @@ get_server_meta()
 {
     local domain=${1##*.}
     local host=${1%.*}
+    if [[ -z "$domain" ]]; then
+        domain="ibd"
+    fi
     if [[ "$host" == "$domain" ]]; then
-        domain=""
+        domain="proxied"
     fi
     case $domain in
         ibd|ebd )
@@ -28,7 +31,7 @@ get_server_meta()
             SSH_USERNAME=dictxiong
             SSH_OPTIONS='-o ProxyJump="ssh@nasp.ob.ac.cn:36022"'
             ;;
-        "" )
+        proxied )
             SERVER=proxy.beardic.cn
             local tmp=$(sha256sum <<< "$host" | tr -cd "[:digit:]")
             tmp=${tmp:0:4}