// @flow import { observable, action, computed, ObservableMap, runInAction, autorunAsync, } from 'mobx'; import { client } from 'utils/ApiClient'; import _ from 'lodash'; import invariant from 'invariant'; import BaseStore from 'stores/BaseStore'; import stores from 'stores'; import Document from 'models/Document'; import ErrorsStore from 'stores/ErrorsStore'; import CacheStore from 'stores/CacheStore'; const DOCUMENTS_CACHE_KEY = 'DOCUMENTS_CACHE_KEY'; type Options = { cache: CacheStore, }; class DocumentsStore extends BaseStore { @observable recentlyViewedIds: Array = []; @observable data: Map = new ObservableMap([]); @observable isLoaded: boolean = false; @observable isFetching: boolean = false; errors: ErrorsStore; cache: CacheStore; /* Computed */ @computed get recentlyViewed(): Array { return _.filter(this.data.values(), ({ id }) => this.recentlyViewedIds.includes(id) ); } @computed get recentlyEdited(): Array { // $FlowIssue return this.data.values(); } @computed get starred(): Array { return _.filter(this.data.values(), 'starred'); } /* Actions */ @action fetchAll = async (request: string = 'list'): Promise<*> => { this.isFetching = true; try { const res = await client.post(`/documents.${request}`); invariant(res && res.data, 'Document list not available'); const { data } = res; runInAction('DocumentsStore#fetchAll', () => { data.forEach(document => { this.data.set(document.id, new Document(document)); }); this.isLoaded = true; }); return data; } catch (e) { this.errors.add('Failed to load documents'); } finally { this.isFetching = false; } }; @action fetchRecentlyViewed = async (): Promise<*> => { const data = await this.fetchAll('viewed'); runInAction('DocumentsStore#fetchRecentlyViewed', () => { this.recentlyViewedIds = _.map(data, 'id'); }); }; @action fetchStarred = async (): Promise<*> => { await this.fetchAll('starred'); }; @action fetch = async (id: string): Promise<*> => { this.isFetching = true; try { const res = await client.post('/documents.info', { id }); invariant(res && res.data, 'Document not available'); const { data } = res; const document = new Document(data); runInAction('DocumentsStore#fetch', () => { this.data.set(data.id, document); this.isLoaded = true; }); return document; } catch (e) { this.errors.add('Failed to load documents'); } finally { this.isFetching = false; } }; @action add = (document: Document): void => { this.data.set(document.id, document); }; @action remove = (id: string): void => { this.data.delete(id); }; getById = (id: string): ?Document => { return this.data.get(id); }; getByUrl = (url: string): ?Document => { return _.find(this.data.values(), { url }); }; constructor(options: Options) { super(); this.errors = stores.errors; this.cache = options.cache; this.cache.getItem(DOCUMENTS_CACHE_KEY).then(data => { if (data) { data.forEach(document => this.add(new Document(document))); } }); this.on('document.delete', (data: { id: string }) => { this.remove(data.id); }); autorunAsync('DocumentsStore.persists', () => { if (this.data.size) { this.cache.setItem( DOCUMENTS_CACHE_KEY, Array.from(this.data.values()).map(collection => collection.data) ); } }); } } export default DocumentsStore;