firefish/packages/backend/src/server/api/mastodon/endpoints/status.ts
cutestnekoaqua 717aa899b1
feat: v1 Mastodon API
This commit adds (maybe unstable) support for Mastodons v1 api
also some v2 endpoints, maybe I miss stuff, I dont know.
We will need to test this but it should be kinda stable
and work like (old) butter.

Co-authored-by: Natty <natty.sh.git@gmail.com>
Co-authored-by: cutls <web-pro@cutls.com>
2023-02-09 23:21:50 +01:00

404 lines
16 KiB
TypeScript

import Router from "@koa/router";
import { koaBody } from 'koa-body';
import megalodon, { MegalodonInterface } from '@cutls/megalodon';
import { getClient } from '../ApiMastodonCompatibleService.js';
import fs from 'fs'
import { pipeline } from 'node:stream';
import { promisify } from 'node:util';
import { createTemp } from '@/misc/create-temp.js';
import { emojiRegex, emojiRegexAtStartToEnd } from '@/misc/emoji-regex.js';
import axios from 'axios';
const pump = promisify(pipeline);
export function apiStatusMastodon(router: Router): void {
router.post('/v1/statuses', koaBody(), async (ctx, reply) => {
const BASE_URL = `${ctx.protocol}://${ctx.hostname}`;
const accessTokens = ctx.headers.authorization;
const client = getClient(BASE_URL, accessTokens);
try {
const body: any = ctx.request.body
const text = body.status
const removed = text.replace(/@\S+/g, '').replaceAll(' ', '')
const isDefaultEmoji = emojiRegexAtStartToEnd.test(removed)
const isCustomEmoji = /^:[a-zA-Z0-9@_]+:$/.test(removed)
if (body.in_reply_to_id && isDefaultEmoji || isCustomEmoji) {
const a = await client.createEmojiReaction(body.in_reply_to_id, removed)
ctx.body = a.data
}
if (body.in_reply_to_id && removed === '/unreact') {
try {
const id = body.in_reply_to_id
const post = await client.getStatus(id)
const react = post.data.emoji_reactions.filter((e) => e.me)[0].name
const data = await client.deleteEmojiReaction(id, react);
ctx.body = data.data;
} catch (e: any) {
console.error(e)
ctx.status = (401);
ctx.body = e.response.data;
}
}
if (!body.media_ids) body.media_ids = undefined
if (body.media_ids && !body.media_ids.length) body.media_ids = undefined
const data = await client.postStatus(text, body);
ctx.body = data.data;
} catch (e: any) {
console.error(e)
ctx.status = (401);
ctx.body = e.response.data;
}
});
router.get<{ Params: { id: string } }>('/v1/statuses/:id', async (ctx, reply) => {
const BASE_URL = `${ctx.protocol}://${ctx.hostname}`;
const accessTokens = ctx.headers.authorization;
const client = getClient(BASE_URL, accessTokens);
try {
const data = await client.getStatus(ctx.params.id);
ctx.body = data.data;
} catch (e: any) {
console.error(e)
ctx.status = (401);
ctx.body = e.response.data;
}
});
router.delete<{ Params: { id: string } }>('/v1/statuses/:id', async (ctx, reply) => {
const BASE_URL = `${ctx.protocol}://${ctx.hostname}`;
const accessTokens = ctx.headers.authorization;
const client = getClient(BASE_URL, accessTokens);
try {
const data = await client.deleteStatus(ctx.params.id);
ctx.body = data.data;
} catch (e: any) {
console.error(e)
ctx.status = (401);
ctx.body = e.response.data;
}
});
interface IReaction {
id: string
createdAt: string
user: MisskeyEntity.User,
type: string
}
router.get<{ Params: { id: string } }>('/v1/statuses/:id/context', async (ctx, reply) => {
const BASE_URL = `${ctx.protocol}://${ctx.hostname}`;
const accessTokens = ctx.headers.authorization;
const client = getClient(BASE_URL, accessTokens);
try {
const id = ctx.params.id
const data = await client.getStatusContext(id, ctx.query as any);
const status = await client.getStatus(id);
const reactionsAxios = await axios.get(`${BASE_URL}/api/notes/reactions?noteId=${id}`)
const reactions: IReaction[] = reactionsAxios.data
const text = reactions.map((r) => `${r.type.replace('@.', '')} ${r.user.username}`).join('<br />')
data.data.descendants.unshift(statusModel(status.data.id, status.data.account.id, status.data.emojis, text))
ctx.body = data.data;
} catch (e: any) {
console.error(e)
ctx.status = (401);
ctx.body = e.response.data;
}
});
router.get<{ Params: { id: string } }>('/v1/statuses/:id/reblogged_by', async (ctx, reply) => {
const BASE_URL = `${ctx.protocol}://${ctx.hostname}`;
const accessTokens = ctx.headers.authorization;
const client = getClient(BASE_URL, accessTokens);
try {
const data = await client.getStatusRebloggedBy(ctx.params.id);
ctx.body = data.data;
} catch (e: any) {
console.error(e)
ctx.status = (401);
ctx.body = e.response.data;
}
});
router.get<{ Params: { id: string } }>('/v1/statuses/:id/favourited_by', async (ctx, reply) => {
ctx.body = []
});
router.post<{ Params: { id: string } }>('/v1/statuses/:id/favourite', async (ctx, reply) => {
const BASE_URL = `${ctx.protocol}://${ctx.hostname}`;
const accessTokens = ctx.headers.authorization;
const client = getClient(BASE_URL, accessTokens);
const react = await getFirstReaction(BASE_URL, accessTokens);
try {
const a = await client.createEmojiReaction(ctx.params.id, react) as any;
//const data = await client.favouriteStatus(ctx.params.id) as any;
ctx.body = a.data;
} catch (e: any) {
console.error(e)
console.error(e.response.data)
ctx.status = (401);
ctx.body = e.response.data;
}
});
router.post<{ Params: { id: string } }>('/v1/statuses/:id/unfavourite', async (ctx, reply) => {
const BASE_URL = `${ctx.protocol}://${ctx.hostname}`;
const accessTokens = ctx.headers.authorization;
const client = getClient(BASE_URL, accessTokens);
const react = await getFirstReaction(BASE_URL, accessTokens);
try {
const data = await client.deleteEmojiReaction(ctx.params.id, react);
ctx.body = data.data;
} catch (e: any) {
console.error(e)
ctx.status = (401);
ctx.body = e.response.data;
}
});
router.post<{ Params: { id: string } }>('/v1/statuses/:id/reblog', async (ctx, reply) => {
const BASE_URL = `${ctx.protocol}://${ctx.hostname}`;
const accessTokens = ctx.headers.authorization;
const client = getClient(BASE_URL, accessTokens);
try {
const data = await client.reblogStatus(ctx.params.id);
ctx.body = data.data;
} catch (e: any) {
console.error(e)
ctx.status = (401);
ctx.body = e.response.data;
}
});
router.post<{ Params: { id: string } }>('/v1/statuses/:id/unreblog', async (ctx, reply) => {
const BASE_URL = `${ctx.protocol}://${ctx.hostname}`;
const accessTokens = ctx.headers.authorization;
const client = getClient(BASE_URL, accessTokens);
try {
const data = await client.unreblogStatus(ctx.params.id);
ctx.body = data.data;
} catch (e: any) {
console.error(e)
ctx.status = (401);
ctx.body = e.response.data;
}
});
router.post<{ Params: { id: string } }>('/v1/statuses/:id/bookmark', async (ctx, reply) => {
const BASE_URL = `${ctx.protocol}://${ctx.hostname}`;
const accessTokens = ctx.headers.authorization;
const client = getClient(BASE_URL, accessTokens);
try {
const data = await client.bookmarkStatus(ctx.params.id);
ctx.body = data.data;
} catch (e: any) {
console.error(e)
ctx.status = (401);
ctx.body = e.response.data;
}
});
router.post<{ Params: { id: string } }>('/v1/statuses/:id/unbookmark', async (ctx, reply) => {
const BASE_URL = `${ctx.protocol}://${ctx.hostname}`;
const accessTokens = ctx.headers.authorization;
const client = getClient(BASE_URL, accessTokens);
try {
const data = await client.unbookmarkStatus(ctx.params.id) as any;
ctx.body = data.data;
} catch (e: any) {
console.error(e)
ctx.status = (401);
ctx.body = e.response.data;
}
});
router.post<{ Params: { id: string } }>('/v1/statuses/:id/pin', async (ctx, reply) => {
const BASE_URL = `${ctx.protocol}://${ctx.hostname}`;
const accessTokens = ctx.headers.authorization;
const client = getClient(BASE_URL, accessTokens);
try {
const data = await client.pinStatus(ctx.params.id);
ctx.body = data.data;
} catch (e: any) {
console.error(e)
ctx.status = (401);
ctx.body = e.response.data;
}
});
router.post<{ Params: { id: string } }>('/v1/statuses/:id/unpin', async (ctx, reply) => {
const BASE_URL = `${ctx.protocol}://${ctx.hostname}`;
const accessTokens = ctx.headers.authorization;
const client = getClient(BASE_URL, accessTokens);
try {
const data = await client.unpinStatus(ctx.params.id);
ctx.body = data.data;
} catch (e: any) {
console.error(e)
ctx.status = (401);
ctx.body = e.response.data;
}
});
router.post('/v1/media', async (ctx, reply) => {
const BASE_URL = `${ctx.protocol}://${ctx.hostname}`;
const accessTokens = ctx.headers.authorization;
const client = getClient(BASE_URL, accessTokens);
try {
const multipartData = await ctx.file;
if (!multipartData) {
ctx.body = { error: 'No image' };
return;
}
const [path] = await createTemp();
await pump(multipartData.buffer, fs.createWriteStream(path));
const image = fs.readFileSync(path);
const data = await client.uploadMedia(image);
ctx.body = data.data;
} catch (e: any) {
console.error(e)
ctx.status = (401);
ctx.body = e.response.data;
}
});
router.post('/v2/media', async (ctx, reply) => {
const BASE_URL = `${ctx.protocol}://${ctx.hostname}`;
const accessTokens = ctx.headers.authorization;
const client = getClient(BASE_URL, accessTokens);
try {
const multipartData = await ctx.file;
if (!multipartData) {
ctx.body = { error: 'No image' };
return;
}
const [path] = await createTemp();
await pump(multipartData.buffer, fs.createWriteStream(path));
const image = fs.readFileSync(path);
const data = await client.uploadMedia(image);
ctx.body = data.data;
} catch (e: any) {
console.error(e)
ctx.status = (401);
ctx.body = e.response.data;
}
});
router.get<{ Params: { id: string } }>('/v1/media/:id', async (ctx, reply) => {
const BASE_URL = `${ctx.protocol}://${ctx.hostname}`;
const accessTokens = ctx.headers.authorization;
const client = getClient(BASE_URL, accessTokens);
try {
const data = await client.getMedia(ctx.params.id);
ctx.body = data.data;
} catch (e: any) {
console.error(e)
ctx.status = (401);
ctx.body = e.response.data;
}
});
router.put<{ Params: { id: string } }>('/v1/media/:id', koaBody(), async (ctx, reply) => {
const BASE_URL = `${ctx.protocol}://${ctx.hostname}`;
const accessTokens = ctx.headers.authorization;
const client = getClient(BASE_URL, accessTokens);
try {
const data = await client.updateMedia(ctx.params.id, ctx.request.body as any);
ctx.body = data.data;
} catch (e: any) {
console.error(e)
ctx.status = (401);
ctx.body = e.response.data;
}
});
router.get<{ Params: { id: string } }>('/v1/polls/:id', async (ctx, reply) => {
const BASE_URL = `${ctx.protocol}://${ctx.hostname}`;
const accessTokens = ctx.headers.authorization;
const client = getClient(BASE_URL, accessTokens);
try {
const data = await client.getPoll(ctx.params.id);
ctx.body = data.data;
} catch (e: any) {
console.error(e)
ctx.status = (401);
ctx.body = e.response.data;
}
});
router.post<{ Params: { id: string } }>('/v1/polls/:id/votes', koaBody(), async (ctx, reply) => {
const BASE_URL = `${ctx.protocol}://${ctx.hostname}`;
const accessTokens = ctx.headers.authorization;
const client = getClient(BASE_URL, accessTokens);
try {
const data = await client.votePoll(ctx.params.id, (ctx.request.body as any).choices);
ctx.body = data.data;
} catch (e: any) {
console.error(e)
ctx.status = (401);
ctx.body = e.response.data;
}
});
}
async function getFirstReaction(BASE_URL: string, accessTokens: string | undefined) {
const accessTokenArr = accessTokens?.split(' ') ?? [null];
const accessToken = accessTokenArr[accessTokenArr.length - 1];
let react = '👍'
try {
const api = await axios.post(`${BASE_URL}/api/i/registry/get-unsecure`, {
scope: ['client', 'base'],
key: 'reactions',
i: accessToken
})
const reactRaw = api.data
react = Array.isArray(reactRaw) ? api.data[0] : '👍'
console.log(api.data)
return react
} catch (e) {
return react
}
}
export function statusModel(id: string | null, acctId: string | null, emojis: MastodonEntity.Emoji[], content: string) {
const now = "1970-01-02T00:00:00.000Z"
return {
id: '9atm5frjhb',
uri: 'https://http.cat/404', // ""
url: 'https://http.cat/404', // "",
account: {
id: '9arzuvv0sw',
username: 'ReactionBot',
acct: 'ReactionBot',
display_name: 'ReactionOfThisPost',
locked: false,
created_at: now,
followers_count: 0,
following_count: 0,
statuses_count: 0,
note: '',
url: 'https://http.cat/404',
avatar: 'https://http.cat/404',
avatar_static: 'https://http.cat/404',
header: 'https://http.cat/404', // ""
header_static: 'https://http.cat/404', // ""
emojis: [],
fields: [],
moved: null,
bot: false,
},
in_reply_to_id: id,
in_reply_to_account_id: acctId,
reblog: null,
content: `<p>${content}</p>`,
plain_content: null,
created_at: now,
emojis: emojis,
replies_count: 0,
reblogs_count: 0,
favourites_count: 0,
favourited: false,
reblogged: false,
muted: false,
sensitive: false,
spoiler_text: '',
visibility: 'public' as const,
media_attachments: [],
mentions: [],
tags: [],
card: null,
poll: null,
application: null,
language: null,
pinned: false,
emoji_reactions: [],
bookmarked: false,
quote: false,
}
}