Add util url validator

Author: code-yeongyu

twitter_video_tools/utils/__init__.py

@@ -1,4 +1,5 @@
 # pylint: disable=useless-import-alias
 
 from .execute_parallel import execute_parallel as execute_parallel
+from .url_validator import URLValidator as URLValidator
 from .youtube_dl_wrapper import youtube_dl_wrapper as youtube_dl_wrapper

twitter_video_tools/utils/url_validator.py

@@ -0,0 +1,20 @@
+class URLValidator:
+    link: str
+
+    def __init__(self, link: str) -> None:
+        self.link = link
+
+    def is_valid_twitter_media_link(self) -> bool:
+        return self.is_valid_link() and self._is_twitter_media_link()
+
+    def is_valid_twitter_liked_link(self) -> bool:
+        return self.is_valid_link() and self._is_twitter_liked_link()
+
+    def is_valid_link(self) -> bool:
+        return self.link.startswith('https://twitter.com/') or self.link.startswith('https://monsnode.com/')
+
+    def _is_twitter_media_link(self) -> bool:
+        return self.link.endswith('/media')
+
+    def _is_twitter_liked_link(self) -> bool:
+        return self.link.endswith('/likes')