# 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 main https://github.com/edmcman/DIRTY # 10-3-24
#ADD ./DIRTY /DIRTY

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 (cd /root/.cache; (echo "b1e812b758eccf402271607c40fa491b5486742abf3706be174dc3f4fe87b9dd data1.tar.bz2" | sha256sum -c || wget -O '/root/.cache/data1.tar.bz2' 'https://cmu.box.com/shared/static/nx9fyn8jx0i9p4bftw8f2giqlufnoyj5')) && tar -xvjf /root/.cache/data1.tar.bz2 -C /DIRTY/dirty
RUN pip install --break-system-packages huggingface_hub[cli]

RUN --mount=type=cache,target=/tmp/model-dl huggingface-cli download --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"]