Preserves urls in multi-line snakesay.

Co-authored-by: Piotr Kaznowski <piotr@kazno.dev>

Author: filiplajszczak

snakesay/snakesay.py

@@ -1,6 +1,7 @@
 #!/usr/bin/env python
 import sys
 import textwrap
+import re
 from typing import Tuple, List, Iterator
 
 MESSAGE = '\n'.join([
@@ -33,7 +34,11 @@ def speech_bubble_lines(speech) -> Iterator[str]:
 
 
 def rewrap(speech: str) -> Tuple[List[str], int]:
-    lines = textwrap.wrap(speech)
+    url_pattern = r'https?://[^\s]+'
+    if re.search(url_pattern, speech):
+        lines = textwrap.wrap(speech, break_long_words=False, break_on_hyphens=False)
+    else:
+        lines = textwrap.wrap(speech)
     width = max(len(l) for l in lines) if lines else 0
     return [line.ljust(width) for line in lines], width
 

tests/test_snakesay.py

@@ -35,3 +35,15 @@ def test_three_lines():
 
 def test_multiple_arguments():
     assert '< hi there >' in snakesay('hi', 'there')
+
+
+def test_long_url_not_split():
+    long_url = 'https://www.example.com/very/long/path/that/exceeds/normal/line/length/and/should/not/be/split/across/multiple/lines'
+
+    message = snakesay(long_url)
+
+    assert long_url in message
+
+    lines = message.split('\n')
+    url_containing_lines = [line for line in lines if 'https://' in line]
+    assert len(url_containing_lines) == 1, "URL should appear on exactly one line"