Update video logic.

Author: deadblue

agent.go

@@ -20,6 +20,9 @@ type Agent struct {
 
 	// Common parameters
 	common types.CommonParams
+
+	// Is agent use web credential?
+	isWeb bool
 }
 
 // Default creates an Agent with default settings.

internal/protocol/cookie.go

@@ -1,5 +1,7 @@
 package protocol
 
+import "strings"
+
 const (
 	CookieUrl = "https://115.com"
 
@@ -14,3 +16,8 @@ var (
 		".anxia.com",
 	}
 )
+
+func IsWebCredential(uid string) bool {
+	parts := strings.Split(uid, "_")
+	return len(parts) == 3 && parts[1][0] == 'A'
+}

login.go

@@ -35,7 +35,7 @@ func (a *Agent) CredentialImport(cr *Credential) (err error) {
 		protocol.CookieNameSEID: cr.SEID,
 	}
 	a.llc.ImportCookies(cookies, protocol.CookieDomains...)
-	return a.afterSignIn()
+	return a.afterSignIn(cr.UID)
 }
 
 // CredentialExport exports current credentials for future-use.
@@ -46,11 +46,12 @@ func (a *Agent) CredentialExport(cr *Credential) {
 	cr.SEID = cookies[protocol.CookieNameSEID]
 }
 
-func (a *Agent) afterSignIn() (err error) {
+func (a *Agent) afterSignIn(uid string) (err error) {
 	// Call UploadInfo API to get userId and userKey
 	spec := (&api.UploadInfoSpec{}).Init()
 	if err = a.llc.CallApi(spec, context.Background()); err == nil {
 		a.common.SetUserInfo(spec.Result.UserId, spec.Result.UserKey)
+		a.isWeb = protocol.IsWebCredential(uid)
 	}
 	return
 }

lowlevel/errors/media.go

@@ -0,0 +1,7 @@
+package errors
+
+import "errors"
+
+var (
+	ErrUnsupportedPlatform = errors.New("unsupported platform")
+)

media.go

@@ -45,6 +45,9 @@ type VideoTicket struct {
 
 // VideoCreateTicket creates a PlayTicket to play the cloud video.
 func (a *Agent) VideoCreateTicket(pickcode string, ticket *VideoTicket) (err error) {
+	if !a.isWeb {
+		return errors.ErrUnsupportedPlatform
+	}
 	spec := (&api.VideoPlayWebSpec{}).Init(pickcode)
 	if err = a.llc.CallApi(spec, context.Background()); err != nil {
 		return
@@ -63,6 +66,9 @@ func (a *Agent) VideoCreateTicket(pickcode string, ticket *VideoTicket) (err err
 			ticket.Url = video.PlayUrl
 		}
 	}
+	ticket.Headers = map[string]string{
+		"User-Agent": a.llc.GetUserAgent(),
+	}
 	return
 }
 

qrcode.go

@@ -85,7 +85,7 @@ func (a *Agent) qrcodeSignIn(session *QrcodeSession) (err error) {
 	if err = a.llc.CallApi(spec, context.Background()); err != nil {
 		return
 	}
-	return a.afterSignIn()
+	return a.afterSignIn(spec.Result.Cookie.UID)
 }
 
 // QrcodePoll polls the session state, and automatically sin