feature(manage): add stop onl option

Author: zdszero

manage

@@ -2,7 +2,7 @@
 
 function print_help {
   echo "usage:"
-  echo "    ./manage [makemigrations|migrate|clean|rebuild|run]"
+  echo "    ./manage [makemigrations|migrate|clean|rebuild|run|stop]"
 }
 
 [[ $# -ne 1 ]] && print_help && exit
@@ -27,30 +27,50 @@ function migrate {
   python3 manage.py inituser --username=root --password=rootroot --action=create_super_admin
 }
 
-function run {
-  # run dramatiq
-  pgrep dramatiq > /dev/null
+function run_onl {
+  [[ ! -d data/log ]] && mkdir -p data/log && echo 'mkdir data/log'
 
   # using postgres
   # docker run -it -d -e POSTGRES_DB=onlinejudge -e POSTGRES_USER=onlinejudge -e POSTGRES_PASSWORD=onlinejudge -p 5435:5432 --name oj-postgres-dev postgres:10-alpine
 
+  # run dramatiq
+  pgrep dramatiq > /dev/null
   if [[ $? -ne 0 ]]; then
     echo "running dramatiq ..."
     python3 manage.py rundramatiq &>data/log/dramatiq.log &
   else
     echo "dramatiq is already running ..."
   fi
-  docker container ls > /dev/null
-  
+
+  # run redis
+  docker container ls | grep redis > /dev/null
   if [[ $? -ne 0 ]]; then
     echo "running redis in docker ..."
-    docker run -it -d -p 6380:6379 --name oj-redis-dev redis:4.0-alpine
+    docker ps -a | grep redis > /dev/null
+    if [[ $? -ne 0 ]]; then
+      docker run -it -d -p 6380:6379 --name oj-redis-dev redis:4.0-alpine > /dev/null
+    else
+      container_id=$(docker ps -a | grep redis | awk '{print $1}')
+      docker container start $container_id
+    fi
   else
     echo "redis is already running"
   fi
 
-  echo "running onl ..."
-  python3 manage.py runserver 0.0.0.0:7890 &>data/log/onl.log &
+  # run onl
+  pgrep -f "python3 manage.py runserver" > /dev/null
+  if [[ $? -ne 0 ]]; then
+    echo "running onl ..."
+    python3 manage.py runserver 0.0.0.0:7890 &>data/log/onl.log &
+  else
+    echo "onl is already running ..."
+  fi
+}
+
+function stop_onl {
+  pkill dramatiq
+  pkill -f "python3 manage.py runserver"
+  docker container stop oj-redis-dev > /dev/null
 }
 
 if [[ $cmd == "make" || $cmd == "makemigrations" ]]; then
@@ -65,5 +85,7 @@ elif [[ $cmd == "rebuild" ]]; then
   clean
   migrate
 elif [[ $cmd == "run" ]]; then
-  run
+  run_onl
+elif [[ $cmd == "stop" ]]; then
+  stop_onl
 fi