dbg.go

// This file was AUTOMATICALLY GENERATED by dbg-import (smuggol) from github.com/extrame/dbg

/*
Package dbg is a println/printf/log-debugging utility library.

	import (
	    Dbg "github.com/extrame/dbg"
	)

	dbg, dbgf := Dbg.New()

	dbg("Emit some debug stuff", []byte{120, 121, 122, 122, 121}, math.Pi)
	# "2013/01/28 16:50:03 Emit some debug stuff [120 121 122 122 121] 3.141592653589793"

	dbgf("With a %s formatting %.2f", "little", math.Pi)
	# "2013/01/28 16:51:55 With a little formatting (3.14)"

	dbgf("%/fatal//A fatal debug statement: should not be here")
	# "A fatal debug statement: should not be here"
	# ...and then, os.Exit(1)

	dbgf("%/panic//Can also panic %s", "this")
	# "Can also panic this"
	# ...as a panic, equivalent to: panic("Can also panic this")

	dbgf("Any %s arguments without a corresponding %%", "extra", "are treated like arguments to dbg()")
	# "2013/01/28 17:14:40 Any extra arguments (without a corresponding %) are treated like arguments to dbg()"

	dbgf("%d %d", 1, 2, 3, 4, 5)
	# "2013/01/28 17:16:32 Another example: 1 2 3 4 5"

	dbgf("%@: Include the function name for a little context (via %s)", "%@")
	# "2013... github.com/extrame/dbg.TestSynopsis: Include the function name for a little context (via %@)"

By default, dbg uses log (log.Println, log.Printf, log.Panic, etc.) for output.
However, you can also provide your own output destination by invoking dbg.New with
a customization function:

	import (
	    "bytes"
	    Dbg "github.com/extrame/dbg"
	    "os"
	)

	# dbg to os.Stderr
	dbg, dbgf := Dbg.New(func(dbgr *Dbgr) {
	    dbgr.SetOutput(os.Stderr)
	})

	# A slightly contrived example:
	var buffer bytes.Buffer
	dbg, dbgf := New(func(dbgr *Dbgr) {
	    dbgr.SetOutput(&buffer)
	})
*/
package dbg

import (
	"bytes"
	"fmt"
	"io"
	"log"
	"os"
	"regexp"
	goruntime "runtime"
	"strings"
	"unicode"
)

type _frmt struct {
	ctl          string
	format       string
	operandCount int
	panic        bool
	fatal        bool
	check        bool
}

var (
	ctlTest = regexp.MustCompile(`^\s*%/`)
	ctlScan = regexp.MustCompile(`%?/(panic|fatal|check)(?:\s|$)`)
)

func operandCount(format string) int {
	count := 0
	end := len(format)
	for at := 0; at < end; {
		for at < end && format[at] != '%' {
			at++
		}
		at++
		if at < end {
			if format[at] != '%' && format[at] != '@' {
				count++
			}
			at++
		}
	}
	return count
}

func parseFormat(format string) (frmt _frmt) {
	if ctlTest.MatchString(format) {
		format = strings.TrimLeftFunc(format, unicode.IsSpace)
		index := strings.Index(format, "//")
		if index != -1 {
			frmt.ctl = format[0:index]
			format = format[index+2:] // Skip the second slash via +2 (instead of +1)
		} else {
			frmt.ctl = format
			format = ""
		}
		for _, tmp := range ctlScan.FindAllStringSubmatch(frmt.ctl, -1) {
			for _, value := range tmp[1:] {
				switch value {
				case "panic":
					frmt.panic = true
				case "fatal":
					frmt.fatal = true
				case "check":
					frmt.check = true
				}
			}
		}
	}
	frmt.format = format
	frmt.operandCount = operandCount(format)
	return
}

type Dbgr struct {
	emit emit
}

type DbgFunction func(values ...interface{})

func NewDbgr() *Dbgr {
	return &Dbgr{}
}

/*
New will create and return a pair of debugging functions. You can customize where
they output to by passing in an (optional) customization function:

	import (
	    Dbg "github.com/extrame/dbg"
	    "os"
	)

	# dbg to os.Stderr
	dbg, dbgf := Dbg.New(func(dbgr *Dbgr) {
	    dbgr.SetOutput(os.Stderr)
	})
*/
func New(options ...interface{}) (dbg DbgFunction, dbgf DbgFunction) {
	dbgr := NewDbgr()
	if len(options) > 0 {
		if fn, ok := options[0].(func(*Dbgr)); ok {
			fn(dbgr)
		}
	}
	return dbgr.DbgDbgf()
}

func (d Dbgr) Dbg(values ...interface{}) {
	d.getEmit().emit(_frmt{}, "", values...)
}

func (d Dbgr) Dbgf(values ...interface{}) {
	d.dbgf(values...)
}

func (d Dbgr) DbgDbgf() (dbg DbgFunction, dbgf DbgFunction) {
	dbg = func(vl ...interface{}) {
		d.Dbg(vl...)
	}
	dbgf = func(vl ...interface{}) {
		d.dbgf(vl...)
	}
	return dbg, dbgf // Redundant, but...
}

func (d Dbgr) dbgf(values ...interface{}) {
	var frmt _frmt
	if len(values) > 0 {
		tmp := fmt.Sprint(values[0])
		frmt = parseFormat(tmp)
		values = values[1:]
	}

	buf := bytes.Buffer{}
	format := frmt.format
	end := len(format)
	for at := 0; at < end; {
		last := at
		for at < end && format[at] != '%' {
			at++
		}
		if at > last {
			buf.WriteString(format[last:at])
		}
		if at >= end {
			break
		}
		// format[at] == '%'
		at++
		// format[at] == ?
		if format[at] == '@' {
			depth := 2
			pc, _, _, _ := goruntime.Caller(depth)
			name := goruntime.FuncForPC(pc).Name()
			buf.WriteString(name)
		} else {
			buf.WriteString(format[at-1 : at+1])
		}
		at++
	}

	//valuesF := append([]interface{}{}, values[0:frmt.operandCount]...)
	valuesF := values[0:frmt.operandCount]
	valuesDbg := values[frmt.operandCount:]
	if len(valuesDbg) > 0 {
		// Adjust frmt.format:
		// (%v instead of %s because: frmt.check)
		tmp := format
		if len(tmp) > 0 {
			if unicode.IsSpace(rune(tmp[len(tmp)-1])) {
				buf.WriteString("%v")
			} else {
				buf.WriteString(" %v")
			}
		} else if frmt.check {
			// Performing a check, so no output
		} else {
			buf.WriteString("%v")
		}

		// Adjust valuesF:
		if !frmt.check {
			tmp := []string{}
			for _, value := range valuesDbg {
				tmp = append(tmp, fmt.Sprintf("%v", value))
			}
			// First, make a copy of valuesF, so we avoid overwriting valuesDbg when appending
			valuesF = append([]interface{}{}, valuesF...)
			valuesF = append(valuesF, strings.Join(tmp, " "))
		}
	}

	format = buf.String()
	if frmt.check {
		// We do not actually emit to the log, but panic if
		// a non-nil value is detected (e.g. a non-nil error)
		for _, value := range valuesDbg {
			if value != nil {
				if format == "" {
					panic(value)
				} else {
					panic(fmt.Sprintf(format, append(valuesF, value)...))
				}
			}
		}
	} else {
		d.getEmit().emit(frmt, format, valuesF...)
	}
}

// Idiot-proof &Dbgr{}, etc.
func (d *Dbgr) getEmit() emit {
	if d.emit == nil {
		d.emit = standardEmit()
	}
	return d.emit
}

// SetOutput will accept the following as a destination for output:
//
//	*log.Logger         Print*/Panic*/Fatal* of the logger
//	io.Writer           -
//	nil                 Reset to the default output (os.Stderr)
//	"log"               Print*/Panic*/Fatal* via the "log" package
func (d *Dbgr) SetOutput(output interface{}) {
	if output == nil {
		d.emit = standardEmit()
		return
	}
	switch output := output.(type) {
	case *log.Logger:
		d.emit = emitLogger{
			logger: output,
		}
		return
	case io.Writer:
		d.emit = emitWriter{
			writer: output,
		}
		return
	case string:
		if output == "log" {
			d.emit = emitLog{}
			return
		}
	}
	panic(output)
}

// ======== //
// = emit = //
// ======== //

func standardEmit() emit {
	return emitWriter{
		writer: os.Stderr,
	}
}

func ln(tmp string) string {
	length := len(tmp)
	if length > 0 && tmp[length-1] != '\n' {
		return tmp + "\n"
	}
	return tmp
}

type emit interface {
	emit(_frmt, string, ...interface{})
}

type emitWriter struct {
	writer io.Writer
}

func (ew emitWriter) emit(frmt _frmt, format string, values ...interface{}) {
	if format == "" {
		fmt.Fprintln(ew.writer, values...)
	} else {
		if frmt.panic {
			panic(fmt.Sprintf(format, values...))
		}
		fmt.Fprintf(ew.writer, ln(format), values...)
		if frmt.fatal {
			os.Exit(1)
		}
	}
}

type emitLogger struct {
	logger *log.Logger
}

func (el emitLogger) emit(frmt _frmt, format string, values ...interface{}) {
	if format == "" {
		el.logger.Println(values...)
	} else {
		if frmt.panic {
			el.logger.Panicf(format, values...)
		} else if frmt.fatal {
			el.logger.Fatalf(format, values...)
		} else {
			el.logger.Printf(format, values...)
		}
	}
}

type emitLog struct {
}

func (el emitLog) emit(frmt _frmt, format string, values ...interface{}) {
	if format == "" {
		log.Println(values...)
	} else {
		if frmt.panic {
			log.Panicf(format, values...)
		} else if frmt.fatal {
			log.Fatalf(format, values...)
		} else {
			log.Printf(format, values...)
		}
	}
}