FROM ubuntu:latest

# 避免安装过程中出现交互提示
ENV DEBIAN_FRONTEND=noninteractive

# 安装 SSH 服务、Python 以及 webssh 所需的依赖
RUN apt-get update && \
    apt-get install -y openssh-server python3 python3-pip && \
    apt-get clean && \
    rm -rf /var/lib/apt/lists/*

# 配置 SSH 服务
RUN mkdir /var/run/sshd
RUN echo 'PermitRootLogin yes' >> /etc/ssh/sshd_config
RUN echo 'PasswordAuthentication yes' >> /etc/ssh/sshd_config

# 安装 webssh (添加参数以适配较新版 Ubuntu 的 Python 环境限制)
RUN pip3 install webssh --break-system-packages

# 编写启动脚本，动态设置密码并同时启动 sshd 与 webssh
RUN echo '#!/bin/bash\n\
# 检查是否设置了密码环境变量\n\
if [ -z "$SSH_PASSWORD" ]; then\n\
  echo "请在 Hugging Face 设置中配置 SSH_PASSWORD 环境变量"\n\
  exit 1\n\
fi\n\
\n\
# 将环境变量的值赋予 root 用户作为登录密码\n\
echo "root:$SSH_PASSWORD" | chpasswd\n\
\n\
# 启动原生 SSH 服务\n\
/usr/sbin/sshd\n\
\n\
# 在前台启动 webssh，并绑定 Hugging Face 默认的 7860 端口\n\
wssh --address=0.0.0.0 --port=7860 --xsrf=False\n\
' > /start.sh

# 赋予启动脚本执行权限
RUN chmod +x /start.sh

# 暴露 Hugging Face 要求的默认网络端口
EXPOSE 7860

# 设置容器启动时执行的命令
CMD ["/start.sh"]