miiverse-api/src/util/xmlResponseGenerator.js

const xmlbuilder = require("xmlbuilder");
const moment = require("moment");
const database = require('../database');

class XmlResponseGenerator {
    /**
     * Generate response to reply request
     * @param posts
     * @param options
     * @returns xml
     * @constructor
     */
    static async RepliesResponse(posts, options) {
        let xml = xmlbuilder.create("result", { encoding: 'UTF-8' })
            .e("has_error", "0").up()
            .e("version", "1").up()
            .e("request_name", "replies").up()
            .e("posts");
        for (const post of posts) {
            postObj(xml, post, options);
        }
        xml = xml.up();
        return xml.end({ pretty: true, allowEmpty: true });
    }

    /**
     * Generate response to community posts response
     * @param posts
     * @param community
     * @param options
     * @returns xml
     * @constructor
     */
    static async PostsResponse(posts, community, options) {
        let xml = xmlbuilder.create("result", { encoding: 'UTF-8' })
            .e("has_error", "0").up()
            .e("version", "1").up()
            .e("request_name", options.name).up()
            .e("topic")
            .e("community_id", community.app_id ? community.app_id : community.community_id).up()
            .up()
            .e("posts");
        for (const post of posts) {
            postObj(xml, post, options);
        }
        xml = xml.up();
        return xml.end({ pretty: true, allowEmpty: true });
    }

    /**
     * Generate empty xml response
     * @returns xml
     * @constructor
     */
    static async EmptyResponse() {
        const xml = xmlbuilder.create("result", { encoding: 'UTF-8' })
            .e("has_error", "0").up()
            .e("version", "1").up();
        return xml.end({ pretty: true, allowEmpty: true });
    }

    /**
     * Generates response to list of communities request
     * @param communities
     * @returns xml
     * @constructor
     */
    static async Communities(communities) {
        let parent = communities[0].community_id;
        let xml = xmlbuilder.create("result", { encoding: 'UTF-8' })
            .e("has_error", "0").up()
            .e("version", "1").up()
            .e("request_name", "communities").up()
            .e("communities");
        for(let community of communities) {
        xml = xml.e("community")
                .e('olive_community_id', parent).up()
                .e('community_id', community.app_id ? community.app_id.padStart(6, '0') : community.community_id).up()
                .e("name", community.name).up()
                .e("description", community.description).up()
                .e("icon").up()
                .e("icon_3ds").up()
                .e("pid").up()
                .e("app_data", community.app_data).up()
                .e("is_user_community", 0).up()
                .up()
        }
        return xml.up().end({ pretty: true, allowEmpty: true});
    }

    /**
     * Generate response to request for single post
     * @param post
     * @returns xml
     * @constructor
     */
    static async SinglePostResponse(post) {
        let xml = xmlbuilder.create("result", { encoding: 'UTF-8' })
            .e("has_error", "0").up()
            .e("version", "1").up()
            .e("post");
        postObj(xml, post, { with_mii: true });
        xml = xml.up();
        return xml.end({ pretty: true, allowEmpty: true });
    }

    /**
     * Generate response to search for post
     * @param post
     * @returns xml
     * @constructor
     */
    static async QueryResponse(post) {
        let xml = xmlbuilder.create("result", { encoding: 'UTF-8' })
            .e("has_error", "0").up()
            .e("version", "1").up()
            .e("request_name", "posts.search").up()
            .e("posts");
        postObj(xml, post, { with_mii: true });
        xml = xml.up();
        return xml.end({ pretty: true, allowEmpty: true });
    }

    /**
     * Generate response to /v1/topics
     * @param communities
     * @returns xml
     */
    static async topics(communities) {
        const expirationDate = moment().add(1, 'days');
        let xml = xmlbuilder.create("result", { encoding: 'UTF-8' })
            .e("has_error", "0").up()
            .e("version", "1").up()
            .e("request_name", "topics").up()
            .e("expire", expirationDate.format('YYYY-MM-DD HH:MM:SS')).up()
            .e("topics");
        for (const community of communities) {
            let posts = await database.getNumberNewCommunityPostsByID(community, 30);
            xml = xml.e('topic')
                .e('empathy_count', community.empathy_count).up()
                .e('has_shop_page', community.has_shop_page).up()
                .e('icon', community.icon).up()
                .e('title_ids');
            community.title_id.forEach(function (title_id) {
                if(title_id !== '')
                    xml = xml.e('title_id', title_id).up();
            })
            xml = xml.up()
                .e('title_id', community.title_id[0]).up()
                .e('community_id', community.community_id).up()
                .e('is_recommended', community.is_recommended).up()
                .e('name', community.name).up()
                .e("people");
            for (const post of posts) {
                xml = xml.e("person")
                    .e("posts")
                    postObj(xml, post, { with_mii: true });
                    xml = xml.up().up();
            }
            xml = xml.up().up()
        }
        return xml.end({ pretty: false, allowEmpty: true });
    }

    /**
     * Generate response to /v1/people
     * @param people
     * @returns xml
     * @constructor
     */
    static async People(people) {
        let xml = xmlbuilder.create("result", { encoding: 'UTF-8' })
            .e("has_error", "0").up()
            .e("version", "1").up()
            .e("request_name", "user_infos").up()
            .e("people");
        for(let person of people) {
            xml = xml.e("person")
                .e('pid', person.pid).up()
                .e('screen_name', person.screen_name).up()
                .up()
        }
        return xml.up().end({ pretty: true, allowEmpty: true});
    }
}

/**
 * Generate xml for individual post
 * @param xml
 * @param post
 * @param options
 */
function postObj(xml, post, options) {
    xml = xml.e("post");
    if (post.app_data) {
        xml.e("app_data", post.app_data.replace(/[^A-Za-z0-9+/=]/g, "").replace(/[\n\r]+/gm, '').trim()).up();
    }
    xml.e("body", post.body ? post.body.replace(/[^A-Za-z\d\s-_!@#$%^&*(){}+=,.<>/?;:'"\[\]]/g, "") : "").up()
        .e("community_id", post.community_id).up()
        .e("country_id", post.country_id).up()
        .e("created_at", post.created_at).up()
        .e("feeling_id", post.feeling_id).up()
        .e("id", post.id).up()
        .e("is_autopost", post.is_autopost).up()
        .e("is_community_private_autopost", post.is_community_private_autopost).up()
        .e("is_spoiler", post.is_spoiler).up()
        .e("is_app_jumpable", post.is_app_jumpable).up()
        .e("empathy_count", post.empathy_count).up()
        .e("language_id", post.language_id).up();
    if(post.mii && options.with_mii) {
        xml.e("mii", post.mii.replace(/[^A-Za-z0-9+/=]/g, "").replace(/[\n\r]+/gm, '').trim()).up()
            .e("mii_face_url", post.mii_face_url).up()
    }
    xml.e("number", "0").up();
    if (post.painting) {
        xml.e("painting")
            .e("format", "tga").up()
            .e("content", post.painting.replace(/\r?\n|\r/g, "").trim()).up()
            .e("size", post.painting.length).up()
            .e("url", `https://pretendo-cdn.b-cdn.net/paintings/${post.pid}/${post.id}.png`).up()
            .up();
    }
    xml.e("pid", post.pid).up()
        .e("platform_id", post.platform_id).up()
        .e("region_id", post.region_id).up()
        .e("reply_count", post.reply_count).up()
        .e("screen_name", post.screen_name).up();
    if (post.screenshot && post.screenshot_length) {
        xml.e("screenshot")
            .e("size", post.screenshot_length).up()
            .e("url", `https://pretendo-cdn.b-cdn.net/screenshots/${post.pid}/${post.id}.jpg`).up()
            .up();
    }
    xml.e("title_id", post.title_id).up().up()
}

if (typeof module !== "undefined") {
    module.exports = XmlResponseGenerator;
}