package main

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


    "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, videoFormat string, audioFormat string, outDir string, sizeLimit float64) (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{})
    if err != nil {
        return err
    }

    path := outDir+strconv.Itoa(int(video.Info.Timestamp))+" - "+video.Info.Title

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

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

            defer videoDLResult.Close()

            file, err := os.OpenFile(path+"-vid", os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0644)
            defer file.Close()

            io.Copy(file, videoDLResult)

            videoDLResult.Close()
            file.Close()
        }

        /* AUDIO */
        {
            log.Debugf("Downloading audio \"%s\" with format \"%s\"", video.Info.ID, audioFormat)

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

            defer audioDLResult.Close()

            file, err := os.OpenFile(path+"-audio", os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0644)
            defer file.Close()

            io.Copy(file, audioDLResult)

            audioDLResult.Close()
            file.Close()
        }

        mergeStreams(path+"-vid",path+"-audio",path+".mkv")

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

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

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

    return nil
}