LBRYCommunity/lbry.go
1
0
Fork 0
Code Issues 5 Pull requests Projects Releases 46 Packages Wiki Activity

add support for channel updates

Browse source
This commit is contained in:
Niko Storni 2018-05-22 17:47:03 -04:00
parent d0cfcaa3e5
commit c8eebbfae1
No known key found for this signature in database
GPG key ID: F37FE63398800368
2 changed files with 18 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
cmd/selfsync.go
View file

@ -20,8 +20,8 @@ import (
func init() { func init() {
var selfSyncCmd = &cobra.Command{ var selfSyncCmd = &cobra.Command{
Use: "selfsync <youtube_api_key> <auth_token>", Use: "selfsync <youtube_api_key> <auth_token> [<sync_status>]",
Args: cobra.RangeArgs(2, 2), Args: cobra.RangeArgs(2, 3),
Short: "Publish youtube channels into LBRY network automatically.", Short: "Publish youtube channels into LBRY network automatically.",
Run: selfSync, Run: selfSync,
} }
@ -30,6 +30,7 @@ func init() {
selfSyncCmd.Flags().BoolVar(&takeOverExistingChannel, "takeover-existing-channel", false, "If channel exists and we don't own it, take over the channel") selfSyncCmd.Flags().BoolVar(&takeOverExistingChannel, "takeover-existing-channel", false, "If channel exists and we don't own it, take over the channel")
selfSyncCmd.Flags().IntVar(&limit, "limit", 0, "limit the amount of channels to sync") selfSyncCmd.Flags().IntVar(&limit, "limit", 0, "limit the amount of channels to sync")
selfSyncCmd.Flags().BoolVar(&skipSpaceCheck, "skip-space-check", false, "Do not perform free space check on startup") selfSyncCmd.Flags().BoolVar(&skipSpaceCheck, "skip-space-check", false, "Do not perform free space check on startup")
selfSyncCmd.Flags().BoolVar(&skipSpaceCheck, "update", false, "Update previously synced channels instead of syncing new ones")
RootCmd.AddCommand(selfSyncCmd) RootCmd.AddCommand(selfSyncCmd)
} }
@ -48,9 +49,11 @@ type APIYoutubeChannel struct {
} }
//PoC //PoC
func fetchChannels(authToken string) ([]APIYoutubeChannel, error) { func fetchChannels(authToken string, status string) ([]APIYoutubeChannel, error) {
url := "https://api.lbry.io/yt/jobs" url := "https://api.lbry.io/yt/jobs"
payload := strings.NewReader("------WebKitFormBoundary7MA4YWxkTrZu0gW\r\nContent-Disposition: form-data; name=\"auth_token\"\r\n\r\n" + authToken + "\r\n------WebKitFormBoundary7MA4YWxkTrZu0gW--") payload := strings.NewReader("------WebKitFormBoundary7MA4YWxkTrZu0gW\r\n" +
"Content-Disposition: form-data; name=\"auth_token\"\r\n\r\n" + authToken + "\r\n------WebKitFormBoundary7MA4YWxkTrZu0gW\r\n" +
"Content-Disposition: form-data; name=\"sync_status\"\r\n\r\n" + status + "\r\n------WebKitFormBoundary7MA4YWxkTrZu0gW--")
req, _ := http.NewRequest("POST", url, payload) req, _ := http.NewRequest("POST", url, payload)
req.Header.Add("content-type", "multipart/form-data; boundary=----WebKitFormBoundary7MA4YWxkTrZu0gW") req.Header.Add("content-type", "multipart/form-data; boundary=----WebKitFormBoundary7MA4YWxkTrZu0gW")
res, _ := http.DefaultClient.Do(req) res, _ := http.DefaultClient.Do(req)
@ -131,6 +134,15 @@ func selfSync(cmd *cobra.Command, args []string) {
ytAPIKey := args[0] ytAPIKey := args[0]
authToken := args[1] authToken := args[1]
status := StatusQueued
if len(args) > 2 {
if util.InSlice(args[2], SyncStatuses) {
status = args[2]
} else {
log.Errorf("status must be one of the following: %v\n", SyncStatuses)
return
}
}
if stopOnError && maxTries != defaultMaxTries { if stopOnError && maxTries != defaultMaxTries {
log.Errorln("--stop-on-error and --max-tries are mutually exclusive") log.Errorln("--stop-on-error and --max-tries are mutually exclusive")
return return
@ -144,7 +156,7 @@ func selfSync(cmd *cobra.Command, args []string) {
log.Errorln("setting --limit less than 0 (unlimited) doesn't make sense") log.Errorln("setting --limit less than 0 (unlimited) doesn't make sense")
return return
} }
channelsToSync, err := fetchChannels(authToken) channelsToSync, err := fetchChannels(authToken, status)
if err != nil { if err != nil {
util.SendToSlackError("failed to fetch channels: %v", err) util.SendToSlackError("failed to fetch channels: %v", err)
return return
@ -193,7 +205,6 @@ func selfSync(cmd *cobra.Command, args []string) {
fatalErrors := []string{ fatalErrors := []string{
"default_wallet already exists", "default_wallet already exists",
"WALLET HAS NOT BEEN MOVED TO THE WALLET BACKUP DIR", "WALLET HAS NOT BEEN MOVED TO THE WALLET BACKUP DIR",
"Initial wallet setup failed! Manual Intervention is required.",
} }
if util.InSliceContains(err.Error(), fatalErrors) { if util.InSliceContains(err.Error(), fatalErrors) {
util.SendToSlackError("@Nikooo777 this requires manual intervention! Exiting...") util.SendToSlackError("@Nikooo777 this requires manual intervention! Exiting...")

2
cmd/selfsync_test.go
View file

@ -6,7 +6,7 @@ import (
) )
func TestFetchChannels(t *testing.T) { func TestFetchChannels(t *testing.T) {
res, err := fetchChannels("620280") res, err := fetchChannels("620280", StatusFailed)
if err != nil { if err != nil {
t.Error(err) t.Error(err)
} }