From 56bbdf0370297cd5e861354666d876831daad6cb Mon Sep 17 00:00:00 2001
From: yuin <yuin@inforno.net>
Date: Sun, 7 Feb 2021 20:36:39 +0900
Subject: [PATCH] refactoring

---
 parser/code_block.go  | 20 ++++++++++++--------
 parser/fcode_block.go |  9 +--------
 2 files changed, 13 insertions(+), 16 deletions(-)

diff --git a/parser/code_block.go b/parser/code_block.go
index 8781bb0..4b18639 100644
--- a/parser/code_block.go
+++ b/parser/code_block.go
@@ -52,14 +52,7 @@ func (b *codeBlockParser) Continue(node ast.Node, reader text.Reader, pc Context
 
 	// if code block line starts with a tab, keep a tab as it is.
 	if segment.Padding != 0 {
-		offsetWithPadding := reader.LineOffset()
-		sl, ss := reader.Position()
-		reader.SetPosition(sl, text.NewSegment(ss.Start-1, ss.Stop))
-		if offsetWithPadding == reader.LineOffset() {
-			segment.Padding = 0
-			segment.Start--
-		}
-		reader.SetPosition(sl, ss)
+		preserveLeadingTabInCodeBlock(&segment, reader)
 	}
 
 	node.Lines().Append(segment)
@@ -90,3 +83,14 @@ func (b *codeBlockParser) CanInterruptParagraph() bool {
 func (b *codeBlockParser) CanAcceptIndentedLine() bool {
 	return true
 }
+
+func preserveLeadingTabInCodeBlock(segment *text.Segment, reader text.Reader) {
+	offsetWithPadding := reader.LineOffset()
+	sl, ss := reader.Position()
+	reader.SetPosition(sl, text.NewSegment(ss.Start-1, ss.Stop))
+	if offsetWithPadding == reader.LineOffset() {
+		segment.Padding = 0
+		segment.Start--
+	}
+	reader.SetPosition(sl, ss)
+}
diff --git a/parser/fcode_block.go b/parser/fcode_block.go
index 75410d7..4801449 100644
--- a/parser/fcode_block.go
+++ b/parser/fcode_block.go
@@ -73,14 +73,7 @@ func (b *fencedCodeBlockParser) Continue(node ast.Node, reader text.Reader, pc C
 	fdata := pc.Get(fencedCodeBlockInfoKey).(*fenceData)
 	// if code block line starts with a tab, keep a tab as it is.
 	if segment.Padding != 0 {
-		offsetWithPadding := reader.LineOffset()
-		sl, ss := reader.Position()
-		reader.SetPosition(sl, text.NewSegment(ss.Start-1, ss.Stop))
-		if offsetWithPadding == reader.LineOffset() {
-			segment.Padding = 0
-			segment.Start--
-		}
-		reader.SetPosition(sl, ss)
+		preserveLeadingTabInCodeBlock(&segment, reader)
 	}
 	w, pos := util.IndentWidth(line, reader.LineOffset())
 	if w < 4 {