false DOCUMENT_ACTION_SAVE editor 2023-03-31T17:34:57.216+02:00 Generate AI Variants GenerateAIImageVariants false subshell-content-nt:imageObject def IMAGE_AI_URI = 'https://imageai.example.com:8080' def IGNORE_USERNAMES = ['importer'] class Response { CropHint[] cropHints String cropHintsComment } class CropHint { Rectangle rectangle } class Rectangle { int x int y int width int height } if (IGNORE_USERNAMES.contains(sophoraClient.username)) { return } def doc = context.document if (!context.isUpdatable(doc)) { return } if (context.trigger != ClientScriptTrigger.DIRECT && doc.UUID != null) { return } def variantNodes = doc.getNodesByName('sophora-extension:imagedata') def originalVariantNode = variantNodes.find { it.getString('sophora-extension:imagetype') == 'original' } if (!originalVariantNode) { return } def originalBinary = originalVariantNode.getBinaryData('sophora-extension:binarydata') if (!originalBinary) { return } def originalWidth = originalVariantNode.getLong('sophora-extension:width') def originalHeight = originalVariantNode.getLong('sophora-extension:height') def originalAspectRatio = (double) originalWidth / (double) originalHeight def siteUUID = doc.structureInfo.siteUUID def siteVariants = sophoraClient.getImageVariantsForSite(siteUUID) .findAll { it.name != 'original' } .findAll { it.getStateForSite(siteUUID) == com.subshell.sophora.api.structure.ImageVariant.State.ACTIVE } def aspectRatios = siteVariants.collect { it.width != null && it.height != null ? (double) it.width / (double) it.height : originalAspectRatio } def clientBuilder = context.getHttpClientBuilder() .userAgent('Image AI Generate AI Variants') .timeout(java.time.Duration.ofSeconds(30)) def response def error = false context.showBusyWhile('Calculate image variants', { try (def client = clientBuilder.build()) { def restClient = new HttpClientRestWrapper(client) def request = [ image: java.util.Base64.encoder.encodeToString(originalBinary.bytes), cropHints: [ aspectRatios: aspectRatios ] ] response = restClient.post(IMAGE_AI_URI + '/rest/v1/image-ai', request, Response.class) } catch (Exception e) { error = true } }) if (error) { context.dialog().showMessage('An error has occurred. Please try again later.') return } response.cropHints.eachWithIndex { cropHint, cropHintIdx -> def variantName = siteVariants[cropHintIdx].name def variantNode = variantNodes.find { it.getString('sophora-extension:imagetype') == variantName } variantNode.setLong('sophora-extension:clipX', cropHint.rectangle.x) variantNode.setLong('sophora-extension:clipY', cropHint.rectangle.y) variantNode.setLong('sophora-extension:width', cropHint.rectangle.width) variantNode.setLong('sophora-extension:height', cropHint.rectangle.height) variantNode.setBoolean('sophora-extension:disabled', false) } def historyItem = doc.addNode('sophora-extension:imageAIHistory', 'sophora-extension-nt:imageAIHistoryItem') historyItem.setDate('sophora-extension:eventDate', java.util.GregorianCalendar.instance) historyItem.setString('sophora-extension:description', response.cropHintsComment) context.updateDocument(doc) default Image AI: Generate AI Variants system / clientscriptdocument true false