From 74b3c953def27d91761cb487c1984dcebd346d29 Mon Sep 17 00:00:00 2001 From: Lunny Xiao Date: Wed, 26 Mar 2014 17:57:13 +0800 Subject: [PATCH] improved diff --- models/git.go | 144 +++++++++++++++++++++++++++++++++++++++++++++++--- 1 file changed, 138 insertions(+), 6 deletions(-) diff --git a/models/git.go b/models/git.go index b0b0dd2f8..5b2a4060b 100644 --- a/models/git.go +++ b/models/git.go @@ -9,8 +9,10 @@ import ( "fmt" "path" "strings" - - "github.com/Unknwon/com" + "io" + "bufio" + "os" + "os/exec" "github.com/gogits/git" ) @@ -226,20 +228,150 @@ func GetCommits(userName, reposName, branchname string) (*list.List, error) { return r.AllCommits() } +const ( + PlainLine = iota + 1 + AddLine + DelLine + SectionLine +) + +const ( + AddFile = iota + 1 + ChangeFile + DelFile +) + +type DiffLine struct { + LeftIdx int + RightIdx int + Type int + Content string +} + +type DiffSection struct { + Name string + Lines []*DiffLine +} + type DiffFile struct { Name string Addition, Deletion int - Type string - Content []string + Type int + Sections []*DiffSection } type Diff struct { - NumFiles int // Number of file has been changed. TotalAddition, TotalDeletion int Files []*DiffFile } +func (diff *Diff) NumFiles() int { + return len(diff.Files) +} + +const diffHead = "diff --git " + +func ParsePatch(reader io.Reader) (*Diff, error) { + scanner := bufio.NewScanner(reader) + var totalAdd, totalDel int + var curFile *DiffFile + var curSection * DiffSection + //var leftLine, rightLine int + diff := &Diff{Files:make([]*DiffFile, 0)} + var i int + for scanner.Scan() { + line := scanner.Text() + fmt.Println(i, line) + i = i + 1 + if line == "" { + continue + } + if line[0] == ' ' { + diffLine := &DiffLine{Type: PlainLine, Content:line} + curSection.Lines = append(curSection.Lines, diffLine) + continue + } else if line[0] == '@' { + ss := strings.Split(line, "@@") + diffLine := &DiffLine{Type: SectionLine, Content:"@@ "+ss[len(ss)-2]} + curSection.Lines = append(curSection.Lines, diffLine) + + + + diffLine = &DiffLine{Type: PlainLine, Content:ss[len(ss)-1]} + curSection.Lines = append(curSection.Lines, diffLine) + continue + } else if line[0] == '+' { + diffLine := &DiffLine{Type: AddLine, Content:line} + curSection.Lines = append(curSection.Lines, diffLine) + continue + } else if line[0] == '-' { + diffLine := &DiffLine{Type: DelLine, Content:line} + curSection.Lines = append(curSection.Lines, diffLine) + continue + } + + if strings.HasPrefix(line, diffHead) { + if curFile != nil { + curFile.Addition, totalAdd = totalAdd, 0 + curFile.Deletion, totalDel = totalDel, 0 + curFile = nil + } + fs := strings.Split(line[len(diffHead):], " ") + a := fs[0] + + curFile = &DiffFile{ + Name:a[strings.Index(a, "/")+1:], + Type: ChangeFile, + Sections:make([]*DiffSection, 0), + } + diff.Files = append(diff.Files, curFile) + scanner.Scan() + scanner.Scan() + if scanner.Text() == "--- /dev/null" { + curFile.Type = AddFile + } + scanner.Scan() + } + } + + return diff, nil +} + func GetDiff(repoPath, commitid string) (*Diff, error) { + repo, err := git.OpenRepository(repoPath) + if err != nil { + return nil, err + } + + commit, err := repo.GetCommit("", commitid) + if err != nil { + return nil, err + } + + if commit.ParentCount() == 0 { + return nil, err + } + + rd, wr := io.Pipe() + go func() { + cmd := exec.Command("git", "diff", commitid, commit.Parent(0).Oid.String()) + cmd.Dir = repoPath + cmd.Stdout = wr + cmd.Stdin = os.Stdin + cmd.Stderr = os.Stderr + cmd.Run() + //if err != nil { + // return nil, err + //} + wr.Close() + }() + + defer rd.Close() + + return ParsePatch(rd) +} + +/*func GetDiff(repoPath, commitid string) (*Diff, error) { stdout, _, err := com.ExecCmdDir(repoPath, "git", "show", commitid) if err != nil { return nil, err @@ -271,4 +403,4 @@ func GetDiff(repoPath, commitid string) (*Diff, error) { diff.Files = append(diff.Files, file) } return diff, nil -} +}*/