rtty-soda 0.1.3__tar.gz → 0.1.4__tar.gz

{rtty_soda-0.1.3 → rtty_soda-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: rtty-soda
-Version: 0.1.3
+Version: 0.1.4
 Summary: A CLI tool for Unix-like environments to encrypt a RTTY session using NaCl
 Keywords: cli,encryption,libsodium,nacl,rtty
 Author: Theo Saveliev
@@ -54,8 +54,8 @@ A CLI tool for Unix-like environments to encrypt a RTTY session using NaCl.
 #### Docker
 
 ```
-% docker run -it --rm -h rtty-soda -v .:/app/host nett/rtty-soda:0.1.3
-% docker run -it --rm -h rtty-soda -v .:/app/host nett/rtty-soda:0.1.3-tools
+% docker run -it --rm -h rtty-soda -v .:/app/host nett/rtty-soda:0.1.4
+% docker run -it --rm -h rtty-soda -v .:/app/host nett/rtty-soda:0.1.4-tools
 ```
 
 

{rtty_soda-0.1.3 → rtty_soda-0.1.4}/README.md

@@ -29,8 +29,8 @@ A CLI tool for Unix-like environments to encrypt a RTTY session using NaCl.
 #### Docker
 
 ```
-% docker run -it --rm -h rtty-soda -v .:/app/host nett/rtty-soda:0.1.3
-% docker run -it --rm -h rtty-soda -v .:/app/host nett/rtty-soda:0.1.3-tools
+% docker run -it --rm -h rtty-soda -v .:/app/host nett/rtty-soda:0.1.4
+% docker run -it --rm -h rtty-soda -v .:/app/host nett/rtty-soda:0.1.4-tools
 ```
 
 

{rtty_soda-0.1.3 → rtty_soda-0.1.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "rtty-soda"
-version = "0.1.3"
+version = "0.1.4"
 description = "A CLI tool for Unix-like environments to encrypt a RTTY session using NaCl"
 readme = "README.md"
 license = { text = "MIT" }

{rtty_soda-0.1.3 → rtty_soda-0.1.4}/src/rtty_soda/cli_io.py

@@ -1,4 +1,6 @@
+import random
 import re
+import string
 from pathlib import Path
 from typing import TextIO, cast
 
@@ -14,6 +16,7 @@ __all__ = [
     "read_plaintext_bytes",
     "read_str",
     "remove_whitespace",
+    "write_bytes_atomic",
     "write_output",
 ]
 
@@ -51,6 +54,13 @@ def read_ciphertext_bytes(source: Path, in_enc: Encoder) -> bytes:
     return read_clean_bytes(source)
 
 
+def write_bytes_atomic(target: Path, data: bytes) -> None:
+    temp_name = "".join(random.choices(string.ascii_lowercase, k=10))  # noqa: S311
+    temp_path = target.parent / temp_name
+    temp_path.write_bytes(data)
+    temp_path.replace(target)
+
+
 def write_output(target: Path | None, data: bytes, out_enc: Encoder) -> None:
     if target is None or target.stem == "-":
         if out_enc == RawEncoder:
@@ -67,7 +77,7 @@ def write_output(target: Path | None, data: bytes, out_enc: Encoder) -> None:
                 f"Overwrite the output file? ({target})", default=False, abort=True
             )
 
-        target.write_bytes(data)
+        write_bytes_atomic(target, data)
 
 
 def print_stats(plaintext: bytes, ciphertext: bytes) -> None:

{rtty_soda-0.1.3 → rtty_soda-0.1.4}/src/rtty_soda/encoders/functions.py

@@ -35,6 +35,9 @@ def bytes_to_base(source: bytes, alphabet: str) -> str:
 
 
 def int_to_bytes(number: int) -> bytes:
+    if number == 0:
+        return b"\x00"
+
     buffer = bytearray()
     while number:
         buffer.append(number & 0xFF)