snail-lang 0.5.3__tar.gz → 0.6.0__tar.gz

{snail_lang-0.5.3 → snail_lang-0.6.0}/Cargo.lock

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

{snail_lang-0.5.3 → snail_lang-0.6.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snail-lang
-Version: 0.5.3
+Version: 0.6.0
 Requires-Dist: jmespath>=1.0.1
 Requires-Dist: maturin>=1.5 ; extra == 'dev'
 Requires-Dist: pytest ; extra == 'dev'
@@ -62,11 +62,21 @@ Process files line-by-line with familiar awk semantics:
 { print($1, "->", $2) }
 ```
 
-Built-in variables: `$0` (line), `$1`, `$2` etc (access fields), `$n` (line number), `$fn` (per-file line number), `$p` (file path), `$m` (last match).
+**Built-in variables:**
 
-Begin/end blocks use CLI flags for setup and teardown:
+| Variable | Description |
+|----------|-------------|
+| `$0` | Current line (with newline stripped) |
+| `$1`, `$2`, ... | Individual fields (whitespace-split) |
+| `$f` | All fields as a list |
+| `$n` | Global line number (across all files) |
+| `$fn` | Per-file line number |
+| `$p` | Current file path |
+| `$m` | Last regex match object |
+
+Begin/end blocks use CLI flags (`-b`/`--begin`, `-e`/`--end`) for setup and teardown:
 ```bash
-echo -e "5\n4\n3\n2\n1" | snail --awk -b 'total = 0' -e 'print("Sum:", total)' '/^[0-9]+/ { total = total + int($1) }'
+echo -e "5\n4\n3\n2\n1" | snail --awk --begin 'total = 0' --end 'print("Sum:", total)' '/^[0-9]+/ { total = total + int($1) }'
 ```
 
 ### Compact Error Handling

{snail_lang-0.5.3 → snail_lang-0.6.0}/README.md

@@ -50,11 +50,21 @@ Process files line-by-line with familiar awk semantics:
 { print($1, "->", $2) }
 ```
 
-Built-in variables: `$0` (line), `$1`, `$2` etc (access fields), `$n` (line number), `$fn` (per-file line number), `$p` (file path), `$m` (last match).
-
-Begin/end blocks use CLI flags for setup and teardown:
+**Built-in variables:**
+
+| Variable | Description |
+|----------|-------------|
+| `$0` | Current line (with newline stripped) |
+| `$1`, `$2`, ... | Individual fields (whitespace-split) |
+| `$f` | All fields as a list |
+| `$n` | Global line number (across all files) |
+| `$fn` | Per-file line number |
+| `$p` | Current file path |
+| `$m` | Last regex match object |
+
+Begin/end blocks use CLI flags (`-b`/`--begin`, `-e`/`--end`) for setup and teardown:
 ```bash
-echo -e "5\n4\n3\n2\n1" | snail --awk -b 'total = 0' -e 'print("Sum:", total)' '/^[0-9]+/ { total = total + int($1) }'
+echo -e "5\n4\n3\n2\n1" | snail --awk --begin 'total = 0' --end 'print("Sum:", total)' '/^[0-9]+/ { total = total + int($1) }'
 ```
 
 ### Compact Error Handling

{snail_lang-0.5.3 → snail_lang-0.6.0}/crates/snail-ast/Cargo.toml

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

{snail_lang-0.5.3 → snail_lang-0.6.0}/crates/snail-ast/src/awk.rs

@@ -4,6 +4,7 @@ use crate::ast::{Expr, SourceSpan, Stmt};
 pub enum CompileMode {
     Snail,
     Awk,
+    Map,
 }
 
 #[derive(Debug, Clone, PartialEq)]

{snail_lang-0.5.3 → snail_lang-0.6.0}/crates/snail-core/Cargo.toml

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

{snail_lang-0.5.3 → snail_lang-0.6.0}/crates/snail-core/src/lib.rs

@@ -32,6 +32,11 @@ pub fn compile_snail_source_with_auto_print(
             let module = lower_awk_program_with_auto_print(py, &program, auto_print_last)?;
             Ok(module)
         }
+        CompileMode::Map => {
+            let program = parse_map_program(source)?;
+            let module = lower_map_program_with_auto_print(py, &program, auto_print_last)?;
+            Ok(module)
+        }
     }
 }
 

{snail_lang-0.5.3 → snail_lang-0.6.0}/crates/snail-error/Cargo.toml

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

{snail_lang-0.5.3 → snail_lang-0.6.0}/crates/snail-lower/Cargo.toml

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

{snail_lang-0.5.3 → snail_lang-0.6.0}/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**: `$0`, `$<num>`, `$n`, `$fn`, `$p`, `$m` mapped to Python variable names
+- **Awk variables**: `$0`, `$<num>`, `$n`, `$fn`, `$f`, `$p`, `$m` mapped to Python variable names
 
 ## Awk Mode Lowering
 

{snail_lang-0.5.3 → snail_lang-0.6.0}/crates/snail-lower/src/constants.rs

@@ -20,6 +20,7 @@ pub(crate) const SNAIL_AWK_NR: &str = "$n";
 pub(crate) const SNAIL_AWK_FNR: &str = "$fn";
 pub(crate) const SNAIL_AWK_PATH: &str = "$p";
 pub(crate) const SNAIL_AWK_MATCH: &str = "$m";
+pub(crate) const SNAIL_AWK_FIELDS: &str = "$f";
 pub(crate) const SNAIL_AWK_LINE_PYVAR: &str = "__snail_line";
 pub(crate) const SNAIL_AWK_FIELDS_PYVAR: &str = "__snail_fields";
 pub(crate) const SNAIL_AWK_NR_PYVAR: &str = "__snail_nr_user";
@@ -27,12 +28,27 @@ pub(crate) const SNAIL_AWK_FNR_PYVAR: &str = "__snail_fnr_user";
 pub(crate) const SNAIL_AWK_PATH_PYVAR: &str = "__snail_path_user";
 pub(crate) const SNAIL_AWK_MATCH_PYVAR: &str = "__snail_match";
 
+// Map-related constants (public within crate)
+pub(crate) const SNAIL_MAP_SRC: &str = "$src";
+pub(crate) const SNAIL_MAP_FD: &str = "$fd";
+pub(crate) const SNAIL_MAP_TEXT: &str = "$text";
+pub(crate) const SNAIL_MAP_SRC_PYVAR: &str = "__snail_src";
+pub(crate) const SNAIL_MAP_FD_PYVAR: &str = "__snail_fd";
+pub(crate) const SNAIL_MAP_TEXT_PYVAR: &str = "__snail_text";
+pub const SNAIL_LAZY_TEXT_CLASS: &str = "__SnailLazyText";
+
 pub(crate) fn injected_py_name(name: &str) -> Option<&'static str> {
     match name {
+        // Awk variables
         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_AWK_MATCH => Some(SNAIL_AWK_MATCH_PYVAR),
+        SNAIL_AWK_FIELDS => Some(SNAIL_AWK_FIELDS_PYVAR),
+        // Map variables
+        SNAIL_MAP_SRC => Some(SNAIL_MAP_SRC_PYVAR),
+        SNAIL_MAP_FD => Some(SNAIL_MAP_FD_PYVAR),
+        SNAIL_MAP_TEXT => Some(SNAIL_MAP_TEXT_PYVAR),
         _ => None,
     }
 }

{snail_lang-0.5.3 → snail_lang-0.6.0}/crates/snail-lower/src/lib.rs

@@ -3,6 +3,7 @@ mod awk;
 mod constants;
 mod expr;
 mod helpers;
+mod map;
 mod operators;
 mod program;
 mod py_ast;
@@ -10,6 +11,7 @@ mod stmt;
 
 // Re-export public API
 pub use constants::*;
+pub use map::{lower_map_program, lower_map_program_with_auto_print};
 pub use program::{
     lower_awk_program, lower_awk_program_with_auto_print, lower_program,
     lower_program_with_auto_print,

snail_lang-0.6.0/crates/snail-lower/src/map.rs

@@ -0,0 +1,235 @@
+use pyo3::prelude::*;
+use pyo3::types::PyList;
+use snail_ast::*;
+use snail_error::LowerError;
+
+use crate::constants::*;
+use crate::helpers::{assign_name, name_expr, string_expr};
+use crate::py_ast::{AstBuilder, py_err_to_lower};
+use crate::stmt::lower_block_with_auto_print;
+
+pub fn lower_map_program(py: Python<'_>, program: &Program) -> Result<PyObject, LowerError> {
+    lower_map_program_with_auto_print(py, program, false)
+}
+
+pub fn lower_map_program_with_auto_print(
+    py: Python<'_>,
+    program: &Program,
+    auto_print_last: bool,
+) -> Result<PyObject, LowerError> {
+    let builder = AstBuilder::new(py).map_err(py_err_to_lower)?;
+    let span = program.span.clone();
+    let mut body = Vec::new();
+
+    // import sys
+    let sys_import = builder
+        .call_node(
+            "Import",
+            vec![
+                PyList::new_bound(
+                    builder.py(),
+                    vec![
+                        builder
+                            .call_node_no_loc(
+                                "alias",
+                                vec![
+                                    "sys".to_string().into_py(builder.py()),
+                                    builder.py().None().into_py(builder.py()),
+                                ],
+                            )
+                            .map_err(py_err_to_lower)?,
+                    ],
+                )
+                .into_py(builder.py()),
+            ],
+            &span,
+        )
+        .map_err(py_err_to_lower)?;
+    body.push(sys_import);
+
+    // __snail_paths = sys.argv[1:] or [line.rstrip('\n') for line in sys.stdin if line.rstrip('\n')]
+    let paths_expr = lower_paths_source(&builder, &span)?;
+    body.push(assign_name(&builder, "__snail_paths", paths_expr, &span)?);
+
+    // Generate file loop
+    let file_loop = lower_map_file_loop(&builder, program, &span, auto_print_last)?;
+    body.push(file_loop);
+
+    builder.module(body, &span).map_err(py_err_to_lower)
+}
+
+fn lower_paths_source(builder: &AstBuilder<'_>, span: &SourceSpan) -> Result<PyObject, LowerError> {
+    // sys.argv[1:]
+    builder
+        .call_node(
+            "Subscript",
+            vec![
+                builder
+                    .call_node(
+                        "Attribute",
+                        vec![
+                            name_expr(
+                                builder,
+                                "sys",
+                                span,
+                                builder.load_ctx().map_err(py_err_to_lower)?,
+                            )?,
+                            "argv".to_string().into_py(builder.py()),
+                            builder.load_ctx().map_err(py_err_to_lower)?,
+                        ],
+                        span,
+                    )
+                    .map_err(py_err_to_lower)?,
+                builder
+                    .call_node(
+                        "Slice",
+                        vec![
+                            builder
+                                .call_node("Constant", vec![1i64.into_py(builder.py())], span)
+                                .map_err(py_err_to_lower)?,
+                            builder.py().None().into_py(builder.py()),
+                            builder.py().None().into_py(builder.py()),
+                        ],
+                        span,
+                    )
+                    .map_err(py_err_to_lower)?,
+                builder.load_ctx().map_err(py_err_to_lower)?,
+            ],
+            span,
+        )
+        .map_err(py_err_to_lower)
+}
+
+fn lower_map_file_loop(
+    builder: &AstBuilder<'_>,
+    program: &Program,
+    span: &SourceSpan,
+    auto_print_last: bool,
+) -> Result<PyObject, LowerError> {
+    // for __snail_src in __snail_paths:
+    //     with open(__snail_src, 'r') as __snail_fd:
+    //         __snail_text = __SnailLazyText(__snail_fd)
+    //         # user code
+
+    // Build the with statement body
+    let mut with_body = Vec::new();
+
+    // __snail_text = __SnailLazyText(__snail_fd)
+    let lazy_text_call = builder
+        .call_node(
+            "Call",
+            vec![
+                name_expr(
+                    builder,
+                    SNAIL_LAZY_TEXT_CLASS,
+                    span,
+                    builder.load_ctx().map_err(py_err_to_lower)?,
+                )?,
+                PyList::new_bound(
+                    builder.py(),
+                    vec![name_expr(
+                        builder,
+                        SNAIL_MAP_FD_PYVAR,
+                        span,
+                        builder.load_ctx().map_err(py_err_to_lower)?,
+                    )?],
+                )
+                .into_py(builder.py()),
+                PyList::empty_bound(builder.py()).into_py(builder.py()),
+            ],
+            span,
+        )
+        .map_err(py_err_to_lower)?;
+    with_body.push(assign_name(
+        builder,
+        SNAIL_MAP_TEXT_PYVAR,
+        lazy_text_call,
+        span,
+    )?);
+
+    // Lower user code
+    let user_code =
+        lower_block_with_auto_print(builder, &program.stmts, auto_print_last, &program.span)?;
+    with_body.extend(user_code);
+
+    // open(__snail_src, 'r')
+    let open_call = builder
+        .call_node(
+            "Call",
+            vec![
+                name_expr(
+                    builder,
+                    "open",
+                    span,
+                    builder.load_ctx().map_err(py_err_to_lower)?,
+                )?,
+                PyList::new_bound(
+                    builder.py(),
+                    vec![
+                        name_expr(
+                            builder,
+                            SNAIL_MAP_SRC_PYVAR,
+                            span,
+                            builder.load_ctx().map_err(py_err_to_lower)?,
+                        )?,
+                        string_expr(builder, "r", false, StringDelimiter::Double, span)?,
+                    ],
+                )
+                .into_py(builder.py()),
+                PyList::empty_bound(builder.py()).into_py(builder.py()),
+            ],
+            span,
+        )
+        .map_err(py_err_to_lower)?;
+
+    // with open(...) as __snail_fd:
+    let with_item = builder
+        .call_node_no_loc(
+            "withitem",
+            vec![
+                open_call,
+                name_expr(
+                    builder,
+                    SNAIL_MAP_FD_PYVAR,
+                    span,
+                    builder.store_ctx().map_err(py_err_to_lower)?,
+                )?,
+            ],
+        )
+        .map_err(py_err_to_lower)?;
+
+    let with_stmt = builder
+        .call_node(
+            "With",
+            vec![
+                PyList::new_bound(builder.py(), vec![with_item]).into_py(builder.py()),
+                PyList::new_bound(builder.py(), with_body).into_py(builder.py()),
+            ],
+            span,
+        )
+        .map_err(py_err_to_lower)?;
+
+    // for __snail_src in __snail_paths:
+    builder
+        .call_node(
+            "For",
+            vec![
+                name_expr(
+                    builder,
+                    SNAIL_MAP_SRC_PYVAR,
+                    span,
+                    builder.store_ctx().map_err(py_err_to_lower)?,
+                )?,
+                name_expr(
+                    builder,
+                    "__snail_paths",
+                    span,
+                    builder.load_ctx().map_err(py_err_to_lower)?,
+                )?,
+                PyList::new_bound(builder.py(), vec![with_stmt]).into_py(builder.py()),
+                PyList::empty_bound(builder.py()).into_py(builder.py()),
+            ],
+            span,
+        )
+        .map_err(py_err_to_lower)
+}

{snail_lang-0.5.3 → snail_lang-0.6.0}/crates/snail-parser/Cargo.toml

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