feat: added additional data

stsg/build.py

@@ -5,7 +5,7 @@ import os
 import markdown
 from typing import Optional
 
-from .config import SOURCE_DIRECTORY, DIST_DIRECTORY, STATIC_DIRECTORY
+from .config import SOURCE_DIRECTORY, DIST_DIRECTORY, LANGUAGE_INFORMATION
 
 
 logger = logging.getLogger("stsg.build")
@@ -96,9 +96,28 @@ class CustomLanguageCode:
     def __repr__(self) -> str:
         return f"{self.language_code}"
 
+    def _get_additional_data(self) -> dict:
+        parsed_language_code = self.language_code.lower().replace("-", "_")
+        if parsed_language_code in LANGUAGE_INFORMATION:
+            return LANGUAGE_INFORMATION[parsed_language_code]
+        
+        parsed_language_code = parsed_language_code.split("_")[0]
+        if parsed_language_code in LANGUAGE_INFORMATION:
+            return LANGUAGE_INFORMATION[parsed_language_code]
+
+        return {}
+    
+    @property
+    def flag(self) -> str:
+        return self._get_additional_data()["flag"]
+    
+    @property
+    def native_name(self) -> str:
+        return self._get_additional_data()["native_name"]
+
     @property
     def html_code(self) -> str:
-        return f'<ul><a href="{self.relative_url}"><bold>{self.language_code}</bold></a></ul>'
+        return f'<ul><a href="{self.relative_url}"><bold>{self.flag} {self.native_name}</bold></a></ul>'
 
 
 def walk_directory(root):