diff options
| -rw-r--r-- | main.go | 88 | ||||
| -rw-r--r-- | pkg/app/app.go | 106 |
2 files changed, 112 insertions, 82 deletions
@@ -1,97 +1,21 @@ package main
import (
- "html/template"
"log"
- "net/http"
- "github.com/gorilla/mux"
- "github.com/wybiral/tube/pkg/media"
+ "github.com/wybiral/tube/pkg/app"
)
const addr = "127.0.0.1:40404"
-var templates *template.Template
-
-var library *media.Library
-var playlist media.Playlist
-
func main() {
- library = media.NewLibrary()
- err := library.Import("./videos")
+ a, err := app.NewApp()
if err != nil {
log.Fatal(err)
}
- playlist = library.Playlist()
- if len(playlist) == 0 {
- log.Fatal("No valid videos found")
- }
- templates = template.Must(template.ParseGlob("templates/*"))
- r := mux.NewRouter().StrictSlash(true)
- r.HandleFunc("/", index).Methods("GET")
- r.HandleFunc("/v/{id}.mp4", video).Methods("GET")
- r.HandleFunc("/t/{id}", thumb).Methods("GET")
- r.HandleFunc("/{id}", page).Methods("GET")
- r.PathPrefix("/static/").Handler(
- http.StripPrefix("/static/", http.FileServer(http.Dir("./static/"))),
- ).Methods("GET")
- log.Printf("Serving at %s", addr)
- http.ListenAndServe(addr, r)
-}
-
-func index(w http.ResponseWriter, r *http.Request) {
- log.Printf("/index")
- http.Redirect(w, r, "/"+playlist[0].ID, 302)
-}
-
-func page(w http.ResponseWriter, r *http.Request) {
- vars := mux.Vars(r)
- id := vars["id"]
- log.Printf(id)
- playing, ok := library.Videos[id]
- if !ok {
- log.Print(ok)
- return
- }
- w.Header().Set("Content-Type", "text/html; charset=utf-8")
- templates.ExecuteTemplate(w, "index.html", &struct {
- Playing *media.Video
- Playlist media.Playlist
- }{
- Playing: playing,
- Playlist: playlist,
- })
-}
-
-func video(w http.ResponseWriter, r *http.Request) {
- vars := mux.Vars(r)
- id := vars["id"]
- log.Print("/v/" + id)
- m, ok := library.Videos[id]
- if !ok {
- return
- }
- title := m.Title
- disposition := "attachment; filename=\"" + title + ".mp4\""
- w.Header().Set("Content-Disposition", disposition)
- w.Header().Set("Content-Type", "video/mp4")
- http.ServeFile(w, r, "./videos/"+id+".mp4")
-}
-
-func thumb(w http.ResponseWriter, r *http.Request) {
- vars := mux.Vars(r)
- id := vars["id"]
- log.Printf("/t/" + id)
- m, ok := library.Videos[id]
- if !ok {
- return
- }
- w.Header().Set("Cache-Control", "public, max-age=7776000")
- if m.ThumbType == "" {
- w.Header().Set("Content-Type", "image/jpeg")
- http.ServeFile(w, r, "static/defaulticon.jpg")
- } else {
- w.Header().Set("Content-Type", m.ThumbType)
- w.Write(m.Thumb)
+ log.Printf("Serving at http://%s", addr)
+ err = a.Run(addr)
+ if err != nil {
+ log.Fatal(err)
}
}
diff --git a/pkg/app/app.go b/pkg/app/app.go new file mode 100644 index 0000000..befe1a1 --- /dev/null +++ b/pkg/app/app.go @@ -0,0 +1,106 @@ +package app + +import ( + "errors" + "html/template" + "log" + "net/http" + + "github.com/gorilla/mux" + "github.com/wybiral/tube/pkg/media" +) + +type App struct { + Library *media.Library + Playlist media.Playlist + Templates *template.Template + Router *mux.Router +} + +func NewApp() (*App, error) { + a := &App{} + lib := media.NewLibrary() + err := lib.Import("./videos") + if err != nil { + return nil, err + } + a.Library = lib + pl := lib.Playlist() + if len(pl) == 0 { + return nil, errors.New("No valid videos found") + } + a.Playlist = pl + a.Templates = template.Must(template.ParseGlob("templates/*")) + r := mux.NewRouter().StrictSlash(true) + r.HandleFunc("/", a.indexHandler).Methods("GET") + r.HandleFunc("/v/{id}.mp4", a.videoHandler).Methods("GET") + r.HandleFunc("/t/{id}", a.thumbHandler).Methods("GET") + r.HandleFunc("/{id}", a.pageHandler).Methods("GET") + fsHandler := http.StripPrefix( + "/static/", + http.FileServer(http.Dir("./static/")), + ) + r.PathPrefix("/static/").Handler(fsHandler).Methods("GET") + a.Router = r + return a, nil +} + +func (a *App) Run(addr string) error { + return http.ListenAndServe(addr, a.Router) +} + +func (a *App) indexHandler(w http.ResponseWriter, r *http.Request) { + log.Printf("/") + http.Redirect(w, r, "/"+a.Playlist[0].ID, 302) +} + +func (a *App) pageHandler(w http.ResponseWriter, r *http.Request) { + vars := mux.Vars(r) + id := vars["id"] + log.Printf("/%s", id) + playing, ok := a.Library.Videos[id] + if !ok { + return + } + w.Header().Set("Content-Type", "text/html; charset=utf-8") + a.Templates.ExecuteTemplate(w, "index.html", &struct { + Playing *media.Video + Playlist media.Playlist + }{ + Playing: playing, + Playlist: a.Playlist, + }) +} + +func (a *App) videoHandler(w http.ResponseWriter, r *http.Request) { + vars := mux.Vars(r) + id := vars["id"] + log.Printf("/v/%s", id) + m, ok := a.Library.Videos[id] + if !ok { + return + } + title := m.Title + disposition := "attachment; filename=\"" + title + ".mp4\"" + w.Header().Set("Content-Disposition", disposition) + w.Header().Set("Content-Type", "video/mp4") + http.ServeFile(w, r, "./videos/"+id+".mp4") +} + +func (a *App) thumbHandler(w http.ResponseWriter, r *http.Request) { + vars := mux.Vars(r) + id := vars["id"] + log.Printf("/t/%s", id) + m, ok := a.Library.Videos[id] + if !ok { + return + } + w.Header().Set("Cache-Control", "public, max-age=7776000") + if m.ThumbType == "" { + w.Header().Set("Content-Type", "image/jpeg") + http.ServeFile(w, r, "static/defaulticon.jpg") + } else { + w.Header().Set("Content-Type", m.ThumbType) + w.Write(m.Thumb) + } +} |