310 lines
8.8 KiB
Go
310 lines
8.8 KiB
Go
package webcontroller
|
|
|
|
import (
|
|
"fmt"
|
|
"io/ioutil"
|
|
"math/rand"
|
|
"mime"
|
|
"net/http"
|
|
"strconv"
|
|
"strings"
|
|
"time"
|
|
|
|
"fornaxian.com/pixeldrain-api/api/apiclient"
|
|
"fornaxian.com/pixeldrain-api/api/apitype"
|
|
"github.com/Fornaxian/log"
|
|
pdmimetype "github.com/Fornaxian/pd_mime_type"
|
|
"github.com/julienschmidt/httprouter"
|
|
)
|
|
|
|
func (wc *WebController) viewTokenOrBust() (t string) {
|
|
var err error
|
|
if t, err = wc.systemPixelAPI.GetMiscViewToken(); err != nil {
|
|
log.Error("Could not get viewtoken: %s", err)
|
|
}
|
|
return t
|
|
}
|
|
|
|
func browserCompat(ua string) bool {
|
|
return strings.Contains(ua, "MSIE") || strings.Contains(ua, "Trident/7.0")
|
|
}
|
|
|
|
func adType() (i int) {
|
|
// Intn returns a number up to n, but never n itself. So it get a random 0
|
|
// or 1 we need to give it n=2. We can use this function to make other
|
|
// splits like 1/3 1/4, etc
|
|
i = rand.Intn(4)
|
|
|
|
// The return value corresponds to the type of ad shown:
|
|
// 0: A-ads
|
|
// 1: Amarulasolutions
|
|
// 2: Patreon
|
|
|
|
switch i {
|
|
case 0, 1, 2: // 75% of the traffic
|
|
return 1
|
|
case 3:
|
|
return 2
|
|
default:
|
|
panic(fmt.Errorf(
|
|
"random number generator returned unrecognised number: %d", i),
|
|
)
|
|
}
|
|
}
|
|
|
|
type viewerData struct {
|
|
Type string // file or list
|
|
CaptchaKey string
|
|
ViewToken string
|
|
AdType int
|
|
ShowAds bool
|
|
APIResponse interface{}
|
|
}
|
|
|
|
// ServeFileViewer controller for GET /u/:id
|
|
func (wc *WebController) serveFileViewer(w http.ResponseWriter, r *http.Request, p httprouter.Params) {
|
|
var err error
|
|
if p.ByName("id") == "demo" {
|
|
wc.serveFileViewerDemo(w, r) // Required for a-ads.com quality check
|
|
return
|
|
}
|
|
|
|
var ids = strings.Split(p.ByName("id"), ",")
|
|
|
|
templateData := wc.newTemplateData(w, r)
|
|
|
|
var finfo []apitype.ListFile
|
|
for _, id := range ids {
|
|
inf, err := templateData.PixelAPI.GetFileInfo(id)
|
|
if err != nil {
|
|
continue
|
|
}
|
|
finfo = append(finfo, apitype.ListFile{FileInfo: inf})
|
|
}
|
|
|
|
if len(finfo) == 0 {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
wc.templates.Get().ExecuteTemplate(w, "file_not_found", templateData)
|
|
return
|
|
}
|
|
|
|
showAds := true
|
|
if (templateData.Authenticated && templateData.User.DisableAdDisplay) || finfo[0].ShowAds == false {
|
|
showAds = false
|
|
}
|
|
|
|
templateData.OGData = metadataFromFile(finfo[0].FileInfo)
|
|
if len(ids) > 1 {
|
|
templateData.Title = fmt.Sprintf("%d files on pixeldrain", len(finfo))
|
|
templateData.Other = viewerData{
|
|
Type: "list",
|
|
CaptchaKey: wc.captchaKey(),
|
|
ViewToken: wc.viewTokenOrBust(),
|
|
AdType: adType(),
|
|
ShowAds: showAds,
|
|
APIResponse: apitype.ListInfo{
|
|
Success: true,
|
|
Title: "Multiple files",
|
|
DateCreated: time.Now(),
|
|
Files: finfo,
|
|
},
|
|
}
|
|
} else {
|
|
templateData.Title = fmt.Sprintf("%s ~ pixeldrain", finfo[0].Name)
|
|
templateData.Other = viewerData{
|
|
Type: "file",
|
|
CaptchaKey: wc.captchaKey(),
|
|
ViewToken: wc.viewTokenOrBust(),
|
|
AdType: adType(),
|
|
ShowAds: showAds,
|
|
APIResponse: finfo[0].FileInfo,
|
|
}
|
|
}
|
|
|
|
var templateName = "file_viewer"
|
|
if browserCompat(r.UserAgent()) {
|
|
templateName = "file_viewer_compat"
|
|
}
|
|
|
|
err = wc.templates.Get().ExecuteTemplate(w, templateName, templateData)
|
|
if err != nil && !strings.Contains(err.Error(), "broken pipe") {
|
|
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, r *http.Request) {
|
|
templateData := wc.newTemplateData(w, r)
|
|
templateData.Other = viewerData{
|
|
Type: "file",
|
|
CaptchaKey: wc.captchaSiteKey,
|
|
AdType: 0, // Always show a-ads on the demo page
|
|
ShowAds: true,
|
|
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,
|
|
"bandwidth_used": 123456789,
|
|
"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",
|
|
},
|
|
}
|
|
err := wc.templates.Get().ExecuteTemplate(w, "file_viewer", templateData)
|
|
if err != nil && !strings.Contains(err.Error(), "broken pipe") {
|
|
log.Error("Error rendering demo file: %s", err)
|
|
}
|
|
}
|
|
|
|
// ServeListViewer controller for GET /l/:id
|
|
func (wc *WebController) serveListViewer(w http.ResponseWriter, r *http.Request, p httprouter.Params) {
|
|
var templateData = wc.newTemplateData(w, r)
|
|
var list, err = templateData.PixelAPI.GetList(p.ByName("id"))
|
|
if err != nil {
|
|
if err, ok := err.(apiclient.Error); ok && err.Status == http.StatusNotFound {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
wc.templates.Get().ExecuteTemplate(w, "list_not_found", templateData)
|
|
} else {
|
|
log.Error("API request error occurred: %s", err)
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
wc.templates.Get().ExecuteTemplate(w, "500", templateData)
|
|
}
|
|
return
|
|
}
|
|
if len(list.Files) == 0 {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
wc.templates.Get().ExecuteTemplate(w, "list_not_found", templateData)
|
|
return
|
|
}
|
|
|
|
showAds := true
|
|
if (templateData.Authenticated && templateData.User.DisableAdDisplay) || list.Files[0].ShowAds == false {
|
|
showAds = false
|
|
}
|
|
|
|
templateData.Title = fmt.Sprintf("%s ~ pixeldrain", list.Title)
|
|
templateData.OGData = metadataFromList(list)
|
|
templateData.Other = viewerData{
|
|
Type: "list",
|
|
CaptchaKey: wc.captchaSiteKey,
|
|
ViewToken: wc.viewTokenOrBust(),
|
|
AdType: adType(),
|
|
ShowAds: showAds,
|
|
APIResponse: list,
|
|
}
|
|
|
|
var templateName = "file_viewer"
|
|
if browserCompat(r.UserAgent()) {
|
|
templateName = "file_viewer_compat"
|
|
}
|
|
|
|
err = wc.templates.Get().ExecuteTemplate(w, templateName, templateData)
|
|
if err != nil && !strings.Contains(err.Error(), "broken pipe") {
|
|
log.Error("Error executing template file_viewer: %s", err)
|
|
}
|
|
}
|
|
|
|
// ServeFileViewer controller for GET /s/:id
|
|
func (wc *WebController) serveSkynetViewer(w http.ResponseWriter, r *http.Request, p httprouter.Params) {
|
|
var err error
|
|
templateData := wc.newTemplateData(w, r)
|
|
|
|
// Get the first few bytes from the file to probe the content type and
|
|
// length
|
|
rq, err := http.NewRequest("GET", "https://skydrain.net/file/"+p.ByName("id"), nil)
|
|
if err != nil {
|
|
log.Warn("Failed to make request to sia portal: %s", err)
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
wc.templates.Get().ExecuteTemplate(w, "500", templateData)
|
|
return
|
|
}
|
|
|
|
// Range header limits the number of bytes which will be read
|
|
rq.Header.Set("Range", "bytes=0-1023")
|
|
|
|
resp, err := wc.httpClient.Do(rq)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
if resp.StatusCode >= 500 {
|
|
head, _ := ioutil.ReadAll(resp.Body)
|
|
log.Warn("Sia portal returned error: %s", head)
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
wc.templates.Get().ExecuteTemplate(w, "500", templateData)
|
|
return
|
|
} else if resp.StatusCode >= 400 {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
wc.templates.Get().ExecuteTemplate(w, "file_not_found", templateData)
|
|
return
|
|
}
|
|
|
|
head, err := ioutil.ReadAll(resp.Body)
|
|
if err != nil {
|
|
log.Warn("Failed to read file header from Sia portal: %s", err)
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
wc.templates.Get().ExecuteTemplate(w, "500", templateData)
|
|
return
|
|
}
|
|
|
|
var fileType = resp.Header.Get("Content-Type")
|
|
if fileType == "application/octet-stream" || fileType == "" {
|
|
fileType = pdmimetype.Detect(head)
|
|
}
|
|
|
|
// Now get the size of the file from the content-range header
|
|
contentRange := resp.Header.Get("Content-Range")
|
|
if contentRange == "" {
|
|
log.Warn("Sia portal didn't send Content-Range")
|
|
w.WriteHeader(http.StatusInternalServerError)
|
|
wc.templates.Get().ExecuteTemplate(w, "500", templateData)
|
|
return
|
|
}
|
|
contentRange = strings.TrimPrefix(contentRange, "bytes ")
|
|
size, err := strconv.ParseInt(strings.Split(contentRange, "/")[1], 10, 64)
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
var name string
|
|
_, params, err := mime.ParseMediaType(resp.Header.Get("Content-Disposition"))
|
|
if err != nil {
|
|
name = "skynet_file"
|
|
} else {
|
|
name = params["filename"]
|
|
}
|
|
|
|
templateData.OGData = ""
|
|
templateData.Title = fmt.Sprintf("name ~ Skynet")
|
|
templateData.Other = viewerData{
|
|
Type: "skylink",
|
|
AdType: adType(),
|
|
APIResponse: apitype.FileInfo{
|
|
Success: true,
|
|
ID: p.ByName("id"),
|
|
Name: name,
|
|
Size: size,
|
|
DateUpload: time.Now(),
|
|
DateLastView: time.Now(),
|
|
MimeType: fileType,
|
|
},
|
|
}
|
|
|
|
var templateName = "file_viewer"
|
|
if browserCompat(r.UserAgent()) {
|
|
templateName = "file_viewer_compat"
|
|
}
|
|
|
|
err = wc.templates.Get().ExecuteTemplate(w, templateName, templateData)
|
|
if err != nil && !strings.Contains(err.Error(), "broken pipe") {
|
|
log.Error("Error executing template file_viewer: %s", err)
|
|
}
|
|
}
|