szurubooru/client/js/models/pool.js

"use strict";

const api = require("../api.js");
const uri = require("../util/uri.js");
const events = require("../events.js");
const misc = require("../util/misc.js");

class Pool extends events.EventTarget {
    constructor() {
        const PostList = require("./post_list.js");

        super();
        this._orig = {};

        for (let obj of [this, this._orig]) {
            obj._posts = new PostList();
        }

        this._updateFromResponse({});
    }

    get id() {
        return this._id;
    }

    get names() {
        return this._names;
    }

    get category() {
        return this._category;
    }

    get description() {
        return this._description;
    }

    get posts() {
        return this._posts;
    }

    get postCount() {
        return this._postCount;
    }

    get creationTime() {
        return this._creationTime;
    }

    get lastEditTime() {
        return this._lastEditTime;
    }

    set names(value) {
        this._names = value;
    }

    set category(value) {
        this._category = value;
    }

    set description(value) {
        this._description = value;
    }

    static fromResponse(response) {
        const ret = new Pool();
        ret._updateFromResponse(response);
        return ret;
    }

    static get(id) {
        return api.get(uri.formatApiLink("pool", id)).then((response) => {
            return Promise.resolve(Pool.fromResponse(response));
        });
    }

    save() {
        const detail = { version: this._version };

        // send only changed fields to avoid user privilege violation
        if (misc.arraysDiffer(this._names, this._orig._names, true)) {
            detail.names = this._names;
        }
        if (this._category !== this._orig._category) {
            detail.category = this._category;
        }
        if (this._description !== this._orig._description) {
            detail.description = this._description;
        }
        if (misc.arraysDiffer(this._posts, this._orig._posts)) {
            detail.posts = this._posts.map((post) => post.id);
        }

        let promise = this._id
            ? api.put(uri.formatApiLink("pool", this._id), detail)
            : api.post(uri.formatApiLink("pools"), detail);
        return promise.then((response) => {
            this._updateFromResponse(response);
            this.dispatchEvent(
                new CustomEvent("change", {
                    detail: {
                        pool: this,
                    },
                })
            );
            return Promise.resolve();
        });
    }

    merge(targetId, addAlias) {
        return api
            .get(uri.formatApiLink("pool", targetId))
            .then((response) => {
                return api.post(uri.formatApiLink("pool-merge"), {
                    removeVersion: this._version,
                    remove: this._id,
                    mergeToVersion: response.version,
                    mergeTo: targetId,
                });
            })
            .then((response) => {
                if (!addAlias) {
                    return Promise.resolve(response);
                }
                return api.put(uri.formatApiLink("pool", targetId), {
                    version: response.version,
                    names: response.names.concat(this._names),
                });
            })
            .then((response) => {
                this._updateFromResponse(response);
                this.dispatchEvent(
                    new CustomEvent("change", {
                        detail: {
                            pool: this,
                        },
                    })
                );
                return Promise.resolve();
            });
    }

    delete() {
        return api
            .delete(uri.formatApiLink("pool", this._id), {
                version: this._version,
            })
            .then((response) => {
                this.dispatchEvent(
                    new CustomEvent("delete", {
                        detail: {
                            pool: this,
                        },
                    })
                );
                return Promise.resolve();
            });
    }

    _updateFromResponse(response) {
        const map = {
            _id: response.id,
            _version: response.version,
            _origName: response.names ? response.names[0] : null,
            _names: response.names,
            _category: response.category,
            _description: response.description,
            _creationTime: response.creationTime,
            _lastEditTime: response.lastEditTime,
            _postCount: response.postCount || 0,
        };

        for (let obj of [this, this._orig]) {
            obj._posts.sync(response.posts);
        }

        Object.assign(this, map);
        Object.assign(this._orig, map);
    }
}

module.exports = Pool;