webui-aria2/js/ctrls/main.js

angular
.module('webui.ctrls.download', [
  "ui.bootstrap",
  'webui.services.utils', 'webui.services.rpc', 'webui.services.alerts',
  'webui.services.settings', 'webui.services.modals', 'webui.services.configuration',
])
.controller('MainCtrl', [
  '$scope', '$name', '$enable', '$rpc', '$utils', '$alerts', '$modals',
  '$fileSettings', '$activeInclude', '$waitingExclude', '$pageSize',
  // for document title
  '$window',
function(
  scope, name, enable, rpc, utils, alerts, modals,
  fsettings, activeInclude, waitingExclude, pageSize,
  window
) {

  scope.name = name;   // default UI name
  scope.enable = enable; // UI enable options

  var re_slashes = /\\/g;
  var slash = "/";
  var allStopped = [];

  scope.active = [], scope.waiting = [], scope.stopped = [];
  scope.gstats = {};
  scope.hideLinkedMetadata = true;

  // pause the download
  // d: the download ctx
  scope.pause = function(d) {
    rpc.once('forcePause', [d.gid]);
  }

  // resume the download
  // d: the download ctx
  scope.resume = function(d) {
    rpc.once('unpause', [d.gid]);
  }

  scope.restart = function(d) {
    // assumes downloads which are started by URIs, not torrents.
    // the preferences are also not transferred, just simple restart

    rpc.once('getFiles', [d.gid], function(data) {
      var files = data[0];
      var uris =
        _.chain(files).map(function(f) { return f.uris })
        .filter(function(uris) { return uris && uris.length })
        .map(function(uris) {
          var u = _.chain(uris)
            .map(function(u) { return u.uri })
            .uniq().value();
          return u;
        }).value();

      if (uris.length > 0) {
        console.log('adding uris:', uris);
        scope.remove(d, function() {
          rpc.once('addUri', uris, angular.noop, true);
        }, true);
      }
    });
  }



  scope.canRestart = function(d) {
    return (['active', 'paused'].indexOf(d.status) == -1
      && !d.bittorrent);
  }

  // remove the download,
  // put it in stopped list if active,
  // otherwise permanantly remove it
  // d: the download ctx
  scope.remove = function(d, cb, noConfirm) {
    if (!noConfirm && !confirm("Remove %s and associated meta-data?".replace("%s", d.name))) {
      return;
    }

    var method = 'remove';

    if (scope.getType(d) == 'stopped')
      method = 'removeDownloadResult';

    if (d.followedFrom) {
      scope.remove(d.followedFrom, function() {}, true);
      d.followedFrom = null;
    }
    rpc.once(method, [d.gid], cb);
  }

// start filling in the model of active,
  // waiting and stopped download
  rpc.subscribe('tellActive', [], function(data) {
    scope.$apply(function() {
      utils.mergeMap(data[0], scope.active, scope.getCtx);
    });
  });

  rpc.subscribe('tellWaiting', [0, 1000], function(data) {
    scope.$apply(function() {
      utils.mergeMap(data[0], scope.waiting, scope.getCtx);
    });
  });


  rpc.subscribe('tellStopped', [0, 1000], function(data) {
    scope.$apply(function() {
      if (!scope.hideLinkedMetadata) {
        utils.mergeMap(data[0], scope.stopped, scope.getCtx);
        return;
      }
      utils.mergeMap(data[0], allStopped, scope.getCtx);
      var gids = {};
      _.forEach(allStopped, function(e) {
        gids[e.gid] = e;
      });
      _.forEach(scope.active, function(e) {
        gids[e.gid] = e;
      });
      _.forEach(scope.waiting, function(e) {
        gids[e.gid] = e;
      });
      scope.stopped = _.filter(allStopped, function(e) {
        if (!e.metadata || !e.followedBy || !(e.followedBy in gids)) {
          return true;
        }
        var linked = gids[e.followedBy];
        linked.followedFrom = e;
        return false;
      });
    });
  });

  rpc.subscribe('getGlobalStat', [], function(data) {
    scope.$apply(function() {
      scope.gstats = data[0];
      window.document.title = utils.getTitle(scope.gstats);
    });
  });

  rpc.once('getVersion', [], function(data) {
    scope.$apply(function() {
      scope.miscellaneous = data[0];
    });
  });

  // total number of downloads, updates dynamically as downloads are
  // stored in scope
  scope.totalDownloads = 0;

  // download search filter
  scope.downloadFilter = "";
  scope.downloadFilterCommitted = "";

  scope.onDownloadFilter = function() {
    if (scope.downloadFilterTimer) {
      clearTimeout(scope.downloadFilterTimer);
    }
    scope.downloadFilterTimer = setTimeout(function() {
      delete scope.downloadFilterTimer;
      if (scope.downloadFilterCommitted !== scope.downloadFilter) {
        scope.downloadFilterCommitted = scope.downloadFilter;
        scope.$digest();
      }
    }, 500);
  };

  scope.filterDownloads = function(downloads) {
    if (!scope.downloadFilterCommitted) {
      return downloads;
    }
    var filter = scope.downloadFilterCommitted.
      replace(/[{}()\[\]\\^$.?]/g, "\\$&").
      replace(/\*/g, ".*").
      replace(/\./g, ".");
    filter = new RegExp(filter, "i");
    return _.filter(downloads, function(d) {
      if (filter.test(d.name)) return true;
      return _.filter(d.files, function(f) {
        return filter.test(f.relpath);
      }).length;
    });
  };

  scope.clearFilter = function() {
    scope.downloadFilter = scope.downloadFilterCommitted = "";
  };

  scope.toggleStateFilters = function() {
    scope.filterSpeed = !scope.filterSpeed;
    scope.filterActive = !scope.filterActive;
    scope.filterWaiting = !scope.filterWaiting;
    scope.filterComplete = !scope.filterComplete;
    scope.filterError = !scope.filterError;
    scope.filterPaused = !scope.filterPaused;
    scope.filterRemoved = !scope.filterRemoved;
    scope.persistFilters();
  };

  scope.resetFilters = function() {
    scope.filterSpeed =
      scope.filterActive =
      scope.filterWaiting =
      scope.filterComplete =
      scope.filterError =
      scope.filterPaused =
      scope.filterRemoved =
      true;
    scope.clearFilter();
    scope.persistFilters();
  };

  scope.persistFilters = function() {
    var o = JSON.stringify({
      s: scope.filterSpeed,
      a: scope.filterActive,
      w: scope.filterWaiting,
      c: scope.filterComplete,
      e: scope.filterError,
      p: scope.filterPaused,
      r: scope.filterRemoved
    });
    utils.setCookie("aria2filters", o);
  };

  scope.loadFilters = function() {
    var o = JSON.parse(utils.getCookie("aria2filters"));
    if (!o) {
      scope.resetFilters();
      return;
    }
    scope.filterSpeed = !!o.s;
    scope.filterActive = !!o.a;
    scope.filterWaiting = !!o.w;
    scope.filterComplete = !!o.c;
    scope.filterError = !!o.e;
    scope.filterPaused = !!o.p;
    scope.filterRemoved = !!o.r;
  };

  scope.loadFilters();

  scope.toggleCollapsed = function(download) {
    if (!download.collapsed) {
      download.animCollapsed = true;
      // ng-unswitch after half a second.
      // XXX hacky way, because I'm to lazy right now to wire up proper
      // transitionend events.
      setTimeout(function() {
        scope.$apply(function() {
          download.collapsed = true;
        });
      }, 500);
      return;
    }

    download.collapsed = false;
    setTimeout(function() {
      scope.$apply(function() {
        download.animCollapsed = false;
      });
    }, 0);
  };


  // max downloads shown in one page
  scope.pageSize = pageSize;

  // current displayed page
  scope.currentPage = 1;

  // total amount of downloads returned by aria2
  scope.totalAria2Downloads = function() {
    return scope.active.length
      + scope.waiting.length
      + scope.stopped.length;
  }

  // actual downloads used by the view
  scope.getDownloads = function() {
    var downloads = [];
    if (scope.filterActive) {
      if (!scope.filterSpeed) {
        downloads = _.filter(scope.active, function (e) {
          return !+e.uploadSpeed && !+e.downloadSpeed;
        });
      }
      else {
        downloads = scope.active;
      }
    }
    else if (scope.filterSpeed) {
      downloads = _.filter(scope.active, function (e) {
        return +e.uploadSpeed || +e.downloadSpeed;
      });
    }
    if (scope.filterWaiting) {
      downloads = downloads.concat(_.filter(scope.waiting, function (e) {
        return e.status == "waiting";
      }));
    }
    if (scope.filterPaused) {
      downloads = downloads.concat(_.filter(scope.waiting, function (e) {
        return e.status == "paused";
      }));
    }
    if (scope.filterError) {
      downloads = downloads.concat(_.filter(scope.stopped, function (e) {
        return e.status == "error";
      }));
    }
    if (scope.filterComplete) {
      downloads = downloads.concat(_.filter(scope.stopped, function (e) {
        return e.status == "complete";
      }));
    }
    if (scope.filterRemoved) {
      downloads = downloads.concat(_.filter(scope.stopped, function (e) {
        return e.status == "removed";
      }));
    }

    downloads = scope.filterDownloads(downloads);

    scope.totalDownloads = downloads.length;

    downloads = downloads.slice( (scope.currentPage - 1) * scope.pageSize );
    downloads.splice( scope.pageSize );

    return downloads;
  }

  // convert the donwload form aria2 to once used by the view,
  // minor additions of some fields and checks
  scope.getCtx = function(d, ctx) {
    if (!ctx) {
      ctx = {
        dir: d.dir,
        status: d.status,
        gid: d.gid,
        followedBy: (d.followedBy && d.followedBy.length == 1
          ? d.followedBy[0] : null),
        followedFrom: null,
        numPieces: d.numPieces,
        connections: d.connections,
        bitfield: d.bitfield,
        totalLength: d.totalLength,
        fmtTotalLength: utils.fmtsize(d.totalLength),
        completedLength: d.completedLength,
        fmtCompletedLength: utils.fmtsize(d.completedLength),
        uploadLength: d.uploadLength,
        fmtUploadLength: utils.fmtsize(d.uploadLength),
        pieceLength: d.pieceLength,
        fmtPieceLength: utils.fmtsize(d.pieceLength),
        downloadSpeed: d.downloadSpeed,
        fmtDownloadSpeed: utils.fmtspeed(d.downloadSpeed),
        uploadSpeed: d.uploadSpeed,
        fmtUploadSpeed: utils.fmtspeed(d.uploadSpeed),
        collapsed: true,
        animCollapsed: true,
        files: [],
      };
    }
    else {
      ctx.dir = d.dir;
      ctx.status = d.status;
      ctx.gid = d.gid;
      ctx.followedBy = (d.followedBy && d.followedBy.length == 1
        ? d.followedBy[0] : null);
      ctx.followedFrom = null;
      ctx.numPieces = d.numPieces;
      ctx.connections = d.connections;
      ctx.bitfield = d.bitfield;
      if (ctx.totalLength !== d.totalLength) {
        ctx.totalLength = d.totalLength;
        ctx.fmtTotalLength = utils.fmtsize(d.totalLength);
      }
      if (ctx.completedLength !== d.completedLength) {
        ctx.completedLength = d.completedLength;
        ctx.fmtCompletedLength = utils.fmtsize(d.completedLength);
      }
      if (ctx.uploadLength !== d.uploadength) {
        ctx.uploadLength = d.uploadlength;
        ctx.fmtUploadLength = utils.fmtsize(d.uploadLength);
      }
      if (ctx.pieceLength !== d.pieceLength) {
        ctx.pieceLength = d.pieceLength;
        ctx.fmtPieceLength = utils.fmtsize(d.pieceLength);
      }
      if (ctx.downloadSpeed !== d.downloadSpeed) {
        ctx.downloadSpeed =  d.downloadSpeed;
        ctx.fmtDownloadSpeed = utils.fmtspeed(d.downloadSpeed);
      }
      if (ctx.uploadSpeed !== d.uploadSpeed) {
        ctx.uploadSpeed =  d.uploadSpeed;
        ctx.fmtUploadSpeed = utils.fmtspeed(d.uploadSpeed);
      }
    }

    var dlName;
    var files = d.files;
    if (files) {
      var cfiles = ctx.files;
      for (var i = 0; i < files.length; ++i) {
        var cfile = cfiles[i] || (cfiles[i] = {});
        var file = files[i];
        if (file.path !== cfile.path) {
          cfile.path = file.path;
          cfile.length = file.length;
          cfile.fmtLength = utils.fmtsize(file.length);
          cfile.relpath = file.path.replace(re_slashes, slash);
          if (!cfile.relpath) {
            cfile.relpath = (file.uris && file.uris[0] && file.uris[0].uri) || "Unknown";
          }
          else if (!cfile.relpath.startsWith("[")) { // METADATA
            cfile.relpath = cfile.relpath.substr(ctx.dir.length + 1);
          }
        }
      }
      cfiles.length = files.length;
      if (cfiles.length) {
        dlName = cfiles[0].relpath;
      }
    }
    else {
      delete ctx.files;
    }

    var btName;
    if (d.bittorrent) {
      btName = d.bittorrent.info && d.bittorrent.info.name;
      ctx.bittorrent = true;
    }
    else {
      delete ctx.bittorrent;
    }

    ctx.name = btName || dlName || "Unknown";
    ctx.metadata = ctx.name.startsWith("[METADATA]");
    if (ctx.metadata) {
      ctx.name = ctx.name.substr(10);
    }

    return ctx;
  };

  scope.hasStatus = function hasStatus(d, status) {
    if (_.isArray(status)) {
      if (status.length == 0) return false;
      return hasStatus(d, status[0]) || hasStatus(d, status.slice(1));
    }
    else {
      return d.status == status;
    }
  };

  // get time left for the download with
  // current download speed,
  // could be smarter by different heuristics
  scope.getEta = function(d) {
    return (d.totalLength-d.completedLength) / d.downloadSpeed;
  }

  scope.getProgressClass = function(d) {
    switch (d.status) {
      case "paused":
        return "progress-info";
      case "error":
        return "progress-danger";
      case "removed":
        return "progress-warning";
      case "active":
        return "progress-active";
      case "complete":
        return "progress-success";
      default:
        return "";
    }
  };

  // gets the progress in percentages
  scope.getProgress = function(d) {
    var percentage = (d.completedLength / d.totalLength)*100 || 0;
    percentage = percentage.toFixed(2);
    if(!percentage) percentage = 0;

    return percentage;
  };

  // gets the type for the download as classified by the aria2 rpc calls
  scope.getType = function(d) {
    var type = d.status;
    if (type == "paused") type = "waiting";
    if (["error", "removed", "complete"].indexOf(type) != -1)
      type = "stopped";
    return type;
  };

  scope.showSettings = function(d) {
    var type = scope.getType(d)
      , settings = {};

    rpc.once('getOption', [d.gid], function(data) {
      var vals = data[0];

      for (var i in fsettings) {
        if (type == 'active' && activeInclude.indexOf(i) == -1) continue;

        if (type == 'waiting' && waitingExclude.indexOf(i) != -1) continue;

        settings[i] = fsettings[i];
        settings[i].val = vals[i] || settings[i].val;
      }

      modals.invoke('settings', settings, scope.name + ' settings', function(settings) {
        var sets = {};
        for (var i in settings) { sets[i] = settings[i].val };

        rpc.once('changeOption', [d.gid, sets]);
      });
    });

    return false;
  }

}]);