forked from BrianAllred/goydl
/
goydl.go
132 lines (105 loc) · 3.42 KB
/
goydl.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
/*
* @Author: brian.allred
* @Date: 2017-06-28 22:05:04
* @Last Modified by: brian.allred
* @Last Modified time: 2017-06-28 22:05:04
* Copyright (c) 2017-06-28 22:05:04 brian.allred
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
package goydl
import (
"encoding/json"
"errors"
"io"
"os/exec"
"strings"
)
// Represents the youtube-dl object
type youtubeDl struct {
VideoURL string
YoutubeDlPath string
Info Info
Stderr, Stdout io.ReadCloser
Options options
Err error
cmd *exec.Cmd
}
// NewYoutubeDl returns a newly instantiated youtubeDl object
func NewYoutubeDl() youtubeDl {
return youtubeDl{YoutubeDlPath: "youtube-dl", Info: Info{}, Options: NewOptions(), cmd: new(exec.Cmd)}
}
// Download will download either the given url or the youtubeDl.VideoUrl from youtube
func (ydl *youtubeDl) Download(urls ...string) (*exec.Cmd, error) {
if ydl.Options.Update.Value {
return ydl.Update()
}
if len(urls) > 1 {
return nil, errors.New("invalid argument")
}
if urls != nil {
ydl.VideoURL = urls[0]
}
// Get the info first
if _, err := ydl.GetInfo(); err != nil {
return nil, err
}
ydl.cmd = exec.Command(ydl.YoutubeDlPath, ydl.VideoURL)
ydl.cmd.Args = append(ydl.cmd.Args, strings.Fields(ydl.Options.OptionsToCliParameters())...)
// Setup stderr pipe
ydl.Stderr, ydl.Err = ydl.cmd.StderrPipe()
if ydl.Err != nil {
return nil, ydl.Err
}
// Setup stdout pipe
ydl.Stdout, ydl.Err = ydl.cmd.StdoutPipe()
if ydl.Err != nil {
return nil, ydl.Err
}
// Return the command and any error from starting the command
return ydl.cmd, ydl.cmd.Start()
}
// GetInfo gets the info of a particular video or playlist
func (ydl *youtubeDl) GetInfo() (Info, error) {
// Setup command with '-J' argument
cmd := exec.Command(ydl.YoutubeDlPath, "-J", ydl.VideoURL)
stdOut, err := cmd.StdoutPipe()
if err != nil {
return Info{}, err
}
if err := cmd.Start(); err != nil {
return Info{}, err
}
if err := json.NewDecoder(stdOut).Decode(&ydl.Info); err != nil {
return Info{}, err
}
return ydl.Info, cmd.Wait()
}
// Update updates the youtube-dl binary
func (ydl *youtubeDl) Update() (*exec.Cmd, error) {
ydl.cmd = exec.Command(ydl.YoutubeDlPath, "-U")
ydl.Stderr, ydl.Err = ydl.cmd.StderrPipe()
if ydl.Err != nil {
return nil, ydl.Err
}
ydl.Stdout, ydl.Err = ydl.cmd.StdoutPipe()
if ydl.Err != nil {
return nil, ydl.Err
}
return ydl.cmd, ydl.cmd.Start()
}