Please provide the git diff text so that I can generate the commit message for you.

This commit is contained in:
KaySar12 2025-05-15 18:00:17 +07:00
parent f82e8af9ce
commit 06400f62b2

View File

@ -8,64 +8,57 @@ load_dotenv()
def get_git_credentials(): def get_git_credentials():
"""Retrieve Git credentials and project path from command-line arguments."""
if len(sys.argv) != 4: if len(sys.argv) != 4:
print("Usage: python gen_commit.py <GIT_USER> <GIT_PASS> <PROJECT_PATH>") print("Usage: python gen_commit.py <GIT_USER> <GIT_PASS> <PROJECT_PATH>")
sys.exit(1) sys.exit(1)
return sys.argv[1], sys.argv[2], sys.argv[3] return sys.argv[1], sys.argv[2], sys.argv[3]
def get_git_diff(project_path):
"""Run git diff and return the output."""
result = os.popen(f'cd "{project_path}" && git diff').read()
return result.strip()
def gen_commit(project_path): def gen_commit(project_path):
"""Generate commit message using OpenInterpreter based on git diff.""" """Generate commit message using OpenInterpreter based on git diff output."""
agent = OpenInterpreter() agent = OpenInterpreter()
agent.llm.model = "gpt-4o" agent.llm.model = "gpt-4o"
agent.auto_run = True agent.auto_run = True
# Read convention and git commit guidance
convention_path = os.path.join(os.path.dirname(__file__), "resources/commit_convention.md") convention_path = os.path.join(os.path.dirname(__file__), "resources/commit_convention.md")
git_commit_path = os.path.join(os.path.dirname(__file__), "resources/git_commit.md")
with open(convention_path, "r") as f: with open(convention_path, "r") as f:
convention_content = f.read() convention_content = f.read()
with open(git_commit_path, "r") as f:
git_commit_content = f.read()
agent.system_message = f""" agent.system_message = f"""
You are a helpful assistant that generates commit messages based on uncommitted code. You are a helpful assistant that generates concise, conventional commit messages based on code diffs.
Follow these rules: Guidelines:
1. Use English. - Use English.
2. Be concise and follow this format: - Do not include explanationsonly output a single-line commit message.
- Follow this convention:
{convention_content} {convention_content}
3. Use these git commands to analyze changes as you see fit:
{git_commit_content}
4. Only return the commit message. Do not add explanation or extra output.
""" """
diff_output = get_git_diff(project_path)
if not diff_output:
return "chore: no changes to commit"
prompt = f""" prompt = f"""
Generate a commit message based on the following changes in the project directory: Analyze the following git diff and generate a commit message:
cd {project_path}
Check uncommitted changes and generate a commit message.
""" """
response = agent.chat(prompt) response = agent.chat(prompt)
if isinstance(response, list) and response and isinstance(response[-1], dict) and "content" in response[-1]:
if isinstance(response, list) and len(response) > 0: return response[-1]["content"].strip()
if isinstance(response[-1], dict) and "content" in response[-1]:
return response[-1]["content"].strip()
return str(response).strip() return str(response).strip()
async def get_submodule_info(): async def get_submodule_info():
"""Retrieve mapping of submodule paths to their names."""
proc = await asyncio.create_subprocess_exec( proc = await asyncio.create_subprocess_exec(
"git", "config", "--file", ".gitmodules", "--get-regexp", r"^submodule\..*\.path$", "git", "config", "--file", ".gitmodules", "--get-regexp", r"^submodule\..*\.path$",
stdout=asyncio.subprocess.PIPE, stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE,
stderr=asyncio.subprocess.PIPE,
) )
stdout, stderr = await proc.communicate() stdout, stderr = await proc.communicate()
@ -85,17 +78,14 @@ async def get_submodule_info():
async def commit_and_push_submodules(): async def commit_and_push_submodules():
"""Generate commits and push changes for each modified submodule."""
print("Checking for submodule changes...") print("Checking for submodule changes...")
submodule_info = await get_submodule_info() submodule_info = await get_submodule_info()
proc = await asyncio.create_subprocess_shell( proc = await asyncio.create_subprocess_shell(
"git submodule foreach --quiet 'if [ -n \"$(git status --porcelain)\" ]; then echo $path; fi'", "git submodule foreach --quiet 'if [ -n \"$(git status --porcelain)\" ]; then echo $path; fi'",
stdout=asyncio.subprocess.PIPE, stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE,
stderr=asyncio.subprocess.PIPE,
) )
stdout, stderr = await proc.communicate() stdout, stderr = await proc.communicate()
if stderr: if stderr:
print(f"Error checking submodules:\n{stderr.decode()}") print(f"Error checking submodules:\n{stderr.decode()}")
return return
@ -111,15 +101,11 @@ async def commit_and_push_submodules():
name = submodule_info.get(abs_path, submodule_path) name = submodule_info.get(abs_path, submodule_path)
print(f"Processing submodule: {name} at {abs_path}") print(f"Processing submodule: {name} at {abs_path}")
# Generate commit message
commit_msg = gen_commit(abs_path) commit_msg = gen_commit(abs_path)
# Get remote URL
proc = await asyncio.create_subprocess_exec( proc = await asyncio.create_subprocess_exec(
"git", "remote", "get-url", "origin", "git", "remote", "get-url", "origin",
cwd=abs_path, cwd=abs_path, stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE,
stdout=asyncio.subprocess.PIPE,
stderr=asyncio.subprocess.PIPE,
) )
stdout, stderr = await proc.communicate() stdout, stderr = await proc.communicate()
if stderr: if stderr:
@ -129,7 +115,6 @@ async def commit_and_push_submodules():
remote_url = stdout.decode().strip().replace(".git", "").replace("https://", "") remote_url = stdout.decode().strip().replace(".git", "").replace("https://", "")
remote_url = f"https://{git_user}:{git_pass}@{remote_url}" remote_url = f"https://{git_user}:{git_pass}@{remote_url}"
# Commit and push
commands = [ commands = [
f"cd {abs_path} && git add .", f"cd {abs_path} && git add .",
f'cd {abs_path} && git commit -m "{commit_msg}" || true', f'cd {abs_path} && git commit -m "{commit_msg}" || true',
@ -148,17 +133,13 @@ async def commit_and_push_submodules():
async def push_code(): async def push_code():
"""Main routine: commit and push changes for main repo and submodules."""
git_user, git_pass, project_path = get_git_credentials() git_user, git_pass, project_path = get_git_credentials()
os.chdir(project_path) os.chdir(project_path)
print(f"Changed to project path: {project_path}") print(f"Changed to project path: {project_path}")
# Get main repo remote URL
proc = await asyncio.create_subprocess_exec( proc = await asyncio.create_subprocess_exec(
"git", "remote", "get-url", "origin", "git", "remote", "get-url", "origin",
stdout=asyncio.subprocess.PIPE, stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE,
stderr=asyncio.subprocess.PIPE,
) )
stdout, stderr = await proc.communicate() stdout, stderr = await proc.communicate()
if stderr: if stderr:
@ -166,10 +147,8 @@ async def push_code():
git_repo = stdout.decode().strip().replace(".git", "").replace("https://", "") git_repo = stdout.decode().strip().replace(".git", "").replace("https://", "")
remote_url = f"https://{git_user}:{git_pass}@{git_repo}" remote_url = f"https://{git_user}:{git_pass}@{git_repo}"
# Handle submodules
await commit_and_push_submodules() await commit_and_push_submodules()
# Generate and commit main repo
commit_msg = gen_commit(project_path) commit_msg = gen_commit(project_path)
print(f"Generated commit message:\n{commit_msg}\n") print(f"Generated commit message:\n{commit_msg}\n")