import sys
import git
import manim as m
from git_sim.git_sim_base_command import GitSimBaseCommand
from git_sim.settings import settings
class Commit(GitSimBaseCommand):
def __init__(self, message: str, amend: bool):
super().__init__()
self.message = message
self.amend = amend
self.n_default = 4 if not self.amend else 5
self.n = self.n_default
self.hide_first_tag = True
settings.hide_merged_branches = True
try:
self.selected_branches.append(self.repo.active_branch.name)
except TypeError:
pass
if self.amend and self.message == "New commit":
print(
"git-sim error: The --amend flag must be used with the -m flag to specify the amended commit message."
)
sys.exit(1)
self.cmd += (
f"{type(self).__name__.lower()} {'--amend ' if self.amend else ''}"
+ '-m "'
+ self.message
+ '"'
)
def construct(self):
if not settings.stdout and not settings.output_only_path and not settings.quiet:
print(f"{settings.INFO_STRING} {self.cmd}")
self.show_intro()
head_commit = self.get_commit()
if self.amend:
tree = self.repo.tree()
amended = git.Commit.create_from_tree(
self.repo,
tree,
self.message,
)
head_commit = amended
self.parse_commits(head_commit)
self.center_frame_on_commit(head_commit)
if not self.amend:
self.setup_and_draw_parent(head_commit, self.message)
else:
self.draw_ref(head_commit, self.drawnCommitIds[amended.hexsha])
self.draw_ref(
head_commit,
self.drawnRefs["HEAD"],
text=self.repo.active_branch.name,
color=m.GREEN,
)
self.recenter_frame()
self.scale_frame()
if not self.amend:
self.reset_head_branch("abcdef")
self.vsplit_frame()
self.setup_and_draw_zones(
first_column_name="Working directory",
second_column_name="Staged files",
third_column_name="New commit",
)
self.show_command_as_title()
self.fadeout()
self.show_outro()
def populate_zones(
self,
firstColumnFileNames,
secondColumnFileNames,
thirdColumnFileNames,
firstColumnArrowMap={},
secondColumnArrowMap={},
thirdColumnArrowMap={},
):
for x in self.repo.index.diff(None):
if "git-sim_media" not in x.a_path:
firstColumnFileNames.add(x.a_path)
if self.head_exists():
for y in self.repo.index.diff("HEAD"):
if "git-sim_media" not in y.a_path:
secondColumnFileNames.add(y.a_path)
thirdColumnFileNames.add(y.a_path)
secondColumnArrowMap[y.a_path] = m.Arrow(
stroke_width=3, color=self.fontColor
)
else:
for y in self.repo.index.diff(None, staged=True):
if "git-sim_media" not in y.a_path:
secondColumnFileNames.add(y.a_path)
thirdColumnFileNames.add(y.a_path)
secondColumnArrowMap[y.a_path] = m.Arrow(
stroke_width=3, color=self.fontColor
)