From c139d899edbd4179bfad7f826492958e6a08847c Mon Sep 17 00:00:00 2001
From: Wu Zhenyu <wuzhenyu@ustc.edu>
Date: Tue, 5 Dec 2023 09:46:48 +0800
Subject: [PATCH] :sparkles: Allow to customize language

---
 src/repl_python_wakatime/hooks/codestats.py | 18 ++++++++++++++----
 src/repl_python_wakatime/hooks/wakatime.py  |  5 +++--
 2 files changed, 17 insertions(+), 6 deletions(-)

diff --git a/src/repl_python_wakatime/hooks/codestats.py b/src/repl_python_wakatime/hooks/codestats.py
index a5055bf..a5cd2a9 100644
--- a/src/repl_python_wakatime/hooks/codestats.py
+++ b/src/repl_python_wakatime/hooks/codestats.py
@@ -30,7 +30,8 @@
 def codestats_hook(
     api_key: str = "",
     url: str = "https://codestats.net/api/my/pulses",
-    language_type: str = "Terminal (python)",
+    language: str = "python",
+    language_type: str = "Terminal (%s)",
     service_name: str = "codestats",
     user_name: str = gethostname(),
     *args: Any,
@@ -42,12 +43,18 @@ def codestats_hook(
     :type api_key: str
     :param url:
     :type url: str
+    :param language:
+    :type language: str
     :param language_type:
     :type language_type: str
     :param service_name:
     :type service_name: str
     :param user_name:
     :type user_name: str
+    :param args:
+    :type args: Any
+    :param kwargs:
+    :type kwargs: Any
     :rtype: None
     """
     global codestats
@@ -56,7 +63,7 @@ def codestats_hook(
             from ..utils.api import get_api_key
 
             api_key = get_api_key(service_name, user_name)
-        codestats = CodeStats(api_key, url, language_type)
+        codestats = CodeStats(api_key, url, language_type % language)
     codestats.add_xp()
 
 
@@ -67,7 +74,8 @@ def __init__(
         self,
         api_key: str,
         url: str = "https://codestats.net/api/my/pulses",
-        language_type: str = "Terminal (python)",
+        language: str = "python",
+        language_type: str = "Terminal (%s)",
     ) -> None:
         """Init.
 
@@ -75,13 +83,15 @@ def __init__(
         :type api_key: str
         :param url:
         :type url: str
+        :param language:
+        :type language: str
         :param language_type:
         :type language_type: str
         :rtype: None
         """
         self.url = url
         self.api_key = api_key
-        self.language_type = language_type
+        self.language_type = language_type % language
         self.xp_dict = {language_type: 0}
 
         self.sem = threading.Semaphore()
diff --git a/src/repl_python_wakatime/hooks/wakatime.py b/src/repl_python_wakatime/hooks/wakatime.py
index e275bae..552e137 100644
--- a/src/repl_python_wakatime/hooks/wakatime.py
+++ b/src/repl_python_wakatime/hooks/wakatime.py
@@ -10,6 +10,7 @@
 
 def wakatime_hook(
     project: str = "",
+    language: str = "python",
     category: str = "coding",
     plugin: str = "repl-python-wakatime",
     filenames: list[str] = [".git"],
@@ -47,8 +48,8 @@ def wakatime_hook(
             f"--category={category}",
             f"--plugin={plugin}",
             "--entity-type=app",
-            "--entity=python",
-            "--alternate-language=python",
+            f"--entity={language}",
+            f"--alternate-language={language}",
             f"--project={project}",
         ]
     )