package com.lego.web.controller.system;


import cn.dev33.satoken.annotation.SaCheckPermission;
import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.util.ObjectUtil;
import com.aliyun.vod.upload.impl.UploadVideoImpl;
import com.aliyun.vod.upload.req.UploadStreamRequest;
import com.aliyun.vod.upload.resp.UploadStreamResponse;
import com.lego.common.annotation.Log;
import com.lego.common.core.controller.BaseController;
import com.lego.common.core.domain.PageQuery;
import com.lego.common.core.domain.R;
import com.lego.common.core.page.TableDataInfo;
import com.lego.common.core.validate.QueryGroup;
import com.lego.common.enums.BusinessType;
import com.lego.common.utils.EncryptUtils;
import com.lego.common.utils.StringUtils;
import com.lego.oss.config.AppConfig;
import com.lego.system.domain.SysOss;
import com.lego.system.domain.bo.SysOssBo;
import com.lego.system.domain.vo.SysOssVo;
import com.lego.system.mapper.SysOssMapper;
import com.lego.system.service.ISysOssService;
import lombok.RequiredArgsConstructor;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.MediaType;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

import javax.servlet.http.HttpServletResponse;
import javax.validation.constraints.NotEmpty;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStream;
import java.net.URL;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * 文件上传 控制层
 *
 * @author Lion Li
 */
@Validated
@RequiredArgsConstructor
@RestController
@RequestMapping("/system/oss")
public class SysOssController extends BaseController {

    private final ISysOssService iSysOssService;
    private final SysOssMapper baseMapper;
    @Autowired
    public AppConfig config;

    /**
     * 查询OSS对象存储列表
     */
    @SaCheckPermission("system:oss:list")
    @GetMapping("/list")
    public TableDataInfo<SysOssVo> list(@Validated(QueryGroup.class) SysOssBo bo, PageQuery pageQuery) {
        return iSysOssService.queryPageList(bo, pageQuery);
    }

    /**
     * 查询OSS对象基于id串
     *
     * @param ossIds OSS对象ID串
     */
    @SaCheckPermission("system:oss:list")
    @GetMapping("/listByIds/{ossIds}")
    public R<List<SysOssVo>> listByIds(@NotEmpty(message = "主键不能为空")
                                       @PathVariable Long[] ossIds) {
        List<SysOssVo> list = iSysOssService.listByIds(Arrays.asList(ossIds));
        return R.ok(list);
    }

    /**
     * 上传OSS对象存储
     *
     * @param file 文件
     */
    @SaCheckPermission("system:oss:upload")
    @Log(title = "OSS对象存储", businessType = BusinessType.INSERT)
    @PostMapping(value = "/upload", consumes = MediaType.MULTIPART_FORM_DATA_VALUE)
    public R<Map<String, String>> upload(@RequestPart("file") MultipartFile file) {
        if (ObjectUtil.isNull(file)) {
            return R.fail("上传文件不能为空");
        }
        SysOssVo oss = iSysOssService.upload(file);
        Map<String, String> map = new HashMap<>(2);
        map.put("url", oss.getUrl());
        map.put("fileName", oss.getOriginalName());
        map.put("ossId", oss.getOssId().toString());
        return R.ok(map);
    }

    /**
     * 同步Oss至阿里云
     *
     */
    @GetMapping("/sync")
    public R<String> sync() {
        List<SysOssVo> list = iSysOssService.list();
        String accessKeyId = config.getAliyunAccessKeyId();
        String accessKeySecret = config.getAliyunAccessKeySecret();
        for(SysOssVo vo : list){
            if(StringUtils.isBlank(vo.getVideoId())) {
                if (".mp4".equals(vo.getFileSuffix()) || ".avi".equals(vo.getFileSuffix()) || ".mov".equals(vo.getFileSuffix()) || ".flv".equals(vo.getFileSuffix())) {
                    try {
                        InputStream inputStream = new URL(vo.getUrl()).openStream();
                        ByteArrayOutputStream buffer = new ByteArrayOutputStream();
                        byte[] data = new byte[1024];
                        int nRead;
                        while ((nRead = inputStream.read(data, 0, data.length)) != -1) {
                            buffer.write(data, 0, nRead);
                        }
                        buffer.flush();
                        byte[] byteArray = EncryptUtils.decryptByAes(buffer.toByteArray());
                        UploadStreamRequest request = new UploadStreamRequest(accessKeyId, accessKeySecret, vo.getOriginalName(), vo.getFileName(), new ByteArrayInputStream(byteArray));
                        /* 是否使用默认水印(可选)，指定模板组ID时，根据模板组配置确定是否使用默认水印*/
                        //request.setShowWaterMark(true);
                        /* 自定义消息回调设置，参数说明参考文档 https://help.aliyun.com/document_detail/86952.html#UserData */
                        //request.setUserData(""{\"Extend\":{\"test\":\"www\",\"localId\":\"xxxx\"},\"MessageCallback\":{\"CallbackURL\":\"http://test.test.com\"}}"");
                        /* 视频分类ID(可选) */
                        //request.setCateId(0);
                        /* 视频标签,多个用逗号分隔(可选) */
                        //request.setTags("标签1,标签2");
                        /* 视频描述(可选) */
                        //request.setDescription("视频描述");
                        /* 封面图片(可选) */
                        //request.setCoverURL("http://cover.sample.com/sample.jpg");
                        /* 模板组ID(可选) */
                        //request.setTemplateGroupId("8c4792cbc8694e7084fd5330e56a33d");
                        /* 工作流ID(可选) */
                        //request.setWorkflowId("d4430d07361f0*be1339577859b0177b");
                        /* 存储区域(可选) */
                        //request.setStorageLocation("in-201703232118266-5sejdln9o.oss-cn-shanghai.aliyuncs.com");
                        /* 开启默认上传进度回调 */
                        // request.setPrintProgress(true);
                        /* 设置自定义上传进度回调 (必须继承 VoDProgressListener) */
                        // request.setProgressListener(new PutObjectProgressListener());
                        /* 设置应用ID*/
                        //request.setAppId("app-1000000");
                        /* 点播服务接入点 */
                        //request.setApiRegionId("cn-shanghai");
                        /* ECS部署区域*/
                        // request.setEcsRegionId("cn-shanghai");
                        UploadVideoImpl uploader = new UploadVideoImpl();
                        UploadStreamResponse response = uploader.uploadStream(request);
                        System.out.print("RequestId=" + response.getRequestId() + "\n");  //请求视频点播服务的请求ID
                        if (response.isSuccess()) {
                            System.out.print("VideoId=" + response.getVideoId() + "\n");
                            vo.setVideoId(response.getVideoId());
                            SysOss sysOss = BeanUtil.toBean(vo, SysOss.class);
                            baseMapper.updateById(sysOss);
                        } else { //如果设置回调URL无效，不影响视频上传，可以返回VideoId同时会返回错误码。其他情况上传失败时，VideoId为空，此时需要根据返回错误码分析具体错误原因
                            System.out.print("VideoId=" + response.getVideoId() + "\n");
                            System.out.print("ErrorCode=" + response.getCode() + "\n");
                            System.out.print("ErrorMessage=" + response.getMessage() + "\n");
                        }
                    } catch (Exception e) {

                    }
                }
            }
        }
        return R.ok("同步媒体问题至阿里云成功");
    }

    /**
     * 下载OSS对象
     *
     * @param ossId OSS对象ID
     */
    @SaCheckPermission("system:oss:download")
    @GetMapping("/download/{ossId}")
    public void download(@PathVariable Long ossId, HttpServletResponse response) throws IOException {
        iSysOssService.download(ossId,response);
    }

    /**
     * 删除OSS对象存储
     *
     * @param ossIds OSS对象ID串
     */
    @SaCheckPermission("system:oss:remove")
    @Log(title = "OSS对象存储", businessType = BusinessType.DELETE)
    @DeleteMapping("/{ossIds}")
    public R<Void> remove(@NotEmpty(message = "主键不能为空")
                          @PathVariable Long[] ossIds) {
        for(Long id:ossIds){
            iSysOssService.delete(id);
        }
        return toAjax(iSysOssService.deleteWithValidByIds(Arrays.asList(ossIds), true));
    }

}