Add file viewer branding options

webcontroller/file_viewer.go

@@ -97,6 +97,8 @@ func (wc *WebController) serveFileViewer(w http.ResponseWriter, r *http.Request,
 
 	templateData.Other = vd
 
+	fileStyleOverride(templateData, files)
+
 	for _, file := range files {
 		if file.AbuseType != "" {
 			w.WriteHeader(http.StatusUnavailableForLegalReasons)
@@ -157,6 +159,8 @@ func (wc *WebController) serveListViewer(w http.ResponseWriter, r *http.Request,
 	}
 	templateData.Other = vd
 
+	fileStyleOverride(templateData, list.Files)
+
 	for _, file := range list.Files {
 		if file.AbuseType != "" {
 			w.WriteHeader(http.StatusUnavailableForLegalReasons)
@@ -175,6 +179,16 @@ func (wc *WebController) serveListViewer(w http.ResponseWriter, r *http.Request,
 	}
 }
 
+func fileStyleOverride(td *TemplateData, files []pixelapi.ListFile) {
+	if len(files) == 0 {
+		return
+	}
+
+	if files[0].CustomTheme != "" {
+		td.setStyle(userStyle(files[0].CustomTheme))
+	}
+}
+
 // 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.