import { app } from "../../scripts/app.js" ;
import { api } from "../../scripts/api.js"
export async function sleep ( ms ) {
return new Promise ( resolve => setTimeout ( resolve , ms ) ) ;
}
export function rebootAPI ( ) {
if ( confirm ( "Are you sure you'd like to reboot the server?" ) ) {
try {
api . fetchApi ( "/manager/reboot" ) ;
}
catch ( exception ) {
}
return true ;
}
return false ;
}
export async function install _checked _custom _node ( grid _rows , target _i , caller , mode ) {
if ( caller ) {
let failed = '' ;
caller . disableButtons ( ) ;
for ( let i in grid _rows ) {
if ( ! grid _rows [ i ] . checkbox . checked && i != target _i )
continue ;
var target ;
if ( grid _rows [ i ] . data . custom _node ) {
target = grid _rows [ i ] . data . custom _node ;
}
else {
target = grid _rows [ i ] . data ;
}
caller . startInstall ( target ) ;
try {
const response = await api . fetchApi ( ` /customnode/ ${ mode } ` , {
method : 'POST' ,
headers : { 'Content-Type' : 'application/json' } ,
body : JSON . stringify ( target )
} ) ;
if ( response . status == 400 ) {
app . ui . dialog . show ( ` ${ mode } failed: ${ target . title } ` ) ;
app . ui . dialog . element . style . zIndex = 10010 ;
continue ;
}
const status = await response . json ( ) ;
app . ui . dialog . close ( ) ;
target . installed = 'True' ;
continue ;
}
catch ( exception ) {
failed += ` <BR> ${ target . title } ` ;
}
}
if ( failed != '' ) {
app . ui . dialog . show ( ` ${ mode } failed: ${ failed } ` ) ;
app . ui . dialog . element . style . zIndex = 10010 ;
}
await caller . invalidateControl ( ) ;
caller . updateMessage ( "<BR>To apply the installed/updated/disabled/enabled custom node, please <button id='cm-reboot-button' class='cm-small-button'>RESTART</button> ComfyUI. And refresh browser." , 'cm-reboot-button' ) ;
}
} ;
export var manager _instance = null ;
export function setManagerInstance ( obj ) {
manager _instance = obj ;
}
function isValidURL ( url ) {
if ( url . includes ( '&' ) )
return false ;
const pattern = /^(https?|ftp):\/\/[^\s/$.?#].[^\s]*$/ ;
return pattern . test ( url ) ;
}
export async function install _pip ( packages ) {
if ( packages . includes ( '&' ) )
app . ui . dialog . show ( ` Invalid PIP package enumeration: ' ${ packages } ' ` ) ;
const res = await api . fetchApi ( ` /customnode/install/pip?packages= ${ packages } ` ) ;
if ( res . status == 200 ) {
app . ui . dialog . show ( ` PIP package installation is processed.<br>To apply the pip packages, please click the <button id='cm-reboot-button'><font size='3px'>RESTART</font></button> button in ComfyUI. ` ) ;
const rebootButton = document . getElementById ( 'cm-reboot-button' ) ;
const self = this ;
rebootButton . addEventListener ( "click" , rebootAPI ) ;
app . ui . dialog . element . style . zIndex = 10010 ;
}
else {
app . ui . dialog . show ( ` Failed to install ' ${ packages } '<BR>See terminal log. ` ) ;
app . ui . dialog . element . style . zIndex = 10010 ;
}
}
export async function install _via _git _url ( url , manager _dialog ) {
if ( ! url ) {
return ;
}
if ( ! isValidURL ( url ) ) {
app . ui . dialog . show ( ` Invalid Git url ' ${ url } ' ` ) ;
app . ui . dialog . element . style . zIndex = 10010 ;
return ;
}
app . ui . dialog . show ( ` Wait...<BR><BR>Installing ' ${ url } ' ` ) ;
app . ui . dialog . element . style . zIndex = 10010 ;
const res = await api . fetchApi ( ` /customnode/install/git_url?url= ${ url } ` ) ;
if ( res . status == 200 ) {
app . ui . dialog . show ( ` ' ${ url } ' is installed<BR>To apply the installed custom node, please <button id='cm-reboot-button'><font size='3px'>RESTART</font></button> ComfyUI. ` ) ;
const rebootButton = document . getElementById ( 'cm-reboot-button' ) ;
const self = this ;
rebootButton . addEventListener ( "click" ,
function ( ) {
if ( rebootAPI ( ) ) {
manager _dialog . close ( ) ;
}
} ) ;
app . ui . dialog . element . style . zIndex = 10010 ;
}
else {
app . ui . dialog . show ( ` Failed to install ' ${ url } '<BR>See terminal log. ` ) ;
app . ui . dialog . element . style . zIndex = 10010 ;
}
}