own-rag-cli 0.0.1-snapshot

package/bin/postinstall.sh

@@ -0,0 +1,102 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+PACKAGE_ROOT="$(dirname "${SCRIPT_DIR}")"
+
+LOCAL_BIN_DIR="${HOME}/.local/bin"
+WRAPPER_SRC="${PACKAGE_ROOT}/bin/rag-wrapper.sh"
+WRAPPER_DEST="${LOCAL_BIN_DIR}/rag-wrapper.sh"
+RAG_BIN_DEST="${LOCAL_BIN_DIR}/rag"
+RAG_SETUP_SRC="${PACKAGE_ROOT}/rag-setup.run"
+RAG_SETUP_DEST="${LOCAL_BIN_DIR}/rag-setup.run"
+RAG_SETUP_MACOS_SRC="${PACKAGE_ROOT}/rag-setup-macos.run"
+RAG_SETUP_MACOS_DEST="${LOCAL_BIN_DIR}/rag-setup-macos.run"
+MONITOR_SRC="${PACKAGE_ROOT}/chroma_monitor.sh"
+MONITOR_DEST="${LOCAL_BIN_DIR}/chroma_monitor.sh"
+REMOVE_SRC="${PACKAGE_ROOT}/bin/rag-remove.sh"
+REMOVE_DEST="${LOCAL_BIN_DIR}/rag-remove.sh"
+
+COMPOSE_SOURCE="${PACKAGE_ROOT}/bin/docker-compose.yml"
+COMPOSE_DIR="${HOME}/docker-chromadb"
+COMPOSE_TARGET="${COMPOSE_DIR}/docker-compose.yml"
+
+ALIAS_LINE="alias rag='~/.local/bin/rag-wrapper.sh'"
+
+log_info() { printf "[+] %s\n" "$*"; }
+log_warn() { printf "[!] %s\n" "$*" >&2; }
+
+mkdir -p "${LOCAL_BIN_DIR}"
+
+if [[ -f "${WRAPPER_SRC}" ]]; then
+    cp "${WRAPPER_SRC}" "${WRAPPER_DEST}"
+    chmod +x "${WRAPPER_DEST}"
+    ln -sf "${WRAPPER_DEST}" "${RAG_BIN_DEST}"
+    log_info "Wrapper instalado: ${WRAPPER_DEST}"
+fi
+
+if [[ -f "${RAG_SETUP_SRC}" ]]; then
+    cp "${RAG_SETUP_SRC}" "${RAG_SETUP_DEST}"
+    chmod +x "${RAG_SETUP_DEST}"
+    log_info "Instalador Linux copiado: ${RAG_SETUP_DEST}"
+fi
+
+if [[ -f "${RAG_SETUP_MACOS_SRC}" ]]; then
+    cp "${RAG_SETUP_MACOS_SRC}" "${RAG_SETUP_MACOS_DEST}"
+    chmod +x "${RAG_SETUP_MACOS_DEST}"
+    log_info "Instalador macOS copiado: ${RAG_SETUP_MACOS_DEST}"
+fi
+
+if [[ -f "${MONITOR_SRC}" ]]; then
+    cp "${MONITOR_SRC}" "${MONITOR_DEST}"
+    chmod +x "${MONITOR_DEST}"
+    log_info "Monitor copiado: ${MONITOR_DEST}"
+fi
+
+if [[ -f "${REMOVE_SRC}" ]]; then
+    cp "${REMOVE_SRC}" "${REMOVE_DEST}"
+    chmod +x "${REMOVE_DEST}"
+    log_info "Remoção copiada: ${REMOVE_DEST}"
+fi
+
+DOCKER_COMPOSE_CMD=""
+if command -v docker >/dev/null 2>&1; then
+    if docker compose version >/dev/null 2>&1; then
+        DOCKER_COMPOSE_CMD="docker compose"
+    elif command -v docker-compose >/dev/null 2>&1; then
+        DOCKER_COMPOSE_CMD="docker-compose"
+    fi
+fi
+
+if [[ -n "${DOCKER_COMPOSE_CMD}" && -f "${COMPOSE_SOURCE}" ]]; then
+    mkdir -p "${COMPOSE_DIR}"
+    cp "${COMPOSE_SOURCE}" "${COMPOSE_TARGET}"
+    if docker info >/dev/null 2>&1; then
+        ${DOCKER_COMPOSE_CMD} -f "${COMPOSE_TARGET}" up -d || log_warn "Falha ao subir ChromaDB automaticamente."
+        log_info "ChromaDB iniciado com ${DOCKER_COMPOSE_CMD}."
+    else
+        log_warn "Docker instalado, mas daemon não está ativo. ChromaDB não foi iniciado."
+    fi
+else
+    log_warn "Docker Compose não encontrado. Pulei o auto-start do ChromaDB."
+fi
+
+for rc_file in "${HOME}/.bashrc" "${HOME}/.profile"; do
+    touch "${rc_file}"
+    if ! grep -qF "${ALIAS_LINE}" "${rc_file}"; then
+        printf "\n# own-rag npm install\n%s\n" "${ALIAS_LINE}" >> "${rc_file}"
+        log_info "Alias adicionado em ${rc_file}"
+    fi
+done
+
+if [[ -f "${HOME}/.bashrc" ]]; then
+    # shellcheck disable=SC1090
+    source "${HOME}/.bashrc" || true
+fi
+
+if [[ -f "${HOME}/.profile" ]]; then
+    # shellcheck disable=SC1090
+    source "${HOME}/.profile" || true
+fi
+
+log_info "Postinstall concluído. Use: rag run [path] | rag monitor | rag remove"

package/bin/rag-remove.sh

@@ -0,0 +1,198 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+usage() {
+    cat <<'EOF'
+Uso:
+  rag remove
+  rag remove --force
+
+Descrição:
+  Remove toda a instalação local do own-rag (venv, banco, docker, binários e configuração MCP).
+
+Opções:
+  --force, -f    pula a confirmação dupla.
+EOF
+}
+
+detect_os() {
+    case "$(uname -s)" in
+        Darwin) echo "macos" ;;
+        Linux) echo "linux" ;;
+        *) echo "unknown" ;;
+    esac
+}
+
+log_info() { printf "[+] %s\n" "$*"; }
+log_warn() { printf "[!] %s\n" "$*" >&2; }
+
+FORCE=false
+for arg in "$@"; do
+    case "$arg" in
+        --force|-f) FORCE=true ;;
+        --help|-h) usage; exit 0 ;;
+        *)
+            echo "Opção desconhecida: $arg" >&2
+            usage
+            exit 1
+            ;;
+    esac
+done
+
+OS_NAME="$(detect_os)"
+if [[ "${OS_NAME}" == "unknown" ]]; then
+    echo "[ERRO] Sistema operacional não suportado. Use Linux ou macOS." >&2
+    exit 1
+fi
+
+if [[ "${FORCE}" != "true" ]]; then
+    if [[ ! -t 0 ]]; then
+        echo "[ERRO] Terminal não interativo. Use --force para continuar." >&2
+        exit 1
+    fi
+
+    echo "[AVISO] Esta ação removerá completamente o own-rag deste computador."
+    echo "        Isso inclui venv, banco local, binários e configuração MCP."
+    read -r -p "Confirma a remoção completa? [s/N] " answer
+    answer="$(echo "${answer}" | tr '[:upper:]' '[:lower:]')"
+    if [[ "${answer}" != "s" && "${answer}" != "sim" && "${answer}" != "y" && "${answer}" != "yes" ]]; then
+        echo "Remoção cancelada."
+        exit 0
+    fi
+
+    read -r -p "Digite REMOVER para confirmar definitivamente: " confirm_word
+    if [[ "${confirm_word}" != "REMOVER" ]]; then
+        echo "Confirmação inválida. Remoção cancelada."
+        exit 1
+    fi
+fi
+
+log_info "Iniciando remoção completa (OS: ${OS_NAME})..."
+
+DOCKER_COMPOSE_CMD=""
+if command -v docker >/dev/null 2>&1; then
+    if docker compose version >/dev/null 2>&1; then
+        DOCKER_COMPOSE_CMD="docker compose"
+    elif command -v docker-compose >/dev/null 2>&1; then
+        DOCKER_COMPOSE_CMD="docker-compose"
+    fi
+fi
+
+if [[ -n "${DOCKER_COMPOSE_CMD}" && -f "${HOME}/docker-chromadb/docker-compose.yml" ]]; then
+    if docker info >/dev/null 2>&1; then
+        (cd "${HOME}/docker-chromadb" && ${DOCKER_COMPOSE_CMD} down -v) || log_warn "Falha ao derrubar docker compose."
+    else
+        log_warn "Docker encontrado, mas daemon não está ativo; pulando compose down."
+    fi
+fi
+
+if command -v docker >/dev/null 2>&1; then
+    if docker ps -a --format '{{.Names}}' | grep -qx 'chromadb-rag'; then
+        docker rm -f chromadb-rag >/dev/null 2>&1 || log_warn "Falha ao remover container chromadb-rag."
+    fi
+fi
+
+declare -a REMOVE_PATHS=(
+    "${HOME}/.rag_venv"
+    "${HOME}/.rag_db"
+    "${HOME}/docker-chromadb"
+    "${HOME}/.cache/my-custom-rag-python"
+    "${HOME}/.cache/ny-custom-rag-python"
+    "${HOME}/.local/bin/mcp-rag-server"
+    "${HOME}/.local/bin/download_model_from_hugginface.py"
+    "${HOME}/.local/bin/download_model_from_modelscope.py"
+    "${HOME}/.local/bin/rag-wrapper.sh"
+    "${HOME}/.local/bin/rag-remove.sh"
+    "${HOME}/.local/bin/rag-setup.run"
+    "${HOME}/.local/bin/rag-setup-macos.run"
+    "${HOME}/.local/bin/chroma_monitor.sh"
+    "${HOME}/.local/bin/rag"
+)
+
+declare -a FAILED_PATHS=()
+
+remove_path() {
+    local target="$1"
+    if [[ -e "${target}" || -L "${target}" ]]; then
+        if rm -rf "${target}" 2>/dev/null; then
+            log_info "Removido: ${target}"
+        else
+            FAILED_PATHS+=("${target}")
+            log_warn "Falha ao remover (permissão): ${target}"
+        fi
+    fi
+}
+
+for target in "${REMOVE_PATHS[@]}"; do
+    remove_path "${target}"
+done
+
+python3 - <<'PY'
+import json
+from pathlib import Path
+
+home = Path.home()
+config_paths = [
+    home / ".claude.json",
+    home / ".cursor" / "mcp.json",
+    home / "Library" / "Application Support" / "Cursor" / "User" / "mcp.json",
+]
+for path in config_paths:
+    if not path.exists():
+        continue
+    try:
+        data = json.loads(path.read_text(encoding="utf-8"))
+    except Exception:
+        continue
+    servers = data.get("mcpServers")
+    if isinstance(servers, dict) and "rag-codebase" in servers:
+        servers.pop("rag-codebase", None)
+        try:
+            path.write_text(json.dumps(data, ensure_ascii=False, indent=2) + "\n", encoding="utf-8")
+            print(f"[+] MCP removido de: {path}")
+        except Exception as exc:
+            print(f"[!] Não foi possível atualizar {path}: {exc}")
+
+rc_files = [
+    home / ".bashrc",
+    home / ".profile",
+    home / ".zshrc",
+    home / ".zprofile",
+]
+for rc in rc_files:
+    if not rc.exists():
+        continue
+    try:
+        lines = rc.read_text(encoding="utf-8").splitlines()
+    except Exception:
+        continue
+
+    filtered = [
+        line for line in lines
+        if "alias rag=" not in line
+        and "rag-wrapper.sh" not in line
+        and "own-rag npm install" not in line
+    ]
+    if filtered != lines:
+        try:
+            rc.write_text("\n".join(filtered).rstrip() + "\n", encoding="utf-8")
+            print(f"[+] Limpeza de alias em: {rc}")
+        except Exception as exc:
+            print(f"[!] Não foi possível limpar {rc}: {exc}")
+PY
+
+if command -v npm >/dev/null 2>&1; then
+    npm uninstall -g own-rag >/dev/null 2>&1 || true
+fi
+
+if [[ ${#FAILED_PATHS[@]} -gt 0 ]]; then
+    echo ""
+    log_warn "Remoção concluída com pendências de permissão:"
+    for path in "${FAILED_PATHS[@]}"; do
+        echo "    - ${path}"
+    done
+    log_warn "Execute com sudo para limpar pendências, por exemplo:"
+    echo "    sudo rm -rf ${FAILED_PATHS[0]}"
+else
+    log_info "Remoção completa concluída."
+fi

package/bin/rag-wrapper.sh

@@ -0,0 +1,186 @@
+#!/usr/bin/env bash
+set -euo pipefail
+
+SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+LOCAL_BIN_DIR="${HOME}/.local/bin"
+LOCAL_SETUP_RUN_LINUX="${LOCAL_BIN_DIR}/rag-setup.run"
+LOCAL_SETUP_RUN_MACOS="${LOCAL_BIN_DIR}/rag-setup-macos.run"
+PACKAGE_SETUP_RUN_LINUX="${SCRIPT_DIR}/../rag-setup.run"
+PACKAGE_SETUP_RUN_MACOS="${SCRIPT_DIR}/../rag-setup-macos.run"
+LOCAL_MONITOR_SCRIPT="${LOCAL_BIN_DIR}/chroma_monitor.sh"
+PACKAGE_MONITOR_SCRIPT="${SCRIPT_DIR}/../chroma_monitor.sh"
+LOCAL_REMOVE_SCRIPT="${LOCAL_BIN_DIR}/rag-remove.sh"
+PACKAGE_REMOVE_SCRIPT="${SCRIPT_DIR}/../bin/rag-remove.sh"
+
+detect_os() {
+    case "$(uname -s)" in
+        Darwin) echo "macos" ;;
+        Linux) echo "linux" ;;
+        *) echo "unknown" ;;
+    esac
+}
+
+find_setup_runner() {
+    local os_name="$1"
+    local local_runner=""
+    local package_runner=""
+
+    case "${os_name}" in
+        macos)
+            local_runner="${LOCAL_SETUP_RUN_MACOS}"
+            package_runner="${PACKAGE_SETUP_RUN_MACOS}"
+            ;;
+        linux)
+            local_runner="${LOCAL_SETUP_RUN_LINUX}"
+            package_runner="${PACKAGE_SETUP_RUN_LINUX}"
+            ;;
+        *)
+            return 1
+            ;;
+    esac
+
+    if [[ -x "${local_runner}" ]]; then
+        echo "${local_runner}"
+        return 0
+    fi
+    if [[ -x "${package_runner}" ]]; then
+        echo "${package_runner}"
+        return 0
+    fi
+    if [[ -f "${package_runner}" ]]; then
+        chmod +x "${package_runner}" || true
+        echo "${package_runner}"
+        return 0
+    fi
+    return 1
+}
+
+find_monitor_runner() {
+    local os_name="$1"
+    if [[ "${os_name}" == "unknown" ]]; then
+        return 1
+    fi
+
+    if [[ -x "${LOCAL_MONITOR_SCRIPT}" ]]; then
+        echo "${LOCAL_MONITOR_SCRIPT}"
+        return 0
+    fi
+    if [[ -x "${PACKAGE_MONITOR_SCRIPT}" ]]; then
+        echo "${PACKAGE_MONITOR_SCRIPT}"
+        return 0
+    fi
+    if [[ -f "${PACKAGE_MONITOR_SCRIPT}" ]]; then
+        chmod +x "${PACKAGE_MONITOR_SCRIPT}" || true
+        echo "${PACKAGE_MONITOR_SCRIPT}"
+        return 0
+    fi
+    return 1
+}
+
+find_remove_runner() {
+    local os_name="$1"
+    if [[ "${os_name}" == "unknown" ]]; then
+        return 1
+    fi
+
+    if [[ -x "${LOCAL_REMOVE_SCRIPT}" ]]; then
+        echo "${LOCAL_REMOVE_SCRIPT}"
+        return 0
+    fi
+    if [[ -x "${PACKAGE_REMOVE_SCRIPT}" ]]; then
+        echo "${PACKAGE_REMOVE_SCRIPT}"
+        return 0
+    fi
+    if [[ -f "${PACKAGE_REMOVE_SCRIPT}" ]]; then
+        chmod +x "${PACKAGE_REMOVE_SCRIPT}" || true
+        echo "${PACKAGE_REMOVE_SCRIPT}"
+        return 0
+    fi
+    return 1
+}
+
+usage() {
+    cat <<'EOF'
+Uso:
+  rag run [path]
+  rag monitor [command]
+  rag remove
+
+Exemplos:
+  rag run .
+  rag run /caminho/do/projeto
+  rag monitor
+  rag monitor full
+  rag remove
+EOF
+}
+
+if [[ $# -lt 1 ]]; then
+    usage
+    exit 1
+fi
+
+command_name="$1"
+shift || true
+os_name="$(detect_os)"
+
+if [[ "${os_name}" == "unknown" ]]; then
+    echo "Erro: sistema operacional não suportado por este instalador." >&2
+    echo "Suportados: Linux e macOS." >&2
+    exit 1
+fi
+
+case "${command_name}" in
+    run)
+        target_path="${1:-}"
+        if [[ -z "${target_path}" ]]; then
+            printf "⚠️ Pasta não informada. Deseja usar a pasta atual: %s? (s/n)\n" "${PWD}"
+            read -r answer
+            answer="$(echo "${answer}" | tr '[:upper:]' '[:lower:]')"
+            if [[ "${answer}" == "s" || "${answer}" == "sim" || "${answer}" == "y" || "${answer}" == "yes" ]]; then
+                target_path="${PWD}"
+            else
+                echo "Dica: use 'rag run /caminho/do/projeto'."
+                exit 1
+            fi
+        fi
+
+        if [[ ! -d "${target_path}" ]]; then
+            echo "Erro: caminho não encontrado ou não é diretório: ${target_path}" >&2
+            exit 1
+        fi
+
+        if ! setup_runner="$(find_setup_runner "${os_name}")"; then
+            if [[ "${os_name}" == "macos" ]]; then
+                echo "Erro: rag-setup-macos.run não encontrado." >&2
+                echo "Esperado em: ${LOCAL_SETUP_RUN_MACOS} ou ${PACKAGE_SETUP_RUN_MACOS}" >&2
+            else
+                echo "Erro: rag-setup.run não encontrado." >&2
+                echo "Esperado em: ${LOCAL_SETUP_RUN_LINUX} ou ${PACKAGE_SETUP_RUN_LINUX}" >&2
+            fi
+            exit 1
+        fi
+
+        exec "${setup_runner}" "${target_path}"
+        ;;
+    monitor)
+        if ! monitor_runner="$(find_monitor_runner "${os_name}")"; then
+            echo "Erro: chroma_monitor.sh não encontrado." >&2
+            echo "Esperado em: ${LOCAL_MONITOR_SCRIPT} ou ${PACKAGE_MONITOR_SCRIPT}" >&2
+            exit 1
+        fi
+        exec "${monitor_runner}" "$@"
+        ;;
+    remove)
+        if ! remove_runner="$(find_remove_runner "${os_name}")"; then
+            echo "Erro: rag-remove.sh não encontrado." >&2
+            echo "Esperado em: ${LOCAL_REMOVE_SCRIPT} ou ${PACKAGE_REMOVE_SCRIPT}" >&2
+            exit 1
+        fi
+        exec "${remove_runner}" "$@"
+        ;;
+    *)
+        usage
+        exit 1
+        ;;
+esac

package/bin/requirements.txt

@@ -0,0 +1,21 @@
+# Cliente HTTP do ChromaDB (conecta ao servidor Docker)
+chromadb>=0.5.0
+
+# Modelo de embeddings local rodando na CPU
+sentence-transformers>=3.0.0
+# Dependência usada por modelos Jina v3 com código remoto
+einops>=0.7.0
+# Evita incompatibilidades conhecidas do Jina v3 com APIs novas do transformers 5.x
+transformers<5
+
+# Splitter de texto para chunking inteligente
+langchain-text-splitters>=0.2.0
+
+# Barra de progresso visual no terminal
+tqdm>=4.66.0
+
+# Monitor de memória em tempo real durante indexação
+psutil>=5.9.0
+
+# SDK oficial do MCP (Model Context Protocol) da Anthropic
+mcp>=1.0.0