Interrupt AUTO1111 gens on cancel

7e4abaabff3e130ab1e50c8b16842a6b271c5a5f

Cohee <18619528+Cohee1207@users.noreply.github.com>

src/endpoints/stable-diffusion.js+13 -0 Viewed

         const url = new URL(request.body.url);
         url.pathname = '/sdapi/v1/txt2img';
 +        const controller = new AbortController();
 +        request.socket.removeAllListeners('close');
 +        request.socket.on('close', function () {
 +            if (!response.writableEnded) {
 +                const url = new URL(request.body.url);
 +                url.pathname = '/sdapi/v1/interrupt';
 +                fetch(url, { method: 'POST', headers: { 'Authorization': getBasicAuthHeader(request.body.auth) } });
 +            }
 +            controller.abort();
 +        });
++
         const result = await fetch(url, {
             method: 'POST',
             body: JSON.stringify(request.body),

                 'Authorization': getBasicAuthHeader(request.body.auth),
             },
             timeout: 0,
 +            // @ts-ignore
 +            signal: controller.signal,
         });
         if (!result.ok) {