api_drive_spaces.go

/*
 * Copyright 2020 zhaoyunxing.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package dingtalk

import (
	"fmt"
	"net/http"
	"net/url"
	"strconv"

	"github.com/zhaoyunxing92/dingtalk/v2/constant"
	"github.com/zhaoyunxing92/dingtalk/v2/constant/policy"
	"github.com/zhaoyunxing92/dingtalk/v2/constant/spaces"
	"github.com/zhaoyunxing92/dingtalk/v2/request"
	"github.com/zhaoyunxing92/dingtalk/v2/response"
)

// CreateDriveSpaces 新建空间
func (ding *DingTalk) CreateDriveSpaces(name, unionId string) (rsp response.CreateDriveSpaces, err error) {
	return rsp, ding.Request(http.MethodPost, constant.CreateDriveSpacesKey, nil,
		request.NewCreateDriveSpaces(name, unionId), &rsp)
}

// DeleteDriveSpaces 删除空间
func (ding *DingTalk) DeleteDriveSpaces(spaceId, unionId string) (rsp response.Response, err error) {
	query := url.Values{}
	query.Set("unionId", unionId)

	return rsp, ding.Request(http.MethodDelete, fmt.Sprintf(constant.DeleteDriveSpacesKey, spaceId), query,
		nil, &rsp)
}

// GetDriveSpaces 获取空间列表
func (ding *DingTalk) GetDriveSpaces(unionId string, spaceType spaces.SpaceType, token string,
	size int) (rsp response.GetDriveSpaces, err error,
) {
	query := url.Values{}
	query.Set("spaceType", string(spaceType))
	query.Set("unionId", unionId)
	query.Set("nextToken", token)
	query.Set("maxResults", strconv.Itoa(size))

	return rsp, ding.Request(http.MethodGet, constant.GetDriveSpacesKey, query, nil, &rsp)
}

// GetDriveSpacesInfo 获取空间信息
func (ding *DingTalk) GetDriveSpacesInfo(spaceId, unionId string) (rsp response.CreateDriveSpaces, err error) {
	query := url.Values{}
	query.Set("unionId", unionId)

	return rsp, ding.Request(http.MethodGet, fmt.Sprintf(constant.GetDriveSpacesInfoKey, spaceId), query, nil,
		&rsp)
}

// GetDriveSpacesFiles 查询文件（夹）列表
// order 取值：
// createTimeAsc：按创建时间升序
// createTimeDesc：按创建时间降序
// modifyTimeAsc：按修改时间升序
// modifyTimeDesc：按修改时间降序
// nameAsc：按名称升序
// nameDesc：按名称降序
// sizeAsc：按大小升序
// sizeDesc：按大小降序
// 默认值：createTimeDesc
func (ding *DingTalk) GetDriveSpacesFiles(res *request.GetDriveSpacesFiles) (rsp response.GetDriveSpacesFiles,
	err error,
) {
	query := url.Values{}
	query.Set("parentId", res.ParentId)
	query.Set("unionId", res.UnionId)
	query.Set("nextToken", res.Token)
	query.Set("maxResults", strconv.Itoa(res.Size))
	query.Set("orderType", res.OrderType)

	return rsp, ding.Request(http.MethodGet, fmt.Sprintf(constant.GetDriveSpacesFilesKey, res.SpaceId), query,
		nil, &rsp)
}

// GetDriveSpacesFileInfo 查询文件（夹）信息
func (ding *DingTalk) GetDriveSpacesFileInfo(spaceId, fileId, unionId string) (rsp response.GetDriveSpacesFileInfo,
	err error) {
	query := url.Values{}
	query.Set("unionId", unionId)

	return rsp, ding.Request(http.MethodGet, fmt.Sprintf(constant.GetDriveSpacesFileInfoKey, spaceId, fileId),
		query, nil, &rsp)
}

// CreateDriveSpacesFiles 添加文件（夹）
func (ding *DingTalk) CreateDriveSpacesFiles(res *request.CreateDriveSpacesFiles) (rsp response.GetDriveSpacesFileInfo,
	err error,
) {
	return rsp, ding.Request(http.MethodPost, fmt.Sprintf(constant.CreateDriveSpacesFileKey, res.SpaceId), nil,
		res, &rsp)
}

// DeleteDriveSpacesFiles 删除文件（夹）
func (ding *DingTalk) DeleteDriveSpacesFiles(spaceId, fileId, unionId string,
	policy policy.DeletePolicy) (rsp response.Response, err error,
) {
	query := url.Values{}
	query.Set("unionId", unionId)
	query.Set("deletePolicy", string(policy))

	return rsp, ding.Request(http.MethodDelete, fmt.Sprintf(constant.DeleteDriveSpacesFilesKey, spaceId, fileId), query,
		nil, &rsp)
}

// MoveDriveSpacesFiles 移动文件（夹）
func (ding *DingTalk) MoveDriveSpacesFiles(res *request.MoveDriveSpacesFiles) (rsp response.GetDriveSpacesFileInfo,
	err error,
) {
	return rsp, ding.Request(http.MethodPost, fmt.Sprintf(constant.MoveDriveSpacesFilesKey, res.SpaceId, res.FileId),
		nil, res, &rsp)
}

// RenameDriveSpacesFiles 修改文件（夹）名
func (ding *DingTalk) RenameDriveSpacesFiles(spaceId, fileId, newFileName,
	unionId string) (rsp response.GetDriveSpacesFileInfo, err error,
) {
	return rsp, ding.Request(http.MethodPost, fmt.Sprintf(constant.RenameDriveSpacesFilesKey, spaceId, fileId),
		nil, request.NewRenameDriveSpacesFiles(newFileName, unionId), &rsp)
}

// GetDriveSpacesFilesDownloadInfo 获取文件下载信息
func (ding *DingTalk) GetDriveSpacesFilesDownloadInfo(spaceId, fileId,
	unionId string) (rsp response.GetDriveSpacesFileInfo, err error,
) {
	query := url.Values{}
	query.Set("unionId", unionId)
	return rsp, ding.Request(http.MethodGet, fmt.Sprintf(constant.GetDriveSpacesFilesDownloadInfoKey, spaceId, fileId),
		query, nil, &rsp)
}

// GetDriveSpacesFilesUploadInfo 获取文件上传信息
func (ding *DingTalk) GetDriveSpacesFilesUploadInfo(res *request.GetDriveSpacesFilesUploadInfo) (rsp response.GetDriveSpacesFileInfo, err error) {
	if err = validate(res); err != nil {
		return response.GetDriveSpacesFileInfo{}, err
	}

	query := url.Values{}
	query.Set("md5", res.Md5)
	query.Set("unionId", res.UnionId)
	query.Set("fileName", res.FileName)
	query.Set("fileSize", strconv.Itoa(res.FileSize))

	if len(res.ConflictPolicy) >= 0 {
		query.Set("addConflictPolicy", res.ConflictPolicy)
	}

	if len(res.MediaId) >= 0 {
		query.Set("mediaId", res.MediaId)
	}

	return rsp, ding.Request(http.MethodPost, fmt.Sprintf(constant.GetDriveSpacesFilesUploadInfoKey, res.SpaceId, res.ParentId),
		query, nil, &rsp)
}

// GetDriveSpacesFilesPermissions 获取权限列表
func (ding *DingTalk) GetDriveSpacesFilesPermissions(spaceId, fileId, unionId string) (rsp response.DriveSpacesFilesPermissions,
	err error,
) {
	query := url.Values{}
	query.Set("unionId", unionId)

	return rsp, ding.Request(http.MethodGet, fmt.Sprintf(constant.GetDriveSpacesFilesPermissionsKey, spaceId, fileId),
		query, nil, &rsp)
}

// AddDriveSpacesFilesPermissions 添加权限
func (ding *DingTalk) AddDriveSpacesFilesPermissions(res *request.AddDriveSpacesFilesPermissions) (rsp response.DriveSpacesFilesPermissions,
	err error,
) {
	return rsp, ding.Request(http.MethodPost, fmt.Sprintf(constant.AddDriveSpacesFilesPermissionsKey, res.SpaceId, res.FileId),
		nil, res, &rsp)
}