package com.test.app;

import com.aliyun.oss.OSS;
import com.aliyun.oss.OSSClientBuilder;
import com.aliyun.oss.model.*;
import org.apache.commons.codec.digest.DigestUtils;
import java.io.*;
import java.util.List;


public class SendToOSS {

    public static void main(String[] args) throws IOException {
        //主程序入口，调用getFiles方法
        //"./public"是你hexo生成博客文件的相对路径
        getFiles("./public");
    }


    public static void getFiles(String path) throws IOException {

        // Endpoint是OSS的仓库地区URL，请按实际情况填写，***一般填OSS所在地区的地名拼音，如"hangzhou","shenzhen"等
        String endpoint = "http://oss-cn-***.aliyuncs.com";
        // 阿里云主账号AccessKey拥有所有API的访问权限，风险很高。强烈建议您创建并使用RAM账号进行API访问或日常运维，请登录RAM控制台创建RAM账号。
        String accessKeyId = "***";
        String accessKeySecret = "***";
        //bucket填你的bucket名称
        String bucket = "***";

        // 创建OSSClient实例。
        OSS ossClient = new OSSClientBuilder().build(endpoint, accessKeyId, accessKeySecret);

        //开始查询文件
        File file = new File(path);
        File[] files = file.listFiles();
        for (int i = 0; i < files.length; i++) {
            if (files[i].isDirectory()) {
                // 如果还是文件夹，递归获取文件夹里面的文件
                getFiles(files[i].getPath());
            } else {
                //优化上传到OSS的文件名，否则上传到OSS后文件路径会有问题
                String OSSfiles = files[i].getPath().substring(9);

                //为了输出日志好看而写的...
                System.out.println("");
                System.out.println("");
                System.out.println("-----------------------------------");
                
                //开始处理文件
                System.out.println("\033[32;2m"+"Begin to Upload file：" +"\033[35;1m"+ OSSfiles+"\033[0m");
                
                // 查询OSS是否有与本地重名的文件
                final int maxKeys = 1;
                ObjectListing objectListing = ossClient.listObjects(new ListObjectsRequest(bucket).withPrefix(OSSfiles));
                List<OSSObjectSummary> sums = objectListing.getObjectSummaries();
                //查询有重名则赋值到"OSSfile"
                String OSSfile=new String();
                for (OSSObjectSummary s : sums) {
                    OSSfile = s.getKey();
                }

                    if (OSSfile.equals(OSSfiles)){
                        //重名则执行以下

                        System.out.println("\033[33;2m"+"文件已存在，验证MD5"+"\033[0m");

                        // 获取OSS文件的Etag。
                        SimplifiedObjectMeta objectMeta = ossClient.getSimplifiedObjectMeta(bucket, OSSfiles);
                        String Etag = objectMeta.getETag();
                        System.out.println("\033[37;2m"+"Etag值：" + Etag+"\033[0m");

                        //计算本地文件md5值
                        String filemd5 = DigestUtils.md5Hex(new FileInputStream(files[i].getPath()));
                        String fileMD5 = filemd5.toUpperCase();
                        System.out.println("\033[37;2m"+"文件MD5值:" + fileMD5+"\033[0m");

                        if (Etag.equals(fileMD5)) {
                            System.out.println("\033[33;2m"+"文件一致，不覆盖"+"\033[0m");
                        } else {
                            System.out.println("\033[32;2m"+"不一致，上传文件"+"\033[0m");
                            // 创建PutObjectRequest对象
                            PutObjectRequest putObjectRequest = new PutObjectRequest(bucket, OSSfiles, new File(files[i].getPath()));
                            // 上传文件。
                            ossClient.putObject(putObjectRequest);
                            //输出上传成功
                            System.out.println("\033[34;1m"+"Upload done,Succeeded!"+"\033[0m");
                        }
                    } else if (OSSfile != OSSfiles){
                        //不重名执行以下

                        //文件不存在，直接上传
                        System.out.println("\033[32;2m"+"File not found,Upload begin"+"\033[0m");
                        // 创建PutObjectRequest对象
                        PutObjectRequest putObjectRequest = new PutObjectRequest(bucket, OSSfiles, new File(files[i].getPath()));
                        // 上传文件。
                        ossClient.putObject(putObjectRequest);
                        //输出上传成功
                        System.out.println("\033[34;1m"+"Upload done,Succeeded!"+"\033[0m");
                    }
                    else {
                        //出问题则会显示上传失败（一般不存在这种情况，只会直接中断进程）
                        System.out.println("\033[31;1m"+"Upload Faild,please check ,,Ծ‸Ծ,,"+"\033[0m");
                    }
                    }
        }
        // 关闭OSSClient。
        ossClient.shutdown();
    }
}