go/lazygit
1
0
Fork 0
mirror of https://github.com/jesseduffield/lazygit.git synced 2025-04-04 22:34:39 +02:00
Code Issues Releases Activity

extract out function

Browse Source
This commit is contained in:
Jesse Duffield 2023-04-02 11:37:02 +10:00
parent 3422b1e218
commit 711be78811
1 changed files with 20 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
pkg/app/daemon/daemon.go
View File

@ -79,25 +79,11 @@ type rebaseDaemon struct {
func (self *rebaseDaemon) Run() error { func (self *rebaseDaemon) Run() error {
self.c.Log.Info("Lazygit invoked as interactive rebase demon") self.c.Log.Info("Lazygit invoked as interactive rebase demon")
self.c.Log.Info("args: ", os.Args) self.c.Log.Info("args: ", os.Args)
filePath := os.Args[1] path := os.Args[1]
if strings.HasSuffix(filePath, "git-rebase-todo") { if strings.HasSuffix(path, "git-rebase-todo") {
todoEnvKey := os.Getenv(RebaseTODOEnvKey) return self.writeTodoFile(path)
} else if strings.HasSuffix(path, filepath.Join(gitDir(), "COMMIT_EDITMSG")) { // TODO: test
var todoContent []byte
if v := os.Getenv(PrependLinesEnvKey); v != "" {
existingContent, err := os.ReadFile(filePath)
if err != nil {
return err
}
todoContent = append([]byte(todoEnvKey), existingContent...)
} else {
todoContent = []byte(todoEnvKey)
}
return os.WriteFile(filePath, todoContent, 0o644)
} else if strings.HasSuffix(filePath, filepath.Join(gitDir(), "COMMIT_EDITMSG")) { // TODO: test
// if we are rebasing and squashing, we'll see a COMMIT_EDITMSG // if we are rebasing and squashing, we'll see a COMMIT_EDITMSG
// but in this case we don't need to edit it, so we'll just return // but in this case we don't need to edit it, so we'll just return
} else { } else {
@ -107,6 +93,22 @@ func (self *rebaseDaemon) Run() error {
return nil return nil
} }
func (self *rebaseDaemon) writeTodoFile(path string) error {
todoContent := []byte(os.Getenv(RebaseTODOEnvKey))
prependLines := os.Getenv(PrependLinesEnvKey) != ""
if prependLines {
existingContent, err := os.ReadFile(path)
if err != nil {
return err
}
todoContent = append(todoContent, existingContent...)
}
return os.WriteFile(path, todoContent, 0o644)
}
func gitDir() string { func gitDir() string {
dir := env.GetGitDirEnv() dir := env.GetGitDirEnv()
if dir == "" { if dir == "" {