From 17efcf6b51cfd80594ee7827cec75842dbeca3b3 Mon Sep 17 00:00:00 2001
From: Mar0xy <marie@kaifa.ch>
Date: Sun, 24 Sep 2023 02:48:16 +0200
Subject: [PATCH] upd: Masto API accounts endpoint finished

---
 .../api/mastodon/MastodonApiServerService.ts  | 214 +++++++++++++++++-
 .../server/api/mastodon/endpoints/account.ts  | 139 ++++++++++++
 2 files changed, 351 insertions(+), 2 deletions(-)

diff --git a/packages/backend/src/server/api/mastodon/MastodonApiServerService.ts b/packages/backend/src/server/api/mastodon/MastodonApiServerService.ts
index 320fdf52d5..fd802b9a73 100644
--- a/packages/backend/src/server/api/mastodon/MastodonApiServerService.ts
+++ b/packages/backend/src/server/api/mastodon/MastodonApiServerService.ts
@@ -3,9 +3,9 @@ import { Inject, Injectable } from '@nestjs/common';
 import type { UsersRepository } from '@/models/_.js';
 import { DI } from '@/di-symbols.js';
 import { bindThis } from '@/decorators.js';
-import megalodon, { MegalodonInterface } from "megalodon";
+import megalodon, { Entity, MegalodonInterface } from "megalodon";
 import type { FastifyInstance, FastifyPluginOptions } from 'fastify';
-import { convertId, IdConvertType as IdType, convertAccount, convertAnnouncement, convertFilter, convertAttachment, convertFeaturedTag } from './converters.js';
+import { convertId, IdConvertType as IdType, convertAccount, convertAnnouncement, convertFilter, convertAttachment, convertFeaturedTag, convertList } from './converters.js';
 import { IsNull } from 'typeorm';
 import type { Config } from '@/config.js';
 import { getInstance } from './endpoints/meta.js';
@@ -339,6 +339,216 @@ export class MastodonApiServerService {
                 reply.code(401).send(e.response.data);
             }
         });
+
+        fastify.get<{ Params: { id: string } }>("/v1/accounts/:id/lists", async (_request, reply) => {
+            const BASE_URL = `${_request.protocol}://${_request.hostname}`;
+            const accessTokens = _request.headers.authorization;
+            const client = getClient(BASE_URL, accessTokens);
+            try {
+                const data = await client.getAccountLists(convertId(_request.params.id, IdType.SharkeyId));
+                reply.send(data.data.map((list) => convertList(list)));
+            } catch (e: any) {
+                console.error(e);
+                console.error(e.response.data);
+                reply.code(401).send(e.response.data);
+            }
+        });
+
+        fastify.post<{ Params: { id: string } }>("/v1/accounts/:id/follow", async (_request, reply) => {
+            const BASE_URL = `${_request.protocol}://${_request.hostname}`;
+            const accessTokens = _request.headers.authorization;
+            const client = getClient(BASE_URL, accessTokens);
+            try {
+                const account = new apiAccountMastodon(_request, client, BASE_URL);
+                reply.send(await account.addFollow());
+            } catch (e: any) {
+                console.error(e);
+                console.error(e.response.data);
+                reply.code(401).send(e.response.data);
+            }
+        });
+
+        fastify.post<{ Params: { id: string } }>("/v1/accounts/:id/unfollow", async (_request, reply) => {
+            const BASE_URL = `${_request.protocol}://${_request.hostname}`;
+            const accessTokens = _request.headers.authorization;
+            const client = getClient(BASE_URL, accessTokens);
+            try {
+                const account = new apiAccountMastodon(_request, client, BASE_URL);
+                reply.send(await account.rmFollow());
+            } catch (e: any) {
+                console.error(e);
+                console.error(e.response.data);
+                reply.code(401).send(e.response.data);
+            }
+        });
+
+        fastify.post<{ Params: { id: string } }>("/v1/accounts/:id/block", async (_request, reply) => {
+            const BASE_URL = `${_request.protocol}://${_request.hostname}`;
+            const accessTokens = _request.headers.authorization;
+            const client = getClient(BASE_URL, accessTokens);
+            try {
+                const account = new apiAccountMastodon(_request, client, BASE_URL);
+                reply.send(await account.addBlock());
+            } catch (e: any) {
+                console.error(e);
+                console.error(e.response.data);
+                reply.code(401).send(e.response.data);
+            }
+        });
+
+        fastify.post<{ Params: { id: string } }>("/v1/accounts/:id/unblock", async (_request, reply) => {
+            const BASE_URL = `${_request.protocol}://${_request.hostname}`;
+            const accessTokens = _request.headers.authorization;
+            const client = getClient(BASE_URL, accessTokens);
+            try {
+                const account = new apiAccountMastodon(_request, client, BASE_URL);
+                reply.send(await account.rmBlock());
+            } catch (e: any) {
+                console.error(e);
+                console.error(e.response.data);
+                reply.code(401).send(e.response.data);
+            }
+        });
+
+        fastify.post<{ Params: { id: string } }>("/v1/accounts/:id/mute", async (_request, reply) => {
+            const BASE_URL = `${_request.protocol}://${_request.hostname}`;
+            const accessTokens = _request.headers.authorization;
+            const client = getClient(BASE_URL, accessTokens);
+            try {
+                const account = new apiAccountMastodon(_request, client, BASE_URL);
+                reply.send(await account.addMute());
+            } catch (e: any) {
+                console.error(e);
+                console.error(e.response.data);
+                reply.code(401).send(e.response.data);
+            }
+        });
+
+        fastify.post<{ Params: { id: string } }>("/v1/accounts/:id/unmute", async (_request, reply) => {
+            const BASE_URL = `${_request.protocol}://${_request.hostname}`;
+            const accessTokens = _request.headers.authorization;
+            const client = getClient(BASE_URL, accessTokens);
+            try {
+                const account = new apiAccountMastodon(_request, client, BASE_URL);
+                reply.send(await account.rmMute());
+            } catch (e: any) {
+                console.error(e);
+                console.error(e.response.data);
+                reply.code(401).send(e.response.data);
+            }
+        });
+
+        fastify.get("/v1/followed_tags", async (_request, reply) => {
+            const BASE_URL = `${_request.protocol}://${_request.hostname}`;
+            const accessTokens = _request.headers.authorization;
+            const client = getClient(BASE_URL, accessTokens);
+            try {
+                const data = await client.getFollowedTags();
+                reply.send(data.data);
+            } catch (e: any) {
+                console.error(e);
+                console.error(e.response.data);
+                reply.code(401).send(e.response.data);
+            }
+        });
+
+        fastify.get("/v1/bookmarks", async (_request, reply) => {
+            const BASE_URL = `${_request.protocol}://${_request.hostname}`;
+            const accessTokens = _request.headers.authorization;
+            const client = getClient(BASE_URL, accessTokens);
+            try {
+                const account = new apiAccountMastodon(_request, client, BASE_URL);
+                reply.send(await account.getBookmarks());
+            } catch (e: any) {
+                console.error(e);
+                console.error(e.response.data);
+                reply.code(401).send(e.response.data);
+            }
+        });
+
+        fastify.get("/v1/favourites", async (_request, reply) => {
+            const BASE_URL = `${_request.protocol}://${_request.hostname}`;
+            const accessTokens = _request.headers.authorization;
+            const client = getClient(BASE_URL, accessTokens);
+            try {
+                const account = new apiAccountMastodon(_request, client, BASE_URL);
+                reply.send(await account.getFavourites());
+            } catch (e: any) {
+                console.error(e);
+                console.error(e.response.data);
+                reply.code(401).send(e.response.data);
+            }
+        });
+
+        fastify.get("/v1/mutes", async (_request, reply) => {
+            const BASE_URL = `${_request.protocol}://${_request.hostname}`;
+            const accessTokens = _request.headers.authorization;
+            const client = getClient(BASE_URL, accessTokens);
+            try {
+                const account = new apiAccountMastodon(_request, client, BASE_URL);
+                reply.send(await account.getMutes());
+            } catch (e: any) {
+                console.error(e);
+                console.error(e.response.data);
+                reply.code(401).send(e.response.data);
+            }
+        });
+
+        fastify.get("/v1/blocks", async (_request, reply) => {
+            const BASE_URL = `${_request.protocol}://${_request.hostname}`;
+            const accessTokens = _request.headers.authorization;
+            const client = getClient(BASE_URL, accessTokens);
+            try {
+                const account = new apiAccountMastodon(_request, client, BASE_URL);
+                reply.send(await account.getBlocks());
+            } catch (e: any) {
+                console.error(e);
+                console.error(e.response.data);
+                reply.code(401).send(e.response.data);
+            }
+        });
+
+        fastify.get("/v1/follow_requests", async (_request, reply) => {
+            const BASE_URL = `${_request.protocol}://${_request.hostname}`;
+            const accessTokens = _request.headers.authorization;
+            const client = getClient(BASE_URL, accessTokens);
+            try {
+                const data = await client.getFollowRequests( ((_request.query as any) || { limit: 20 }).limit );
+                reply.send(data.data.map((account) => convertAccount(account as Entity.Account)));
+            } catch (e: any) {
+                console.error(e);
+                console.error(e.response.data);
+                reply.code(401).send(e.response.data);
+            }
+        });
+
+        fastify.post<{ Params: { id: string } }>("/v1/follow_requests/:id/authorize", async (_request, reply) => {
+            const BASE_URL = `${_request.protocol}://${_request.hostname}`;
+            const accessTokens = _request.headers.authorization;
+            const client = getClient(BASE_URL, accessTokens);
+            try {
+                const account = new apiAccountMastodon(_request, client, BASE_URL);
+                reply.send(await account.acceptFollow());
+            } catch (e: any) {
+                console.error(e);
+                console.error(e.response.data);
+                reply.code(401).send(e.response.data);
+            }
+        });
+
+        fastify.post<{ Params: { id: string } }>("/v1/follow_requests/:id/reject", async (_request, reply) => {
+            const BASE_URL = `${_request.protocol}://${_request.hostname}`;
+            const accessTokens = _request.headers.authorization;
+            const client = getClient(BASE_URL, accessTokens);
+            try {
+                const account = new apiAccountMastodon(_request, client, BASE_URL);
+                reply.send(await account.rejectFollow());
+            } catch (e: any) {
+                console.error(e);
+                console.error(e.response.data);
+                reply.code(401).send(e.response.data);
+            }
+        });
         //#endregion
 		done();
 	}
diff --git a/packages/backend/src/server/api/mastodon/endpoints/account.ts b/packages/backend/src/server/api/mastodon/endpoints/account.ts
index 205543f186..a9580aeedb 100644
--- a/packages/backend/src/server/api/mastodon/endpoints/account.ts
+++ b/packages/backend/src/server/api/mastodon/endpoints/account.ts
@@ -144,4 +144,143 @@ export class apiAccountMastodon {
             return e.response.data;
         }
     }
+
+    public async addFollow() {
+        try {
+            const data = await this.client.followAccount( convertId((this.request.params as any).id, IdType.SharkeyId) );
+            let acct = convertRelationship(data.data);
+            acct.following = true;
+			return acct;
+        } catch (e: any) {
+            console.error(e);
+			console.error(e.response.data);
+            return e.response.data;
+        }
+    }
+
+    public async rmFollow() {
+        try {
+            const data = await this.client.unfollowAccount( convertId((this.request.params as any).id, IdType.SharkeyId) );
+            let acct = convertRelationship(data.data);
+            acct.following = false;
+			return acct;
+        } catch (e: any) {
+            console.error(e);
+			console.error(e.response.data);
+            return e.response.data;
+        }
+    }
+
+    public async addBlock() {
+        try {
+            const data = await this.client.blockAccount( convertId((this.request.params as any).id, IdType.SharkeyId) );
+			return convertRelationship(data.data);
+        } catch (e: any) {
+            console.error(e);
+			console.error(e.response.data);
+            return e.response.data;
+        }
+    }
+
+    public async rmBlock() {
+        try {
+            const data = await this.client.unblockAccount( convertId((this.request.params as any).id, IdType.SharkeyId) );
+			return convertRelationship(data.data);
+        } catch (e: any) {
+            console.error(e);
+			console.error(e.response.data);
+            return e.response.data;
+        }
+    }
+
+    public async addMute() {
+        try {
+            const data = await this.client.muteAccount(
+                convertId((this.request.params as any).id, IdType.SharkeyId),
+                this.request.body as any
+            );
+			return convertRelationship(data.data);
+        } catch (e: any) {
+            console.error(e);
+			console.error(e.response.data);
+            return e.response.data;
+        }
+    }
+
+    public async rmMute() {
+        try {
+            const data = await this.client.unmuteAccount( convertId((this.request.params as any).id, IdType.SharkeyId) );
+			return convertRelationship(data.data);
+        } catch (e: any) {
+            console.error(e);
+			console.error(e.response.data);
+            return e.response.data;
+        }
+    }
+
+    public async getBookmarks() {
+        try {
+            const data = await this.client.getBookmarks( convertTimelinesArgsId(limitToInt(this.request.query as any)) );
+			return data.data.map((status) => convertStatus(status));
+        } catch (e: any) {
+            console.error(e);
+			console.error(e.response.data);
+            return e.response.data;
+        }
+    }
+
+    public async getFavourites() {
+        try {
+            const data = await this.client.getFavourites( convertTimelinesArgsId(limitToInt(this.request.query as any)) );
+			return data.data.map((status) => convertStatus(status));
+        } catch (e: any) {
+            console.error(e);
+			console.error(e.response.data);
+            return e.response.data;
+        }
+    }
+
+    public async getMutes() {
+        try {
+            const data = await this.client.getMutes( convertTimelinesArgsId(limitToInt(this.request.query as any)) );
+			return data.data.map((account) => convertAccount(account));
+        } catch (e: any) {
+            console.error(e);
+			console.error(e.response.data);
+            return e.response.data;
+        }
+    }
+
+    public async getBlocks() {
+        try {
+            const data = await this.client.getBlocks( convertTimelinesArgsId(limitToInt(this.request.query as any)) );
+			return data.data.map((account) => convertAccount(account));
+        } catch (e: any) {
+            console.error(e);
+			console.error(e.response.data);
+            return e.response.data;
+        }
+    }
+
+    public async acceptFollow() {
+        try {
+            const data = await this.client.acceptFollowRequest( convertId((this.request.params as any).id, IdType.SharkeyId) );
+			return convertRelationship(data.data);
+        } catch (e: any) {
+            console.error(e);
+			console.error(e.response.data);
+            return e.response.data;
+        }
+    }
+
+    public async rejectFollow() {
+        try {
+            const data = await this.client.rejectFollowRequest( convertId((this.request.params as any).id, IdType.SharkeyId) );
+			return convertRelationship(data.data);
+        } catch (e: any) {
+            console.error(e);
+			console.error(e.response.data);
+            return e.response.data;
+        }
+    }
 }
\ No newline at end of file