iframe-proxy | Sunbelt Computer Software

89 lines (73 loc) · 2.79 KB

import settings from '../settings';
import { S3Settings } from './settings';
import * as mime from 'mime-types';
import * as path from 'path';
import * as crypto from 'crypto';
import S3 from 'aws-sdk/clients/s3';
import GitlabHelper from './gitlabHelper';
export const sleep = (milliseconds: number) => {
  return new Promise((resolve) => setTimeout(resolve, milliseconds));
 * Generate regular expression which finds userid and cross-project issue references
 * from usermap and projectmap
export const generateUserProjectRegex = () => {
  let reString = '';
  if (settings.usermap !== null && Object.keys(settings.usermap).length > 0) {
    reString = '@' + Object.keys(settings.usermap).join('|@');
    settings.projectmap !== null &&
    Object.keys(settings.projectmap).length > 0
    if (reString.length > 0) {
      reString += '|';
    reString += Object.keys(settings.projectmap).join('#|') + '#';
  return new RegExp(reString, 'g');
// Creates new attachments and replaces old links
export const migrateAttachments = async (body: string, githubRepoId: number | undefined, s3: S3Settings, gitlabHelper: GitlabHelper) => {
  const regexp = /!\[([^\]]+)\]\((\/uploads[^)]+)\)/g;
  // Maps link offset to a new name in S3
  const offsetToAttachment: {
    [key: number]: string;
  // Find all local links
  const matches = body.matchAll(regexp);
  for (const match of matches) {
    const name = match[1];
    const url = match[2];
    const basename = path.basename(url);
    const extension = path.extname(url);
    const mimeType = mime.lookup(basename);
    const attachmentBuffer = await gitlabHelper.getAttachment(url);
    // Generate new random file name for S3 bucket
    const id = crypto.randomBytes(16).toString('hex');
    const newFileName = id + extension;
    const relativePath = githubRepoId ? `${githubRepoId}/${newFileName}` : newFileName;
    // Doesn't seem like it is easy to upload an issue to github, so upload to S3
    //https://stackoverflow.com/questions/41581151/how-to-upload-an-image-to-use-in-issue-comments-via-github-api
    const s3url = `https://${s3.bucket}.s3.amazonaws.com/${relativePath}`;
    const s3bucket = new S3();
    s3bucket.createBucket(() => {
      const params: S3.PutObjectRequest = {
        Key: relativePath,
        Body: attachmentBuffer,
        ContentType: mimeType === false ? null : mimeType,
        Bucket: s3.bucket,
      s3bucket.upload(params, function (err, data) {
        console.log(`\tUploaded ${basename} to ${s3url}`);
        if (err) {
          console.log('ERROR MSG: ', err);
    // Add the new URL to the map
    offsetToAttachment[match.index] = `![${name}](${s3url})`;
  return body.replace(regexp, ({},{},{},offset,{}) => offsetToAttachment[offset]);

Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

utils.ts

Latest commit

History

utils.ts

File metadata and controls