Add dummy shells for init and config subcommands

initialcommit-io · initialcommit-io · commit 9039e0d891ec · 2024-02-16T12:46:24.000-08:00
Signed-off-by: Jacob Stopak &lt;jacob@initialcommit.io&gt;
diff --git a/src/git_sim/__main__.py b/src/git_sim/__main__.py
@@ -251,7 +251,9 @@ def main(
 app.command()(git_sim.commands.clean)
 app.command()(git_sim.commands.clone)
 app.command()(git_sim.commands.commit)
+app.command()(git_sim.commands.config)
 app.command()(git_sim.commands.fetch)
+app.command()(git_sim.commands.init)
 app.command()(git_sim.commands.log)
 app.command()(git_sim.commands.merge)
 app.command()(git_sim.commands.mv)
diff --git a/src/git_sim/commands.py b/src/git_sim/commands.py
@@ -117,6 +117,18 @@ def commit(
     handle_animations(scene=scene)
 
 
+def config(
+    settings: List[str] = typer.Argument(
+        default=None,
+        help="The names and values of one or more config settings to set",
+    )
+):
+    from git_sim.config import Config
+
+    scene = Config(settings=settings)
+    handle_animations(scene=scene)
+
+
 def fetch(
     remote: str = typer.Argument(
         default=None,
@@ -133,6 +145,13 @@ def fetch(
     handle_animations(scene=scene)
 
 
+def init():
+    from git_sim.init import Init
+
+    scene = Init()
+    handle_animations(scene=scene)
+
+
 def log(
     ctx: typer.Context,
     n: int = typer.Option(
diff --git a/src/git_sim/commit.py b/src/git_sim/commit.py
@@ -75,7 +75,7 @@ def construct(self):
             self.vsplit_frame()
             self.setup_and_draw_zones(
                 first_column_name="Working directory",
-                second_column_name="Staging area",
+                second_column_name="Staged files",
                 third_column_name="New commit",
             )
 
@@ -96,10 +96,19 @@ def populate_zones(
             if "git-sim_media" not in x.a_path:
                 firstColumnFileNames.add(x.a_path)
 
-        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
-                )
+        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
+                    )
diff --git a/src/git_sim/config.py b/src/git_sim/config.py
@@ -0,0 +1,72 @@
+import sys
+import os
+import git
+import numpy
+import tempfile
+import shutil
+import stat
+import re
+
+import manim as m
+
+from typing import List
+from argparse import Namespace
+
+from git.exc import GitCommandError, InvalidGitRepositoryError
+from git.repo import Repo
+
+from git_sim.git_sim_base_command import GitSimBaseCommand
+from git_sim.settings import settings
+
+
+class Config(GitSimBaseCommand):
+    def __init__(self, settings: List[str]):
+        super().__init__()
+        self.settings = settings
+
+        for i, setting in enumerate(self.settings):
+            if " " in setting:
+                self.settings[i] = f"\"{setting}\""
+
+        self.cmd += f"{type(self).__name__.lower()} {' '.join(self.settings)}"
+
+    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()
+        #self.recenter_frame()
+        #self.scale_frame()
+        #self.add_details(repo_name)
+        self.show_command_as_title()
+        self.fadeout()
+        self.show_outro()
+
+    def add_details(self, repo_name):
+        text1 = m.Text(
+            f"Successfully cloned from {self.url} into ./{repo_name}",
+            font=self.font,
+            font_size=20,
+            color=self.fontColor,
+            weight=m.BOLD,
+        )
+        text1.move_to([self.camera.frame.get_center()[0], 4, 0])
+
+        text2 = m.Text(
+            f"Cloned repo log:",
+            font=self.font,
+            font_size=20,
+            color=self.fontColor,
+            weight=m.BOLD,
+        )
+        text2.move_to(text1.get_center()).shift(m.DOWN / 2)
+
+        self.toFadeOut.add(text1)
+        self.toFadeOut.add(text2)
+        self.recenter_frame()
+        self.scale_frame()
+
+        if settings.animate:
+            self.play(m.AddTextLetterByLetter(text1), m.AddTextLetterByLetter(text2))
+        else:
+            self.add(text1, text2)
diff --git a/src/git_sim/git_sim_base_command.py b/src/git_sim/git_sim_base_command.py
@@ -101,7 +101,9 @@ def construct(self):
         self.show_outro()
 
     def get_commit(self, sha_or_ref="HEAD"):
-        return self.repo.commit(sha_or_ref)
+        if self.head_exists():
+            return self.repo.commit(sha_or_ref)
+        return "dark"
 
     def get_default_commits(self):
         defaultCommits = [self.get_commit()]
@@ -646,7 +648,9 @@ def setup_and_draw_zones(
         reverse=False,
     ):
         if self.check_all_dark():
-            self.zone_title_offset = 2.0 if platform.system() == "Windows" else 6.0
+            self.zone_title_offset = 2.0 if platform.system() == "Windows" else 2.0
+        elif len(self.drawnCommits) == 1:
+            self.zone_title_offset = 2.0 if platform.system() == "Windows" else 2.0
 
         horizontal = m.Line(
             (
@@ -928,6 +932,9 @@ def populate_zones(
                 firstColumnFileNames.add(z)
 
     def center_frame_on_commit(self, commit):
+        if not commit or commit == "dark":
+            return
+
         if settings.animate:
             self.play(
                 self.camera.frame.animate.move_to(
@@ -938,6 +945,9 @@ def center_frame_on_commit(self, commit):
             self.camera.frame.move_to(self.drawnCommits[commit.hexsha].get_center())
 
     def reset_head_branch(self, hexsha, shift=numpy.array([0.0, 0.0, 0.0])):
+        if not self.head_exists():
+            return
+
         if settings.animate:
             self.play(
                 self.drawnRefs["HEAD"].animate.move_to(
@@ -1046,25 +1056,28 @@ def setup_and_draw_parent(
             fill_opacity=self.ref_fill_opacity,
         )
         circle.height = 1
-        circle.next_to(
-            self.drawnCommits[child.hexsha],
-            m.LEFT if settings.reverse else m.RIGHT,
-            buff=1.5,
-        )
+        if child != "dark":
+            circle.next_to(
+                self.drawnCommits[child.hexsha],
+                m.LEFT if settings.reverse else m.RIGHT,
+                buff=1.5,
+            )
+
         circle.shift(shift)
 
-        start = circle.get_center()
-        end = self.drawnCommits[child.hexsha].get_center()
-        arrow = m.Arrow(
-            start,
-            end,
-            color=self.fontColor,
-            stroke_width=self.arrow_stroke_width,
-            tip_shape=self.arrow_tip_shape,
-            max_stroke_width_to_length_ratio=1000,
-        )
-        length = numpy.linalg.norm(start - end) - (1.5 if start[1] == end[1] else 3)
-        arrow.set_length(length)
+        if child != "dark":
+            start = circle.get_center()
+            end = self.drawnCommits[child.hexsha].get_center()
+            arrow = m.Arrow(
+                start,
+                end,
+                color=self.fontColor,
+                stroke_width=self.arrow_stroke_width,
+                tip_shape=self.arrow_tip_shape,
+                max_stroke_width_to_length_ratio=1000,
+            )
+            length = numpy.linalg.norm(start - end) - (1.5 if start[1] == end[1] else 3)
+            arrow.set_length(length)
 
         commitId = m.Text(
             "abcdef",
@@ -1102,7 +1115,7 @@ def setup_and_draw_parent(
         self.drawnCommits["abcdef"] = circle
         self.toFadeOut.add(circle)
 
-        if draw_arrow:
+        if draw_arrow and child != "dark":
             if settings.animate:
                 self.play(m.Create(arrow), run_time=1 / settings.speed)
             else:
diff --git a/src/git_sim/init.py b/src/git_sim/init.py
@@ -0,0 +1,64 @@
+import sys
+import os
+from argparse import Namespace
+
+import git
+import manim as m
+import numpy
+import tempfile
+import shutil
+import stat
+import re
+
+from git.exc import GitCommandError, InvalidGitRepositoryError
+from git.repo import Repo
+
+from git_sim.git_sim_base_command import GitSimBaseCommand
+from git_sim.settings import settings
+
+
+class Init(GitSimBaseCommand):
+    def __init__(self):
+        super().__init__()
+        self.cmd += f"{type(self).__name__.lower()}"
+
+    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()
+        #self.recenter_frame()
+        #self.scale_frame()
+        #self.add_details(repo_name)
+        self.show_command_as_title()
+        self.fadeout()
+        self.show_outro()
+
+    def add_details(self, repo_name):
+        text1 = m.Text(
+            f"Successfully cloned from {self.url} into ./{repo_name}",
+            font=self.font,
+            font_size=20,
+            color=self.fontColor,
+            weight=m.BOLD,
+        )
+        text1.move_to([self.camera.frame.get_center()[0], 4, 0])
+
+        text2 = m.Text(
+            f"Cloned repo log:",
+            font=self.font,
+            font_size=20,
+            color=self.fontColor,
+            weight=m.BOLD,
+        )
+        text2.move_to(text1.get_center()).shift(m.DOWN / 2)
+
+        self.toFadeOut.add(text1)
+        self.toFadeOut.add(text2)
+        self.recenter_frame()
+        self.scale_frame()
+
+        if settings.animate:
+            self.play(m.AddTextLetterByLetter(text1), m.AddTextLetterByLetter(text2))
+        else:
+            self.add(text1, text2)
