imp-git 0.0.26__py3-none-any.whl

imp/commands/fix.py

@@ -0,0 +1,78 @@
+import json
+import shutil
+import subprocess
+
+import typer
+
+from imp import ai, console, git, prompts, validate
+
+
+def fix (
+   issue: int = typer.Argument (..., help="GitHub issue number"),
+   yes: bool = typer.Option (False, "--yes", "-y", help="Accept suggested branch name"),
+   whisper: str = typer.Option ("", "--whisper", "-w", help="Hint to guide the AI"),
+):
+   """Create a branch from a GitHub issue.
+
+   Fetches the issue title and body from GitHub using the gh CLI, then
+   uses AI to generate a branch name. After creation, displays the issue
+   context so you can start working immediately.
+   """
+
+   git.require ()
+
+   if not shutil.which ("gh"):
+      console.err ("GitHub CLI (gh) not installed")
+      console.hint ("https://cli.github.com")
+      raise typer.Exit (1)
+
+   console.header (f"Fix Issue #{issue}")
+
+   try:
+      result = console.spin (
+         "Fetching issue...",
+         subprocess.run,
+         [ "gh", "issue", "view", str (issue), "--json", "title,body,labels" ],
+         capture_output=True,
+         text=True,
+         check=True,
+      )
+      data = json.loads (result.stdout)
+   except (subprocess.CalledProcessError, json.JSONDecodeError, OSError) as e:
+      console.err (f"Could not fetch issue #{issue}: {e}")
+      raise typer.Exit (1) from None
+
+   title = data.get ("title", "")
+   body = (data.get ("body", "") or "") [:500]
+
+   console.label ("Issue")
+   console.item (f"#{issue}: {title}")
+   console.out.print ()
+
+   name = ai.fast (prompts.fix (title, body, whisper))
+   name = ai.oneline (name)
+
+   if not validate.branch (name):
+      console.err (f"Invalid branch name: {name}")
+      raise typer.Exit (1)
+
+   console.label ("Branch")
+   console.item (name)
+   console.out.print ()
+
+   if yes or console.confirm ("Create branch?"):
+      git.checkout (name, create=True)
+      console.success (f"Switched to {name}")
+      console.out.print ()
+      console.label ("Context")
+      console.divider ()
+      console.out.print (title)
+      console.out.print ()
+      body_preview = "\n".join (body.splitlines () [:10])
+      if body_preview:
+         console.out.print (body_preview)
+      console.divider ()
+      console.hint ("make changes, then imp commit")
+   else:
+      console.muted ("Cancelled")
+      raise typer.Exit (0)

imp/commands/help.py

@@ -0,0 +1,103 @@
+from imp import console
+
+
+def help ():
+   """Show workflow guide and common commands.
+
+   Prints a quick-reference of all imp commands organized by workflow
+   phase (starting, working, syncing, shipping) with common flow
+   examples for solo, feature-branch, and hotfix patterns.
+   """
+
+   console.header ("imp workflow")
+
+   console.out.print ("imp wraps git with AI. You commit locally as you work,")
+   console.out.print ("then squash everything into a clean release when ready.")
+   console.out.print ()
+
+   console.divider ()
+   console.out.print ()
+
+   console.out.print ("[bold]Starting a feature[/bold]")
+   console.out.print ('  imp branch "add auth"                     [muted]# create branch from description[/muted]')
+   console.out.print ("  imp branch                                [muted]# switch between branches[/muted]")
+   console.out.print ("  imp fix 42                                [muted]# or from a GitHub issue[/muted]")
+   console.out.print ()
+
+   console.out.print ("[bold]While working[/bold]")
+   console.out.print ("  imp review                                [muted]# AI code review[/muted]")
+   console.out.print ("  imp commit -a                             [muted]# stage all + AI commit message[/muted]")
+   console.out.print ("  imp split                                 [muted]# group changes into logical commits[/muted]")
+   console.out.print ("  imp amend                                 [muted]# fix last commit[/muted]")
+   console.out.print ("  imp undo [N]                              [muted]# undo last N commits[/muted]")
+   console.out.print ("  imp revert <hash>                         [muted]# safely undo a pushed commit[/muted]")
+   console.out.print ()
+
+   console.out.print ("[bold]Staying in sync[/bold]")
+   console.out.print ("  imp sync                                  [muted]# pull, rebase, push[/muted]")
+   console.out.print ("  imp resolve                               [muted]# AI merge conflict resolution[/muted]")
+   console.out.print ("  imp status                                [muted]# repo overview[/muted]")
+   console.out.print ("  imp log                                   [muted]# pretty commit graph[/muted]")
+   console.out.print ()
+
+   console.out.print ("[bold]Shipping[/bold]")
+   console.out.print ("  imp pr                                    [muted]# create pull request[/muted]")
+   console.out.print ("  imp done                                  [muted]# clean up after PR merge[/muted]")
+   console.out.print ("  imp clean                                 [muted]# delete merged branches[/muted]")
+   console.out.print ("  imp release                               [muted]# squash + changelog + tag + push[/muted]")
+   console.out.print ("  imp ship [level]                           [muted]# commit + release, no prompts[/muted]")
+   console.out.print ()
+
+   console.out.print ("[bold]Setup[/bold]")
+   console.out.print ("  imp config                                [muted]# configure AI provider and models[/muted]")
+   console.out.print ("  imp doctor                                [muted]# verify tools and connection[/muted]")
+   console.out.print ()
+
+   console.divider ()
+   console.out.print ()
+
+   console.out.print ("[bold]Commit format[/bold] [muted](Conventional Commits)[/muted]")
+   console.out.print ()
+   console.out.print ("  [muted]type: message[/muted]")
+   console.out.print ("  [muted]type(scope): message[/muted]")
+   console.out.print ("  [muted]type!: message                          breaking change[/muted]")
+   console.out.print ()
+   console.out.print ("  feat        [muted]new feature[/muted]        build    [muted]build system, deps[/muted]")
+   console.out.print ("  fix         [muted]bug fix[/muted]            chore    [muted]maintenance, config[/muted]")
+   console.out.print ("  refactor    [muted]restructure code[/muted]   docs     [muted]documentation[/muted]")
+   console.out.print ("  test        [muted]add/update tests[/muted]   style    [muted]formatting, whitespace[/muted]")
+   console.out.print ("  perf        [muted]performance[/muted]        ci       [muted]CI/CD pipelines[/muted]")
+   console.out.print ()
+   console.out.print ("  [muted]Tickets go after the colon:[/muted]  fix: IMP-123 resolve timeout")
+   console.out.print ("  [muted]Scopes are optional:[/muted]         refactor(auth): simplify flow")
+   console.out.print ()
+   console.out.print ("  [muted]All lowercase after colon (except ticket IDs)[/muted]")
+   console.out.print ("  [muted]Imperative mood (add, not added). Max 72 chars, no period.[/muted]")
+   console.out.print ()
+
+   console.divider ()
+   console.out.print ()
+
+   console.out.print ("[bold]AI whisper[/bold]")
+   console.out.print ()
+   console.out.print ('  [muted]Any AI command accepts[/muted] --whisper / -w [muted]to hint the AI:[/muted]')
+   console.out.print ()
+   console.out.print ('  imp commit -a -w "use IMP-99999 as ticket"')
+   console.out.print ('  imp branch "auth flow" -w "use feat/ prefix"')
+   console.out.print ('  imp review -w "focus on error handling"')
+   console.out.print ()
+
+   console.divider ()
+   console.out.print ()
+
+   console.out.print ("[bold]Common flows[/bold]")
+   console.out.print ()
+   console.muted ("Solo (trunk-based):")
+   console.out.print ("  imp commit -a  →  imp commit -a  →  imp release")
+   console.out.print ()
+   console.muted ("Feature branch:")
+   console.out.print ("  imp branch  →  imp commit -a  →  imp pr  →  imp done")
+   console.out.print ()
+   console.muted ("Hotfix:")
+   console.out.print ("  imp fix 42  →  imp commit -a  →  imp pr  →  imp done")
+   console.out.print ()

imp/commands/log.py

@@ -0,0 +1,25 @@
+import typer
+
+from imp import console, git
+
+
+def log (
+   count: int = typer.Option (20, "-n", help="Number of commits"),
+   ref: str | None = typer.Argument (None, help="Branch or commit ref"),
+):
+   """Show pretty commit graph.
+
+   Displays a decorated commit graph with branch topology. Defaults to
+   the last 20 commits; use -n to adjust. Optionally pass a branch or
+   ref to view its history instead of the current branch.
+   """
+
+   git.require ()
+
+   console.header ("Log")
+
+   output = git.log_graph (count, ref or "")
+   if output:
+      print (output)
+   else:
+      console.muted ("No commits")

imp/commands/pr.py

@@ -0,0 +1,125 @@
+import shutil
+import subprocess
+
+import typer
+
+from imp import ai, console, git, prompts
+
+
+def _parse_response (content: str) -> tuple [str, str]:
+   title = ""
+   for line in content.splitlines ():
+      if line.startswith ("TITLE:"):
+         title = line [6:].strip ()
+         break
+
+   description = ""
+   parts = content.split ("DESCRIPTION:", 1)
+   if len (parts) > 1:
+      description = parts [1].strip ()
+
+   return title, description
+
+
+def pr (
+   yes: bool = typer.Option (False, "--yes", "-y", help="Accept AI description without review"),
+   whisper: str = typer.Option ("", "--whisper", "-w", help="Hint to guide the AI"),
+):
+   """Create a GitHub pull request with AI-generated description.
+
+   Diffs the current branch against the base branch, then uses AI to
+   generate a PR title and description. Pushes to origin if needed and
+   creates the PR via the gh CLI. Requires gh to be installed.
+   """
+
+   git.require ()
+
+   if not shutil.which ("gh"):
+      console.err ("GitHub CLI (gh) not installed")
+      console.hint ("https://cli.github.com")
+      raise typer.Exit (1)
+
+   console.header ("Pull Request")
+
+   b = git.branch ()
+   base = git.base_branch ()
+
+   if b == base:
+      console.err (f"Cannot create PR from {base}")
+      console.hint ("imp branch <description>")
+      raise typer.Exit (1)
+
+   log = git.log_oneline (rev_range=f"{base}..{b}")
+
+   if not log:
+      console.err (f"No commits on {b}")
+      raise typer.Exit (1)
+
+   console.label ("Branch")
+   console.item (f"{b} → {base}")
+   console.out.print ()
+
+   console.items ("Commits", log)
+
+   d = git.diff_range (f"{base}..{b}", max_lines=ai.MAX_DIFF_LINES)
+
+   pr_content = console.spin (
+      "Thinking...",
+      ai.smart,
+      prompts.pr (b, log, d, whisper),
+      False,
+   )
+
+   title, description = _parse_response (pr_content)
+
+   if not title:
+      console.warn ("Could not parse title, using branch name")
+      title = b
+
+   console.label ("Title")
+   console.item (title)
+   console.out.print ()
+
+   console.label ("Description")
+   console.divider ()
+   console.md (description)
+   console.divider ()
+   console.out.print ()
+
+   if not yes:
+      choice = console.choose ("Create PR?", [ "Yes", "Edit", "No" ])
+
+      if choice == "Edit":
+         edited = console.edit (f"{title}\n\n{description}")
+         lines = edited.splitlines ()
+         title = lines [0] if lines else title
+         description = "\n".join (lines [2:]) if len (lines) > 2 else description
+      elif choice == "No":
+         console.muted ("Cancelled")
+         raise typer.Exit (0)
+
+   if not git.has_upstream ():
+      console.spin ("Pushing to origin...", git.push, False, True, b)
+
+   try:
+      result = subprocess.run (
+         [
+            "gh", "pr", "create",
+            "--title", title,
+            "--body", description,
+            "--base", base,
+         ],
+         capture_output=True,
+         text=True,
+         check=True,
+      )
+      pr_url = result.stdout.strip ()
+   except subprocess.CalledProcessError:
+      console.err ("Failed to create PR")
+      raise typer.Exit (1) from None
+
+   console.out.print ()
+   console.success ("Created PR")
+   console.item (pr_url)
+
+   console.hint ("gh pr view --web to open in browser")

imp/commands/push.py

@@ -0,0 +1,42 @@
+import subprocess
+
+import typer
+
+from imp import console, git
+
+
+def push ():
+   """Push commits to origin.
+
+   Pushes the current branch. Sets upstream tracking on first push.
+   Does not create tags, changelogs, or releases.
+   """
+
+   git.require ()
+   git.require_clean ()
+
+   console.header ("Push")
+
+   b = git.branch ()
+
+   if not git.remote_exists ():
+      console.err ("No remote configured")
+      console.hint ("git remote add origin <url>")
+      raise typer.Exit (1)
+
+   ahead = 0
+   if git.has_upstream ():
+      git.fetch ()
+      ahead = git.count_ahead ()
+
+      if ahead == 0:
+         console.success ("Nothing to push")
+         raise typer.Exit (0)
+
+      console.item (f"{ahead} commits on {b}")
+      git.push ()
+   else:
+      console.item (f"Setting upstream for {b}")
+      git.push (set_upstream=True, target=b)
+
+   console.success ("Pushed to origin")

imp/commands/release.py

@@ -0,0 +1,273 @@
+import shutil
+import subprocess
+from datetime import date
+from pathlib import Path
+
+import typer
+
+from imp import console, git, version
+
+
+def _rollback_release (
+   ver: str,
+   original_head: str,
+   changelog_path: Path,
+   original_changelog: str,
+   committed: bool,
+):
+   console.warn ("Rolling back...")
+   git.tag_delete (f"v{ver}")
+   if committed:
+      git.reset (original_head, hard=True)
+   if original_changelog:
+      changelog_path.write_text (original_changelog)
+   elif changelog_path.is_file ():
+      changelog_path.unlink ()
+   console.err ("Release failed")
+
+
+def _write_changelog (path: Path, new_entry: str):
+   if path.is_file ():
+      content = path.read_text ()
+
+      lines = content.splitlines (keepends=True)
+      insert_at = None
+      for i, line in enumerate (lines):
+         if line.lstrip ().startswith ("## "):
+            insert_at = i
+            break
+
+      if insert_at is not None:
+         before = "".join (lines [:insert_at])
+         after = "".join (lines [insert_at:])
+         content = before + new_entry + "\n\n" + after
+      else:
+         content = content + "\n" + new_entry + "\n"
+
+      path.write_text (content)
+   else:
+      path.write_text (
+         f"# Changelog\n\n"
+         f"All notable changes to this project will be documented in this file.\n\n"
+         f"{new_entry}\n"
+      )
+
+
+def _squash_commits (tag: str, summary: str, changelog_path: str, count: int) -> bool:
+   can_squash = False
+   if tag:
+      if git.has_upstream ():
+         unpushed = git.log_oneline (rev_range="@{u}..HEAD")
+         if unpushed:
+            can_squash = True
+      else:
+         can_squash = True
+
+   if can_squash:
+      git.reset (tag, soft=True)
+      git.stage (all=True)
+      git.commit (summary)
+      console.success (f"Squashed {count} commits")
+   else:
+      git.add ([ str (changelog_path) ])
+      git.commit (summary)
+      if tag:
+         console.muted ("Commits already pushed, skipped squash")
+
+   return can_squash
+
+
+def _push_release (ver: str, entry: str, can_squash: bool):
+   if git.has_upstream ():
+      if can_squash:
+         git.push (force_lease=True)
+      else:
+         git.push ()
+   else:
+      b = git.branch ()
+      git.push (set_upstream=True, target=b)
+
+   git.push (ref=f"v{ver}")
+   console.success ("Pushed to origin")
+
+   if shutil.which ("gh"):
+      try:
+         subprocess.run (
+            [
+               "gh", "release", "create",
+               f"v{ver}",
+               "--title", f"v{ver}",
+               "--notes", entry,
+            ],
+            capture_output=True,
+            text=True,
+            check=True,
+         )
+         console.success ("Created GitHub release")
+      except subprocess.CalledProcessError:
+         console.muted ("GitHub release skipped (gh auth or repo issue)")
+
+
+def release ():
+   """Squash, changelog, tag, and push a release.
+
+   Collects commits since the last tag, lets you pick a semver bump,
+   generates a changelog entry, squashes unpushed commits into one, tags
+   the release, and optionally pushes with a GitHub release. Rolls back
+   automatically if anything fails.
+   """
+
+   git.require ()
+
+   base = git.base_branch ()
+   current = git.branch ()
+   if current != base:
+      console.warn (f"Releasing from {current}, not {base}")
+      if not console.confirm ("Continue?"):
+         console.muted ("Cancelled")
+         raise typer.Exit (0)
+
+   git.require_clean ("imp commit first")
+
+   tag = git.last_tag ()
+
+   log = ""
+   if tag:
+      log = git.log_oneline (rev_range=f"{tag}..HEAD")
+
+   if not log:
+      log = git.log_oneline (count=20)
+      tag = ""
+
+   if not log:
+      console.muted ("No commits to release")
+      raise typer.Exit (0)
+
+   count = len (log.splitlines ())
+
+   console.header ("Release")
+
+   console.items (f"Commits since {tag or 'beginning'}", log)
+
+   highest = git.highest_tag ()
+   current = highest.lstrip ("v") if highest else "0.0.0"
+   if not current:
+      current = "0.0.0"
+
+   patch_ver = version.bump (current, "patch")
+   minor_ver = version.bump (current, "minor")
+   major_ver = version.bump (current, "major")
+
+   console.muted (f"Current: {current}")
+   console.out.print ()
+
+   choice = console.choose (
+      "Version bump",
+      [
+         f"patch  {patch_ver}",
+         f"minor  {minor_ver}",
+         f"major  {major_ver}",
+         "custom",
+         "quit",
+      ],
+   )
+
+   if choice.startswith ("patch"):
+      new_version = patch_ver
+   elif choice.startswith ("minor"):
+      new_version = minor_ver
+   elif choice.startswith ("major"):
+      new_version = major_ver
+   elif choice == "custom":
+      new_version = console.prompt ("Version:", patch_ver)
+   else:
+      console.muted ("Cancelled")
+      raise typer.Exit (0)
+
+   if git.tag_exists (f"v{new_version}"):
+      console.err (f"Tag v{new_version} already exists")
+      console.hint (f"pick a different version, or: git tag -d v{new_version}")
+      raise typer.Exit (1)
+
+   if tag:
+      subjects = git.log_subjects (rev_range=f"{tag}..HEAD")
+   else:
+      subjects = git.log_subjects (count=count)
+
+   entry = version.changelog_from_commits (subjects)
+
+   console.label (f"v{new_version}")
+   console.divider ()
+   console.md (entry)
+   console.divider ()
+   console.out.print ()
+
+   choice = console.choose (
+      f"Release v{new_version}?",
+      [ "Yes", "Edit", "No" ],
+   )
+
+   if choice == "Edit":
+      entry = console.edit (entry)
+      console.out.print ()
+      console.label (f"v{new_version} (edited)")
+      console.divider ()
+      console.md (entry)
+      console.divider ()
+      console.out.print ()
+      if not console.confirm (f"Release v{new_version}?"):
+         console.muted ("Cancelled")
+         raise typer.Exit (0)
+   elif choice == "No":
+      console.muted ("Cancelled")
+      raise typer.Exit (0)
+
+   will_push = console.confirm ("Push after release?")
+
+   if will_push and not git.remote_exists ():
+      console.err ("No remote configured")
+      console.hint ("git remote add origin <url>")
+      raise typer.Exit (1)
+
+   summary = f"chore: release v{new_version}"
+   today = date.today ().isoformat ()
+   new_entry = f"## [{new_version}] - {today}\n\n{entry}"
+
+   root = git.repo_root ()
+   changelog_path = Path (root) / "CHANGELOG.md"
+
+   original_head = git.rev_parse ("HEAD")
+   original_changelog = ""
+   if changelog_path.is_file ():
+      original_changelog = changelog_path.read_text ()
+
+   committed = False
+
+   try:
+      _write_changelog (changelog_path, new_entry)
+      console.success ("Updated CHANGELOG.md")
+
+      can_squash = _squash_commits (tag, summary, changelog_path, count)
+      committed = True
+
+      git.tag (f"v{new_version}")
+      console.success (f"Tagged v{new_version}")
+
+   except (subprocess.CalledProcessError, OSError) as e:
+      msg = getattr (e, "stderr", "") or str (e)
+      console.err (f"Release failed: {msg.strip ()}")
+      _rollback_release (
+         new_version, original_head, changelog_path,
+         original_changelog, committed,
+      )
+      raise typer.Exit (1) from None
+
+   if will_push:
+      try:
+         _push_release (new_version, entry, can_squash)
+      except (subprocess.CalledProcessError, OSError) as e:
+         msg = getattr (e, "stderr", "") or str (e)
+         console.err (f"Push failed: {msg.strip ()}")
+         raise typer.Exit (1) from None
+
+   console.hint ("make changes, then imp commit")