szurubooru/client/js/models/post.js

'use strict';

const api = require('../api.js');
const uri = require('../util/uri.js');
const tags = require('../tags.js');
const events = require('../events.js');
const TagList = require('./tag_list.js');
const NoteList = require('./note_list.js');
const CommentList = require('./comment_list.js');
const PoolList = require('./pool_list.js');
const Pool = require('./pool.js');
const misc = require('../util/misc.js');

class Post extends events.EventTarget {
    constructor() {
        super();
        this._orig = {};

        for (let obj of [this, this._orig]) {
            obj._tags = new TagList();
            obj._notes = new NoteList();
            obj._comments = new CommentList();
            obj._pools = new PoolList();
        }

        this._updateFromResponse({});
    }

    get id() {
        return this._id;
    }

    get type() {
        return this._type;
    }

    get mimeType() {
        return this._mimeType;
    }

    get creationTime() {
        return this._creationTime;
    }

    get user() {
        return this._user;
    }

    get safety() {
        return this._safety;
    }

    get contentUrl() {
        return this._contentUrl;
    }

    get fullContentUrl() {
        return this._fullContentUrl;
    }

    get thumbnailUrl() {
        return this._thumbnailUrl;
    }

    get source() {
        return this._source;
    }

    get sourceSplit() {
        return this._source.split('\n');
    }

    get canvasWidth() {
        return this._canvasWidth || 800;
    }

    get canvasHeight() {
        return this._canvasHeight || 450;
    }

    get fileSize() {
        return this._fileSize || 0;
    }

    get newContent() {
        throw 'Invalid operation';
    }

    get newThumbnail() {
        throw 'Invalid operation';
    }

    get flags() {
        return this._flags;
    }

    get tags() {
        return this._tags;
    }

    get tagNames() {
        return this._tags.map(tag => tag.names[0]);
    }

    get notes() {
        return this._notes;
    }

    get comments() {
        return this._comments;
    }

    get relations() {
        return this._relations;
    }

    get pools() {
        return this._pools;
    }

    get score() {
        return this._score;
    }

    get commentCount() {
        return this._commentCount;
    }

    get favoriteCount() {
        return this._favoriteCount;
    }

    get ownFavorite() {
        return this._ownFavorite;
    }

    get ownScore() {
        return this._ownScore;
    }

    get hasCustomThumbnail() {
        return this._hasCustomThumbnail;
    }

    set flags(value) {
        this._flags = value;
    }

    set safety(value) {
        this._safety = value;
    }

    set relations(value) {
        this._relations = value;
    }

    set newContent(value) {
        this._newContent = value;
    }

    set newThumbnail(value) {
        this._newThumbnail = value;
    }

    set source(value) {
        this._source = value;
    }

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

    static reverseSearch(content) {
        let apiPromise = api.post(
            uri.formatApiLink('posts', 'reverse-search'),
            {},
            {content: content});
        let returnedPromise = apiPromise
            .then(response => {
                if (response.exactPost) {
                    response.exactPost = Post.fromResponse(response.exactPost);
                }
                for (let item of response.similarPosts) {
                    item.post = Post.fromResponse(item.post);
                }
                return Promise.resolve(response);
            });
        returnedPromise.abort = () => apiPromise.abort();
        return returnedPromise;
    }

    static get(id) {
        return api.get(uri.formatApiLink('post', id))
            .then(response => {
                return Promise.resolve(Post.fromResponse(response));
            });
    }

    _savePoolPosts() {
        const difference = (a, b) => a.filter(post => !b.hasPoolId(post.id));

        // find the pools where the post was added or removed
        const added = difference(this.pools, this._orig._pools);
        const removed = difference(this._orig._pools, this.pools);

        let ops = [];

        // update each pool's list of posts
        for (let pool of added) {
            let op = Pool.get(pool.id).then(response => {
                if (!response.posts.hasPostId(this._id)) {
                    response.posts.addById(this._id);
                    return response.save();
                } else {
                    return Promise.resolve(response);
                }
            });
            ops.push(op);
        }

        for (let pool of removed) {
            let op = Pool.get(pool.id).then(response => {
                if (response.posts.hasPostId(this._id)) {
                    response.posts.removeById(this._id);
                    return response.save();
                } else {
                    return Promise.resolve(response);
                }
            });
            ops.push(op);
        }

        return Promise.all(ops);
    }

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

        // send only changed fields to avoid user privilege violation
        if (anonymous === true) {
            detail.anonymous = true;
        }
        if (this._safety !== this._orig._safety) {
            detail.safety = this._safety;
        }
        if (misc.arraysDiffer(this._flags, this._orig._flags)) {
            detail.flags = this._flags;
        }
        if (misc.arraysDiffer(this._tags, this._orig._tags)) {
            detail.tags = this._tags.map(tag => tag.names[0]);
        }
        if (misc.arraysDiffer(this._relations, this._orig._relations)) {
            detail.relations = this._relations;
        }
        if (misc.arraysDiffer(this._notes, this._orig._notes)) {
            detail.notes = this._notes.map(note => ({
                polygon: note.polygon.map(point => [point.x, point.y]),
                text: note.text,
            }));
        }
        if (this._newContent) {
            files.content = this._newContent;
        }
        if (this._newThumbnail !== undefined) {
            files.thumbnail = this._newThumbnail;
        }
        if (this._source !== this._orig._source) {
            detail.source = this._source;
        }

        let apiPromise = this._id ?
            api.put(uri.formatApiLink('post', this.id), detail, files) :
            api.post(uri.formatApiLink('posts'), detail, files);

        return apiPromise.then(response => {
            if (misc.arraysDiffer(this._pools, this._orig._pools)) {
                return this._savePoolPosts()
                    .then(() => Promise.resolve(response));
            }
            return Promise.resolve(response);
        }).then(response => {
            this._updateFromResponse(response);
            this.dispatchEvent(
                new CustomEvent('change', {detail: {post: this}}));
            if (this._newContent) {
                this.dispatchEvent(
                    new CustomEvent('changeContent', {detail: {post: this}}));
            }
            if (this._newThumbnail) {
                this.dispatchEvent(
                    new CustomEvent('changeThumbnail', {detail: {post: this}}));
            }

            return Promise.resolve();
        }, error => {
            if (error.response &&
                error.response.name === 'PostAlreadyUploadedError') {
                error.message =
                  `Post already uploaded (@${error.response.otherPostId})`;
            }
            return Promise.reject(error);
        });
    }

    feature() {
        return api.post(
            uri.formatApiLink('featured-post'),
            {id: this._id})
            .then(response => {
                return Promise.resolve();
            });
    }

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

    merge(targetId, useOldContent) {
        return api.get(uri.formatApiLink('post', targetId))
            .then(response => {
                return api.post(uri.formatApiLink('post-merge'), {
                    removeVersion: this._version,
                    remove: this._id,
                    mergeToVersion: response.version,
                    mergeTo: targetId,
                    replaceContent: useOldContent,
                });
            }).then(response => {
                this._updateFromResponse(response);
                this.dispatchEvent(new CustomEvent('change', {
                    detail: {
                        post: this,
                    },
                }));
                return Promise.resolve();
            });
    }

    setScore(score) {
        return api.put(
            uri.formatApiLink('post', this.id, 'score'),
            {score: score})
            .then(response => {
                const prevFavorite = this._ownFavorite;
                this._updateFromResponse(response);
                if (this._ownFavorite !== prevFavorite) {
                    this.dispatchEvent(new CustomEvent('changeFavorite', {
                        detail: {
                            post: this,
                        },
                    }));
                }
                this.dispatchEvent(new CustomEvent('changeScore', {
                    detail: {
                        post: this,
                    },
                }));
                return Promise.resolve();
            });
    }

    addToFavorites() {
        return api.post(uri.formatApiLink('post', this.id, 'favorite'))
            .then(response => {
                const prevScore = this._ownScore;
                this._updateFromResponse(response);
                if (this._ownScore !== prevScore) {
                    this.dispatchEvent(new CustomEvent('changeScore', {
                        detail: {
                            post: this,
                        },
                    }));
                }
                this.dispatchEvent(new CustomEvent('changeFavorite', {
                    detail: {
                        post: this,
                    },
                }));
                return Promise.resolve();
            });
    }

    removeFromFavorites() {
        return api.delete(uri.formatApiLink('post', this.id, 'favorite'))
            .then(response => {
                const prevScore = this._ownScore;
                this._updateFromResponse(response);
                if (this._ownScore !== prevScore) {
                    this.dispatchEvent(new CustomEvent('changeScore', {
                        detail: {
                            post: this,
                        },
                    }));
                }
                this.dispatchEvent(new CustomEvent('changeFavorite', {
                    detail: {
                        post: this,
                    },
                }));
                return Promise.resolve();
            });
    }

    mutateContentUrl() {
        this._contentUrl =
          this._orig._contentUrl +
          '?bypass-cache=' +
          Math.round(Math.random() * 1000);
    }

    _updateFromResponse(response) {
        const map = () => ({
            _version: response.version,
            _id: response.id,
            _type: response.type,
            _mimeType: response.mimeType,
            _creationTime: response.creationTime,
            _user: response.user,
            _safety: response.safety,
            _contentUrl: response.contentUrl,
            _fullContentUrl: new URL(response.contentUrl, document.getElementsByTagName('base')[0].href).href,
            _thumbnailUrl: response.thumbnailUrl,
            _source: response.source,
            _canvasWidth: response.canvasWidth,
            _canvasHeight: response.canvasHeight,
            _fileSize: response.fileSize,

            _flags: [...response.flags || []],
            _relations: [...response.relations || []],

            _score: response.score,
            _commentCount: response.commentCount,
            _favoriteCount: response.favoriteCount,
            _ownScore: response.ownScore,
            _ownFavorite: response.ownFavorite,
            _hasCustomThumbnail: response.hasCustomThumbnail,
        });

        for (let obj of [this, this._orig]) {
            obj._tags.sync(response.tags);
            obj._notes.sync(response.notes);
            obj._comments.sync(response.comments);
            obj._pools.sync(response.pools);
        }

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

module.exports = Post;