Fit Studio 2 inputs and report live generation progress

tests/studio2-progress.test.mjs

@@ -0,0 +1,19 @@
+import test from 'node:test'
+import assert from 'node:assert/strict'
+import fs from 'node:fs'
+import ts from 'typescript'
+test('progress accepts this client, rejects other prompts and closes on completion',()=>{
+ const source=fs.readFileSync(new URL('../server/utils/studio2/progress.ts',import.meta.url),'utf8')
+ const tree=ts.createSourceFile('progress.ts',source,ts.ScriptTarget.Latest,true)
+ const fn=tree.statements.find(n=>ts.isFunctionDeclaration(n)&&n.name.text==='watchProgress')
+ const code=ts.transpileModule(fn.getText(tree).replace(/^export /,''),{compilerOptions:{target:ts.ScriptTarget.ES2022}}).outputText
+ let socket,saves=0
+ class Socket { listeners={};constructor(){socket=this}addEventListener(name,fn){this.listeners[name]=fn}close(){this.closed=true} }
+ const watch=new Function('WebSocket','comfyWsUrl','saveRecord',`${code};return watchProgress`)(Socket,()=>'/fake',()=>saves++)
+ const record={promptId:'mine'},stop=watch(record,{clientId:'client'})
+ const send=data=>socket.listeners.message({data:JSON.stringify({type:'progress',data})})
+ send({prompt_id:'another',value:1,max:8});assert.equal(saves,0)
+ send({prompt_id:'mine',value:3,max:8});assert.equal(record.progress.step,3);assert.equal(record.progress.maxStep,8)
+ send({value:4,max:8});assert.equal(record.progress.percent,50)
+ stop();assert.equal(socket.closed,true)
+})