ytva/download.go

package main

import (
    "context"
    "fmt"
    "os"
    "io"
    "errors"
    "net/http"
    //"strconv"
    "path/filepath"


    "github.com/charmbracelet/log"
    "github.com/wader/goutubedl"
)

func checkURL(input string) (error) {
    request, err := http.NewRequest("GET", input, nil)
    if err != nil {
        return err
    }

    client := &http.Client{}
    resp, err := client.Do(request)
    if err != nil {
        return err
    }

    if resp.StatusCode == http.StatusNotFound {
        return errors.New(fmt.Sprintf("Video does not exist!"))
    }

    return nil
}

func downloadVideo(url string, path string, group GroupConfig) (err error) {
    goutubedl.Path = "yt-dlp"

    if err := checkURL(url); err != nil {
        return err
    }

    log.Debugf("URL \"%s\" exists.", url)

    video, err := goutubedl.New(context.Background(), url, goutubedl.Options{
        MergeOutputFormat: "",
        DownloadSubtitles: group.DownloadSubtitles,
        DownloadThumbnail: group.DownloadThumbnails,
    })
    if err != nil {
        return err
    }

    tempPath := filepath.Join(MainConfig.TempDir, sanitizeFilename(replaceDelimiters(video.Info.ID)))
    tempPath = filepath.Clean(tempPath)

    log.Debugf("Using temp path \"%s\".", tempPath)

    /* VIDEO */
    if group.VideoFormat != "" {
        log.Debugf("Downloading video \"%s\" with format \"%s\"", video.Info.ID, group.VideoFormat)

        videoDLResult, err := video.Download(context.Background(), group.VideoFormat)
        if err != nil {
            return err
        }

        defer videoDLResult.Close()

        file, err := os.OpenFile(tempPath+"-vid", os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0644)
        if err != nil {
            return err
        }

        defer file.Close()

        io.Copy(file, videoDLResult)

        videoDLResult.Close()
        file.Close()
    }

    /* AUDIO */
    if group.ExtractFormat != "" {
        log.Debugf("Extracting audio from \"%s\" with format \"%s\"", video.Info.ID, group.ExtractFormat)

        audioDLResult, err := video.Download(context.Background(), group.ExtractFormat)
        if err != nil {
            return err
        }

        defer audioDLResult.Close()

        file, err := os.OpenFile(tempPath+"-extract", os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0644)
        if err != nil {
            return err
        }

        defer file.Close()

        io.Copy(file, audioDLResult)

        audioDLResult.Close()
        file.Close()
    } else if group.AudioFormat != "" {
        log.Debugf("Downloading audio \"%s\" with format \"%s\"", video.Info.ID, group.AudioFormat)

        audioDLResult, err := video.Download(context.Background(), group.AudioFormat)
        if err != nil {
            return err
        }

        defer audioDLResult.Close()

        file, err := os.OpenFile(tempPath+"-audio", os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0644)
        if err != nil {
            return err
        }

        defer file.Close()

        io.Copy(file, audioDLResult)

        audioDLResult.Close()
        file.Close()
    }

    if group.VideoFormat != "" && group.AudioFormat != "" {
        if err := mergeStreams(tempPath+"-vid",tempPath+"-audio", group.CFormat, path); err != nil {
            log.Errorf("Could not merge files \"%s\" and \"%s\" to \"%s\"", tempPath+"-vid", tempPath+"-audio", path)
        }

        if err := os.Remove(tempPath+"-vid"); err != nil {
            log.Warnf("Could not remove file \"%s\"", tempPath+"-vid")
        }
        if err := os.Remove(tempPath+"-audio"); err != nil {
            log.Warnf("Could not remove file \"%s\"", tempPath+"-audio")
        }
    } else if group.ExtractFormat != "" {
        if err := extractAudio(tempPath+"-extract", group.CFormat, path); err != nil {
            log.Errorf("Could not extract audio from \"%s\" to \"%s\"", tempPath+"-extract", path)
        }

        if err := os.Remove(tempPath+"-extract"); err != nil {
            log.Warnf("Could not remove file \"%s\"", tempPath+"-extract")
        }
    } else if group.VideoFormat != "" {
        copyFile(tempPath+"-vid", path)

        if err := os.Remove(tempPath+"-vid"); err != nil {
            log.Warnf("Could not remove file \"%s\"", tempPath+"-vid")
        }
    } else if group.AudioFormat != "" {
        if err := copyFile(tempPath+"-audio", path); err != nil {
            log.Errorf("Could not copy file \"%s\" to \"%s\"", tempPath+"-audio", path)
        }
        if err := os.Remove(tempPath+"-audio"); err != nil {
            log.Warnf("Could not remove file \"%s\"", tempPath+"-audio")
        }
    }

    /*
    extension, err := getExtension(path)
    if err != nil {
        return err
    }

    os.Rename(path, path+"."+extension)
    */

    return nil
}