Add metadata collection + comment rendering

DirectoryIndexers/DirectoryIndexer.go

@@ -5,6 +5,7 @@ import (
 	"github.com/dlclark/regexp2"
 	"io/ioutil"
 	"log"
+	"os"
 	"path/filepath"
 	"strings"
 	"sync"
@@ -20,6 +21,7 @@ type VideoFile struct {
 	Extension string
 	Title string
 	Id string
+	Metadata Metadata
 }
 
 func Index(path string, results chan FileList) {
@@ -56,11 +58,34 @@ func Index(path string, results chan FileList) {
 
 		id := filenameToID(video.Name())
 
+		videoObject := VideoFile{
+			Filename:  video.Name(),
+			Extension: extension,
+			Title:     filenameToTitle(video.Name(), extension),
+			Id:        id,
+		}
+
+		metadataFilename := strings.TrimSuffix(video.Name(), filepath.Ext(video.Name())) + ".info.json"
+		jsonMetadataFile, err := os.Open(path + metadataFilename)
+		if err != nil {
+			FL.Files[id] = videoObject
+			continue
+		}
+		metadataBytes, _ := ioutil.ReadAll(jsonMetadataFile)
+		_ = jsonMetadataFile.Close()
+
+		metadata, err := ParseMetadata(metadataBytes)
+		if err != nil {
+			FL.Files[id] = videoObject
+			continue
+		}
+
 		FL.Files[id] = VideoFile{
 			Filename:  video.Name(),
 			Extension: extension,
 			Title:     filenameToTitle(video.Name(), extension),
 			Id:        id,
+			Metadata:  metadata,
 		}
 	}
 

DirectoryIndexers/MetadataParser.go

@@ -0,0 +1,52 @@
+package DirectoryIndexers
+
+import (
+	"encoding/json"
+	"fmt"
+)
+
+type Metadata struct {
+	Description        string       `json:"description"`
+	UploadDate         string       `json:"upload_date"`
+	UploaderURL        string       `json:"uploader_url"`
+	ChannelID          string       `json:"channel_id"`
+	ChannelURL         string       `json:"channel_url"`
+	ViewCount          int          `json:"view_count"`
+	AverageRating      float64      `json:"average_rating"`
+	AgeLimit           int          `json:"age_limit"`
+	WebpageURL         string       `json:"webpage_url"`
+	LikeCount          int          `json:"like_count"`
+	DislikeCount       int          `json:"dislike_count"`
+	Channel            string       `json:"channel"`
+	Thumbnail          string       `json:"thumbnail"`
+	DisplayID          string       `json:"display_id"`
+	Width              int          `json:"width"`
+	Height             int          `json:"height"`
+	Fps                int          `json:"fps"`
+	Vcodec             string       `json:"vcodec"`
+	Acodec             string       `json:"acodec"`
+	Abr                float64      `json:"abr"`
+	Fulltitle          string       `json:"fulltitle"`
+	Comments           []Comments   `json:"comments"`
+	CommentCount       int          `json:"comment_count"`
+}
+type Comments struct {
+	ID               string `json:"id"`
+	Text             string `json:"text"`
+	Timestamp        int    `json:"timestamp"`
+	LikeCount        int    `json:"like_count"`
+	Author           string `json:"author"`
+	AuthorID         string `json:"author_id"`
+	AuthorThumbnail  string `json:"author_thumbnail"`
+	AuthorIsUploader bool   `json:"author_is_uploader"`
+}
+
+func ParseMetadata(jsonBytes []byte) (Metadata, error) {
+	var meta Metadata
+	err := json.Unmarshal(jsonBytes, &meta)
+	if err != nil {
+		fmt.Println(err)
+		return Metadata{}, err
+	}
+	return meta, err
+}

PageHandlers/Index.go

@@ -22,8 +22,6 @@ func Index(writer http.ResponseWriter, request *http.Request, FL *DirectoryIndex
 		Files:     FL.Files,
 	}
 
-	fmt.Println(tmpl["index.html"].Name())
-
 	err := tmpl["index.html"].ExecuteTemplate(writer, "base", data)
 	if err != nil {
 		fmt.Println(err)

PageHandlers/View.go

@@ -13,6 +13,7 @@ type ViewPageData struct {
 	Filename 	string
 	Id 			string
 	Extension 	string
+	Metadata 	DirectoryIndexers.Metadata
 }
 
 func View(writer http.ResponseWriter, request *http.Request, FL *DirectoryIndexers.FileList) {
@@ -35,6 +36,7 @@ func View(writer http.ResponseWriter, request *http.Request, FL *DirectoryIndexe
 		Filename:  video.Filename,
 		Id:        video.Id,
 		Extension: video.Extension,
+		Metadata:  video.Metadata,
 	}
 
 	err := tmpl["view.html"].ExecuteTemplate(writer, "base", data)

PageHandlers/templates/base.html

@@ -14,8 +14,8 @@
     {{template "head" .}}
 </head>
 <body style="background-color: #212121">
-    <div style="background-color: #424242; padding: 8px; border-radius: 15px; margin-bottom: 50px">
+    <div style="background-color: #424242; padding: 8px; border-radius: 15px;">
-        <h1 style="float: left">yt-dlp Archive Viewer</h1>
+        <a href="/" style="text-decoration: none"><h1 style="float: left">yt-dlp Archive Viewer</h1></a>
         <form action="/search" method="get" style="float: right; margin: auto">
             <label>
                 Search term:

PageHandlers/templates/index.html

@@ -1,9 +1,9 @@
 {{define "head"}}
-    <title>ytdlp-viewer | {{.FileCount}}</title>
+    <title>ytdlp-viewer | {{.FileCount}} videos</title>
 {{end}}
 
 {{define "body"}}
-<div style="text-align: center;">
+<div>
     <h1>yt-dlp Viewer</h1>
     <h2>{{.FileCount}} files in this archive</h2>
 
@@ -15,7 +15,9 @@
     </form>
 
     {{range .Files}}
-    <a href="/view?id={{.Id}}">{{.Id}} - {{.Title}}</a><br />
+        <p><a href="/view?id={{.Id}}">{{.Id}} - {{.Title}}</a>
+            {{ if .Metadata.ChannelID }} [M] {{ end }}
+            {{ if .Metadata.ChannelID }} - {{.Metadata.Channel }} {{ end }}<br /></p>
     {{end}}
 </div>
 {{end}}

PageHandlers/templates/view.html

@@ -8,10 +8,49 @@
 {{end}}
 
 {{define "body"}}
+    <div style="max-width: 1280px">
         <video class="video-js vjs-big-play-centered centeredvideo" id="player" height="720" width="1280" controls autoplay data-setup="{}">
             <source src="/videos/{{.Filename}}">
         </video>
         <h1>{{.Title}}</h1>
+
+        {{ if .Metadata.ChannelID }}
+            <div>
+                <p style="float: right">{{ .Metadata.LikeCount }} likes | {{ .Metadata.DislikeCount }} dislikes</p>
+
+                <p>{{ .Metadata.ViewCount }} views</p>
+                <p>Uploaded by: <a href="{{ .Metadata.UploaderURL }}">{{ .Metadata.Channel }}</a></p>
+                <hr />
+
+                <div style="clear: both"></div>
+            </div>
+            <br />
+
+            <div style="white-space: pre-line;">
+                {{ .Metadata.Description }}
+            </div>
+
+            <p>Comments: ({{ .Metadata.CommentCount }}</p>
+            <hr />
+            <div>
+                {{range .Metadata.Comments}}
+                    <div>
+                        <img src="{{ .AuthorThumbnail }}" style="border-radius: 50%; max-height: 75px; max-width: 75px; float: left" />
+                        <p>{{ .Author }}</p>
+                        <div style="clear: both"></div>
+                        <p>{{ .LikeCount }} likes</p>
+                    </div>
+                    <div style="white-space: pre-line;">
+                        {{ .Text }}
+                    </div>
+                    <hr />
+                {{end}}
+            </div>
+        {{ end }}
+    </div>
+    <br />
+
+
     <p>Filetype: {{.Extension}}</p>
     <p>Youtube ID: {{.Id}}</p>
     <p>Filename: {{.Filename}}</p>

go.sum

@@ -0,0 +1,2 @@
+github.com/dlclark/regexp2 v1.4.0 h1:F1rxgk7p4uKjwIQxBs9oAXe5CqrXlCduYEJvrF4u93E=
+github.com/dlclark/regexp2 v1.4.0/go.mod h1:2pZnwuY/m+8K6iRw6wQdMtk+rH5tNGR1i55kozfMjCc=