minidrama.go

package minidrama

import (
	"context"
	"github.com/donetkit/wechat/util"
	"strconv"
)

// SingleFileUpload 单文件上传
func (s *MiniDrama) SingleFileUpload(ctx context.Context, in *SingleFileUploadRequest) (out SingleFileUploadResponse, err error) {
	var address string
	if address, err = s.requestAddress(ctx, singleFileUpload); err != nil {
		return
	}
	var (
		fields = []util.MultipartFormField{
			{
				IsFile:    true,
				Fieldname: "media_data",
				Filename:  string(in.MediaData),
			}, {
				IsFile:    false,
				Fieldname: "media_name",
				Value:     []byte(in.MediaName),
			}, {
				IsFile:    false,
				Fieldname: "media_type",
				Value:     []byte(in.MediaType),
			},
		}
		response []byte
	)

	if in.CoverType != "" && in.CoverData != nil {
		fields = append(fields, util.MultipartFormField{
			IsFile:    false,
			Fieldname: "cover_type",
			Value:     []byte(in.CoverType),
		})
		fields = append(fields, util.MultipartFormField{
			IsFile:    true,
			Fieldname: "cover_data",
			Filename:  string(in.CoverData),
		})
	}

	if in.SourceContext != "" {
		fields = append(fields, util.MultipartFormField{
			IsFile:    false,
			Fieldname: "source_context",
			Value:     []byte(in.SourceContext),
		})
	}

	if response, err = util.PostMultipartForm(fields, address); err != nil {
		return
	}
	// 使用通用方法返回错误
	err = util.DecodeWithError(response, &out, "SingleFileUpload")
	return
}

// PullUpload 拉取上传
func (s *MiniDrama) PullUpload(ctx context.Context, in *PullUploadRequest) (out PullUploadResponse, err error) {
	var address string
	if address, err = s.requestAddress(ctx, pullUpload); err != nil {
		return
	}
	var response []byte
	if response, err = util.PostJSONContext(ctx, address, in); err != nil {
		return
	}

	// 使用通用方法返回错误
	err = util.DecodeWithError(response, &out, "PullUpload")
	return
}

// GetTask 查询任务状态
func (s *MiniDrama) GetTask(ctx context.Context, in *GetTaskRequest) (out GetTaskResponse, err error) {
	var address string
	if address, err = s.requestAddress(ctx, getTask); err != nil {
		return
	}

	var response []byte
	if response, err = util.PostJSONContext(ctx, address, in); err != nil {
		return
	}

	// 使用通用方法返回错误
	err = util.DecodeWithError(response, &out, "GetTask")
	return
}

// ApplyUpload 申请分片上传
func (s *MiniDrama) ApplyUpload(ctx context.Context, in *ApplyUploadRequest) (out ApplyUploadResponse, err error) {
	var address string
	if address, err = s.requestAddress(ctx, applyUpload); err != nil {
		return
	}

	var response []byte
	if response, err = util.PostJSONContext(ctx, address, in); err != nil {
		return
	}

	// 使用通用方法返回错误
	err = util.DecodeWithError(response, &out, "ApplyUpload")
	return
}

// UploadPart 上传分片
// Content-Type 需要指定为 multipart/form-data; boundary=<delimiter>，<箭头括号>表示必须替换为有效值的变量。
func (s *MiniDrama) UploadPart(ctx context.Context, in *UploadPartRequest) (out UploadPartResponse, err error) {
	var address string
	if address, err = s.requestAddress(ctx, uploadPart); err != nil {
		return
	}

	var (
		fields = []util.MultipartFormField{
			{
				IsFile:    true,
				Fieldname: "data",
				Filename:  string(in.Data),
			}, {
				IsFile:    false,
				Fieldname: "upload_id",
				Value:     []byte(in.UploadID),
			}, {
				IsFile:    false,
				Fieldname: "part_number",
				Value:     []byte(strconv.Itoa(in.PartNumber)),
			}, {
				IsFile:    false,
				Fieldname: "resource_type",
				Value:     []byte(strconv.Itoa(in.PartNumber)),
			},
		}
		response []byte
	)
	if response, err = util.PostMultipartForm(fields, address); err != nil {
		return
	}

	// 使用通用方法返回错误
	err = util.DecodeWithError(response, &out, "UploadPart")
	return
}

// CommitUpload 确认上传
func (s *MiniDrama) CommitUpload(ctx context.Context, in *CommitUploadRequest) (out CommitUploadResponse, err error) {
	var address string
	if address, err = s.requestAddress(ctx, commitUpload); err != nil {
		return
	}

	var response []byte
	if response, err = util.PostJSONContext(ctx, address, in); err != nil {
		return
	}

	// 使用通用方法返回错误
	err = util.DecodeWithError(response, &out, "CommitUpload")
	return
}

// ListMedia 获取媒体列表
func (s *MiniDrama) ListMedia(ctx context.Context, in *ListMediaRequest) (out ListMediaResponse, err error) {
	var address string
	if address, err = s.requestAddress(ctx, listMedia); err != nil {
		return
	}

	var response []byte
	if response, err = util.PostJSONContext(ctx, address, in); err != nil {
		return
	}

	// 使用通用方法返回错误
	err = util.DecodeWithError(response, &out, "ListMedia")
	return
}

// GetMedia 获取媒资详细信息
func (s *MiniDrama) GetMedia(ctx context.Context, in *GetMediaRequest) (out GetMediaResponse, err error) {
	var address string
	if address, err = s.requestAddress(ctx, getMedia); err != nil {
		return
	}

	var response []byte
	if response, err = util.PostJSONContext(ctx, address, in); err != nil {
		return
	}

	// 使用通用方法返回错误
	err = util.DecodeWithError(response, &out, "GetMedia")
	return
}

// GetMediaLink 获取媒资播放链接
func (s *MiniDrama) GetMediaLink(ctx context.Context, in *GetMediaLinkRequest) (out GetMediaLinkResponse, err error) {
	var address string
	if address, err = s.requestAddress(ctx, getMediaLink); err != nil {
		return
	}

	var response []byte
	if response, err = util.PostJSONContext(ctx, address, in); err != nil {
		return
	}

	// 使用通用方法返回错误
	err = util.DecodeWithError(response, &out, "GetMediaLink")
	return
}

// DeleteMedia 删除媒体
func (s *MiniDrama) DeleteMedia(ctx context.Context, in *DeleteMediaRequest) (out DeleteMediaResponse, err error) {
	var address string
	if address, err = s.requestAddress(ctx, deleteMedia); err != nil {
		return
	}

	var response []byte
	if response, err = util.PostJSONContext(ctx, address, in); err != nil {
		return
	}

	// 使用通用方法返回错误
	err = util.DecodeWithError(response, &out, "DeleteMedia")
	return
}

// AuditDrama 审核剧本
func (s *MiniDrama) AuditDrama(ctx context.Context, in *AuditDramaRequest) (out AuditDramaResponse, err error) {
	var address string
	if address, err = s.requestAddress(ctx, auditDrama); err != nil {
		return
	}

	var response []byte
	if response, err = util.PostJSONContext(ctx, address, in); err != nil {
		return
	}

	// 使用通用方法返回错误
	err = util.DecodeWithError(response, &out, "AuditDrama")
	return
}

// ListDramas 获取剧目列表
func (s *MiniDrama) ListDramas(ctx context.Context, in *ListDramasRequest) (out ListDramasResponse, err error) {
	var address string
	if address, err = s.requestAddress(ctx, listDramas); err != nil {
		return
	}

	var response []byte
	if response, err = util.PostJSONContext(ctx, address, in); err != nil {
		return
	}

	// 使用通用方法返回错误
	err = util.DecodeWithError(response, &out, "ListDramas")
	return
}

// GetDrama 获取剧目信息
func (s *MiniDrama) GetDrama(ctx context.Context, in *GetDramaRequest) (out GetDramaResponse, err error) {
	var address string
	if address, err = s.requestAddress(ctx, getDrama); err != nil {
		return
	}

	var response []byte
	if response, err = util.PostJSONContext(ctx, address, in); err != nil {
		return
	}
	// 使用通用方法返回错误
	err = util.DecodeWithError(response, &out, "GetDrama")
	return
}

// GetCdnUsageData 查询 CDN 用量数据
func (s *MiniDrama) GetCdnUsageData(ctx context.Context, in *GetCdnUsageDataRequest) (out GetCdnUsageDataResponse, err error) {
	var address string
	if address, err = s.requestAddress(ctx, getCdnUsageData); err != nil {
		return
	}

	var response []byte
	if response, err = util.PostJSONContext(ctx, address, in); err != nil {
		return
	}
	// 使用通用方法返回错误
	err = util.DecodeWithError(response, &out, "GetCdnUsageData")
	return
}

// GetCdnLogs 查询 CDN 日志
func (s *MiniDrama) GetCdnLogs(ctx context.Context, in *GetCdnLogsRequest) (out GetCdnLogsResponse, err error) {
	var address string
	if address, err = s.requestAddress(ctx, getCdnLogs); err != nil {
		return
	}

	var response []byte
	if response, err = util.PostJSONContext(ctx, address, in); err != nil {
		return
	}
	// 使用通用方法返回错误
	err = util.DecodeWithError(response, &out, "GetCdnLogs")
	return
}

// requestAddress 请求地址
func (s *MiniDrama) requestAddress(ctx context.Context, url string) (string, error) {
	accessToken, err := s.ctx.GetAccessToken(ctx)
	if err != nil {
		return "", err
	}
	return url + accessToken, nil
}