@jesec/flood 0.0.0-master.4b23b16 → 0.0.0-master.4b36bdb

package/server/middleware/clientActivityStream.js

@@ -1,150 +0,0 @@
-const clientGatewayServiceEvents = require('../constants/clientGatewayServiceEvents');
-const historyServiceEvents = require('../constants/historyServiceEvents');
-const historySnapshotTypes = require('../../shared/constants/historySnapshotTypes');
-const notificationServiceEvents = require('../constants/notificationServiceEvents');
-const ServerEvent = require('../models/ServerEvent');
-const serverEventTypes = require('../../shared/constants/serverEventTypes');
-const services = require('../services');
-const taxonomyServiceEvents = require('../constants/taxonomyServiceEvents');
-const torrentServiceEvents = require('../constants/torrentServiceEvents');
-const diskUsageServiceEvents = require('../constants/diskUsageServiceEvents');
-const DiskUsageService = require('../services/diskUsageService');
-
-module.exports = (req, res) => {
-  const {
-    query: {historySnapshot = historySnapshotTypes.FIVE_MINUTE},
-    user,
-  } = req;
-
-  const serviceInstances = services.getAllServices(user);
-  const serverEvent = new ServerEvent(res);
-  const taxonomy = serviceInstances.taxonomyService.getTaxonomy();
-  const torrentList = serviceInstances.torrentService.getTorrentList();
-  const transferSummary = serviceInstances.historyService.getTransferSummary();
-
-  // Hook into events and stop listening when connection is closed
-  const handleEvents = (emitter, event, handler) => {
-    emitter.on(event, handler);
-    res.on('close', () => {
-      emitter.removeListener(event, handler);
-    });
-  };
-
-  // Emit current state immediately on connection.
-  serverEvent.setID(Date.now());
-  serverEvent.setType(serverEventTypes.CLIENT_CONNECTIVITY_STATUS_CHANGE);
-  serverEvent.addData({isConnected: !serviceInstances.clientGatewayService.hasError});
-  serverEvent.emit();
-
-  const handleDiskUsageChange = (diskUsageChange) => {
-    serverEvent.setID(diskUsageChange.id);
-    serverEvent.setType(serverEventTypes.DISK_USAGE_CHANGE);
-    serverEvent.addData(diskUsageChange.disks);
-    serverEvent.emit();
-  };
-
-  DiskUsageService.updateDisks().then(() => {
-    const diskUsage = DiskUsageService.getDiskUsage();
-    serverEvent.setID(diskUsage.id);
-    serverEvent.setType(serverEventTypes.DISK_USAGE_CHANGE);
-    serverEvent.addData(diskUsage.disks);
-    serverEvent.emit();
-    handleEvents(DiskUsageService, diskUsageServiceEvents.DISK_USAGE_CHANGE, handleDiskUsageChange);
-  });
-
-  serverEvent.setID(torrentList.id);
-  serverEvent.setType(serverEventTypes.TORRENT_LIST_FULL_UPDATE);
-  serverEvent.addData(torrentList.torrents);
-  serverEvent.emit();
-
-  serverEvent.setID(taxonomy.id);
-  serverEvent.setType(serverEventTypes.TAXONOMY_FULL_UPDATE);
-  serverEvent.addData(taxonomy.taxonomy);
-  serverEvent.emit();
-
-  serverEvent.setID(transferSummary.id);
-  serverEvent.setType(serverEventTypes.TRANSFER_SUMMARY_FULL_UPDATE);
-  serverEvent.addData(transferSummary.transferSummary);
-  serverEvent.emit();
-
-  serverEvent.setID(Date.now());
-  serverEvent.setType(serverEventTypes.NOTIFICATION_COUNT_CHANGE);
-  serverEvent.addData(serviceInstances.notificationService.getNotificationCount());
-  serverEvent.emit();
-
-  handleEvents(serviceInstances.clientGatewayService, clientGatewayServiceEvents.CLIENT_CONNECTION_STATE_CHANGE, () => {
-    serverEvent.setID(Date.now());
-    serverEvent.setType(serverEventTypes.CLIENT_CONNECTIVITY_STATUS_CHANGE);
-    serverEvent.addData({isConnected: !serviceInstances.clientGatewayService.hasError});
-    serverEvent.emit();
-  });
-
-  if (serviceInstances.clientGatewayService.hasError) {
-    serviceInstances.clientGatewayService.testGateway();
-  }
-
-  // TODO: Handle empty or sub-optimal history states.
-  // Get user's specified history snapshot current history.
-  serviceInstances.historyService.getHistory({snapshot: historySnapshot}, (snapshot, error) => {
-    const {timestamps: lastTimestamps = []} = snapshot;
-    const lastTimestamp = lastTimestamps[lastTimestamps.length - 1];
-
-    if (error == null) {
-      serverEvent.setID(lastTimestamp);
-      serverEvent.setType(serverEventTypes.TRANSFER_HISTORY_FULL_UPDATE);
-      serverEvent.addData(snapshot);
-      serverEvent.emit();
-    }
-  });
-
-  // Add user's specified history snapshot change event listener.
-  handleEvents(
-    serviceInstances.historyService,
-    historyServiceEvents[`${historySnapshotTypes[historySnapshot]}_SNAPSHOT_FULL_UPDATE`],
-    (payload) => {
-      const {data, id} = payload;
-
-      serverEvent.setID(id);
-      serverEvent.setType(serverEventTypes.TRANSFER_HISTORY_FULL_UPDATE);
-      serverEvent.addData(data);
-      serverEvent.emit();
-    },
-  );
-
-  handleEvents(serviceInstances.notificationService, notificationServiceEvents.NOTIFICATION_COUNT_CHANGE, (payload) => {
-    const {data, id} = payload;
-
-    serverEvent.setID(id);
-    serverEvent.setType(serverEventTypes.NOTIFICATION_COUNT_CHANGE);
-    serverEvent.addData(data);
-    serverEvent.emit();
-  });
-
-  // Add diff event listeners.
-  handleEvents(serviceInstances.historyService, historyServiceEvents.TRANSFER_SUMMARY_DIFF_CHANGE, (payload) => {
-    const {diff, id} = payload;
-
-    serverEvent.setID(id);
-    serverEvent.setType(serverEventTypes.TRANSFER_SUMMARY_DIFF_CHANGE);
-    serverEvent.addData(diff);
-    serverEvent.emit();
-  });
-
-  handleEvents(serviceInstances.taxonomyService, taxonomyServiceEvents.TAXONOMY_DIFF_CHANGE, (payload) => {
-    const {diff, id} = payload;
-
-    serverEvent.setID(id);
-    serverEvent.setType(serverEventTypes.TAXONOMY_DIFF_CHANGE);
-    serverEvent.addData(diff);
-    serverEvent.emit();
-  });
-
-  handleEvents(serviceInstances.torrentService, torrentServiceEvents.TORRENT_LIST_DIFF_CHANGE, (payload) => {
-    const {diff, id} = payload;
-
-    serverEvent.setID(id);
-    serverEvent.setType(serverEventTypes.TORRENT_LIST_DIFF_CHANGE);
-    serverEvent.addData(diff);
-    serverEvent.emit();
-  });
-};

package/server/middleware/eventStream.js

@@ -1,27 +0,0 @@
-module.exports = (req, res, next) => {
-  req.socket.setKeepAlive(true);
-  req.socket.setTimeout(0);
-
-  res.set({
-    'Content-Type': 'text/event-stream',
-    'Cache-Control': 'no-cache',
-    Connection: 'keep-alive',
-    'Cross-Origin-Resource-Policy': 'same-origin',
-    'X-Accel-Buffering': 'no',
-  });
-  res.status(200);
-  res.write('retry: 500\n\n');
-
-  // Keep the connection open by sending a message every so often.
-  const keepAliveTimeout = setInterval(() => {
-    res.write(':keep-alive\n\n');
-    res.flush();
-  }, 500);
-
-  // cleanup on close
-  res.on('close', () => {
-    clearInterval(keepAliveTimeout);
-  });
-
-  next();
-};

package/server/middleware/requireAdmin.js

@@ -1,6 +0,0 @@
-module.exports = (req, res, next) => {
-  if (req.user == null || !req.user.isAdmin) {
-    return res.status(403).json({message: 'User is not admin.'}).send();
-  }
-  next();
-};

package/server/models/ClientRequest.js

@@ -1,389 +0,0 @@
-/**
- * This file is deprecated in favor of clientGatewayService.
- */
-const mv = require('mv');
-const path = require('path');
-const util = require('util');
-
-const clientSettingsMap = require('../../shared/constants/clientSettingsMap');
-const rTorrentPropMap = require('../util/rTorrentPropMap');
-const torrentStatusMap = require('../../shared/constants/torrentStatusMap');
-
-const addTagsToRequest = (tagsArr, requestParameters) => {
-  if (tagsArr && tagsArr.length) {
-    const tags = tagsArr
-      .reduce((accumulator, currentTag) => {
-        const tag = encodeURIComponent(currentTag.trim());
-
-        if (tag !== '' && accumulator.indexOf(tag) === -1) {
-          accumulator.push(tag);
-        }
-
-        return accumulator;
-      }, [])
-      .join(',');
-
-    requestParameters.push(`d.custom1.set="${tags}"`);
-  }
-
-  return requestParameters;
-};
-
-const getEnsuredArray = (item) => {
-  if (!util.isArray(item)) {
-    return [item];
-  }
-  return item;
-};
-
-const getMethodCall = (methodName, params) => {
-  params = params || [];
-  return {methodName, params};
-};
-
-class ClientRequest {
-  constructor(user, services, options) {
-    options = options || {};
-
-    this.services = services;
-    this.user = user;
-    this.clientRequestManager = this.services.clientRequestManager;
-
-    this.onCompleteFn = null;
-    this.postProcessFn = null;
-    this.requests = [];
-
-    if (options.onComplete) {
-      this.onCompleteFn = options.onComplete;
-    }
-
-    if (options.postProcess) {
-      this.postProcessFn = options.postProcess;
-    }
-
-    if (options.name) {
-      this.name = options.name;
-    }
-  }
-
-  clearRequestQueue() {
-    this.requests = [];
-  }
-
-  handleError(error) {
-    if (error.code === 'ECONNREFUSED') {
-      console.error(
-        `Connection refused at ${error.address}${error.port ? `:${error.port}` : ''}. ` +
-          'Check these values in config.js and ensure that rTorrent is running.',
-      );
-    }
-
-    this.clearRequestQueue();
-
-    if (this.onCompleteFn) {
-      this.onCompleteFn(null, error);
-    }
-  }
-
-  handleSuccess(data) {
-    let response = data;
-
-    this.clearRequestQueue();
-
-    if (this.postProcessFn) {
-      response = this.postProcessFn(data);
-    }
-
-    if (this.onCompleteFn) {
-      this.onCompleteFn(response);
-    }
-  }
-
-  onComplete(fn) {
-    this.onCompleteFn = fn;
-  }
-
-  postProcess(fn) {
-    this.postProcessFn = fn;
-  }
-
-  send() {
-    const handleSuccess = this.handleSuccess.bind(this);
-    const handleError = this.handleError.bind(this);
-
-    this.clientRequestManager.methodCall('system.multicall', [this.requests]).then(handleSuccess).catch(handleError);
-  }
-
-  // TODO: Separate these and add support for additional clients.
-  // rTorrent method calls.
-  addFiles(options) {
-    const files = getEnsuredArray(options.files);
-    const {path: destinationPath, isBasePath, start, tags: tagsArr} = options;
-
-    files.forEach((file) => {
-      let methodCall = 'load.raw_start';
-      let parameters = ['', file.buffer];
-      const timeAdded = Math.floor(Date.now() / 1000);
-
-      if (destinationPath) {
-        if (isBasePath) {
-          parameters.push(`d.directory_base.set="${destinationPath}"`);
-        } else {
-          parameters.push(`d.directory.set="${destinationPath}"`);
-        }
-      }
-
-      parameters = addTagsToRequest(tagsArr, parameters);
-
-      parameters.push(`d.custom.set=x-filename,${file.originalname}`);
-      parameters.push(`d.custom.set=addtime,${timeAdded}`);
-
-      // The start value is a string because it was appended to a FormData
-      // object.
-      if (start === 'false') {
-        methodCall = 'load.raw';
-      }
-
-      this.requests.push(getMethodCall(methodCall, parameters));
-    });
-  }
-
-  addURLs(options) {
-    const {path: destinationPath, isBasePath, start, tags: tagsArr} = options;
-    const urls = getEnsuredArray(options.urls);
-
-    urls.forEach((url) => {
-      let methodCall = 'load.start';
-      let parameters = ['', url];
-      const timeAdded = Math.floor(Date.now() / 1000);
-
-      if (destinationPath) {
-        if (isBasePath) {
-          parameters.push(`d.directory_base.set="${destinationPath}"`);
-        } else {
-          parameters.push(`d.directory.set="${destinationPath}"`);
-        }
-      }
-
-      parameters = addTagsToRequest(tagsArr, parameters);
-
-      parameters.push(`d.custom.set=addtime,${timeAdded}`);
-
-      if (!start) {
-        methodCall = 'load.normal';
-      }
-
-      this.requests.push(getMethodCall(methodCall, parameters));
-    });
-  }
-
-  checkHash(options) {
-    const {torrentService} = this.services;
-    const hashes = getEnsuredArray(options.hashes);
-    const stoppedHashes = hashes.filter((hash) =>
-      torrentService.getTorrent(hash).status.includes(torrentStatusMap.stopped),
-    );
-
-    const hashesToStart = [];
-
-    this.stopTorrents({hashes});
-
-    hashes.forEach((hash) => {
-      this.requests.push(getMethodCall('d.check_hash', [hash]));
-
-      if (!stoppedHashes.includes(hash)) {
-        hashesToStart.push(hash);
-      }
-    });
-
-    if (hashesToStart.length) {
-      this.startTorrents({hashes: hashesToStart});
-    }
-  }
-
-  fetchSettings(options) {
-    let {requestedSettings} = options;
-
-    if (requestedSettings == null) {
-      requestedSettings = clientSettingsMap.defaults.map((settingsKey) => clientSettingsMap[settingsKey]);
-    }
-
-    // Ensure client's response gets mapped to the correct requested keys.
-    if (options.setRequestedKeysArr) {
-      options.setRequestedKeysArr(requestedSettings);
-    }
-
-    requestedSettings.forEach((settingsKey) => {
-      this.requests.push(getMethodCall(settingsKey));
-    });
-  }
-
-  getTorrentDetails(options) {
-    const peerParams = [options.hash, ''].concat(options.peerProps);
-    const fileParams = [options.hash, ''].concat(options.fileProps);
-    const trackerParams = [options.hash, ''].concat(options.trackerProps);
-
-    this.requests.push(getMethodCall('p.multicall', peerParams));
-    this.requests.push(getMethodCall('f.multicall', fileParams));
-    this.requests.push(getMethodCall('t.multicall', trackerParams));
-  }
-
-  getTorrentList(options) {
-    this.requests.push(getMethodCall('d.multicall2', options.props));
-  }
-
-  getTransferData() {
-    Object.keys(rTorrentPropMap.transferData).forEach((key) => {
-      this.requests.push(getMethodCall(rTorrentPropMap.transferData[key]));
-    });
-  }
-
-  listMethods(options) {
-    const args = getEnsuredArray(options.args);
-    this.requests.push(getMethodCall(options.method, [args]));
-  }
-
-  moveTorrents(options) {
-    const {destinationPath} = options;
-    const filenames = getEnsuredArray(options.filenames);
-    const sourcePaths = getEnsuredArray(options.sourcePaths);
-
-    sourcePaths.forEach((source, index) => {
-      let callback = () => {};
-      const destination = `${destinationPath}${path.sep}${filenames[index]}`;
-      const isLastRequest = index + 1 === sourcePaths.length;
-
-      if (isLastRequest) {
-        callback = this.handleSuccess.bind(this);
-      }
-
-      if (source !== destination) {
-        mv(source, destination, {mkdirp: true}, callback);
-      } else if (isLastRequest) {
-        callback();
-      }
-    });
-  }
-
-  setDownloadPath(options) {
-    const hashes = getEnsuredArray(options.hashes);
-
-    let pathMethod;
-    if (options.isBasePath) {
-      pathMethod = 'd.directory_base.set';
-    } else {
-      pathMethod = 'd.directory.set';
-    }
-
-    hashes.forEach((hash) => {
-      this.requests.push(getMethodCall(pathMethod, [hash, options.path]));
-      this.requests.push(getMethodCall('d.open', [hash]));
-      this.requests.push(getMethodCall('d.close', [hash]));
-    });
-  }
-
-  setFilePriority(options) {
-    const fileIndices = getEnsuredArray(options.fileIndices);
-    const hashes = getEnsuredArray(options.hashes);
-
-    hashes.forEach((hash) => {
-      fileIndices.forEach((fileIndex) => {
-        this.requests.push(getMethodCall('f.priority.set', [`${hash}:f${fileIndex}`, options.priority]));
-      });
-      this.requests.push(getMethodCall('d.update_priorities', [hash]));
-    });
-  }
-
-  setPriority(options) {
-    const hashes = getEnsuredArray(options.hashes);
-
-    hashes.forEach((hash) => {
-      this.requests.push(getMethodCall('d.priority.set', [hash, options.priority]));
-      this.requests.push(getMethodCall('d.update_priorities', [hash]));
-    });
-  }
-
-  setSettings(options) {
-    const settings = getEnsuredArray(options.settings);
-
-    settings.forEach((setting) => {
-      if (setting.overrideLocalSetting) {
-        this.requests.push(getMethodCall(setting.id, setting.data));
-      } else {
-        this.requests.push(getMethodCall(`${clientSettingsMap[setting.id]}.set`, ['', setting.data]));
-      }
-    });
-  }
-
-  setTaxonomy(options) {
-    const methodName = 'd.custom1.set';
-
-    const tags = options.tags
-      .reduce((memo, currentTag) => {
-        const tag = encodeURIComponent(currentTag.trim());
-
-        if (tag !== '' && memo.indexOf(tag) === -1) {
-          memo.push(tag);
-        }
-
-        return memo;
-      }, [])
-      .join(',');
-
-    getEnsuredArray(options.hashes).forEach((hash) => {
-      this.requests.push(getMethodCall(methodName, [hash, tags]));
-    });
-  }
-
-  setTracker(options) {
-    const existingTrackerIndex = 0;
-    const {tracker} = options;
-
-    getEnsuredArray(options.hashes).forEach((hash) => {
-      // Disable existing tracker
-      this.requests.push(getMethodCall('t.disable', [`${hash}:t${existingTrackerIndex}`]));
-      // Insert new tracker
-      this.requests.push(getMethodCall('d.tracker.insert', [hash, `${existingTrackerIndex}`, tracker]));
-      // Save full session to apply tracker change
-      this.requests.push(getMethodCall('d.save_full_session', [hash]));
-    });
-  }
-
-  setThrottle(options) {
-    let methodName = 'throttle.global_down.max_rate.set';
-    if (options.direction === 'upload') {
-      methodName = 'throttle.global_up.max_rate.set';
-    }
-    this.requests.push(getMethodCall(methodName, ['', options.throttle]));
-  }
-
-  startTorrents(options) {
-    if (!options.hashes) {
-      console.error("startTorrents requires key 'hashes'.");
-      return;
-    }
-
-    getEnsuredArray(options.hashes).forEach((hash) => {
-      this.requests.push(getMethodCall('d.open', [hash]));
-      this.requests.push(getMethodCall('d.start', [hash]));
-    });
-  }
-
-  stopTorrents(options) {
-    if (!options.hashes) {
-      console.error("stopTorrents requires key 'hashes'.");
-      return;
-    }
-
-    getEnsuredArray(options.hashes).forEach((hash) => {
-      this.requests.push(getMethodCall('d.stop', [hash]));
-      this.requests.push(getMethodCall('d.close', [hash]));
-    });
-  }
-
-  getSessionPath() {
-    this.requests.push(getMethodCall('session.path'));
-  }
-}
-
-module.exports = ClientRequest;

package/server/models/Feed.js

@@ -1,74 +0,0 @@
-const FeedSub = require('feedsub');
-
-class Feed {
-  constructor(options) {
-    this.options = options || {};
-    this.options.maxItemHistory = options.maxItemHistory || 100;
-    this.items = [];
-
-    if (!options.url) {
-      console.error('Feed URL must be defined.');
-      return null;
-    }
-
-    this.reader = new FeedSub(options.url, {
-      autoStart: true,
-      emitOnStart: true,
-      maxHistory: this.options.maxItemHistory,
-      interval: options.interval ? Number(options.interval) : 15,
-      forceInterval: true,
-      readEveryItem: true,
-    });
-
-    this.initReader();
-  }
-
-  modify(options) {
-    Object.assign(this.options, options);
-    this.items = [];
-
-    this.reader = new FeedSub(options.url, {
-      autoStart: true,
-      emitOnStart: true,
-      maxHistory: this.options.maxItemHistory,
-      interval: options.interval ? Number(options.interval) : 15,
-      forceInterval: true,
-      readEveryItem: true,
-    });
-
-    this.initReader();
-  }
-
-  getItems() {
-    return this.items;
-  }
-
-  handleFeedItems(items) {
-    const nextLength = this.items.length + items.length;
-    if (nextLength >= this.options.maxItemHistory) {
-      const diff = nextLength - this.options.maxHistory;
-      this.items = this.items.splice(0, diff);
-    }
-
-    this.items = this.items.concat(items);
-
-    this.options.onNewItems({
-      feed: this.options,
-      items,
-    });
-  }
-
-  initReader() {
-    this.reader.on('items', this.handleFeedItems.bind(this));
-    this.reader.on('error', (error) => {
-      console.log('Feed reader error:', error);
-    });
-    this.reader.start();
-  }
-
-  stopReader() {
-    this.reader.stop();
-  }
-}
-
-module.exports = Feed;

package/server/models/Filesystem.js

@@ -1,45 +0,0 @@
-const fs = require('fs');
-const os = require('os');
-const path = require('path');
-
-const fileUtil = require('../util/fileUtil');
-
-const getDirectoryList = (options, callback) => {
-  const sourcePath = (options.path || '/').replace(/^~/, os.homedir());
-
-  const resolvedPath = fileUtil.sanitizePath(sourcePath);
-  if (!fileUtil.isAllowedPath(resolvedPath)) {
-    callback(null, fileUtil.accessDeniedError());
-    return;
-  }
-
-  try {
-    const directories = [];
-    const files = [];
-
-    fs.readdirSync(resolvedPath).forEach((item) => {
-      const joinedPath = path.join(resolvedPath, item);
-      if (fs.existsSync(joinedPath)) {
-        if (fs.statSync(joinedPath).isDirectory()) {
-          directories.push(item);
-        } else {
-          files.push(item);
-        }
-      }
-    });
-
-    const hasParent = /^.{0,}:?(\/|\\){1,1}\S{1,}/.test(resolvedPath);
-
-    callback({
-      directories,
-      files,
-      hasParent,
-      path: resolvedPath,
-      separator: path.sep,
-    });
-  } catch (error) {
-    callback(null, error);
-  }
-};
-
-module.exports = {getDirectoryList};