gmsg 0.1.0__tar.gz

gmsg-0.1.0/PKG-INFO

@@ -0,0 +1,15 @@
+Metadata-Version: 2.3
+Name: gmsg
+Version: 0.1.0
+Summary: Generates git commit messages for you using AI.
+License: MIT
+Author: shiyan99s@gmail.com
+Requires-Python: >=3.12
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Requires-Dist: google-genai (>=1.9.0,<2.0.0)
+Description-Content-Type: text/markdown
+
+

gmsg-0.1.0/gmsg/__init__.py

@@ -0,0 +1 @@
+from .gmsg import main

gmsg-0.1.0/gmsg/api_key.py

@@ -0,0 +1,20 @@
+import sys
+import os
+
+from pathlib import Path
+
+
+def get_or_set_api_key():
+    try:
+        rc = os.path.join(Path.home(), ".gmsgrc")
+        if not os.path.isfile(rc):
+            key_inp = input("Enter your Gemini API KEY: ").strip()
+            with open(rc, 'w') as f:
+                f.write(key_inp)
+
+            return key_inp
+
+        with open(rc, 'r') as f:
+            return f.read().strip()
+    except KeyboardInterrupt:
+        sys.exit(0)

gmsg-0.1.0/gmsg/gmsg.py

@@ -0,0 +1,104 @@
+import sys
+import subprocess
+from google import genai
+from pathlib import Path
+from .api_key import get_or_set_api_key
+
+
+def is_git_repo() -> bool:
+    path = Path(Path.cwd())
+    for parent in [path] + list(path.parents):
+        if (parent / ".git").exists():
+            return True
+    return False
+
+
+def git_diff() -> str | None:
+    try:
+        result = subprocess.run(["git", "diff", "--cached"],
+                                stdout=subprocess.PIPE,
+                                stderr=subprocess.PIPE,
+                                check=True,
+                                text=True)
+
+        if not result.stdout:
+            return None
+        return result.stdout
+    except subprocess.CalledProcessError:
+        sys.exit(1)
+
+
+def make_query(diff) -> str:
+    prompt = f"Generate a one liner git commit message for these changes, Below are the changes:\n {diff}"
+    return prompt
+
+
+def generate_commit_message(diff: str) -> str:
+    try:
+        query = make_query(diff)
+        while True:
+            client = genai.Client(api_key=USER_API_KEY)
+            msg = client.models.generate_content(model="gemini-2.0-flash",
+                                                 contents=query)
+
+            print(msg.text)
+            # generated_msg = "feat: Add poetry.lock and pyproject.toml files."
+            # print(generated_msg)
+            if continue_with_this_prompt(msg.text):
+                break
+    except genai.errors.ClientError as error:
+        if error.code == 400:
+            print(dict(error="Gemini API key not valid, check ~/.gmsg"))
+        else:
+            print(error)
+            sys.exit(1)
+
+
+def continue_with_this_prompt(commit_msg: str) -> bool:
+    action = input("Do you want to continue with this message [Y/n]: ")
+    if action in ("", "y", "Y"):
+        print(f"Running: `git commit -m {commit_msg}`")
+        print(
+            f"Message commited to git, to confirm you can run git commit --amend."
+        )
+        commit_message_to_git(commit_msg)
+        return True
+    elif action in ("n", "N"):
+        return False
+
+
+def commit_message_to_git(generated_msg: str) -> str | None:
+    try:
+        result = subprocess.run(["git", "commit", "-m", generated_msg],
+                                stdout=subprocess.PIPE,
+                                stderr=subprocess.PIPE,
+                                check=True,
+                                text=True)
+
+        return result.stdout
+    except subprocess.CalledProcessError:
+        return None
+
+
+def main():
+    global USER_API_KEY
+
+    USER_API_KEY = get_or_set_api_key()
+    try:
+        if not is_git_repo():
+            print("Current directory does not have git initialized",
+                  file=sys.stdout)
+            sys.exit(1)
+        diff = git_diff()
+        if not diff:
+            print("There's nothing in git diff.")
+            sys.exit(1)
+
+        generate_commit_message(diff)
+
+    except KeyboardInterrupt:
+        sys.exit(0)
+
+
+if __name__ == "__main__":
+    main()

gmsg-0.1.0/pyproject.toml

@@ -0,0 +1,21 @@
+[project]
+name = "gmsg"
+version = "0.1.0"
+description = "Generates git commit messages for you using AI."
+authors = [
+    {name = "shiyan99s@gmail.com"}
+]
+license = {text = "MIT"}
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = [
+    "google-genai (>=1.9.0,<2.0.0)"
+]
+
+[project.scripts]
+gmsg = "gmsg.gmsg:main"
+
+
+[build-system]
+requires = ["poetry-core>=2.0.0,<3.0.0"]
+build-backend = "poetry.core.masonry.api"