snail-lang 0.3.9__tar.gz → 0.4.1__tar.gz

{snail_lang-0.3.9 → snail_lang-0.4.1}/Cargo.lock

@@ -485,11 +485,11 @@ checksum = "67b1b7a3b5fe4f1376887184045fcf45c69e92af734b7aaddc05fb777b6fbd03"
 
 [[package]]
 name = "snail-ast"
-version = "0.3.9"
+version = "0.4.1"
 
 [[package]]
 name = "snail-core"
-version = "0.3.9"
+version = "0.4.1"
 dependencies = [
  "pyo3",
  "snail-ast",
@@ -500,14 +500,14 @@ dependencies = [
 
 [[package]]
 name = "snail-error"
-version = "0.3.9"
+version = "0.4.1"
 dependencies = [
  "snail-ast",
 ]
 
 [[package]]
 name = "snail-lower"
-version = "0.3.9"
+version = "0.4.1"
 dependencies = [
  "pyo3",
  "snail-ast",
@@ -516,7 +516,7 @@ dependencies = [
 
 [[package]]
 name = "snail-parser"
-version = "0.3.9"
+version = "0.4.1"
 dependencies = [
  "pest",
  "pest_derive",
@@ -526,7 +526,7 @@ dependencies = [
 
 [[package]]
 name = "snail-proptest"
-version = "0.3.9"
+version = "0.4.1"
 dependencies = [
  "proptest",
  "pyo3",
@@ -540,7 +540,7 @@ dependencies = [
 
 [[package]]
 name = "snail-python"
-version = "0.3.9"
+version = "0.4.1"
 dependencies = [
  "pyo3",
  "snail-core",

{snail_lang-0.3.9 → snail_lang-0.4.1}/PKG-INFO

@@ -1,6 +1,7 @@
 Metadata-Version: 2.4
 Name: snail-lang
-Version: 0.3.9
+Version: 0.4.1
+Requires-Dist: jmespath>=1.0.1
 Requires-Dist: maturin>=1.5 ; extra == 'dev'
 Requires-Dist: pytest ; extra == 'dev'
 Provides-Extra: dev
@@ -12,17 +13,11 @@ Description-Content-Type: text/markdown; charset=UTF-8; variant=GFM
 <p align="center">
   <img src="logo.png" alt="Snail logo" width="200">
 </p>
-
-<h1 align="center">Snail</h1>
 <p align="center"><em>What do you get when you shove a snake in a shell?</em></p>
 
-<h1>Snail, while I hope it is useful to myself and others, is my attempt at
-improving my knowledge of AI code developement. Things are probably broken
-in interesting and horrible ways.</h1>
-
----
+<h1 align="center">Snail</h1>
 
-**Snail** is a programming language that compiles to Python, combining Python's power with Perl/awk-inspired syntax for quick scripts and one-liners. No more whitespace sensitivity—just curly braces and concise expressions.
+**Snail** is a programming language that compiles to Python, combining Python's familiarity and extensive libraries with Perl/awk-inspired syntax for quick scripts and one-liners.
 
 ## Installing Snail
 
@@ -62,7 +57,7 @@ BEGIN { total = 0 }
 END { print("Sum:", total); assert total == 15}
 ```
 
-Built-in variables: `$l` (line), `$f` (fields), `$n` (line number), `$fn` (per-file line number), `$p` (file path), `$m` (last match).
+Built-in variables: `$0` (line), `$1`, `$2` etc (access fields), `$n` (line number), `$fn` (per-file line number), `$p` (file path), `$m` (last match).
 
 
 ### Compact Error Handling
@@ -77,7 +72,7 @@ err = risky()?
 err = risky():$e?
 
 # Provide a fallback value (exception available as $e)
-value = js("malformed json"):{}?
+value = js("malformed json"):{"error": "invalid json"}?
 details = fetch_url("foo.com"):"default html"?
 exception_info = fetch_url("example.com"):$e.http_response_code?
 
@@ -192,7 +187,7 @@ snail 'result = int("oops"):"bad int {$e}"?; print(result)'
 snail 'm = "user@example.com" in /^[\\w.]+@([\\w.]+)$/; if m { print(m[1]) }'
 
 # Awk mode: print line numbers for matches
-rg -n "TODO" README.md | snail --awk '/TODO/ { print("{$n}: {$l}") }'
+rg -n "TODO" README.md | snail --awk '/TODO/ { print("{$n}: {$0}") }'
 ```
 
 ## 🏗️ Architecture
@@ -239,8 +234,6 @@ Installation per platform:
 - **macOS**: `brew install python@3.12` (or use the system Python 3)
 - **Windows**: Download from [python.org](https://www.python.org/downloads/)
 
-**No Python packages required**: Snail vendors jmespath under `snail.vendor`.
-
 ### Build, Test, and Install
 
 ```bash

{snail_lang-0.3.9 → snail_lang-0.4.1}/README.md

@@ -1,17 +1,11 @@
 <p align="center">
   <img src="logo.png" alt="Snail logo" width="200">
 </p>
-
-<h1 align="center">Snail</h1>
 <p align="center"><em>What do you get when you shove a snake in a shell?</em></p>
 
-<h1>Snail, while I hope it is useful to myself and others, is my attempt at
-improving my knowledge of AI code developement. Things are probably broken
-in interesting and horrible ways.</h1>
-
----
+<h1 align="center">Snail</h1>
 
-**Snail** is a programming language that compiles to Python, combining Python's power with Perl/awk-inspired syntax for quick scripts and one-liners. No more whitespace sensitivity—just curly braces and concise expressions.
+**Snail** is a programming language that compiles to Python, combining Python's familiarity and extensive libraries with Perl/awk-inspired syntax for quick scripts and one-liners.
 
 ## Installing Snail
 
@@ -51,7 +45,7 @@ BEGIN { total = 0 }
 END { print("Sum:", total); assert total == 15}
 ```
 
-Built-in variables: `$l` (line), `$f` (fields), `$n` (line number), `$fn` (per-file line number), `$p` (file path), `$m` (last match).
+Built-in variables: `$0` (line), `$1`, `$2` etc (access fields), `$n` (line number), `$fn` (per-file line number), `$p` (file path), `$m` (last match).
 
 
 ### Compact Error Handling
@@ -66,7 +60,7 @@ err = risky()?
 err = risky():$e?
 
 # Provide a fallback value (exception available as $e)
-value = js("malformed json"):{}?
+value = js("malformed json"):{"error": "invalid json"}?
 details = fetch_url("foo.com"):"default html"?
 exception_info = fetch_url("example.com"):$e.http_response_code?
 
@@ -181,7 +175,7 @@ snail 'result = int("oops"):"bad int {$e}"?; print(result)'
 snail 'm = "user@example.com" in /^[\\w.]+@([\\w.]+)$/; if m { print(m[1]) }'
 
 # Awk mode: print line numbers for matches
-rg -n "TODO" README.md | snail --awk '/TODO/ { print("{$n}: {$l}") }'
+rg -n "TODO" README.md | snail --awk '/TODO/ { print("{$n}: {$0}") }'
 ```
 
 ## 🏗️ Architecture
@@ -228,8 +222,6 @@ Installation per platform:
 - **macOS**: `brew install python@3.12` (or use the system Python 3)
 - **Windows**: Download from [python.org](https://www.python.org/downloads/)
 
-**No Python packages required**: Snail vendors jmespath under `snail.vendor`.
-
 ### Build, Test, and Install
 
 ```bash

{snail_lang-0.3.9 → snail_lang-0.4.1}/crates/snail-ast/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "snail-ast"
-version = "0.3.9"
+version = "0.4.1"
 edition = "2024"
 readme = "README.md"
 

{snail_lang-0.3.9 → snail_lang-0.4.1}/crates/snail-core/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "snail-core"
-version = "0.3.9"
+version = "0.4.1"
 edition.workspace = true
 readme = "README.md"
 

{snail_lang-0.3.9 → snail_lang-0.4.1}/crates/snail-error/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "snail-error"
-version = "0.3.9"
+version = "0.4.1"
 edition = "2024"
 readme = "README.md"
 

{snail_lang-0.3.9 → snail_lang-0.4.1}/crates/snail-lower/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "snail-lower"
-version = "0.3.9"
+version = "0.4.1"
 edition = "2024"
 readme = "README.md"
 

{snail_lang-0.3.9 → snail_lang-0.4.1}/crates/snail-lower/README.md

@@ -27,7 +27,7 @@ This crate is the semantic transformation core of the Snail compiler. It takes S
 - **Regex expressions** (`/pattern/`): Transformed into `__snail_regex_compile(pattern)` call
 - **Regex matching** (`string in /pattern/`): Transformed into `__snail_regex_search(string, pattern)` call
 - **Structured accessors** (`$[query]`): Transformed into `__SnailStructuredAccessor(query)` instance
-- **Awk variables**: `$l`, `$f`, `$n`, `$fn`, `$p`, `$m` mapped to Python variable names
+- **Awk variables**: `$0`, `$<num>`, `$n`, `$fn`, `$p`, `$m` mapped to Python variable names
 
 ## Awk Mode Lowering
 
@@ -35,7 +35,7 @@ When lowering awk programs, generates a complete Python AST that:
 1. Imports `sys` for accessing command-line arguments and stdin
 2. Executes BEGIN blocks before processing input
 3. Creates a main loop that reads lines from files or stdin
-4. Updates awk variables (`$l`, `$f`, `$n`, etc.) for each line
+4. Updates awk variables (`$0`, `$<num>`, `$n`, etc.) for each line
 5. Evaluates patterns and executes actions for matching lines
 6. Executes END blocks after all input is processed
 

{snail_lang-0.3.9 → snail_lang-0.4.1}/crates/snail-lower/src/awk.rs

@@ -365,7 +365,12 @@ pub(crate) fn lower_awk_rules_with_auto_print(
     let mut stmts = Vec::new();
     for rule in rules {
         let mut action = if rule.has_explicit_action() {
-            lower_block_with_auto_print(builder, rule.action.as_ref().unwrap(), auto_print)?
+            lower_block_with_auto_print(
+                builder,
+                rule.action.as_ref().unwrap(),
+                auto_print,
+                &rule.span,
+            )?
         } else {
             vec![awk_default_print(builder, &rule.span)?]
         };
@@ -427,8 +432,8 @@ fn regex_pattern_components(pattern: &Expr) -> Option<(Expr, RegexPattern, Sourc
             span,
         } => Some((*value.clone(), pattern.clone(), span.clone())),
         Expr::Regex { pattern, span } => Some((
-            Expr::Name {
-                name: SNAIL_AWK_LINE.to_string(),
+            Expr::FieldIndex {
+                index: "0".to_string(),
                 span: span.clone(),
             },
             pattern.clone(),

{snail_lang-0.3.9 → snail_lang-0.4.1}/crates/snail-lower/src/constants.rs

@@ -9,8 +9,6 @@ pub const SNAIL_JMESPATH_QUERY: &str = "__snail_jmespath_query";
 pub const SNAIL_PARTIAL_HELPER: &str = "__snail_partial";
 
 // Awk-related constants (public within crate)
-pub(crate) const SNAIL_AWK_LINE: &str = "$l";
-pub(crate) const SNAIL_AWK_FIELDS: &str = "$f";
 pub(crate) const SNAIL_AWK_NR: &str = "$n";
 pub(crate) const SNAIL_AWK_FNR: &str = "$fn";
 pub(crate) const SNAIL_AWK_PATH: &str = "$p";
@@ -24,8 +22,6 @@ pub(crate) const SNAIL_AWK_MATCH_PYVAR: &str = "__snail_match";
 
 pub(crate) fn injected_py_name(name: &str) -> Option<&'static str> {
     match name {
-        SNAIL_AWK_LINE => Some(SNAIL_AWK_LINE_PYVAR),
-        SNAIL_AWK_FIELDS => Some(SNAIL_AWK_FIELDS_PYVAR),
         SNAIL_AWK_NR => Some(SNAIL_AWK_NR_PYVAR),
         SNAIL_AWK_FNR => Some(SNAIL_AWK_FNR_PYVAR),
         SNAIL_AWK_PATH => Some(SNAIL_AWK_PATH_PYVAR),

{snail_lang-0.3.9 → snail_lang-0.4.1}/crates/snail-lower/src/program.rs

@@ -18,7 +18,8 @@ pub fn lower_program_with_auto_print(
     auto_print_last: bool,
 ) -> Result<PyObject, LowerError> {
     let builder = AstBuilder::new(py).map_err(py_err_to_lower)?;
-    let body = lower_block_with_auto_print(&builder, &program.stmts, auto_print_last)?;
+    let body =
+        lower_block_with_auto_print(&builder, &program.stmts, auto_print_last, &program.span)?;
     builder.module(body, &program.span).map_err(py_err_to_lower)
 }
 
@@ -62,7 +63,7 @@ pub fn lower_awk_program_with_auto_print(
 
     let mut main_body = Vec::new();
     for block in &program.begin_blocks {
-        let lowered = lower_block_with_auto_print(&builder, block, auto_print)?;
+        let lowered = lower_block_with_auto_print(&builder, block, auto_print, &span)?;
         main_body.extend(lowered);
     }
 
@@ -166,7 +167,7 @@ pub fn lower_awk_program_with_auto_print(
     main_body.push(for_loop);
 
     for block in &program.end_blocks {
-        let lowered = lower_block_with_auto_print(&builder, block, auto_print)?;
+        let lowered = lower_block_with_auto_print(&builder, block, auto_print, &span)?;
         main_body.extend(lowered);
     }
 

{snail_lang-0.3.9 → snail_lang-0.4.1}/crates/snail-lower/src/py_ast.rs

@@ -69,9 +69,9 @@ impl<'py> AstBuilder<'py> {
 
 pub fn set_location(node: &Bound<'_, PyAny>, span: &SourceSpan) -> PyResult<()> {
     node.setattr("lineno", span.start.line)?;
-    node.setattr("col_offset", span.start.column)?;
+    node.setattr("col_offset", span.start.column.saturating_sub(1))?;
     node.setattr("end_lineno", span.end.line)?;
-    node.setattr("end_col_offset", span.end.column)?;
+    node.setattr("end_col_offset", span.end.column.saturating_sub(1))?;
     Ok(())
 }
 

{snail_lang-0.3.9 → snail_lang-0.4.1}/crates/snail-lower/src/stmt.rs

@@ -24,10 +24,10 @@ pub(crate) fn lower_stmt(builder: &AstBuilder<'_>, stmt: &Stmt) -> Result<PyObje
             span,
         } => {
             let test = lower_expr(builder, cond)?;
-            let body = lower_block(builder, body)?;
+            let body = lower_block(builder, body, span)?;
             let orelse = else_body
                 .as_ref()
-                .map(|items| lower_block(builder, items))
+                .map(|items| lower_block(builder, items, span))
                 .transpose()?
                 .unwrap_or_default();
             builder
@@ -51,10 +51,10 @@ pub(crate) fn lower_stmt(builder: &AstBuilder<'_>, stmt: &Stmt) -> Result<PyObje
         } => {
             let target = lower_assign_target(builder, target)?;
             let iter = lower_expr(builder, iter)?;
-            let body = lower_block(builder, body)?;
+            let body = lower_block(builder, body, span)?;
             let orelse = else_body
                 .as_ref()
-                .map(|items| lower_block(builder, items))
+                .map(|items| lower_block(builder, items, span))
                 .transpose()?
                 .unwrap_or_default();
             builder
@@ -77,7 +77,7 @@ pub(crate) fn lower_stmt(builder: &AstBuilder<'_>, stmt: &Stmt) -> Result<PyObje
             span,
         } => {
             let args = lower_parameters(builder, params)?;
-            let body = lower_block(builder, body)?;
+            let body = lower_block(builder, body, span)?;
             builder
                 .call_node(
                     "FunctionDef",
@@ -94,7 +94,7 @@ pub(crate) fn lower_stmt(builder: &AstBuilder<'_>, stmt: &Stmt) -> Result<PyObje
                 .map_err(py_err_to_lower)
         }
         Stmt::Class { name, body, span } => {
-            let body = lower_block(builder, body)?;
+            let body = lower_block(builder, body, span)?;
             builder
                 .call_node(
                     "ClassDef",
@@ -116,19 +116,19 @@ pub(crate) fn lower_stmt(builder: &AstBuilder<'_>, stmt: &Stmt) -> Result<PyObje
             finally_body,
             span,
         } => {
-            let body = lower_block(builder, body)?;
+            let body = lower_block(builder, body, span)?;
             let handlers = handlers
                 .iter()
                 .map(|handler| lower_except_handler(builder, handler))
                 .collect::<Result<Vec<_>, _>>()?;
             let orelse = else_body
                 .as_ref()
-                .map(|items| lower_block(builder, items))
+                .map(|items| lower_block(builder, items, span))
                 .transpose()?
                 .unwrap_or_default();
             let finalbody = finally_body
                 .as_ref()
-                .map(|items| lower_block(builder, items))
+                .map(|items| lower_block(builder, items, span))
                 .transpose()?
                 .unwrap_or_default();
             builder
@@ -149,7 +149,7 @@ pub(crate) fn lower_stmt(builder: &AstBuilder<'_>, stmt: &Stmt) -> Result<PyObje
                 .iter()
                 .map(|item| lower_with_item(builder, item))
                 .collect::<Result<Vec<_>, _>>()?;
-            let body = lower_block(builder, body)?;
+            let body = lower_block(builder, body, span)?;
             builder
                 .call_node(
                     "With",
@@ -313,14 +313,16 @@ pub(crate) fn lower_stmt(builder: &AstBuilder<'_>, stmt: &Stmt) -> Result<PyObje
 pub(crate) fn lower_block(
     builder: &AstBuilder<'_>,
     block: &[Stmt],
+    span: &SourceSpan,
 ) -> Result<Vec<PyObject>, LowerError> {
-    lower_block_with_auto_print(builder, block, false)
+    lower_block_with_auto_print(builder, block, false, span)
 }
 
 pub(crate) fn lower_block_with_auto_print(
     builder: &AstBuilder<'_>,
     block: &[Stmt],
     auto_print: bool,
+    span: &SourceSpan,
 ) -> Result<Vec<PyObject>, LowerError> {
     let mut stmts = Vec::new();
     for (idx, stmt) in block.iter().enumerate() {
@@ -340,6 +342,13 @@ pub(crate) fn lower_block_with_auto_print(
         }
         stmts.push(lower_stmt(builder, stmt)?);
     }
+    if stmts.is_empty() {
+        stmts.push(
+            builder
+                .call_node("Pass", Vec::new(), span)
+                .map_err(py_err_to_lower)?,
+        );
+    }
     Ok(stmts)
 }
 
@@ -352,7 +361,7 @@ fn lower_if(
     span: &SourceSpan,
 ) -> Result<PyObject, LowerError> {
     let test = lower_expr(builder, cond)?;
-    let body = lower_block(builder, body)?;
+    let body = lower_block(builder, body, span)?;
     let orelse = if let Some((elif_cond, elif_body)) = elifs.first() {
         vec![lower_if(
             builder,
@@ -363,7 +372,7 @@ fn lower_if(
             span,
         )?]
     } else if let Some(else_body) = else_body {
-        lower_block(builder, else_body)?
+        lower_block(builder, else_body, span)?
     } else {
         Vec::new()
     };
@@ -468,7 +477,7 @@ fn lower_except_handler(
         .as_ref()
         .map(|name| name.to_string().into_py(builder.py()))
         .unwrap_or_else(|| builder.py().None().into_py(builder.py()));
-    let body = lower_block(builder, &handler.body)?;
+    let body = lower_block(builder, &handler.body, &handler.span)?;
     builder
         .call_node(
             "ExceptHandler",

{snail_lang-0.3.9 → snail_lang-0.4.1}/crates/snail-parser/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "snail-parser"
-version = "0.3.9"
+version = "0.4.1"
 edition = "2024"
 readme = "README.md"
 

{snail_lang-0.3.9 → snail_lang-0.4.1}/crates/snail-parser/src/lib.rs

@@ -86,7 +86,7 @@ pub fn parse_awk_program(source: &str) -> Result<AwkProgram, ParseError> {
     })
 }
 
-const AWK_ONLY_NAMES: [&str; 6] = ["$l", "$f", "$n", "$fn", "$p", "$m"];
+const AWK_ONLY_NAMES: [&str; 4] = ["$n", "$fn", "$p", "$m"];
 const AWK_ONLY_MESSAGE: &str = "awk variables are only valid in awk mode; use --awk";
 
 fn validate_no_awk_syntax(program: &Program, source: &str) -> Result<(), ParseError> {

{snail_lang-0.3.9 → snail_lang-0.4.1}/crates/snail-parser/src/snail.pest

@@ -191,7 +191,7 @@ tuple_literal = { "(" ~ ")" | "(" ~ expr ~ "," ~ (expr ~ ("," ~ expr)* ~ ","?)?
 list_comp = { "[" ~ expr ~ comp_for ~ "]" }
 list_literal = { "[" ~ (expr ~ ("," ~ expr)*)? ~ "]" }
 dict_comp = { "{" ~ expr ~ ":" ~ expr ~ comp_for ~ "}" }
-dict_literal = { "{" ~ (dict_entry ~ ("," ~ dict_entry)*)? ~ "}" }
+dict_literal = { "{" ~ dict_entry ~ ("," ~ dict_entry)* ~ "}" }
 dict_entry = { expr ~ ":" ~ expr }
 
 // Comprehension clauses
@@ -206,7 +206,7 @@ none = { "None" }
 // Special variables: exception, AWK fields, injected vars
 exception_var = { "$e" }
 field_index_var = @{ "$" ~ ASCII_DIGIT+ }
-injected_var = { "$fn" | "$l" | "$f" | "$n" | "$p" | "$m" }
+injected_var = { "$fn" | "$n" | "$p" | "$m" }
 
 // Number, string, and regex literals
 number = @{ ASCII_DIGIT+ ~ ("." ~ ASCII_DIGIT+)? }

{snail_lang-0.3.9 → snail_lang-0.4.1}/crates/snail-parser/tests/errors.rs

@@ -9,7 +9,7 @@ fn reports_parse_error_with_location() {
     let message = err.to_string();
     assert!(message.contains("-->"));
     assert!(message.contains("if"));
-    expect_err_span(&err, 1, 7);
+    expect_err_span(&err, 1, 6);
 }
 
 #[test]
@@ -22,9 +22,9 @@ fn rejects_user_defined_dollar_identifiers() {
 
 #[test]
 fn rejects_awk_only_variables_in_regular_mode() {
-    let err = parse_err("value = $l");
+    let err = parse_err("value = $n");
     let message = err.to_string();
-    assert!(message.contains("$l"));
+    assert!(message.contains("$n"));
     assert!(message.contains("--awk"));
     expect_err_span(&err, 1, 9);
 }
@@ -84,6 +84,12 @@ fn parser_reports_error_on_missing_colon_in_dict() {
     assert!(err.span.is_some());
 }
 
+#[test]
+fn parser_rejects_empty_dict_literal() {
+    let err = parse_err("d = {}");
+    assert!(err.span.is_some());
+}
+
 #[test]
 fn parser_rejects_incomplete_function_def() {
     let err = parse_err("def foo");

{snail_lang-0.3.9 → snail_lang-0.4.1}/crates/snail-parser/tests/parser.rs

@@ -683,7 +683,7 @@ fn parses_class_followed_by_stmt_without_separator() {
 
 #[test]
 fn parses_try_followed_by_stmt_without_separator() {
-    // Note: using explicit exception type since bare `except { }` is ambiguous with set literals
+    // Note: using explicit exception type since bare `except { }` is ambiguous
     let source = "try { x } except Exception { y } z";
     let program = parse_ok(source);
     assert_eq!(program.stmts.len(), 2);

{snail_lang-0.3.9 → snail_lang-0.4.1}/crates/snail-python/Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "snail-python"
-version = "0.3.9"
+version = "0.4.1"
 edition.workspace = true
 
 [lib]

{snail_lang-0.3.9 → snail_lang-0.4.1}/crates/snail-python/src/lib.rs

@@ -3,12 +3,14 @@
 use pyo3::Bound;
 use pyo3::exceptions::{PyRuntimeError, PySyntaxError, PySystemExit};
 use pyo3::prelude::*;
-use pyo3::types::{PyDict, PyList, PyModule};
+use pyo3::types::{PyDict, PyList, PyModule, PyTuple};
 use snail_core::{
     CompileMode, compile_snail_source_with_auto_print, format_snail_error, parse_awk_program,
     parse_program,
 };
 
+const SNAIL_TRACE_PREFIX: &str = "snail:";
+
 fn parse_mode(mode: &str) -> PyResult<CompileMode> {
     match mode {
         "snail" => Ok(CompileMode::Snail),
@@ -19,6 +21,53 @@ fn parse_mode(mode: &str) -> PyResult<CompileMode> {
     }
 }
 
+fn display_filename(filename: &str) -> String {
+    if filename.starts_with(SNAIL_TRACE_PREFIX) {
+        filename.to_string()
+    } else {
+        format!("{SNAIL_TRACE_PREFIX}{filename}")
+    }
+}
+
+fn strip_display_prefix(filename: &str) -> &str {
+    filename
+        .strip_prefix(SNAIL_TRACE_PREFIX)
+        .unwrap_or(filename)
+}
+
+fn split_source_lines(source: &str) -> Vec<String> {
+    let mut lines = Vec::new();
+    let mut start = 0;
+    for (idx, ch) in source.char_indices() {
+        if ch == '\n' {
+            let end = idx + 1;
+            lines.push(source[start..end].to_string());
+            start = end;
+        }
+    }
+    if start < source.len() {
+        lines.push(source[start..].to_string());
+    }
+    lines
+}
+
+fn register_linecache(py: Python<'_>, filename: &str, source: &str) -> PyResult<()> {
+    let linecache = py.import_bound("linecache")?;
+    let cache = linecache.getattr("cache")?;
+    let lines = split_source_lines(source);
+    let entry = PyTuple::new_bound(
+        py,
+        vec![
+            source.len().into_py(py),
+            py.None().into_py(py),
+            PyList::new_bound(py, lines).into_py(py),
+            filename.into_py(py),
+        ],
+    );
+    cache.set_item(filename, entry)?;
+    Ok(())
+}
+
 fn compile_source(
     py: Python<'_>,
     source: &str,
@@ -74,10 +123,12 @@ fn compile_py(
 ) -> PyResult<PyObject> {
     let mode = parse_mode(mode)?;
     let python_ast = compile_source(py, source, mode, auto_print, filename)?;
+    let display = display_filename(filename);
+    register_linecache(py, &display, source)?;
     let builtins = py.import_bound("builtins")?;
     let code = builtins
         .getattr("compile")?
-        .call1((python_ast, filename, "exec"))?;
+        .call1((python_ast, display, "exec"))?;
     Ok(code.unbind())
 }
 
@@ -94,11 +145,13 @@ fn exec_py(
 ) -> PyResult<i32> {
     let mode = parse_mode(mode)?;
     let python_ast = compile_source(py, source, mode, auto_print, filename)?;
+    let display = display_filename(filename);
+    register_linecache(py, &display, source)?;
     let builtins = py.import_bound("builtins")?;
     let code = builtins
         .getattr("compile")?
-        .call1((python_ast, filename, "exec"))?;
-    let globals = prepare_globals(py, filename, &argv, auto_import)?;
+        .call1((python_ast, display, "exec"))?;
+    let globals = prepare_globals(py, strip_display_prefix(filename), &argv, auto_import)?;
 
     match builtins.getattr("exec")?.call1((code.as_any(), &globals)) {
         Ok(_) => Ok(0),

{snail_lang-0.3.9 → snail_lang-0.4.1}/pyproject.toml

@@ -4,11 +4,14 @@ build-backend = "maturin"
 
 [project]
 name = "snail-lang"
-version = "0.3.9"
+version = "0.4.1"
 description = "Snail programming language interpreter"
 readme = "README.md"
 requires-python = ">=3.10"
 license = { file = "LICENSE" }
+dependencies = [
+    "jmespath>=1.0.1",
+]
 
 [project.optional-dependencies]
 dev = [