trash.go

package trash

import (
	"encoding/base64"
	"fmt"
	"io"
	"log"
	"os"
	"strings"
	"time"
)

// DefaultLogger is the logger use by default by trash
var DefaultLogger = log.New(os.Stdout, "[TRASH] ", 0)

// Trash defined the Trash data structure
type Trash struct {
	Type   string
	format interface{}
	dump   *Dump
	Logger *log.Logger
}

// New create a new Trash with the provided logger and data format
func New(logger *log.Logger, format string) *Trash {
	t := &Trash{Logger: logger}
	switch strings.ToLower(format) {
	case "json":
		t.Type = "json"
		t.format = t.jsonErr
	case "xml":
		t.Type = "xml"
		t.format = t.xmlErr
	default:
		logger.Printf("[!] %s is a invalid format.\n", format)
		return nil
	}
	return t
}

// NewErr generate a standard Err
func (t *Trash) NewErr(err string, message interface{}) Err {
	er := t.format.(func(string, string) genErr)
	e := er(err, extractMessage(message))
	if t.dump != nil {
		t.dump.errChan <- e
	}
	return e
}

// NewHTTPErr generate a new HTTPErr
func (t *Trash) NewHTTPErr(err string, message interface{}) HTTPErr {
	er := t.format.(func(string, string) genErr)
	e := er(err, extractMessage(message))
	if t.dump != nil {
		t.dump.errChan <- e
	}
	return e
}

func (t *Trash) NewShortErr(err string, message interface{}, rw io.Writer) {
	er := t.format.(func(string, string) genErr)
	e := er(err, extractMessage(message))
	if t.dump != nil {
		t.dump.errChan <- e
	}
	e.Log()
	e.Send(rw)
}

// NewErr generate a new Err
func (t *Trash) jsonErr(err string, message string) genErr {
	checksum := base64.StdEncoding.EncodeToString([]byte(time.Now().String()))
	return JsonErr{Logger: t.Logger, errData: errData{checksum, err, message, 0}}
}

// NewHTTPErr generate a new HTTPErr
func (t *Trash) xmlErr(err string, message string) genErr {
	checksum := base64.StdEncoding.EncodeToString([]byte(time.Now().String()))
	return XmlErr{Logger: t.Logger, errData: errData{checksum, err, message, 0}}
}

func extractMessage(m interface{}) string {
	switch mss := m.(type) {
	case string:
		return mss
	case error:
		return mss.Error()
	case fmt.Stringer:
		return mss.String()
	}
	return fmt.Sprintf("Invalid message type %s", m)
}