132 lines
No EOL
4.3 KiB
TypeScript
132 lines
No EOL
4.3 KiB
TypeScript
import type { MegalodonInterface } from "megalodon";
|
|
import { Converter } from "megalodon";
|
|
import type { FastifyRequest } from 'fastify';
|
|
import { convertTimelinesArgsId, limitToInt } from "./timeline.js";
|
|
import { convertAccount, convertStatus } from '../converters.js';
|
|
|
|
async function getHighlight(
|
|
BASE_URL: string,
|
|
domain: string,
|
|
accessTokens: string | undefined,
|
|
) {
|
|
const accessTokenArr = accessTokens?.split(" ") ?? [null];
|
|
const accessToken = accessTokenArr[accessTokenArr.length - 1];
|
|
try {
|
|
|
|
const apicall = await fetch(`${BASE_URL}/api/notes/featured`,
|
|
{
|
|
method: 'POST',
|
|
headers: {
|
|
'Accept': 'application/json, text/plain, */*',
|
|
'Content-Type': 'application/json'
|
|
},
|
|
body: JSON.stringify({i: accessToken})
|
|
});
|
|
const api = await apicall.json();
|
|
const data: MisskeyEntity.Note[] = api;
|
|
return data.map((note) => Converter.note(note, domain));
|
|
} catch (e: any) {
|
|
console.log(e);
|
|
console.log(e.response.data);
|
|
return [];
|
|
}
|
|
}
|
|
|
|
async function getFeaturedUser( BASE_URL: string, host: string, accessTokens: string | undefined, limit: number ) {
|
|
const accessTokenArr = accessTokens?.split(" ") ?? [null];
|
|
const accessToken = accessTokenArr[accessTokenArr.length - 1];
|
|
try {
|
|
const apicall = await fetch(`${BASE_URL}/api/users`,
|
|
{
|
|
method: 'POST',
|
|
headers: {
|
|
'Accept': 'application/json, text/plain, */*',
|
|
'Content-Type': 'application/json'
|
|
},
|
|
body: JSON.stringify({i: accessToken, limit, origin: "local", sort: "+follower", state: "alive"})
|
|
});
|
|
const api = await apicall.json();
|
|
const data: MisskeyEntity.UserDetail[] = api;
|
|
return data.map((u) => {
|
|
return {
|
|
source: "past_interactions",
|
|
account: Converter.userDetail(u, host),
|
|
};
|
|
});
|
|
} catch (e: any) {
|
|
console.log(e);
|
|
console.log(e.response.data);
|
|
return [];
|
|
}
|
|
}
|
|
export class apiSearchMastodon {
|
|
private request: FastifyRequest;
|
|
private client: MegalodonInterface;
|
|
private BASE_URL: string;
|
|
|
|
constructor(request: FastifyRequest, client: MegalodonInterface, BASE_URL: string) {
|
|
this.request = request;
|
|
this.client = client;
|
|
this.BASE_URL = BASE_URL;
|
|
}
|
|
|
|
public async SearchV1() {
|
|
try {
|
|
const query: any = convertTimelinesArgsId(limitToInt(this.request.query as any));
|
|
const type = query.type || "";
|
|
const data = await this.client.search(query.q, { type: type, ...query });
|
|
return data.data;
|
|
} catch (e: any) {
|
|
console.error(e);
|
|
return e.response.data;
|
|
}
|
|
}
|
|
|
|
public async SearchV2() {
|
|
try {
|
|
const query: any = convertTimelinesArgsId(limitToInt(this.request.query as any));
|
|
const type = query.type;
|
|
const acct = !type || type === "accounts" ? await this.client.search(query.q, { type: "accounts", ...query }) : null;
|
|
const stat = !type || type === "statuses" ? await this.client.search(query.q, { type: "statuses", ...query }) : null;
|
|
const tags = !type || type === "hashtags" ? await this.client.search(query.q, { type: "hashtags", ...query }) : null;
|
|
const data = {
|
|
accounts: acct?.data.accounts.map((account) => convertAccount(account)) ?? [],
|
|
statuses: stat?.data.statuses.map((status) => convertStatus(status)) ?? [],
|
|
hashtags: tags?.data.hashtags ?? []
|
|
};
|
|
return data;
|
|
} catch (e: any) {
|
|
console.error(e);
|
|
return e.response.data;
|
|
}
|
|
}
|
|
|
|
public async getStatusTrends() {
|
|
try {
|
|
const data = await getHighlight(
|
|
this.BASE_URL,
|
|
this.request.hostname,
|
|
this.request.headers.authorization,
|
|
);
|
|
return data.map((status) => convertStatus(status));
|
|
} catch (e: any) {
|
|
console.error(e);
|
|
return e.response.data;
|
|
}
|
|
}
|
|
|
|
public async getSuggestions() {
|
|
try {
|
|
const data = await getFeaturedUser(
|
|
this.BASE_URL,
|
|
this.request.hostname,
|
|
this.request.headers.authorization,
|
|
(this.request.query as any).limit || 20,
|
|
);
|
|
return data.map((suggestion) => { suggestion.account = convertAccount(suggestion.account); return suggestion; });
|
|
} catch (e: any) {
|
|
console.error(e);
|
|
return e.response.data;
|
|
}
|
|
}
|
|
} |