added stats page and password reset

pkg/server/handler.go

@@ -174,51 +174,19 @@ func rawHandler(ctx *Context) {
 	http.ServeContent(ctx.Response, ctx.Request, fi.Name(), fi.ModTime(), f)
 }
 
-func scanFile(srv *HTTPServer, path string) (err error) {
-	err = srv.FS.AddScan(path)
-	if err != nil {
-		return
-	}
-	go func() {
-		defer srv.FS.RemoveScan(path)
-		l := srv.Log
-		file, txt, err := srv.Scanner.Scan(path)
-		if err != nil {
-			l.Printf("scanner: %s: %s", path, err)
-			return
-		}
-		tags, err := srv.DB.GetAllRTags()
-		if err != nil {
-			l.Printf("getTags: %s: %s", path, err)
-		}
-		found := tags.Match(txt)
-		id, err := srv.DB.Index.Add(txt, found)
-		if err != nil {
-			l.Printf("index: %s: %s", path, err)
-			return
-		}
-		err = srv.DB.NewFile(id, file, found)
-		if err != nil {
-			l.Printf("newFile: %s: %s", path, err)
-		}
-	}()
-	return
-}
-
 func indexHandler(ctx *Context) {
 	switch ctx.Method() {
 	case "GET":
 		var (
 			err   error
+			ok    bool
 			fpath string
 		)
 		path := ctx.Path()
-		if strings.HasPrefix(path, core.IndexPrefix) {
-			fpath = strings.TrimPrefix(path, core.IndexPrefix)
+		if fpath, ok = hasTrimPrefix(path, core.IndexPrefix); ok {
 			err = scanFile(ctx.Srv, fpath)
 		}
-		if strings.HasPrefix(path, core.RetagPrefix) {
-			fpath = strings.TrimPrefix(path, core.RetagPrefix)
+		if fpath, ok = hasTrimPrefix(path, core.RetagPrefix); ok {
 			if fpath == "" {
 				go func() {
 					err := ctx.Srv.DB.RetagAll()
@@ -859,6 +827,19 @@ func tagsDelHandler(ctx *Context) {
 	}
 }
 
+func statsHandler(ctx *Context) {
+	ctx.Template("statsHandler")
+	ctx.Data = webData{
+		Title:     "Stats",
+		BodyTitle: "Stats",
+	}
+	switch ctx.Method() {
+	case "GET":
+		ctx.Data.Data = getStats(ctx.Srv.DB.Index)
+		ctx.Exec()
+	}
+}
+
 func logoutHandler(ctx *Context) {
 	user := ctx.User()
 	if user == "" {

pkg/server/helper.go

@@ -0,0 +1,68 @@
+// Copyright (C) 2019 Marius Schellenberger
+
+package server
+
+import (
+	"runtime"
+	"strconv"
+
+	"git.giftfish.de/ston1th/docstore/pkg/core"
+	bleve "git.giftfish.de/ston1th/docstore/pkg/index"
+)
+
+const div = 1048576
+
+func fmtMem(i uint64) string {
+	return strconv.FormatFloat(float64(i)/div, 'f', 2, 64)
+}
+
+func getStats(i *bleve.Index) *core.Stats {
+	mem := new(runtime.MemStats)
+	runtime.ReadMemStats(mem)
+	return &core.Stats{
+		Goroutines: runtime.NumGoroutine(),
+		Alloc:      fmtMem(mem.Alloc),
+		Sys:        fmtMem(mem.Sys),
+		Docs:       i.DocCount(),
+		GoVersion:  runtime.Version(),
+	}
+}
+
+func scanFile(srv *HTTPServer, path string) (err error) {
+	err = srv.FS.AddScan(path)
+	if err != nil {
+		return
+	}
+	go func() {
+		defer srv.FS.RemoveScan(path)
+		l := srv.Log
+		file, txt, err := srv.Scanner.Scan(path)
+		if err != nil {
+			l.Printf("scanner: %s: %s", path, err)
+			return
+		}
+		tags, err := srv.DB.GetAllRTags()
+		if err != nil {
+			l.Printf("getTags: %s: %s", path, err)
+		}
+		found := tags.Match(txt)
+		id, err := srv.DB.Index.Add(txt, found)
+		if err != nil {
+			l.Printf("index: %s: %s", path, err)
+			return
+		}
+		err = srv.DB.NewFile(id, file, found)
+		if err != nil {
+			l.Printf("newFile: %s: %s", path, err)
+		}
+	}()
+	return
+}
+
+// Parts taken from strings.TrimPrefix
+func hasTrimPrefix(s, prefix string) (string, bool) {
+	if len(s) >= len(prefix) && s[0:len(prefix)] == prefix {
+		return s[len(prefix):], true
+	}
+	return s, false
+}

pkg/server/routes.go

@@ -143,6 +143,13 @@ var routes = append(static, []route{
 				tagsHandler)),
 		[]string{"GET"},
 	},
+	{
+		core.StatsURI,
+		jwtHandler(
+			authHandler(
+				statsHandler)),
+		[]string{"GET"},
+	},
 	{
 		"/tags/new",
 		jwtHandler(

pkg/server/templates.go

@@ -325,6 +325,7 @@ const (
 {{end}}
 <ul class="nav navbar-nav ml-auto">
 {{if .Login}}
+    <li class="nav-item"><a class="nav-link" href="/stats">Stats</a></li>
     <li class="nav-item"><a class="nav-link" href="/user/edit">Profile</a></li>
     <li class="nav-item"><a class="nav-link" href="/logout">Logout</a></li>
 {{else}}
@@ -501,6 +502,39 @@ created:&lt;="2016-09-21"</code></pre>
     </div>
     {{end}}
 </div>
+{{end}}`
+	stats = `{{define "body"}}
+<div class="page-header">
+    <div class="row">
+        <h2>{{.BodyTitle}}</h2>
+    </div>
+</div>
+<div class="row">
+    <table class="table table-stripped table-hover">
+    <tbody>
+        <tr>
+            <td><b>Goroutines</b></td>
+            <td>{{.Data.Goroutines}}</td>
+        </tr>
+        <tr>
+            <td><b>Memory usage</b></td>
+            <td>{{.Data.Alloc}} MiB</td>
+        </tr>
+        <tr>
+            <td><b>Memory allocated</b></td>
+            <td>{{.Data.Sys}} MiB</td>
+        </tr>
+        <tr>
+            <td><b>Documents</b></td>
+            <td>{{.Data.Docs}}</td>
+        </tr>
+        <tr>
+            <td><b>Go Version</b></td>
+            <td>{{.Data.GoVersion}}</td>
+        </tr>
+    </tbody>
+    </table>
+</div>
 {{end}}`
 	tagsDel = `{{define "body"}}
 <div class="row">
@@ -662,7 +696,7 @@ created:&lt;="2016-09-21"</code></pre>
             <form class="form-horizontal" action="/user/edit" method="post">
                 <input type="hidden" name="token" value="{{.Token}}">
                 <div class="form-group">
-                    <label class="col-form-label" for="password">Password</label>
+                    <label class="col-form-label" for="password">New Password</label>
                     <input class="form-control input-sm" type="password" id="password" name="password" autofocus>
                 </div>
                 <div class="form-group">
@@ -946,6 +980,8 @@ func (s *HTTPServer) loadTemplates() {
 	s.templ["tagsNewHandler"] = parse(index, menu, tagsNew)
 	s.templ["tagsEditHandler"] = parse(index, menu, tagsEdit)
 	s.templ["tagsDelHandler"] = parse(index, menu, tagsDel)
+	// stats
+	s.templ["statsHandler"] = parse(index, menu, stats)
 	// logs
 	s.templ["logsHandler"] = parse(index, menu, logs)