| Name | " + escapeHTML(file.name) + " | |
| URL | "+domainURL()+"/u/" + file.id + " | |
| URL | "+file.link+" | |
| Mime Type | " + escapeHTML(file.mime_type) + " | |
| ID | " + file.id + " | |
| Size | " + formatDataVolume(file.size, 4) + " | |
| Bandwidth | " + formatDataVolume(file.bandwidth_used, 4) + " | |
| Upload Date | " + file.date_upload + " | |
| Upload Date | " + printDate(file.date_created, true, true, true) + " | |
| Description | " + escapeHTML(desc) + " |
Paste any pixeldrain file links in here to remove them
"), +// Fields: []Field{ +// { +// Name: "files", +// Label: "Files to delete", +// Type: FieldTypeTextarea, +// }, +// }, +// BackLink: "/admin", +// SubmitLabel: "Submit", +// } + +// if f.ReadInput(r) { +// filesText := f.FieldVal("files") + +// // Get all links from the text +// strings.Index(filesText, "/u/") + +// if len(f.SubmitMessages) == 0 { +// // Request was a success +// f.SubmitSuccess = true +// f.SubmitMessages = []template.HTML{template.HTML( +// fmt.Sprintf("Success! %d values updated", successfulUpdates), +// )} +// } +// } +// return f +// } diff --git a/webcontroller/file_viewer.go b/webcontroller/file_viewer.go index 331f8a7..c6432e2 100644 --- a/webcontroller/file_viewer.go +++ b/webcontroller/file_viewer.go @@ -2,10 +2,14 @@ package webcontroller import ( "fmt" + "io/ioutil" "net/http" + "strconv" "strings" "time" + pdmimetype "github.com/Fornaxian/pd_mime_type" + "fornaxian.com/pixeldrain-web/pixelapi" "github.com/Fornaxian/log" "github.com/julienschmidt/httprouter" @@ -152,3 +156,90 @@ func (wc *WebController) serveListViewer(w http.ResponseWriter, r *http.Request, 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://siasky.net/"+p.ByName("id"), nil) + if err != nil { + panic(err) + } + + // 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 { + 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 { + 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 == "" { + w.WriteHeader(http.StatusInternalServerError) + wc.templates.Get().ExecuteTemplate(w, "500", templateData) + return + } + contentRange = strings.TrimPrefix(contentRange, "bytes ") + size, err := strconv.ParseUint(strings.Split(contentRange, "/")[1], 10, 64) + if err != nil { + panic(err) + } + + templateData.OGData = "" + templateData.Title = fmt.Sprintf("Skylink ~ pixeldrain") + templateData.Other = viewerData{ + Type: "skylink", + APIResponse: pixelapi.FileInfo{ + Success: true, + ID: p.ByName("id"), + Name: "skynet_file.dat", + Size: size, + Views: 0, + BandwidthUsed: 0, + DateUpload: time.Now(), + DateLastView: time.Now(), + MimeType: fileType, + MimeImage: "", + ThumbnailHREF: "", + Availability: "", + }, + } + + 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) + } +} diff --git a/webcontroller/web_controller.go b/webcontroller/web_controller.go index 6d8e8ef..a7dcc72 100644 --- a/webcontroller/web_controller.go +++ b/webcontroller/web_controller.go @@ -6,6 +6,7 @@ import ( "net/http" "os" "strings" + "time" "github.com/google/uuid" @@ -30,6 +31,8 @@ type WebController struct { // page-specific variables captchaSiteKey string + + httpClient *http.Client } // New initializes a new WebController by registering all the request handlers @@ -50,6 +53,7 @@ func New( apiURLInternal: apiURLInternal, apiURLExternal: apiURLExternal, sessionCookieDomain: sessionCookieDomain, + httpClient: &http.Client{Timeout: time.Minute * 10}, } wc.templates = NewTemplateManager(resourceDir, apiURLExternal, debugMode) wc.templates.ParseTemplates(false) @@ -88,6 +92,7 @@ func New( r.GET(p+"/u/:id" /* */, wc.serveFileViewer) r.GET(p+"/u/:id/preview" /**/, wc.serveFilePreview) r.GET(p+"/l/:id" /* */, wc.serveListViewer) + r.GET(p+"/s/:id" /* */, wc.serveSkynetViewer) r.GET(p+"/t" /* */, wc.serveTemplate("paste", false)) r.GET(p+"/donation" /* */, wc.serveTemplate("donation", false)) r.GET(p+"/widgets" /* */, wc.serveTemplate("widgets", false))