From 41ae8591c4b4410e720d0ca66578b928252c2388 Mon Sep 17 00:00:00 2001
From: xiaoz <xiaoz93@outlook.com>
Date: Thu, 9 Dec 2021 22:47:38 +0800
Subject: [PATCH] =?UTF-8?q?=E4=B8=80=E4=BA=9B=E4=BC=98=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 node_exporter.sh | 32 ++++++++++++++++++++++++++------
 1 file changed, 26 insertions(+), 6 deletions(-)

diff --git a/node_exporter.sh b/node_exporter.sh
index ecb9ee7..39dd0d8 100644
--- a/node_exporter.sh
+++ b/node_exporter.sh
@@ -14,19 +14,21 @@ INSTALL_PATH="/opt/node_exporter"
 
 #安装前准备
 depend(){
+	echo "Pre-installation preparation is in progress..."
 	if [ -e "/usr/bin/yum" ]
 	then
-		yum -y install wget
+		yum -y install wget curl
 	else
 		#更新软件，否则可能make命令无法安装
 		apt-get -y update
-		apt-get install -y wget
+		apt-get install -y wget curl
 	fi
 	#创建目录
 	#mkdir -p ${INSTALL_PATH}
 }
 #下载
 download(){
+	echo "Ready to download the installation package..."
 	wget -P /opt http://soft.xiaoz.org/linux/node_exporter-${VERSION}.linux-amd64.tar.gz
 	cd /opt && tar -xvf node_exporter-${VERSION}.linux-amd64.tar.gz
 	mv node_exporter-${VERSION}.linux-amd64 node_exporter
@@ -34,7 +36,7 @@ download(){
 
 #一些额外的配置
 setting(){
-	
+	#设置密码访问
 cat >> $INSTALL_PATH/config.yaml << EOF
 basic_auth_users:
   $USERNAME_PASSWORD
@@ -44,8 +46,19 @@ EOF
 
 #放行端口
 release_port(){
-	firewall-cmd --zone=public --add-port=29100/tcp --permanent
-	firewall-cmd --reload
+	echo "Detecting firewall type..."
+	#检测防火墙类型
+	which firewall-cmd
+	if [ $? -eq 0 ]
+	then
+		firewall-cmd --zone=public --add-port=29100/tcp --permanent
+		firewall-cmd --reload
+	fi
+	which ufw
+	if [ $? -eq 0 ]
+	then
+		ufw allow 29100/tcp
+	fi
 }
 
 
@@ -68,10 +81,17 @@ WantedBy=default.target" > /etc/systemd/system/node_exporter.service
 	systemctl start node_exporter.service
 }
 
+#安装完成
+install_success(){
+	myip=$(curl ip.sb)
+	echo "Installation is complete, please visit http://${myip}:29100"
+}
+
 #清理工作
 clean_work() {
+	echo "Cleaning installation packages..."
 	rm -rf /opt/node_exporter-${VERSION}.linux-amd64.tar.gz
 }
 
-depend && download && setting && release_port && reg_systemd
+depend && download && setting && release_port && reg_systemd && install_success
 clean_work
\ No newline at end of file