@jesec/flood 0.0.0-master.bf6f05c → 0.0.0-master.c0cce62

package/server/models/client.js

@@ -1,417 +0,0 @@
-const fs = require('fs');
-const path = require('path');
-const series = require('run-series');
-const tar = require('tar-stream');
-
-const ClientRequest = require('./ClientRequest');
-const clientResponseUtil = require('../util/clientResponseUtil');
-const clientSettingsMap = require('../../shared/constants/clientSettingsMap');
-const fileUtil = require('../util/fileUtil');
-const settings = require('./settings');
-const torrentFilePropsMap = require('../../shared/constants/torrentFilePropsMap');
-const torrentPeerPropsMap = require('../../shared/constants/torrentPeerPropsMap');
-const torrentStatusMap = require('../../shared/constants/torrentStatusMap');
-const torrentTrackerPropsMap = require('../../shared/constants/torrentTrackerPropsMap');
-
-const client = {
-  addFiles(user, services, req, callback) {
-    const {files} = req;
-    const {destination: destinationPath, isBasePath, start} = req.body;
-    let {tags} = req.body;
-    const request = new ClientRequest(user, services);
-
-    if (!Array.isArray(tags)) {
-      tags = tags.split(',');
-    }
-
-    const resolvedPath = fileUtil.sanitizePath(destinationPath);
-    if (!fileUtil.isAllowedPath(resolvedPath)) {
-      callback(null, fileUtil.accessDeniedError());
-      return;
-    }
-
-    fileUtil.createDirectory({path: resolvedPath});
-    request.send();
-
-    // Each torrent is sent individually because rTorrent accepts a total
-    // filesize of 524 kilobytes or less. This allows the user to send many
-    // torrent files reliably.
-    files.forEach((file, index) => {
-      file.originalname = encodeURIComponent(file.originalname);
-
-      const fileRequest = new ClientRequest(user, services);
-      fileRequest.addFiles({
-        files: file,
-        path: resolvedPath,
-        isBasePath,
-        start,
-        tags,
-      });
-
-      // Set the callback for only the last request.
-      if (index === files.length - 1) {
-        fileRequest.onComplete((response, error) => {
-          services.torrentService.fetchTorrentList();
-          callback(response, error);
-        });
-      }
-
-      fileRequest.send();
-    });
-
-    settings.set(user, {id: 'startTorrentsOnLoad', data: start === 'true' || start === true});
-  },
-
-  addUrls(user, services, data, callback) {
-    const {urls, destination, isBasePath, start, tags} = data;
-    const request = new ClientRequest(user, services);
-    const resolvedPath = fileUtil.sanitizePath(destination);
-    if (!fileUtil.isAllowedPath(resolvedPath)) {
-      callback(null, fileUtil.accessDeniedError());
-      return;
-    }
-    fileUtil.createDirectory({path: resolvedPath});
-    request.addURLs({
-      urls,
-      path: resolvedPath,
-      isBasePath,
-      start,
-      tags,
-    });
-    request.onComplete(callback);
-    request.send();
-
-    settings.set(user, {id: 'startTorrentsOnLoad', data: start});
-  },
-
-  checkHash(user, services, hashes, callback) {
-    const request = new ClientRequest(user, services);
-
-    request.checkHash({hashes});
-    request.onComplete((response, error) => {
-      services.torrentService.fetchTorrentList();
-      callback(response, error);
-    });
-    request.send();
-  },
-
-  downloadFiles(user, services, hash, fileString, res) {
-    try {
-      const selectedTorrent = services.torrentService.getTorrent(hash);
-      if (!selectedTorrent) return res.status(404).json({error: 'Torrent not found.'});
-
-      this.getTorrentDetails(user, services, hash, (torrentDetails) => {
-        if (!torrentDetails) return res.status(404).json({error: 'Torrent details not found'});
-
-        let files;
-        if (!fileString) {
-          files = torrentDetails.fileTree.files.map((x, i) => `${i}`);
-        } else {
-          files = fileString.split(',');
-        }
-
-        const filePathsToDownload = this.findFilesByIndicies(files, torrentDetails.fileTree).map((file) =>
-          path.join(selectedTorrent.directory, file.path),
-        );
-
-        if (filePathsToDownload.length === 1) {
-          const file = filePathsToDownload[0];
-          if (!fs.existsSync(file)) return res.status(404).json({error: 'File not found.'});
-
-          res.attachment(path.basename(file));
-          return res.download(file);
-        }
-
-        res.attachment(`${selectedTorrent.name}.tar`);
-
-        const pack = tar.pack();
-        pack.pipe(res);
-
-        const tasks = filePathsToDownload.map((filePath) => {
-          const filename = path.basename(filePath);
-
-          return (next) => {
-            fs.stat(filePath, (err, stats) => {
-              if (err) return next(err);
-
-              const stream = fs.createReadStream(filePath);
-              const entry = pack.entry(
-                {
-                  name: filename,
-                  size: stats.size,
-                },
-                next,
-              );
-              stream.pipe(entry);
-            });
-          };
-        });
-
-        series(tasks, (error) => {
-          if (error) return res.status(500).json(error);
-
-          pack.finalize();
-        });
-      });
-    } catch (error) {
-      res.status(500).json(error);
-    }
-  },
-
-  findFilesByIndicies(indices, fileTree = {}) {
-    const {directories, files = []} = fileTree;
-
-    let selectedFiles = files.filter((file) => indices.includes(`${file.index}`));
-
-    if (directories != null) {
-      selectedFiles = selectedFiles.concat(
-        Object.keys(directories).reduce(
-          (accumulator, directory) => accumulator.concat(this.findFilesByIndicies(indices, directories[directory])),
-          [],
-        ),
-      );
-    }
-
-    return selectedFiles;
-  },
-
-  getSettings(user, services, options, callback) {
-    let requestedSettingsKeys = [];
-    const request = new ClientRequest(user, services);
-    const response = {};
-
-    const outboundTransformation = {
-      throttleGlobalDownMax: (apiResponse) => Number(apiResponse) / 1024,
-      throttleGlobalUpMax: (apiResponse) => Number(apiResponse) / 1024,
-      piecesMemoryMax: (apiResponse) => Number(apiResponse) / (1024 * 1024),
-    };
-
-    request.fetchSettings({
-      options,
-      setRequestedKeysArr: (requestedSettingsKeysArr) => {
-        requestedSettingsKeys = requestedSettingsKeysArr;
-      },
-    });
-
-    request.postProcess((data) => {
-      if (!data) {
-        return null;
-      }
-
-      data.forEach((datum, index) => {
-        let value = datum[0];
-        const settingsKey = clientSettingsMap[requestedSettingsKeys[index]];
-
-        if (outboundTransformation[settingsKey]) {
-          value = outboundTransformation[settingsKey](value);
-        }
-
-        response[settingsKey] = value;
-      });
-
-      return response;
-    });
-    request.onComplete(callback);
-    request.send();
-  },
-
-  getTorrentDetails(user, services, hash, callback) {
-    const request = new ClientRequest(user, services);
-
-    request.getTorrentDetails({
-      hash,
-      fileProps: torrentFilePropsMap.methods,
-      peerProps: torrentPeerPropsMap.methods,
-      trackerProps: torrentTrackerPropsMap.methods,
-    });
-    request.postProcess(clientResponseUtil.processTorrentDetails);
-    request.onComplete(callback);
-    request.send();
-  },
-
-  listMethods(user, services, method, args, callback) {
-    const request = new ClientRequest(user, services);
-
-    request.listMethods({method, args});
-    request.onComplete(callback);
-    request.send();
-  },
-
-  moveTorrents(user, services, data, callback) {
-    const destinationPath = data.destination;
-    const {isBasePath, hashes, filenames, moveFiles, sourcePaths, isCheckHash} = data;
-    const mainRequest = new ClientRequest(user, services);
-
-    const resolvedPath = fileUtil.sanitizePath(destinationPath);
-    if (!fileUtil.isAllowedPath(resolvedPath)) {
-      callback(null, fileUtil.accessDeniedError());
-      return;
-    }
-
-    const hashesToRestart = hashes.filter(
-      (hash) => !services.torrentService.getTorrent(hash).status.includes(torrentStatusMap.stopped),
-    );
-
-    let afterCheckHash;
-
-    if (hashesToRestart.length) {
-      afterCheckHash = () => {
-        const startTorrentsRequest = new ClientRequest(user, services);
-        startTorrentsRequest.startTorrents({hashes: hashesToRestart});
-        startTorrentsRequest.onComplete(callback);
-        startTorrentsRequest.send();
-      };
-    } else {
-      afterCheckHash = callback;
-    }
-
-    let checkHash;
-
-    if (isCheckHash) {
-      checkHash = () => {
-        const checkHashRequest = new ClientRequest(user, services);
-        checkHashRequest.checkHash({hashes});
-        checkHashRequest.onComplete(afterCheckHash);
-        checkHashRequest.send();
-      };
-    } else {
-      checkHash = afterCheckHash;
-    }
-
-    const moveTorrents = () => {
-      const moveTorrentsRequest = new ClientRequest(user, services);
-      moveTorrentsRequest.onComplete(checkHash);
-      moveTorrentsRequest.moveTorrents({
-        filenames,
-        sourcePaths,
-        resolvedPath,
-      });
-    };
-
-    let afterSetPath = checkHash;
-
-    if (moveFiles) {
-      afterSetPath = moveTorrents;
-    }
-
-    mainRequest.stopTorrents({hashes});
-    mainRequest.setDownloadPath({hashes, path: resolvedPath, isBasePath});
-    mainRequest.onComplete(afterSetPath);
-    mainRequest.send();
-  },
-
-  setFilePriority(user, services, hashes, data, callback) {
-    // TODO Add support for multiple hashes.
-    const {fileIndices} = data;
-    const request = new ClientRequest(user, services);
-
-    request.setFilePriority({hashes, fileIndices, priority: data.priority});
-    request.onComplete((response, error) => {
-      services.torrentService.fetchTorrentList();
-      callback(response, error);
-    });
-    request.send();
-  },
-
-  setPriority(user, services, hashes, data, callback) {
-    const request = new ClientRequest(user, services);
-
-    request.setPriority({hashes, priority: data.priority});
-    request.onComplete((response, error) => {
-      services.torrentService.fetchTorrentList();
-      callback(response, error);
-    });
-    request.send();
-  },
-
-  setSettings(user, services, payloads, callback) {
-    const request = new ClientRequest(user, services);
-    if (payloads.length === 0) return callback({});
-
-    const inboundTransformations = new Map();
-    inboundTransformations
-      .set('throttleGlobalDownMax', (userInput) => ({
-        id: userInput.id,
-        data: Number(userInput.data) * 1024,
-      }))
-      .set('throttleGlobalUpMax', (userInput) => ({
-        id: userInput.id,
-        data: Number(userInput.data) * 1024,
-      }))
-      .set('piecesMemoryMax', (userInput) => ({
-        id: userInput.id,
-        data: (Number(userInput.data) * 1024 * 1024).toString(),
-      }));
-
-    const transformedPayloads = payloads.map((payload) => {
-      if (inboundTransformations.has(payload.id)) {
-        const inboundTransformation = inboundTransformations.get(payload.id);
-        return inboundTransformation(payload);
-      }
-
-      return payload;
-    });
-
-    request.setSettings({settings: transformedPayloads});
-    request.onComplete(callback);
-    request.send();
-  },
-
-  setSpeedLimits(user, services, data, callback) {
-    const request = new ClientRequest(user, services);
-
-    request.setThrottle({
-      direction: data.direction,
-      throttle: data.throttle,
-    });
-    request.onComplete(callback);
-    request.send();
-  },
-
-  setTaxonomy(user, services, data, callback) {
-    const request = new ClientRequest(user, services);
-
-    request.setTaxonomy(data);
-    request.onComplete((response, error) => {
-      // Fetch the latest torrent list to re-index the taxonomy.
-      services.torrentService.fetchTorrentList();
-      callback(response, error);
-    });
-    request.send();
-  },
-
-  setTracker(user, services, data, callback) {
-    const request = new ClientRequest(user, services);
-
-    request.setTracker(data);
-    request.onComplete((response, error) => {
-      // Fetch the latest torrent list to re-index trackerURI.
-      services.torrentService.fetchTorrentList();
-      callback(response, error);
-    });
-    request.send();
-  },
-
-  stopTorrent(user, services, hashes, callback) {
-    const request = new ClientRequest(user, services);
-    request.stopTorrents({hashes});
-    request.onComplete((response, error) => {
-      services.torrentService.fetchTorrentList();
-      callback(response, error);
-    });
-    request.send();
-  },
-
-  startTorrent(user, services, hashes, callback) {
-    const request = new ClientRequest(user, services);
-
-    request.startTorrents({hashes});
-    request.onComplete((response, error) => {
-      services.torrentService.fetchTorrentList();
-      callback(response, error);
-    });
-    request.send();
-  },
-};
-
-module.exports = client;

package/server/models/settings.js

@@ -1,138 +0,0 @@
-const _ = require('lodash');
-const Datastore = require('nedb');
-const path = require('path');
-
-const config = require('../../config');
-
-const databases = new Map();
-
-const changedKeys = {
-  downloadRate: 'downRate',
-  downloadTotal: 'downTotal',
-  uploadRate: 'upRate',
-  uploadTotal: 'upTotal',
-  connectedPeers: 'peersConnected',
-  totalPeers: 'peersTotal',
-  connectedSeeds: 'seedsConnected',
-  totalSeeds: 'seedsTotal',
-  added: 'dateAdded',
-  creationDate: 'dateCreated',
-  trackers: 'trackerURIs',
-};
-
-const removedKeys = ['freeDiskSpace'];
-
-/**
- * Check settings for old torrent propery keys. If the old keys exist and have
- * been assigned values, then check that the new key doesn't also exist. When
- * the new key does not exist, add the new key and assign it the old key's
- * value.
- *
- * @param  {Object} settings - the stored settings object.
- * @return {Object} - the settings object, altered if legacy keys exist.
- */
-const transformLegacyKeys = (settings) => {
-  if (settings.sortTorrents && settings.sortTorrents.property in changedKeys) {
-    settings.sortTorrents.property = changedKeys[settings.sortTorrents.property];
-  }
-
-  if (settings.torrentDetails) {
-    settings.torrentDetails = settings.torrentDetails.reduce((accumulator, detailItem) => {
-      if (
-        detailItem &&
-        detailItem.id in changedKeys &&
-        !settings.torrentDetails.some((subDetailItem) => subDetailItem.id === changedKeys[detailItem.id])
-      ) {
-        detailItem.id = changedKeys[detailItem.id];
-      }
-
-      if (!removedKeys.includes(detailItem.id)) {
-        accumulator.push(detailItem);
-      }
-
-      return accumulator;
-    }, []);
-  }
-
-  if (settings.torrentListColumnWidths) {
-    Object.keys(settings.torrentListColumnWidths).forEach((columnID) => {
-      if (columnID in changedKeys && !(changedKeys[columnID] in settings.torrentListColumnWidths)) {
-        settings.torrentListColumnWidths[changedKeys[columnID]] = settings.torrentListColumnWidths[columnID];
-      }
-    });
-  }
-
-  return settings;
-};
-
-function getDb(user) {
-  const userId = user._id;
-
-  if (databases.has(userId)) {
-    return databases.get(userId);
-  }
-
-  const database = new Datastore({
-    autoload: true,
-    filename: path.join(config.dbPath, userId, 'settings', 'settings.db'),
-  });
-
-  databases.set(userId, database);
-
-  return database;
-}
-
-const settings = {
-  get: (user, opts, callback) => {
-    const query = {};
-    const settingsToReturn = {};
-
-    if (opts.property) {
-      query.id = opts.property;
-    }
-
-    getDb(user)
-      .find(query)
-      .exec((err, docs) => {
-        if (err) {
-          callback(null, err);
-          return;
-        }
-
-        docs.forEach((doc) => {
-          settingsToReturn[doc.id] = doc.data;
-        });
-
-        callback(transformLegacyKeys(settingsToReturn));
-      });
-  },
-
-  set: (user, payloads, callback = _.noop) => {
-    const docsResponse = [];
-
-    if (!Array.isArray(payloads)) {
-      payloads = [payloads];
-    }
-
-    if (payloads && payloads.length) {
-      let error;
-      payloads.forEach((payload) => {
-        getDb(user).update({id: payload.id}, {$set: {data: payload.data}}, {upsert: true}, (err, docs) => {
-          docsResponse.push(docs);
-          if (err) {
-            error = err;
-          }
-        });
-      });
-      if (error) {
-        callback(null, error);
-      } else {
-        callback(docsResponse);
-      }
-    } else {
-      callback();
-    }
-  },
-};
-
-module.exports = settings;

package/server/routes/api.js

@@ -1,86 +0,0 @@
-const express = require('express');
-const passport = require('passport');
-
-const router = express.Router();
-
-const appendUserServices = require('../middleware/appendUserServices');
-const ajaxUtil = require('../util/ajaxUtil');
-const client = require('../models/client');
-const clientRoutes = require('./client');
-const clientActivityStream = require('../middleware/clientActivityStream');
-const eventStream = require('../middleware/eventStream');
-const Filesystem = require('../models/Filesystem');
-const mediainfo = require('../util/mediainfo');
-const settings = require('../models/settings');
-
-router.use('/', passport.authenticate('jwt', {session: false}), appendUserServices);
-
-router.use('/client', clientRoutes);
-
-router.get('/activity-stream', eventStream, clientActivityStream);
-
-router.get('/download', (req, res) => {
-  client.downloadFiles(req.user, req.services, req.query.hash, req.query.files, res);
-});
-
-router.delete('/feed-monitor/:id', (req, res) => {
-  req.services.feedService.removeItem(req.params.id, ajaxUtil.getResponseFn(res));
-});
-
-router.get('/feed-monitor', (req, res) => {
-  req.services.feedService.getAll(ajaxUtil.getResponseFn(res));
-});
-
-router.get('/feed-monitor/feeds', (req, res) => {
-  req.services.feedService.getFeeds(req.body.query, ajaxUtil.getResponseFn(res));
-});
-
-router.put('/feed-monitor/feeds', (req, res) => {
-  req.services.feedService.addFeed(req.body, ajaxUtil.getResponseFn(res));
-});
-
-router.put('/feed-monitor/feeds/:id', (req, res) => {
-  req.services.feedService.modifyFeed(req.params.id, req.body, ajaxUtil.getResponseFn(res));
-});
-
-router.get('/feed-monitor/rules', (req, res) => {
-  req.services.feedService.getRules(req.body.query, ajaxUtil.getResponseFn(res));
-});
-
-router.put('/feed-monitor/rules', (req, res) => {
-  req.services.feedService.addRule(req.body, ajaxUtil.getResponseFn(res));
-});
-
-router.get('/feed-monitor/items', (req, res) => {
-  req.services.feedService.getItems(req.query, ajaxUtil.getResponseFn(res));
-});
-
-router.get('/directory-list', (req, res) => {
-  Filesystem.getDirectoryList(req.query, ajaxUtil.getResponseFn(res));
-});
-
-router.get('/history', (req, res) => {
-  req.services.historyService.getHistory(req.query, ajaxUtil.getResponseFn(res));
-});
-
-router.get('/mediainfo', (req, res) => {
-  mediainfo.getMediainfo(req.user, req.query, ajaxUtil.getResponseFn(res));
-});
-
-router.get('/notifications', (req, res) => {
-  req.services.notificationService.getNotifications(req.query, ajaxUtil.getResponseFn(res));
-});
-
-router.delete('/notifications', (req, res) => {
-  req.services.notificationService.clearNotifications(req.query, ajaxUtil.getResponseFn(res));
-});
-
-router.get('/settings', (req, res) => {
-  settings.get(req.user, req.query, ajaxUtil.getResponseFn(res));
-});
-
-router.patch('/settings', (req, res) => {
-  settings.set(req.user, req.body, ajaxUtil.getResponseFn(res));
-});
-
-module.exports = router;