com.ajaxjs.cloudstroage.NsoHttpUpload Maven / Gradle / Ivy

Go to download
/**
 * 版权所有 2017 Sp42 [email protected] 根据 2.0 版本 Apache 许可证("许可证")授权；
 * 根据本许可证，用户可以不使用此文件。 用户可从下列网址获得许可证副本：
 * http://www.apache.org/licenses/LICENSE-2.0
 * 除非因适用法律需要或书面同意，根据许可证分发的软件是基于"按原样"基础提供，
 * 无任何明示的或暗示的保证或条件。详见根据许可证许可下，特定语言的管辖权限和限制。
 */
package com.ajaxjs.cloudstroage;

import java.io.ByteArrayInputStream;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.net.HttpURLConnection;
import java.nio.file.Files;
import java.nio.file.StandardOpenOption;
import java.security.MessageDigest;
import java.security.NoSuchAlgorithmException;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.Locale;
import java.util.Map;
import java.util.TimeZone;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.util.Base64Utils;

import com.ajaxjs.config.GetConfig;
import com.ajaxjs.net.http.Delete;
import com.ajaxjs.net.http.Get;
import com.ajaxjs.net.http.Post;
import com.ajaxjs.util.Encode;
import com.ajaxjs.util.io.FileHelper;

/**
 * 网易云对象存储 HTTP 文件上传
 * 
 * @author sp42 [email protected]
 *
 */
public class NsoHttpUpload {
	@Autowired
	private GetConfig cfg;

	/**
	 * 列出所有的桶
	 * 
	 * @return XML 结果
	 */
	public Map listBuk() {
		String now = getDate();
		String canonicalizedHeaders = "", canonicalizedResource = "/";
		String data = "GET\n" + "\n" + "\n" + now + "\n" + canonicalizedHeaders + canonicalizedResource;
		String authorization = getAuthorization(data);

		Map xml = Get.apiXML("http://nos-eastchina1.126.net", conn -> {
			conn.addRequestProperty("Authorization", authorization);
			conn.addRequestProperty("Date", now);
			conn.addRequestProperty("Host", "nos-eastchina1.126.net");
		});

		return xml;
	}

	/**
	 * 请求的时间戳，格式必须符合 RFC1123 的日期格式
	 * 
	 * @return 当前日期
	 */
	private static String getDate() {
		SimpleDateFormat format = new SimpleDateFormat("EEE, dd MMM yyyy HH:mm:ss z", Locale.US);
		format.setTimeZone(TimeZone.getTimeZone("GMT"));

		return format.format(new Date());
	}

	/**
	 * 生成验证的字符串
	 * 
	 * @param data
	 * @return
	 */
	private String getAuthorization(String data) {
		String accessKey = cfg.get("uploadFile.ObjectStorageService.NOS.accessKey");
		String secretKey = cfg.get("uploadFile.ObjectStorageService.NOS.secretKey");

		String signature = Base64Utils.encodeToString(Encode.HMACSHA256(data, secretKey));
		String authorization = "NOS " + accessKey + ":" + signature;

		return authorization;
	}

	/**
	 * 创建空文件
	 * 
	 * @param filename 文件名
	 */
	public void createEmptyFile(String filename) {
		String bucket = cfg.get("uploadFile.ObjectStorageService.NOS.bucket");
		String now = getDate();
		String canonicalizedHeaders = "", canonicalizedResource = "/" + bucket + "/" + filename;
		String data = "PUT\n" + "\n\n" + now + "\n" + canonicalizedHeaders + canonicalizedResource;
		String authorization = getAuthorization(data);
		String api = cfg.get("uploadFile.ObjectStorageService.NOS.api");

		Post.put(api + filename, new byte[0], conn -> {
			conn.addRequestProperty("Authorization", authorization);
			conn.addRequestProperty("Content-Length", "0");
			conn.addRequestProperty("Date", now);
//			conn.addRequestProperty("Host", "ajaxjs.nos-eastchina1.126.net");
		});
	}

	public boolean delete(String filename) {
		String bucket = cfg.get("uploadFile.ObjectStorageService.NOS.bucket");
		String now = getDate();
		String canonicalizedHeaders = "", canonicalizedResource = "/" + bucket + "/" + filename;
		String data = "DELETE\n" + "\n\n" + now + "\n" + canonicalizedHeaders + canonicalizedResource;
		String authorization = getAuthorization(data);
		String api = cfg.get("uploadFile.ObjectStorageService.NOS.api");

		Delete.del(api + filename, conn -> {
			conn.addRequestProperty("Authorization", authorization);
			conn.addRequestProperty("Date", now);
		});

		return false;
	}

	/**
	 * 上传文件
	 * 
	 * @param filePath 文件路径
	 */
	public boolean uploadFile(String filePath) {
		return uploadFile(filePath, null);
	}

	/**
	 * 上传文件
	 * 
	 * @param filePath 文件路径
	 * @param filename 文件名，若不指定则按原来的文件名
	 */
	public boolean uploadFile(String filePath, String filename) {

		File file = new File(filePath);
		if (filename == null)
			filename = file.getName();

		return uploadFile(FileHelper.openAsByte(file), filename, calcMD5(file, null));
	}

	/**
	 * 
	 * @param bytes
	 * @param filename
	 * @param md5
	 */
	public boolean uploadFile(byte[] bytes, String filename, String md5) {
		String bucket = cfg.get("uploadFile.ObjectStorageService.NOS.bucket");

		String now = getDate();
		String canonicalizedHeaders = "", canonicalizedResource = "/" + bucket + "/" + filename;
		String data = "PUT\n" + md5 + "\n\n" + now + "\n" + canonicalizedHeaders + canonicalizedResource;
		String authorization = getAuthorization(data);
		String api = cfg.get("uploadFile.ObjectStorageService.NOS.api");
		// "https://ajaxjs.nos-eastchina1.126.net/"

		final TempClz clz = new TempClz();
		Post.put(api + filename, bytes, conn -> {
			conn.addRequestProperty("Authorization", authorization);
			conn.addRequestProperty("Content-Length", bytes.length + "");
//			conn.addRequestProperty("Content-Type", "");
			conn.addRequestProperty("Content-MD5", md5);
			conn.addRequestProperty("Date", now);
//			conn.addRequestProperty("HOST", "gdhdc-org.nos-eastchina1.126.net/cover");
			// conn.addRequestProperty("x-nos-entity-type", "json");
			clz.conn = conn;
		});

		// 判定是否上传成功
		try {
			String ETag = clz.conn.getHeaderField("ETag");
			if (ETag == null)
				return false;

			if (clz.conn.getResponseCode() == 200 && ETag.equalsIgnoreCase("\"" + md5 + "\""))
				return true;
		} catch (IOException e) {
			e.printStackTrace();
		}

		return false;
	}

	/**
	 * 解决 lambda final var
	 * 
	 * @author sp42 [email protected]
	 *
	 */
	private static class TempClz {
		HttpURLConnection conn;
	}

	/**
	 * 计算文件 MD5
	 * 
	 * @param file
	 * @return 返回文件的md5字符串，如果计算过程中任务的状态变为取消或暂停，返回null， 如果有其他异常，返回空字符串
	 */
	public static String calcMD5(File file, byte[] bytes) {
		try (InputStream stream = file != null ? Files.newInputStream(file.toPath(), StandardOpenOption.READ) : new ByteArrayInputStream(bytes)) {
			byte[] buf = new byte[8192];
			int len;
			MessageDigest digest = MessageDigest.getInstance("MD5");

			while ((len = stream.read(buf)) > 0)
				digest.update(buf, 0, len);

			return toHexString(digest.digest());
		} catch (IOException | NoSuchAlgorithmException e) {
			e.printStackTrace();
			return "";
		}
	}

	private static final char[] hexCode = "0123456789ABCDEF".toCharArray();

	/**
	 * 
	 * @param data
	 * @return
	 */
	private static String toHexString(byte[] data) {
		StringBuilder r = new StringBuilder(data.length * 2);

		for (byte b : data) {
			r.append(hexCode[(b >> 4) & 0xF]);
			r.append(hexCode[(b & 0xF)]);
		}

		return r.toString();
	}
}