iceshrimp-161sh/src/server/api/call.ts

import { performance } from 'perf_hooks';
import limitter from './limitter';
import { IUser } from '../../models/user';
import { IApp } from '../../models/app';
import endpoints from './endpoints';

export default (endpoint: string, user: IUser, app: IApp, data: any, file?: any) => new Promise<any>(async (ok, rej) => {
	const isSecure = user != null && app == null;

	const ep = endpoints.find(e => e.name === endpoint);

	if (ep == null) {
		return rej('ENDPOINT_NOT_FOUND');
	}

	if (ep.meta.secure && !isSecure) {
		return rej('ACCESS_DENIED');
	}

	if (ep.meta.requireCredential && user == null) {
		return rej('SIGNIN_REQUIRED');
	}

	if (ep.meta.requireCredential && user.isSuspended) {
		return rej('YOUR_ACCOUNT_HAS_BEEN_SUSPENDED');
	}

	if (ep.meta.requireAdmin && !user.isAdmin) {
		return rej('YOU_ARE_NOT_ADMIN');
	}

	if (app && ep.meta.kind && !app.permission.some(p => p === ep.meta.kind)) {
		return rej('PERMISSION_DENIED');
	}

	if (ep.meta.requireCredential && ep.meta.limit) {
		try {
			await limitter(ep, user); // Rate limit
		} catch (e) {
			// drop request if limit exceeded
			return rej('RATE_LIMIT_EXCEEDED');
		}
	}

	let exec = ep.exec;

	if (ep.meta.requireFile && file) {
		exec = exec.bind(null, file);
	}

	let res;

	// API invoking
	try {
		const before = performance.now();
		res = await exec(data, user, app);
		const after = performance.now();

		const time = after - before;

		if (time > 1000) {
			console.warn(`SLOW API CALL DETECTED: ${ep.name} (${time}ms)`);
		}
	} catch (e) {
		rej(e);
		return;
	}

	ok(res);
});