Skip to content

Commit acba10e

Browse files
mniakmniak
committed
Generate files
1 parent 7009dcb commit acba10e

File tree

10 files changed

+129
-66
lines changed

10 files changed

+129
-66
lines changed

‎.vscode/multistreamer.code-workspace

+11
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
{
2+
"folders": [
3+
{
4+
"path": "..\\tools"
5+
},
6+
{
7+
"path": ".."
8+
}
9+
],
10+
"settings": {}
11+
}

‎Dockerfile

-1
Original file line numberDiff line numberDiff line change
@@ -76,7 +76,6 @@ EXPOSE 1935
7676

7777
# Set up configuration
7878
RUN sed -i "s/ENABLED=.*/ENABLED=1/" /etc/default/stunnel4
79-
COPY stunnel.conf /etc/stunnel/stunnel.conf
8079

8180
COPY scripts/ /scripts/
8281
COPY --from=tools-builder /app/app.bin /tools/tools.bin

‎nginx.conf

-15
This file was deleted.

‎scripts/init.sh

+7-1
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,12 @@
1-
env
1+
# Generate configuration files
22
/tools/tools.bin mkconfig nginx -o /etc/nginx/nginx.conf
3+
/tools/tools.bin mkconfig stunnel -o /etc/stunnel/stunnel.conf
4+
5+
# Start STunnel4
36
stunnel
47

8+
# Start tools server
59
/tools/tools.bin server &
10+
11+
# Start NGINX
612
nginx -g "daemon off;"

‎stunnel.conf

-17
This file was deleted.

‎tools/cmd/mkconfig.go

+71-10
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@ import (
55
"os"
66
"text/template"
77

8+
"github.com/mniak/multistreamer/tools/internal/utils"
89
"github.com/spf13/cobra"
910
"github.com/spf13/viper"
1011
)
@@ -31,42 +32,45 @@ rtmp {
3132
on_publish http://127.0.0.1:8080/on_publish;
3233
}
3334
}
34-
}`))
35+
}
36+
`))
3537
var err error
3638
data := make(map[string]interface{})
3739
if data["Port"], err = cmd.Flags().GetInt("port"); err != nil {
38-
fmt.Println("Invalid port", err.Error())
40+
fmt.Println("Invalid port.", err.Error())
3941
os.Exit(1)
4042
}
4143
if data["STunnelPort"], err = cmd.Flags().GetInt("stunnel_port"); err != nil {
42-
fmt.Println("Invalid STunnel port", err.Error())
44+
fmt.Println("Invalid STunnel port.", err.Error())
4345
os.Exit(1)
4446
}
45-
if data["YoutubeURL"], err = cmd.Flags().GetString("youtube_url"); err != nil {
46-
fmt.Println("Invalid YouTube URL", err.Error())
47+
if yturl, err := utils.GetRTMPURLFlag(cmd, "youtube_url"); err == nil {
48+
data["YoutubeURL"] = yturl.String()
49+
} else {
50+
fmt.Println("Invalid YouTube URL.", err.Error())
4751
os.Exit(1)
4852
}
4953
if data["YoutubeKey"], err = cmd.Flags().GetString("youtube_key"); err != nil {
50-
fmt.Println("Invalid YouTube Key", err.Error())
54+
fmt.Println("Invalid YouTube Key.", err.Error())
5155
os.Exit(1)
5256
}
5357
if data["FacebookKey"], err = cmd.Flags().GetString("facebook_key"); err != nil {
54-
fmt.Println("Invalid Facebook Key", err.Error())
58+
fmt.Println("Invalid Facebook Key.", err.Error())
5559
os.Exit(1)
5660
}
5761
var filename string
5862
if filename, err = cmd.Flags().GetString("output"); err != nil {
59-
fmt.Println("Invalid output filename", err.Error())
63+
fmt.Println("Invalid output filename.", err.Error())
6064
os.Exit(1)
6165
}
6266
var file *os.File
6367
if file, err = os.Create(filename); err != nil {
64-
fmt.Println("Could not open file", err.Error())
68+
fmt.Println("Could not open file.", err.Error())
6569
os.Exit(1)
6670
}
6771
defer file.Close()
6872
if err = t.Execute(file, data); err != nil {
69-
fmt.Println("Could not process template", err.Error())
73+
fmt.Println("Could not process template.", err.Error())
7074
os.Exit(1)
7175
}
7276
},
@@ -75,6 +79,63 @@ rtmp {
7579
var mkconfigSTunnelCmd = &cobra.Command{
7680
Use: "stunnel",
7781
Run: func(cmd *cobra.Command, args []string) {
82+
83+
t := template.Must(template.New("nginx-config").Parse(`
84+
pid = /var/run/stunnel4/stunnel.pid
85+
output = /var/log/stunnel4/stunnel.log
86+
87+
setuid = stunnel4
88+
setgid = stunnel4
89+
90+
# https://www.stunnel.org/faq.html
91+
socket = r:TCP_NODELAY=1
92+
socket = l:TCP_NODELAY=1
93+
94+
debug = 4
95+
96+
[fb-live]
97+
client = yes
98+
accept = {{.STunnelPort}}
99+
connect = {{.FacebookHost}}:{{.FacebookPort}}
100+
verifyChain = no
101+
`))
102+
var err error
103+
data := make(map[string]interface{})
104+
if data["STunnelPort"], err = cmd.Flags().GetInt("stunnel_port"); err != nil {
105+
fmt.Println("Invalid STunnel port.", err.Error())
106+
os.Exit(1)
107+
}
108+
if url, err := utils.GetRTMPURLFlag(cmd, "facebook_url"); err == nil {
109+
data["FacebookHost"] = url.Hostname()
110+
port := url.Port()
111+
if port == "" {
112+
if url.Scheme == "rtmps" {
113+
port = "443"
114+
} else {
115+
fmt.Println("The Facebook URL must use scheme rtmps://.")
116+
os.Exit(1)
117+
}
118+
}
119+
data["FacebookPort"] = port
120+
} else {
121+
fmt.Println("Invalid Facebook URL.", err.Error())
122+
os.Exit(1)
123+
}
124+
var filename string
125+
if filename, err = cmd.Flags().GetString("output"); err != nil {
126+
fmt.Println("Invalid output filename.", err.Error())
127+
os.Exit(1)
128+
}
129+
var file *os.File
130+
if file, err = os.Create(filename); err != nil {
131+
fmt.Println("Could not open file.", err.Error())
132+
os.Exit(1)
133+
}
134+
defer file.Close()
135+
if err = t.Execute(file, data); err != nil {
136+
fmt.Println("Could not process template.", err.Error())
137+
os.Exit(1)
138+
}
78139
},
79140
}
80141

‎tools/cmd/root.go

+1-20
Original file line numberDiff line numberDiff line change
@@ -12,17 +12,11 @@ import (
1212

1313
var cfgFile string
1414

15-
// rootCmd represents the base command when called without any subcommands
1615
var rootCmd = &cobra.Command{
1716
Use: "tools",
1817
Short: "A brief description of your application",
19-
// Uncomment the following line if your bare application
20-
// has an action associated with it:
21-
// Run: func(cmd *cobra.Command, args []string) { },
2218
}
2319

24-
// Execute adds all child commands to the root command and sets flags appropriately.
25-
// This is called by main.main(). It only needs to happen once to the rootCmd.
2620
func Execute() {
2721
if err := rootCmd.Execute(); err != nil {
2822
fmt.Println(err)
@@ -32,39 +26,26 @@ func Execute() {
3226

3327
func init() {
3428
cobra.OnInitialize(initConfig)
35-
36-
// Here you will define your flags and configuration settings.
37-
// Cobra supports persistent flags, which, if defined here,
38-
// will be global for your application.
39-
4029
rootCmd.PersistentFlags().StringVar(&cfgFile, "config", "", "config file (default is $HOME/.tools.yaml)")
41-
42-
// Cobra also supports local flags, which will only run
43-
// when this action is called directly.
4430
rootCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
4531
}
4632

47-
// initConfig reads in config file and ENV variables if set.
4833
func initConfig() {
4934
if cfgFile != "" {
50-
// Use config file from the flag.
5135
viper.SetConfigFile(cfgFile)
5236
} else {
53-
// Find home directory.
5437
home, err := homedir.Dir()
5538
if err != nil {
5639
fmt.Println(err)
5740
os.Exit(1)
5841
}
5942

60-
// Search config in home directory with name ".tools" (without extension).
6143
viper.AddConfigPath(home)
6244
viper.SetConfigName(".tools")
6345
}
6446

65-
viper.AutomaticEnv() // read in environment variables that match
47+
viper.AutomaticEnv()
6648

67-
// If a config file is found, read it in.
6849
if err := viper.ReadInConfig(); err == nil {
6950
fmt.Println("Using config file:", viper.ConfigFileUsed())
7051
}

‎tools/go.mod

+1-2
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,7 @@ require (
88
github.com/golang/protobuf v1.4.2 // indirect
99
github.com/json-iterator/go v1.1.10 // indirect
1010
github.com/mitchellh/go-homedir v1.1.0
11-
github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
12-
github.com/modern-go/reflect2 v1.0.1 // indirect
11+
github.com/pkg/errors v0.8.1
1312
github.com/spf13/cobra v1.0.0
1413
github.com/spf13/viper v1.7.1
1514
golang.org/x/sys v0.0.0-20200819171115-d785dc25833f // indirect

‎tools/go.sum

+15
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@ cloud.google.com/go/firestore v1.1.0/go.mod h1:ulACoGHTpvq5r8rxGJ4ddJZBZqakUQqCl
1111
cloud.google.com/go/pubsub v1.0.1/go.mod h1:R0Gpsv3s54REJCy4fxDixWD93lHJMoZTyQ2kNxGRt3I=
1212
cloud.google.com/go/storage v1.0.0/go.mod h1:IhtSnM/ZTZV8YYJWCY8RULGVqBDmpoyjwiyrjsg+URw=
1313
dmitri.shuralyov.com/gpu/mtl v0.0.0-20190408044501-666a987793e9/go.mod h1:H6x//7gZCb22OMCxBHrMx7a5I7Hp++hsVxbQ4BYO7hU=
14+
github.com/BurntSushi/toml v0.3.1 h1:WXkYYl6Yr3qBf1K79EBnL4mak0OimBfB0XUf9Vl28OQ=
1415
github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU=
1516
github.com/BurntSushi/xgb v0.0.0-20160522181843-27f122750802/go.mod h1:IVnqGOEym/WlBOVXweHU+Q+/VP0lqqI8lqeDx9IjBqo=
1617
github.com/OneOfOne/xxhash v1.2.2/go.mod h1:HSdplMjZKSmBqAxg5vPj2TmRDmfkzw+cTzAElWljhcU=
@@ -36,6 +37,7 @@ github.com/coreos/go-systemd v0.0.0-20190321100706-95778dfbb74e/go.mod h1:F5haX7
3637
github.com/coreos/pkg v0.0.0-20180928190104-399ea9e2e55f/go.mod h1:E3G3o1h8I7cfcXa63jLwjI0eiQQMgzzUDFVpN/nH/eA=
3738
github.com/cpuguy83/go-md2man/v2 v2.0.0/go.mod h1:maD7wRr/U5Z6m/iR4s+kqSMx2CaBsrgA7czyZG/E6dU=
3839
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
40+
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
3941
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
4042
github.com/dgrijalva/jwt-go v3.2.0+incompatible/go.mod h1:E3ru+11k8xSBh+hMPgOLZmtrrCbhqsmaPHjLKYnJCaQ=
4143
github.com/dgryski/go-sip13 v0.0.0-20181026042036-e10d5fee7954/go.mod h1:vAd38F8PWV+bWy6jNmig1y/TA+kYO4g3RSRF0IAv0no=
@@ -53,6 +55,7 @@ github.com/go-gl/glfw v0.0.0-20190409004039-e6da0acd62b1/go.mod h1:vR7hzQXu2zJy9
5355
github.com/go-kit/kit v0.8.0/go.mod h1:xBxKIO96dXMWWy0MnWVtmwkA9/13aqxPnvrjFYMA2as=
5456
github.com/go-logfmt/logfmt v0.3.0/go.mod h1:Qt1PoO58o5twSAckw1HlFXLmHsOX5/0LbT9GBnD5lWE=
5557
github.com/go-logfmt/logfmt v0.4.0/go.mod h1:3RMwSq7FuexP4Kalkev3ejPJsZTpXXBr9+V4qmtdjCk=
58+
github.com/go-playground/assert/v2 v2.0.1 h1:MsBgLAaY856+nPRTKrp3/OZK38U/wa0CcBYNjji3q3A=
5659
github.com/go-playground/assert/v2 v2.0.1/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4=
5760
github.com/go-playground/locales v0.13.0 h1:HyWk6mgj5qFqCT5fjGBuRArbVDfE4hi8+e8ceBS/t7Q=
5861
github.com/go-playground/locales v0.13.0/go.mod h1:taPMhCMXrRLJO55olJkUXHZBHCxTMfnGwq/HNwmWNS8=
@@ -89,6 +92,7 @@ github.com/google/go-cmp v0.2.0/go.mod h1:oXzfMopK8JAjlY9xF4vHSVASa0yLyX7SntLO5a
8992
github.com/google/go-cmp v0.3.0/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=
9093
github.com/google/go-cmp v0.3.1/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU=
9194
github.com/google/go-cmp v0.4.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
95+
github.com/google/go-cmp v0.5.0 h1:/QaMHBdZ26BB3SSst0Iwl10Epc+xhTquomWX0oZEB6w=
9296
github.com/google/go-cmp v0.5.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE=
9397
github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg=
9498
github.com/google/martian v2.1.0+incompatible/go.mod h1:9I4somxYTbIHy5NJKHRl3wXiIaQGbYVAs8BPL6v8lEs=
@@ -97,6 +101,7 @@ github.com/google/pprof v0.0.0-20190515194954-54271f7e092f/go.mod h1:zfwlbNMJ+OI
97101
github.com/google/renameio v0.1.0/go.mod h1:KWCgfxg9yswjAJkECMjeO8J8rahYeXnNhOm40UhjYkI=
98102
github.com/googleapis/gax-go/v2 v2.0.4/go.mod h1:0Wqv26UfaUD9n4G6kQubkQ+KchISgw+vpHVxEJEs9eg=
99103
github.com/googleapis/gax-go/v2 v2.0.5/go.mod h1:DWXyrwAJ9X0FpwwEdw+IPEYBICEFu5mhpdKc/us6bOk=
104+
github.com/gopherjs/gopherjs v0.0.0-20181017120253-0766667cb4d1 h1:EGx4pi6eqNxGaHF6qqu48+N2wcFQ5qg5FXgOdqsJ5d8=
100105
github.com/gopherjs/gopherjs v0.0.0-20181017120253-0766667cb4d1/go.mod h1:wJfORRmW1u3UXTncJ5qlYoELFm8eSnnEO6hX4iZ3EWY=
101106
github.com/gorilla/websocket v1.4.0/go.mod h1:E7qHFY5m1UJ88s3WnNqhKjPHQ0heANvMoAMk2YaljkQ=
102107
github.com/gorilla/websocket v1.4.2/go.mod h1:YR8l580nyteQvAITg2hZ9XVh4b55+EU/adAjf1fMHhE=
@@ -133,14 +138,17 @@ github.com/json-iterator/go v1.1.9/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/u
133138
github.com/json-iterator/go v1.1.10 h1:Kz6Cvnvv2wGdaG/V8yMvfkmNiXq9Ya2KUv4rouJJr68=
134139
github.com/json-iterator/go v1.1.10/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4=
135140
github.com/jstemmer/go-junit-report v0.0.0-20190106144839-af01ea7f8024/go.mod h1:6v2b51hI/fHJwM22ozAgKL4VKDeJcHhJFhtBdhmNjmU=
141+
github.com/jtolds/gls v4.20.0+incompatible h1:xdiiI2gbIgH/gLH7ADydsJ1uDOEzR8yvV7C0MuV77Wo=
136142
github.com/jtolds/gls v4.20.0+incompatible/go.mod h1:QJZ7F/aHp+rZTRtaJ1ow/lLfFfVYBRgL+9YlvaHOwJU=
137143
github.com/julienschmidt/httprouter v1.2.0/go.mod h1:SYymIcj16QtmaHHD7aYtjjsJG7VTCxuUUipMqKk8s4w=
138144
github.com/kisielk/errcheck v1.1.0/go.mod h1:EZBBE59ingxPouuu3KfxchcWSUPOHkagtvWXihfKN4Q=
139145
github.com/kisielk/gotool v1.0.0/go.mod h1:XhKaO+MFFWcvkIS/tQcRk01m1F5IRFswLeQ+oQHNcck=
140146
github.com/konsorten/go-windows-terminal-sequences v1.0.1/go.mod h1:T0+1ngSBFLxvqU3pZ+m/2kptfBszLMUkC4ZK/EgS/cQ=
141147
github.com/kr/logfmt v0.0.0-20140226030751-b84e30acd515/go.mod h1:+0opPa2QZZtGFBFZlji/RkVcI2GknAs/DXo4wKdlNEc=
148+
github.com/kr/pretty v0.1.0 h1:L/CwN0zerZDmRFUapSPitk6f+Q3+0za1rQkzVuMiMFI=
142149
github.com/kr/pretty v0.1.0/go.mod h1:dAy3ld7l9f0ibDNOQOHHMYYIIbhfbHSm3C4ZsoJORNo=
143150
github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ=
151+
github.com/kr/text v0.1.0 h1:45sCR5RtlFHMR4UwH9sdQ5TC8v0qDQCHnXt+kaKSTVE=
144152
github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI=
145153
github.com/leodido/go-urn v1.2.0 h1:hpXL4XnriNwQ/ABnpepYM/1vCLWNDfUNts8dX3xTG6Y=
146154
github.com/leodido/go-urn v1.2.0/go.mod h1:+8+nEpDfqqsY+g338gtMEUOtuK+4dEMhiQEgxpxOKII=
@@ -177,7 +185,9 @@ github.com/pascaldekloe/goe v0.0.0-20180627143212-57f6aae5913c/go.mod h1:lzWF7FI
177185
github.com/pelletier/go-toml v1.2.0 h1:T5zMGML61Wp+FlcbWjRDT7yAxhJNAiPPLOFECq181zc=
178186
github.com/pelletier/go-toml v1.2.0/go.mod h1:5z9KED0ma1S8pY6P1sdut58dfprrGBbd/94hg7ilaic=
179187
github.com/pkg/errors v0.8.0/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
188+
github.com/pkg/errors v0.8.1 h1:iURUrRGxPUNPdy5/HRSm+Yj6okJ6UtLINN0Q9M4+h3I=
180189
github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
190+
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
181191
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
182192
github.com/posener/complete v1.1.1/go.mod h1:em0nMJCgc9GFtwrmVmEMR/ZL6WyhyjMBndrE9hABlRI=
183193
github.com/prometheus/client_golang v0.9.1/go.mod h1:7SWBe2y4D6OKWSNQJUaRYU/AaXPKyh/dDVn+NZz0KFw=
@@ -197,7 +207,9 @@ github.com/ryanuber/columnize v0.0.0-20160712163229-9b3edd62028f/go.mod h1:sm1tb
197207
github.com/sean-/seed v0.0.0-20170313163322-e2103e2c3529/go.mod h1:DxrIzT+xaE7yg65j358z/aeFdxmN0P9QXhEzd20vsDc=
198208
github.com/shurcooL/sanitized_anchor_name v1.0.0/go.mod h1:1NzhyTcUVG4SuEtjjoZeVRXNmyL/1OwPU0+IJeTBvfc=
199209
github.com/sirupsen/logrus v1.2.0/go.mod h1:LxeOpSwHxABJmUn/MG1IvRgCAasNZTLOkJPxbbu5VWo=
210+
github.com/smartystreets/assertions v0.0.0-20180927180507-b2de0cb4f26d h1:zE9ykElWQ6/NYmHa3jpm/yHnI4xSofP+UP6SpjHcSeM=
200211
github.com/smartystreets/assertions v0.0.0-20180927180507-b2de0cb4f26d/go.mod h1:OnSkiWE9lh6wB0YB77sQom3nweQdgAjqCqsofrRNTgc=
212+
github.com/smartystreets/goconvey v1.6.4 h1:fv0U8FUIMPNf1L9lnHLvLhgicrIVChEkdzIKYqbNC9s=
201213
github.com/smartystreets/goconvey v1.6.4/go.mod h1:syvi0/a8iFYH4r/RixwvyeAJjdLS9QV7WQ/tjFTllLA=
202214
github.com/soheilhy/cmux v0.1.4/go.mod h1:IM3LyeVVIOuxMH7sFAkER9+bJ4dT7Ms6E4xg4kGIyLM=
203215
github.com/spaolacci/murmur3 v0.0.0-20180118202830-f09979ecbc72/go.mod h1:JwIasOWyU6f++ZhiEuf87xNszmSA2myDM2Kzu9HwQUA=
@@ -218,6 +230,7 @@ github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+
218230
github.com/stretchr/objx v0.1.1/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
219231
github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs=
220232
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
233+
github.com/stretchr/testify v1.4.0 h1:2E4SXV/wtOkTonXsotYi4li6zVWxYlZuYNCXe9XRJyk=
221234
github.com/stretchr/testify v1.4.0/go.mod h1:j7eGeouHqKxXV5pUuKE4zz7dFj8WfuZ+81PSLYec5m4=
222235
github.com/subosito/gotenv v1.2.0 h1:Slr1R9HxAlEKefgq5jn9U+DnETlIUa6HfgEzj0g5d7s=
223236
github.com/subosito/gotenv v1.2.0/go.mod h1:N0PQaV/YGNqwC0u51sEeR/aUtSLEXKX9iv69rRypqCw=
@@ -323,6 +336,7 @@ golang.org/x/tools v0.0.0-20190911174233-4f2ddba30aff/go.mod h1:b+2E5dAYhXwXZwtn
323336
golang.org/x/tools v0.0.0-20191012152004-8de300cfc20a/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
324337
golang.org/x/tools v0.0.0-20191112195655-aa38f8e97acc/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo=
325338
golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
339+
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543 h1:E7g+9GITq07hpfrRu66IVDexMakfv52eLZ2CXBWiKr4=
326340
golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0=
327341
google.golang.org/api v0.4.0/go.mod h1:8k5glujaEP+g9n7WNsDg8QP6cUVNI86fCNMcbazEtwE=
328342
google.golang.org/api v0.7.0/go.mod h1:WtwebWUNSVBH/HAw79HIFXZNqEvBhG+Ra+ax0hx3E3M=
@@ -362,6 +376,7 @@ google.golang.org/protobuf v1.25.0 h1:Ejskq+SyPohKW+1uil0JJMtmHCgJPJ/qWTxr8qp+R4
362376
google.golang.org/protobuf v1.25.0/go.mod h1:9JNX74DMeImyA3h4bdi1ymwjUzf21/xIlbajtzgsN7c=
363377
gopkg.in/alecthomas/kingpin.v2 v2.2.6/go.mod h1:FMv+mEhP44yOT+4EoQTLFTRgOQ1FBLkstjWtayDeSgw=
364378
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
379+
gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127 h1:qIbj1fsPNlZgppZ+VLlY7N33q108Sa+fhmuc+sWQYwY=
365380
gopkg.in/check.v1 v1.0.0-20180628173108-788fd7840127/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
366381
gopkg.in/errgo.v2 v2.1.0/go.mod h1:hNsd1EY+bozCKY1Ytp96fpM3vjJbqLJn88ws8XvfDNI=
367382
gopkg.in/ini.v1 v1.51.0 h1:AQvPpx3LzTDM0AjnIRlVFwFFGC+npRopjZxLJj6gdno=

‎tools/internal/utils/url.go

+23
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
package utils
2+
3+
import (
4+
"net/url"
5+
6+
"github.com/pkg/errors"
7+
"github.com/spf13/cobra"
8+
)
9+
10+
func GetRTMPURLFlag(cmd *cobra.Command, flagname string) (url.URL, error) {
11+
urlstr, err := cmd.Flags().GetString(flagname)
12+
if err != nil {
13+
return url.URL{}, errors.Wrap(err, "could not read flag")
14+
}
15+
u, err := url.Parse(urlstr)
16+
if err != nil {
17+
return url.URL{}, errors.Wrap(err, "could not parse url")
18+
}
19+
if u.Scheme != "rtmp" && u.Scheme != "rtmps" {
20+
return url.URL{}, errors.New("invalid url scheme. it must be rtmp:// or rmtps://")
21+
}
22+
return *u, nil
23+
}

0 commit comments

Comments
 (0)