snake_case

2019-02-13 21:35:21 +01:00
parent cae7883c73
commit 6fa16218f6
9 changed files with 3 additions and 3 deletions
--- a/webcontroller/file_viewer.go
+++ b/webcontroller/file_viewer.go
@@ -0,0 +1,95 @@
+package webcontroller
+
+import (
+	"fmt"
+	"html/template"
+	"net/http"
+	"strings"
+
+	"fornaxian.com/pixeldrain-web/pixelapi"
+	"github.com/Fornaxian/log"
+	"github.com/julienschmidt/httprouter"
+)
+
+// ServeFileViewer controller for GET /u/:id
+func (wc *WebController) serveFileViewer(w http.ResponseWriter, r *http.Request, p httprouter.Params) {
+	if p.ByName("id") == "demo" {
+		wc.serveFileViewerDemo(w) // Required for a-ads.com quality check
+		return
+	}
+
+	var list = strings.Contains(p.ByName("id"), ",")
+	var ids []string
+
+	if list {
+		ids = strings.Split(p.ByName("id"), ",")
+	} else {
+		ids = append(ids, p.ByName("id"))
+	}
+
+	var api = pixelapi.New(wc.conf.APIURLInternal, "")
+	var finfo []*pixelapi.FileInfo
+	for _, id := range ids {
+		inf, err := api.GetFileInfo(id, "")
+		if err != nil {
+			continue
+		}
+		finfo = append(finfo, inf)
+	}
+
+	if len(finfo) == 0 {
+		wc.serveNotFound(w, r)
+		return
+	}
+
+	var ogData OGData
+	var err error
+	if list {
+		listdata := map[string]interface{}{
+			"data":          finfo,
+			"date_created":  "now",
+			"title":         "Concatenation of files",
+			"date_lastview": "now",
+			"views":         0,
+		}
+		err = wc.templates.Get().ExecuteTemplate(w, "file_viewer", map[string]interface{}{
+			"Title":       fmt.Sprintf("%d files in Pixeldrain", len(finfo)),
+			"APIResponse": listdata,
+			"Type":        "list",
+			"OGData":      ogData.FromFile(*finfo[0]),
+			"APIEndpoint": template.URL(wc.conf.APIURLExternal),
+		})
+	} else {
+		err = wc.templates.Get().ExecuteTemplate(w, "file_viewer", map[string]interface{}{
+			"Title":       fmt.Sprintf("%s ~ Pixeldrain file", finfo[0].Name),
+			"APIResponse": finfo[0],
+			"Type":        "file",
+			"OGData":      ogData.FromFile(*finfo[0]),
+			"APIEndpoint": template.URL(wc.conf.APIURLExternal),
+		})
+	}
+	if err != nil {
+		log.Error("Error executing template file_viewer: %s", err)
+	}
+}
+
+// ServeFileViewerDemo is a dummy API response that responds with info about a
+// non-existent demo file. This is required by the a-ads ad network to allow for
+// automatic checking of the presence of the ad unit on this page.
+func (wc *WebController) serveFileViewerDemo(w http.ResponseWriter) {
+	wc.templates.Get().ExecuteTemplate(w, "file_viewer", map[string]interface{}{
+		"APIResponse": map[string]interface{}{
+			"id":            "demo",
+			"name":          "Demo file",
+			"date_upload":   "2017-01-01 12:34:56",
+			"date_lastview": "2017-01-01 12:34:56",
+			"size":          123456789,
+			"views":         1,
+			"mime_type":     "text/demo",
+			"description":   "A file to demonstrate the viewer page",
+			"mime_image":    "/res/img/mime/text.png",
+			"thumbnail":     "/res/img/mime/text.png",
+		},
+		"Type": "file",
+	})
+}