index.js 8.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337
  1. "use strict";
  2. var _interopRequireWildcard = require("@babel/runtime-corejs2/helpers/interopRequireWildcard");
  3. var _interopRequireDefault = require("@babel/runtime-corejs2/helpers/interopRequireDefault");
  4. var _Object$defineProperty = require("@babel/runtime-corejs2/core-js/object/define-property");
  5. _Object$defineProperty(exports, "__esModule", {
  6. value: true
  7. });
  8. exports.getUpHosts = getUpHosts;
  9. exports.initUploadParts = initUploadParts;
  10. exports.uploadChunk = uploadChunk;
  11. exports.uploadComplete = uploadComplete;
  12. exports.deleteUploadedChunks = deleteUploadedChunks;
  13. exports.direct = direct;
  14. exports.getUploadUrl = getUploadUrl;
  15. var _iterator = _interopRequireDefault(require("@babel/runtime-corejs2/core-js/symbol/iterator"));
  16. var _symbol = _interopRequireDefault(require("@babel/runtime-corejs2/core-js/symbol"));
  17. var _promise = _interopRequireDefault(require("@babel/runtime-corejs2/core-js/promise"));
  18. var _assign = _interopRequireDefault(require("@babel/runtime-corejs2/core-js/object/assign"));
  19. var _querystring = require("querystring");
  20. var utils = _interopRequireWildcard(require("../utils"));
  21. var __assign = void 0 && (void 0).__assign || function () {
  22. __assign = _assign["default"] || function (t) {
  23. for (var s, i = 1, n = arguments.length; i < n; i++) {
  24. s = arguments[i];
  25. for (var p in s) {
  26. if (Object.prototype.hasOwnProperty.call(s, p)) t[p] = s[p];
  27. }
  28. }
  29. return t;
  30. };
  31. return __assign.apply(this, arguments);
  32. };
  33. var __awaiter = void 0 && (void 0).__awaiter || function (thisArg, _arguments, P, generator) {
  34. function adopt(value) {
  35. return value instanceof P ? value : new P(function (resolve) {
  36. resolve(value);
  37. });
  38. }
  39. return new (P || (P = _promise["default"]))(function (resolve, reject) {
  40. function fulfilled(value) {
  41. try {
  42. step(generator.next(value));
  43. } catch (e) {
  44. reject(e);
  45. }
  46. }
  47. function rejected(value) {
  48. try {
  49. step(generator["throw"](value));
  50. } catch (e) {
  51. reject(e);
  52. }
  53. }
  54. function step(result) {
  55. result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected);
  56. }
  57. step((generator = generator.apply(thisArg, _arguments || [])).next());
  58. });
  59. };
  60. var __generator = void 0 && (void 0).__generator || function (thisArg, body) {
  61. var _ = {
  62. label: 0,
  63. sent: function sent() {
  64. if (t[0] & 1) throw t[1];
  65. return t[1];
  66. },
  67. trys: [],
  68. ops: []
  69. },
  70. f,
  71. y,
  72. t,
  73. g;
  74. return g = {
  75. next: verb(0),
  76. "throw": verb(1),
  77. "return": verb(2)
  78. }, typeof _symbol["default"] === "function" && (g[_iterator["default"]] = function () {
  79. return this;
  80. }), g;
  81. function verb(n) {
  82. return function (v) {
  83. return step([n, v]);
  84. };
  85. }
  86. function step(op) {
  87. if (f) throw new TypeError("Generator is already executing.");
  88. while (_) {
  89. try {
  90. if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
  91. if (y = 0, t) op = [op[0] & 2, t.value];
  92. switch (op[0]) {
  93. case 0:
  94. case 1:
  95. t = op;
  96. break;
  97. case 4:
  98. _.label++;
  99. return {
  100. value: op[1],
  101. done: false
  102. };
  103. case 5:
  104. _.label++;
  105. y = op[1];
  106. op = [0];
  107. continue;
  108. case 7:
  109. op = _.ops.pop();
  110. _.trys.pop();
  111. continue;
  112. default:
  113. if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) {
  114. _ = 0;
  115. continue;
  116. }
  117. if (op[0] === 3 && (!t || op[1] > t[0] && op[1] < t[3])) {
  118. _.label = op[1];
  119. break;
  120. }
  121. if (op[0] === 6 && _.label < t[1]) {
  122. _.label = t[1];
  123. t = op;
  124. break;
  125. }
  126. if (t && _.label < t[2]) {
  127. _.label = t[2];
  128. _.ops.push(op);
  129. break;
  130. }
  131. if (t[2]) _.ops.pop();
  132. _.trys.pop();
  133. continue;
  134. }
  135. op = body.call(thisArg, _);
  136. } catch (e) {
  137. op = [6, e];
  138. y = 0;
  139. } finally {
  140. f = t = 0;
  141. }
  142. }
  143. if (op[0] & 5) throw op[1];
  144. return {
  145. value: op[0] ? op[1] : void 0,
  146. done: true
  147. };
  148. }
  149. };
  150. function getUpHosts(accessKey, bucketName, protocol) {
  151. return __awaiter(this, void 0, void 0, function () {
  152. var params, url;
  153. return __generator(this, function (_a) {
  154. params = (0, _querystring.stringify)({
  155. ak: accessKey,
  156. bucket: bucketName
  157. });
  158. url = protocol + "://api.qiniu.com/v2/query?" + params;
  159. return [2
  160. /*return*/
  161. , utils.request(url, {
  162. method: 'GET'
  163. })];
  164. });
  165. });
  166. }
  167. /**
  168. * @param bucket 空间名
  169. * @param key 目标文件名
  170. * @param uploadInfo 上传信息
  171. */
  172. function getBaseUrl(bucket, key, uploadInfo) {
  173. var url = uploadInfo.url,
  174. id = uploadInfo.id;
  175. return url + "/buckets/" + bucket + "/objects/" + (key != null ? utils.urlSafeBase64Encode(key) : '~') + "/uploads/" + id;
  176. }
  177. /**
  178. * @param token 上传鉴权凭证
  179. * @param bucket 上传空间
  180. * @param key 目标文件名
  181. * @param uploadUrl 上传地址
  182. */
  183. function initUploadParts(token, bucket, key, uploadUrl) {
  184. var url = uploadUrl + "/buckets/" + bucket + "/objects/" + (key != null ? utils.urlSafeBase64Encode(key) : '~') + "/uploads";
  185. return utils.request(url, {
  186. method: 'POST',
  187. headers: utils.getAuthHeaders(token)
  188. });
  189. }
  190. /**
  191. * @param token 上传鉴权凭证
  192. * @param index 当前 chunk 的索引
  193. * @param uploadInfo 上传信息
  194. * @param options 请求参数
  195. */
  196. function uploadChunk(token, key, index, uploadInfo, options) {
  197. var bucket = utils.getPutPolicy(token).bucketName;
  198. var url = getBaseUrl(bucket, key, uploadInfo) + ("/" + index);
  199. var headers = utils.getHeadersForChunkUpload(token);
  200. if (options.md5) headers['Content-MD5'] = options.md5;
  201. return utils.request(url, __assign(__assign({}, options), {
  202. method: 'PUT',
  203. headers: headers
  204. }));
  205. }
  206. /**
  207. * @param token 上传鉴权凭证
  208. * @param key 目标文件名
  209. * @param uploadInfo 上传信息
  210. * @param options 请求参数
  211. */
  212. function uploadComplete(token, key, uploadInfo, options) {
  213. var bucket = utils.getPutPolicy(token).bucketName;
  214. var url = getBaseUrl(bucket, key, uploadInfo);
  215. return utils.request(url, __assign(__assign({}, options), {
  216. method: 'POST',
  217. headers: utils.getHeadersForMkFile(token)
  218. }));
  219. }
  220. /**
  221. * @param token 上传鉴权凭证
  222. * @param key 目标文件名
  223. * @param uploadInfo 上传信息
  224. */
  225. function deleteUploadedChunks(token, key, uploadinfo) {
  226. var bucket = utils.getPutPolicy(token).bucketName;
  227. var url = getBaseUrl(bucket, key, uploadinfo);
  228. return utils.request(url, {
  229. method: 'DELETE',
  230. headers: utils.getAuthHeaders(token)
  231. });
  232. }
  233. /**
  234. * @param {string} url
  235. * @param {FormData} data
  236. * @param {Partial<utils.RequestOptions>} options
  237. * @returns Promise
  238. * @description 直传接口
  239. */
  240. function direct(url, data, options) {
  241. return utils.request(url, __assign({
  242. method: 'POST',
  243. body: data
  244. }, options));
  245. }
  246. /**
  247. * @param {UploadUrlConfig} config
  248. * @param {string} token
  249. * @returns Promise
  250. * @description 获取上传 url
  251. */
  252. function getUploadUrl(_config, token) {
  253. return __awaiter(this, void 0, void 0, function () {
  254. var config, protocol, putPolicy, res, hosts;
  255. return __generator(this, function (_a) {
  256. switch (_a.label) {
  257. case 0:
  258. config = (0, utils.normalizeUploadConfig)(_config);
  259. protocol = config.upprotocol;
  260. if (config.uphost.length > 0) {
  261. return [2
  262. /*return*/
  263. , protocol + "://" + config.uphost[0]];
  264. }
  265. putPolicy = utils.getPutPolicy(token);
  266. return [4
  267. /*yield*/
  268. , getUpHosts(putPolicy.assessKey, putPolicy.bucketName, protocol)];
  269. case 1:
  270. res = _a.sent();
  271. hosts = res.data.up.acc.main;
  272. return [2
  273. /*return*/
  274. , protocol + "://" + hosts[0]];
  275. }
  276. });
  277. });
  278. }