Added Docker support, setup, and uninstall scripts for CVE Scanner CLI

Author: DebaA17

.dockerignore

@@ -0,0 +1,8 @@
+myenv/
+__pycache__/
+*.pyc
+*.pyo
+*.pyd
+.git/
+.gitignore
+demo/

Dockerfile

@@ -0,0 +1,15 @@
+FROM python:3.13-slim
+
+WORKDIR /app
+
+ENV PYTHONDONTWRITEBYTECODE=1 \
+    PYTHONUNBUFFERED=1
+
+COPY requirements.txt ./
+
+RUN pip install --no-cache-dir --upgrade pip \
+    && pip install --no-cache-dir -r requirements.txt
+
+COPY cve_search_cli.py ./
+
+ENTRYPOINT ["python", "/app/cve_search_cli.py"]

README.md

@@ -30,9 +30,43 @@ bash setup.sh
 ```
 This will install all dependencies in a virtual environment and create a `cvecli` command in your `~/.local/bin` directory.
 
-Make sure `~/.local/bin` is in your PATH:
+The setup script also adds `~/.local/bin` to your shell startup file if it is missing. If your current terminal was already open before setup, reload it with:
 ```bash
-export PATH="$HOME/.local/bin:$PATH"
+source ~/.bashrc
+```
+
+If you use zsh, run:
+
+```bash
+source ~/.zshrc
+```
+
+## Uninstall
+To remove the global `cvecli` command and the PATH entry added by setup, run:
+
+```bash
+bash uninstall.sh
+```
+
+This removes the `~/.local/bin/cvecli` launcher and cleans the PATH line from your shell startup files. It does not delete the repository folder or `myenv`.
+
+## Docker Setup
+Build and run the CLI with Docker in one line:
+
+```bash
+docker build -t cvecli . && docker run --rm -it cvecli
+```
+
+Run a direct CVE lookup without entering interactive mode:
+
+```bash
+docker build -t cvecli . && docker run --rm cvecli --id CVE-2025-55184
+```
+
+Run a keyword search:
+
+```bash
+docker build -t cvecli . && docker run --rm cvecli --keyword wordpress
 ```
 
 ## Usage

cve_search_cli.py

@@ -122,4 +122,8 @@ def main():
         interactive_menu()
 
 if __name__ == "__main__":
-    main()
+    try:
+        main()
+    except KeyboardInterrupt:
+        console.print("\n[bold yellow]Interrupted by user. Exiting.[/]")
+        sys.exit(130)

setup.sh

@@ -25,6 +25,37 @@ chmod +x cve_search_cli.py
 mkdir -p "$HOME/.local/bin"
 ln -sf "$PWD/cve_search_cli.py" "$HOME/.local/bin/cvecli"
 
+PATH_EXPORT='export PATH="$HOME/.local/bin:$PATH"'
+SHELL_NAME="$(basename "${SHELL:-}")"
+RC_FILE=""
+
+case "$SHELL_NAME" in
+    bash)
+        if [ -f "$HOME/.bashrc" ]; then
+            RC_FILE="$HOME/.bashrc"
+        else
+            RC_FILE="$HOME/.profile"
+        fi
+        ;;
+    zsh)
+        RC_FILE="$HOME/.zshrc"
+        ;;
+    *)
+        if [ -f "$HOME/.profile" ]; then
+            RC_FILE="$HOME/.profile"
+        fi
+        ;;
+esac
+
+if [ -n "$RC_FILE" ]; then
+    touch "$RC_FILE"
+    if ! grep -Fqx "$PATH_EXPORT" "$RC_FILE"; then
+        printf '\n%s\n' "$PATH_EXPORT" >> "$RC_FILE"
+        printf "\033[0;32mAdded ~/.local/bin to PATH in %s\033[0m\n" "$RC_FILE"
+        printf "\033[0;33mOpen a new terminal or run: source %s\033[0m\n" "$RC_FILE"
+    fi
+fi
+
 # Print green success message
 printf "\033[0;32mSetup complete! Use 'cvecli' from anywhere to run the tool.\033[0m\n"
 printf "\033[0;32mExample: cvecli --id CVE-2025-55184\033[0m\n"

uninstall.sh

@@ -0,0 +1,42 @@
+#!/usr/bin/env bash
+set -e
+
+CLI_LINK="$HOME/.local/bin/cvecli"
+PATH_EXPORT='export PATH="$HOME/.local/bin:$PATH"'
+REMOVED_PATH_LINE=0
+
+remove_path_line() {
+    local rc_file="$1"
+    local temp_file
+
+    if [ ! -f "$rc_file" ]; then
+        return
+    fi
+
+    if ! grep -Fqx "$PATH_EXPORT" "$rc_file"; then
+        return
+    fi
+
+    temp_file="$(mktemp)"
+    grep -Fvx "$PATH_EXPORT" "$rc_file" > "$temp_file" || true
+    mv "$temp_file" "$rc_file"
+    REMOVED_PATH_LINE=1
+    printf "\033[0;32mRemoved PATH entry from %s\033[0m\n" "$rc_file"
+}
+
+if [ -L "$CLI_LINK" ] || [ -f "$CLI_LINK" ]; then
+    rm -f "$CLI_LINK"
+    printf "\033[0;32mRemoved %s\033[0m\n" "$CLI_LINK"
+else
+    printf "\033[0;33mNo cvecli launcher found at %s\033[0m\n" "$CLI_LINK"
+fi
+
+remove_path_line "$HOME/.bashrc"
+remove_path_line "$HOME/.zshrc"
+remove_path_line "$HOME/.profile"
+
+printf "\033[0;32mUninstall complete.\033[0m\n"
+
+if [ "$REMOVED_PATH_LINE" -eq 1 ]; then
+    printf "\033[0;33mOpen a new terminal or reload your shell config to apply the PATH change.\033[0m\n"
+fi