1  /**
 2   * Yollomi background removal — POST /api/ai/remove-bg (free, 0 credits)
 3   */
 4  import * as path from 'node:path';
 5  import { cli, Strategy } from '@jackwener/opencli/registry';
 6  import { CliError } from '@jackwener/opencli/errors';
 7  import { log } from '@jackwener/opencli/logger';
 8  import { YOLLOMI_DOMAIN, yollomiPost, downloadOutput, fmtBytes } from './utils.js';
 9  cli({
10      site: 'yollomi',
11      name: 'remove-bg',
12      description: 'Remove image background with AI (free)',
13      domain: YOLLOMI_DOMAIN,
14      strategy: Strategy.COOKIE,
15      args: [
16          { name: 'image', positional: true, required: true, help: 'Image URL to remove background from' },
17          { name: 'output', default: './yollomi-output', help: 'Output directory' },
18          { name: 'no-download', type: 'boolean', default: false, help: 'Only show URL' },
19      ],
20      columns: ['status', 'file', 'size', 'url'],
21      func: async (page, kwargs) => {
22          log.status('Removing background...');
23          const data = await yollomiPost(page, '/api/ai/remove-bg', { imageUrl: kwargs.image });
24          const url = data.image || (data.images?.[0]);
25          if (!url)
26              throw new CliError('EMPTY_RESPONSE', 'No result', 'Check the input image URL');
27          if (kwargs['no-download'])
28              return [{ status: 'processed', file: '-', size: '-', url }];
29          try {
30              const filename = `yollomi_nobg_${Date.now()}.png`;
31              const { path: fp, size } = await downloadOutput(url, kwargs.output, filename);
32              return [{ status: 'saved', file: path.relative('.', fp), size: fmtBytes(size), url }];
33          }
34          catch {
35              return [{ status: 'download-failed', file: '-', size: '-', url }];
36          }
37      },
38  });