mirror of https://github.com/goccy/go-json.git
113 lines
2.5 KiB
Go
113 lines
2.5 KiB
Go
package encoder
|
|
|
|
import (
|
|
"bytes"
|
|
|
|
"github.com/goccy/go-json/internal/errors"
|
|
)
|
|
|
|
func Indent(dst *bytes.Buffer, src []byte, prefix, indentStr string) error {
|
|
length := int64(len(src))
|
|
indentNum := 0
|
|
indentBytes := []byte(indentStr)
|
|
for cursor := int64(0); cursor < length; cursor++ {
|
|
c := src[cursor]
|
|
switch c {
|
|
case ' ', '\t', '\n', '\r':
|
|
continue
|
|
case '"':
|
|
if err := dst.WriteByte(c); err != nil {
|
|
return err
|
|
}
|
|
for {
|
|
cursor++
|
|
if err := dst.WriteByte(src[cursor]); err != nil {
|
|
return err
|
|
}
|
|
switch src[cursor] {
|
|
case '\\':
|
|
cursor++
|
|
if err := dst.WriteByte(src[cursor]); err != nil {
|
|
return err
|
|
}
|
|
case '"':
|
|
goto LOOP_END
|
|
case '\000':
|
|
return errors.ErrUnexpectedEndOfJSON("string", length)
|
|
}
|
|
}
|
|
case '{':
|
|
if cursor+1 < length && src[cursor+1] == '}' {
|
|
if _, err := dst.Write([]byte{'{', '}'}); err != nil {
|
|
return err
|
|
}
|
|
cursor++
|
|
} else {
|
|
indentNum++
|
|
b := []byte{c, '\n'}
|
|
b = append(b, prefix...)
|
|
b = append(b, bytes.Repeat(indentBytes, indentNum)...)
|
|
if _, err := dst.Write(b); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
case '}':
|
|
indentNum--
|
|
if indentNum < 0 {
|
|
return errors.ErrInvalidCharacter('}', "}", cursor)
|
|
}
|
|
b := []byte{'\n'}
|
|
b = append(b, prefix...)
|
|
b = append(b, bytes.Repeat(indentBytes, indentNum)...)
|
|
b = append(b, c)
|
|
if _, err := dst.Write(b); err != nil {
|
|
return err
|
|
}
|
|
case '[':
|
|
if cursor+1 < length && src[cursor+1] == ']' {
|
|
if _, err := dst.Write([]byte{'[', ']'}); err != nil {
|
|
return err
|
|
}
|
|
cursor++
|
|
} else {
|
|
indentNum++
|
|
b := []byte{c, '\n'}
|
|
b = append(b, prefix...)
|
|
b = append(b, bytes.Repeat(indentBytes, indentNum)...)
|
|
if _, err := dst.Write(b); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
case ']':
|
|
indentNum--
|
|
if indentNum < 0 {
|
|
return errors.ErrInvalidCharacter(']', "]", cursor)
|
|
}
|
|
b := []byte{'\n'}
|
|
b = append(b, prefix...)
|
|
b = append(b, bytes.Repeat(indentBytes, indentNum)...)
|
|
b = append(b, c)
|
|
if _, err := dst.Write(b); err != nil {
|
|
return err
|
|
}
|
|
case ':':
|
|
if _, err := dst.Write([]byte{':', ' '}); err != nil {
|
|
return err
|
|
}
|
|
case ',':
|
|
b := []byte{',', '\n'}
|
|
b = append(b, prefix...)
|
|
b = append(b, bytes.Repeat(indentBytes, indentNum)...)
|
|
if _, err := dst.Write(b); err != nil {
|
|
return err
|
|
}
|
|
default:
|
|
if err := dst.WriteByte(c); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
LOOP_END:
|
|
}
|
|
return nil
|
|
}
|