450 lines
16 KiB
TypeScript
450 lines
16 KiB
TypeScript
import { spawn } from 'node:child_process';
|
|
import { existsSync } from 'node:fs';
|
|
import path from 'node:path';
|
|
import { beforeAll, describe, expect, it } from 'vitest';
|
|
import { resolveCredentials } from '../../src/lib/cookies.js';
|
|
import { TwitterClient } from '../../src/lib/twitter-client.js';
|
|
|
|
type RunResult = { exitCode: number; stdout: string; stderr: string; signal: NodeJS.Signals | null };
|
|
|
|
const LIVE = process.env.BIRD_LIVE === '1';
|
|
|
|
const relayBaseUrl = (process.env.TWITTER_RELAY_BASE_URL ?? '').trim();
|
|
|
|
const CLI_PATH = path.resolve(process.cwd(), 'dist', 'cli.js');
|
|
|
|
const WHOAMI_HANDLE_REGEX = /^user:\s*@([A-Za-z0-9_]+)/m;
|
|
const WHOAMI_USER_ID_REGEX = /^user_id:\s*([0-9]+)/m;
|
|
const TWEET_ID_REGEX = /^\d+$/;
|
|
const LIVE_NODE_ENV = (process.env.BIRD_LIVE_NODE_ENV ?? 'production').trim() || 'production';
|
|
|
|
function runBird(args: string[], options: { timeoutMs?: number } = {}): Promise<RunResult> {
|
|
if (!LIVE) {
|
|
throw new Error('runBird() called without BIRD_LIVE=1');
|
|
}
|
|
|
|
return new Promise((resolve) => {
|
|
const child = spawn(process.execPath, [CLI_PATH, ...args], {
|
|
stdio: ['ignore', 'pipe', 'pipe'],
|
|
env: {
|
|
...process.env,
|
|
NODE_ENV: LIVE_NODE_ENV,
|
|
},
|
|
});
|
|
|
|
const stdoutChunks: Buffer[] = [];
|
|
const stderrChunks: Buffer[] = [];
|
|
|
|
child.stdout.on('data', (chunk: Buffer) => stdoutChunks.push(chunk));
|
|
child.stderr.on('data', (chunk: Buffer) => stderrChunks.push(chunk));
|
|
|
|
const killTimer =
|
|
typeof options.timeoutMs === 'number' && options.timeoutMs > 0
|
|
? setTimeout(() => {
|
|
child.kill('SIGKILL');
|
|
}, options.timeoutMs)
|
|
: null;
|
|
|
|
child.on('close', (code, signal) => {
|
|
if (killTimer) {
|
|
clearTimeout(killTimer);
|
|
}
|
|
resolve({
|
|
exitCode: code ?? (signal ? 1 : 0),
|
|
stdout: Buffer.concat(stdoutChunks).toString('utf8'),
|
|
stderr: Buffer.concat(stderrChunks).toString('utf8'),
|
|
signal: signal ?? null,
|
|
});
|
|
});
|
|
});
|
|
}
|
|
|
|
function parseJson<T>(stdout: string): T {
|
|
try {
|
|
return JSON.parse(stdout) as T;
|
|
} catch (error) {
|
|
throw new Error(`Invalid JSON output: ${error instanceof Error ? error.message : String(error)}\n${stdout}`);
|
|
}
|
|
}
|
|
|
|
async function sleep(ms: number): Promise<void> {
|
|
await new Promise((resolve) => setTimeout(resolve, ms));
|
|
}
|
|
|
|
const d = LIVE ? describe : describe.skip;
|
|
|
|
d('live CLI (Twitter/X) all commands', () => {
|
|
const timeoutArg = (process.env.BIRD_LIVE_TIMEOUT_MS ?? '20000').trim();
|
|
const baseArgs = ['--plain', '--timeout', timeoutArg, '--quote-depth', '0'];
|
|
|
|
let whoamiStdout = '';
|
|
let handle = '';
|
|
let userId = '';
|
|
let tweetId = '';
|
|
let searchStdout = '';
|
|
let listId = '';
|
|
let listTimelineAvailable = false;
|
|
|
|
beforeAll(async () => {
|
|
if (!existsSync(CLI_PATH)) {
|
|
throw new Error(`Missing ${CLI_PATH}. Run: pnpm run build:dist`);
|
|
}
|
|
|
|
if (!relayBaseUrl) {
|
|
throw new Error('Missing TWITTER_RELAY_BASE_URL for live tests.');
|
|
}
|
|
|
|
const check = await runBird([...baseArgs, 'check'], { timeoutMs: 45_000 });
|
|
if (check.exitCode !== 0) {
|
|
throw new Error(`bird check failed:\n${check.stdout}\n${check.stderr}`);
|
|
}
|
|
|
|
const who = await runBird([...baseArgs, 'whoami'], { timeoutMs: 45_000 });
|
|
if (who.exitCode !== 0) {
|
|
throw new Error(`whoami failed (exit ${who.exitCode}, signal ${who.signal ?? 'none'}):\n${who.stderr}`);
|
|
}
|
|
whoamiStdout = who.stdout;
|
|
|
|
const handleMatch = WHOAMI_HANDLE_REGEX.exec(who.stdout);
|
|
if (!handleMatch?.[1]) {
|
|
throw new Error(`Could not parse whoami handle:\n${who.stdout}`);
|
|
}
|
|
handle = handleMatch[1];
|
|
|
|
const userIdMatch = WHOAMI_USER_ID_REGEX.exec(who.stdout);
|
|
if (!userIdMatch?.[1]) {
|
|
throw new Error(`Could not parse whoami user_id:\n${who.stdout}`);
|
|
}
|
|
userId = userIdMatch[1];
|
|
|
|
const forcedTweetId = (process.env.BIRD_LIVE_TWEET_ID ?? '').trim();
|
|
if (forcedTweetId) {
|
|
if (!TWEET_ID_REGEX.test(forcedTweetId)) {
|
|
throw new Error(`Invalid BIRD_LIVE_TWEET_ID (expected digits): "${forcedTweetId}"`);
|
|
}
|
|
tweetId = forcedTweetId;
|
|
} else {
|
|
const searchQuery = (
|
|
process.env.BIRD_LIVE_SEARCH_QUERY ?? `from:${handle} -filter:replies -filter:retweets`
|
|
).trim();
|
|
const search = await runBird([...baseArgs, 'search', searchQuery, '-n', '25', '--json'], {
|
|
timeoutMs: 45_000,
|
|
});
|
|
if (search.exitCode !== 0) {
|
|
throw new Error(
|
|
`search failed (exit ${search.exitCode}, signal ${search.signal ?? 'none'}):\n${search.stderr}`,
|
|
);
|
|
}
|
|
searchStdout = search.stdout;
|
|
const tweets = parseJson<Array<{ id?: string }>>(search.stdout);
|
|
const first = String(tweets[0]?.id ?? '');
|
|
if (!TWEET_ID_REGEX.test(first)) {
|
|
throw new Error(
|
|
`Search returned no usable tweets. Query: "${searchQuery}". ` +
|
|
`Override with BIRD_LIVE_SEARCH_QUERY.\n${search.stdout}`,
|
|
);
|
|
}
|
|
tweetId = first;
|
|
}
|
|
|
|
listId = (process.env.BIRD_LIVE_LIST_ID ?? '').trim();
|
|
if (!listId) {
|
|
const lists = await runBird([...baseArgs, 'lists', '-n', '1', '--json'], {
|
|
timeoutMs: 45_000,
|
|
});
|
|
if (lists.exitCode === 0) {
|
|
const parsed = parseJson<Array<{ id?: string }>>(lists.stdout);
|
|
listId = String(parsed[0]?.id ?? '').trim();
|
|
}
|
|
}
|
|
if (!listId) {
|
|
const lists = await runBird([...baseArgs, 'lists', '--member-of', '-n', '1', '--json'], {
|
|
timeoutMs: 45_000,
|
|
});
|
|
if (lists.exitCode === 0) {
|
|
const parsed = parseJson<Array<{ id?: string }>>(lists.stdout);
|
|
listId = String(parsed[0]?.id ?? '').trim();
|
|
}
|
|
}
|
|
listTimelineAvailable = Boolean(listId);
|
|
});
|
|
|
|
it('requires env credentials + built dist CLI', async () => {
|
|
expect(existsSync(CLI_PATH)).toBe(true);
|
|
expect(whoamiStdout.length).toBeGreaterThan(0);
|
|
});
|
|
|
|
it('check works', async () => {
|
|
const check = await runBird([...baseArgs, 'check'], { timeoutMs: 45_000 });
|
|
expect(check.exitCode).toBe(0);
|
|
expect(check.stdout.length).toBeGreaterThan(0);
|
|
});
|
|
|
|
it('whoami works', async () => {
|
|
expect(whoamiStdout).toContain('user: @');
|
|
expect(whoamiStdout).toContain('user_id:');
|
|
});
|
|
|
|
it('search returns JSON array', async () => {
|
|
const search = searchStdout
|
|
? { exitCode: 0, stdout: searchStdout }
|
|
: await runBird([...baseArgs, 'search', `from:${handle}`, '-n', '10', '--json'], { timeoutMs: 45_000 });
|
|
expect(search.exitCode).toBe(0);
|
|
const tweets = parseJson<Array<{ id?: string }>>(search.stdout);
|
|
expect(Array.isArray(tweets)).toBe(true);
|
|
});
|
|
|
|
it('read returns tweet JSON', async () => {
|
|
const read = await runBird([...baseArgs, 'read', tweetId, '--json'], {
|
|
timeoutMs: 45_000,
|
|
});
|
|
expect(read.exitCode).toBe(0);
|
|
const readTweet = parseJson<{ id?: string; author?: { username?: string } }>(read.stdout);
|
|
expect(readTweet.id).toBe(tweetId);
|
|
expect(readTweet.author?.username?.length).toBeGreaterThan(0);
|
|
});
|
|
|
|
it('tweet-id shorthand returns tweet JSON', async () => {
|
|
const shorthand = await runBird([...baseArgs, tweetId, '--json'], {
|
|
timeoutMs: 45_000,
|
|
});
|
|
expect(shorthand.exitCode).toBe(0);
|
|
const shorthandTweet = parseJson<{ id?: string }>(shorthand.stdout);
|
|
expect(shorthandTweet.id).toBe(tweetId);
|
|
});
|
|
|
|
it('replies returns JSON array', async () => {
|
|
const replies = await runBird([...baseArgs, 'replies', tweetId, '--json'], {
|
|
timeoutMs: 45_000,
|
|
});
|
|
expect(replies.exitCode).toBe(0);
|
|
const replyTweets = parseJson<Array<{ id?: string }>>(replies.stdout);
|
|
expect(Array.isArray(replyTweets)).toBe(true);
|
|
});
|
|
|
|
it('thread returns JSON array', async () => {
|
|
const thread = await runBird([...baseArgs, 'thread', tweetId, '--json'], {
|
|
timeoutMs: 45_000,
|
|
});
|
|
expect(thread.exitCode).toBe(0);
|
|
const threadTweets = parseJson<Array<{ id?: string }>>(thread.stdout);
|
|
expect(Array.isArray(threadTweets)).toBe(true);
|
|
});
|
|
|
|
it('mentions returns JSON array', async () => {
|
|
const mentions = await runBird([...baseArgs, 'mentions', '-n', '10', '--json'], {
|
|
timeoutMs: 45_000,
|
|
});
|
|
expect(mentions.exitCode).toBe(0);
|
|
const mentionTweets = parseJson<Array<{ id?: string }>>(mentions.stdout);
|
|
expect(Array.isArray(mentionTweets)).toBe(true);
|
|
});
|
|
|
|
it('bookmarks returns JSON array', async () => {
|
|
const bookmarks = await runBird([...baseArgs, 'bookmarks', '-n', '10', '--json'], {
|
|
timeoutMs: 45_000,
|
|
});
|
|
expect(bookmarks.exitCode).toBe(0);
|
|
const bookmarkTweets = parseJson<Array<{ id?: string }>>(bookmarks.stdout);
|
|
expect(Array.isArray(bookmarkTweets)).toBe(true);
|
|
});
|
|
|
|
it('bookmarks full-chain-only with ancestor branches returns JSON array', async () => {
|
|
const bookmarks = await runBird(
|
|
[
|
|
...baseArgs,
|
|
'--cookie-timeout',
|
|
cookieTimeoutArg,
|
|
'bookmarks',
|
|
'-n',
|
|
'1',
|
|
'--full-chain-only',
|
|
'--include-ancestor-branches',
|
|
'--json',
|
|
],
|
|
{
|
|
timeoutMs: 45_000,
|
|
},
|
|
);
|
|
expect(bookmarks.exitCode).toBe(0);
|
|
const bookmarkTweets = parseJson<Array<{ id?: string }>>(bookmarks.stdout);
|
|
expect(Array.isArray(bookmarkTweets)).toBe(true);
|
|
});
|
|
|
|
it('likes returns JSON array', async () => {
|
|
const likes = await runBird([...baseArgs, 'likes', '-n', '10', '--json'], {
|
|
timeoutMs: 45_000,
|
|
});
|
|
expect(likes.exitCode).toBe(0);
|
|
const likeTweets = parseJson<Array<{ id?: string }>>(likes.stdout);
|
|
expect(Array.isArray(likeTweets)).toBe(true);
|
|
});
|
|
|
|
it('following returns JSON array', async () => {
|
|
const following = await runBird([...baseArgs, 'following', '--user', userId, '-n', '10', '--json'], {
|
|
timeoutMs: 45_000,
|
|
});
|
|
expect(following.exitCode).toBe(0);
|
|
const followingUsers = parseJson<Array<{ id?: string; username?: string }>>(following.stdout);
|
|
expect(Array.isArray(followingUsers)).toBe(true);
|
|
});
|
|
|
|
it('followers returns JSON array', async () => {
|
|
const followers = await runBird([...baseArgs, 'followers', '--user', userId, '-n', '10', '--json'], {
|
|
timeoutMs: 45_000,
|
|
});
|
|
expect(followers.exitCode).toBe(0);
|
|
const followerUsers = parseJson<Array<{ id?: string; username?: string }>>(followers.stdout);
|
|
expect(Array.isArray(followerUsers)).toBe(true);
|
|
});
|
|
|
|
it('user-tweets returns JSON array', async () => {
|
|
const testHandle = process.env.BIRD_LIVE_USER_TWEETS_HANDLE || 'X';
|
|
const userTweets = await runBird([...baseArgs, 'user-tweets', testHandle, '-n', '5', '--json'], {
|
|
timeoutMs: 45_000,
|
|
});
|
|
expect(userTweets.exitCode).toBe(0);
|
|
const tweets = parseJson<Array<{ id?: string; author?: { username?: string } }>>(userTweets.stdout);
|
|
expect(Array.isArray(tweets)).toBe(true);
|
|
expect(tweets.length).toBeGreaterThan(0);
|
|
});
|
|
|
|
it('user-tweets paged JSON returns { tweets, nextCursor }', async () => {
|
|
const testHandle =
|
|
process.env.BIRD_LIVE_USER_TWEETS_PAGED_HANDLE ?? process.env.BIRD_LIVE_USER_TWEETS_HANDLE ?? 'X';
|
|
const userTweets = await runBird(
|
|
[...baseArgs, 'user-tweets', testHandle, '-n', '50', '--max-pages', '3', '--delay', '0', '--json'],
|
|
{
|
|
timeoutMs: 45_000,
|
|
},
|
|
);
|
|
expect(userTweets.exitCode).toBe(0);
|
|
const payload = parseJson<{ tweets?: Array<{ id?: string }>; nextCursor?: string | null }>(userTweets.stdout);
|
|
expect(Array.isArray(payload.tweets)).toBe(true);
|
|
});
|
|
|
|
it('lists returns JSON array', async () => {
|
|
const lists = await runBird([...baseArgs, 'lists', '-n', '10', '--json'], {
|
|
timeoutMs: 45_000,
|
|
});
|
|
expect(lists.exitCode).toBe(0);
|
|
const payload = parseJson<Array<{ id?: string }>>(lists.stdout);
|
|
expect(Array.isArray(payload)).toBe(true);
|
|
});
|
|
|
|
it('list-timeline returns JSON array', async () => {
|
|
if (!listTimelineAvailable) {
|
|
return;
|
|
}
|
|
const timeline = await runBird([...baseArgs, 'list-timeline', listId, '-n', '5', '--json'], { timeoutMs: 45_000 });
|
|
expect(timeline.exitCode).toBe(0);
|
|
const payload = parseJson<Array<{ id?: string }>>(timeline.stdout);
|
|
expect(Array.isArray(payload)).toBe(true);
|
|
});
|
|
|
|
it('home for-you returns JSON array', async () => {
|
|
const home = await runBird([...baseArgs, 'home', '-n', '5', '--json'], {
|
|
timeoutMs: 45_000,
|
|
});
|
|
expect(home.exitCode).toBe(0);
|
|
const payload = parseJson<Array<{ id?: string }>>(home.stdout);
|
|
expect(Array.isArray(payload)).toBe(true);
|
|
}, 60_000);
|
|
|
|
it('home following returns JSON array', async () => {
|
|
let home = await runBird([...baseArgs, 'home', '--following', '-n', '5', '--json'], {
|
|
timeoutMs: 45_000,
|
|
});
|
|
if (home.exitCode !== 0 && home.stderr.includes('OverCapacity')) {
|
|
await sleep(1500);
|
|
home = await runBird([...baseArgs, 'home', '--following', '-n', '5', '--json'], {
|
|
timeoutMs: 45_000,
|
|
});
|
|
}
|
|
expect(home.exitCode).toBe(0);
|
|
const payload = parseJson<Array<{ id?: string }>>(home.stdout);
|
|
expect(Array.isArray(payload)).toBe(true);
|
|
}, 60_000);
|
|
|
|
it('news returns JSON array', async () => {
|
|
const news = await runBird([...baseArgs, 'news', '-n', '5', '--ai-only', '--json'], { timeoutMs: 45_000 });
|
|
expect(news.exitCode).toBe(0);
|
|
const payload = parseJson<Array<{ id?: string }>>(news.stdout);
|
|
expect(Array.isArray(payload)).toBe(true);
|
|
});
|
|
|
|
it('trending returns JSON array', async () => {
|
|
const trending = await runBird([...baseArgs, 'trending', '-n', '5', '--json'], { timeoutMs: 45_000 });
|
|
expect(trending.exitCode).toBe(0);
|
|
const payload = parseJson<Array<{ id?: string }>>(trending.stdout);
|
|
expect(Array.isArray(payload)).toBe(true);
|
|
});
|
|
|
|
it('query-ids returns JSON', async () => {
|
|
const queryIds = await runBird([...baseArgs, 'query-ids', '--json'], {
|
|
timeoutMs: 60_000,
|
|
});
|
|
expect(queryIds.exitCode).toBe(0);
|
|
const snapshot = parseJson<{ cached?: boolean; ids?: Record<string, string> }>(queryIds.stdout);
|
|
expect(typeof snapshot.cached).toBe('boolean');
|
|
});
|
|
|
|
it('engagement mutations work (opt-in)', async () => {
|
|
const engagementTweetId = (process.env.BIRD_LIVE_ENGAGEMENT_TWEET_ID ?? '').trim();
|
|
if (!engagementTweetId) {
|
|
return;
|
|
}
|
|
if (!TWEET_ID_REGEX.test(engagementTweetId)) {
|
|
throw new Error(`Invalid BIRD_LIVE_ENGAGEMENT_TWEET_ID (expected digits): "${engagementTweetId}"`);
|
|
}
|
|
|
|
const cookieTimeoutMs = Number.parseInt(cookieTimeoutArg, 10);
|
|
const { cookies, warnings } = await resolveCredentials({
|
|
authToken,
|
|
ct0,
|
|
cookieTimeoutMs: Number.isFinite(cookieTimeoutMs) ? cookieTimeoutMs : undefined,
|
|
});
|
|
for (const warning of warnings) {
|
|
console.error(`${warning}`);
|
|
}
|
|
if (!cookies.authToken || !cookies.ct0) {
|
|
throw new Error('Missing live credentials for engagement test.');
|
|
}
|
|
|
|
const originalNodeEnv = process.env.NODE_ENV;
|
|
process.env.NODE_ENV = LIVE_NODE_ENV;
|
|
try {
|
|
const timeoutMs = Number.parseInt(timeoutArg, 10);
|
|
const client = new TwitterClient({
|
|
cookies,
|
|
timeoutMs: Number.isFinite(timeoutMs) ? timeoutMs : undefined,
|
|
});
|
|
|
|
// Cleanup (best-effort) to make the mutation sequence deterministic.
|
|
await client.unbookmark(engagementTweetId);
|
|
await client.unlike(engagementTweetId);
|
|
await client.unretweet(engagementTweetId);
|
|
|
|
const like = await client.like(engagementTweetId);
|
|
expect(like.success).toBe(true);
|
|
const retweet = await client.retweet(engagementTweetId);
|
|
expect(retweet.success).toBe(true);
|
|
const bookmark = await client.bookmark(engagementTweetId);
|
|
expect(bookmark.success).toBe(true);
|
|
|
|
const unbookmark = await client.unbookmark(engagementTweetId);
|
|
expect(unbookmark.success).toBe(true);
|
|
const unlike = await client.unlike(engagementTweetId);
|
|
expect(unlike.success).toBe(true);
|
|
const unretweet = await client.unretweet(engagementTweetId);
|
|
expect(unretweet.success).toBe(true);
|
|
} finally {
|
|
if (originalNodeEnv === undefined) {
|
|
delete process.env.NODE_ENV;
|
|
} else {
|
|
process.env.NODE_ENV = originalNodeEnv;
|
|
}
|
|
}
|
|
});
|
|
});
|