Simplify previous fix

This commit is contained in:
Buu Nguyen 2017-09-01 13:27:29 -07:00
parent c0e271d1d5
commit 879a2bbd31
1 changed files with 5 additions and 6 deletions

View File

@ -21,11 +21,10 @@ class TreeView {
const $jstree = this.$jstree
$jstree.settings.core.data = (node, cb) => {
const loadAll = this.adapter.canLoadEntireTree() &&
this.store.get(STORE.LOADALL)
// No lazy loading in PR tree view
const isPR = this.store.get(STORE.PR) && repo.pullNumber
node = !loadAll && !isPR && (node.id === '#' ? {path: ''} : node.original)
const prMode = this.store.get(STORE.PR) && repo.pullNumber
const loadAll = this.adapter.canLoadEntireTree() && (prMode || this.store.get(STORE.LOADALL))
node = !loadAll && (node.id === '#' ? {path: ''} : node.original)
this.adapter.loadCodeTree({repo, token, node}, (err, treeData) => {
if (err) {
@ -33,7 +32,7 @@ class TreeView {
}
else {
treeData = this._sort(treeData)
if (loadAll || isPR) {
if (loadAll) {
treeData = this._collapse(treeData)
}
cb(treeData)