separate gofumpt code into an importable package

To allow offering a drop-in replacement for goimports.
mvdan · Apr 9, 2019 · 9ef2c07 · 9ef2c07
1 parent bdb9bda
commit 9ef2c07
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/gofmt.go b/gofmt.go
@@ -21,6 +21,8 @@ import (
 	"runtime"
 	"runtime/pprof"
 	"strings"
+
+	"mvdan.cc/gofumpt/internal"
 )
 
 var (
@@ -116,7 +118,7 @@ func processFile(filename string, in io.Reader, out io.Writer, stdin bool) error
 
 	// This is the only gofumpt change on gofmt's codebase, besides changing
 	// the name in the usage text.
-	gofumpt(fileSet, file)
+	internal.Gofumpt(fileSet, file)
 
 	res, err := format(fileSet, file, sourceAdj, indentAdj, src, printer.Config{Mode: printerMode, Tabwidth: tabWidth})
 	if err != nil {

diff --git a/gofumpt.go → internal/gofumpt.go b/gofumpt.go → internal/gofumpt.go
@@ -1,7 +1,7 @@
 // Copyright (c) 2019, Daniel Martí <mvdan@mvdan.cc>
 // See LICENSE for licensing information
 
-package main
+package internal
 
 import (
 	"fmt"
@@ -14,7 +14,7 @@ import (
 	"unicode/utf8"
 )
 
-func gofumpt(fset *token.FileSet, file *ast.File) {
+func Gofumpt(fset *token.FileSet, file *ast.File) {
 	f := &fumpter{
 		fset:    fset,
 		file:    fset.File(file.Pos()),