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

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

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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: snail-lang
-Version: 0.6.0
+Version: 0.6.2
 Requires-Dist: jmespath>=1.0.1
 Requires-Dist: maturin>=1.5 ; extra == 'dev'
 Requires-Dist: pytest ; extra == 'dev'
@@ -13,11 +13,10 @@ Description-Content-Type: text/markdown; charset=UTF-8; variant=GFM
 <p align="center">
   <img src="logo.png" alt="Snail logo" width="200">
 </p>
-<p align="center"><em>What do you get when you shove a snake in a shell?</em></p>
 
 <h1 align="center">Snail</h1>
 
-**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.
+**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. Its what you get when you shove a snake in a shell.
 
 ## AI Slop!
 
@@ -58,8 +57,10 @@ semicolons are optional. You can separate statements with newlines.
 Process files line-by-line with familiar awk semantics:
 
 ```snail-awk("hello world\nfoo bar\n")
+BEGIN { print("start") }
 /hello/ { print("matched:", $0) }
 { print($1, "->", $2) }
+END { print("done") }
 ```
 
 **Built-in variables:**
@@ -74,11 +75,43 @@ Process files line-by-line with familiar awk semantics:
 | `$p` | Current file path |
 | `$m` | Last regex match object |
 
-Begin/end blocks use CLI flags (`-b`/`--begin`, `-e`/`--end`) for setup and teardown:
+Begin/end blocks can live in the source file (`BEGIN { ... }` / `END { ... }`) or be supplied
+via CLI flags (`-b`/`--begin`, `-e`/`--end`) for setup and teardown. CLI BEGIN blocks run
+before in-file BEGIN blocks; CLI END blocks run after in-file END blocks.
+`BEGIN` and `END` are reserved keywords in all modes.
+BEGIN/END blocks are regular Snail blocks, so awk/map-only `$` variables are not available inside them.
 ```bash
 echo -e "5\n4\n3\n2\n1" | snail --awk --begin 'total = 0' --end 'print("Sum:", total)' '/^[0-9]+/ { total = total + int($1) }'
 ```
 
+### Map Mode
+
+Process files one at a time instead of line-by-line:
+
+```snail-map
+BEGIN { print("start") }
+print("File:", $src)
+print("Size:", len($text), "bytes")
+END { print("done") }
+```
+
+**Built-in variables:**
+
+| Variable | Description |
+|----------|-------------|
+| `$src` | Current file path |
+| `$fd` | Open file handle for the current file |
+| `$text` | Lazy text view of the current file contents |
+
+Begin/end blocks can live in the source file (`BEGIN { ... }` / `END { ... }`) or be supplied
+via CLI flags (`-b`/`--begin`, `-e`/`--end`) for setup and teardown. CLI BEGIN blocks run
+before in-file BEGIN blocks; CLI END blocks run after in-file END blocks.
+BEGIN/END blocks are regular Snail blocks, so awk/map-only `$` variables are not available inside them.
+`BEGIN` and `END` are reserved keywords in all modes.
+```bash
+snail --map --begin "print('start')" --end "print('done')" "print($src)" *.txt
+```
+
 ### Compact Error Handling
 
 The `?` operator makes error handling terse yet expressive:
@@ -238,6 +271,7 @@ Documentation is WIP
 - **[Language Reference](docs/REFERENCE.md)** — Complete syntax and semantics
 - **[examples/all_syntax.snail](examples/all_syntax.snail)** — Every feature in one file
 - **[examples/awk.snail](examples/awk.snail)** — Awk mode examples
+- **[examples/map.snail](examples/map.snail)** — Map mode examples
 
 ## 🔌 Editor Support
 

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

@@ -1,11 +1,10 @@
 <p align="center">
   <img src="logo.png" alt="Snail logo" width="200">
 </p>
-<p align="center"><em>What do you get when you shove a snake in a shell?</em></p>
 
 <h1 align="center">Snail</h1>
 
-**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.
+**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. Its what you get when you shove a snake in a shell.
 
 ## AI Slop!
 
@@ -46,8 +45,10 @@ semicolons are optional. You can separate statements with newlines.
 Process files line-by-line with familiar awk semantics:
 
 ```snail-awk("hello world\nfoo bar\n")
+BEGIN { print("start") }
 /hello/ { print("matched:", $0) }
 { print($1, "->", $2) }
+END { print("done") }
 ```
 
 **Built-in variables:**
@@ -62,11 +63,43 @@ Process files line-by-line with familiar awk semantics:
 | `$p` | Current file path |
 | `$m` | Last regex match object |
 
-Begin/end blocks use CLI flags (`-b`/`--begin`, `-e`/`--end`) for setup and teardown:
+Begin/end blocks can live in the source file (`BEGIN { ... }` / `END { ... }`) or be supplied
+via CLI flags (`-b`/`--begin`, `-e`/`--end`) for setup and teardown. CLI BEGIN blocks run
+before in-file BEGIN blocks; CLI END blocks run after in-file END blocks.
+`BEGIN` and `END` are reserved keywords in all modes.
+BEGIN/END blocks are regular Snail blocks, so awk/map-only `$` variables are not available inside them.
 ```bash
 echo -e "5\n4\n3\n2\n1" | snail --awk --begin 'total = 0' --end 'print("Sum:", total)' '/^[0-9]+/ { total = total + int($1) }'
 ```
 
+### Map Mode
+
+Process files one at a time instead of line-by-line:
+
+```snail-map
+BEGIN { print("start") }
+print("File:", $src)
+print("Size:", len($text), "bytes")
+END { print("done") }
+```
+
+**Built-in variables:**
+
+| Variable | Description |
+|----------|-------------|
+| `$src` | Current file path |
+| `$fd` | Open file handle for the current file |
+| `$text` | Lazy text view of the current file contents |
+
+Begin/end blocks can live in the source file (`BEGIN { ... }` / `END { ... }`) or be supplied
+via CLI flags (`-b`/`--begin`, `-e`/`--end`) for setup and teardown. CLI BEGIN blocks run
+before in-file BEGIN blocks; CLI END blocks run after in-file END blocks.
+BEGIN/END blocks are regular Snail blocks, so awk/map-only `$` variables are not available inside them.
+`BEGIN` and `END` are reserved keywords in all modes.
+```bash
+snail --map --begin "print('start')" --end "print('done')" "print($src)" *.txt
+```
+
 ### Compact Error Handling
 
 The `?` operator makes error handling terse yet expressive:
@@ -226,6 +259,7 @@ Documentation is WIP
 - **[Language Reference](docs/REFERENCE.md)** — Complete syntax and semantics
 - **[examples/all_syntax.snail](examples/all_syntax.snail)** — Every feature in one file
 - **[examples/awk.snail](examples/awk.snail)** — Awk mode examples
+- **[examples/map.snail](examples/map.snail)** — Map mode examples
 
 ## 🔌 Editor Support
 

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

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

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

@@ -216,6 +216,22 @@ pub enum Expr {
         right: Box<Expr>,
         span: SourceSpan,
     },
+    AugAssign {
+        target: Box<AssignTarget>,
+        op: AugAssignOp,
+        value: Box<Expr>,
+        span: SourceSpan,
+    },
+    PrefixIncr {
+        op: IncrOp,
+        target: Box<AssignTarget>,
+        span: SourceSpan,
+    },
+    PostfixIncr {
+        op: IncrOp,
+        target: Box<AssignTarget>,
+        span: SourceSpan,
+    },
     Compare {
         left: Box<Expr>,
         ops: Vec<CompareOp>,
@@ -403,6 +419,23 @@ pub enum BinaryOp {
     Pipeline,
 }
 
+#[derive(Debug, Clone, Copy, PartialEq, Eq)]
+pub enum AugAssignOp {
+    Add,
+    Sub,
+    Mul,
+    Div,
+    FloorDiv,
+    Mod,
+    Pow,
+}
+
+#[derive(Debug, Clone, Copy, PartialEq, Eq)]
+pub enum IncrOp {
+    Increment,
+    Decrement,
+}
+
 #[derive(Debug, Clone, Copy, PartialEq, Eq)]
 pub enum CompareOp {
     Eq,

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

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

snail_lang-0.6.2/crates/snail-core/src/lib.rs

@@ -0,0 +1,101 @@
+use pyo3::prelude::*;
+
+// Re-export all workspace crates for unified API
+pub use snail_ast::*;
+pub use snail_error::*;
+pub use snail_lower::*;
+pub use snail_parser::*;
+
+/// Compilation API
+pub fn compile_snail_source(
+    py: Python<'_>,
+    source: &str,
+    mode: CompileMode,
+) -> Result<PyObject, SnailError> {
+    compile_snail_source_with_auto_print(py, source, mode, false)
+}
+
+pub fn compile_snail_source_with_auto_print(
+    py: Python<'_>,
+    source: &str,
+    mode: CompileMode,
+    auto_print_last: bool,
+) -> Result<PyObject, SnailError> {
+    match mode {
+        CompileMode::Snail => {
+            let program = parse_program(source)?;
+            let module = lower_program_with_auto_print(py, &program, auto_print_last)?;
+            Ok(module)
+        }
+        CompileMode::Awk => {
+            let program = parse_awk_program(source)?;
+            let module = lower_awk_program_with_auto_print(py, &program, auto_print_last)?;
+            Ok(module)
+        }
+        CompileMode::Map => {
+            let (program, begin_blocks, end_blocks) = parse_map_program_with_begin_end(source)?;
+            let module = lower_map_program_with_begin_end(
+                py,
+                &program,
+                &begin_blocks,
+                &end_blocks,
+                auto_print_last,
+            )?;
+            Ok(module)
+        }
+    }
+}
+
+/// Compile an awk program with separate begin and end code blocks.
+/// In-file BEGIN/END blocks are merged with CLI blocks so CLI BEGIN runs first
+/// and CLI END runs last. CLI blocks are parsed as regular Snail programs.
+pub fn compile_awk_source_with_begin_end(
+    py: Python<'_>,
+    main_source: &str,
+    begin_sources: &[&str],
+    end_sources: &[&str],
+    auto_print_last: bool,
+) -> Result<PyObject, SnailError> {
+    let program = parse_awk_program_with_begin_end(main_source, begin_sources, end_sources)?;
+    let module = lower_awk_program_with_auto_print(py, &program, auto_print_last)?;
+    Ok(module)
+}
+
+/// Compile a map program with separate begin and end code blocks.
+/// In-file BEGIN/END blocks are merged with CLI blocks so CLI BEGIN runs first
+/// and CLI END runs last. CLI blocks are parsed as regular Snail programs.
+pub fn compile_map_source_with_begin_end(
+    py: Python<'_>,
+    main_source: &str,
+    begin_sources: &[&str],
+    end_sources: &[&str],
+    auto_print_last: bool,
+) -> Result<PyObject, SnailError> {
+    let (program, mut begin_blocks, mut end_blocks) =
+        parse_map_program_with_begin_end(main_source)?;
+
+    let mut cli_begin_blocks = Vec::new();
+    for source in begin_sources {
+        let begin_program = parse_program(source)?;
+        if !begin_program.stmts.is_empty() {
+            cli_begin_blocks.push(begin_program.stmts);
+        }
+    }
+    cli_begin_blocks.extend(begin_blocks);
+    begin_blocks = cli_begin_blocks;
+
+    for source in end_sources {
+        let end_program = parse_program(source)?;
+        if !end_program.stmts.is_empty() {
+            end_blocks.push(end_program.stmts);
+        }
+    }
+    let module = lower_map_program_with_begin_end(
+        py,
+        &program,
+        &begin_blocks,
+        &end_blocks,
+        auto_print_last,
+    )?;
+    Ok(module)
+}

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

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

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

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

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

@@ -9,11 +9,16 @@ pub const SNAIL_JMESPATH_QUERY: &str = "__snail_jmespath_query";
 pub const SNAIL_PARTIAL_HELPER: &str = "__snail_partial";
 pub const SNAIL_CONTAINS_HELPER: &str = "__snail_contains__";
 pub const SNAIL_CONTAINS_NOT_HELPER: &str = "__snail_contains_not__";
+pub const SNAIL_INCR_ATTR: &str = "__snail_incr_attr";
+pub const SNAIL_INCR_INDEX: &str = "__snail_incr_index";
+pub const SNAIL_AUG_ATTR: &str = "__snail_aug_attr";
+pub const SNAIL_AUG_INDEX: &str = "__snail_aug_index";
 pub(crate) const SNAIL_LET_VALUE: &str = "__snail_let_value";
 pub(crate) const SNAIL_LET_OK: &str = "__snail_let_ok";
 pub(crate) const SNAIL_LET_KEEP: &str = "__snail_let_keep";
 pub(crate) const SNAIL_COMPARE_LEFT: &str = "__snail_compare_left";
 pub(crate) const SNAIL_COMPARE_RIGHT: &str = "__snail_compare_right";
+pub(crate) const SNAIL_INCR_TMP: &str = "__snail_incr_tmp";
 
 // Awk-related constants (public within crate)
 pub(crate) const SNAIL_AWK_NR: &str = "$n";