PrivateGitHubProvider.js 7.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185
  1. "use strict";
  2. Object.defineProperty(exports, "__esModule", {
  3. value: true
  4. });
  5. exports.PrivateGitHubProvider = undefined;
  6. var _bluebirdLst;
  7. function _load_bluebirdLst() {
  8. return _bluebirdLst = require("bluebird-lst");
  9. }
  10. var _electron;
  11. function _load_electron() {
  12. return _electron = require("electron");
  13. }
  14. var _electronBuilderHttp;
  15. function _load_electronBuilderHttp() {
  16. return _electronBuilderHttp = require("electron-builder-http");
  17. }
  18. var _CancellationToken;
  19. function _load_CancellationToken() {
  20. return _CancellationToken = require("electron-builder-http/out/CancellationToken");
  21. }
  22. var _jsYaml;
  23. function _load_jsYaml() {
  24. return _jsYaml = require("js-yaml");
  25. }
  26. var _path = _interopRequireWildcard(require("path"));
  27. var _url;
  28. function _load_url() {
  29. return _url = require("url");
  30. }
  31. var _api;
  32. function _load_api() {
  33. return _api = require("./api");
  34. }
  35. var _electronHttpExecutor;
  36. function _load_electronHttpExecutor() {
  37. return _electronHttpExecutor = require("./electronHttpExecutor");
  38. }
  39. var _GenericProvider;
  40. function _load_GenericProvider() {
  41. return _GenericProvider = require("./GenericProvider");
  42. }
  43. var _GitHubProvider;
  44. function _load_GitHubProvider() {
  45. return _GitHubProvider = require("./GitHubProvider");
  46. }
  47. function _interopRequireWildcard(obj) { if (obj && obj.__esModule) { return obj; } else { var newObj = {}; if (obj != null) { for (var key in obj) { if (Object.prototype.hasOwnProperty.call(obj, key)) newObj[key] = obj[key]; } } newObj.default = obj; return newObj; } }
  48. class PrivateGitHubProvider extends (_GitHubProvider || _load_GitHubProvider()).BaseGitHubProvider {
  49. constructor(options, token) {
  50. super(options, "api.github.com");
  51. this.token = token;
  52. this.netSession = (_electron || _load_electron()).session.fromPartition((_electronHttpExecutor || _load_electronHttpExecutor()).NET_SESSION_NAME);
  53. this.registerHeaderRemovalListener();
  54. }
  55. getLatestVersion() {
  56. var _this = this;
  57. return (0, (_bluebirdLst || _load_bluebirdLst()).coroutine)(function* () {
  58. const basePath = _this.basePath;
  59. const cancellationToken = new (_CancellationToken || _load_CancellationToken()).CancellationToken();
  60. const channelFile = (0, (_api || _load_api()).getChannelFilename)((0, (_api || _load_api()).getDefaultChannelName)());
  61. const assets = yield _this.getLatestVersionInfo(basePath, cancellationToken);
  62. const requestOptions = Object.assign({
  63. headers: _this.configureHeaders("application/octet-stream"),
  64. session: _this.netSession
  65. }, (0, (_url || _load_url()).parse)(assets.find(function (it) {
  66. return it.name == channelFile;
  67. }).url));
  68. let result;
  69. try {
  70. result = yield (0, (_electronBuilderHttp || _load_electronBuilderHttp()).request)(requestOptions, cancellationToken);
  71. if (typeof result === "string") {
  72. if ((0, (_api || _load_api()).getCurrentPlatform)() === "darwin") {
  73. result = JSON.parse(result);
  74. } else {
  75. result = (0, (_jsYaml || _load_jsYaml()).safeLoad)(result);
  76. }
  77. }
  78. } catch (e) {
  79. if (e instanceof (_electronBuilderHttp || _load_electronBuilderHttp()).HttpError && e.response.statusCode === 404) {
  80. throw new Error(`Cannot find ${channelFile} in the latest release artifacts (${(0, (_api || _load_api()).formatUrl)(requestOptions)}): ${e.stack || e.message}`);
  81. }
  82. throw e;
  83. }
  84. (0, (_GenericProvider || _load_GenericProvider()).validateUpdateInfo)(result);
  85. if ((0, (_api || _load_api()).getCurrentPlatform)() === "darwin") {
  86. result.releaseJsonUrl = `${_this.options.protocol || "https"}://${_this.options.host || "api.github.com"}${requestOptions.path}`;
  87. }
  88. result.assets = assets;
  89. return result;
  90. })();
  91. }
  92. registerHeaderRemovalListener() {
  93. const filter = {
  94. urls: ["*://*.amazonaws.com/*"]
  95. };
  96. this.netSession.webRequest.onBeforeSendHeaders(filter, (details, callback) => {
  97. if (details.requestHeaders.Authorization != null) {
  98. delete details.requestHeaders.Authorization;
  99. }
  100. callback({ cancel: false, requestHeaders: details.requestHeaders });
  101. });
  102. }
  103. configureHeaders(accept) {
  104. return Object.assign({
  105. Accept: accept,
  106. Authorization: `token ${this.token}`
  107. }, this.requestHeaders);
  108. }
  109. getLatestVersionInfo(basePath, cancellationToken) {
  110. var _this2 = this;
  111. return (0, (_bluebirdLst || _load_bluebirdLst()).coroutine)(function* () {
  112. const requestOptions = Object.assign({
  113. path: `${basePath}/latest`,
  114. headers: _this2.configureHeaders("application/vnd.github.v3+json")
  115. }, _this2.baseUrl);
  116. try {
  117. return (yield (0, (_electronBuilderHttp || _load_electronBuilderHttp()).request)(requestOptions, cancellationToken)).assets;
  118. } catch (e) {
  119. throw new Error(`Unable to find latest version on GitHub (${(0, (_api || _load_api()).formatUrl)(requestOptions)}), please ensure a production release exists: ${e.stack || e.message}`);
  120. }
  121. })();
  122. }
  123. get basePath() {
  124. return `/repos/${this.options.owner}/${this.options.repo}/releases`;
  125. }
  126. getUpdateFile(versionInfo) {
  127. var _this3 = this;
  128. return (0, (_bluebirdLst || _load_bluebirdLst()).coroutine)(function* () {
  129. const headers = {
  130. Accept: "application/octet-stream",
  131. Authorization: `token ${_this3.token}`
  132. };
  133. // space is not supported on GitHub
  134. if ((0, (_api || _load_api()).getCurrentPlatform)() === "darwin") {
  135. const info = versionInfo;
  136. const name = info.url.split("/").pop();
  137. const assetPath = (0, (_url || _load_url()).parse)(versionInfo.assets.find(function (it) {
  138. return it.name == name;
  139. }).url).path;
  140. info.url = (0, (_api || _load_api()).formatUrl)(Object.assign({ path: `${assetPath}` }, _this3.baseUrl));
  141. info.headers = headers;
  142. return info;
  143. } else {
  144. const name = versionInfo.githubArtifactName || _path.posix.basename(versionInfo.path).replace(/ /g, "-");
  145. return {
  146. name: name,
  147. url: versionInfo.assets.find(function (it) {
  148. return it.name == name;
  149. }).url,
  150. sha2: versionInfo.sha2,
  151. headers: headers,
  152. session: _this3.netSession
  153. };
  154. }
  155. })();
  156. }
  157. }
  158. exports.PrivateGitHubProvider = PrivateGitHubProvider; //# sourceMappingURL=PrivateGitHubProvider.js.map