customize bash prompt in our docker images

a16z · Nov 13, 2024 · b1ebbc2 · b1ebbc2
1 parent 12232f8
commit b1ebbc2
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 1 deletion.
diff --git a/packages/halmos-builder/Dockerfile b/packages/halmos-builder/Dockerfile
@@ -23,3 +23,7 @@ RUN curl -sSf -L https://raw.githubusercontent.com/foundry-rs/foundry/master/fou
     -o /root/.foundry/bin/foundryup && \
     chmod +x /root/.foundry/bin/foundryup && \
     foundryup
+
+# Set a nicer prompt
+ENV IMAGE_NAME=halmos-builder
+RUN echo 'PS1="($IMAGE_NAME) \[\033[1;32m\]\u@\h \[\033[1;35m\]\w \$\[\033[0m\] "' >> /root/.bashrc
diff --git a/packages/halmos/Dockerfile b/packages/halmos/Dockerfile
@@ -11,5 +11,9 @@ RUN --mount=type=bind,source=../..,target=/src,readonly=false \
     uv pip install --no-cache /src && \
     uv pip install --no-cache -r /src/requirements-dev.txt
 
+# Set a nicer prompt
+ENV IMAGE_NAME=halmos
+RUN echo 'PS1="($IMAGE_NAME) \[\033[1;32m\]\u@\h \[\033[1;35m\]\w \$\[\033[0m\] "' >> /root/.bashrc
+
 WORKDIR /workspace
 CMD ["halmos"]
diff --git a/packages/solvers/Dockerfile b/packages/solvers/Dockerfile
@@ -90,9 +90,13 @@ COPY --from=ghcr.io/astral-sh/uv:latest /uv /uvx /bin/
 # Install jsi
 RUN uv tool install just-solve-it
 
-# make installed tools available
+# Make installed uv tools available
 ENV PATH=/root/.local/bin:$PATH
 
+# Set a nicer prompt
+ENV IMAGE_NAME=solvers
+RUN echo 'PS1="($IMAGE_NAME) \[\033[1;32m\]\u@\h \[\033[1;35m\]\w \$\[\033[0m\] "' >> /root/.bashrc
+
 # Set the default command for the container
 WORKDIR /workspace
 CMD ["/bin/bash"]