refactor

gkwa · Jul 25, 2024 · fe49d5b · fe49d5b
1 parent 209490c
commit fe49d5b
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 16 deletions.
diff --git a/core/cleanup_links.go b/core/cleanup_links.go
@@ -1,6 +1,7 @@
 package core
 
 import (
+	"bytes"
 	"fmt"
 	"io"
 	"regexp"
@@ -46,3 +47,19 @@ func RemoveTitlesFromMarkdownLinks(r io.Reader, w io.Writer) error {
 
 	return nil
 }
+
+func ApplyTransforms(content []byte, transforms ...func(io.Reader, io.Writer) error) ([]byte, error) {
+	var processedContent bytes.Buffer
+	currentContent := bytes.NewReader(content)
+
+	for _, transform := range transforms {
+		processedContent.Reset()
+		err := transform(currentContent, &processedContent)
+		if err != nil {
+			return nil, err
+		}
+		currentContent = bytes.NewReader(processedContent.Bytes())
+	}
+
+	return processedContent.Bytes(), nil
+}
diff --git a/core/file.go b/core/file.go
@@ -40,22 +40,6 @@ func ProcessFile(ctx context.Context, path string, transforms ...func(io.Reader,
 	return nil
 }
 
-func ApplyTransforms(content []byte, transforms ...func(io.Reader, io.Writer) error) ([]byte, error) {
-	var processedContent bytes.Buffer
-	currentContent := bytes.NewReader(content)
-
-	for _, transform := range transforms {
-		processedContent.Reset()
-		err := transform(currentContent, &processedContent)
-		if err != nil {
-			return nil, err
-		}
-		currentContent = bytes.NewReader(processedContent.Bytes())
-	}
-
-	return processedContent.Bytes(), nil
-}
-
 func ReadPathsFromStdin() ([]string, error) {
 	var paths []string
 	scanner := bufio.NewScanner(os.Stdin)