add fragment in issue

takashi · May 8, 2014 · 79732f6 · 79732f6
1 parent 457d444
commit 79732f6
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 1 deletion.
diff --git a/issue.go b/issue.go
@@ -4,6 +4,7 @@ type Issue struct {
 	Title    string
 	FileName string
 	FilePath string
+	Fragment string
 	Label    string
 	Line     int
 	Md5      string

diff --git a/parser.go b/parser.go
@@ -68,7 +68,27 @@ func ParseIssuesByLang(lang *language.Lang, path string, info os.FileInfo) error
 		one := re.Find([]byte(scanner.Text()))
 		if one != nil {
 			title := strings.TrimPrefix(string(commnentLabelRe.ReplaceAll(one, []byte(""))), " ")
-			issue := &Issue{Title: title, FileName: info.Name(), Label: config.Label, Line: line, FilePath: path}
+			frag := scanner.Text() + "\n"
+
+			// copy scanner instance
+			s := *scanner
+			i := 0
+			// get 10 line under the todo.
+			for s.Scan() {
+				if i > 10 {
+					break
+				}
+				frag += (s.Text() + "\n")
+				i++
+			}
+
+			issue := &Issue{
+				Title:    title,
+				FileName: info.Name(),
+				Label:    config.Label,
+				Line:     line,
+				FilePath: path,
+				Fragment: frag}
 			IssueList = append(IssueList, issue)
 		}
 		line += 1