1import { APIError, ImageGenerationError } from '../exceptions.js';
2import { sleep } from './http.js';
3 
4export function running(retry: number = 0) {
5  return <TArgs extends unknown[], TResult>(
6    fn: (client: any, ...args: TArgs) => Promise<TResult>,
7  ): ((client: any, ...args: TArgs) => Promise<TResult>) => {
8    const wrap = async (client: any, ...args: TArgs): Promise<TResult> => {
9      try {
10        if (!client?._running) {
11          await client.init?.({
12            timeout: client.timeout,
13            auto_close: client.auto_close,
14            close_delay: client.close_delay,
15            auto_refresh: client.auto_refresh,
16            refresh_interval: client.refresh_interval,
17            verbose: false,
18          });
19        }
20        return await fn(client, ...args);
21      } catch (e) {
22        let r = retry;
23        if (e instanceof ImageGenerationError) r = Math.min(1, r);
24        if (e instanceof APIError && r > 0) {
25          await sleep(1000);
26          return await running(r - 1)(fn)(client, ...args);
27        }
28        throw e;
29      }
30    };
31    return wrap;
32  };
33}
34 
35