docgen-mcp-server 0.1.0__tar.gz

docgen_mcp_server-0.1.0/PKG-INFO

@@ -0,0 +1,154 @@
+Metadata-Version: 2.4
+Name: docgen-mcp-server
+Version: 0.1.0
+Summary: Add your description here
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+Requires-Dist: beautifulsoup4>=4.14.3
+Requires-Dist: mammoth>=1.12.0
+Requires-Dist: markdownify>=1.2.2
+Requires-Dist: mcp>=1.27.2
+Requires-Dist: openpyxl>=3.1.5
+Requires-Dist: playwright>=1.60.0
+Requires-Dist: pypdf>=6.12.2
+Requires-Dist: python-docx>=1.2.0
+Requires-Dist: python-dotenv>=1.2.2
+Requires-Dist: reportlab>=4.5.1
+
+# DocGen MCP Server
+
+## O que é
+
+Servidor [Model Context Protocol (MCP)](https://modelcontextprotocol.io) via **stdio** para leitura e escrita de documentos, planilhas, renderização HTML→PDF (Puppeteer) e utilitários de arquivo. Está publicado no npm como **`docgen-mcp-server`** e requer **Node.js 18+**.
+
+Recomenda-se executar com **`npx -y docgen-mcp-server@latest`** para o `npx` resolver sempre o dist-tag **latest** (evita reutilizar cache de instalação antiga). Para travar em uma versão: `docgen-mcp-server@<versão>` no lugar de `@latest`.
+
+Na primeira execução, **Puppeteer** pode baixar Chromium (tools `render_slide` e `render_page`).
+
+## Ferramentas
+
+| Módulo | Ferramenta | Descrição resumida |
+|--------|------------|-------------------|
+| **read_** | `read_doc` | `.docx`/`.pdf`/`.odt` → Markdown; `previewOnly` / `maxChars` limitam saída. |
+| | `read_sheet` | Planilhas → JSON ou Markdown; `range` tipo `A1:D10`; `previewOnly` / `maxRows`. |
+| | `read_archive` | Lista árvore de entradas em `.zip`. |
+| **write_** | `write_doc` | `.docx`/`.pdf`; **Markdown** (`#`, listas, \`\`\`) ou `contentFormat: plain`; template `{{campo}}`. |
+| | `write_sheet` | `.xlsx` ou `.csv`; `append` em `.xlsx` para logs. |
+| **render_** | `render_slide` | HTML/CSS → PDF ou ZIP de slides (use `.slide` por página). |
+| | `render_page` | HTML/CSS → PDF A4 (índice opcional). |
+| **patch_** | `patch_doc` | PDF: merge, split, watermark; DOCX: `replace_text` em XML. |
+| | `patch_sheet` | Atualiza células em `.xlsx`. |
+| **system_** | `scan_dir` | Busca por regex em diretório ou em arquivos/ZIPs. |
+| | `diff_file` | Diff texto ou dados (planilhas). |
+| | `bundle_zip` | Compacta lista de arquivos em um ZIP. |
+
+### Segurança
+
+- Sem `..` nos caminhos; leitura limitada por tamanho de ficheiro.
+- **Escrita bloqueada** em pastas do sistema (ex.: `Windows`, `Program Files`, `.ssh`, `.aws` no home).
+- Opcional: **`DOCGEN_ALLOWED_ROOTS`** — lista separada por vírgulas de pastas absolutas; só é permitido ler/escrever dentro delas (útil em monorepos/CI).
+
+### Variáveis de ambiente (opcional)
+
+| Variável | Efeito |
+|----------|--------|
+| `DOCGEN_ALLOWED_ROOTS` | Ex.: `C:\repo\my-app,C:\tmp` — restrição de caminhos. |
+| `DOCGEN_SCAN_MAX_MATCHES` | Máximo de correspondências em `scan_dir` (padrão 500). |
+| `DOCGEN_READ_SHEET_MAX_ROWS` | Teto de linhas de dados em `read_sheet` quando não usas `maxRows` (padrão 10000). |
+
+Erros das tools devolvem **`structuredContent`** com `ok: false`, `code`, `tool`, `message` e às vezes `hint`.
+
+## Como usar nos clientes (recomendado: npm publicado)
+
+Em qualquer cliente MCP com transporte **stdio**:
+
+- **Comando**: `npx` (no Windows, se necessário, use o caminho completo de `npx.cmd`).
+- **Argumentos**: `["-y", "docgen-mcp-server@latest"]` (ou versão fixa: `["-y", "docgen-mcp-server@3.0.0"]`).
+- **Variáveis de ambiente**: opcional; veja variáveis do Puppeteer/Chromium se precisar de proxy ou caminho de browser.
+
+Exemplo (Cursor, VS Code com MCP, Claude Desktop, etc.):
+
+```json
+{
+  "mcpServers": {
+    "docgen": {
+      "command": "npx",
+      "args": ["-y", "docgen-mcp-server@latest"],
+      "env": {}
+    }
+  }
+}
+```
+
+No repositório há um exemplo em [`.cursor/mcp.json.example`](.cursor/mcp.json.example).
+
+### Cursor
+
+**Configurações → MCP** (ou JSON de MCP do projeto): use `command`, `args` e `env` como acima.
+
+### Claude Desktop
+
+Mesmo esquema de `command`, `args` e `env`. Detalhes de caminho do arquivo de configuração variam por SO; veja a [documentação da Anthropic sobre MCP](https://docs.anthropic.com/en/docs/agents-and-tools/mcp).
+
+### Problemas comuns no Windows com `npx`
+
+Se aparecer **`EPERM`**, **`TAR_ENTRY_ERROR`** ou erros tipo **`Cannot find package '...\node_modules\yauzl\index.js'`**, o cache do **`npx`** costuma estar **corrompido** (extração interrompida quando o Cursor recarrega o MCP no meio do `npm install`).
+
+1. Desligue ou desative temporariamente o servidor Docgen no MCP.
+2. Apague **`%LocalAppData%\npm-cache\_npx`** (pasta inteira ou só o subdiretório do pacote).
+3. Suba o MCP de novo com `npx -y docgen-mcp-server@latest`.
+
+Alternativa estável: `npm install -g docgen-mcp-server` e no MCP use **`command`: `docgen-mcp-server`** (sem `npx`), ou **`node`** com o caminho absoluto do `cli.js` global.
+
+## Desenvolvimento a partir do clone
+
+```bash
+git clone <repo>
+cd docgen-mcp-server
+npm install
+npm run build
+npm start
+```
+
+Sem build prévio (local):
+
+```bash
+npm install
+npm run dev
+```
+
+| Script | Ação |
+|--------|------|
+| `npm run build` | Compila `src/` → `dist/` (`tsc`) |
+| `npm start` | `node dist/cli.js` |
+| `npm run dev` | `tsx src/cli.ts` |
+
+## Publicação no npm (mantenedores)
+
+O pacote não inclui `node_modules`; o tarball contém só `dist/` + `README.md` + `package.json` (dependências são instaladas pelo cliente ao rodar `npx`).
+
+```bash
+npm whoami
+npm publish --dry-run
+npm publish
+```
+
+Se a conta tiver **2FA “Auth and writes”**, o npm exige OTP:
+
+```bash
+npm publish --otp=123456
+```
+
+Erro **403 Forbidden** com nome livre costuma ser: OTP ausente, `npm login` em outra conta, ou registro apontando para outro servidor (`npm config get registry` deve ser `https://registry.npmjs.org/`).
+
+Versões subsequentes (como no Nautilus):
+
+```bash
+npm run release        # patch
+npm run release:minor
+npm run release:major
+```
+
+## Licença
+
+ISC

docgen_mcp_server-0.1.0/README.md

@@ -0,0 +1,137 @@
+# DocGen MCP Server
+
+## O que é
+
+Servidor [Model Context Protocol (MCP)](https://modelcontextprotocol.io) via **stdio** para leitura e escrita de documentos, planilhas, renderização HTML→PDF (Puppeteer) e utilitários de arquivo. Está publicado no npm como **`docgen-mcp-server`** e requer **Node.js 18+**.
+
+Recomenda-se executar com **`npx -y docgen-mcp-server@latest`** para o `npx` resolver sempre o dist-tag **latest** (evita reutilizar cache de instalação antiga). Para travar em uma versão: `docgen-mcp-server@<versão>` no lugar de `@latest`.
+
+Na primeira execução, **Puppeteer** pode baixar Chromium (tools `render_slide` e `render_page`).
+
+## Ferramentas
+
+| Módulo | Ferramenta | Descrição resumida |
+|--------|------------|-------------------|
+| **read_** | `read_doc` | `.docx`/`.pdf`/`.odt` → Markdown; `previewOnly` / `maxChars` limitam saída. |
+| | `read_sheet` | Planilhas → JSON ou Markdown; `range` tipo `A1:D10`; `previewOnly` / `maxRows`. |
+| | `read_archive` | Lista árvore de entradas em `.zip`. |
+| **write_** | `write_doc` | `.docx`/`.pdf`; **Markdown** (`#`, listas, \`\`\`) ou `contentFormat: plain`; template `{{campo}}`. |
+| | `write_sheet` | `.xlsx` ou `.csv`; `append` em `.xlsx` para logs. |
+| **render_** | `render_slide` | HTML/CSS → PDF ou ZIP de slides (use `.slide` por página). |
+| | `render_page` | HTML/CSS → PDF A4 (índice opcional). |
+| **patch_** | `patch_doc` | PDF: merge, split, watermark; DOCX: `replace_text` em XML. |
+| | `patch_sheet` | Atualiza células em `.xlsx`. |
+| **system_** | `scan_dir` | Busca por regex em diretório ou em arquivos/ZIPs. |
+| | `diff_file` | Diff texto ou dados (planilhas). |
+| | `bundle_zip` | Compacta lista de arquivos em um ZIP. |
+
+### Segurança
+
+- Sem `..` nos caminhos; leitura limitada por tamanho de ficheiro.
+- **Escrita bloqueada** em pastas do sistema (ex.: `Windows`, `Program Files`, `.ssh`, `.aws` no home).
+- Opcional: **`DOCGEN_ALLOWED_ROOTS`** — lista separada por vírgulas de pastas absolutas; só é permitido ler/escrever dentro delas (útil em monorepos/CI).
+
+### Variáveis de ambiente (opcional)
+
+| Variável | Efeito |
+|----------|--------|
+| `DOCGEN_ALLOWED_ROOTS` | Ex.: `C:\repo\my-app,C:\tmp` — restrição de caminhos. |
+| `DOCGEN_SCAN_MAX_MATCHES` | Máximo de correspondências em `scan_dir` (padrão 500). |
+| `DOCGEN_READ_SHEET_MAX_ROWS` | Teto de linhas de dados em `read_sheet` quando não usas `maxRows` (padrão 10000). |
+
+Erros das tools devolvem **`structuredContent`** com `ok: false`, `code`, `tool`, `message` e às vezes `hint`.
+
+## Como usar nos clientes (recomendado: npm publicado)
+
+Em qualquer cliente MCP com transporte **stdio**:
+
+- **Comando**: `npx` (no Windows, se necessário, use o caminho completo de `npx.cmd`).
+- **Argumentos**: `["-y", "docgen-mcp-server@latest"]` (ou versão fixa: `["-y", "docgen-mcp-server@3.0.0"]`).
+- **Variáveis de ambiente**: opcional; veja variáveis do Puppeteer/Chromium se precisar de proxy ou caminho de browser.
+
+Exemplo (Cursor, VS Code com MCP, Claude Desktop, etc.):
+
+```json
+{
+  "mcpServers": {
+    "docgen": {
+      "command": "npx",
+      "args": ["-y", "docgen-mcp-server@latest"],
+      "env": {}
+    }
+  }
+}
+```
+
+No repositório há um exemplo em [`.cursor/mcp.json.example`](.cursor/mcp.json.example).
+
+### Cursor
+
+**Configurações → MCP** (ou JSON de MCP do projeto): use `command`, `args` e `env` como acima.
+
+### Claude Desktop
+
+Mesmo esquema de `command`, `args` e `env`. Detalhes de caminho do arquivo de configuração variam por SO; veja a [documentação da Anthropic sobre MCP](https://docs.anthropic.com/en/docs/agents-and-tools/mcp).
+
+### Problemas comuns no Windows com `npx`
+
+Se aparecer **`EPERM`**, **`TAR_ENTRY_ERROR`** ou erros tipo **`Cannot find package '...\node_modules\yauzl\index.js'`**, o cache do **`npx`** costuma estar **corrompido** (extração interrompida quando o Cursor recarrega o MCP no meio do `npm install`).
+
+1. Desligue ou desative temporariamente o servidor Docgen no MCP.
+2. Apague **`%LocalAppData%\npm-cache\_npx`** (pasta inteira ou só o subdiretório do pacote).
+3. Suba o MCP de novo com `npx -y docgen-mcp-server@latest`.
+
+Alternativa estável: `npm install -g docgen-mcp-server` e no MCP use **`command`: `docgen-mcp-server`** (sem `npx`), ou **`node`** com o caminho absoluto do `cli.js` global.
+
+## Desenvolvimento a partir do clone
+
+```bash
+git clone <repo>
+cd docgen-mcp-server
+npm install
+npm run build
+npm start
+```
+
+Sem build prévio (local):
+
+```bash
+npm install
+npm run dev
+```
+
+| Script | Ação |
+|--------|------|
+| `npm run build` | Compila `src/` → `dist/` (`tsc`) |
+| `npm start` | `node dist/cli.js` |
+| `npm run dev` | `tsx src/cli.ts` |
+
+## Publicação no npm (mantenedores)
+
+O pacote não inclui `node_modules`; o tarball contém só `dist/` + `README.md` + `package.json` (dependências são instaladas pelo cliente ao rodar `npx`).
+
+```bash
+npm whoami
+npm publish --dry-run
+npm publish
+```
+
+Se a conta tiver **2FA “Auth and writes”**, o npm exige OTP:
+
+```bash
+npm publish --otp=123456
+```
+
+Erro **403 Forbidden** com nome livre costuma ser: OTP ausente, `npm login` em outra conta, ou registro apontando para outro servidor (`npm config get registry` deve ser `https://registry.npmjs.org/`).
+
+Versões subsequentes (como no Nautilus):
+
+```bash
+npm run release        # patch
+npm run release:minor
+npm run release:major
+```
+
+## Licença
+
+ISC

docgen_mcp_server-0.1.0/config.py

@@ -0,0 +1,21 @@
+import os
+
+def get_scan_max_matches() -> int:
+    raw = os.environ.get("DOCGEN_SCAN_MAX_MATCHES")
+    if not raw:
+        return 500
+    try:
+        n = int(raw)
+        return min(max(1, n), 50000)
+    except ValueError:
+        return 500
+
+def get_read_sheet_max_rows() -> int:
+    raw = os.environ.get("DOCGEN_READ_SHEET_MAX_ROWS")
+    if not raw:
+        return 10000
+    try:
+        n = int(raw)
+        return min(max(1, n), 100000)
+    except ValueError:
+        return 10000

docgen_mcp_server-0.1.0/docgen_mcp_server.egg-info/PKG-INFO

@@ -0,0 +1,154 @@
+Metadata-Version: 2.4
+Name: docgen-mcp-server
+Version: 0.1.0
+Summary: Add your description here
+Requires-Python: >=3.11
+Description-Content-Type: text/markdown
+Requires-Dist: beautifulsoup4>=4.14.3
+Requires-Dist: mammoth>=1.12.0
+Requires-Dist: markdownify>=1.2.2
+Requires-Dist: mcp>=1.27.2
+Requires-Dist: openpyxl>=3.1.5
+Requires-Dist: playwright>=1.60.0
+Requires-Dist: pypdf>=6.12.2
+Requires-Dist: python-docx>=1.2.0
+Requires-Dist: python-dotenv>=1.2.2
+Requires-Dist: reportlab>=4.5.1
+
+# DocGen MCP Server
+
+## O que é
+
+Servidor [Model Context Protocol (MCP)](https://modelcontextprotocol.io) via **stdio** para leitura e escrita de documentos, planilhas, renderização HTML→PDF (Puppeteer) e utilitários de arquivo. Está publicado no npm como **`docgen-mcp-server`** e requer **Node.js 18+**.
+
+Recomenda-se executar com **`npx -y docgen-mcp-server@latest`** para o `npx` resolver sempre o dist-tag **latest** (evita reutilizar cache de instalação antiga). Para travar em uma versão: `docgen-mcp-server@<versão>` no lugar de `@latest`.
+
+Na primeira execução, **Puppeteer** pode baixar Chromium (tools `render_slide` e `render_page`).
+
+## Ferramentas
+
+| Módulo | Ferramenta | Descrição resumida |
+|--------|------------|-------------------|
+| **read_** | `read_doc` | `.docx`/`.pdf`/`.odt` → Markdown; `previewOnly` / `maxChars` limitam saída. |
+| | `read_sheet` | Planilhas → JSON ou Markdown; `range` tipo `A1:D10`; `previewOnly` / `maxRows`. |
+| | `read_archive` | Lista árvore de entradas em `.zip`. |
+| **write_** | `write_doc` | `.docx`/`.pdf`; **Markdown** (`#`, listas, \`\`\`) ou `contentFormat: plain`; template `{{campo}}`. |
+| | `write_sheet` | `.xlsx` ou `.csv`; `append` em `.xlsx` para logs. |
+| **render_** | `render_slide` | HTML/CSS → PDF ou ZIP de slides (use `.slide` por página). |
+| | `render_page` | HTML/CSS → PDF A4 (índice opcional). |
+| **patch_** | `patch_doc` | PDF: merge, split, watermark; DOCX: `replace_text` em XML. |
+| | `patch_sheet` | Atualiza células em `.xlsx`. |
+| **system_** | `scan_dir` | Busca por regex em diretório ou em arquivos/ZIPs. |
+| | `diff_file` | Diff texto ou dados (planilhas). |
+| | `bundle_zip` | Compacta lista de arquivos em um ZIP. |
+
+### Segurança
+
+- Sem `..` nos caminhos; leitura limitada por tamanho de ficheiro.
+- **Escrita bloqueada** em pastas do sistema (ex.: `Windows`, `Program Files`, `.ssh`, `.aws` no home).
+- Opcional: **`DOCGEN_ALLOWED_ROOTS`** — lista separada por vírgulas de pastas absolutas; só é permitido ler/escrever dentro delas (útil em monorepos/CI).
+
+### Variáveis de ambiente (opcional)
+
+| Variável | Efeito |
+|----------|--------|
+| `DOCGEN_ALLOWED_ROOTS` | Ex.: `C:\repo\my-app,C:\tmp` — restrição de caminhos. |
+| `DOCGEN_SCAN_MAX_MATCHES` | Máximo de correspondências em `scan_dir` (padrão 500). |
+| `DOCGEN_READ_SHEET_MAX_ROWS` | Teto de linhas de dados em `read_sheet` quando não usas `maxRows` (padrão 10000). |
+
+Erros das tools devolvem **`structuredContent`** com `ok: false`, `code`, `tool`, `message` e às vezes `hint`.
+
+## Como usar nos clientes (recomendado: npm publicado)
+
+Em qualquer cliente MCP com transporte **stdio**:
+
+- **Comando**: `npx` (no Windows, se necessário, use o caminho completo de `npx.cmd`).
+- **Argumentos**: `["-y", "docgen-mcp-server@latest"]` (ou versão fixa: `["-y", "docgen-mcp-server@3.0.0"]`).
+- **Variáveis de ambiente**: opcional; veja variáveis do Puppeteer/Chromium se precisar de proxy ou caminho de browser.
+
+Exemplo (Cursor, VS Code com MCP, Claude Desktop, etc.):
+
+```json
+{
+  "mcpServers": {
+    "docgen": {
+      "command": "npx",
+      "args": ["-y", "docgen-mcp-server@latest"],
+      "env": {}
+    }
+  }
+}
+```
+
+No repositório há um exemplo em [`.cursor/mcp.json.example`](.cursor/mcp.json.example).
+
+### Cursor
+
+**Configurações → MCP** (ou JSON de MCP do projeto): use `command`, `args` e `env` como acima.
+
+### Claude Desktop
+
+Mesmo esquema de `command`, `args` e `env`. Detalhes de caminho do arquivo de configuração variam por SO; veja a [documentação da Anthropic sobre MCP](https://docs.anthropic.com/en/docs/agents-and-tools/mcp).
+
+### Problemas comuns no Windows com `npx`
+
+Se aparecer **`EPERM`**, **`TAR_ENTRY_ERROR`** ou erros tipo **`Cannot find package '...\node_modules\yauzl\index.js'`**, o cache do **`npx`** costuma estar **corrompido** (extração interrompida quando o Cursor recarrega o MCP no meio do `npm install`).
+
+1. Desligue ou desative temporariamente o servidor Docgen no MCP.
+2. Apague **`%LocalAppData%\npm-cache\_npx`** (pasta inteira ou só o subdiretório do pacote).
+3. Suba o MCP de novo com `npx -y docgen-mcp-server@latest`.
+
+Alternativa estável: `npm install -g docgen-mcp-server` e no MCP use **`command`: `docgen-mcp-server`** (sem `npx`), ou **`node`** com o caminho absoluto do `cli.js` global.
+
+## Desenvolvimento a partir do clone
+
+```bash
+git clone <repo>
+cd docgen-mcp-server
+npm install
+npm run build
+npm start
+```
+
+Sem build prévio (local):
+
+```bash
+npm install
+npm run dev
+```
+
+| Script | Ação |
+|--------|------|
+| `npm run build` | Compila `src/` → `dist/` (`tsc`) |
+| `npm start` | `node dist/cli.js` |
+| `npm run dev` | `tsx src/cli.ts` |
+
+## Publicação no npm (mantenedores)
+
+O pacote não inclui `node_modules`; o tarball contém só `dist/` + `README.md` + `package.json` (dependências são instaladas pelo cliente ao rodar `npx`).
+
+```bash
+npm whoami
+npm publish --dry-run
+npm publish
+```
+
+Se a conta tiver **2FA “Auth and writes”**, o npm exige OTP:
+
+```bash
+npm publish --otp=123456
+```
+
+Erro **403 Forbidden** com nome livre costuma ser: OTP ausente, `npm login` em outra conta, ou registro apontando para outro servidor (`npm config get registry` deve ser `https://registry.npmjs.org/`).
+
+Versões subsequentes (como no Nautilus):
+
+```bash
+npm run release        # patch
+npm run release:minor
+npm run release:major
+```
+
+## Licença
+
+ISC

docgen_mcp_server-0.1.0/docgen_mcp_server.egg-info/SOURCES.txt

@@ -0,0 +1,12 @@
+README.md
+config.py
+pyproject.toml
+security.py
+server.py
+services.py
+docgen_mcp_server.egg-info/PKG-INFO
+docgen_mcp_server.egg-info/SOURCES.txt
+docgen_mcp_server.egg-info/dependency_links.txt
+docgen_mcp_server.egg-info/entry_points.txt
+docgen_mcp_server.egg-info/requires.txt
+docgen_mcp_server.egg-info/top_level.txt

docgen_mcp_server-0.1.0/docgen_mcp_server.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

docgen_mcp_server-0.1.0/docgen_mcp_server.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+docgen-mcp-server = server:main

docgen_mcp_server-0.1.0/docgen_mcp_server.egg-info/requires.txt

@@ -0,0 +1,10 @@
+beautifulsoup4>=4.14.3
+mammoth>=1.12.0
+markdownify>=1.2.2
+mcp>=1.27.2
+openpyxl>=3.1.5
+playwright>=1.60.0
+pypdf>=6.12.2
+python-docx>=1.2.0
+python-dotenv>=1.2.2
+reportlab>=4.5.1

docgen_mcp_server-0.1.0/docgen_mcp_server.egg-info/top_level.txt

@@ -0,0 +1,4 @@
+config
+security
+server
+services

docgen_mcp_server-0.1.0/pyproject.toml

@@ -0,0 +1,28 @@
+[project]
+name = "docgen-mcp-server"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.11"
+dependencies = [
+    "beautifulsoup4>=4.14.3",
+    "mammoth>=1.12.0",
+    "markdownify>=1.2.2",
+    "mcp>=1.27.2",
+    "openpyxl>=3.1.5",
+    "playwright>=1.60.0",
+    "pypdf>=6.12.2",
+    "python-docx>=1.2.0",
+    "python-dotenv>=1.2.2",
+    "reportlab>=4.5.1",
+]
+
+[project.scripts]
+docgen-mcp-server = "server:main"
+
+[tool.setuptools]
+py-modules = ["server", "config", "security", "services"]
+
+
+
+

docgen_mcp_server-0.1.0/security.py

@@ -0,0 +1,69 @@
+import os
+import sys
+
+MAX_FILE_SIZE_BYTES = 50 * 1024 * 1024
+
+BLOCKED_DIR_NAMES = {
+    ".ssh", ".aws", ".gnupg", ".credentials", ".kube", ".docker", ".azure"
+}
+
+def get_blocked_roots() -> list:
+    blocked = []
+    is_win = sys.platform == "win32"
+    if is_win:
+        sys_root = os.environ.get("SystemRoot", "C:\\Windows")
+        blocked.append(os.path.abspath(sys_root))
+        blocked.append(os.path.abspath("C:/Program Files"))
+        blocked.append(os.path.abspath("C:/Program Files (x86)"))
+    else:
+        for p in ["/bin", "/sbin", "/usr", "/etc", "/boot", "/sys", "/proc"]:
+            try:
+                blocked.append(os.path.abspath(p))
+            except Exception:
+                pass
+    home = os.path.expanduser("~")
+    for name in BLOCKED_DIR_NAMES:
+        blocked.append(os.path.abspath(os.path.join(home, name)))
+    return blocked
+
+def get_allowed_roots() -> list:
+    raw = os.environ.get("DOCGEN_ALLOWED_ROOTS", "").strip()
+    if not raw:
+        return []
+    return [os.path.abspath(s.strip()) for s in raw.split(",") if s.strip()]
+
+def enforce_allowed_roots(resolved: str) -> None:
+    roots = get_allowed_roots()
+    if not roots:
+        return
+    norm = os.path.normpath(resolved)
+    ok = False
+    for r in roots:
+        if norm == r or norm.startswith(r + os.sep):
+            ok = True
+            break
+    if not ok:
+        raise ValueError(f"Caminho fora de DOCGEN_ALLOWED_ROOTS. Recebido: {resolved}")
+
+def validate_path(file_path: str, must_exist: bool) -> str:
+    if ".." in file_path:
+        raise ValueError(f"Path traversal detectado: {file_path}")
+    resolved = os.path.abspath(file_path)
+    enforce_allowed_roots(resolved)
+    if must_exist and not os.path.exists(resolved):
+        raise ValueError(f"Arquivo nao encontrado: {file_path}")
+    if must_exist:
+        if os.path.isfile(resolved):
+            sz = os.path.getsize(resolved)
+            if sz > MAX_FILE_SIZE_BYTES:
+                raise ValueError(f"Arquivo excede limite de {MAX_FILE_SIZE_BYTES / (1024*1024)}MB: {sz / (1024*1024):.1f}MB")
+    return resolved
+
+def validate_write_path(file_path: str) -> str:
+    resolved = validate_path(file_path, False)
+    norm = os.path.normpath(resolved)
+    for b in get_blocked_roots():
+        nb = os.path.normpath(b)
+        if norm == nb or norm.startswith(nb + os.sep):
+            raise PermissionError(f"Escrita bloqueada em diretorio protegido: {b}")
+    return resolved