# From https://huggingface.co/spaces/sayakpaul/demo-docker-gradio/blob/main/Dockerfile

FROM blacktop/ghidra:latest

WORKDIR /code

COPY ./requirements.txt /code/requirements.txt

RUN --mount=type=cache,target=/var/cache/apt,sharing=locked \
  --mount=type=cache,target=/var/lib/apt,sharing=locked \
  apt-get -y update && apt-get -y install -y python3-pip python-is-python3 \
  git pkg-config libsentencepiece-dev libprotobuf-dev nano sudo unzip

RUN --mount=type=cache,target=/root/.cache pip install --break-system-packages --upgrade -r /code/requirements.txt

# Install Ghidrathon

WORKDIR /tmp/

RUN wget https://github.com/mandiant/Ghidrathon/releases/download/v4.0.0/Ghidrathon-v4.0.0.zip
RUN unzip Ghidrathon-v4.0.0.zip -d ghidrathon
RUN --mount=type=cache,target=/root/.cache pip install --break-system-packages -r ghidrathon/requirements.txt
RUN python ghidrathon/ghidrathon_configure.py /ghidra
RUN unzip ghidrathon/Ghidrathon-v4.0.0.zip -d /ghidra/Ghidra/Extensions

# Install DIRTY Ghidra

WORKDIR /

RUN git clone -b development https://github.com/edmcman/DIRTY-ghidra /DIRTY # 11/25/24

RUN --mount=type=cache,target=/root/.cache pip install --break-system-packages --upgrade -r /DIRTY/requirements.txt

# Download the DIRTY model
RUN --mount=type=cache,target=/root/.cache pip install --break-system-packages huggingface_hub[cli]

RUN --mount=type=cache,target=/tmp/model-dl huggingface-cli download --revision release/0.2 --repo-type model ejschwartz/dirty-ghidra --local-dir /tmp/model-dl && cp -R /tmp/model-dl/data1 /DIRTY/dirty/data1

# Set up a new user named "user"
RUN useradd -m -o -u 1000 user

# Switch to the "user" user
USER user

# Set home to the user's home directory
ENV HOME=/home/user \
    PATH=/home/user/.local/bin:$PATH

# Set the working directory to the user's home directory
WORKDIR $HOME/app

# Copy the current directory contents into the container at $HOME/app setting the owner to the user
COPY --chown=user . $HOME/app

CMD ["python", "main.py"]