refactor(interpreter): enhance submodule handling and simplify credential usage
This commit is contained in:
@@ -2,8 +2,6 @@ REGISTRY_PATH = $(shell pwd)/registry
|
|||||||
REGISTRY_REPO = git.nextzenos.com/CDN/bifrost-registry.git
|
REGISTRY_REPO = git.nextzenos.com/CDN/bifrost-registry.git
|
||||||
PYTHON = $(REGISTRY_PATH)/venv/bin/python
|
PYTHON = $(REGISTRY_PATH)/venv/bin/python
|
||||||
update_registry: package_registry upload_registry clean_registry
|
update_registry: package_registry upload_registry clean_registry
|
||||||
( cd $(REGISTRY_PATH) && \
|
|
||||||
$(PYTHON) $(REGISTRY_PATH)/scripts/interpreter/gen_commit.py $(GIT_USER) $(GIT_PASS) $(REGISTRY_REPO) $(REGISTRY_PATH) )
|
|
||||||
|
|
||||||
package_registry:
|
package_registry:
|
||||||
(cd $(REGISTRY_PATH) && \
|
(cd $(REGISTRY_PATH) && \
|
||||||
|
|||||||
@@ -1,96 +1,178 @@
|
|||||||
|
import asyncio
|
||||||
import os
|
import os
|
||||||
import sys
|
import sys
|
||||||
from dotenv import load_dotenv
|
from dotenv import load_dotenv
|
||||||
from interpreter import OpenInterpreter
|
from interpreter import OpenInterpreter
|
||||||
|
|
||||||
load_dotenv() # Load environment variables from .env
|
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) != 5:
|
print("Usage: python gen_commit.py <GIT_USER> <GIT_PASS> <PROJECT_PATH>")
|
||||||
print(
|
|
||||||
"Usage: python gen_commit.py <GIT_USER> <GIT_PASS> <GIT_REPO> <PROJECT_PATH>"
|
|
||||||
)
|
|
||||||
sys.exit(1)
|
sys.exit(1)
|
||||||
return sys.argv[1], sys.argv[2], sys.argv[3], sys.argv[4]
|
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):
|
||||||
"""Use OpenInterpreter to generate a commit message 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 both convention and git commit files
|
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"""
|
||||||
Your name is Bifrost.
|
You are a helpful assistant that generates concise, conventional commit messages based on code diffs.
|
||||||
You are a helpful assistant that generates commit messages based on uncommitted code.
|
|
||||||
|
|
||||||
Follow these rules:
|
Guidelines:
|
||||||
1. Use English.
|
- Use English.
|
||||||
2. Be concise and follow this format:
|
- Do not include explanations—only output commit message in under 100 characters.
|
||||||
{convention_content}
|
- Follow this convention:
|
||||||
|
{convention_content}
|
||||||
3. Use these git commands to analyze changes as you see fit:
|
"""
|
||||||
{git_commit_content}
|
|
||||||
|
diff_output = get_git_diff(project_path)
|
||||||
4. Only return the commit message. Do not add explanation or extra output.
|
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:
|
||||||
|
|
||||||
First, change directory:
|
|
||||||
cd {project_path}
|
|
||||||
|
|
||||||
Then check the uncommitted changes.
|
{diff_output}
|
||||||
|
"""
|
||||||
Then generate a commit message according to the changes.
|
|
||||||
"""
|
|
||||||
|
|
||||||
response = agent.chat(prompt)
|
response = agent.chat(prompt)
|
||||||
|
if isinstance(response, list) and response and isinstance(response[-1], dict) and "content" in response[-1]:
|
||||||
# Extract just the content from the response
|
return response[-1]["content"].strip()
|
||||||
if isinstance(response, list) and len(response) > 0:
|
|
||||||
if isinstance(response[-1], dict) and "content" in response[-1]:
|
|
||||||
return response[-1]["content"].strip()
|
|
||||||
|
|
||||||
return str(response).strip()
|
return str(response).strip()
|
||||||
|
|
||||||
|
|
||||||
def push_code():
|
async def get_submodule_info():
|
||||||
"""Pull, commit, and push code with AI-generated message."""
|
proc = await asyncio.create_subprocess_exec(
|
||||||
git_user, git_pass, git_repo, project_path = get_git_credentials()
|
"git", "config", "--file", ".gitmodules", "--get-regexp", r"^submodule\..*\.path$",
|
||||||
|
stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE,
|
||||||
|
)
|
||||||
|
stdout, stderr = await proc.communicate()
|
||||||
|
|
||||||
# Change to project path
|
if stderr:
|
||||||
|
print(f"Error getting submodule paths:\n{stderr.decode()}")
|
||||||
|
return {}
|
||||||
|
|
||||||
|
submodule_info = {}
|
||||||
|
for line in stdout.decode().splitlines():
|
||||||
|
parts = line.split()
|
||||||
|
if len(parts) == 2:
|
||||||
|
name = parts[0].split('.')[1]
|
||||||
|
path = parts[1]
|
||||||
|
abs_path = os.path.abspath(path)
|
||||||
|
submodule_info[abs_path] = name
|
||||||
|
return submodule_info
|
||||||
|
|
||||||
|
|
||||||
|
async def commit_and_push_submodules():
|
||||||
|
print("Checking for submodule changes...")
|
||||||
|
submodule_info = await get_submodule_info()
|
||||||
|
|
||||||
|
proc = await asyncio.create_subprocess_shell(
|
||||||
|
"git submodule foreach --quiet 'if [ -n \"$(git status --porcelain)\" ]; then echo $path; fi'",
|
||||||
|
stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE,
|
||||||
|
)
|
||||||
|
stdout, stderr = await proc.communicate()
|
||||||
|
if stderr:
|
||||||
|
print(f"Error checking submodules:\n{stderr.decode()}")
|
||||||
|
return
|
||||||
|
|
||||||
|
submodule_paths = stdout.decode().strip().split('\n')
|
||||||
|
git_user, git_pass, _ = get_git_credentials()
|
||||||
|
|
||||||
|
for submodule_path in submodule_paths:
|
||||||
|
if not submodule_path:
|
||||||
|
continue
|
||||||
|
|
||||||
|
abs_path = os.path.abspath(submodule_path)
|
||||||
|
name = submodule_info.get(abs_path, submodule_path)
|
||||||
|
print(f"Processing submodule: {name} at {abs_path}")
|
||||||
|
|
||||||
|
commit_msg = gen_commit(abs_path)
|
||||||
|
|
||||||
|
proc = await asyncio.create_subprocess_exec(
|
||||||
|
"git", "remote", "get-url", "origin",
|
||||||
|
cwd=abs_path, stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE,
|
||||||
|
)
|
||||||
|
stdout, stderr = await proc.communicate()
|
||||||
|
if stderr:
|
||||||
|
print(f"Error getting remote URL for {name}:\n{stderr.decode()}")
|
||||||
|
continue
|
||||||
|
|
||||||
|
remote_url = stdout.decode().strip().replace(".git", "").replace("https://", "")
|
||||||
|
remote_url = f"https://{git_user}:{git_pass}@{remote_url}"
|
||||||
|
|
||||||
|
commands = [
|
||||||
|
f"cd {abs_path} && git add .",
|
||||||
|
f'cd {abs_path} && git commit -m "{commit_msg}" || true',
|
||||||
|
f"cd {abs_path} && git push {remote_url} || true",
|
||||||
|
]
|
||||||
|
|
||||||
|
for cmd in commands:
|
||||||
|
proc = await asyncio.create_subprocess_shell(
|
||||||
|
cmd, stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE
|
||||||
|
)
|
||||||
|
stdout, stderr = await proc.communicate()
|
||||||
|
if stdout:
|
||||||
|
print(stdout.decode())
|
||||||
|
if stderr:
|
||||||
|
print(f"Submodule '{name}' stderr:\n{stderr.decode()}")
|
||||||
|
|
||||||
|
|
||||||
|
async def push_code():
|
||||||
|
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}")
|
||||||
|
|
||||||
# Generate commit message
|
proc = await asyncio.create_subprocess_exec(
|
||||||
|
"git", "remote", "get-url", "origin",
|
||||||
|
stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE,
|
||||||
|
)
|
||||||
|
stdout, stderr = await proc.communicate()
|
||||||
|
if stderr:
|
||||||
|
print(f"Failed to get remote URL: {stderr.decode()}")
|
||||||
|
git_repo = stdout.decode().strip().replace(".git", "").replace("https://", "")
|
||||||
|
remote_url = f"https://{git_user}:{git_pass}@{git_repo}"
|
||||||
|
|
||||||
|
await commit_and_push_submodules()
|
||||||
|
|
||||||
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")
|
||||||
|
|
||||||
# Pull, add, commit, and push
|
commands = [
|
||||||
remote_url = f"https://{git_user}:{git_pass}@{git_repo}"
|
f"git pull {remote_url} || true",
|
||||||
|
"git add .",
|
||||||
|
f'git commit -m "{commit_msg}" || true',
|
||||||
|
f"git push {remote_url} || true",
|
||||||
|
]
|
||||||
|
|
||||||
# Execute git commands in sequence
|
for cmd in commands:
|
||||||
os.system(f"git pull {remote_url} || true")
|
proc = await asyncio.create_subprocess_shell(
|
||||||
os.system("git add .")
|
cmd, stdout=asyncio.subprocess.PIPE, stderr=asyncio.subprocess.PIPE
|
||||||
os.system(f'git commit -m "{commit_msg}" || true')
|
)
|
||||||
os.system(f"git push {remote_url} || true")
|
stdout, stderr = await proc.communicate()
|
||||||
|
if proc.returncode != 0:
|
||||||
|
print(f"Command failed: {cmd}")
|
||||||
|
print(stderr.decode())
|
||||||
|
elif stderr:
|
||||||
|
# Not an error; just info output
|
||||||
|
print(stderr.decode())
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
push_code()
|
asyncio.run(push_code())
|
||||||
|
|||||||
Reference in New Issue
Block a user