46 lines
833 B
Go
46 lines
833 B
Go
package format
|
|
|
|
import (
|
|
"bytes"
|
|
"io"
|
|
"os"
|
|
|
|
"github.com/zeromicro/go-zero/tools/goctl/pkg/parser/api/parser"
|
|
)
|
|
|
|
// File formats the api file.
|
|
func File(filename string) error {
|
|
data, err := os.ReadFile(filename)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
buffer := bytes.NewBuffer(nil)
|
|
if err := Source(data, buffer); err != nil {
|
|
return err
|
|
}
|
|
return os.WriteFile(filename, buffer.Bytes(), 0666)
|
|
}
|
|
|
|
// Source formats the api source.
|
|
func Source(source []byte, w io.Writer) error {
|
|
p := parser.New("", source)
|
|
result := p.Parse()
|
|
if err := p.CheckErrors(); err != nil {
|
|
return err
|
|
}
|
|
|
|
result.Format(w)
|
|
return nil
|
|
}
|
|
|
|
func formatForUnitTest(source []byte, w io.Writer) error {
|
|
p := parser.New("", source)
|
|
result := p.Parse()
|
|
if err := p.CheckErrors(); err != nil {
|
|
return err
|
|
}
|
|
|
|
result.FormatForUnitTest(w)
|
|
return nil
|
|
}
|