Pulse to bolt

Signed-off-by: Vishal Rana <vr@labstack.com>

Author: vishr

bolt/bolt.go

@@ -0,0 +1,119 @@
+package bolt
+
+import (
+	"os"
+
+	"github.com/go-errors/errors"
+	"github.com/go-resty/resty"
+	"github.com/labstack/gommon/log"
+)
+
+type (
+	Bolt struct {
+		Options
+		// mutex  sync.RWMutex
+		client *resty.Client
+		logger *log.Logger
+	}
+
+	Options struct {
+		App    *App
+		Device *Device
+	}
+
+	App struct {
+		Version string `json:"version"`
+	}
+
+	Device struct {
+		Hostname string `json:"hostname"`
+	}
+)
+
+var (
+	b *Bolt
+)
+
+// Severity
+var (
+	SeverityInfo  = Severity("info")
+	SeverityWarn  = Severity("warn")
+	SeverityError = Severity("error")
+)
+
+func Register(apiKey string) {
+	RegisterWithOptions(apiKey, Options{})
+}
+
+func RegisterWithOptions(apiKey string, options Options) {
+	b = &Bolt{
+		client: resty.New().
+			SetHostURL("https://api.labstack.com").
+			SetAuthToken(apiKey).
+			SetHeader("User-Agent", "labstack/bolt"),
+		logger: log.New("bolt"),
+	}
+	b.Options = options
+
+	// Defaults
+	if b.Device == nil {
+		b.Device = new(Device)
+	}
+	if b.Device.Hostname == "" {
+		b.Device.Hostname, _ = os.Hostname()
+	}
+}
+
+func (b *Bolt) dispatch(err *errors.Error, data *Data) {
+	// fmt.Println(err.ErrorStack())
+	event := newEvent(err, data)
+	r, e := b.client.R().
+		SetBody(event).
+		// SetError(err).
+		Post("/bolt")
+	if e != nil {
+		b.logger.Error(err)
+		return
+	}
+	if r.StatusCode() < 200 || r.StatusCode() >= 300 {
+		b.logger.Errorf("Failed to send error report: %s", r.Body())
+	}
+}
+
+func Report(err error) {
+	ReportWithData(err, new(Data))
+}
+
+func ReportWithData(err error, data *Data) {
+	if ee, ok := err.(*errors.Error); ok {
+		b.dispatch(ee, data)
+	} else {
+		b.dispatch(errors.Wrap(err, 1), data)
+	}
+}
+
+func AutoReport() {
+	if err := recover(); err != nil {
+		ReportWithData(errors.Wrap(err, 1), new(Data))
+		panic(err)
+	}
+}
+
+func AutoReportWithData(data *Data) {
+	if err := recover(); err != nil {
+		ReportWithData(errors.Wrap(err, 1), data)
+		panic(err)
+	}
+}
+
+func Recover() {
+	if err := recover(); err != nil {
+		ReportWithData(errors.Wrap(err, 1), new(Data))
+	}
+}
+
+func RecoverWithData(data *Data) {
+	if err := recover(); err != nil {
+		ReportWithData(errors.Wrap(err, 1), data)
+	}
+}

pulse/event.go renamed to bolt/event.go

@@ -1,23 +1,29 @@
-package pulse
+package bolt
 
-import "github.com/go-errors/errors"
+import (
+	"path/filepath"
+
+	"github.com/go-errors/errors"
+	labstack "github.com/labstack/labstack-go"
+)
 
 type (
 	Event struct {
-		*Context
-		App       *App       `json:"app" db:"app"`
-		Device    *Device    `json:"device" db:"device"`
-		Exception *Exception `json:"exception" db:"exception"`
+		*Data
+		App       *App       `json:"app"`
+		Device    *Device    `json:"device"`
+		Exception *Exception `json:"exception"`
 	}
 
-	Context struct {
-		Severity severity `json:"severity" db:"severity"`
-		User     *User    `json:"user" db:"user"`
-		Request  *Request `json:"request" db:"request"`
-		Data     Data     `json:"data" db:"data"`
+	Data struct {
+		Severity Severity     `json:"severity"`
+		User     *User        `json:"user"`
+		Request  *Request     `json:"request"`
+		Tags     []string     `json:"tags"`
+		Extra    labstack.Map `json:"extra"`
 	}
 
-	severity string
+	Severity string
 
 	User struct {
 		ID string `json:"id"`
@@ -26,8 +32,6 @@ type (
 	Request struct {
 	}
 
-	Data map[string]interface{}
-
 	Exception struct {
 		Class      string        `json:"class"`
 		Message    string        `json:"message"`
@@ -43,11 +47,11 @@ type (
 	}
 )
 
-func newEvent(err *errors.Error, ctx *Context) (e *Event) {
+func newEvent(err *errors.Error, data *Data) (e *Event) {
 	e = &Event{
-		Context: ctx,
-		App:     p.App,
-		Device:  p.Device,
+		Data:   data,
+		App:    b.App,
+		Device: b.Device,
 		Exception: &Exception{
 			Class:      err.TypeName(),
 			Message:    err.Error(),
@@ -62,7 +66,7 @@ func newEvent(err *errors.Error, ctx *Context) (e *Event) {
 
 	for i, f := range err.StackFrames() {
 		e.Exception.StackTrace[i] = &StackFrame{
-			File:     f.File,
+			File:     filepath.Base(f.File),
 			Line:     f.LineNumber,
 			Function: f.Name,
 		}

labstack.go

@@ -0,0 +1,5 @@
+package labstack
+
+type (
+	Map map[string]interface{}
+)