allow specifying KEY_FILE and CLIENT_ID_FILE environment variables

This commit is contained in:
Christopher Tarry 2022-09-17 23:29:00 -04:00
parent 77b6b192bc
commit bc3ab4bbcb
1 changed files with 12 additions and 3 deletions

15
main.go
View File

@ -67,6 +67,9 @@ download [id] - prints the MPD url the video is available at and returns the mp4
return
}
key, _ := os.ReadFile(os.Getenv("KEY_FILE"))
clientID, _ := os.ReadFile(os.Getenv("CLIENT_ID_FILE"))
client := hulu.NewDefaultClient(huluSession, huluGUID)
w := tabwriter.NewWriter(os.Stdout, 8, 8, 0, '\t', 0)
defer w.Flush()
@ -138,9 +141,15 @@ download [id] - prints the MPD url the video is available at and returns the mp4
panic(err)
}
cdm, err := widevine.NewDefaultCDM(initData)
if err != nil {
panic(err)
var cdm widevine.CDM
if len(key) > 0 && len(clientID) > 0 {
if cdm, err = widevine.NewCDM(key, clientID, initData); err != nil {
panic(err)
}
} else {
if cdm, err = widevine.NewDefaultCDM(initData); err != nil {
panic(err)
}
}
licenseRequest, err := cdm.GetLicenseRequest()