go-library/vendor/github.com/upyun/go-sdk/v3/upyun/process.go

package upyun

import (
	"bytes"
	"encoding/json"
	"fmt"
	"io/ioutil"
	"net/http"
	"path"
	"strings"
	"time"
)

type CommitTasksConfig struct {
	AppName   string
	Accept    string
	Source    string
	NotifyUrl string
	Tasks     []interface{}
}

type LiveauditCreateTask struct {
	Source    string
	SaveAs    string
	NotifyUrl string
	Interval  string
	Resize    string
}

type LiveauditCancelTask struct {
	TaskId string
}

type SyncCommonTask struct {
	Kwargs  map[string]interface{}
	TaskUri string
}

func (up *UpYun) CommitTasks(config *CommitTasksConfig) (taskIds []string, err error) {
	b, err := json.Marshal(config.Tasks)
	if err != nil {
		return nil, err
	}

	kwargs := map[string]string{
		"app_name":   config.AppName,
		"tasks":      base64ToStr(b),
		"notify_url": config.NotifyUrl,

		// for naga
		"source": config.Source,
	}
	if config.Accept != "" {
		kwargs["accept"] = config.Accept
	}

	err = up.doProcessRequest("POST", "/pretreatment/", kwargs, &taskIds)
	return
}

func (up *UpYun) GetProgress(taskIds []string) (result map[string]int, err error) {
	kwargs := map[string]string{
		"task_ids": strings.Join(taskIds, ","),
	}
	v := map[string]map[string]int{}
	err = up.doProcessRequest("GET", "/status/", kwargs, &v)
	if err != nil {
		return
	}

	if r, ok := v["tasks"]; ok {
		return r, err
	}
	return nil, fmt.Errorf("no tasks")
}

func (up *UpYun) GetResult(taskIds []string) (result map[string]interface{}, err error) {
	kwargs := map[string]string{
		"task_ids": strings.Join(taskIds, ","),
	}
	v := map[string]map[string]interface{}{}
	err = up.doProcessRequest("GET", "/result/", kwargs, &v)
	if err != nil {
		return
	}

	if r, ok := v["tasks"]; ok {
		return r, err
	}
	return nil, fmt.Errorf("no tasks")
}

func (up *UpYun) doProcessRequest(method, uri string,
	kwargs map[string]string, v interface{}) error {
	if _, ok := kwargs["service"]; !ok {
		kwargs["service"] = up.Bucket
	}

	if method == "GET" {
		uri = addQueryToUri(uri, kwargs)
	}

	headers := make(map[string]string)
	headers["Date"] = makeRFC1123Date(time.Now())
	headers["Content-Type"] = "application/x-www-form-urlencoded"
	if up.deprecated {
		headers["Authorization"] = up.MakeProcessAuth(kwargs)
	} else {
		headers["Authorization"] = up.MakeUnifiedAuth(&UnifiedAuthConfig{
			Method:  method,
			Uri:     uri,
			DateStr: headers["Date"],
		})
	}

	var resp *http.Response
	var err error
	endpoint := up.doGetEndpoint("p0.api.upyun.com")
	rawurl := fmt.Sprintf("http://%s%s", endpoint, uri)
	switch method {
	case "GET":
		resp, err = up.doHTTPRequest(method, rawurl, headers, nil)
	case "POST":
		payload := encodeQueryToPayload(kwargs)
		resp, err = up.doHTTPRequest(method, rawurl, headers, bytes.NewBufferString(payload))
	default:
		return fmt.Errorf("Unknown method")
	}

	if err != nil {
		return errorOperation("process", err)
	}

	b, err := ioutil.ReadAll(resp.Body)
	resp.Body.Close()
	if err != nil {
		return errorOperation("process read body", err)
	}

	return json.Unmarshal(b, v)
}

func (up *UpYun) CommitSyncTasks(commitTask interface{}) (result map[string]interface{}, err error) {
	var kwargs map[string]interface{}
	var uri string
	var payload string

	switch taskConfig := commitTask.(type) {
	case LiveauditCreateTask:
		kwargs = map[string]interface{}{
			"source":     taskConfig.Source,
			"save_as":    taskConfig.SaveAs,
			"notify_url": taskConfig.NotifyUrl,
			"service":    up.Bucket,
		}
		if taskConfig.Interval != "" {
			kwargs["interval"] = taskConfig.Interval
		}
		if taskConfig.Resize != "" {
			kwargs["resize"] = taskConfig.Resize
		}
		uri = path.Join("/", up.Bucket, "/liveaudit/create")

	case LiveauditCancelTask:
		kwargs = map[string]interface{}{
			"task_id": taskConfig.TaskId,
			"service": up.Bucket,
		}
		uri = path.Join("/", up.Bucket, "/liveaudit/cancel")

	case SyncCommonTask:
		kwargs = taskConfig.Kwargs
		uri = path.Join("/", up.Bucket, taskConfig.TaskUri)

	default:
		err = fmt.Errorf("don't match any task")
		return nil, err
	}

	if _, exist := kwargs["service"]; !exist {
		kwargs["service"] = up.Bucket
	}

	body, err := json.Marshal(kwargs)
	if err != nil {
		return nil, fmt.Errorf("can't encode the json")
	}
	payload = string(body)
	return up.doSyncProcessRequest("POST", uri, payload)
}

func (up *UpYun) doSyncProcessRequest(method, uri string, payload string) (map[string]interface{}, error) {
	headers := make(map[string]string)
	headers["Date"] = makeRFC1123Date(time.Now())
	headers["Content-Type"] = "application/json"
	headers["Content-MD5"] = md5Str(payload)
	headers["Authorization"] = up.MakeUnifiedAuth(&UnifiedAuthConfig{
		Method:     method,
		Uri:        uri,
		DateStr:    headers["Date"],
		ContentMD5: headers["Content-MD5"],
	})

	var resp *http.Response
	var err error
	endpoint := up.doGetEndpoint("p1.api.upyun.com")
	rawurl := fmt.Sprintf("http://%s%s", endpoint, uri)
	switch method {
	case "POST":
		resp, err = up.doHTTPRequest(method, rawurl, headers, strings.NewReader(payload))
	default:
		return nil, fmt.Errorf("Unknown method")
	}
	if err != nil {
		return nil, errorOperation("sync process", err)
	}

	b, err := ioutil.ReadAll(resp.Body)
	resp.Body.Close()
	if err != nil {
		return nil, err
	}

	var v map[string]interface{}
	err = json.Unmarshal(b, &v)
	if err != nil {
		fmt.Println("can't unmarshal the data", string(b))
	}
	return v, err
}