RocketChat/Rocket.Chat

import http from 'http';
import https from 'https';

import _ from 'underscore';

import { settings } from '../../../settings/server';
import { FileUploadClass, FileUpload } from '../lib/FileUpload';
import '../../ufs/GoogleStorage/server';
import { forceDownload } from './helper';

const get: FileUploadClass['get'] = async function (this: FileUploadClass, file, req, res) {
    const forcedDownload = forceDownload(req);

    const fileUrl = await this.store.getRedirectURL(file, forcedDownload);
    if (!fileUrl || !file.store) {
        res.end();
        return;
    }

    const storeType = file.store.split(':').pop();
    if (settings.get(`FileUpload_GoogleStorage_Proxy_${storeType}`)) {
        const request = /^https:/.test(fileUrl) ? https : http;

        FileUpload.proxyFile(file.name || '', fileUrl, forcedDownload, request, req, res);
        return;
    }

    FileUpload.redirectToFile(fileUrl, req, res);
};

const copy: FileUploadClass['copy'] = async function (this: FileUploadClass, file, out) {
    const fileUrl = await this.store.getRedirectURL(file, false);

    if (!fileUrl) {
        out.end();
        return;
    }

    const request = /^https:/.test(fileUrl) ? https : http;
    return new Promise((resolve) => {
        request.get(fileUrl, (fileRes) => fileRes.pipe(out).on('finish', () => resolve()));
    });
};

const GoogleCloudStorageUploads = new FileUploadClass({
    name: 'GoogleCloudStorage:Uploads',
    get,
    copy,
    // store setted bellow
});

const GoogleCloudStorageAvatars = new FileUploadClass({
    name: 'GoogleCloudStorage:Avatars',
    get,
    copy,
    // store setted bellow
});

const GoogleCloudStorageUserDataFiles = new FileUploadClass({
    name: 'GoogleCloudStorage:UserDataFiles',
    get,
    copy,
    // store setted bellow
});

const configure = _.debounce(() => {
    const bucket = settings.get('FileUpload_GoogleStorage_Bucket');
    const projectId = settings.get('FileUpload_GoogleStorage_ProjectId');
    const accessId = settings.get('FileUpload_GoogleStorage_AccessId');
    const secret = settings.get('FileUpload_GoogleStorage_Secret');
    const URLExpiryTimeSpan = settings.get('FileUpload_S3_URLExpiryTimeSpan');

    if (!bucket || !accessId || !secret) {
        return;
    }

    const config = {
        connection: {
            credentials: {
                client_email: accessId,
                private_key: secret,
            },
            projectId,
        },
        bucket,
        URLExpiryTimeSpan,
    };

    GoogleCloudStorageUploads.store = FileUpload.configureUploadsStore('GoogleStorage', GoogleCloudStorageUploads.name, config);
    GoogleCloudStorageAvatars.store = FileUpload.configureUploadsStore('GoogleStorage', GoogleCloudStorageAvatars.name, config);
    GoogleCloudStorageUserDataFiles.store = FileUpload.configureUploadsStore('GoogleStorage', GoogleCloudStorageUserDataFiles.name, config);
}, 500);

settings.watchByRegex(/^FileUpload_GoogleStorage_/, configure);