package cn.exrick.xboot.file.manage.impl;

import cn.exrick.xboot.core.common.constant.SettingConstant;
import cn.exrick.xboot.core.common.exception.XbootException;
import cn.exrick.xboot.core.entity.Setting;
import cn.exrick.xboot.core.service.SettingService;
import cn.exrick.xboot.core.vo.OssSetting;
import cn.exrick.xboot.file.manage.FileManage;
import cn.hutool.core.util.StrUtil;
import com.google.gson.Gson;
import com.qiniu.common.QiniuException;
import com.qiniu.http.Response;
import com.qiniu.storage.BucketManager;
import com.qiniu.storage.Configuration;
import com.qiniu.storage.Region;
import com.qiniu.storage.UploadManager;
import com.qiniu.storage.model.DefaultPutRet;
import com.qiniu.storage.model.FileInfo;
import com.qiniu.util.Auth;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.web.multipart.MultipartFile;

import java.io.InputStream;

/**
 * @author Exrick
 */
@Slf4j
@Component
public class QiniuFileManage implements FileManage {

    @Autowired
    private SettingService settingService;

    @Override
    public OssSetting getOssSetting() {

        Setting setting = settingService.get(SettingConstant.QINIU_OSS);
        if (setting == null || StrUtil.isBlank(setting.getValue())) {
            throw new XbootException("您还未配置七牛云对象存储");
        }
        return new Gson().fromJson(setting.getValue(), OssSetting.class);
    }

    public Configuration getConfiguration(Integer zone) {

        Configuration cfg = null;
        if (zone.equals(SettingConstant.ZONE_ZERO)) {
            cfg = new Configuration(Region.region0());
        } else if (zone.equals(SettingConstant.ZONE_ONE)) {
            cfg = new Configuration(Region.region1());
        } else if (zone.equals(SettingConstant.ZONE_TWO)) {
            cfg = new Configuration(Region.region2());
        } else if (zone.equals(SettingConstant.ZONE_THREE)) {
            cfg = new Configuration(Region.regionNa0());
        } else if (zone.equals(SettingConstant.ZONE_FOUR)) {
            cfg = new Configuration(Region.regionAs0());
        } else {
            cfg = new Configuration(Region.autoRegion());
        }
        return cfg;
    }

    public UploadManager getUploadManager(Configuration cfg) {

        UploadManager uploadManager = new UploadManager(cfg);
        return uploadManager;
    }

    @Override
    public String inputStreamUpload(InputStream inputStream, String key, MultipartFile file) {

        OssSetting os = getOssSetting();
        Auth auth = Auth.create(os.getAccessKey(), os.getSecretKey());
        String upToken = auth.uploadToken(os.getBucket());
        try {
            Response response = getUploadManager(getConfiguration(os.getZone())).put(inputStream, key, upToken, null, null);
            DefaultPutRet putRet = new Gson().fromJson(response.bodyString(), DefaultPutRet.class);
            return os.getHttp() + os.getEndpoint() + "/" + putRet.key;
        } catch (QiniuException ex) {
            Response r = ex.response;
            throw new XbootException("上传文件出错，请检查七牛云配置，" + r.toString());
        }
    }

    @Override
    public String renameFile(String fromKey, String toKey) {

        OssSetting os = getOssSetting();
        Auth auth = Auth.create(os.getAccessKey(), os.getSecretKey());
        BucketManager bucketManager = new BucketManager(auth, getConfiguration(os.getZone()));
        try {
            bucketManager.move(os.getBucket(), fromKey, os.getBucket(), toKey);
            return os.getHttp() + os.getEndpoint() + "/" + toKey;
        } catch (QiniuException ex) {
            throw new XbootException("重命名文件失败，" + ex.response.error);
        }
    }

    @Override
    public String copyFile(String fromKey, String toKey) {

        OssSetting os = getOssSetting();
        Auth auth = Auth.create(os.getAccessKey(), os.getSecretKey());
        BucketManager bucketManager = new BucketManager(auth, getConfiguration(os.getZone()));
        try {
            bucketManager.copy(os.getBucket(), fromKey, os.getBucket(), toKey);
            return os.getHttp() + os.getEndpoint() + "/" + toKey;
        } catch (QiniuException ex) {
            throw new XbootException("复制文件失败，" + ex.response.error);
        }
    }

    @Override
    public void deleteFile(String key) {

        OssSetting os = getOssSetting();
        Auth auth = Auth.create(os.getAccessKey(), os.getSecretKey());
        BucketManager bucketManager = new BucketManager(auth, getConfiguration(os.getZone()));
        FileInfo fileInfo = null;
        try {
            fileInfo = bucketManager.stat(os.getBucket(), key);
        } catch (QiniuException ex) {
            log.error("获取文件失败，" + ex.response.error);
        }
        try {
            if (fileInfo != null) {
                bucketManager.delete(os.getBucket(), key);
            }
        } catch (QiniuException ex) {
            throw new XbootException("删除文件失败，" + ex.response.error);
        }
    }
}