From 82dbf653fd1da185e3e4cd6c4d707846704d4d51 Mon Sep 17 00:00:00 2001
From: derekdowling <thederekdowling@gmail.com>
Date: Tue, 4 Nov 2014 09:04:58 -0700
Subject: [PATCH] Adding logtext formatter

---
 logtext.go | 48 ++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 48 insertions(+)
 create mode 100644 logtext.go

diff --git a/logtext.go b/logtext.go
new file mode 100644
index 000000000..0cc9b2c22
--- /dev/null
+++ b/logtext.go
@@ -0,0 +1,48 @@
+package logtext
+
+// A Logrus hook that adds filename/line number/stack trace to our log outputs
+
+import (
+	"github.com/Sirupsen/logrus"
+	"runtime"
+)
+
+// Log depth is how many levels to ascend to find where the actual log call occurred
+// while debugOnly sets whether or not stack traces should be printed outside of
+// debug prints
+type Logtext struct{
+	Formatter logrus.Formatter
+	LogDepth int
+	DebugOnly bool
+}
+
+func NewLogtext(formatter logrus.Formatter, debugOnly bool) *Logtext {
+	return &Logtext{
+		LogDepth: 4,
+		Formatter: formatter,
+		DebugOnly: debugOnly,
+	}
+}
+
+// Creates a hook to be added to an instance of logger. This is called with
+func (hook *Logtext) Format(entry *logrus.Entry) ([]byte, error) {
+
+	if _, file, line, ok := runtime.Caller(hook.LogDepth); ok {
+		entry.Data["line"] = line
+		entry.Data["file"] = file
+	}
+
+	if !hook.DebugOnly || entry.Level == logrus.DebugLevel {
+		stack := getTrace()
+		entry.Data["stack"] = stack
+	}
+
+	return hook.Formatter.Format(entry)
+}
+
+// handles getting the stack trace and returns it as a string
+func getTrace() string {
+	stack := make([]byte, 2048)
+	size := runtime.Stack(stack, false)
+	return string(stack[:size])
+}