add acc-lc dl

utils/runv3/key/key.go

@@ -0,0 +1,73 @@
+package wv
+
+import (
+	"context"
+	"encoding/base64"
+	"encoding/hex"
+	"github.com/gospider007/requests"
+	"log/slog"
+	"main/utils/runv3/cdm"
+)
+
+type Key struct {
+	ReqCli        *requests.Client
+	BeforeRequest func(cl *requests.Client, preCtx context.Context, method string, href string, options ...requests.RequestOption) (resp *requests.Response, err error)
+	AfterRequest  func(*requests.Response) ([]byte, error)
+}
+
+func (w *Key) CdmInit() {
+	wv.InitConstants()
+}
+func (w *Key) GetKey(ctx context.Context, licenseServerURL string, PSSH string, headers map[string][]string) (string, []byte, error) {
+	initData, err := base64.StdEncoding.DecodeString(PSSH)
+	var keybt []byte
+	if err != nil {
+		slog.Error("pssh decode error: %v", err)
+		return "", keybt, err
+	}
+	cdm, err := wv.NewDefaultCDM(initData)
+	if err != nil {
+		slog.Error("cdm init error: %v", err)
+		return "", keybt, err
+	}
+	licenseRequest, err := cdm.GetLicenseRequest()
+	if err != nil {
+		slog.Error("license request error: %v", err)
+		return "", keybt, err
+	}
+	var response *requests.Response
+	if w.BeforeRequest != nil {
+		response, err = w.BeforeRequest(w.ReqCli, ctx, "post", licenseServerURL, requests.RequestOption{
+			Data: licenseRequest,
+		})
+	} else {
+		response, err = w.ReqCli.Request(nil, "post", licenseServerURL, requests.RequestOption{
+			Data: licenseRequest,
+		})
+	}
+
+	if err != nil {
+		slog.Error("license request error: %s", err)
+		return "", keybt, err
+	}
+	var licenseResponse []byte
+	if w.AfterRequest != nil {
+		licenseResponse, err = w.AfterRequest(response)
+		if err != nil {
+			return "", keybt, err
+		}
+	} else {
+		licenseResponse = response.Content()
+	}
+	keys, err := cdm.GetLicenseKeys(licenseRequest, licenseResponse)
+	command := ""
+
+	for _, key := range keys {
+		if key.Type == wv.License_KeyContainer_CONTENT {
+			// command += "--key " + hex.EncodeToString(key.ID) + ":" + hex.EncodeToString(key.Value)
+			command += hex.EncodeToString(key.Value)
+			keybt = key.Value
+		}
+	}
+	return command, keybt, nil
+}