Breaking into different files

gui.go

@@ -7,37 +7,29 @@
 package main
 
 import (
-  "fmt"
-  "strings"
+
   // "io"
   // "io/ioutil"
+
   "log"
   // "strings"
-  "os"
 
-  "github.com/fatih/color"
   "github.com/jroimartin/gocui"
 )
 
 type stateType struct {
-  GitFiles []GitFile
-  Branches []Branch
+  GitFiles     []GitFile
+  Branches     []Branch
+  PreviousView string
 }
 
-var state = stateType{GitFiles: make([]GitFile, 0)}
+var state = stateType{
+  GitFiles:     make([]GitFile, 0),
+  PreviousView: "files",
+}
 
 var cyclableViews = []string{"files", "branches"}
 
-func stagedFiles(files []GitFile) []GitFile {
-  result := make([]GitFile, 0)
-  for _, file := range files {
-    if file.HasStagedChanges {
-      result = append(result, file)
-    }
-  }
-  return result
-}
-
 func nextView(g *gocui.Gui, v *gocui.View) error {
   var focusedViewName string
   if v == nil || v.Name() == cyclableViews[len(cyclableViews)-1] {
@@ -61,86 +53,7 @@ func nextView(g *gocui.Gui, v *gocui.View) error {
   return switchFocus(g, v, focusedView)
 }
 
-func switchFocus(g *gocui.Gui, oldView, newView *gocui.View) error {
-  if oldView != nil {
-    oldView.Highlight = false
-  }
-  newView.Highlight = true
-  devLog(newView.Name())
-  _, err := g.SetCurrentView(newView.Name()) // not mega proud of the delayed
-  // return of err
-  itemSelected(g, newView)
-  showViewOptions(g, newView.Name())
-  return err
-}
-
-func showViewOptions(g *gocui.Gui, viewName string) error {
-  optionsMap := map[string]string{
-    "files":    "space: toggle staged, c: commit changes, shift+d: remove",
-    "branches": "space: checkout",
-    "prompt":   "esc: cancel, enter: commit",
-  }
-  g.Update(func(*gocui.Gui) error {
-    v, err := g.View("options")
-    if err != nil {
-      panic(err)
-    }
-    v.Clear()
-    fmt.Fprint(v, optionsMap[viewName])
-    return nil
-  })
-  return nil
-}
-
-func getItemPosition(v *gocui.View) int {
-  _, cy := v.Cursor()
-  _, oy := v.Origin()
-  return oy + cy
-}
-
-func cursorUp(g *gocui.Gui, v *gocui.View) error {
-  if v == nil {
-    return nil
-  }
-
-  ox, oy := v.Origin()
-  cx, cy := v.Cursor()
-  if err := v.SetCursor(cx, cy-1); err != nil && oy > 0 {
-    if err := v.SetOrigin(ox, oy-1); err != nil {
-      return err
-    }
-  }
-
-  itemSelected(g, v)
-  return nil
-}
-
-func resetOrigin(v *gocui.View) error {
-  if err := v.SetCursor(0, 0); err != nil {
-    return err
-  }
-  return v.SetOrigin(0, 0)
-}
-
-func cursorDown(g *gocui.Gui, v *gocui.View) error {
-  if v != nil {
-    cx, cy := v.Cursor()
-    ox, oy := v.Origin()
-    if cy+oy >= len(v.BufferLines())-2 {
-      return nil
-    }
-    if err := v.SetCursor(cx, cy+1); err != nil {
-      if err := v.SetOrigin(ox, oy+1); err != nil {
-        return err
-      }
-    }
-  }
-
-  itemSelected(g, v)
-  return nil
-}
-
-func itemSelected(g *gocui.Gui, v *gocui.View) error {
+func newLineFocused(g *gocui.Gui, v *gocui.View) error {
   mainView, _ := g.View("main")
   mainView.SetOrigin(0, 0)
 
@@ -149,14 +62,18 @@ func itemSelected(g *gocui.Gui, v *gocui.View) error {
     return handleFileSelect(g, v)
   case "branches":
     return handleBranchSelect(g, v)
-  case "prompt":
+  case "commit":
+    return handleCommitPromptFocus(g, v)
+  case "confirmation":
+    return nil
+  case "main":
     return nil
   default:
-    panic("No view matching itemSelected switch statement")
+    panic("No view matching newLineFocused switch statement")
   }
 }
 
-func scrollUp(g *gocui.Gui, v *gocui.View) error {
+func scrollUpMain(g *gocui.Gui, v *gocui.View) error {
   mainView, _ := g.View("main")
   ox, oy := mainView.Origin()
   if oy >= 1 {
@@ -165,7 +82,7 @@ func scrollUp(g *gocui.Gui, v *gocui.View) error {
   return nil
 }
 
-func scrollDown(g *gocui.Gui, v *gocui.View) error {
+func scrollDownMain(g *gocui.Gui, v *gocui.View) error {
   mainView, _ := g.View("main")
   ox, oy := mainView.Origin()
   if oy < len(mainView.BufferLines()) {
@@ -174,144 +91,6 @@ func scrollDown(g *gocui.Gui, v *gocui.View) error {
   return nil
 }
 
-func devLog(s string) {
-  f, _ := os.OpenFile("/Users/jesseduffieldduffield/go/src/github.com/jesseduffield/gitgot/development.log", os.O_APPEND|os.O_WRONLY, 0644)
-  defer f.Close()
-
-  f.WriteString(s + "\n")
-}
-
-func handleBranchPress(g *gocui.Gui, v *gocui.View) error {
-  branch := getSelectedBranch(v)
-  if err := gitCheckout(branch.Name, false); err != nil {
-    panic(err)
-  }
-  refreshBranches(v)
-  refreshFiles(g)
-  return nil
-}
-
-func handleFilePress(g *gocui.Gui, v *gocui.View) error {
-  file := getSelectedFile(v)
-
-  if file.HasUnstagedChanges {
-    stageFile(file.Name)
-  } else {
-    unStageFile(file.Name)
-  }
-
-  if err := refreshFiles(g); err != nil {
-    return err
-  }
-  if err := handleFileSelect(g, v); err != nil {
-    return err
-  }
-
-  return nil
-}
-
-func handleCommitPrompt(g *gocui.Gui, currentView *gocui.View) error {
-  devLog(fmt.Sprint(stagedFiles(state.GitFiles)))
-  if len(stagedFiles(state.GitFiles)) == 0 {
-    return nil
-  }
-  maxX, maxY := g.Size()
-  // var v *gocui.View
-  if v, err := g.SetView("prompt", maxX/2-30, maxY/2-1, maxX/2+30, maxY/2+1); err != nil {
-    if err != gocui.ErrUnknownView {
-      return err
-    }
-    v.Title = "Commit Message: "
-    v.Editable = true
-    v.Highlight = true
-    v.Autoscroll = true
-    v.Wrap = true
-    v.Overwrite = true
-    v.Caret = true
-    // fmt.Fprintln(v, "commit message: ")
-    if _, err := g.SetCurrentView("prompt"); err != nil {
-      return err
-    }
-    switchFocus(g, currentView, v)
-  }
-  return nil
-}
-
-func handleCommitSubmit(g *gocui.Gui, v *gocui.View) error {
-  if len(v.BufferLines()) == 0 {
-    return closePrompt(g, v)
-  }
-  message := fmt.Sprint(v.BufferLines()[0])
-  // for whatever reason, a successful commit returns an error, so we're not
-  // going to check for an error here
-  if err := gitCommit(message); err != nil {
-    devLog(fmt.Sprint(err))
-    panic(err)
-  }
-  refreshFiles(g)
-  return closePrompt(g, v)
-}
-
-func handleFileRemove(g *gocui.Gui, v *gocui.View) error {
-  file := getSelectedFile(v)
-  removeFile(file)
-  refreshFiles(g)
-  return nil
-}
-
-func getSelectedFile(v *gocui.View) GitFile {
-  lineNumber := getItemPosition(v)
-  if len(state.GitFiles) == 0 {
-    return GitFile{
-      Name:               "noFile",
-      DisplayString:      "none",
-      HasStagedChanges:   false,
-      HasUnstagedChanges: false,
-      Tracked:            false,
-      Deleted:            false,
-    }
-  }
-  return state.GitFiles[lineNumber]
-}
-
-func getSelectedBranch(v *gocui.View) Branch {
-  lineNumber := getItemPosition(v)
-  return state.Branches[lineNumber]
-}
-
-func handleBranchSelect(g *gocui.Gui, v *gocui.View) error {
-  lineNumber := getItemPosition(v)
-  branch := state.Branches[lineNumber]
-  diff, _ := getBranchDiff(branch.Name, branch.BaseBranch)
-  if err := renderString(g, diff); err != nil {
-    return err
-  }
-  return nil
-}
-
-func handleFileSelect(g *gocui.Gui, v *gocui.View) error {
-  item := getSelectedFile(v)
-  diff := getDiff(item)
-  return renderString(g, diff)
-}
-
-func closePrompt(g *gocui.Gui, v *gocui.View) error {
-  filesView, _ := g.View("files")
-  switchFocus(g, v, filesView)
-  devLog("test prompt close")
-  if err := g.DeleteView("prompt"); err != nil {
-    return err
-  }
-  if _, err := g.SetCurrentView("files"); err != nil {
-    return err
-  }
-  return nil
-}
-
-func quit(g *gocui.Gui, v *gocui.View) error {
-  return gocui.ErrQuit
-}
-
 func keybindings(g *gocui.Gui) error {
   if err := g.SetKeybinding("", gocui.KeyTab, gocui.ModNone, nextView); err != nil {
     return err
@@ -328,25 +107,31 @@ func keybindings(g *gocui.Gui) error {
   if err := g.SetKeybinding("", gocui.KeyArrowUp, gocui.ModNone, cursorUp); err != nil {
     return err
   }
-  if err := g.SetKeybinding("", gocui.KeyPgup, gocui.ModNone, scrollUp); err != nil {
+  if err := g.SetKeybinding("", gocui.KeyPgup, gocui.ModNone, scrollUpMain); err != nil {
     return err
   }
-  if err := g.SetKeybinding("", gocui.KeyPgdn, gocui.ModNone, scrollDown); err != nil {
+  if err := g.SetKeybinding("", gocui.KeyPgdn, gocui.ModNone, scrollDownMain); err != nil {
     return err
   }
-  if err := g.SetKeybinding("", 'C', gocui.ModNone, handleCommitPrompt); err != nil {
+  if err := g.SetKeybinding("", 'ç', gocui.ModNone, handleCommitPress); err != nil {
     return err
   }
   if err := g.SetKeybinding("files", gocui.KeySpace, gocui.ModNone, handleFilePress); err != nil {
     return err
   }
-  if err := g.SetKeybinding("files", 'D', gocui.ModNone, handleFileRemove); err != nil {
+  if err := g.SetKeybinding("files", '®', gocui.ModNone, handleFileRemove); err != nil {
     return err
   }
-  if err := g.SetKeybinding("prompt", gocui.KeyEsc, gocui.ModNone, closePrompt); err != nil {
+  if err := g.SetKeybinding("files", 'ø', gocui.ModNone, handleFileOpen); err != nil {
     return err
   }
-  if err := g.SetKeybinding("prompt", gocui.KeyEnter, gocui.ModNone, handleCommitSubmit); err != nil {
+  if err := g.SetKeybinding("files", 'ß', gocui.ModNone, handleSublimeFileOpen); err != nil {
+    return err
+  }
+  if err := g.SetKeybinding("commit", gocui.KeyEsc, gocui.ModNone, closeCommitPrompt); err != nil {
+    return err
+  }
+  if err := g.SetKeybinding("commit", gocui.KeyEnter, gocui.ModNone, handleCommitSubmit); err != nil {
     return err
   }
   if err := g.SetKeybinding("branches", gocui.KeySpace, gocui.ModNone, handleBranchPress); err != nil {
@@ -355,94 +140,16 @@ func keybindings(g *gocui.Gui) error {
   return nil
 }
 
-func splitLines(multilineString string) []string {
-  if multilineString == "" || multilineString == "\n" {
-    return make([]string, 0)
-  }
-  lines := strings.Split(multilineString, "\n")
-  if lines[len(lines)-1] == "" {
-    return lines[:len(lines)-1]
-  }
-  return lines
-}
-
-func refreshBranches(v *gocui.View) error {
-  state.Branches = getGitBranches()
-  yellow := color.New(color.FgYellow)
-  red := color.New(color.FgRed)
-  white := color.New(color.FgWhite)
-  green := color.New(color.FgGreen)
-
-  v.Clear()
-  for _, branch := range state.Branches {
-    if branch.Type == "feature" {
-      green.Fprintln(v, branch.DisplayString)
-      continue
-    }
-    if branch.Type == "bugfix" {
-      yellow.Fprintln(v, branch.DisplayString)
-      continue
-    }
-    if branch.Type == "hotfix" {
-      red.Fprintln(v, branch.DisplayString)
-      continue
-    }
-    white.Fprintln(v, branch.DisplayString)
-  }
-  resetOrigin(v)
-  return nil
-}
-
-// if the cursor down past the last item, move it up one
-func correctCursor(v *gocui.View) error {
-  cx, cy := v.Cursor()
-  _, oy := v.Origin()
-  lineCount := len(v.BufferLines()) - 2
-  if cy >= lineCount-oy {
-    return v.SetCursor(cx, lineCount-oy)
-  }
-  return nil
-}
-
-func refreshFiles(g *gocui.Gui) error {
-  filesView, err := g.View("files")
-  if err != nil {
-    return err
-  }
-
-  // get files to stage
-  gitFiles := getGitStatusFiles()
-  state.GitFiles = mergeGitStatusFiles(state.GitFiles, gitFiles)
-
-  filesView.Clear()
-  red := color.New(color.FgRed)
-  green := color.New(color.FgGreen)
-  for _, gitFile := range state.GitFiles {
-    if !gitFile.Tracked {
-      red.Fprintln(filesView, gitFile.DisplayString)
-      continue
-    }
-    green.Fprint(filesView, gitFile.DisplayString[0:1])
-    red.Fprint(filesView, gitFile.DisplayString[1:3])
-    if gitFile.HasUnstagedChanges {
-      red.Fprintln(filesView, gitFile.Name)
-    } else {
-      green.Fprintln(filesView, gitFile.Name)
-    }
-  }
-  correctCursor(filesView)
-  return nil
-}
-
 func layout(g *gocui.Gui) error {
-  maxX, maxY := g.Size()
-  leftSideWidth := maxX / 3
-  filesBranchesBoundary := maxY - 10
+  width, height := g.Size()
+  leftSideWidth := width / 3
+  logsBranchesBoundary := height - 10
+  filesBranchesBoundary := height - 20
 
-  optionsTop := maxY - 3
+  optionsTop := height - 3
   // hiding options if there's not enough space
-  if maxY < 30 {
-    optionsTop = maxY
+  if height < 30 {
+    optionsTop = height
   }
 
   sideView, err := g.SetView("files", 0, 0, leftSideWidth, filesBranchesBoundary-1)
@@ -455,19 +162,27 @@ func layout(g *gocui.Gui) error {
     refreshFiles(g)
   }
 
-  if v, err := g.SetView("main", leftSideWidth+2, 0, maxX-1, optionsTop-1); err != nil {
+  if v, err := g.SetView("main", leftSideWidth+2, 0, width-1, optionsTop-1); err != nil {
     if err != gocui.ErrUnknownView {
       return err
     }
     v.Title = "Diff"
     v.Wrap = true
-    if _, err := g.SetCurrentView("files"); err != nil {
-      return err
-    }
+    switchFocus(g, nil, v)
     handleFileSelect(g, sideView)
   }
 
-  if v, err := g.SetView("branches", 0, filesBranchesBoundary, leftSideWidth, optionsTop-1); err != nil {
+  if v, err := g.SetView("logs", 0, logsBranchesBoundary, leftSideWidth, optionsTop-1); err != nil {
+    if err != gocui.ErrUnknownView {
+      return err
+    }
+    v.Title = "Log"
+
+    // these are only called once
+    refreshLogs(g)
+  }
+
+  if v, err := g.SetView("branches", 0, filesBranchesBoundary, leftSideWidth, logsBranchesBoundary-1); err != nil {
     if err != gocui.ErrUnknownView {
       return err
     }
@@ -478,7 +193,7 @@ func layout(g *gocui.Gui) error {
     nextView(g, nil)
   }
 
-  if v, err := g.SetView("options", 0, optionsTop, maxX-1, optionsTop+2); err != nil {
+  if v, err := g.SetView("options", 0, optionsTop, width-1, optionsTop+2); err != nil {
     if err != gocui.ErrUnknownView {
       return err
     }
@@ -488,20 +203,6 @@ func layout(g *gocui.Gui) error {
   return nil
 }
 
-func renderString(g *gocui.Gui, s string) error {
-  g.Update(func(*gocui.Gui) error {
-    v, err := g.View("main")
-    if err != nil {
-      panic(err)
-    }
-    v.Clear()
-    fmt.Fprint(v, s)
-    v.Wrap = true
-    return nil
-  })
-  return nil
-}
-
 func run() {
   g, err := gocui.NewGui(gocui.OutputNormal)
   if err != nil {
@@ -509,8 +210,6 @@ func run() {
   }
   defer g.Close()
 
-  // g.Cursor = true
-
   g.SetManagerFunc(layout)
 
   if err := keybindings(g); err != nil {
@@ -522,6 +221,10 @@ func run() {
   }
 }
 
+func quit(g *gocui.Gui, v *gocui.View) error {
+  return gocui.ErrQuit
+}
+
 // const mcRide = "
 //                                                     `.-::-`
 //                                                  -/o+oossys+:.