cwpearson/ytdlp-site
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

always re-encode to mp4

Browse Source
This commit is contained in:
Carl Pearson
2024-09-07 15:24:42 -06:00
parent 9cc496ba74
commit 689ce8280e
5 changed files with 19 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
README.md
View File

@@ -23,6 +23,7 @@ docker build -t server .
docker run --rm -it \
-p 3000:8080 \
--env YTDLP_SITE_ADMIN_INITIAL_PASSWORD=abc123 \
--env YTDLP_SITE_SESSION_AUTH_KEY=avowt7n8 \
server
docker run --rm -it \

33
handlers.go
View File

@@ -2,7 +2,6 @@ package main
import (
"bytes"
"errors"
"fmt"
"net/http"
"os"
@@ -99,32 +98,22 @@ func downloadPostHandler(c echo.Context) error {
type Meta struct {
title string
ext string
}
func getMeta(url string) (Meta, error) {
cmd := exec.Command("yt-dlp", "--simulate", "--print", "%(title)s.%(ext)s", url)
cmd := exec.Command("yt-dlp",
"--simulate", "--print", "%(title)s",
url)
var stdout bytes.Buffer
cmd.Stdout = &stdout
err := cmd.Run()
if err != nil {
fmt.Println("getTitle error:", err)
fmt.Println("getTitle error:", err, stdout.String())
return Meta{}, err
} else {
isDot := func(r rune) bool {
return r == '.'
}
fields := strings.FieldsFunc(strings.TrimSpace(stdout.String()), isDot)
if len(fields) < 2 {
return Meta{}, errors.New("couldn't parse ytdlp output")
}
return Meta{
title: strings.Join(fields[:len(fields)-1], "."),
ext: fields[len(fields)-1],
title: strings.TrimSpace(stdout.String()),
}, nil
}
}
@@ -183,7 +172,7 @@ func humanSize(bytes int64) string {
}
func startDownload(videoID uint, videoURL string) {
db.Model(&Video{}).Where("id = ?", videoID).Update("status", "downloading")
db.Model(&Video{}).Where("id = ?", videoID).Update("status", "metadata")
meta, err := getMeta(videoURL)
if err != nil {
@@ -193,15 +182,21 @@ func startDownload(videoID uint, videoURL string) {
fmt.Println("set video title:", meta.title)
db.Model(&Video{}).Where("id = ?", videoID).Update("title", meta.title)
videoFilename := fmt.Sprintf("%d-%s.%s", videoID, meta.title, meta.ext)
db.Model(&Video{}).Where("id = ?", videoID).Update("status", "downloading")
videoFilename := fmt.Sprintf("%d-%s.mp4", videoID, meta.title)
videoFilepath := filepath.Join(getDownloadDir(), "video", videoFilename)
cmd := exec.Command("yt-dlp", "-o", videoFilepath, videoURL)
cmd := exec.Command("yt-dlp",
"-f", "bestvideo[height<=720]+bestaudio/best[height<=720]",
"--recode-video", "mp4",
"-o", videoFilepath,
videoURL)
err = cmd.Run()
if err != nil {
db.Model(&Video{}).Where("id = ?", videoID).Update("status", "failed")
return
}
db.Model(&Video{}).Where("id = ?", videoID).Update("status", "audio")
audioFilename := fmt.Sprintf("%d-%s.mp3", videoID, meta.title)
audioFilepath := filepath.Join(getDownloadDir(), "audio", audioFilename)
audioDir := filepath.Dir(audioFilepath)

1
middleware.go
View File

@@ -19,6 +19,7 @@ func authMiddleware(next echo.HandlerFunc) echo.HandlerFunc {
userID, ok := session.Values["user_id"]
if !ok {
fmt.Println("authMiddleware: session does not contain user_id. Redirect to /login")
// return c.String(http.StatusForbidden, "not logged in")
return c.Redirect(http.StatusSeeOther, "/login")
}
fmt.Println("set user_id", userID, "in context")

4
templates/video.html
View File

@@ -10,10 +10,8 @@
<body>
<h1>Downloaded Video {{.video.Title}}</h1>
<video controls width="250">
<video controls playsinline width="250" preload="metadata">
<source src="/downloads/video/{{.video.VideoFilename}}" type="video/mp4" />
<!-- <source src="/media/cc0-videos/flower.mp4" type="video/mp4" /> -->
</video>
</body>

4
templates/videos.html
View File

@@ -35,8 +35,8 @@
</tr>
{{range .videos}}
<tr>
<td>{{.ID}} {{.Title}}</td>
<td>{{.URL}}</td>
<td><a href="/video/{{.ID}}">{{.Title}}</a></td>
<td><a href="{{.URL}}">{{.URL}}</a></td>
<td>{{.Length}}</td>
<td>{{.Status}}</td>
<td>