diff options
author | Volpeon <git@volpeon.ink> | 2021-10-17 16:54:53 +0200 |
---|---|---|
committer | Volpeon <git@volpeon.ink> | 2021-10-17 16:54:53 +0200 |
commit | 30b0adcacef48ac53aea13cbdc3288db0bd8d103 (patch) | |
tree | 402f4cc31a2fa4f0e746c81755b537b5f5137e4c /src/api | |
download | feralbot-30b0adcacef48ac53aea13cbdc3288db0bd8d103.tar.gz feralbot-30b0adcacef48ac53aea13cbdc3288db0bd8d103.tar.bz2 feralbot-30b0adcacef48ac53aea13cbdc3288db0bd8d103.zip |
Init
Diffstat (limited to 'src/api')
-rw-r--r-- | src/api/e621/index.ts | 50 | ||||
-rw-r--r-- | src/api/mastodon/index.ts | 58 | ||||
-rw-r--r-- | src/api/misskey/index.ts | 26 |
3 files changed, 134 insertions, 0 deletions
diff --git a/src/api/e621/index.ts b/src/api/e621/index.ts new file mode 100644 index 0000000..6aa6a35 --- /dev/null +++ b/src/api/e621/index.ts | |||
@@ -0,0 +1,50 @@ | |||
1 | import got from "got"; | ||
2 | import config from "../../config"; | ||
3 | |||
4 | export interface Post { | ||
5 | id: number; | ||
6 | file: { | ||
7 | url: string; | ||
8 | }; | ||
9 | sources: readonly string[]; | ||
10 | |||
11 | tags: { | ||
12 | general: readonly string[]; | ||
13 | species: readonly string[]; | ||
14 | character: readonly string[]; | ||
15 | copyright: readonly string[]; | ||
16 | artist: readonly string[]; | ||
17 | invalid: readonly string[]; | ||
18 | lore: readonly string[]; | ||
19 | meta: readonly string[]; | ||
20 | }; | ||
21 | } | ||
22 | |||
23 | export const client = got.extend({ | ||
24 | headers: { | ||
25 | "User-Agent": config.e621.userAgent, | ||
26 | }, | ||
27 | }); | ||
28 | |||
29 | export async function randomPost() { | ||
30 | const page = Math.floor(Math.random() * (config.e621.maxPage - 1)) + 1; | ||
31 | |||
32 | const response = await client | ||
33 | .get("https://e926.net/posts.json", { | ||
34 | searchParams: { | ||
35 | limit: 75, | ||
36 | page, | ||
37 | tags: config.e621.tags.join(" "), | ||
38 | }, | ||
39 | }) | ||
40 | .json<{ posts: readonly Post[] }>(); | ||
41 | |||
42 | if (!response.posts.length) { | ||
43 | throw new Error("No posts received"); | ||
44 | } | ||
45 | |||
46 | const postIndex = Math.floor(Math.random() * response.posts.length); | ||
47 | const post = response.posts[postIndex]; | ||
48 | |||
49 | return post; | ||
50 | } | ||
diff --git a/src/api/mastodon/index.ts b/src/api/mastodon/index.ts new file mode 100644 index 0000000..2d8636e --- /dev/null +++ b/src/api/mastodon/index.ts | |||
@@ -0,0 +1,58 @@ | |||
1 | import got from "got"; | ||
2 | import FormData from "form-data"; | ||
3 | import fileType from "file-type"; | ||
4 | import { nanoid } from "nanoid"; | ||
5 | import config from "../../config"; | ||
6 | |||
7 | export interface Attachment { | ||
8 | id: string; | ||
9 | } | ||
10 | |||
11 | export interface Status { | ||
12 | url: string; | ||
13 | } | ||
14 | |||
15 | export const client = got.extend({ | ||
16 | prefixUrl: config.mastodon.instance, | ||
17 | headers: { | ||
18 | Authorization: `Bearer ${config.mastodon.token}`, | ||
19 | }, | ||
20 | }); | ||
21 | |||
22 | export async function upload(buf: Buffer, filename: string) { | ||
23 | const type = await fileType.fromBuffer(buf); | ||
24 | |||
25 | const body = new FormData(); | ||
26 | body.append("i", config.mastodon.token); | ||
27 | body.append("file", buf, { filename: `${filename}.${type.ext}`, contentType: type.mime }); | ||
28 | |||
29 | return client.post("api/v1/media", { body }).json<Attachment>(); | ||
30 | } | ||
31 | |||
32 | export async function createStatus( | ||
33 | postUrl: string, | ||
34 | sourceUrl: string | undefined, | ||
35 | spoiler: string[], | ||
36 | attachmentId: string | ||
37 | ) { | ||
38 | let lines = [postUrl]; | ||
39 | if (sourceUrl) { | ||
40 | lines.push(`Source: ${sourceUrl}`); | ||
41 | } | ||
42 | |||
43 | const spoilerText = spoiler.length ? `CW: ${spoiler.join(", ")}` : undefined; | ||
44 | |||
45 | return client | ||
46 | .post("api/v1/statuses", { | ||
47 | headers: { | ||
48 | "Idempotency-Key": nanoid(), | ||
49 | }, | ||
50 | json: { | ||
51 | status: lines.join("\n"), | ||
52 | media_ids: [attachmentId], | ||
53 | sensitive: true, | ||
54 | spoiler_text: spoilerText, | ||
55 | }, | ||
56 | }) | ||
57 | .json<Status>(); | ||
58 | } | ||
diff --git a/src/api/misskey/index.ts b/src/api/misskey/index.ts new file mode 100644 index 0000000..64bfe67 --- /dev/null +++ b/src/api/misskey/index.ts | |||
@@ -0,0 +1,26 @@ | |||
1 | /* | ||
2 | import got from "got"; | ||
3 | import FormData from "form-data"; | ||
4 | import stream from "stream"; | ||
5 | import config from "../../config"; | ||
6 | |||
7 | export interface DriveFile { | ||
8 | id: string; | ||
9 | } | ||
10 | |||
11 | export const client = got.extend({ | ||
12 | prefixUrl: config.misskey.instance, | ||
13 | }); | ||
14 | |||
15 | export function upload(buf: Buffer, filename: string) { | ||
16 | const body = new FormData(); | ||
17 | body.append("i", config.misskey.token); | ||
18 | body.append("file", buf, { filename }); | ||
19 | |||
20 | return client.post("drive/files/create", { body }).json<DriveFile>(); | ||
21 | }; | ||
22 | |||
23 | export function createNote() { | ||
24 | |||
25 | } | ||
26 | */ | ||