package cn.celess.extension.serviceimpl;

import cn.celess.common.entity.vo.QiniuResponse;
import cn.celess.common.service.QiniuService;
import cn.celess.common.util.EnvironmentUtil;
import com.qiniu.common.QiniuException;
import com.qiniu.common.Zone;
import com.qiniu.http.Response;
import com.qiniu.storage.BucketManager;
import com.qiniu.storage.Configuration;
import com.qiniu.storage.UploadManager;
import com.qiniu.storage.model.FileInfo;
import com.qiniu.util.Auth;
import org.springframework.stereotype.Service;

import java.io.InputStream;

/**
 * @author : xiaohai
 * @date : 2019/04/25 18:15
 */
@Service
public class QiniuServiceImpl implements QiniuService {
    private static final Configuration cfg = new Configuration(Zone.zone2());
    private static UploadManager uploadManager;
    private static BucketManager bucketManager;
    private static Auth auth;

    private void init() {
        String accessKey = EnvironmentUtil.getProperties("qiniu.accessKey");
        String secretKey = EnvironmentUtil.getProperties("qiniu.secretKey");
        if (auth == null || uploadManager == null || bucketManager == null) {
            auth = Auth.create(accessKey, secretKey);
            uploadManager = new UploadManager(cfg);
            bucketManager = new BucketManager(auth, cfg);
        }
    }

    @Override
    public QiniuResponse uploadFile(InputStream is, String fileName) {
        init();
        String bucket = EnvironmentUtil.getProperties("qiniu.bucket");
        //文件存在则删除文件
        if (continueFile(fileName)) {
            try {
                System.out.println(bucketManager.delete(bucket, fileName).toString());
            } catch (QiniuException e) {
                e.printStackTrace();
            }
        }
        //上传
        try {
            Response response = uploadManager.put(is, fileName, auth.uploadToken(bucket), null, null);
            return response.jsonToObject(QiniuResponse.class);
        } catch (QiniuException e) {
            Response r = e.response;
            System.err.println(r.toString());
            return null;
        }
    }

    @Override
    public FileInfo[] getFileList() {
        init();
        String bucket = EnvironmentUtil.getProperties("qiniu.bucket");
        BucketManager.FileListIterator fileListIterator = bucketManager.createFileListIterator(bucket, "", 1000, "");
        FileInfo[] items = null;
        while (fileListIterator.hasNext()) {
            //处理获取的file list结果
            items = fileListIterator.next();
        }
        return items;
    }

    private boolean continueFile(String key) {
        FileInfo[] allFile = getFileList();
        for (FileInfo fileInfo : allFile) {
            if (key.equals(fileInfo.key)) {
                return true;
            }
        }
        return false;
    }
}