add dockerization with extension

Dockerfile

@@ -1,29 +1,158 @@
-FROM python:3.11
+# Use Ubuntu 22.04 as the base image with multi-arch support
+FROM --platform=$TARGETPLATFORM ubuntu:22.04
 
-# Download precompiled ttyd binary from GitHub releases
-RUN apt-get update && \
-    apt-get install -y wget && \
-    wget https://github.com/tsl0922/ttyd/releases/download/1.6.3/ttyd.x86_64 -O /usr/bin/ttyd && \
-    chmod +x /usr/bin/ttyd && \
-    apt-get remove -y wget && \
-    apt-get autoremove -y && \
-    rm -rf /var/lib/apt/lists/*
+# Set environment to prevent interactive prompts during builds
+ENV DEBIAN_FRONTEND=noninteractive
+ENV TZ=Etc/UTC
 
-ENV NVM_DIR /root/.nvm
+# Use buildx args for multi-arch support
+ARG TARGETPLATFORM
+ARG BUILDPLATFORM
+ARG commitHash
 
-RUN curl -o- https://raw.githubusercontent.com/nvm-sh/nvm/v0.39.5/install.sh | bash \
-    && . "$NVM_DIR/nvm.sh" \
-    && nvm install node \
-    && nvm use node
+# Update package list and install prerequisites
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    software-properties-common \
+    build-essential \
+    curl \
+    git \
+    gnupg \
+    tzdata \
+    openssh-server \
+    && rm -rf /var/lib/apt/lists/*
 
-WORKDIR /usr/src/app
-COPY . .
-RUN pip install --no-cache-dir -r requirements.txt
-RUN python -m venv venv
-RUN /bin/bash -c "source venv/bin/activate"
+# Add deadsnakes PPA for Python 3.12 and install Python
+RUN add-apt-repository ppa:deadsnakes/ppa -y && apt-get update && \
+    apt-get install -y --no-install-recommends \
+    python3.12 \
+    python3.12-venv \
+    python3.12-dev \
+    python3-pip \
+    && rm -rf /var/lib/apt/lists/*
 
-RUN pip install -r requirements.txt
-WORKDIR /usr/src/app/pilot
 
-EXPOSE 7681
-CMD ["ttyd", "bash"]
+# Set Python 3.12 as the default python3 and python
+RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.12 1 && \
+    update-alternatives --install /usr/bin/python python /usr/bin/python3 1 && \
+    python --version
+
+RUN curl -fsSL https://deb.nodesource.com/setup_lts.x | bash - && \
+    apt-get install -y nodejs && \
+    node --version && npm --version
+
+# MongoDB installation with platform-specific approach
+RUN case "$TARGETPLATFORM" in \
+    "linux/amd64") \
+        curl -fsSL https://www.mongodb.org/static/pgp/server-6.0.asc | gpg --dearmor -o /usr/share/keyrings/mongodb-archive-keyring.gpg && \
+        echo "deb [arch=amd64 signed-by=/usr/share/keyrings/mongodb-archive-keyring.gpg] https://repo.mongodb.org/apt/ubuntu jammy/mongodb-org/6.0 multiverse" | tee /etc/apt/sources.list.d/mongodb-org-6.0.list && \
+        apt-get update && apt-get install -y mongodb-org \
+        ;; \
+    "linux/arm64"|"linux/arm64/v8") \
+        curl -fsSL https://www.mongodb.org/static/pgp/server-6.0.asc | gpg --dearmor -o /usr/share/keyrings/mongodb-archive-keyring.gpg && \
+        echo "deb [arch=arm64 signed-by=/usr/share/keyrings/mongodb-archive-keyring.gpg] https://repo.mongodb.org/apt/ubuntu jammy/mongodb-org/6.0 multiverse" | tee /etc/apt/sources.list.d/mongodb-org-6.0.list && \
+        apt-get update && apt-get install -y mongodb-org \
+        ;; \
+    *) \
+        echo "Unsupported platform: $TARGETPLATFORM" && exit 1 \
+        ;; \
+    esac \
+    && rm -rf /var/lib/apt/lists/*
+
+# Configure SSH
+RUN mkdir -p /run/sshd \
+    && sed -i 's/#PermitRootLogin prohibit-password/PermitRootLogin no/' /etc/ssh/sshd_config \
+    && sed -i 's/#PubkeyAuthentication yes/PubkeyAuthentication yes/' /etc/ssh/sshd_config \
+    && sed -i 's/#PasswordAuthentication yes/PasswordAuthentication no/' /etc/ssh/sshd_config \
+    && sed -i 's/#ChallengeResponseAuthentication yes/ChallengeResponseAuthentication no/' /etc/ssh/sshd_config
+
+# Set up work directory
+WORKDIR /pythagora/gpt-pilot/pythagora-core
+
+# Add Python requirements
+ADD requirements.txt .
+
+# Create and activate a virtual environment, then install dependencies
+RUN python3 -m venv venv && \
+    . venv/bin/activate && \
+    pip install -r requirements.txt
+
+# Copy application files
+ADD main.py .
+ADD core core
+ADD config.json .
+
+# Set the virtual environment to be automatically activated
+ENV VIRTUAL_ENV=/pythagora/gpt-pilot/pythagora-core/venv
+ENV PATH="$VIRTUAL_ENV/bin:$PATH"
+
+# Expose MongoDB and application ports
+EXPOSE 27017 8000
+
+ENV MONGO_DB_DATA=/pythagora/gpt-pilot/pythagora-core/.mongodata
+RUN mkdir -p $MONGO_DB_DATA
+
+# Create a group named "devusergroup" with a specific GID (1000, optional)
+RUN groupadd -g 1000 devusergroup
+
+ARG USERNAME=devuser
+
+# Create a user named "devuser" with a specific UID (1000) and assign it to "devusergroup"
+RUN useradd -m -u 1000 -g devusergroup -s /bin/bash $USERNAME
+
+# Add the user to sudoers for admin privileges
+RUN echo "$USERNAME ALL=(ALL) NOPASSWD:ALL" >> /etc/sudoers
+
+# Create an embedded entrypoint script
+ADD entrypoint.sh /entrypoint.sh
+RUN chmod +x /entrypoint.sh
+
+RUN chown -R $USERNAME:devusergroup /pythagora
+
+# Copy SSH public key from secret
+#RUN --mount=type=secret,id=ssh_public_key \
+#    mkdir -p /home/${USERNAME}/.ssh \
+#    && cat /run/secrets/ssh_public_key > /home/${USERNAME}/.ssh/authorized_keys \
+#    && chown -R ${USERNAME}:devusergroup /home/${USERNAME}/.ssh \
+#    && chmod 700 /home/${USERNAME}/.ssh \
+#    && chmod 600 /home/${USERNAME}/.ssh/authorized_keys
+
+USER $USERNAME
+
+RUN mkdir -p ~/.ssh && touch ~/.ssh/authorized_keys && chmod 600 ~/.ssh/authorized_keys
+
+# VS Code server installation with platform-specific handling
+RUN case "$TARGETPLATFORM" in \
+    "linux/amd64") \
+        mkdir -p ~/.vscode-server/cli/servers/Stable-${commitHash} && \
+        curl -fsSL https://update.code.visualstudio.com/commit:${commitHash}/server-linux-x64/stable -o server-linux-x64.tar.gz && \
+        tar -xz -f server-linux-x64.tar.gz -C ~/.vscode-server/cli/servers/Stable-${commitHash} && \
+        mv ~/.vscode-server/cli/servers/Stable-${commitHash}/vscode-server-linux-x64 ~/.vscode-server/cli/servers/Stable-${commitHash}/server \
+        ;; \
+    "linux/arm64"|"linux/arm64/v8") \
+        mkdir -p ~/.vscode-server/cli/servers/Stable-${commitHash} && \
+        curl -fsSL https://update.code.visualstudio.com/commit:${commitHash}/server-linux-arm64/stable -o server-linux-arm64.tar.gz && \
+        tar -xz -f server-linux-arm64.tar.gz -C ~/.vscode-server/cli/servers/Stable-${commitHash} && \
+        mv ~/.vscode-server/cli/servers/Stable-${commitHash}/vscode-server-linux-arm64 ~/.vscode-server/cli/servers/Stable-${commitHash}/server \
+        ;; \
+    *) \
+        echo "Unsupported platform: $TARGETPLATFORM" && exit 1 \
+        ;; \
+    esac
+
+# Create a workspace directory
+WORKDIR /pythagora/gpt-pilot/workspace
+
+# Create default VS Code settings with dark theme
+# RUN mkdir -p ~/.local/share/code-server/User \
+#    && echo '{"workbench.colorTheme": "Dark", "workbench.startupEditor": "none"}' \
+#    > ~/.local/share/code-server/User/settings.json
+
+# Install VS Code extension (platform-agnostic)
+ADD pythagora-vs-code-1.1.2.vsix pythagora-vs-code.vsix
+RUN ~/.vscode-server/cli/servers/Stable-${commitHash}/server/bin/code-server --install-extension pythagora-vs-code.vsix
+RUN rm pythagora-vs-code.vsix
+
+USER root
+
+# Set the entrypoint to the main application
+ENTRYPOINT ["/entrypoint.sh"]

entrypoint.sh

@@ -0,0 +1,13 @@
+#!/bin/bash
+
+if [ -z "$SSH_KEY_B64" ]; then
+  echo "Environment variable SSH_KEY_B64 is not set. Exiting."
+  exit 1
+fi
+
+echo "$SSH_KEY_B64" | base64 -d >> /home/devuser/.ssh/authorized_keys
+chmod 600 /home/devuser/.ssh/authorized_keys
+
+mongod --dbpath "$MONGO_DB_DATA" --bind_ip_all >> $MONGO_DB_DATA/mongo_logs.txt 2>&1 &
+
+/usr/sbin/sshd -D