diff --git a/app/models/Collection.ts b/app/models/Collection.ts index ec22f5dc8..7761f1e24 100644 --- a/app/models/Collection.ts +++ b/app/models/Collection.ts @@ -66,7 +66,10 @@ export default class Collection extends BaseModel { @computed get isEmpty(): boolean { - return this.documents.length === 0; + return ( + this.documents.length === 0 && + this.store.rootStore.documents.inCollection(this.id).length === 0 + ); } @computed diff --git a/app/stores/DocumentsStore.ts b/app/stores/DocumentsStore.ts index f01c13058..b64f2dff8 100644 --- a/app/stores/DocumentsStore.ts +++ b/app/stores/DocumentsStore.ts @@ -142,7 +142,12 @@ export default class DocumentsStore extends BaseStore { return []; } - return compact(collection.documents.map((node) => this.get(node.id))); + const drafts = this.drafts({ collectionId }); + + return compact([ + ...drafts, + ...collection.documents.map((node) => this.get(node.id)), + ]); } leastRecentlyUpdatedInCollection(collectionId: string): Document[] {