102 lines
3.1 KiB
Go
102 lines
3.1 KiB
Go
|
// Copyright 2013 Google Inc. All rights reserved.
|
||
|
|
//
|
||
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
|
// you may not use this file except in compliance with the License.
|
||
|
|
// You may obtain a copy of the License at
|
||
|
|
//
|
||
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||
|
|
//
|
||
|
|
// Unless required by applicable law or agreed to in writing, software
|
||
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
|
// See the License for the specific language governing permissions and
|
||
|
|
// limitations under the License.
|
||
|
|
|
||
|
|
package pretty
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"fmt"
|
||
|
|
"io"
|
||
|
|
"reflect"
|
||
|
|
|
||
|
|
"github.com/kylelemons/godebug/diff"
|
||
|
|
)
|
||
|
|
|
||
|
|
// A Config represents optional configuration parameters for formatting.
|
||
|
|
//
|
||
|
|
// Some options, notably ShortList, dramatically increase the overhead
|
||
|
|
// of pretty-printing a value.
|
||
|
|
type Config struct {
|
||
|
|
// Verbosity options
|
||
|
|
Compact bool // One-line output. Overrides Diffable.
|
||
|
|
Diffable bool // Adds extra newlines for more easily diffable output.
|
||
|
|
|
||
|
|
// Field and value options
|
||
|
|
IncludeUnexported bool // Include unexported fields in output
|
||
|
|
PrintStringers bool // Call String on a fmt.Stringer
|
||
|
|
|
||
|
|
// Output transforms
|
||
|
|
ShortList int // Maximum character length for short lists if nonzero.
|
||
|
|
}
|
||
|
|
|
||
|
|
var DefaultConfig = &Config{}
|
||
|
|
|
||
|
|
func (cfg *Config) fprint(buf *bytes.Buffer, vals ...interface{}) {
|
||
|
|
for i, val := range vals {
|
||
|
|
if i > 0 {
|
||
|
|
buf.WriteByte('\n')
|
||
|
|
}
|
||
|
|
cfg.val2node(reflect.ValueOf(val)).WriteTo(buf, "", cfg)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Print writes the default representation of the given values to standard output.
|
||
|
|
func Print(vals ...interface{}) {
|
||
|
|
DefaultConfig.Print(vals...)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Print writes the configured presentation of the given values to standard output.
|
||
|
|
func (cfg *Config) Print(vals ...interface{}) {
|
||
|
|
fmt.Println(cfg.Sprint(vals...))
|
||
|
|
}
|
||
|
|
|
||
|
|
// Sprint returns a string representation of the given value according to the default config.
|
||
|
|
func Sprint(vals ...interface{}) string {
|
||
|
|
return DefaultConfig.Sprint(vals...)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Sprint returns a string representation of the given value according to cfg.
|
||
|
|
func (cfg *Config) Sprint(vals ...interface{}) string {
|
||
|
|
buf := new(bytes.Buffer)
|
||
|
|
cfg.fprint(buf, vals...)
|
||
|
|
return buf.String()
|
||
|
|
}
|
||
|
|
|
||
|
|
// Fprint writes the representation of the given value to the writer according to the default config.
|
||
|
|
func Fprint(w io.Writer, vals ...interface{}) (n int64, err error) {
|
||
|
|
return DefaultConfig.Fprint(w, vals...)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Fprint writes the representation of the given value to the writer according to the cfg.
|
||
|
|
func (cfg *Config) Fprint(w io.Writer, vals ...interface{}) (n int64, err error) {
|
||
|
|
buf := new(bytes.Buffer)
|
||
|
|
cfg.fprint(buf, vals...)
|
||
|
|
return buf.WriteTo(w)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Compare returns a string containing a line-by-line unified diff of the
|
||
|
|
// values in got and want. Compare includes unexported fields.
|
||
|
|
//
|
||
|
|
// Each line in the output is prefixed with '+', '-', or ' ' to indicate if it
|
||
|
|
// should be added to, removed from, or is correct for the "got" value with
|
||
|
|
// respect to the "want" value.
|
||
|
|
func Compare(got, want interface{}) string {
|
||
|
|
diffOpt := &Config{
|
||
|
|
Diffable: true,
|
||
|
|
IncludeUnexported: true,
|
||
|
|
}
|
||
|
|
|
||
|
|
return diff.Diff(diffOpt.Sprint(got), diffOpt.Sprint(want))
|
||
|
|
}
|