liberama/client/components/Reader/ServerStorage/ServerStorage.vue

<template>
    <div></div>
</template>

<script>
//-----------------------------------------------------------------------------
import Vue from 'vue';
import Component from 'vue-class-component';
import _ from 'lodash';

import bookManager from '../share/bookManager';
import readerApi from '../../../api/reader';
import * as utils from '../../../share/utils';
import * as cryptoUtils from '../../../share/cryptoUtils';

const maxSetTries = 5;

export default @Component({
    watch: {
        serverSyncEnabled: function() {
            this.serverSyncEnabledChanged();
        },
        serverStorageKey: function() {
            this.serverStorageKeyChanged(true);
        },
        settings: function() {
            this.debouncedSaveSettings();
        },
        profiles: function() {
            this.saveProfiles();
        },
        currentProfile: function() {
            this.currentProfileChanged(true);
        },
    },
})
class ServerStorage extends Vue {
    created() {
        this.inited = false;
        this.keyInited = false;
        this.commit = this.$store.commit;
        this.prevServerStorageKey = null;
        this.$root.$on('generateNewServerStorageKey', () => {this.generateNewServerStorageKey()});

        this.debouncedSaveSettings = _.debounce(() => {
            this.saveSettings();
        }, 500);

        this.oldProfiles = {};
        this.oldSettings = {};
        this.oldRecent = {};
        this.oldRecentLast = {};
        this.oldRecentLastDiff = {};
    }

    async init() {
        try {
            if (!this.serverStorageKey) {
                //генерируем новый ключ
                await this.generateNewServerStorageKey();
            } else {
                await this.serverStorageKeyChanged();
            }
            this.oldRecent = _.cloneDeep(bookManager.recent);
            this.oldRecentLast = _.cloneDeep(bookManager.recentLast) || {};
        } finally {
            this.inited = true;
        }
    }

    async generateNewServerStorageKey() {
        const key = utils.toBase58(utils.randomArray(32));
        this.commit('reader/setServerStorageKey', key);
        await this.serverStorageKeyChanged(true);
    }

    async serverSyncEnabledChanged() {
        if (this.serverSyncEnabled) {
            this.prevServerStorageKey = null;
            if (!this.serverStorageKey) {
                //генерируем новый ключ
                await this.generateNewServerStorageKey();
            } else {
                await this.serverStorageKeyChanged(true);
            }
        }
    }

    async serverStorageKeyChanged(force) {
        if (this.prevServerStorageKey != this.serverStorageKey) {
            this.prevServerStorageKey = this.serverStorageKey;
            this.hashedStorageKey = utils.toBase58(cryptoUtils.sha256(this.serverStorageKey));
            this.keyInited = true;

            await this.loadProfiles(force);
            this.checkCurrentProfile();
            await this.currentProfileChanged(force);
            await this.loadRecent(force);
            if (force)
                await this.saveRecent();
        }
    }

    async currentProfileChanged(force) {
        if (!this.currentProfile)
            return;

        await this.loadSettings(force);
    }

    get serverSyncEnabled() {
        return this.$store.state.reader.serverSyncEnabled;
    }

    get settings() {
        return this.$store.state.reader.settings;
    }

    get settingsRev() {
        return this.$store.state.reader.settingsRev;
    }

    get serverStorageKey() {
        return this.$store.state.reader.serverStorageKey;
    }

    get profiles() {
        return this.$store.state.reader.profiles;
    }

    get profilesRev() {
        return this.$store.state.reader.profilesRev;
    }

    get currentProfile() {
        return this.$store.state.reader.currentProfile;
    }

    get showServerStorageMessages() {
        return this.settings.showServerStorageMessages;
    }

    checkCurrentProfile() {
        if (!this.profiles[this.currentProfile]) {
            this.commit('reader/setCurrentProfile', '');
        }
    }

    notifySuccess() {
        this.success('Данные синхронизированы с сервером');
    }

    success(message) {
        if (this.showServerStorageMessages)
            this.$notify.success({message});
    }

    warning(message) {
        if (this.showServerStorageMessages)
            this.$notify.warning({message});
    }

    error(message) {
        if (this.showServerStorageMessages)
            this.$notify.error({message});
    }

    async loadSettings(force) {
        if (!this.keyInited || !this.serverSyncEnabled || !this.currentProfile)
            return;

        const setsId = `settings-${this.currentProfile}`;
        const oldRev = this.settingsRev[setsId] || 0;
        //проверим ревизию на сервере
        if (!force) {
            try {
                const revs = await this.storageCheck({[setsId]: {}});
                if (revs.state == 'success' && revs.items[setsId].rev == oldRev) {
                    return;
                }
            } catch(e) {
                this.error(`Ошибка соединения с сервером: ${e.message}`);
                return;
            }
        }

        let sets = null;
        try {
            sets = await this.storageGet({[setsId]: {}});
        } catch(e) {
            this.error(`Ошибка соединения с сервером: ${e.message}`);
            return;
        }

        if (sets.state == 'success') {
            sets = sets.items[setsId];

            if (sets.rev == 0)
                sets.data = {};

            this.oldSettings = _.cloneDeep(sets.data);
            this.commit('reader/setSettings', sets.data);
            this.commit('reader/setSettingsRev', {[setsId]: sets.rev});

            this.notifySuccess();
        } else {
            this.warning(`Неверный ответ сервера: ${sets.state}`);
        }
    }

    async saveSettings() {
        if (!this.keyInited || !this.serverSyncEnabled || !this.currentProfile || this.savingSettings)
            return;

        const diff = utils.getObjDiff(this.oldSettings, this.settings);
        if (utils.isEmptyObjDiff(diff))
            return;

        this.savingSettings = true;
        try {
            const setsId = `settings-${this.currentProfile}`;
            let result = {state: ''};
            let tries = 0;
            while (result.state != 'success' && tries < maxSetTries) {
                const oldRev = this.settingsRev[setsId] || 0;
                try {
                    result = await this.storageSet({[setsId]: {rev: oldRev + 1, data: this.settings}});
                } catch(e) {
                    this.savingSettings = false;
                    this.error(`Ошибка соединения с сервером (${e.message}). Данные не сохранены и могут быть перезаписаны.`);
                    return;
                }

                if (result.state == 'reject') {
                    await this.loadSettings(true);
                    const newSettings = utils.applyObjDiff(this.settings, diff);
                    this.commit('reader/setSettings', newSettings);
                }

                tries++;
            }

            if (tries >= maxSetTries) {
                //отменять изменения не будем, просто предупредим
                //this.commit('reader/setSettings', this.oldSettings);
                console.error(result);
                this.error('Не удалось отправить настройки на сервер. Данные не сохранены и могут быть перезаписаны.');
            } else {
                this.oldSettings = _.cloneDeep(this.settings);
                this.commit('reader/setSettingsRev', {[setsId]: this.settingsRev[setsId] + 1});
            }
        } finally {
            this.savingSettings = false;
        }
    }

    async loadProfiles(force) {
        if (!this.keyInited || !this.serverSyncEnabled)
            return;

        const oldRev = this.profilesRev;
        //проверим ревизию на сервере
        if (!force) {
            try {
                const revs = await this.storageCheck({profiles: {}});
                if (revs.state == 'success' && revs.items.profiles.rev == oldRev) {
                    return;
                }
            } catch(e) {
                this.error(`Ошибка соединения с сервером: ${e.message}`);
                return;
            }
        }

        let prof = null;
        try {
            prof = await this.storageGet({profiles: {}});
        } catch(e) {
            this.error(`Ошибка соединения с сервером: ${e.message}`);
            return;
        }

        if (prof.state == 'success') {
            prof = prof.items.profiles;

            if (prof.rev == 0)
                prof.data = {};

            this.oldProfiles = _.cloneDeep(prof.data);
            this.commit('reader/setProfiles', prof.data);
            this.commit('reader/setProfilesRev', prof.rev);

            this.notifySuccess();
        } else {
            this.warning(`Неверный ответ сервера: ${prof.state}`);
        }
    }

    async saveProfiles() {
        if (!this.keyInited || !this.serverSyncEnabled || this.savingProfiles)
            return;

        const diff = utils.getObjDiff(this.oldProfiles, this.profiles);
        if (utils.isEmptyObjDiff(diff))
            return;

        //обнуляются профили во время разработки, подстраховка
        if (!this.$store.state.reader.allowProfilesSave) {
            console.error('Сохранение профилей не санкционировано');
            return;
        }

        this.savingProfiles = true;
        try {
            let result = {state: ''};
            let tries = 0;
            while (result.state != 'success' && tries < maxSetTries) {
                try {
                    result = await this.storageSet({profiles: {rev: this.profilesRev + 1, data: this.profiles}});
                } catch(e) {
                    this.savingProfiles = false;
                    this.commit('reader/setProfiles', this.oldProfiles);
                    this.checkCurrentProfile();
                    this.error(`Ошибка соединения с сервером: (${e.message}). Изменения отменены.`);
                    return;
                }

                if (result.state == 'reject') {
                    await this.loadProfiles(true);
                    const newProfiles = utils.applyObjDiff(this.profiles, diff);
                    this.commit('reader/setProfiles', newProfiles);
                }

                tries++;
            }

            if (tries >= maxSetTries) {
                this.commit('reader/setProfiles', this.oldProfiles);
                this.checkCurrentProfile();
                console.error(result);
                this.error('Не удалось отправить данные на сервер. Изменения отменены.');
            } else {
                this.oldProfiles = _.cloneDeep(this.profiles);
                this.commit('reader/setProfilesRev', this.profilesRev + 1);        
            }
        } finally {
            this.savingProfiles = false;
        }
    }

    async loadRecent(force) {
        if (!this.keyInited || !this.serverSyncEnabled)
            return;

        const oldRev = bookManager.recentRev;
        const oldLastRev = bookManager.recentLastRev;
        const oldLastDiffRev = bookManager.recentLastDiffRev;
        //проверим ревизию на сервере
        let revs = null;
        if (!force) {
            try {
                revs = await this.storageCheck({recent: {}, recentLast: {}, recentLastDiff: {}});
                if (revs.state == 'success' && revs.items.recent.rev == oldRev &&
                    revs.items.recentLast.rev == oldLastRev &&
                    revs.items.recentLastDiff.rev == oldLastDiffRev) {
                    return;
                }
            } catch(e) {
                this.error(`Ошибка соединения с сервером: ${e.message}`);
                return;
            }
        }

        if (force || revs.items.recent.rev != oldRev) {
            let recent = null;
            try {
                recent = await this.storageGet({recent: {}});
            } catch(e) {
                this.error(`Ошибка соединения с сервером: ${e.message}`);
                return;
            }

            if (recent.state == 'success') {
                recent = recent.items.recent;

                if (recent.rev == 0)
                    recent.data = {};

                this.oldRecent = _.cloneDeep(recent.data);
                await bookManager.setRecent(recent.data);
                await bookManager.setRecentRev(recent.rev);
            } else {
                this.warning(`Неверный ответ сервера: ${recent.state}`);
            }
        }

        if (force || revs.items.recentLast.rev != oldLastRev || revs.items.recentLastDiff.rev != oldLastDiffRev) {
            let recentLast = null;
            try {
                recentLast = await this.storageGet({recentLast: {}, recentLastDiff: {}});
            } catch(e) {
                this.error(`Ошибка соединения с сервером: ${e.message}`);
                return;
            }

            if (recentLast.state == 'success') {
                const recentLastDiff = recentLast.items.recentLastDiff;
                recentLast = recentLast.items.recentLast;

                if (recentLast.rev == 0)
                    recentLast.data = {};
                if (recentLastDiff.rev == 0)
                    recentLastDiff.data = {};

                recentLast.data = utils.applyObjDiff(recentLast.data, recentLastDiff.data);

                this.oldRecentLast = _.cloneDeep(recentLast.data);
                await bookManager.setRecentLast(recentLast.data);
                await bookManager.setRecentLastRev(recentLast.rev);
                await bookManager.setRecentLastDiffRev(recentLastDiff.rev);
            } else {
                this.warning(`Неверный ответ сервера: ${recentLast.state}`);
            }
        }

        this.notifySuccess();
    }

    async saveRecent() {
        if (!this.keyInited || !this.serverSyncEnabled || this.savingRecent)
            return;

        const bm = bookManager;

        const diff = utils.getObjDiff(this.oldRecent, bm.recent);
        if (utils.isEmptyObjDiff(diff))
            return;

        this.savingRecent = true;
        try {
            let result = {state: ''};
            let tries = 0;
            while (result.state != 'success' && tries < maxSetTries) {
                try {
                    result = await this.storageSet({recent: {rev: bm.recentRev + 1, data: bm.recent}});
                } catch(e) {
                    this.savingRecent = false;
                    this.error(`Ошибка соединения с сервером: (${e.message}). Изменения не сохранены.`);
                    return;
                }

                if (result.state == 'reject') {
                    await this.loadRecent(true);
                    //похоже это лишнее
                    /*const newRecent = utils.applyObjDiff(bm.recent, diff);
                    await bm.setRecent(newRecent);*/
                }

                tries++;
            }

            if (tries >= maxSetTries) {
                console.error(result);
                this.error('Не удалось отправить данные на сервер. Данные не сохранены и могут быть перезаписаны.');
            } else {
                this.oldRecent = _.cloneDeep(bm.recent);
                await bm.setRecentRev(bm.recentRev + 1);
                await this.saveRecentLast(true);
            }
        } finally {
            this.savingRecent = false;
        }
    }

    async saveRecentLast(force = false) {
        if (!this.keyInited || !this.serverSyncEnabled || this.savingRecentLast)
            return;

        const bm = bookManager;
        let recentLast = bm.recentLast;
        recentLast = (recentLast ? recentLast : {});
        let lastRev = bm.recentLastRev;

        const diff = utils.getObjDiff(this.oldRecentLast, recentLast);
        if (utils.isEmptyObjDiff(diff))
            return;

        if (JSON.stringify(recentLast) > JSON.stringify(diff)) {
            await this.saveRecentLastDiff(diff, force);
            return;
        }

        this.savingRecentLast = true;
        try {
            let result = {state: ''};
            let tries = 0;
            while (result.state != 'success' && tries < maxSetTries) {
                if (force) {
                    try {
                        const revs = await this.storageCheck({recentLast: {}});
                        if (revs.items.recentLast.rev)
                            lastRev = revs.items.recentLast.rev;
                    } catch(e) {
                        this.error(`Ошибка соединения с сервером: ${e.message}`);
                        return;
                    }
                }

                try {
                    result = await this.storageSet({recentLast: {rev: lastRev + 1, data: recentLast}}, force);
                } catch(e) {
                    this.savingRecentLast = false;
                    this.error(`Ошибка соединения с сервером: (${e.message}). Изменения не сохранены.`);
                    return;
                }

                if (result.state == 'reject') {
                    await this.loadRecent(false);
                    this.savingRecentLast = false;//!!!
                    return;//!!!
                }

                tries++;
            }

            if (tries >= maxSetTries) {
                console.error(result);
                this.error('Не удалось отправить данные на сервер. Данные не сохранены и могут быть перезаписаны.');
            } else {
                this.oldRecentLast = _.cloneDeep(recentLast);
                await bm.setRecentLastRev(lastRev + 1);
            }
        } finally {
            this.savingRecentLast = false;
        }
    }

    async saveRecentLastDiff(diff, force = false) {
        if (!this.keyInited || !this.serverSyncEnabled || this.savingRecentLastDiff)
            return;

        const bm = bookManager;
        let lastRev = bm.recentLastDiffRev;

        const d = utils.getObjDiff(this.oldRecentLastDiff, diff);
        if (utils.isEmptyObjDiff(d))
            return;

        this.savingRecentLastDiff = true;
        try {
            let result = {state: ''};
            let tries = 0;
            while (result.state != 'success' && tries < maxSetTries) {
                if (force) {
                    try {
                        const revs = await this.storageCheck({recentLastDiff: {}});
                        if (revs.items.recentLastDiff.rev)
                            lastRev = revs.items.recentLastDiff.rev;
                    } catch(e) {
                        this.error(`Ошибка соединения с сервером: ${e.message}`);
                        return;
                    }
                }

                try {
                    result = await this.storageSet({recentLastDiff: {rev: lastRev + 1, data: diff}}, force);
                } catch(e) {
                    this.savingRecentLastDiff = false;
                    this.error(`Ошибка соединения с сервером: (${e.message}). Изменения не сохранены.`);
                    return;
                }

                if (result.state == 'reject') {
                    await this.loadRecent(false);
                    this.savingRecentLastDiff = false;
                    return;
                }

                tries++;
            }

            if (tries >= maxSetTries) {
                console.error(result);
                this.error('Не удалось отправить данные на сервер. Данные не сохранены и могут быть перезаписаны.');
            } else {
                this.oldRecentLastDiff = _.cloneDeep(diff);
                await bm.setRecentLastDiffRev(lastRev + 1);
            }
        } finally {
            this.savingRecentLastDiff = false;
        }
    }

    async storageCheck(items) {
        return await this.storageApi('check', items);
    }

    async storageGet(items) {
        return await this.storageApi('get', items);
    }

    async storageSet(items, force) {
        return await this.storageApi('set', items, force);
    }

    async storageApi(action, items, force) {
        const request = {action, items};
        if (force)
            request.force = true;
        const encodedRequest = await this.encodeStorageItems(request);
        return await this.decodeStorageItems(await readerApi.storage(encodedRequest));
    }

    async encodeStorageItems(request) {
        if (!this.hashedStorageKey)
            throw new Error('hashedStorageKey is empty');

        if (!_.isObject(request.items))
            throw new Error('items is not an object');

        let result = Object.assign({}, request);
        let items = {};
        for (const id of Object.keys(request.items)) {
            const item = request.items[id];
            if (request.action == 'set' && !_.isObject(item.data))
                throw new Error('encodeStorageItems: data is not an object');

            let encoded = Object.assign({}, item);

            if (item.data) {
                const comp = utils.pako.deflate(JSON.stringify(item.data), {level: 1});
                let encrypted = null;
                try {
                    encrypted = cryptoUtils.aesEncrypt(comp, this.serverStorageKey);
                } catch (e) {
                    throw new Error('encrypt failed');
                }
                encoded.data = '1' + utils.toBase64(encrypted);
            }
            items[`${this.hashedStorageKey}.${utils.toBase58(id)}`] = encoded;
        }

        result.items = items;
        return result;
    }

    async decodeStorageItems(response) {
        if (!this.hashedStorageKey)
            throw new Error('hashedStorageKey is empty');

        let result = Object.assign({}, response);
        let items = {};
        if (response.items) {
            if (!_.isObject(response.items))
                throw new Error('items is not an object');

            for (const id of Object.keys(response.items)) {
                const item = response.items[id];
                let decoded = Object.assign({}, item);
                if (item.data) {
                    if (!_.isString(item.data) || !item.data.length)
                        throw new Error('decodeStorageItems: data is not a string');
                    if (item.data[0] !== '1')
                        throw new Error('decodeStorageItems: unknown data format');

                    const a = utils.fromBase64(item.data.substr(1));
                    let decrypted = null;
                    try {
                        decrypted = cryptoUtils.aesDecrypt(a, this.serverStorageKey);
                    } catch (e) {
                        throw new Error('decrypt failed');
                    }
                    decoded.data = JSON.parse(utils.pako.inflate(decrypted, {to: 'string'}));
                }

                const ids = id.split('.');
                if (!(ids.length == 2) || !(ids[0] == this.hashedStorageKey))
                    throw new Error(`decodeStorageItems: bad id - ${id}`);
                items[utils.fromBase58(ids[1])] = decoded;
            }
        }

        result.items = items;
        return result;
    }
}
//-----------------------------------------------------------------------------
</script>