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