import { readFile } from 'node:fs/promises'; import { isAbsolute, basename } from 'node:path'; import WebSocket from 'ws'; import fetch, { Blob, File, RequestInit } from 'node-fetch'; import { DataSource } from 'typeorm'; import { JSDOM } from 'jsdom'; import { entities } from '../src/postgres.js'; import { loadConfig } from '../src/config.js'; import type * as misskey from 'misskey-js'; export { server as startServer } from '@/boot/common.js'; const config = loadConfig(); export const port = config.port; export const cookie = (me: any): string => { return `token=${me.token};`; }; export const api = async (endpoint: string, params: any, me?: any) => { const normalized = endpoint.replace(/^\//, ''); return await request(`api/${normalized}`, params, me); }; const request = async (path: string, params: any, me?: any): Promise<{ body: any, status: number }> => { const auth = me ? { i: me.token, } : {}; const res = await relativeFetch(path, { method: 'POST', headers: { 'Content-Type': 'application/json', }, body: JSON.stringify(Object.assign(auth, params)), redirect: 'manual', }); const status = res.status; const body = res.headers.get('content-type') === 'application/json; charset=utf-8' ? await res.json() : null; return { body, status, }; }; const relativeFetch = async (path: string, init?: RequestInit | undefined) => { return await fetch(new URL(path, `http://127.0.0.1:${port}/`).toString(), init); }; export const signup = async (params?: any): Promise => { const q = Object.assign({ username: 'test', password: 'test', }, params); const res = await api('signup', q); return res.body; }; export const post = async (user: any, params?: misskey.Endpoints['notes/create']['req']): Promise => { const q = params; const res = await api('notes/create', q, user); return res.body ? res.body.createdNote : null; }; export const react = async (user: any, note: any, reaction: string): Promise => { await api('notes/reactions/create', { noteId: note.id, reaction: reaction, }, user); }; export const page = async (user: any, page: any = {}): Promise => { const res = await api('pages/create', { alignCenter: false, content: [ { id: '2be9a64b-5ada-43a3-85f3-ec3429551ded', text: 'Hello World!', type: 'text', }, ], eyeCatchingImageId: null, font: 'sans-serif', hideTitleWhenPinned: false, name: '1678594845072', script: '', summary: null, title: '', variables: [], ...page, }, user); return res.body; }; export const play = async (user: any, play: any = {}): Promise => { const res = await api('flash/create', { permissions: [], script: 'test', summary: '', title: 'test', ...play, }, user); return res.body; }; export const clip = async (user: any, clip: any = {}): Promise => { const res = await api('clips/create', { description: null, isPublic: true, name: 'test', ...clip, }, user); return res.body; }; export const galleryPost = async (user: any, channel: any = {}): Promise => { const res = await api('gallery/posts/create', { description: null, fileIds: [], isSensitive: false, title: 'test', ...channel, }, user); return res.body; }; export const channel = async (user: any, channel: any = {}): Promise => { const res = await api('channels/create', { bannerId: null, description: null, name: 'test', ...channel, }, user); return res.body; }; interface UploadOptions { /** Optional, absolute path or relative from ./resources/ */ path?: string | URL; /** The name to be used for the file upload */ name?: string; /** A Blob can be provided instead of path */ blob?: Blob; } /** * Upload file * @param user User */ export const uploadFile = async (user: any, { path, name, blob }: UploadOptions = {}): Promise => { const absPath = path == null ? new URL('resources/Lenna.jpg', import.meta.url) : isAbsolute(path.toString()) ? new URL(path) : new URL(path, new URL('resources/', import.meta.url)); const formData = new FormData(); formData.append('i', user.token); formData.append('file', blob ?? new File([await readFile(absPath)], basename(absPath.toString()))); formData.append('force', 'true'); if (name) { formData.append('name', name); } const res = await relativeFetch('api/drive/files/create', { method: 'POST', body: formData, }); const body = res.status !== 204 ? await res.json() : null; return { status: res.status, body, }; }; export const uploadUrl = async (user: any, url: string) => { let file: any; const marker = Math.random().toString(); const ws = await connectStream(user, 'main', (msg) => { if (msg.type === 'urlUploadFinished' && msg.body.marker === marker) { file = msg.body.file; } }); await api('drive/files/upload-from-url', { url, marker, force: true, }, user); await sleep(7000); ws.close(); return file; }; export function connectStream(user: any, channel: string, listener: (message: Record) => any, params?: any): Promise { return new Promise((res, rej) => { const ws = new WebSocket(`ws://127.0.0.1:${port}/streaming?i=${user.token}`); ws.on('open', () => { ws.on('message', data => { const msg = JSON.parse(data.toString()); if (msg.type === 'channel' && msg.body.id === 'a') { listener(msg.body); } else if (msg.type === 'connected' && msg.body.id === 'a') { res(ws); } }); ws.send(JSON.stringify({ type: 'connect', body: { channel: channel, id: 'a', pong: true, params: params, }, })); }); }); } export const waitFire = async (user: any, channel: string, trgr: () => any, cond: (msg: Record) => boolean, params?: any) => { return new Promise(async (res, rej) => { let timer: NodeJS.Timeout | null = null; let ws: WebSocket; try { ws = await connectStream(user, channel, msg => { if (cond(msg)) { ws.close(); if (timer) clearTimeout(timer); res(true); } }, params); } catch (e) { rej(e); } if (!ws!) return; timer = setTimeout(() => { ws.close(); res(false); }, 3000); try { await trgr(); } catch (e) { ws.close(); if (timer) clearTimeout(timer); rej(e); } }); }; export type SimpleGetResponse = { status: number, body: any | JSDOM | null, type: string | null, location: string | null }; export const simpleGet = async (path: string, accept = '*/*', cookie: any = undefined): Promise => { const res = await relativeFetch(path, { headers: { Accept: accept, Cookie: cookie, }, redirect: 'manual', }); const jsonTypes = [ 'application/json; charset=utf-8', 'application/activity+json; charset=utf-8', ]; const htmlTypes = [ 'text/html; charset=utf-8', ]; const body = jsonTypes.includes(res.headers.get('content-type') ?? '') ? await res.json() : htmlTypes.includes(res.headers.get('content-type') ?? '') ? new JSDOM(await res.text()) : null; return { status: res.status, body, type: res.headers.get('content-type'), location: res.headers.get('location'), }; }; export async function initTestDb(justBorrow = false, initEntities?: any[]) { if (process.env.NODE_ENV !== 'test') throw 'NODE_ENV is not a test'; const db = new DataSource({ type: 'postgres', host: config.db.host, port: config.db.port, username: config.db.user, password: config.db.pass, database: config.db.db, synchronize: true && !justBorrow, dropSchema: true && !justBorrow, entities: initEntities ?? entities, }); await db.initialize(); return db; } export function sleep(msec: number) { return new Promise(res => { setTimeout(() => { res(); }, msec); }); }