Add Solidity language support (#563)

Author: buger

Cargo.toml

@@ -44,6 +44,7 @@ tree-sitter-c-sharp = { version = "0.23.1" }
 tree-sitter-html = "0.23.2"
 tree-sitter-md = "0.3.2"
 tree-sitter-yaml = "0.6.1"
+tree-sitter-solidity = "=1.2.10"
 ast-grep-core = "0.36.1"
 ast-grep-language = "0.36.1"
 clap = { version = "4.3", features = ["derive"] }

README.md

@@ -143,7 +143,7 @@ npx -y @probelabs/probe@latest agent "Refactor the login function" --allow-edit
 - **Smart Ranking**: BM25, TF-IDF, and hybrid algorithms with optional BERT reranking
 - **Token-Aware**: `--max-tokens` budget, session-based dedup to avoid repeating context
 - **Built-in Agent**: Multi-provider (Anthropic, OpenAI, Google, Bedrock) with retry, fallback, and context compaction
-- **Multi-Language**: Rust, Python, JavaScript, TypeScript, Go, C/C++, Java, Ruby, PHP, Swift, C#, and more
+- **Multi-Language**: Rust, Python, JavaScript, TypeScript, Go, C/C++, Java, Ruby, PHP, Swift, Solidity, C#, and more
 
 ---
 
@@ -468,6 +468,7 @@ cargo install --path .
 | Ruby | `.rb` |
 | PHP | `.php` |
 | Swift | `.swift` |
+| Solidity | `.sol` |
 | C# | `.cs` |
 | Markdown | `.md` |
 

docs/probe-cli/extraction-reference.md

@@ -177,6 +177,7 @@ Probe supports extraction for many other languages including:
 - **Ruby**: Methods, classes, modules, blocks
 - **PHP**: Functions, classes, namespaces, attributes
 - **Swift**: Functions, classes, structs, protocols, extensions
+- **Solidity**: Contracts, interfaces, libraries, functions, modifiers, events, errors
 - **C#**: Methods, classes, interfaces, namespaces, attributes
 - **Markdown**: Sections, code blocks, lists, tables, frontmatter
 
@@ -433,4 +434,4 @@ probe extract src/api.js:27 --context 10
 - **INTELLIGENT FALLBACKS**: Gracefully handles cases where AST parsing isn't possible
 
 For more information on how Probe works internally, see [How Probe Works](how-it-works.md).
-For details on search capabilities, see [Search Functionality](search-functionality.md).
+For details on search capabilities, see [Search Functionality](search-functionality.md).

docs/probe-cli/query.md

@@ -138,6 +138,7 @@ Required for accurate parsing:
 | Ruby | `ruby`, `rb` | .rb |
 | PHP | `php` | .php |
 | Swift | `swift` | .swift |
+| Solidity | `solidity`, `sol` | .sol |
 | C# | `csharp`, `cs` | .cs |
 
 ```bash

docs/probe-cli/search.md

@@ -208,7 +208,7 @@ probe search "api" ./ --reranker ms-marco-tinybert --question "How is the REST A
 | `-l`, `--language` | String | auto | Limit to programming language |
 
 **Supported Languages:**
-`rust`, `javascript`, `typescript`, `python`, `go`, `c`, `cpp`, `java`, `ruby`, `php`, `swift`, `csharp`, `yaml`, `html`, `markdown`
+`rust`, `javascript`, `typescript`, `python`, `go`, `c`, `cpp`, `java`, `ruby`, `php`, `swift`, `solidity`, `csharp`, `yaml`, `html`, `markdown`
 
 ```bash
 # Search only Python files

docs/reference/adding-languages.md

@@ -95,7 +95,7 @@ Add to language completion lists (2 locations):
 ### File 10: `npm/src/mcp/index.ts`
 Add to supported languages description:
 ```rust
-'Supported languages: rust, javascript, typescript, python, go, c, cpp, java, ruby, php, swift, csharp, yourlang.',
+'Supported languages: rust, javascript, typescript, python, go, c, cpp, java, ruby, php, swift, solidity, csharp, yourlang.',
 ```
 
 ## Step 4: Documentation

docs/reference/architecture.md

@@ -104,7 +104,7 @@ pub trait LanguageImpl {
 }
 ```
 
-**Supported:** Rust, JavaScript, TypeScript, Python, Go, C, C++, Java, Ruby, PHP, Swift, C#, HTML, Markdown, YAML
+**Supported:** Rust, JavaScript, TypeScript, Python, Go, C, C++, Java, Ruby, PHP, Swift, Solidity, C#, HTML, Markdown, YAML
 
 ### Performance Optimizations
 
@@ -418,4 +418,3 @@ agent.events.on('toolCall', (event) => {
 - [API Reference](../probe-agent/sdk/api-reference.md) - SDK API
 - [MCP Protocol](../probe-agent/protocols/mcp.md) - MCP integration
 - [Performance](../probe-cli/performance.md) - Optimization guide
-

docs/reference/faq.md

@@ -62,7 +62,7 @@ Yes, Probe is open source under the Apache 2.0 license.
 
 ### What languages does Probe support?
 
-Probe supports 15+ languages including Rust, JavaScript, TypeScript, Python, Go, C, C++, Java, Ruby, PHP, Swift, C#, HTML, Markdown, and YAML. See [Supported Languages](../supported-languages.md).
+Probe supports 15+ languages including Rust, JavaScript, TypeScript, Python, Go, C, C++, Java, Ruby, PHP, Swift, Solidity, C#, HTML, Markdown, and YAML. See [Supported Languages](../supported-languages.md).
 
 ---
 

docs/reference/language-support.md

@@ -107,6 +107,7 @@ Probe currently supports a wide range of programming languages, including:
 - Ruby
 - PHP
 - Swift
+- Solidity
 - C#
 - Markdown
 
@@ -165,4 +166,4 @@ Tree-sitter provides several advantages for Probe:
 - [Supported Languages](/supported-languages): Detailed information about each supported language
 - [Adding New Languages](/adding-languages): Guide for adding support for new languages
 - [Search Functionality](/search-functionality): How to use Probe's search features
-- [Code Extraction](/code-extraction): Details on how Probe extracts code blocks
+- [Code Extraction](/code-extraction): Details on how Probe extracts code blocks

docs/reference/supported-languages.md

@@ -18,6 +18,7 @@ Probe provides language-aware code search and extraction for a wide range of pro
 | Ruby | `.rb` | ✅ | ✅ |
 | PHP | `.php` | ✅ | ✅ |
 | Swift | `.swift` | ✅ | ✅ |
+| Solidity | `.sol` | ✅ | ✅ |
 | C# | `.cs` | ✅ | ✅ |
 | Markdown | `.md`, `.markdown` | ✅ | ✅ |
 | YAML | `.yaml`, `.yml` | ✅ | ✅ |
@@ -128,6 +129,14 @@ Go also implements special handling for nested struct types.
 - **Attribute Handling**: Properly handles C# attributes
 - **Test Detection**: Identifies test classes and methods using NUnit, MSTest, or xUnit conventions
 
+### Solidity
+
+- **Contract Extraction**: Extracts contracts, interfaces, and libraries
+- **Function Extraction**: Extracts functions, constructors, fallback/receive functions, and modifiers
+- **Type Extraction**: Extracts structs, enums, user-defined value types, events, and custom errors
+- **Comment Handling**: Associates NatSpec and regular comments with Solidity code blocks
+- **Test Detection**: Identifies Foundry-style `.t.sol` files, `*Test.sol` contracts, `setUp`, `test*`, and `invariant_*` functions
+
 ### Markdown
 
 - **Section Extraction**: Extracts complete sections based on headings
@@ -212,4 +221,4 @@ func ($RECEIVER $TYPE) $NAME($$$PARAMS) $$$BODY
 
 ## Adding Support for New Languages
 
-If you're interested in adding support for a language that's not currently supported, see the [Adding New Languages](/adding-languages) page for a detailed guide.
+If you're interested in adding support for a language that's not currently supported, see the [Adding New Languages](/adding-languages) page for a detailed guide.