upload-to-github.ts 3.5 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109
  1. import { Octokit } from '@octokit/rest';
  2. import * as fs from 'node:fs';
  3. import { createGitHubTokenStrategy } from '../github-token';
  4. import { ELECTRON_ORG, ELECTRON_REPO, ElectronReleaseRepo, NIGHTLY_REPO } from '../types';
  5. if (process.argv.length < 6) {
  6. console.log('Usage: upload-to-github filePath fileName releaseId');
  7. process.exit(1);
  8. }
  9. const filePath = process.argv[2];
  10. const fileName = process.argv[3];
  11. const releaseId = parseInt(process.argv[4], 10);
  12. const releaseVersion = process.argv[5];
  13. if (isNaN(releaseId)) {
  14. throw new TypeError('Provided release ID was not a valid integer');
  15. }
  16. const getHeaders = (filePath: string, fileName: string) => {
  17. const extension = fileName.split('.').pop();
  18. if (!extension) {
  19. throw new Error(`Failed to get headers for extensionless file: ${fileName}`);
  20. }
  21. console.log(`About to get size of ${filePath}`);
  22. const size = fs.statSync(filePath).size;
  23. console.log(`Got size of ${filePath}: ${size}`);
  24. const options: Record<string, string> = {
  25. json: 'text/json',
  26. zip: 'application/zip',
  27. txt: 'text/plain',
  28. ts: 'application/typescript'
  29. };
  30. return {
  31. 'content-type': options[extension],
  32. 'content-length': size
  33. };
  34. };
  35. function getRepo (): ElectronReleaseRepo {
  36. return releaseVersion.indexOf('nightly') > 0 ? NIGHTLY_REPO : ELECTRON_REPO;
  37. }
  38. const targetRepo = getRepo();
  39. const uploadUrl = `https://uploads.github.com/repos/electron/${targetRepo}/releases/${releaseId}/assets{?name,label}`;
  40. let retry = 0;
  41. const octokit = new Octokit({
  42. authStrategy: createGitHubTokenStrategy(targetRepo),
  43. log: console
  44. });
  45. function uploadToGitHub () {
  46. console.log(`in uploadToGitHub for ${filePath}, ${fileName}`);
  47. const fileData = fs.createReadStream(filePath);
  48. console.log(`in uploadToGitHub, created readstream for ${filePath}`);
  49. octokit.repos.uploadReleaseAsset({
  50. url: uploadUrl,
  51. headers: getHeaders(filePath, fileName),
  52. data: fileData as any,
  53. name: fileName,
  54. owner: ELECTRON_ORG,
  55. repo: targetRepo,
  56. release_id: releaseId
  57. }).then(() => {
  58. console.log(`Successfully uploaded ${fileName} to GitHub.`);
  59. process.exit();
  60. }).catch((err) => {
  61. if (retry < 4) {
  62. console.log(`Error uploading ${fileName} to GitHub, will retry. Error was:`, err);
  63. retry++;
  64. octokit.repos.listReleaseAssets({
  65. owner: ELECTRON_ORG,
  66. repo: targetRepo,
  67. release_id: releaseId,
  68. per_page: 100
  69. }).then(assets => {
  70. console.log('Got list of assets for existing release:');
  71. console.log(JSON.stringify(assets.data, null, ' '));
  72. const existingAssets = assets.data.filter(asset => asset.name === fileName);
  73. if (existingAssets.length > 0) {
  74. console.log(`${fileName} already exists; will delete before retrying upload.`);
  75. octokit.repos.deleteReleaseAsset({
  76. owner: ELECTRON_ORG,
  77. repo: targetRepo,
  78. asset_id: existingAssets[0].id
  79. }).catch((deleteErr) => {
  80. console.log(`Failed to delete existing asset ${fileName}. Error was:`, deleteErr);
  81. }).then(uploadToGitHub);
  82. } else {
  83. console.log(`Current asset ${fileName} not found in existing assets; retrying upload.`);
  84. uploadToGitHub();
  85. }
  86. }).catch((getReleaseErr) => {
  87. console.log('Fatal: Unable to get current release assets via getRelease! Error was:', getReleaseErr);
  88. });
  89. } else {
  90. console.log(`Error retrying uploading ${fileName} to GitHub:`, err);
  91. process.exitCode = 1;
  92. }
  93. });
  94. }
  95. uploadToGitHub();