Push no remote (#23)

* feat: enhance push functionality with upstream branch setup

---------

Co-authored-by: IanSkelskey <IanSkelskey@users.noreply.github.com>

Author: IanSkelskey

src/main.ts

@@ -14,6 +14,7 @@ import {
     unstageAllFiles,
     getName,
     getEmail,
+    setupUpstreamBranch,
 } from './utils/git';
 import {
     confirmCommitMessage,
@@ -132,7 +133,15 @@ async function executeCommitWorkflow(systemPrompt: string, diff: string) {
     print('success', 'Commit successful.');
 
     if (options.push) {
-        pushChanges();
+        try {
+            pushChanges();
+            print('success', 'Push successful.');
+        } catch (error: any) {
+            if (error.message.includes('The current branch has no upstream branch.')) {
+                setupUpstreamBranch();
+            }
+        }
+
         print('success', 'Push successful.');
     }
 }

src/utils/git.ts

@@ -21,7 +21,18 @@ export function isInGitRepo(): boolean {
 }
 
 export function pushChanges(): void {
-    execSync('git push');
+    try {
+        execSync('git push');
+    } catch (error: any) {
+        if (error.message.includes('fatal: The current branch')) {
+            throw new Error('The current branch has no upstream branch.');
+        }
+    }
+}
+
+export function setupUpstreamBranch() {
+    const branchName = getCurrentBranchName();
+    execSync(`git push --set-upstream origin ${branchName}`);
 }
 
 export function addAllChanges(): void {