blob: 46ad1f01be8de3540dd1d9a009fe363180f5ecd9 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
|
#!/bin/bash
{ # this ensures the entire script is downloaded #
# Find Java
if [[ -n "$JAVA_HOME" ]] && [[ -x "$JAVA_HOME/bin/java" ]]; then
JAVA_EXE="$JAVA_HOME/bin/java"
elif type -p java > /dev/null 2>&1; then
JAVA_EXE=$(type -p java)
elif [[ -x "/usr/bin/java" ]]; then
JAVA_EXE="/usr/bin/java"
else
echo -e "\e[1;31mUnable to find Java\e[0m"
exit 1
fi
echo "java: $JAVA_EXE"
#auto replace parameter start
WEB_TOKEN=${webToken}
DATACENTER_ID=${datacenterID}
TYPE=${type}
REMOTE_IP=${remoteIP}
REMOTE_PORT=${remotePort}
SSL_ENABLED=${sslEnabled}
if [ ! -z "$REMOTE_PORT" ];then
REMOTE_IP=$REMOTE_IP:$REMOTE_PORT
fi
LOCALHOST=${localhost}
#auto replace parameter end
ROOT_PATH=/opt/nezha
AGENT_TOKEN=$(cut -d '-' -f 1 /proc/sys/kernel/random/uuid)
INSTALL_PATH=$ROOT_PATH/nz-agent
WORK_PATH=/tmp/.nezha/nz-agent
rm -rf $WORK_PATH&&mkdir -p $WORK_PATH
function osInfo() {
if [ -e /etc/centos-release ];then
echo 'centos'
elif [ -e /etc/debian_version ]; then
echo 'debian'
elif [ -e /etc/lsb_release ]; then
echo 'ubuntu'
else
echo 'centos';
fi
}
CURRENT_OS=$(osInfo)
function log() {
case $1 in
"red")
echo -e "\e[1;31m$2\e[0m"
;;
"yellow")
echo -e "\e[1;33m$2\e[0m"
;;
"green")
echo -e "\e[1;32m$2\e[0m"
;;
*)
echo $2
;;
esac
}
function isInstall() {
case $CURRENT_OS in
'centos')
if [ $(rpm -aq |grep nz-agent|wc -l) -lt 1 ];then
echo 0
else
echo 1;
fi
;;
esac
}
function getPackageName() {
case $CURRENT_OS in
'centos')
echo "nz-agent.rpm"
;;
esac
}
PACKAGE_NAME=$(getPackageName $CURRENT_OS)
function downloadPackaget() {
cd $WORK_PATH;
log yellow "current os is $CURRENT_OS ,download package from http://$REMOTE_IP/agent/download?os=$CURRENT_OS"
if command -v wget >/dev/null 2>&1;then
if $SSL_ENABLED ;then
wget -O $PACKAGE_NAME --no-check-certificate --header="Authorization:"$WEB_TOKEN "https://$REMOTE_IP/agent/download?os="$CURRENT_OS
else
wget -O $PACKAGE_NAME --header="Authorization:"$WEB_TOKEN "http://$REMOTE_IP/agent/download?os="$CURRENT_OS
fi
return 0
elif command -v curl >/dev/null 2>&1;then
if $SSL_ENABLED ;then
curl -o $PACKAGE_NAME -k -H "Authorization:"$WEB_TOKEN "https://$REMOTE_IP/agent/download?os="$CURRENT_OS
else
curl -o $PACKAGE_NAME -H "Authorization:"$WEB_TOKEN "http://$REMOTE_IP/agent/download?os="$CURRENT_OS
fi
return 0
else
log red "The wget/curl command is required"
exit 1
fi
}
function installPackage() {
case $CURRENT_OS in
'centos')
log yellow 'install package right now...'
rpm -ivh $WORK_PATH/$PACKAGE_NAME && mkdir -p $INSTALL_PATH/config; echo -n $AGENT_TOKEN > $INSTALL_PATH/config/token.auth
;;
esac
AGENT_NAME=$(hostname)
if [ $AGENT_NAME == 'localhost.localdomain' ];then
AGENT_NAME=$(cut -d '-' -f 1 /proc/sys/kernel/random/uuid)
fi
if [ 0 -eq $? ];then
cat > $WORK_PATH/param.tmp <<EOF
{
"name":"$AGENT_NAME",
"dcId":$DATACENTER_ID,
"host":"$LOCALHOST",
"type":$TYPE,
"port":10090,
"token":"$AGENT_TOKEN"
}
EOF
log yellow 'now regist agent to system with information :'
cat $WORK_PATH/param.tmp
if $SSL_ENABLED ;then
local msg=`curl -s -k -X POST -H "Content-Type:application/json" -H "Authorization:"$WEB_TOKEN -d @$WORK_PATH/param.tmp "https://$REMOTE_IP/agent/register"`
else
local msg=`curl -s -X POST -H "Content-Type:application/json" -H "Authorization:"$WEB_TOKEN -d @$WORK_PATH/param.tmp "http://$REMOTE_IP/agent/register"`
fi
local code=`echo $msg |grep -E -o '"code":[0-9]+'|grep -E -o '[0-9]+'`
rm -rf $WORK_PATH/param.tmp
if [ 200 -eq $code ];then
log green 'regist agent success!'
else
log red 'regist agent failed,cause:'
echo $msg
exit 1
fi
else
log red "install nz-agent package may be has some error please try again or install by yourself"
exit 1
fi
}
if [ 1 -eq $(isInstall) ];then
log yellow "The nz-agent is installed , token is "$(cat $INSTALL_PATH/config/token.auth)
else
downloadPackaget && installPackage
log green "The nz-agent install finish,you can check agent info in WEB gui"
fi
} # this ensures the entire script is downloaded #
|
