add server

Dockerfile

@@ -1,28 +1,36 @@
 FROM python:3.9
 
+# Create user first
 RUN useradd -m -u 1000 aim_user
 
-# Switch to the "aim_user" user
+# Perform root operations: Install packages, copy system configs
+RUN apt-get update && apt-get install -y nginx procps && rm -rf /var/lib/apt/lists/*
+COPY nginx.conf /etc/nginx/nginx.conf
+COPY start.sh /start.sh
+RUN chmod +x /start.sh
+
+# Now switch to the final user
 USER aim_user
 
-# Set home to the user's home directory
+# Set home and path
 ENV HOME=/home/aim_user \
 	PATH=/home/aim_user/.local/bin:$PATH
 
-# Set the working directory to the user's home directory
+# Set the working directory
 WORKDIR $HOME
 
-# install the `aim` package on the latest version
-RUN pip install aim
+# install aim (as aim_user)
+RUN pip install aim --no-cache-dir
 
 ENTRYPOINT ["/bin/sh", "-c"]
 
 COPY aim_repo.tar.gz .
 RUN tar xvzf aim_repo.tar.gz
+
 # have to run `aim init` in the directory that stores aim data for
 # otherwise `aim up` will prompt for confirmation to create the directory itself.
-# We run aim listening on 0.0.0.0 to expose all ports. Also, we run
-# using `--dev` to print verbose logs. Port 43800 is the default port of
-# `aim up` but explicit is better than implicit.
-CMD ["aim up --host 0.0.0.0 --port 7860 --workers 2"]
+# We run aim listening on 0.0.0.0 to expose all ports.
+# Port 43800 for UI, 53800 for Server. Nginx listens on 7860.
 
+# Run the startup script as aim_user
+CMD ["/start.sh"]

nginx.conf

@@ -0,0 +1,36 @@
+worker_processes 1;
+pid /home/aim_user/nginx_run/nginx.pid; # Needs a path writable by aim_user
+error_log /home/aim_user/nginx_logs/error.log warn; # Needs a path writable by aim_user
+
+events {
+    worker_connections 1024;
+}
+
+http {
+    access_log /home/aim_user/nginx_logs/access.log; # Needs a path writable by aim_user
+
+    server {
+        listen 7860;
+        server_name localhost;
+
+        location / {
+            proxy_pass http://127.0.0.1:43800;
+            proxy_set_header Host $host;
+            proxy_set_header X-Real-IP $remote_addr;
+            proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
+            proxy_set_header X-Forwarded-Proto $scheme;
+            # Add WebSocket support for Aim UI
+            proxy_http_version 1.1;
+            proxy_set_header Upgrade $http_upgrade;
+            proxy_set_header Connection "upgrade";
+        }
+
+        location /api/ {
+            proxy_pass http://127.0.0.1:53800/api/; # Pass requests to aim server API
+            proxy_set_header Host $host;
+            proxy_set_header X-Real-IP $remote_addr;
+            proxy_set_header X-Forwarded-For $proxy_add_x_forwarded_for;
+            proxy_set_header X-Forwarded-Proto $scheme;
+        }
+    }
+}

start.sh

@@ -0,0 +1,46 @@
+#!/bin/bash
+
+# Create directories for nginx logs and pid file within $HOME
+mkdir -p $HOME/nginx_run $HOME/nginx_logs
+
+# Start Aim UI in the background
+echo "Starting Aim UI on port 43800..."
+# Run directly as aim_user
+aim up --host 0.0.0.0 --port 43800 --repo $HOME &
+AIM_UP_PID=$!
+
+# Start Aim Server in the background
+echo "Starting Aim Server on port 53800..."
+# Run directly as aim_user
+aim server --host 0.0.0.0 --port 53800 --repo $HOME &
+AIM_SERVER_PID=$!
+
+# Wait a few seconds for services to potentially start
+sleep 5
+
+# Check if aim processes are still running
+if ! ps -p $AIM_UP_PID > /dev/null; then
+    echo "Aim UI failed to start." >&2
+    # exit 1 # Optional: exit if a service fails
+fi
+
+if ! ps -p $AIM_SERVER_PID > /dev/null; then
+    echo "Aim Server failed to start." >&2
+    # exit 1 # Optional: exit if a service fails
+fi
+
+echo "Aim services seem to be starting..."
+
+# Start nginx in the foreground using the system config file
+# Run directly as aim_user (no sudo needed)
+echo "Starting nginx..."
+nginx -c /etc/nginx/nginx.conf -g 'daemon off;'
+
+# If nginx exits, stop the background aim processes
+echo "Nginx exited. Stopping Aim services..."
+# Use kill without sudo, as processes are owned by aim_user
+kill $AIM_UP_PID $AIM_SERVER_PID
+# Wait for processes to ensure they are terminated
+wait $AIM_UP_PID $AIM_SERVER_PID
+
+echo "Exiting."