utils

Documentation

Overview

* ChatCLI - Command Line Interface for LLM interaction * Copyright (c) 2024 Edilson Freitas * License: Apache-2.0

* ChatCLI - Console bootstrap (non-Windows no-op). * Copyright (c) 2024 Edilson Freitas * License: Apache-2.0

* ChatCLI - Command Line Interface for LLM interaction * Copyright (c) 2024 Edilson Freitas * License: Apache-2.0

* ChatCLI - Command Line Interface for LLM interaction * Copyright (c) 2024 Edilson Freitas * License: Apache-2.0

* ChatCLI - Command Line Interface for LLM interaction * Copyright (c) 2024 Edilson Freitas * License: Apache-2.0

* ChatCLI - Command Line Interface for LLM interaction * Copyright (c) 2024 Edilson Freitas * License: Apache-2.0

* ChatCLI - Command Line Interface for LLM interaction * Copyright (c) 2024 Edilson Freitas * License: Apache-2.0

* ChatCLI - Command Line Interface for LLM interaction * Copyright (c) 2024 Edilson Freitas * License: Apache-2.0

* ChatCLI - Command Line Interface for LLM interaction * Copyright (c) 2024 Edilson Freitas * License: Apache-2.0

* ChatCLI - Command Line Interface for LLM interaction * Copyright (c) 2024 Edilson Freitas * License: Apache-2.0

* ChatCLI - Command Line Interface for LLM interaction * Copyright (c) 2024 Edilson Freitas * License: Apache-2.0

* ChatCLI - Command Line Interface for LLM interaction * Copyright (c) 2024 Edilson Freitas * License: Apache-2.0

Index

• func ApplyGlobalTLSTrust(logger *zap.Logger)
• func CountMatchingFiles(dirPath string, options DirectoryScanOptions) (int, error)
• func DetectFileType(filePath string) string
• func EnableVirtualTerminal()
• func ExpandPath(path string) (string, error)
• func FormatDirectoryContent(files []FileInfo, maxTotalSize int64) string
• func GetEnvOrDefault(key, defaultValue string) string
• func GetEnvVariablesSanitized() string
• func GetFileBlame(filepath string) (string, error)
• func GetFileDiff(filepath string) (string, error)
• func GetFileHistory(filepath string) (string, error)
• func GetGitInfo(executor CommandExecutor) (string, error)
• func GetHomeDir() (string, error)
• func GetShellConfigFilePath(shellName string) string
• func GetShellHistory() (string, error)
• func GetShellHistoryFile() (string, error)
• func GetUserShell() string
• func GlobalTLSConfig() *tls.Config
• func InitializeLogger() (*zap.Logger, error)
• func IsSensitivePath(absPath string) bool
• func IsTemporaryError(err error) bool
• func LogStartupInfo(logger *zap.Logger)
• func MaybeUnescapeJSONishArgs(input string) (string, bool)
• func NewChromeTLSTransport() http.RoundTripper
• func NewHTTPClient(logger *zap.Logger, timeout time.Duration) *http.Client
• func NewHTTPClientH1(logger *zap.Logger, timeout time.Duration) *http.Client
• func NewHTTPClientWithTransport(logger *zap.Logger, timeout time.Duration, inner http.RoundTripper) *http.Client
• func NewJSONReader(data []byte) io.Reader
• func ParseSize(sizeStr string) (int64, error)
• func ReadFileContent(filePath string, maxSize int64) (string, error)
• func RepairJSONInvalidEscapes(s string) string
• func Retry[T any](ctx context.Context, logger *zap.Logger, maxAttempts int, ...) (T, error)
• func SanitizeSensitiveText(s string) string
• func ShellQuote(s string) string
• func ShouldSkipDir(dirName string) bool
• type APIClient
  • func NewAPIClient(logger *zap.Logger, baseURL string, headers map[string]string) *APIClient
  • func (c *APIClient) Get(ctx context.Context, path string) ([]byte, error)
  • func (c *APIClient) Post(ctx context.Context, path string, payload interface{}) ([]byte, error)
  • func (c *APIClient) Request(ctx context.Context, method, path string, body []byte) ([]byte, error)
• type APIError
  • func (e *APIError) Error() string
• type CommandExecutor
  • func NewOSCommandExecutor() CommandExecutor
• type DirectoryScanOptions
  • func DefaultDirectoryScanOptions(logger *zap.Logger) DirectoryScanOptions
• type FileInfo
  • func ProcessDirectory(ctx context.Context, dirPath string, options DirectoryScanOptions) ([]FileInfo, error)
• type LoggingTransport
  • func (t *LoggingTransport) RoundTrip(req *http.Request) (*http.Response, error)
• type OSCommandExecutor
  • func (e *OSCommandExecutor) Output(name string, arg ...string) ([]byte, error)
  • func (e *OSCommandExecutor) Run(name string, arg ...string) error

Functions

func ApplyGlobalTLSTrust

func ApplyGlobalTLSTrust(logger *zap.Logger)

ApplyGlobalTLSTrust resolves the CHATCLI_CA_BUNDLE / CHATCLI_TLS_INSECURE_SKIP_VERIFY overrides and wires them into both the transports built by NewHTTPClient and http.DefaultTransport — so bare &http.Client{} instances (gateway channels, MCP transports, registries, version check) inherit the corporate trust as well.

Must run after the dotenv load so overrides set only in the .env file are seen. Idempotent; calling it with no overrides set is a no-op that leaves http.DefaultTransport untouched.

func CountMatchingFiles

func CountMatchingFiles(dirPath string, options DirectoryScanOptions) (int, error)

CountMatchingFiles faz uma varredura rápida para contar quantos arquivos em um diretório correspondem aos critérios das opções, sem ler seu conteúdo.

func DetectFileType

func DetectFileType(filePath string) string

DetectFileType detecta o tipo de arquivo com base na extensão

func EnableVirtualTerminal

func EnableVirtualTerminal()

EnableVirtualTerminal is a no-op outside Windows: Unix terminals process ANSI escapes natively. See console_windows.go for the Windows opt-in.

func ExpandPath

func ExpandPath(path string) (string, error)

ExpandPath expande o caractere ~ no início de um caminho para o diretório home do usuário. Se o caminho não começar com ~, ele é retornado sem modificações. A função não suporta a expansão de ~username, retornando um erro nesse caso.

func FormatDirectoryContent

func FormatDirectoryContent(files []FileInfo, maxTotalSize int64) string

FormatDirectoryContent formata o conteúdo de vários arquivos em uma única string organizada com separadores claros e formatação adequada para cada tipo de arquivo.

func GetEnvOrDefault

func GetEnvOrDefault(key, defaultValue string) string

GetEnvOrDefault retorna o valor da variável de ambiente ou um valor padrão se não estiver definida

func GetEnvVariablesSanitized

func GetEnvVariablesSanitized() string

GetEnvVariablesSanitized retorna variáveis de ambiente com valores sensíveis redigidos.

func GetFileBlame

func GetFileBlame(filepath string) (string, error)

Função auxiliar para obter blame de um arquivo

func GetFileDiff

func GetFileDiff(filepath string) (string, error)

Função auxiliar para obter diferenças específicas de um arquivo

func GetFileHistory

func GetFileHistory(filepath string) (string, error)

Função auxiliar para obter o histórico de um arquivo específico

func GetGitInfo

func GetGitInfo(executor CommandExecutor) (string, error)

GetGitInfo agora aceita um CommandExecutor.

func GetHomeDir

func GetHomeDir() (string, error)

GetHomeDir retorna o diretório home do usuário atual.

func GetShellConfigFilePath

func GetShellConfigFilePath(shellName string) string

GetShellConfigFilePath retorna o caminho do arquivo de configuração do shell com base no nome do shell.

func GetShellHistory

func GetShellHistory() (string, error)

GetShellHistory lê o arquivo de histórico do shell e retorna seu conteúdo.

func GetShellHistoryFile

func GetShellHistoryFile() (string, error)

GetShellHistoryFile retorna o caminho do arquivo de histórico do shell.

func GetUserShell

func GetUserShell() string

GetUserShell retorna o shell do usuário atual.

func GlobalTLSConfig

func GlobalTLSConfig() *tls.Config

GlobalTLSConfig returns the process-wide TLS overrides resolved by ApplyGlobalTLSTrust, or nil when none are configured. Callers that build their own transports should Clone() the result before mutating it.

func InitializeLogger

func InitializeLogger() (*zap.Logger, error)

InitializeLogger configura e inicializa um logger robusto com base nas variáveis de ambiente. Em ambiente de desenvolvimento ('dev'), os logs são enviados para o console de forma colorida e legível, e também para um arquivo em formato JSON. Em ambiente de produção ('prod'), os logs são enviados apenas para o arquivo em formato JSON. Suporta rotação de logs com lumberjack.

func IsSensitivePath

func IsSensitivePath(absPath string) bool

IsSensitivePath checks whether a file path points to a known sensitive location. The comparison is case-insensitive on every OS: Windows filesystems are case-insensitive (".SSH" is the same directory as ".ssh"), and on Unix over-matching an odd-cased lookalike only errs on the safe side.

func IsTemporaryError

func IsTemporaryError(err error) bool

IsTemporaryError verifica se o erro é temporário e pode ser retryado. Agora usa errors.Unwrap para desembrulhar e As para checar APIError com status 429 ou 5xx, além de timeouts.

func LogStartupInfo

func LogStartupInfo(logger *zap.Logger)

func MaybeUnescapeJSONishArgs

func MaybeUnescapeJSONishArgs(input string) (string, bool)

MaybeUnescapeJSONishArgs tenta desfazer escape de JSON embutido em uma string quando o conteúdo parece ser um objeto/array JSON com aspas escapadas. Retorna (valor, true) se conseguiu unescape e o resultado parece JSON.

func NewChromeTLSTransport

func NewChromeTLSTransport() http.RoundTripper

NewChromeTLSTransport creates an http.RoundTripper that uses a Chrome-like TLS fingerprint via uTLS with automatic HTTP/1.1 / HTTP/2 support based on ALPN negotiation.

func NewHTTPClient

func NewHTTPClient(logger *zap.Logger, timeout time.Duration) *http.Client

NewHTTPClient cria um cliente HTTP com LoggingTransport e timeout configurado. Cada chamada cria um http.Transport dedicado para evitar compartilhamento de pool de conexões e cache TLS entre providers diferentes.

func NewHTTPClientH1

func NewHTTPClientH1(logger *zap.Logger, timeout time.Duration) *http.Client

NewHTTPClientH1 is like NewHTTPClient but pins the connection to HTTP/1.1.

Go's bundled HTTP/2 client intermittently fails a POST-with-body with "unexpected EOF" against some Cloudflare-fronted hosts (observed on api.openai.com and api.x.ai image endpoints): the edge closes an idle h2 connection and the next request races the GOAWAY, and net/http does NOT retry requests that carry a body. HTTP/1.1 sidesteps the whole race and is rock-solid for request/response calls. Use it for non-streaming endpoints that send/receive large payloads (image generation), where h2 multiplexing buys nothing. Corporate TLS trust (CHATCLI_CA_BUNDLE / CHATCLI_TLS_INSECURE_SKIP_VERIFY) is preserved; only ALPN is constrained.

func NewHTTPClientWithTransport

func NewHTTPClientWithTransport(logger *zap.Logger, timeout time.Duration, inner http.RoundTripper) *http.Client

NewHTTPClientWithTransport cria um cliente HTTP com um transport customizado envolvido pelo LoggingTransport. Use para injetar transports com TLS customizado.

func NewJSONReader

func NewJSONReader(data []byte) io.Reader

NewJSONReader cria um io.Reader a partir de um []byte para enviar em requisições HTTP

func ParseSize

func ParseSize(sizeStr string) (int64, error)

ParseSize converte uma string de tamanho legível (como "50MB") para bytes.

func ReadFileContent

func ReadFileContent(filePath string, maxSize int64) (string, error)

ReadFileContent lê o conteúdo de um arquivo, expandindo ~ para o diretório home, mostrando um indicador de progresso para arquivos grandes.

func RepairJSONInvalidEscapes

func RepairJSONInvalidEscapes(s string) string

RepairJSONInvalidEscapes conserta texto JSON-ish em que barras invertidas não foram escapadas — o caso clássico é o LLM embutir um path Windows ("C:\Users\x") numa string JSON, produzindo escapes inválidos como \U.

A decisão é POR LITERAL DE STRING, tudo-ou-nada: uma string que contém qualquer escape inválido é "suja" — o emissor claramente não estava escapando — e então TODAS as suas barras são tratadas como literais de path e duplicadas, inclusive as que por coincidência formariam escapes válidos (\b em "\builder", \t em "\temp\Users"). Preservam-se apenas \" (aspas são ilegais em nomes de arquivo Windows, logo é um escape real) e \\ (já escapada). Strings totalmente válidas ficam intactas, então um "\n" legítimo numa chave irmã sobrevive ao reparo da chave suja.

Retorna o input inalterado quando não há nada a reparar. Limitação inerente: uma string SÓ com escapes coincidentemente válidos (ex.: "C:\temp") é JSON válido e nunca chega a esta função — o erro segue para o chamador, que devolve o problema ao modelo.

func Retry

func Retry[T any](ctx context.Context, logger *zap.Logger, maxAttempts int, initialBackoff time.Duration, fn func(context.Context) (T, error)) (T, error)

Retry executa uma função com retry exponencial para erros temporários. - maxAttempts: Número máximo de tentativas (lido de ENV ou default). - initialBackoff: Tempo inicial de espera entre tentativas (lido de ENV ou default). - fn: Função a executar, que recebe ctx e retorna um resultado genérico T e erro.

func SanitizeSensitiveText

func SanitizeSensitiveText(s string) string

SanitizeSensitiveText remove/mascara tokens em qualquer texto antes de ir para histórico/LLM

func ShellQuote

func ShellQuote(s string) string

ShellQuote wraps a string in POSIX-compliant single quotes, escaping any embedded single quotes. The result is safe to embed in a shell command without risk of injection.

func ShouldSkipDir

func ShouldSkipDir(dirName string) bool

ShouldSkipDir verifica se um diretório deve ser ignorado

Types

type APIClient

type APIClient struct {
	// contains filtered or unexported fields
}

APIClient é um cliente HTTP especializado para APIs REST

func NewAPIClient

func NewAPIClient(logger *zap.Logger, baseURL string, headers map[string]string) *APIClient

NewAPIClient cria um novo cliente API com cabeçalhos padrão

func (*APIClient) Get

func (c *APIClient) Get(ctx context.Context, path string) ([]byte, error)

Get faz uma requisição GET

func (*APIClient) Post

func (c *APIClient) Post(ctx context.Context, path string, payload interface{}) ([]byte, error)

Post faz uma requisição POST

func (*APIClient) Request

func (c *APIClient) Request(ctx context.Context, method, path string, body []byte) ([]byte, error)

Request faz uma requisição HTTP com retry exponencial

type APIError

type APIError struct {
	StatusCode int
	Message    string
}

APIError é um erro estruturado para respostas HTTP com status code.

func (*APIError) Error

func (e *APIError) Error() string

type CommandExecutor

type CommandExecutor interface {
	Run(name string, arg ...string) error
	Output(name string, arg ...string) ([]byte, error)
}

CommandExecutor define uma interface para executar comandos. Isso nos permite mockar a execução de comandos nos testes.

func NewOSCommandExecutor

func NewOSCommandExecutor() CommandExecutor

NewOSCommandExecutor cria um novo executor de comandos do sistema operacional.

type DirectoryScanOptions

type DirectoryScanOptions struct {
	MaxTotalSize      int64 // Tamanho máximo do conteúdo total (em bytes)
	MaxFilesToProcess int   // Número máximo de arquivos a processar
	Logger            *zap.Logger
	Extensions        []string            // Extensões de arquivo a incluir (vazio = todas)
	ExcludeDirs       []string            // Diretórios a excluir
	ExcludePatterns   []string            // Padrões de nome de arquivo a excluir (ex: "*.tmp")
	IncludeHidden     bool                // Incluir arquivos/diretórios ocultos?
	OnFileProcessed   func(info FileInfo) // Callback para cada arquivo processado
}

DirectoryScanOptions configura opções para escaneamento de diretórios

func DefaultDirectoryScanOptions

func DefaultDirectoryScanOptions(logger *zap.Logger) DirectoryScanOptions

DefaultDirectoryScanOptions retorna opções padrão para escaneamento

type FileInfo

type FileInfo struct {
	Path    string
	Content string
	Size    int64
	Type    string
}

FileInfo armazena informações e conteúdo de um arquivo

func ProcessDirectory

func ProcessDirectory(ctx context.Context, dirPath string, options DirectoryScanOptions) ([]FileInfo, error)

ProcessDirectory processa um diretório recursivamente de forma concorrente e segura.

type LoggingTransport

type LoggingTransport struct {
	Logger      *zap.Logger
	Transport   http.RoundTripper
	MaxBodySize int
}

LoggingTransport é um http.RoundTripper que adiciona logs às requisições e respostas

func (*LoggingTransport) RoundTrip

func (t *LoggingTransport) RoundTrip(req *http.Request) (*http.Response, error)

RoundTrip implementa a interface http.RoundTripper

type OSCommandExecutor

type OSCommandExecutor struct{}

OSCommandExecutor é a implementação real que usa os/exec.

func (*OSCommandExecutor) Output

func (e *OSCommandExecutor) Output(name string, arg ...string) ([]byte, error)

Output executa um comando e retorna sua saída padrão.

func (*OSCommandExecutor) Run

func (e *OSCommandExecutor) Run(name string, arg ...string) error

Run executa um comando e espera ele terminar.