iceshrimp/src/api/endpoints/i/appdata/set.ts

/**
 * Module dependencies
 */
import $ from 'cafy';
import Appdata from '../../../models/appdata';
import User from '../../../models/user';
import serialize from '../../../serializers/user';
import event from '../../../event';

/**
 * Set app data
 *
 * @param {any} params
 * @param {any} user
 * @param {any} app
 * @param {Boolean} isSecure
 * @return {Promise<any>}
 */
module.exports = (params, user, app, isSecure) => new Promise(async (res, rej) => {
	// Get 'set' parameter
	const [set, setError] = $(params.set).optional.object()
		.pipe(obj => {
			return Object.entries(obj).some(kv => {
				const k = kv[0];
				const v = kv[1];
				return $(k).string().match(/[a-z_]+/).isNg() && $(v).string().isNg();
			});
		}).$;
	if (setError) return rej('invalid set param');

	// Get 'key' parameter
	const [key, keyError] = $(params.key).optional.string().match(/[a-z_]+/).$;
	if (keyError) return rej('invalid key param');

	// Get 'value' parameter
	const [value, valueError] = $(params.value).optional.string().$;
	if (valueError) return rej('invalid value param');

	let data = {};
	if (set) {
		data = set;
	} else {
		data[key] = value;
	}

	if (isSecure) {
		const _user = await User.findOneAndUpdate(user._id, {
			$set: { data }
		});

		res(204);

		// Publish i updated event
		event(user._id, 'i_updated', await serialize(_user, user, {
			detail: true,
			includeSecrets: true
		}));
	} else {
		await Appdata.update({
			app_id: app._id,
			user_id: user._id
		}, Object.assign({
			app_id: app._id,
			user_id: user._id
		}, {
			$set: { data }
		}), {
			upsert: true
		});

		res(204);
	}
});