租房小程序前端代码
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

296 lines
8.4 KiB

3 months ago
  1. const assert = require('assert');
  2. const { checkBucketName: _checkBucketName } = require('../common/utils/checkBucketName');
  3. const proto = exports;
  4. function isArray(arr) {
  5. if (Array.isArray) return Array.isArray(arr);
  6. return Object.prototype.toString.call(arr) === '[object Array]';
  7. }
  8. function toArray(obj) {
  9. if (!obj) return [];
  10. if (isArray(obj)) return obj;
  11. return [obj];
  12. }
  13. /**
  14. * Bucket opertaions
  15. */
  16. // TODO: OSS server currently do not support CORS requests for bucket operations
  17. // proto.listBuckets = function* listBuckets(query, options) {
  18. // // prefix, marker, max-keys
  19. // var result = yield this.request({
  20. // method: 'GET',
  21. // query: query,
  22. // timeout: options && options.timeout,
  23. // ctx: options && options.ctx,
  24. // });
  25. //
  26. // if (result.status === 200) {
  27. // var data = yield this.parseXML(result.data);
  28. // var buckets = data.Buckets || null;
  29. // if (buckets) {
  30. // if (buckets.Bucket) {
  31. // buckets = buckets.Bucket;
  32. // }
  33. // if (!isArray(buckets)) {
  34. // buckets = [buckets];
  35. // }
  36. // buckets = buckets.map(function (item) {
  37. // return {
  38. // name: item.Name,
  39. // region: item.Location,
  40. // creationDate: item.CreationDate,
  41. // };
  42. // });
  43. // }
  44. // return {
  45. // buckets: buckets,
  46. // owner: {
  47. // id: data.Owner.ID,
  48. // displayName: data.Owner.DisplayName,
  49. // },
  50. // isTruncated: data.IsTruncated === 'true',
  51. // nextMarker: data.NextMarker || null,
  52. // res: result.res
  53. // };
  54. // }
  55. //
  56. // throw yield this.requestError(result);
  57. // };
  58. proto.useBucket = function useBucket(name) {
  59. _checkBucketName(name);
  60. this.options.bucket = name;
  61. return this;
  62. };
  63. proto.setBucket = function useBucket(name) {
  64. _checkBucketName(name);
  65. this.options.bucket = name;
  66. return this;
  67. };
  68. proto.getBucket = function getBucket() {
  69. return this.options.bucket;
  70. };
  71. proto.deleteBucket = async function deleteBucket(name, options) {
  72. const params = this._bucketRequestParams('DELETE', name, '', options);
  73. const result = await this.request(params);
  74. if (result.status === 200 || result.status === 204) {
  75. return {
  76. res: result.res
  77. };
  78. }
  79. throw await this.requestError(result);
  80. };
  81. // acl
  82. proto.putBucketACL = async function putBucketACL(name, acl, options) {
  83. const params = this._bucketRequestParams('PUT', name, 'acl', options);
  84. params.headers = {
  85. 'x-oss-acl': acl
  86. };
  87. params.successStatuses = [200];
  88. const result = await this.request(params);
  89. return {
  90. bucket: (result.headers.location && result.headers.location.substring(1)) || null,
  91. res: result.res
  92. };
  93. };
  94. proto.getBucketACL = async function getBucketACL(name, options) {
  95. const params = this._bucketRequestParams('GET', name, 'acl', options);
  96. params.successStatuses = [200];
  97. params.xmlResponse = true;
  98. const result = await this.request(params);
  99. return {
  100. acl: result.data.AccessControlList.Grant,
  101. owner: {
  102. id: result.data.Owner.ID,
  103. displayName: result.data.Owner.DisplayName
  104. },
  105. res: result.res
  106. };
  107. };
  108. // logging
  109. proto.putBucketLogging = async function putBucketLogging(name, prefix, options) {
  110. const params = this._bucketRequestParams('PUT', name, 'logging', options);
  111. let xml = `${'<?xml version="1.0" encoding="UTF-8"?>\n<BucketLoggingStatus>\n<LoggingEnabled>\n<TargetBucket>'}${name}</TargetBucket>\n`;
  112. if (prefix) {
  113. xml += `<TargetPrefix>${prefix}</TargetPrefix>\n`;
  114. }
  115. xml += '</LoggingEnabled>\n</BucketLoggingStatus>';
  116. params.content = xml;
  117. params.mime = 'xml';
  118. params.successStatuses = [200];
  119. const result = await this.request(params);
  120. return {
  121. res: result.res
  122. };
  123. };
  124. proto.getBucketLogging = async function getBucketLogging(name, options) {
  125. const params = this._bucketRequestParams('GET', name, 'logging', options);
  126. params.successStatuses = [200];
  127. params.xmlResponse = true;
  128. const result = await this.request(params);
  129. const enable = result.data.LoggingEnabled;
  130. return {
  131. enable: !!enable,
  132. prefix: (enable && enable.TargetPrefix) || null,
  133. res: result.res
  134. };
  135. };
  136. proto.deleteBucketLogging = async function deleteBucketLogging(name, options) {
  137. const params = this._bucketRequestParams('DELETE', name, 'logging', options);
  138. params.successStatuses = [204, 200];
  139. const result = await this.request(params);
  140. return {
  141. res: result.res
  142. };
  143. };
  144. proto.putBucketCORS = async function putBucketCORS(name, rules, options) {
  145. rules = rules || [];
  146. assert(rules.length, 'rules is required');
  147. rules.forEach(rule => {
  148. assert(rule.allowedOrigin, 'allowedOrigin is required');
  149. assert(rule.allowedMethod, 'allowedMethod is required');
  150. });
  151. const params = this._bucketRequestParams('PUT', name, 'cors', options);
  152. let xml = '<?xml version="1.0" encoding="UTF-8"?>\n<CORSConfiguration>';
  153. const parseOrigin = val => {
  154. xml += `<AllowedOrigin>${val}</AllowedOrigin>`;
  155. };
  156. const parseMethod = val => {
  157. xml += `<AllowedMethod>${val}</AllowedMethod>`;
  158. };
  159. const parseHeader = val => {
  160. xml += `<AllowedHeader>${val}</AllowedHeader>`;
  161. };
  162. const parseExposeHeader = val => {
  163. xml += `<ExposeHeader>${val}</ExposeHeader>`;
  164. };
  165. for (let i = 0, l = rules.length; i < l; i++) {
  166. const rule = rules[i];
  167. xml += '<CORSRule>';
  168. toArray(rule.allowedOrigin).forEach(parseOrigin);
  169. toArray(rule.allowedMethod).forEach(parseMethod);
  170. toArray(rule.allowedHeader).forEach(parseHeader);
  171. toArray(rule.exposeHeader).forEach(parseExposeHeader);
  172. if (rule.maxAgeSeconds) {
  173. xml += `<MaxAgeSeconds>${rule.maxAgeSeconds}</MaxAgeSeconds>`;
  174. }
  175. xml += '</CORSRule>';
  176. }
  177. xml += '</CORSConfiguration>';
  178. params.content = xml;
  179. params.mime = 'xml';
  180. params.successStatuses = [200];
  181. const result = await this.request(params);
  182. return {
  183. res: result.res
  184. };
  185. };
  186. proto.getBucketCORS = async function getBucketCORS(name, options) {
  187. const params = this._bucketRequestParams('GET', name, 'cors', options);
  188. params.successStatuses = [200];
  189. params.xmlResponse = true;
  190. const result = await this.request(params);
  191. const rules = [];
  192. if (result.data && result.data.CORSRule) {
  193. let { CORSRule } = result.data;
  194. if (!isArray(CORSRule)) CORSRule = [CORSRule];
  195. CORSRule.forEach(rule => {
  196. const r = {};
  197. Object.keys(rule).forEach(key => {
  198. r[key.slice(0, 1).toLowerCase() + key.slice(1, key.length)] = rule[key];
  199. });
  200. rules.push(r);
  201. });
  202. }
  203. return {
  204. rules,
  205. res: result.res
  206. };
  207. };
  208. proto.deleteBucketCORS = async function deleteBucketCORS(name, options) {
  209. const params = this._bucketRequestParams('DELETE', name, 'cors', options);
  210. params.successStatuses = [204];
  211. const result = await this.request(params);
  212. return {
  213. res: result.res
  214. };
  215. };
  216. // referer
  217. proto.putBucketReferer = async function putBucketReferer(name, allowEmpty, referers, options) {
  218. const params = this._bucketRequestParams('PUT', name, 'referer', options);
  219. let xml = '<?xml version="1.0" encoding="UTF-8"?>\n<RefererConfiguration>\n';
  220. xml += ` <AllowEmptyReferer>${allowEmpty ? 'true' : 'false'}</AllowEmptyReferer>\n`;
  221. if (referers && referers.length > 0) {
  222. xml += ' <RefererList>\n';
  223. for (let i = 0; i < referers.length; i++) {
  224. xml += ` <Referer>${referers[i]}</Referer>\n`;
  225. }
  226. xml += ' </RefererList>\n';
  227. } else {
  228. xml += ' <RefererList />\n';
  229. }
  230. xml += '</RefererConfiguration>';
  231. params.content = xml;
  232. params.mime = 'xml';
  233. params.successStatuses = [200];
  234. const result = await this.request(params);
  235. return {
  236. res: result.res
  237. };
  238. };
  239. proto.getBucketReferer = async function getBucketReferer(name, options) {
  240. const params = this._bucketRequestParams('GET', name, 'referer', options);
  241. params.successStatuses = [200];
  242. params.xmlResponse = true;
  243. const result = await this.request(params);
  244. let referers = result.data.RefererList.Referer || null;
  245. if (referers) {
  246. if (!isArray(referers)) {
  247. referers = [referers];
  248. }
  249. }
  250. return {
  251. allowEmpty: result.data.AllowEmptyReferer === 'true',
  252. referers,
  253. res: result.res
  254. };
  255. };
  256. proto.deleteBucketReferer = async function deleteBucketReferer(name, options) {
  257. return await this.putBucketReferer(name, true, null, options);
  258. };
  259. // private apis
  260. proto._bucketRequestParams = function _bucketRequestParams(method, bucket, subres, options) {
  261. return {
  262. method,
  263. bucket,
  264. subres,
  265. additionalHeaders: options && options.additionalHeaders,
  266. timeout: options && options.timeout,
  267. ctx: options && options.ctx
  268. };
  269. };