Todd
307
scripts/mds
Executable file
307
scripts/mds
Executable file
@@ -0,0 +1,307 @@
|
||||
#!/bin/bash
|
||||
# mds - Media Downloader Services manager
|
||||
# Usage: mds [command] [service(s)...]
|
||||
#
|
||||
# Commands:
|
||||
# status - Show status of services (default)
|
||||
# start - Start service(s)
|
||||
# stop - Stop service(s)
|
||||
# restart - Restart service(s)
|
||||
# logs - Show recent logs for a service
|
||||
#
|
||||
# Services:
|
||||
# all - All services
|
||||
# scheduler - media-downloader (scheduler)
|
||||
# api - media-downloader-api
|
||||
# frontend - media-downloader-frontend
|
||||
# xvfb - xvfb-media-downloader
|
||||
# proxy - unified-proxy (Docker)
|
||||
# cache - media-cache-builder
|
||||
# enrich - media-celebrity-enrichment
|
||||
# embeddings - media-embedding-generator
|
||||
# dbcleanup - media-downloader-db-cleanup
|
||||
# backup - cloud-backup-sync
|
||||
# backupui - backup-central (web UI)
|
||||
#
|
||||
# Examples:
|
||||
# mds # status of all services
|
||||
# mds status # same
|
||||
# mds restart api # restart just the API
|
||||
# mds restart api frontend # restart API and frontend
|
||||
# mds restart all # restart all services
|
||||
# mds stop scheduler # stop the scheduler
|
||||
# mds logs scheduler # show scheduler logs
|
||||
|
||||
set -euo pipefail
|
||||
|
||||
# Color codes
|
||||
RED='\033[0;31m'
|
||||
GREEN='\033[0;32m'
|
||||
YELLOW='\033[1;33m'
|
||||
BLUE='\033[0;34m'
|
||||
CYAN='\033[0;36m'
|
||||
BOLD='\033[1m'
|
||||
DIM='\033[2m'
|
||||
NC='\033[0m' # No Color
|
||||
|
||||
# Service name mapping
|
||||
declare -A SERVICE_MAP=(
|
||||
[scheduler]="media-downloader"
|
||||
[api]="media-downloader-api"
|
||||
[frontend]="media-downloader-frontend"
|
||||
[xvfb]="xvfb-media-downloader"
|
||||
[cache]="media-cache-builder"
|
||||
[enrich]="media-celebrity-enrichment"
|
||||
[embeddings]="media-embedding-generator"
|
||||
[dbcleanup]="media-downloader-db-cleanup"
|
||||
[backup]="cloud-backup-sync"
|
||||
[backupui]="backup-central"
|
||||
)
|
||||
|
||||
ALL_ALIASES=(scheduler api frontend xvfb proxy cache enrich embeddings dbcleanup backup backupui)
|
||||
|
||||
resolve_service() {
|
||||
local alias="$1"
|
||||
if [[ "$alias" == "proxy" ]]; then
|
||||
echo "proxy"
|
||||
elif [[ -n "${SERVICE_MAP[$alias]+x}" ]]; then
|
||||
echo "${SERVICE_MAP[$alias]}"
|
||||
else
|
||||
# Try as a literal service name
|
||||
for key in "${!SERVICE_MAP[@]}"; do
|
||||
if [[ "${SERVICE_MAP[$key]}" == "$alias" ]]; then
|
||||
echo "$alias"
|
||||
return
|
||||
fi
|
||||
done
|
||||
echo ""
|
||||
fi
|
||||
}
|
||||
|
||||
get_alias() {
|
||||
local service="$1"
|
||||
for key in "${!SERVICE_MAP[@]}"; do
|
||||
if [[ "${SERVICE_MAP[$key]}" == "$service" ]]; then
|
||||
echo "$key"
|
||||
return
|
||||
fi
|
||||
done
|
||||
echo "$service"
|
||||
}
|
||||
|
||||
print_status_line() {
|
||||
local alias="$1"
|
||||
local service="$2"
|
||||
|
||||
if [[ "$alias" == "proxy" ]]; then
|
||||
# Docker container
|
||||
local state
|
||||
state=$(docker inspect -f '{{.State.Status}}' unified-proxy 2>/dev/null || echo "not found")
|
||||
local uptime
|
||||
uptime=$(docker inspect -f '{{.State.StartedAt}}' unified-proxy 2>/dev/null || echo "")
|
||||
|
||||
local color="$RED"
|
||||
local symbol="●"
|
||||
if [[ "$state" == "running" ]]; then
|
||||
color="$GREEN"
|
||||
fi
|
||||
|
||||
printf " ${color}${symbol}${NC} %-12s %-30s %s\n" "$alias" "unified-proxy (docker)" "$state"
|
||||
return
|
||||
fi
|
||||
|
||||
local active_state sub_state
|
||||
active_state=$(systemctl show -p ActiveState --value "$service" 2>/dev/null || echo "unknown")
|
||||
sub_state=$(systemctl show -p SubState --value "$service" 2>/dev/null || echo "unknown")
|
||||
|
||||
local color="$RED"
|
||||
local symbol="●"
|
||||
case "$active_state" in
|
||||
active)
|
||||
color="$GREEN"
|
||||
;;
|
||||
inactive)
|
||||
if [[ "$sub_state" == "failed" ]]; then
|
||||
color="$RED"
|
||||
else
|
||||
color="$DIM"
|
||||
fi
|
||||
;;
|
||||
deactivating)
|
||||
color="$YELLOW"
|
||||
;;
|
||||
activating)
|
||||
color="$BLUE"
|
||||
;;
|
||||
failed)
|
||||
color="$RED"
|
||||
;;
|
||||
esac
|
||||
|
||||
# Get memory and uptime
|
||||
local memory=""
|
||||
local pid=""
|
||||
if [[ "$active_state" == "active" ]]; then
|
||||
memory=$(systemctl show -p MemoryCurrent --value "$service" 2>/dev/null || echo "")
|
||||
pid=$(systemctl show -p MainPID --value "$service" 2>/dev/null || echo "")
|
||||
if [[ -n "$memory" && "$memory" != "[not set]" && "$memory" != "infinity" ]]; then
|
||||
# Convert bytes to human readable
|
||||
local mem_mb=$((memory / 1024 / 1024))
|
||||
if [[ $mem_mb -gt 1024 ]]; then
|
||||
local mem_gb=$((mem_mb / 1024))
|
||||
local mem_frac=$(( (mem_mb % 1024) * 10 / 1024 ))
|
||||
memory="${mem_gb}.${mem_frac}G"
|
||||
else
|
||||
memory="${mem_mb}M"
|
||||
fi
|
||||
else
|
||||
memory=""
|
||||
fi
|
||||
fi
|
||||
|
||||
local details
|
||||
if [[ "$active_state" == "inactive" && "$sub_state" == "dead" ]]; then
|
||||
details="stopped"
|
||||
elif [[ "$active_state" == "failed" || "$sub_state" == "failed" ]]; then
|
||||
details="failed"
|
||||
elif [[ "$active_state" == "active" && "$sub_state" == "running" ]]; then
|
||||
details="running"
|
||||
else
|
||||
details="$active_state ($sub_state)"
|
||||
fi
|
||||
if [[ -n "$memory" ]]; then
|
||||
details="$details ${DIM}mem: ${memory}${NC}"
|
||||
fi
|
||||
if [[ -n "$pid" && "$pid" != "0" ]]; then
|
||||
details="$details ${DIM}pid: ${pid}${NC}"
|
||||
fi
|
||||
|
||||
printf " ${color}${symbol}${NC} %-12s %-30s %b\n" "$alias" "$service" "$details"
|
||||
}
|
||||
|
||||
do_status() {
|
||||
local services=("$@")
|
||||
if [[ ${#services[@]} -eq 0 ]]; then
|
||||
services=("${ALL_ALIASES[@]}")
|
||||
fi
|
||||
|
||||
echo -e "\n${BOLD}Media Downloader Services${NC}\n"
|
||||
|
||||
for alias in "${services[@]}"; do
|
||||
local service
|
||||
service=$(resolve_service "$alias")
|
||||
if [[ -z "$service" ]]; then
|
||||
echo -e " ${RED}?${NC} ${alias} (unknown service)"
|
||||
continue
|
||||
fi
|
||||
print_status_line "$alias" "$service"
|
||||
done
|
||||
echo ""
|
||||
}
|
||||
|
||||
do_action() {
|
||||
local action="$1"
|
||||
shift
|
||||
local services=("$@")
|
||||
|
||||
if [[ ${#services[@]} -eq 0 ]]; then
|
||||
echo -e "${RED}Error: specify service(s) or 'all'${NC}"
|
||||
echo "Usage: mds $action [service(s)...]"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
# Expand 'all'
|
||||
if [[ "${services[0]}" == "all" ]]; then
|
||||
services=("${ALL_ALIASES[@]}")
|
||||
fi
|
||||
|
||||
for alias in "${services[@]}"; do
|
||||
local service
|
||||
service=$(resolve_service "$alias")
|
||||
if [[ -z "$service" ]]; then
|
||||
echo -e " ${RED}✗${NC} ${alias}: unknown service"
|
||||
continue
|
||||
fi
|
||||
|
||||
if [[ "$alias" == "proxy" ]]; then
|
||||
case "$action" in
|
||||
start)
|
||||
echo -e " ${CYAN}▶${NC} Starting unified-proxy..."
|
||||
docker start unified-proxy 2>/dev/null && echo -e " ${GREEN}✓${NC} unified-proxy started" || echo -e " ${RED}✗${NC} Failed to start unified-proxy"
|
||||
;;
|
||||
stop)
|
||||
echo -e " ${YELLOW}■${NC} Stopping unified-proxy..."
|
||||
docker stop unified-proxy 2>/dev/null && echo -e " ${GREEN}✓${NC} unified-proxy stopped" || echo -e " ${RED}✗${NC} Failed to stop unified-proxy"
|
||||
;;
|
||||
restart)
|
||||
echo -e " ${CYAN}↻${NC} Restarting unified-proxy..."
|
||||
docker restart unified-proxy 2>/dev/null && echo -e " ${GREEN}✓${NC} unified-proxy restarted" || echo -e " ${RED}✗${NC} Failed to restart unified-proxy"
|
||||
;;
|
||||
esac
|
||||
continue
|
||||
fi
|
||||
|
||||
case "$action" in
|
||||
start)
|
||||
echo -e " ${CYAN}▶${NC} Starting ${alias} (${service})..."
|
||||
sudo systemctl start "$service" && echo -e " ${GREEN}✓${NC} ${alias} started" || echo -e " ${RED}✗${NC} Failed to start ${alias}"
|
||||
;;
|
||||
stop)
|
||||
echo -e " ${YELLOW}■${NC} Stopping ${alias} (${service})..."
|
||||
sudo systemctl stop "$service" && echo -e " ${GREEN}✓${NC} ${alias} stopped" || echo -e " ${RED}✗${NC} Failed to stop ${alias}"
|
||||
;;
|
||||
restart)
|
||||
echo -e " ${CYAN}↻${NC} Restarting ${alias} (${service})..."
|
||||
sudo systemctl restart "$service" && echo -e " ${GREEN}✓${NC} ${alias} restarted" || echo -e " ${RED}✗${NC} Failed to restart ${alias}"
|
||||
;;
|
||||
esac
|
||||
done
|
||||
}
|
||||
|
||||
do_logs() {
|
||||
local alias="${1:-scheduler}"
|
||||
local service
|
||||
service=$(resolve_service "$alias")
|
||||
|
||||
if [[ -z "$service" ]]; then
|
||||
echo -e "${RED}Unknown service: ${alias}${NC}"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
if [[ "$alias" == "proxy" ]]; then
|
||||
docker logs --tail 50 unified-proxy
|
||||
return
|
||||
fi
|
||||
|
||||
sudo journalctl -u "$service" --no-pager -n 50
|
||||
}
|
||||
|
||||
# Main
|
||||
command="${1:-status}"
|
||||
shift 2>/dev/null || true
|
||||
|
||||
case "$command" in
|
||||
status|st|s)
|
||||
do_status "$@"
|
||||
;;
|
||||
start)
|
||||
do_action start "$@"
|
||||
;;
|
||||
stop)
|
||||
do_action stop "$@"
|
||||
;;
|
||||
restart|rs|r)
|
||||
do_action restart "$@"
|
||||
;;
|
||||
logs|log|l)
|
||||
do_logs "$@"
|
||||
;;
|
||||
help|--help|-h)
|
||||
head -27 "$0" | tail -25
|
||||
;;
|
||||
*)
|
||||
echo -e "${RED}Unknown command: ${command}${NC}"
|
||||
echo "Commands: status, start, stop, restart, logs"
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
Reference in New Issue
Block a user