mirror of
https://github.com/cheat/cheat.git
synced 2024-11-24 23:11:36 +01:00
307 lines
12 KiB
Go
307 lines
12 KiB
Go
/*
|
|
* Copyright (c) 2013-2016 Dave Collins <dave@davec.name>
|
|
*
|
|
* Permission to use, copy, modify, and distribute this software for any
|
|
* purpose with or without fee is hereby granted, provided that the above
|
|
* copyright notice and this permission notice appear in all copies.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
|
|
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
|
|
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
|
|
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
|
|
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
|
|
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
|
|
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
|
|
*/
|
|
|
|
package spew
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
)
|
|
|
|
// ConfigState houses the configuration options used by spew to format and
|
|
// display values. There is a global instance, Config, that is used to control
|
|
// all top-level Formatter and Dump functionality. Each ConfigState instance
|
|
// provides methods equivalent to the top-level functions.
|
|
//
|
|
// The zero value for ConfigState provides no indentation. You would typically
|
|
// want to set it to a space or a tab.
|
|
//
|
|
// Alternatively, you can use NewDefaultConfig to get a ConfigState instance
|
|
// with default settings. See the documentation of NewDefaultConfig for default
|
|
// values.
|
|
type ConfigState struct {
|
|
// Indent specifies the string to use for each indentation level. The
|
|
// global config instance that all top-level functions use set this to a
|
|
// single space by default. If you would like more indentation, you might
|
|
// set this to a tab with "\t" or perhaps two spaces with " ".
|
|
Indent string
|
|
|
|
// MaxDepth controls the maximum number of levels to descend into nested
|
|
// data structures. The default, 0, means there is no limit.
|
|
//
|
|
// NOTE: Circular data structures are properly detected, so it is not
|
|
// necessary to set this value unless you specifically want to limit deeply
|
|
// nested data structures.
|
|
MaxDepth int
|
|
|
|
// DisableMethods specifies whether or not error and Stringer interfaces are
|
|
// invoked for types that implement them.
|
|
DisableMethods bool
|
|
|
|
// DisablePointerMethods specifies whether or not to check for and invoke
|
|
// error and Stringer interfaces on types which only accept a pointer
|
|
// receiver when the current type is not a pointer.
|
|
//
|
|
// NOTE: This might be an unsafe action since calling one of these methods
|
|
// with a pointer receiver could technically mutate the value, however,
|
|
// in practice, types which choose to satisify an error or Stringer
|
|
// interface with a pointer receiver should not be mutating their state
|
|
// inside these interface methods. As a result, this option relies on
|
|
// access to the unsafe package, so it will not have any effect when
|
|
// running in environments without access to the unsafe package such as
|
|
// Google App Engine or with the "safe" build tag specified.
|
|
DisablePointerMethods bool
|
|
|
|
// DisablePointerAddresses specifies whether to disable the printing of
|
|
// pointer addresses. This is useful when diffing data structures in tests.
|
|
DisablePointerAddresses bool
|
|
|
|
// DisableCapacities specifies whether to disable the printing of capacities
|
|
// for arrays, slices, maps and channels. This is useful when diffing
|
|
// data structures in tests.
|
|
DisableCapacities bool
|
|
|
|
// ContinueOnMethod specifies whether or not recursion should continue once
|
|
// a custom error or Stringer interface is invoked. The default, false,
|
|
// means it will print the results of invoking the custom error or Stringer
|
|
// interface and return immediately instead of continuing to recurse into
|
|
// the internals of the data type.
|
|
//
|
|
// NOTE: This flag does not have any effect if method invocation is disabled
|
|
// via the DisableMethods or DisablePointerMethods options.
|
|
ContinueOnMethod bool
|
|
|
|
// SortKeys specifies map keys should be sorted before being printed. Use
|
|
// this to have a more deterministic, diffable output. Note that only
|
|
// native types (bool, int, uint, floats, uintptr and string) and types
|
|
// that support the error or Stringer interfaces (if methods are
|
|
// enabled) are supported, with other types sorted according to the
|
|
// reflect.Value.String() output which guarantees display stability.
|
|
SortKeys bool
|
|
|
|
// SpewKeys specifies that, as a last resort attempt, map keys should
|
|
// be spewed to strings and sorted by those strings. This is only
|
|
// considered if SortKeys is true.
|
|
SpewKeys bool
|
|
}
|
|
|
|
// Config is the active configuration of the top-level functions.
|
|
// The configuration can be changed by modifying the contents of spew.Config.
|
|
var Config = ConfigState{Indent: " "}
|
|
|
|
// Errorf is a wrapper for fmt.Errorf that treats each argument as if it were
|
|
// passed with a Formatter interface returned by c.NewFormatter. It returns
|
|
// the formatted string as a value that satisfies error. See NewFormatter
|
|
// for formatting details.
|
|
//
|
|
// This function is shorthand for the following syntax:
|
|
//
|
|
// fmt.Errorf(format, c.NewFormatter(a), c.NewFormatter(b))
|
|
func (c *ConfigState) Errorf(format string, a ...interface{}) (err error) {
|
|
return fmt.Errorf(format, c.convertArgs(a)...)
|
|
}
|
|
|
|
// Fprint is a wrapper for fmt.Fprint that treats each argument as if it were
|
|
// passed with a Formatter interface returned by c.NewFormatter. It returns
|
|
// the number of bytes written and any write error encountered. See
|
|
// NewFormatter for formatting details.
|
|
//
|
|
// This function is shorthand for the following syntax:
|
|
//
|
|
// fmt.Fprint(w, c.NewFormatter(a), c.NewFormatter(b))
|
|
func (c *ConfigState) Fprint(w io.Writer, a ...interface{}) (n int, err error) {
|
|
return fmt.Fprint(w, c.convertArgs(a)...)
|
|
}
|
|
|
|
// Fprintf is a wrapper for fmt.Fprintf that treats each argument as if it were
|
|
// passed with a Formatter interface returned by c.NewFormatter. It returns
|
|
// the number of bytes written and any write error encountered. See
|
|
// NewFormatter for formatting details.
|
|
//
|
|
// This function is shorthand for the following syntax:
|
|
//
|
|
// fmt.Fprintf(w, format, c.NewFormatter(a), c.NewFormatter(b))
|
|
func (c *ConfigState) Fprintf(w io.Writer, format string, a ...interface{}) (n int, err error) {
|
|
return fmt.Fprintf(w, format, c.convertArgs(a)...)
|
|
}
|
|
|
|
// Fprintln is a wrapper for fmt.Fprintln that treats each argument as if it
|
|
// passed with a Formatter interface returned by c.NewFormatter. See
|
|
// NewFormatter for formatting details.
|
|
//
|
|
// This function is shorthand for the following syntax:
|
|
//
|
|
// fmt.Fprintln(w, c.NewFormatter(a), c.NewFormatter(b))
|
|
func (c *ConfigState) Fprintln(w io.Writer, a ...interface{}) (n int, err error) {
|
|
return fmt.Fprintln(w, c.convertArgs(a)...)
|
|
}
|
|
|
|
// Print is a wrapper for fmt.Print that treats each argument as if it were
|
|
// passed with a Formatter interface returned by c.NewFormatter. It returns
|
|
// the number of bytes written and any write error encountered. See
|
|
// NewFormatter for formatting details.
|
|
//
|
|
// This function is shorthand for the following syntax:
|
|
//
|
|
// fmt.Print(c.NewFormatter(a), c.NewFormatter(b))
|
|
func (c *ConfigState) Print(a ...interface{}) (n int, err error) {
|
|
return fmt.Print(c.convertArgs(a)...)
|
|
}
|
|
|
|
// Printf is a wrapper for fmt.Printf that treats each argument as if it were
|
|
// passed with a Formatter interface returned by c.NewFormatter. It returns
|
|
// the number of bytes written and any write error encountered. See
|
|
// NewFormatter for formatting details.
|
|
//
|
|
// This function is shorthand for the following syntax:
|
|
//
|
|
// fmt.Printf(format, c.NewFormatter(a), c.NewFormatter(b))
|
|
func (c *ConfigState) Printf(format string, a ...interface{}) (n int, err error) {
|
|
return fmt.Printf(format, c.convertArgs(a)...)
|
|
}
|
|
|
|
// Println is a wrapper for fmt.Println that treats each argument as if it were
|
|
// passed with a Formatter interface returned by c.NewFormatter. It returns
|
|
// the number of bytes written and any write error encountered. See
|
|
// NewFormatter for formatting details.
|
|
//
|
|
// This function is shorthand for the following syntax:
|
|
//
|
|
// fmt.Println(c.NewFormatter(a), c.NewFormatter(b))
|
|
func (c *ConfigState) Println(a ...interface{}) (n int, err error) {
|
|
return fmt.Println(c.convertArgs(a)...)
|
|
}
|
|
|
|
// Sprint is a wrapper for fmt.Sprint that treats each argument as if it were
|
|
// passed with a Formatter interface returned by c.NewFormatter. It returns
|
|
// the resulting string. See NewFormatter for formatting details.
|
|
//
|
|
// This function is shorthand for the following syntax:
|
|
//
|
|
// fmt.Sprint(c.NewFormatter(a), c.NewFormatter(b))
|
|
func (c *ConfigState) Sprint(a ...interface{}) string {
|
|
return fmt.Sprint(c.convertArgs(a)...)
|
|
}
|
|
|
|
// Sprintf is a wrapper for fmt.Sprintf that treats each argument as if it were
|
|
// passed with a Formatter interface returned by c.NewFormatter. It returns
|
|
// the resulting string. See NewFormatter for formatting details.
|
|
//
|
|
// This function is shorthand for the following syntax:
|
|
//
|
|
// fmt.Sprintf(format, c.NewFormatter(a), c.NewFormatter(b))
|
|
func (c *ConfigState) Sprintf(format string, a ...interface{}) string {
|
|
return fmt.Sprintf(format, c.convertArgs(a)...)
|
|
}
|
|
|
|
// Sprintln is a wrapper for fmt.Sprintln that treats each argument as if it
|
|
// were passed with a Formatter interface returned by c.NewFormatter. It
|
|
// returns the resulting string. See NewFormatter for formatting details.
|
|
//
|
|
// This function is shorthand for the following syntax:
|
|
//
|
|
// fmt.Sprintln(c.NewFormatter(a), c.NewFormatter(b))
|
|
func (c *ConfigState) Sprintln(a ...interface{}) string {
|
|
return fmt.Sprintln(c.convertArgs(a)...)
|
|
}
|
|
|
|
/*
|
|
NewFormatter returns a custom formatter that satisfies the fmt.Formatter
|
|
interface. As a result, it integrates cleanly with standard fmt package
|
|
printing functions. The formatter is useful for inline printing of smaller data
|
|
types similar to the standard %v format specifier.
|
|
|
|
The custom formatter only responds to the %v (most compact), %+v (adds pointer
|
|
addresses), %#v (adds types), and %#+v (adds types and pointer addresses) verb
|
|
combinations. Any other verbs such as %x and %q will be sent to the the
|
|
standard fmt package for formatting. In addition, the custom formatter ignores
|
|
the width and precision arguments (however they will still work on the format
|
|
specifiers not handled by the custom formatter).
|
|
|
|
Typically this function shouldn't be called directly. It is much easier to make
|
|
use of the custom formatter by calling one of the convenience functions such as
|
|
c.Printf, c.Println, or c.Printf.
|
|
*/
|
|
func (c *ConfigState) NewFormatter(v interface{}) fmt.Formatter {
|
|
return newFormatter(c, v)
|
|
}
|
|
|
|
// Fdump formats and displays the passed arguments to io.Writer w. It formats
|
|
// exactly the same as Dump.
|
|
func (c *ConfigState) Fdump(w io.Writer, a ...interface{}) {
|
|
fdump(c, w, a...)
|
|
}
|
|
|
|
/*
|
|
Dump displays the passed parameters to standard out with newlines, customizable
|
|
indentation, and additional debug information such as complete types and all
|
|
pointer addresses used to indirect to the final value. It provides the
|
|
following features over the built-in printing facilities provided by the fmt
|
|
package:
|
|
|
|
* Pointers are dereferenced and followed
|
|
* Circular data structures are detected and handled properly
|
|
* Custom Stringer/error interfaces are optionally invoked, including
|
|
on unexported types
|
|
* Custom types which only implement the Stringer/error interfaces via
|
|
a pointer receiver are optionally invoked when passing non-pointer
|
|
variables
|
|
* Byte arrays and slices are dumped like the hexdump -C command which
|
|
includes offsets, byte values in hex, and ASCII output
|
|
|
|
The configuration options are controlled by modifying the public members
|
|
of c. See ConfigState for options documentation.
|
|
|
|
See Fdump if you would prefer dumping to an arbitrary io.Writer or Sdump to
|
|
get the formatted result as a string.
|
|
*/
|
|
func (c *ConfigState) Dump(a ...interface{}) {
|
|
fdump(c, os.Stdout, a...)
|
|
}
|
|
|
|
// Sdump returns a string with the passed arguments formatted exactly the same
|
|
// as Dump.
|
|
func (c *ConfigState) Sdump(a ...interface{}) string {
|
|
var buf bytes.Buffer
|
|
fdump(c, &buf, a...)
|
|
return buf.String()
|
|
}
|
|
|
|
// convertArgs accepts a slice of arguments and returns a slice of the same
|
|
// length with each argument converted to a spew Formatter interface using
|
|
// the ConfigState associated with s.
|
|
func (c *ConfigState) convertArgs(args []interface{}) (formatters []interface{}) {
|
|
formatters = make([]interface{}, len(args))
|
|
for index, arg := range args {
|
|
formatters[index] = newFormatter(c, arg)
|
|
}
|
|
return formatters
|
|
}
|
|
|
|
// NewDefaultConfig returns a ConfigState with the following default settings.
|
|
//
|
|
// Indent: " "
|
|
// MaxDepth: 0
|
|
// DisableMethods: false
|
|
// DisablePointerMethods: false
|
|
// ContinueOnMethod: false
|
|
// SortKeys: false
|
|
func NewDefaultConfig() *ConfigState {
|
|
return &ConfigState{Indent: " "}
|
|
}
|