1
0
mirror of https://github.com/taigrr/wtf synced 2025-01-18 04:03:14 -08:00

Add a widget to display the log file in reverse order

This commit is contained in:
Chris Cummer 2018-06-16 14:04:48 -07:00
parent 582eba4c7f
commit 09a4139922
3 changed files with 115 additions and 28 deletions

110
logging/log.go Normal file
View File

@ -0,0 +1,110 @@
package logging
import (
"fmt"
//"io/ioutil"
"log"
"os"
"path/filepath"
"strings"
"github.com/senorprogrammer/wtf/wtf"
)
const maxBufferSize int64 = 1024
type Widget struct {
wtf.TextWidget
filePath string
}
func NewWidget() *Widget {
widget := Widget{
TextWidget: wtf.NewTextWidget(" Logs ", "logging", true),
filePath: logFilePath(),
}
return &widget
}
/* -------------------- Exported Functions -------------------- */
func Log(msg string) {
if logFileMissing() {
return
}
f, err := os.OpenFile(logFilePath(), os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666)
if err != nil {
log.Fatalf("error opening file: %v", err)
}
defer f.Close()
log.SetOutput(f)
log.Println(msg)
}
func (widget *Widget) Refresh() {
if logFileMissing() {
return
}
widget.UpdateRefreshedAt()
widget.View.SetTitle(fmt.Sprintf("%s", widget.Name))
widget.View.SetText(fmt.Sprintf("%s", widget.tailFile()))
}
/* -------------------- Unexported Functions -------------------- */
func logFileMissing() bool {
return logFilePath() == ""
}
func logFilePath() string {
dir, err := wtf.Home()
if err != nil {
return ""
}
return filepath.Join(dir, ".wtf", "log.txt")
}
func (widget *Widget) tailFile() string {
file, err := os.Open(widget.filePath)
if err != nil {
return ""
}
defer file.Close()
stat, err := file.Stat()
if err != nil {
return ""
}
bufferSize := maxBufferSize
if maxBufferSize > stat.Size() {
bufferSize = stat.Size()
}
startPos := stat.Size() - bufferSize
buff := make([]byte, bufferSize)
_, err = file.ReadAt(buff, startPos)
if err != nil {
return ""
}
dataArr := strings.Split(string(buff), "\n")
// Reverse the array of lines
// Offset by two to account for the blank line at the end
last := len(dataArr) - 2
for i := 0; i < len(dataArr)/2; i++ {
dataArr[i], dataArr[last-i] = dataArr[last-i], dataArr[i]
}
return fmt.Sprintf("%s\n", strings.Join(dataArr, "\n"))
}

7
wtf.go
View File

@ -29,6 +29,7 @@ import (
"github.com/senorprogrammer/wtf/ipaddresses/ipinfo" "github.com/senorprogrammer/wtf/ipaddresses/ipinfo"
"github.com/senorprogrammer/wtf/jenkins" "github.com/senorprogrammer/wtf/jenkins"
"github.com/senorprogrammer/wtf/jira" "github.com/senorprogrammer/wtf/jira"
"github.com/senorprogrammer/wtf/logging"
"github.com/senorprogrammer/wtf/newrelic" "github.com/senorprogrammer/wtf/newrelic"
"github.com/senorprogrammer/wtf/opsgenie" "github.com/senorprogrammer/wtf/opsgenie"
"github.com/senorprogrammer/wtf/power" "github.com/senorprogrammer/wtf/power"
@ -221,6 +222,8 @@ func addWidget(app *tview.Application, pages *tview.Pages, widgetName string) {
Widgets = append(Widgets, jenkins.NewWidget()) Widgets = append(Widgets, jenkins.NewWidget())
case "jira": case "jira":
Widgets = append(Widgets, jira.NewWidget()) Widgets = append(Widgets, jira.NewWidget())
case "logging":
Widgets = append(Widgets, logging.NewWidget())
case "newrelic": case "newrelic":
Widgets = append(Widgets, newrelic.NewWidget()) Widgets = append(Widgets, newrelic.NewWidget())
case "opsgenie": case "opsgenie":
@ -285,10 +288,10 @@ func main() {
go redrawApp(app) go redrawApp(app)
go watchForConfigChanges(app, flags.Config, display.Grid, pages) go watchForConfigChanges(app, flags.Config, display.Grid, pages)
logging.Log("Running!")
if err := app.SetRoot(pages, true).Run(); err != nil { if err := app.SetRoot(pages, true).Run(); err != nil {
fmt.Printf("Error: %v\n", err) fmt.Printf("Error: %v\n", err)
os.Exit(1) os.Exit(1)
} }
wtf.Log("Running!")
} }

View File

@ -1,26 +0,0 @@
package wtf
import (
"log"
"os"
"path/filepath"
)
//Log basic message logging, defaults to ~/.wtf/log.txt
func Log(message string) {
dir, err := Home()
if err != nil {
return
}
logfile := filepath.Join(dir, ".wtf", "log.txt")
f, err := os.OpenFile(logfile, os.O_RDWR|os.O_CREATE|os.O_APPEND, 0666)
if err != nil {
log.Fatalf("error opening file: %v", err)
}
defer f.Close()
log.SetOutput(f)
log.Println(message)
}