use systemd

Author: mishushakov

template/code-interpreter.service

@@ -0,0 +1,14 @@
+[Unit]
+Description=Code Interpreter Server
+After=jupyter.service
+Requires=jupyter.service
+
+[Service]
+Type=simple
+WorkingDirectory=/root/.server
+ExecStartPre=/bin/bash -c 'until curl -sf http://localhost:8888/api/status > /dev/null 2>&1; do sleep 0.1; done'
+ExecStart=/root/.server/.venv/bin/uvicorn main:app --host 0.0.0.0 --port 49999 --workers 1 --no-access-log --no-use-colors --timeout-keep-alive 640
+Restart=on-failure
+
+[Install]
+WantedBy=multi-user.target

template/jupyter.service

@@ -0,0 +1,13 @@
+[Unit]
+Description=Jupyter Server
+
+[Service]
+Type=simple
+Environment=MATPLOTLIBRC=/root/.config/matplotlib/.matplotlibrc
+ExecStart=/usr/local/bin/jupyter server --IdentityProvider.token=""
+Restart=on-failure
+StandardOutput=journal
+StandardError=journal
+
+[Install]
+WantedBy=multi-user.target

template/start-up.sh

@@ -1,22 +1,5 @@
 #!/bin/bash
 
-function start_jupyter_server() {
-	counter=0
-	response=$(curl -s -o /dev/null -w "%{http_code}" "http://localhost:8888/api/status")
-	while [[ ${response} -ne 200 ]]; do
-		let counter++
-		if ((counter % 20 == 0)); then
-			echo "Waiting for Jupyter Server to start..."
-			sleep 0.1
-		fi
-
-		response=$(curl -s -o /dev/null -w "%{http_code}" "http://localhost:8888/api/status")
-	done
-
-	cd /root/.server/
-	.venv/bin/uvicorn main:app --host 0.0.0.0 --port 49999 --workers 1 --no-access-log --no-use-colors --timeout-keep-alive 640
-}
-
 echo "Starting Code Interpreter server..."
-start_jupyter_server &
-MATPLOTLIBRC=/root/.config/matplotlib/.matplotlibrc jupyter server --IdentityProvider.token="" >/dev/null 2>&1
+systemctl daemon-reload
+systemctl start jupyter code-interpreter

template/template.py

@@ -38,6 +38,7 @@ def make_template(
                 "sudo",
                 "fonts-noto-cjk",
                 "ca-certificates",
+                "systemd",
             ]
         )
         .run_cmd("curl -fsSL https://deb.nodesource.com/setup_20.x | bash -")
@@ -126,6 +127,15 @@ def make_template(
         .make_dir(".ipython/profile_default/startup")
         .copy("ipython_kernel_config.py", ".ipython/profile_default/")
         .copy("startup_scripts", ".ipython/profile_default/startup")
+        # Install systemd service units
+        .copy("jupyter.service", ".jupyter/jupyter.service")
+        .copy("code-interpreter.service", ".jupyter/code-interpreter.service")
+        .run_cmd(
+            [
+                "cp /root/.jupyter/jupyter.service /etc/systemd/system/",
+                "cp /root/.jupyter/code-interpreter.service /etc/systemd/system/",
+            ]
+        )
     )
 
     if is_docker: