Compare commits

..

No commits in common. "master" and "v0.6" have entirely different histories.
master ... v0.6

6 changed files with 22 additions and 53 deletions

View File

@ -6,8 +6,6 @@ WORKDIR /main
COPY . . COPY . .
RUN test -e /main/exec-youtuber
ENV GOPATH="" ENV GOPATH=""
ENV MNT="/mnt/" ENV MNT="/mnt/"
ENTRYPOINT ["/main/exec-youtuber"] ENTRYPOINT ["/main/exec-youtuber"]

View File

@ -4,16 +4,14 @@ import (
"local/args" "local/args"
"local/sandbox/contact/contact" "local/sandbox/contact/contact"
"os" "os"
"time"
) )
var ( var (
Feed string Feed string
Root string Root string
To string To string
Subject string Subject string
Emailer *contact.Emailer Emailer *contact.Emailer
Interval time.Duration
) )
func init() { func init() {
@ -30,7 +28,6 @@ func init() {
as.Append(args.STRING, "feed", "feed URL", "https://www.youtube.com/feeds/videos.xml?channel_id=UCwX0AEx-qIhQ9kgtlNhyIXw") as.Append(args.STRING, "feed", "feed URL", "https://www.youtube.com/feeds/videos.xml?channel_id=UCwX0AEx-qIhQ9kgtlNhyIXw")
as.Append(args.STRING, "root", "root to save videos", "/tmp") as.Append(args.STRING, "root", "root to save videos", "/tmp")
as.Append(args.DURATION, "interval", "check interval", time.Hour)
if err := as.Parse(); err != nil { if err := as.Parse(); err != nil {
panic(err) panic(err)
@ -46,5 +43,4 @@ func init() {
POP3: as.Get("pop3").GetString(), POP3: as.Get("pop3").GetString(),
Password: as.Get("password").GetString(), Password: as.Get("password").GetString(),
} }
Interval = as.GetDuration("interval")
} }

15
main.go
View File

@ -18,7 +18,7 @@ func main() {
if err != nil { if err != nil {
panic(err) panic(err)
} }
interval := config.Interval interval := time.Minute * 30
for true { for true {
err := do(client) err := do(client)
if err != nil { if err != nil {
@ -45,17 +45,18 @@ func do(client *youtubedl.Client) error {
} }
for _, result := range results { for _, result := range results {
seasonNumber := fmt.Sprintf("%02d%02d", result.Date.Year()-2000, result.Date.Month()) target := fmt.Sprintf(
episodeNumber := fmt.Sprintf("%02d", result.Date.Day()) "s%02d%02de%02d_%s.%%(ext)s",
season := fmt.Sprintf("Season_%s", seasonNumber) result.Date.Year()-2000,
episode := fmt.Sprintf("Episode_%s_-_%s", episodeNumber, strings.Join(strings.Split(result.Title, " ")[2:], "_")) result.Date.Month(),
target := fmt.Sprintf("%s/%s.%%(ext)s", season, episode) result.Date.Day(),
strings.Join(strings.Split(result.Title, " ")[2:], "_"),
)
target = path.Join(config.Root, target) target = path.Join(config.Root, target)
if _, err := os.Stat(strings.ReplaceAll(target, "%(ext)s", "mp4")); !os.IsNotExist(err) { if _, err := os.Stat(strings.ReplaceAll(target, "%(ext)s", "mp4")); !os.IsNotExist(err) {
log.Printf("already exists: %s", target) log.Printf("already exists: %s", target)
continue continue
} }
os.MkdirAll(path.Dir(target), os.ModePerm)
if err := client.Download(result.Link, target); err != nil { if err := client.Download(result.Link, target); err != nil {
errs += ", " + err.Error() errs += ", " + err.Error()
} }

Binary file not shown.

View File

@ -3,7 +3,6 @@ package youtubedl
import ( import (
"fmt" "fmt"
"log" "log"
"os"
"os/exec" "os/exec"
"path" "path"
) )
@ -16,17 +15,13 @@ func New() (*Client, error) {
} }
func (c *Client) Download(video, local string) error { func (c *Client) Download(video, local string) error {
if _, ok := os.LookupEnv("TEST"); ok {
log.Println("youtube-dl download", video, local)
return nil
}
errs := []error{} errs := []error{}
for _, extras := range [][]string{ for _, extras := range [][]string{
[]string{"--add-metadata", "--metadata-from-title", ".*[0-9]/[0-9][0-9] (?P<title>.+)"}, []string{"--add-metadata", "--metadata-from-title", ".*[0-9]/[0-9][0-9] (?P<title>.+)"},
[]string{}, []string{},
} { } {
args := append([]string{ args := append([]string{
"/tmp/yt-dlp", "youtube-dl",
"-f", "-f",
"22", "22",
"-o", "-o",
@ -35,7 +30,6 @@ func (c *Client) Download(video, local string) error {
"--write-auto-sub", "--write-auto-sub",
"--sub-format", "--sub-format",
"srt", "srt",
"--no-check-certificate",
}, extras...) }, extras...)
args = append(args, video) args = append(args, video)
cmd := exec.Command(args[0], args[1:]...) cmd := exec.Command(args[0], args[1:]...)
@ -55,11 +49,10 @@ func (c *Client) Download(video, local string) error {
return err return err
} }
cmd := exec.Command( cmd := exec.Command(
"/bin/sh", "/bin/bash",
"-c", "-c",
fmt.Sprintf( fmt.Sprintf(
"true; python3 -m vtt_to_srt %q || python3 /usr/bin/vtt_to_srt.py %q || vtt_to_srt %q", "true; python3 -m vtt_to_srt %q || python3 /usr/bin/vtt_to_srt.py %q",
path.Dir(local),
path.Dir(local), path.Dir(local),
path.Dir(local), path.Dir(local),
), ),

View File

@ -2,16 +2,11 @@ package youtubedl
import ( import (
"errors" "errors"
"fmt"
"log" "log"
"os"
"os/exec" "os/exec"
) )
func install() error { func install() error {
if _, ok := os.LookupEnv("TEST"); ok {
return nil
}
if err := installPyPip3FFMPEG(); err != nil { if err := installPyPip3FFMPEG(); err != nil {
return err return err
} }
@ -53,30 +48,16 @@ func installPyPip3FFMPEG() error {
} }
func installYtdl() error { func installYtdl() error {
cmd := exec.Command("/tmp/yt-dlp", "--version") cmd := exec.Command("youtube-dl", "--version")
if err := cmd.Run(); err == nil { if err := cmd.Run(); err == nil {
return err return err
} }
os.Remove("/tmp/yt-dlp") cmd = exec.Command("sudo", "pip3", "install", "youtube-dl")
cmd = exec.Command(
"wget",
"https://github.com/yt-dlp/yt-dlp/releases/latest/download/yt-dlp",
"--no-check-certificate",
"-O",
"/tmp/yt-dlp",
)
if err := cmd.Run(); err != nil { if err := cmd.Run(); err != nil {
o, _ := cmd.CombinedOutput() cmd = exec.Command("pip3", "install", "youtube-dl")
return fmt.Errorf("failed wget yt-dlp: %v: %s", err, o) if err := cmd.Run(); err != nil {
} return err
cmd = exec.Command( }
"chmod",
"a+rx",
"/tmp/yt-dlp",
)
if err := cmd.Run(); err != nil {
o, _ := cmd.CombinedOutput()
return fmt.Errorf("failed chmod yt-dlp: %v: %s", err, o)
} }
return nil return nil
} }