extern/easydiffusion
1
0
Fork 1
mirror of https://github.com/easydiffusion/easydiffusion.git synced 2025-02-03 03:59:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

Allow plugin buttons for image overlay to decide whether they should be displayed or not

Browse Source
This commit is contained in:
cmdr2 2022-10-19 20:10:45 +05:30
parent 5fa3a7ca44
commit af05d94198
2 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
ui/media/js/main.js
View File

@ -381,18 +381,25 @@ function showImages(reqBody, res, outputContainer, livePreview) {
Object.assign(buttons, PLUGINS['IMAGE_INFO_BUTTONS'])
const imgItemInfo = imageItemElem.querySelector('.imgItemInfo')
const img = imageItemElem.querySelector('img')
const createButton = function(name, btnInfo) {
const newButton = document.createElement('button')
newButton.classList.add(name)
newButton.classList.add('tasksBtns')
newButton.innerText = btnInfo.text
newButton.addEventListener('click', function() {
let img = imageItemElem.querySelector('img')
btnInfo.on_click(req, img)
})
imgItemInfo.appendChild(newButton)
}
Object.keys(buttons).forEach((name) => createButton(name, buttons[name]))
Object.keys(buttons).forEach((name) => {
const btn = buttons[name]
if (btn.filter && btn.filter(req, img) === false) {
return
}
createButton(name, btn)
})
}
})
}

5
ui/media/js/plugins.js
View File

@ -16,6 +16,11 @@ const PLUGINS = {
* })
* newTaskRequest.seed = newTaskRequest.reqBody.seed
* createTask(newTaskRequest)
* },
* filter: function(origRequest, image) {
* // this is an optional function. return true/false to show/hide the button
* // if this function isn't set, the button will always be visible
* return true
* }
* }
*/