allow GNUish macro assignments (`?=`, `+=`)

1 files changed, 10 insertions, 2 deletions

diff --git a/yapymake/makefile/token.py b/yapymake/makefile/token.py
index 57ad4f9..d9ed5ac 100644
--- a/yapymake/makefile/token.py
+++ b/yapymake/makefile/token.py
@@ -54,12 +54,20 @@ class TokenString(Iterable['Token']):
             first_token.text = first_token.text.lstrip()
         self._tokens[0] = first_token
 
-    def rstrip(self) -> None:
+    def rstrip(self, chars: Optional[str] = None) -> None:
         last_token = self._tokens[-1]
         if isinstance(last_token, TextToken):
-            last_token.text = last_token.text.rstrip()
+            last_token.text = last_token.text.rstrip(chars)
         self._tokens[-1] = last_token
 
+    def endswith(self, pattern: str) -> bool:
+        last_token = self._tokens[-1]
+        return isinstance(last_token, TextToken) and last_token.text.endswith(pattern)
+
+    def concat(self, other: 'TokenString') -> 'TokenString':
+        return TokenString([*self._tokens, *other._tokens])
+
+
 @dataclass()
 class Token:
     pass