forked from MNFaizi/Graphite-Project
-
Notifications
You must be signed in to change notification settings - Fork 0
/
run.sh
305 lines (251 loc) · 8.71 KB
/
run.sh
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
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
#!/bin/bash
# Initialize variables
script="neurons/validator.py" # replace this with the relative script path
proc_name="auto_update_graphite_validator"
args=()
version_location="graphite/__init__.py"
version="__version__"
repository="GraphiteAI/Graphite-Subnet"
repository_path="https://github.com/$repository"
old_args=$@
# Locate the path of the pm2 executable
pm2_path=$(which pm2)
# Check if pm2 is found
if [ -z "$pm2_path" ]; then
echo "pm2 could not be found. To install see: https://pm2.keymetrics.io/docs/usage/quick-start/"
exit 1
fi
# Extract the directory containing pm2
pm2_dir=$(dirname "$pm2_path")
# Export the directory to the PATH environment variable
export PATH="$pm2_dir:$PATH"
# Optionally, print the updated PATH
echo "pm2 directory added to PATH: $pm2_dir"
echo "Updated PATH: $PATH"
# Define your function for version comparison and other utilities here
# Checks if $1 is smaller than $2
version_less_than_or_equal() {
[ "$1" = "`echo -e "$1\n$2" | sort -V | head -n1`" ]
}
# Checks if $1 is smaller than $2
version_less_than() {
[ "$1" = "$2" ] && return 1 || version_less_than_or_equal $1 $2
}
are_versions_different() {
local version1="$1"
local version2="$2"
# Check if the versions are different
if [ "$version1" != "$version2" ]; then
return 0 # Return 0 (true) if versions are different
else
return 1 # Return 1 (false) if versions are the same
fi
}
# compares the local version vs the current version on the repository
get_version_difference() {
local tag1="$1"
local tag2="$2"
# Extract the version numbers from the tags
local version1=$(echo "$tag1" | sed 's/v//')
local version2=$(echo "$tag2" | sed 's/v//')
# Split the version numbers into an array
IFS='.' read -ra version1_arr <<< "$version1"
IFS='.' read -ra version2_arr <<< "$version2"
# Calculate the numerical difference
local diff=0
for i in "${!version1_arr[@]}"; do
local num1=${version1_arr[$i]}
local num2=${version2_arr[$i]}
# Compare the numbers and update the difference
if (( num1 > num2 )); then
diff=$((diff + num1 - num2))
elif (( num1 < num2 )); then
diff=$((diff + num2 - num1))
fi
done
strip_quotes $diff
}
# Check for the operating system
check_package_installed() {
local package_name="$1"
os_name=$(uname -s)
if [[ "$os_name" == "Linux" ]]; then
# Use dpkg-query to check if the package is installed
if dpkg-query -W -f='${Status}' "$package_name" 2>/dev/null | grep -q "installed"; then
return 1
else
return 0
fi
elif [[ "$os_name" == "Darwin" ]]; then
if brew list --formula | grep -q "^$package_name$"; then
return 1
else
return 0
fi
else
echo "Unknown operating system"
return 0
fi
}
check_variable_value_on_github() {
local repo="$1"
local file_path="$2"
local variable_name="$3"
local branch="$4"
# Redirect debug statements to stderr
echo "Debugging arguments:" >&2
echo "Repo: $repo" >&2
echo "File Path: $file_path" >&2
echo "Variable Name: $variable_name" >&2
echo "Branch: $branch" >&2
local url="https://api.github.com/repos/$repo/contents/$file_path?ref=$branch"
local response=$(curl -s "$url")
echo "URL: $url" >&2
# Check if the response contains an error message
if [[ $response =~ "message" ]]; then
echo "Error: Failed to retrieve file contents from GitHub." >&2
return 1
fi
# Sanitize JSON content to handle invalid control characters
sanitized_response=$(echo "$response" | tr -d '\000-\037')
# Extract the base64 content and decode it
json_content=$(echo "$sanitized_response" | jq -r '.content' | base64 --decode)
# Extract the desired variable value using grep and awk
variable_line=$(echo "$json_content" | grep -E "^${variable_name}\s*=\s*")
if [ -n "$variable_line" ]; then
variable_value=$(echo "$variable_line" | awk -F '=' '{print $2}' | tr -d '[:space:]')
variable_value=$(strip_quotes "$variable_value")
else
echo "Variable not found" >&2
return 1
fi
# Output only the variable value
echo "$variable_value"
}
strip_quotes() {
local input="$1"
# Remove leading and trailing quotes using parameter expansion
local stripped="${input#\"}"
stripped="${stripped%\"}"
echo "$stripped"
}
read_version_value() {
# Read each line in the file
while IFS= read -r line; do
# Check if the line contains the variable name
if [[ "$line" == *"$version"* ]]; then
# Extract the value of the variable
local value=$(echo "$line" | awk -F '=' '{print $2}' | tr -d ' ')
strip_quotes $value
return
fi
done < "$version_location"
echo "found version $value"
}
check_package_installed "jq"
if [ "$?" -ne 1 ]; then
echo "Missing 'jq'. Please install it first."
exit 1
fi
if [ ! -d "./.git" ]; then
echo "This installation does not seem to be a Git repository. Please install from source."
exit 1
fi
# Similar logic to handle script arguments; adjust as necessary
# Loop through all command line arguments
while [[ $# -gt 0 ]]; do
arg="$1"
# Check if the argument starts with a hyphen (flag)
if [[ "$arg" == -* ]]; then
# Check if the argument has a value
if [[ $# -gt 1 && "$2" != -* ]]; then
if [[ "$arg" == "--script" ]]; then
script="$2";
shift 2
else
# Add '=' sign between flag and value
args+=("'$arg'");
args+=("'$2'");
shift 2
fi
else
# Add '=True' for flags with no value
args+=("'$arg'");
shift
fi
else
# Argument is not a flag, add it as it is
args+=("'$arg '");
shift
fi
done
branch=$(git branch --show-current)
echo "Watching branch: $branch"
echo "PM2 process names: $proc_name, $generate_proc_name"
current_version=$(read_version_value)
echo "Current local version: $current_version"
# Function to check and restart pm2 processes
check_and_restart_pm2() {
local proc_name=$1
local script_path=$2
shift 2
local proc_args=("${@}")
echo "${proc_args[@]}"
if pm2 status | grep -q $proc_name; then
echo "The script $script_path is already running with pm2 under the name $proc_name. Stopping and restarting..."
pm2 delete $proc_name
fi
echo "Running $script_path with the following pm2 config:"
joined_args=$(printf "%s," "${proc_args[@]}")
joined_args=${joined_args%,}
echo "module.exports = {
apps : [{
name : '$proc_name',
script : '$script_path',
interpreter: 'python3',
min_uptime: '5m',
max_restarts: '5',
args: [$joined_args]
}]
}" > $proc_name.app.config.js
cat $proc_name.app.config.js
pm2 start $proc_name.app.config.js
}
check_and_restart_pm2 "$proc_name" "$script" ${args[@]}
# Continuous checking and updating logic
while true; do
# Get the current time
current_minute=$(date +'%M')
current_hour=$(date +'%H')
echo "Current time: $current_hour:$current_minute"
# Check if the current minute is at the top of the hour
if [ "$current_minute" != "00" ]; then
sleep 60 # Sleep for 60 seconds and check again
echo "Not the top of the hour, slept for 60 seconds"
continue
fi
# Proceed with checks only at the top of the hour
remote_version=$(check_variable_value_on_github $repository $version_location $version $branch)
while [ -z "$remote_version" ]; do
echo "Waiting for remote version to be set..."
sleep 1
done
remote_version="${remote_version#"${remote_version%%[![:space:]]*}"}"
current_version="${current_version#"${current_version%%[![:space:]]*}"}"
echo "Remote version: $remote_version, Current version: $current_version"
if [ -n "$remote_version" ] && ! echo "$remote_version" | grep -q "Error" && are_versions_different $current_version $remote_version; then
echo "Updating due to version mismatch. Current: $current_version, Remote: $remote_version"
if git pull origin $branch; then
echo "New version published. Updating the local copy."
pip install -e .
check_and_restart_pm2 "$proc_name" "$script" ${args[@]}
current_version=$(read_version_value)
echo "Update completed. Continuing monitoring..."
else
echo "Please stash your changes using git stash."
fi
else
echo "You are up-to-date with the remote version."
fi
sleep 60 # Sleep for 60 seconds to avoid multiple checks within the same minute
done