gitscope
https://dev.kronoz.win/r/marzneshin.git
script.sh · v0.2.0
Source
1 82ca347 Dawsh 2y ago #!/usr/bin/env bash
2 82ca347 Dawsh 2y ago set -e
3 82ca347 Dawsh 2y ago
4 82ca347 Dawsh 2y ago APP_NAME="marzneshin"
5 82ca347 Dawsh 2y ago CONFIG_DIR="/etc/opt/$APP_NAME"
6 82ca347 Dawsh 2y ago DATA_DIR="/var/lib/$APP_NAME"
7 82ca347 Dawsh 2y ago COMPOSE_FILE="$CONFIG_DIR/docker-compose.yml"
8 82ca347 Dawsh 2y ago
9 82ca347 Dawsh 2y ago
10 82ca347 Dawsh 2y ago colorized_echo() {
11 82ca347 Dawsh 2y ago local color=$1
12 82ca347 Dawsh 2y ago local text=$2
13 82ca347 Dawsh 2y ago
14 82ca347 Dawsh 2y ago case $color in
15 82ca347 Dawsh 2y ago "red")
16 82ca347 Dawsh 2y ago printf "\e[91m${text}\e[0m\n";;
17 82ca347 Dawsh 2y ago "green")
18 82ca347 Dawsh 2y ago printf "\e[92m${text}\e[0m\n";;
19 82ca347 Dawsh 2y ago "yellow")
20 82ca347 Dawsh 2y ago printf "\e[93m${text}\e[0m\n";;
21 82ca347 Dawsh 2y ago "blue")
22 82ca347 Dawsh 2y ago printf "\e[94m${text}\e[0m\n";;
23 82ca347 Dawsh 2y ago "magenta")
24 82ca347 Dawsh 2y ago printf "\e[95m${text}\e[0m\n";;
25 82ca347 Dawsh 2y ago "cyan")
26 82ca347 Dawsh 2y ago printf "\e[96m${text}\e[0m\n";;
27 82ca347 Dawsh 2y ago *)
28 82ca347 Dawsh 2y ago echo "${text}"
29 82ca347 Dawsh 2y ago ;;
30 82ca347 Dawsh 2y ago esac
31 82ca347 Dawsh 2y ago }
32 82ca347 Dawsh 2y ago
33 82ca347 Dawsh 2y ago check_running_as_root() {
34 82ca347 Dawsh 2y ago if [ "$(id -u)" != "0" ]; then
35 82ca347 Dawsh 2y ago colorized_echo red "This command must be run as root."
36 82ca347 Dawsh 2y ago exit 1
37 82ca347 Dawsh 2y ago fi
38 82ca347 Dawsh 2y ago }
39 82ca347 Dawsh 2y ago
40 82ca347 Dawsh 2y ago detect_os() {
41 82ca347 Dawsh 2y ago # Detect the operating system
42 82ca347 Dawsh 2y ago if [ -f /etc/lsb-release ]; then
43 82ca347 Dawsh 2y ago OS=$(lsb_release -si)
44 82ca347 Dawsh 2y ago elif [ -f /etc/os-release ]; then
45 82ca347 Dawsh 2y ago OS=$(awk -F= '/^NAME/{print $2}' /etc/os-release | tr -d '"')
46 82ca347 Dawsh 2y ago elif [ -f /etc/redhat-release ]; then
47 82ca347 Dawsh 2y ago OS=$(cat /etc/redhat-release | awk '{print $1}')
48 82ca347 Dawsh 2y ago elif [ -f /etc/arch-release ]; then
49 82ca347 Dawsh 2y ago OS="Arch"
50 82ca347 Dawsh 2y ago else
51 82ca347 Dawsh 2y ago colorized_echo red "Unsupported operating system"
52 82ca347 Dawsh 2y ago exit 1
53 82ca347 Dawsh 2y ago fi
54 82ca347 Dawsh 2y ago }
55 82ca347 Dawsh 2y ago
56 82ca347 Dawsh 2y ago detect_and_update_package_manager() {
57 82ca347 Dawsh 2y ago colorized_echo blue "Updating package manager"
58 82ca347 Dawsh 2y ago if [[ "$OS" == "Ubuntu"* ]] || [[ "$OS" == "Debian"* ]]; then
59 82ca347 Dawsh 2y ago PKG_MANAGER="apt-get"
60 82ca347 Dawsh 2y ago $PKG_MANAGER update
61 82ca347 Dawsh 2y ago elif [[ "$OS" == "CentOS"* ]] || [[ "$OS" == "AlmaLinux"* ]]; then
62 82ca347 Dawsh 2y ago PKG_MANAGER="yum"
63 82ca347 Dawsh 2y ago $PKG_MANAGER update -y
64 82ca347 Dawsh 2y ago $PKG_MANAGER install -y epel-release
65 82ca347 Dawsh 2y ago elif [ "$OS" == "Fedora"* ]; then
66 82ca347 Dawsh 2y ago PKG_MANAGER="dnf"
67 82ca347 Dawsh 2y ago $PKG_MANAGER update
68 82ca347 Dawsh 2y ago elif [ "$OS" == "Arch" ]; then
69 82ca347 Dawsh 2y ago PKG_MANAGER="pacman"
70 82ca347 Dawsh 2y ago $PKG_MANAGER -Sy
71 82ca347 Dawsh 2y ago else
72 82ca347 Dawsh 2y ago colorized_echo red "Unsupported operating system"
73 82ca347 Dawsh 2y ago exit 1
74 82ca347 Dawsh 2y ago fi
75 82ca347 Dawsh 2y ago }
76 82ca347 Dawsh 2y ago
77 82ca347 Dawsh 2y ago detect_compose() {
78 82ca347 Dawsh 2y ago # Check if docker compose command exists
79 82ca347 Dawsh 2y ago if docker compose >/dev/null 2>&1; then
80 82ca347 Dawsh 2y ago COMPOSE='docker compose'
81 82ca347 Dawsh 2y ago elif docker-compose >/dev/null 2>&1; then
82 82ca347 Dawsh 2y ago COMPOSE='docker-compose'
83 82ca347 Dawsh 2y ago else
84 82ca347 Dawsh 2y ago colorized_echo red "docker compose not found"
85 82ca347 Dawsh 2y ago exit 1
86 82ca347 Dawsh 2y ago fi
87 82ca347 Dawsh 2y ago }
88 82ca347 Dawsh 2y ago
89 82ca347 Dawsh 2y ago install_package () {
90 82ca347 Dawsh 2y ago if [ -z $PKG_MANAGER ]; then
91 82ca347 Dawsh 2y ago detect_and_update_package_manager
92 82ca347 Dawsh 2y ago fi
93 82ca347 Dawsh 2y ago
94 82ca347 Dawsh 2y ago PACKAGE=$1
95 82ca347 Dawsh 2y ago colorized_echo blue "Installing $PACKAGE"
96 82ca347 Dawsh 2y ago if [[ "$OS" == "Ubuntu"* ]] || [[ "$OS" == "Debian"* ]]; then
97 82ca347 Dawsh 2y ago $PKG_MANAGER -y install "$PACKAGE"
98 82ca347 Dawsh 2y ago elif [[ "$OS" == "CentOS"* ]] || [[ "$OS" == "AlmaLinux"* ]]; then
99 82ca347 Dawsh 2y ago $PKG_MANAGER install -y "$PACKAGE"
100 82ca347 Dawsh 2y ago elif [ "$OS" == "Fedora"* ]; then
101 82ca347 Dawsh 2y ago $PKG_MANAGER install -y "$PACKAGE"
102 82ca347 Dawsh 2y ago elif [ "$OS" == "Arch" ]; then
103 82ca347 Dawsh 2y ago $PKG_MANAGER -S --noconfirm "$PACKAGE"
104 82ca347 Dawsh 2y ago else
105 82ca347 Dawsh 2y ago colorized_echo red "Unsupported operating system"
106 82ca347 Dawsh 2y ago exit 1
107 82ca347 Dawsh 2y ago fi
108 82ca347 Dawsh 2y ago }
109 82ca347 Dawsh 2y ago
110 82ca347 Dawsh 2y ago install_docker() {
111 82ca347 Dawsh 2y ago # Install Docker and Docker Compose using the official installation script
112 82ca347 Dawsh 2y ago colorized_echo blue "Installing Docker"
113 82ca347 Dawsh 2y ago curl -fsSL https://get.docker.com | sh
114 82ca347 Dawsh 2y ago colorized_echo green "Docker installed successfully"
115 82ca347 Dawsh 2y ago }
116 82ca347 Dawsh 2y ago
117 82ca347 Dawsh 2y ago install_marzneshin_script() {
118 82ca347 Dawsh 2y ago FETCH_REPO="khodedawsh/marzneshin"
119 82ca347 Dawsh 2y ago SCRIPT_URL="https://github.com/$FETCH_REPO/raw/master/script.sh"
120 82ca347 Dawsh 2y ago colorized_echo blue "Installing marzneshin script"
121 82ca347 Dawsh 2y ago curl -sSL $SCRIPT_URL | install -m 755 /dev/stdin /usr/local/bin/marzneshin
122 82ca347 Dawsh 2y ago colorized_echo green "marzneshin script installed successfully"
123 82ca347 Dawsh 2y ago }
124 82ca347 Dawsh 2y ago
125 82ca347 Dawsh 2y ago install_marzneshin() {
126 82ca347 Dawsh 2y ago # Fetch releases
127 82ca347 Dawsh 2y ago FILES_URL_PREFIX="https://raw.githubusercontent.com/khodedawsh/marzneshin/master"
128 82ca347 Dawsh 2y ago
129 82ca347 Dawsh 2y ago mkdir -p "$DATA_DIR"
130 82ca347 Dawsh 2y ago mkdir -p "$CONFIG_DIR"
131 82ca347 Dawsh 2y ago
132 82ca347 Dawsh 2y ago colorized_echo blue "Fetching compose file"
133 82ca347 Dawsh 2y ago curl -sL "$FILES_URL_PREFIX/docker-compose.yml" -o "$CONFIG_DIR/docker-compose.yml"
134 82ca347 Dawsh 2y ago colorized_echo green "File saved in $CONFIG_DIR/docker-compose.yml"
135 82ca347 Dawsh 2y ago
136 82ca347 Dawsh 2y ago colorized_echo blue "Fetching .env file"
137 82ca347 Dawsh 2y ago curl -sL "$FILES_URL_PREFIX/.env.example" -o "$CONFIG_DIR/.env"
138 82ca347 Dawsh 2y ago sed -i 's/^# \(SQLALCHEMY_DATABASE_URL = .*\)$/\1/' "$CONFIG_DIR/.env"
139 82ca347 Dawsh 2y ago sed -i 's~\(SQLALCHEMY_DATABASE_URL = \).*~\1"sqlite:///'$DATA_DIR'/db.sqlite3"~' "$CONFIG_DIR/.env"
140 82ca347 Dawsh 2y ago colorized_echo green "File saved in $CONFIG_DIR/.env"
141 82ca347 Dawsh 2y ago
142 82ca347 Dawsh 2y ago colorized_echo green "Marzneshin files downloaded successfully"
143 82ca347 Dawsh 2y ago }
144 82ca347 Dawsh 2y ago
145 82ca347 Dawsh 2y ago
146 82ca347 Dawsh 2y ago uninstall_marzneshin_script() {
147 82ca347 Dawsh 2y ago if [ -f "/usr/local/bin/marzneshin" ]; then
148 82ca347 Dawsh 2y ago colorized_echo yellow "Removing marzneshin script"
149 82ca347 Dawsh 2y ago rm "/usr/local/bin/marzneshin"
150 82ca347 Dawsh 2y ago fi
151 82ca347 Dawsh 2y ago }
152 82ca347 Dawsh 2y ago
153 82ca347 Dawsh 2y ago uninstall_marzneshin() {
154 82ca347 Dawsh 2y ago if [ -d "$CONFIG_DIR" ]; then
155 82ca347 Dawsh 2y ago colorized_echo yellow "Removing directory: $CONFIG_DIR"
156 82ca347 Dawsh 2y ago rm -r "$CONFIG_DIR"
157 82ca347 Dawsh 2y ago fi
158 82ca347 Dawsh 2y ago }
159 82ca347 Dawsh 2y ago
160 82ca347 Dawsh 2y ago uninstall_marzneshin_docker_images() {
161 82ca347 Dawsh 2y ago images=$(docker images | grep marzneshin | awk '{print $3}')
162 82ca347 Dawsh 2y ago
163 82ca347 Dawsh 2y ago if [ -n "$images" ]; then
164 82ca347 Dawsh 2y ago colorized_echo yellow "Removing Docker images of Marzneshin"
165 82ca347 Dawsh 2y ago for image in $images; do
166 82ca347 Dawsh 2y ago if docker rmi "$image" >/dev/null 2>&1; then
167 82ca347 Dawsh 2y ago colorized_echo yellow "Image $image removed"
168 82ca347 Dawsh 2y ago fi
169 82ca347 Dawsh 2y ago done
170 82ca347 Dawsh 2y ago fi
171 82ca347 Dawsh 2y ago }
172 82ca347 Dawsh 2y ago
173 82ca347 Dawsh 2y ago uninstall_marzneshin_data_files() {
174 82ca347 Dawsh 2y ago if [ -d "$DATA_DIR" ]; then
175 82ca347 Dawsh 2y ago colorized_echo yellow "Removing directory: $DATA_DIR"
176 82ca347 Dawsh 2y ago rm -r "$DATA_DIR"
177 82ca347 Dawsh 2y ago fi
178 82ca347 Dawsh 2y ago }
179 82ca347 Dawsh 2y ago
180 82ca347 Dawsh 2y ago up_marzneshin() {
181 82ca347 Dawsh 2y ago $COMPOSE -f $COMPOSE_FILE -p "$APP_NAME" up -d --remove-orphans
182 82ca347 Dawsh 2y ago }
183 82ca347 Dawsh 2y ago
184 82ca347 Dawsh 2y ago down_marzneshin() {
185 82ca347 Dawsh 2y ago $COMPOSE -f $COMPOSE_FILE -p "$APP_NAME" down
186 82ca347 Dawsh 2y ago }
187 82ca347 Dawsh 2y ago
188 82ca347 Dawsh 2y ago show_marzneshin_logs() {
189 82ca347 Dawsh 2y ago $COMPOSE -f $COMPOSE_FILE -p "$APP_NAME" logs
190 82ca347 Dawsh 2y ago }
191 82ca347 Dawsh 2y ago
192 82ca347 Dawsh 2y ago follow_marzneshin_logs() {
193 82ca347 Dawsh 2y ago $COMPOSE -f $COMPOSE_FILE -p "$APP_NAME" logs -f
194 82ca347 Dawsh 2y ago }
195 82ca347 Dawsh 2y ago
196 82ca347 Dawsh 2y ago marzneshin_cli() {
197 49562f8 Dawsh 2y ago $COMPOSE -f $COMPOSE_FILE -p "$APP_NAME" exec -e CLI_PROG_NAME="marzneshin cli" marzneshin /app/marzneshin-cli.py "$@"
198 82ca347 Dawsh 2y ago }
199 82ca347 Dawsh 2y ago
200 82ca347 Dawsh 2y ago
201 82ca347 Dawsh 2y ago update_marzneshin_script() {
202 82ca347 Dawsh 2y ago FETCH_REPO="khodedawsh/marzneshin"
203 420cd89 Dawsh 2y ago SCRIPT_URL="https://github.com/$FETCH_REPO/raw/master/script.sh"
204 82ca347 Dawsh 2y ago colorized_echo blue "Updating marzneshin script"
205 82ca347 Dawsh 2y ago curl -sSL $SCRIPT_URL | install -m 755 /dev/stdin /usr/local/bin/marzneshin
206 82ca347 Dawsh 2y ago colorized_echo green "marzneshin script updated successfully"
207 82ca347 Dawsh 2y ago }
208 82ca347 Dawsh 2y ago
209 82ca347 Dawsh 2y ago update_marzneshin() {
210 82ca347 Dawsh 2y ago $COMPOSE -f $COMPOSE_FILE -p "$APP_NAME" pull
211 82ca347 Dawsh 2y ago }
212 82ca347 Dawsh 2y ago
213 82ca347 Dawsh 2y ago is_marzneshin_installed() {
214 82ca347 Dawsh 2y ago if [ -d $CONFIG_DIR ]; then
215 82ca347 Dawsh 2y ago return 0
216 82ca347 Dawsh 2y ago else
217 82ca347 Dawsh 2y ago return 1
218 82ca347 Dawsh 2y ago fi
219 82ca347 Dawsh 2y ago }
220 82ca347 Dawsh 2y ago
221 82ca347 Dawsh 2y ago is_marzneshin_up() {
222 82ca347 Dawsh 2y ago if [ -z "$($COMPOSE -f $COMPOSE_FILE ps -q -a)" ]; then
223 82ca347 Dawsh 2y ago return 1
224 82ca347 Dawsh 2y ago else
225 82ca347 Dawsh 2y ago return 0
226 82ca347 Dawsh 2y ago fi
227 82ca347 Dawsh 2y ago }
228 82ca347 Dawsh 2y ago
229 82ca347 Dawsh 2y ago install_command() {
230 82ca347 Dawsh 2y ago check_running_as_root
231 82ca347 Dawsh 2y ago # Check if marzneshin is already installed
232 82ca347 Dawsh 2y ago if is_marzneshin_installed; then
233 82ca347 Dawsh 2y ago colorized_echo red "Marzneshin is already installed at $CONFIG_DIR"
234 82ca347 Dawsh 2y ago read -p "Do you want to override the previous installation? (y/n) "
235 82ca347 Dawsh 2y ago if [[ ! $REPLY =~ ^[Yy]$ ]]; then
236 82ca347 Dawsh 2y ago colorized_echo red "Aborted installation"
237 82ca347 Dawsh 2y ago exit 1
238 82ca347 Dawsh 2y ago fi
239 82ca347 Dawsh 2y ago fi
240 82ca347 Dawsh 2y ago detect_os
241 82ca347 Dawsh 2y ago if ! command -v jq >/dev/null 2>&1; then
242 82ca347 Dawsh 2y ago install_package jq
243 82ca347 Dawsh 2y ago fi
244 82ca347 Dawsh 2y ago if ! command -v curl >/dev/null 2>&1; then
245 82ca347 Dawsh 2y ago install_package curl
246 82ca347 Dawsh 2y ago fi
247 82ca347 Dawsh 2y ago if ! command -v docker >/dev/null 2>&1; then
248 82ca347 Dawsh 2y ago install_docker
249 82ca347 Dawsh 2y ago fi
250 82ca347 Dawsh 2y ago detect_compose
251 82ca347 Dawsh 2y ago install_marzneshin_script
252 82ca347 Dawsh 2y ago install_marzneshin
253 82ca347 Dawsh 2y ago up_marzneshin
254 82ca347 Dawsh 2y ago follow_marzneshin_logs
255 82ca347 Dawsh 2y ago }
256 82ca347 Dawsh 2y ago
257 82ca347 Dawsh 2y ago uninstall_command() {
258 82ca347 Dawsh 2y ago check_running_as_root
259 82ca347 Dawsh 2y ago # Check if marzneshin is installed
260 82ca347 Dawsh 2y ago if ! is_marzneshin_installed; then
261 82ca347 Dawsh 2y ago colorized_echo red "Marzneshin's not installed!"
262 82ca347 Dawsh 2y ago exit 1
263 82ca347 Dawsh 2y ago fi
264 82ca347 Dawsh 2y ago
265 82ca347 Dawsh 2y ago read -p "Do you really want to uninstall Marzneshin? (y/n) "
266 82ca347 Dawsh 2y ago if [[ ! $REPLY =~ ^[Yy]$ ]]; then
267 82ca347 Dawsh 2y ago colorized_echo red "Aborted"
268 82ca347 Dawsh 2y ago exit 1
269 82ca347 Dawsh 2y ago fi
270 82ca347 Dawsh 2y ago
271 82ca347 Dawsh 2y ago detect_compose
272 82ca347 Dawsh 2y ago if is_marzneshin_up; then
273 82ca347 Dawsh 2y ago down_marzneshin
274 82ca347 Dawsh 2y ago fi
275 82ca347 Dawsh 2y ago uninstall_marzneshin_script
276 82ca347 Dawsh 2y ago uninstall_marzneshin
277 82ca347 Dawsh 2y ago uninstall_marzneshin_docker_images
278 82ca347 Dawsh 2y ago
279 82ca347 Dawsh 2y ago read -p "Do you want to remove Marzneshin's data files too ($DATA_DIR)? (y/n) "
280 82ca347 Dawsh 2y ago if [[ ! $REPLY =~ ^[Yy]$ ]]; then
281 82ca347 Dawsh 2y ago colorized_echo green "Marzneshin uninstalled successfully"
282 82ca347 Dawsh 2y ago else
283 82ca347 Dawsh 2y ago uninstall_marzneshin_data_files
284 82ca347 Dawsh 2y ago colorized_echo green "Marzneshin uninstalled successfully"
285 82ca347 Dawsh 2y ago fi
286 82ca347 Dawsh 2y ago }
287 82ca347 Dawsh 2y ago
288 82ca347 Dawsh 2y ago up_command() {
289 82ca347 Dawsh 2y ago help() {
290 82ca347 Dawsh 2y ago colorized_echo red "Usage: $0 up [options]"
291 82ca347 Dawsh 2y ago echo ""
292 82ca347 Dawsh 2y ago echo "OPTIONS:"
293 82ca347 Dawsh 2y ago echo " -h, --help display this help message"
294 82ca347 Dawsh 2y ago echo " -n, --no-logs do not follow logs after starting"
295 82ca347 Dawsh 2y ago }
296 82ca347 Dawsh 2y ago
297 82ca347 Dawsh 2y ago local no_logs=false
298 82ca347 Dawsh 2y ago while [[ "$#" -gt 0 ]]; do
299 82ca347 Dawsh 2y ago case "$1" in
300 82ca347 Dawsh 2y ago -n|--no-logs)
301 82ca347 Dawsh 2y ago no_logs=true
302 82ca347 Dawsh 2y ago ;;
303 82ca347 Dawsh 2y ago -h|--help)
304 82ca347 Dawsh 2y ago help
305 82ca347 Dawsh 2y ago exit 0
306 82ca347 Dawsh 2y ago ;;
307 82ca347 Dawsh 2y ago *)
308 82ca347 Dawsh 2y ago echo "Error: Invalid option: $1" >&2
309 82ca347 Dawsh 2y ago help
310 82ca347 Dawsh 2y ago exit 0
311 82ca347 Dawsh 2y ago ;;
312 82ca347 Dawsh 2y ago esac
313 82ca347 Dawsh 2y ago shift
314 82ca347 Dawsh 2y ago done
315 82ca347 Dawsh 2y ago
316 82ca347 Dawsh 2y ago # Check if marzneshin is installed
317 82ca347 Dawsh 2y ago if ! is_marzneshin_installed; then
318 82ca347 Dawsh 2y ago colorized_echo red "Marzneshin is not installed!"
319 82ca347 Dawsh 2y ago exit 1
320 82ca347 Dawsh 2y ago fi
321 82ca347 Dawsh 2y ago
322 82ca347 Dawsh 2y ago detect_compose
323 82ca347 Dawsh 2y ago
324 82ca347 Dawsh 2y ago if is_marzneshin_up; then
325 82ca347 Dawsh 2y ago colorized_echo red "Marzneshin is already up"
326 82ca347 Dawsh 2y ago exit 1
327 82ca347 Dawsh 2y ago fi
328 82ca347 Dawsh 2y ago
329 82ca347 Dawsh 2y ago up_marzneshin
330 82ca347 Dawsh 2y ago if [ "$no_logs" = false ]; then
331 82ca347 Dawsh 2y ago follow_marzneshin_logs
332 82ca347 Dawsh 2y ago fi
333 82ca347 Dawsh 2y ago }
334 82ca347 Dawsh 2y ago
335 82ca347 Dawsh 2y ago down_command() {
336 82ca347 Dawsh 2y ago
337 82ca347 Dawsh 2y ago # Check if marzneshin is installed
338 82ca347 Dawsh 2y ago if ! is_marzneshin_installed; then
339 82ca347 Dawsh 2y ago colorized_echo red "Marzneshin's not installed!"
340 82ca347 Dawsh 2y ago exit 1
341 82ca347 Dawsh 2y ago fi
342 82ca347 Dawsh 2y ago
343 82ca347 Dawsh 2y ago detect_compose
344 82ca347 Dawsh 2y ago
345 82ca347 Dawsh 2y ago if ! is_marzneshin_up; then
346 82ca347 Dawsh 2y ago colorized_echo red "Marzneshin's already down"
347 82ca347 Dawsh 2y ago exit 1
348 82ca347 Dawsh 2y ago fi
349 82ca347 Dawsh 2y ago
350 82ca347 Dawsh 2y ago down_marzneshin
351 82ca347 Dawsh 2y ago }
352 82ca347 Dawsh 2y ago
353 82ca347 Dawsh 2y ago restart_command() {
354 82ca347 Dawsh 2y ago help() {
355 82ca347 Dawsh 2y ago colorized_echo red "Usage: $0 restart [options]"
356 82ca347 Dawsh 2y ago echo
357 82ca347 Dawsh 2y ago echo "OPTIONS:"
358 82ca347 Dawsh 2y ago echo " -h, --help display this help message"
359 82ca347 Dawsh 2y ago echo " -n, --no-logs do not follow logs after starting"
360 82ca347 Dawsh 2y ago }
361 82ca347 Dawsh 2y ago
362 82ca347 Dawsh 2y ago local no_logs=false
363 82ca347 Dawsh 2y ago while [[ "$#" -gt 0 ]]; do
364 82ca347 Dawsh 2y ago case "$1" in
365 82ca347 Dawsh 2y ago -n|--no-logs)
366 82ca347 Dawsh 2y ago no_logs=true
367 82ca347 Dawsh 2y ago ;;
368 82ca347 Dawsh 2y ago -h|--help)
369 82ca347 Dawsh 2y ago help
370 82ca347 Dawsh 2y ago exit 0
371 82ca347 Dawsh 2y ago ;;
372 82ca347 Dawsh 2y ago *)
373 82ca347 Dawsh 2y ago echo "Error: Invalid option: $1" >&2
374 82ca347 Dawsh 2y ago help
375 82ca347 Dawsh 2y ago exit 0
376 82ca347 Dawsh 2y ago ;;
377 82ca347 Dawsh 2y ago esac
378 82ca347 Dawsh 2y ago shift
379 82ca347 Dawsh 2y ago done
380 82ca347 Dawsh 2y ago
381 82ca347 Dawsh 2y ago # Check if marzneshin is installed
382 82ca347 Dawsh 2y ago if ! is_marzneshin_installed; then
383 82ca347 Dawsh 2y ago colorized_echo red "Marzneshin's not installed!"
384 82ca347 Dawsh 2y ago exit 1
385 82ca347 Dawsh 2y ago fi
386 82ca347 Dawsh 2y ago
387 82ca347 Dawsh 2y ago detect_compose
388 82ca347 Dawsh 2y ago
389 82ca347 Dawsh 2y ago down_marzneshin
390 82ca347 Dawsh 2y ago up_marzneshin
391 82ca347 Dawsh 2y ago if [ "$no_logs" = false ]; then
392 82ca347 Dawsh 2y ago follow_marzneshin_logs
393 82ca347 Dawsh 2y ago fi
394 82ca347 Dawsh 2y ago }
395 82ca347 Dawsh 2y ago
396 82ca347 Dawsh 2y ago status_command() {
397 82ca347 Dawsh 2y ago
398 82ca347 Dawsh 2y ago # Check if marzneshin is installed
399 82ca347 Dawsh 2y ago if ! is_marzneshin_installed; then
400 82ca347 Dawsh 2y ago echo -n "Status: "
401 82ca347 Dawsh 2y ago colorized_echo red "Not Installed"
402 82ca347 Dawsh 2y ago exit 1
403 82ca347 Dawsh 2y ago fi
404 82ca347 Dawsh 2y ago
405 82ca347 Dawsh 2y ago detect_compose
406 82ca347 Dawsh 2y ago
407 82ca347 Dawsh 2y ago if ! is_marzneshin_up; then
408 82ca347 Dawsh 2y ago echo -n "Status: "
409 82ca347 Dawsh 2y ago colorized_echo blue "Down"
410 82ca347 Dawsh 2y ago exit 1
411 82ca347 Dawsh 2y ago fi
412 82ca347 Dawsh 2y ago
413 82ca347 Dawsh 2y ago echo -n "Status: "
414 82ca347 Dawsh 2y ago colorized_echo green "Up"
415 82ca347 Dawsh 2y ago
416 82ca347 Dawsh 2y ago json=$($COMPOSE -f $COMPOSE_FILE ps -a --format=json)
417 82ca347 Dawsh 2y ago services=$(echo "$json" | jq -r 'if type == "array" then .[] else . end | .Service')
418 82ca347 Dawsh 2y ago states=$(echo "$json" | jq -r 'if type == "array" then .[] else . end | .State')
419 82ca347 Dawsh 2y ago # Print out the service names and statuses
420 82ca347 Dawsh 2y ago for i in $(seq 0 $(expr $(echo $services | wc -w) - 1)); do
421 82ca347 Dawsh 2y ago service=$(echo $services | cut -d' ' -f $(expr $i + 1))
422 82ca347 Dawsh 2y ago state=$(echo $states | cut -d' ' -f $(expr $i + 1))
423 82ca347 Dawsh 2y ago echo -n "- $service: "
424 82ca347 Dawsh 2y ago if [ "$state" == "running" ]; then
425 82ca347 Dawsh 2y ago colorized_echo green $state
426 82ca347 Dawsh 2y ago else
427 82ca347 Dawsh 2y ago colorized_echo red $state
428 82ca347 Dawsh 2y ago fi
429 82ca347 Dawsh 2y ago done
430 82ca347 Dawsh 2y ago }
431 82ca347 Dawsh 2y ago
432 82ca347 Dawsh 2y ago logs_command() {
433 82ca347 Dawsh 2y ago help() {
434 82ca347 Dawsh 2y ago colorized_echo red "Usage: marzneshin logs [options]"
435 82ca347 Dawsh 2y ago echo ""
436 82ca347 Dawsh 2y ago echo "OPTIONS:"
437 82ca347 Dawsh 2y ago echo " -h, --help display this help message"
438 82ca347 Dawsh 2y ago echo " -n, --no-follow do not show follow logs"
439 82ca347 Dawsh 2y ago }
440 82ca347 Dawsh 2y ago
441 82ca347 Dawsh 2y ago local no_follow=false
442 82ca347 Dawsh 2y ago while [[ "$#" -gt 0 ]]; do
443 82ca347 Dawsh 2y ago case "$1" in
444 82ca347 Dawsh 2y ago -n|--no-follow)
445 82ca347 Dawsh 2y ago no_follow=true
446 82ca347 Dawsh 2y ago ;;
447 82ca347 Dawsh 2y ago -h|--help)
448 82ca347 Dawsh 2y ago help
449 82ca347 Dawsh 2y ago exit 0
450 82ca347 Dawsh 2y ago ;;
451 82ca347 Dawsh 2y ago *)
452 82ca347 Dawsh 2y ago echo "Error: Invalid option: $1" >&2
453 82ca347 Dawsh 2y ago help
454 82ca347 Dawsh 2y ago exit 0
455 82ca347 Dawsh 2y ago ;;
456 82ca347 Dawsh 2y ago esac
457 82ca347 Dawsh 2y ago shift
458 82ca347 Dawsh 2y ago done
459 82ca347 Dawsh 2y ago
460 82ca347 Dawsh 2y ago # Check if marzneshin is installed
461 82ca347 Dawsh 2y ago if ! is_marzneshin_installed; then
462 82ca347 Dawsh 2y ago colorized_echo red "Marzneshin is not installed!"
463 82ca347 Dawsh 2y ago exit 1
464 82ca347 Dawsh 2y ago fi
465 82ca347 Dawsh 2y ago
466 82ca347 Dawsh 2y ago detect_compose
467 82ca347 Dawsh 2y ago
468 82ca347 Dawsh 2y ago if ! is_marzneshin_up; then
469 82ca347 Dawsh 2y ago colorized_echo red "Marzneshin is not up."
470 82ca347 Dawsh 2y ago exit 1
471 82ca347 Dawsh 2y ago fi
472 82ca347 Dawsh 2y ago
473 82ca347 Dawsh 2y ago if [ "$no_follow" = true ]; then
474 82ca347 Dawsh 2y ago show_marzneshin_logs
475 82ca347 Dawsh 2y ago else
476 82ca347 Dawsh 2y ago follow_marzneshin_logs
477 82ca347 Dawsh 2y ago fi
478 82ca347 Dawsh 2y ago }
479 82ca347 Dawsh 2y ago
480 82ca347 Dawsh 2y ago cli_command() {
481 82ca347 Dawsh 2y ago # Check if marzneshin is installed
482 82ca347 Dawsh 2y ago if ! is_marzneshin_installed; then
483 82ca347 Dawsh 2y ago colorized_echo red "Marzneshin is not installed!"
484 82ca347 Dawsh 2y ago exit 1
485 82ca347 Dawsh 2y ago fi
486 82ca347 Dawsh 2y ago
487 82ca347 Dawsh 2y ago detect_compose
488 82ca347 Dawsh 2y ago
489 82ca347 Dawsh 2y ago if ! is_marzneshin_up; then
490 82ca347 Dawsh 2y ago colorized_echo red "Marzneshin is not up."
491 82ca347 Dawsh 2y ago exit 1
492 82ca347 Dawsh 2y ago fi
493 82ca347 Dawsh 2y ago
494 82ca347 Dawsh 2y ago marzneshin_cli "$@"
495 82ca347 Dawsh 2y ago }
496 82ca347 Dawsh 2y ago
497 82ca347 Dawsh 2y ago update_command() {
498 82ca347 Dawsh 2y ago check_running_as_root
499 82ca347 Dawsh 2y ago # Check if marzneshin is installed
500 82ca347 Dawsh 2y ago if ! is_marzneshin_installed; then
501 82ca347 Dawsh 2y ago colorized_echo red "Marzneshin is not installed!"
502 82ca347 Dawsh 2y ago exit 1
503 82ca347 Dawsh 2y ago fi
504 82ca347 Dawsh 2y ago
505 82ca347 Dawsh 2y ago detect_compose
506 82ca347 Dawsh 2y ago
507 82ca347 Dawsh 2y ago update_marzneshin_script
508 82ca347 Dawsh 2y ago colorized_echo blue "Pulling latest version"
509 82ca347 Dawsh 2y ago update_marzneshin
510 82ca347 Dawsh 2y ago
511 82ca347 Dawsh 2y ago colorized_echo blue "Restarting Marzneshin's services"
512 82ca347 Dawsh 2y ago down_marzneshin
513 82ca347 Dawsh 2y ago up_marzneshin
514 82ca347 Dawsh 2y ago
515 82ca347 Dawsh 2y ago colorized_echo blue "Marzneshin updated successfully"
516 82ca347 Dawsh 2y ago }
517 82ca347 Dawsh 2y ago
518 82ca347 Dawsh 2y ago
519 82ca347 Dawsh 2y ago usage() {
520 82ca347 Dawsh 2y ago colorized_echo red "Usage: $0 [command]"
521 82ca347 Dawsh 2y ago echo
522 82ca347 Dawsh 2y ago echo "Commands:"
523 82ca347 Dawsh 2y ago echo " up Start services"
524 82ca347 Dawsh 2y ago echo " down Stop services"
525 82ca347 Dawsh 2y ago echo " restart Restart services"
526 82ca347 Dawsh 2y ago echo " status Show status"
527 82ca347 Dawsh 2y ago echo " logs Show logs"
528 82ca347 Dawsh 2y ago echo " cli Marzneshin command-line interface"
529 82ca347 Dawsh 2y ago echo " install Install Marzneshin"
530 82ca347 Dawsh 2y ago echo " update Update latest version"
531 82ca347 Dawsh 2y ago echo " uninstall Uninstall Marzneshin"
532 82ca347 Dawsh 2y ago echo " install-script Install Marzneshin script"
533 82ca347 Dawsh 2y ago echo
534 82ca347 Dawsh 2y ago }
535 82ca347 Dawsh 2y ago
536 82ca347 Dawsh 2y ago case "$1" in
537 82ca347 Dawsh 2y ago up)
538 82ca347 Dawsh 2y ago shift; up_command "$@";;
539 82ca347 Dawsh 2y ago down)
540 82ca347 Dawsh 2y ago shift; down_command "$@";;
541 82ca347 Dawsh 2y ago restart)
542 82ca347 Dawsh 2y ago shift; restart_command "$@";;
543 82ca347 Dawsh 2y ago status)
544 82ca347 Dawsh 2y ago shift; status_command "$@";;
545 82ca347 Dawsh 2y ago logs)
546 82ca347 Dawsh 2y ago shift; logs_command "$@";;
547 82ca347 Dawsh 2y ago cli)
548 82ca347 Dawsh 2y ago shift; cli_command "$@";;
549 82ca347 Dawsh 2y ago install)
550 82ca347 Dawsh 2y ago shift; install_command "$@";;
551 82ca347 Dawsh 2y ago update)
552 82ca347 Dawsh 2y ago shift; update_command "$@";;
553 82ca347 Dawsh 2y ago uninstall)
554 82ca347 Dawsh 2y ago shift; uninstall_command "$@";;
555 82ca347 Dawsh 2y ago install-script)
556 82ca347 Dawsh 2y ago shift; install_marzneshin_script "$@";;
557 82ca347 Dawsh 2y ago *)
558 82ca347 Dawsh 2y ago usage;;
559 82ca347 Dawsh 2y ago esac