FROM kalilinux/kali-rolling

# Install Python and dependencies
RUN apt-get update && apt-get install -y \
    python3 \
    python3-pip \
    python3-flask \
    python3-requests \
    && rm -rf /var/lib/apt/lists/*

# Install MCP SDK
RUN pip3 install --break-system-packages mcp requests

# Install essential Kali tools
RUN apt-get update && apt-get install -y \
    nmap \
    gobuster \
    dirb \
    nikto \
    sqlmap \
    metasploit-framework \
    hydra \
    john \
    wpscan \
    aircrack-ng \
    binwalk \
    hashcat \
    masscan \
    radare2 \
    sherlock \
    wireshark \
    commix \
    cewl \
    crunch \
    exploitdb \
    steghide \
    && rm -rf /var/lib/apt/lists/*

# Set working directory
WORKDIR /app

# Copy application files
COPY . .

# Expose ports
EXPOSE 5000 8000

# Health check
HEALTHCHECK --interval=30s --timeout=10s --start-period=30s --retries=3 \
    CMD python3 -c "import requests; requests.get('http://localhost:5000/health')" || exit 1

# Start the server
CMD ["python3", "server.py"]