ViacheslavKlimov
3 years ago
16 changed files with 825 additions and 17 deletions
@ -0,0 +1,180 @@ |
|||
package org.thingsboard.server.service.install.update; |
|||
|
|||
import com.fasterxml.jackson.databind.JsonNode; |
|||
import com.fasterxml.jackson.databind.node.ArrayNode; |
|||
import com.fasterxml.jackson.databind.node.ObjectNode; |
|||
import lombok.RequiredArgsConstructor; |
|||
import lombok.extern.slf4j.Slf4j; |
|||
import org.apache.commons.lang3.StringUtils; |
|||
import org.springframework.stereotype.Component; |
|||
import org.thingsboard.common.util.JacksonUtil; |
|||
import org.thingsboard.server.common.data.Dashboard; |
|||
import org.thingsboard.server.common.data.ResourceType; |
|||
import org.thingsboard.server.common.data.TbResource; |
|||
import org.thingsboard.server.common.data.TbResourceInfo; |
|||
import org.thingsboard.server.common.data.id.TenantId; |
|||
import org.thingsboard.server.common.data.util.MediaTypeUtils; |
|||
import org.thingsboard.server.common.data.widget.WidgetTypeDetails; |
|||
import org.thingsboard.server.common.data.widget.WidgetsBundle; |
|||
import org.thingsboard.server.dao.resource.ResourceService; |
|||
|
|||
import java.util.List; |
|||
import java.util.Optional; |
|||
|
|||
@Component |
|||
//@Profile("install")
|
|||
@Slf4j |
|||
@RequiredArgsConstructor |
|||
public class ImagesUpdater { |
|||
|
|||
private final ResourceService resourceService; |
|||
|
|||
|
|||
// TODO: solution templates
|
|||
|
|||
|
|||
public void updateWidgetImages(WidgetTypeDetails widgetTypeDetails) { |
|||
String imageName = widgetTypeDetails.getName() + " - image"; |
|||
String imageKey = widgetTypeDetails.getFqn(); |
|||
String imageLink = saveImage(imageName, imageKey, widgetTypeDetails.getImage()); |
|||
widgetTypeDetails.setImage(imageLink); |
|||
|
|||
JsonNode defaultConfig = JacksonUtil.toJsonNode(widgetTypeDetails.getDescriptor().get("defaultConfig").asText()); |
|||
defaultConfig = updateWidgetConfig(TenantId.SYS_TENANT_ID, defaultConfig, imageName, imageKey, imageKey); |
|||
((ObjectNode) widgetTypeDetails.getDescriptor()).put("defaultConfig", defaultConfig.toString()); |
|||
} |
|||
|
|||
public void updateWidgetsBundleImages(WidgetsBundle widgetsBundle) { |
|||
String imageLink = saveImage(widgetsBundle.getTitle(), widgetsBundle.getAlias(), widgetsBundle.getImage()); |
|||
widgetsBundle.setImage(imageLink); |
|||
} |
|||
|
|||
public boolean updateDashboardImages(Dashboard dashboard) { |
|||
String imageNamePrefix = dashboard.getTitle(); |
|||
String imageKeyPrefix = "dashboard_" + dashboard.getUuidId(); |
|||
boolean updated = false; |
|||
|
|||
String imageLink = saveImage(dashboard.getTenantId(), imageNamePrefix + " - image", imageKeyPrefix + ".image", dashboard.getImage(), null); |
|||
dashboard.setImage(imageLink); |
|||
|
|||
for (ObjectNode widgetConfig : dashboard.getWidgetsConfig()) { |
|||
String alias = widgetConfig.get("bundleAlias").asText() + "." + widgetConfig.get("typeAlias").asText(); |
|||
String widgetName = widgetConfig.get("config").get("title").asText(); |
|||
updateWidgetConfig(dashboard.getTenantId(), widgetConfig.get("config"), |
|||
imageNamePrefix + " - " + widgetName + " widget", |
|||
imageKeyPrefix + ".widget." + alias, alias); |
|||
} |
|||
|
|||
return updated; |
|||
} |
|||
|
|||
private JsonNode updateWidgetConfig(TenantId tenantId, JsonNode widgetConfig, |
|||
String imageNamePrefix, String imageKeyPrefix, |
|||
String fqn) { |
|||
ObjectNode widgetSettings = (ObjectNode) widgetConfig.get("settings"); |
|||
ArrayNode markerImages = (ArrayNode) widgetSettings.get("markerImages"); |
|||
if (markerImages != null && !markerImages.isEmpty()) { |
|||
for (int i = 0; i < markerImages.size(); i++) { |
|||
String imageName = imageNamePrefix + " - marker image " + (i + 1); |
|||
String imageKey = imageKeyPrefix + ".marker_image_" + (i + 1); |
|||
String imageLink = saveImage(tenantId, imageName, imageKey, markerImages.get(i).asText(), fqn + ".marker_image_"); |
|||
markerImages.set(i, imageLink); |
|||
} |
|||
} |
|||
|
|||
String mapImage = getText(widgetSettings, "mapImageUrl"); |
|||
if (mapImage != null) { |
|||
String imageName = imageNamePrefix + " - map image"; |
|||
String imageKeySuffix = ".map_image"; |
|||
String imageKey = imageKeyPrefix + imageKeySuffix; |
|||
String imageLink = saveImage(tenantId, imageName, imageKey, mapImage, fqn + imageKeySuffix); |
|||
widgetSettings.put("mapImageUrl", imageLink); |
|||
} |
|||
|
|||
String backgroundImage = getText(widgetSettings, "backgroundImageUrl"); |
|||
if (backgroundImage != null) { |
|||
String imageName = imageNamePrefix + " - background image"; |
|||
String imageKeySuffix = ".background_image"; |
|||
String imageKey = imageKeyPrefix + imageKeySuffix; |
|||
String imageLink = saveImage(tenantId, imageName, imageKey, backgroundImage, fqn + imageKeySuffix); |
|||
widgetSettings.put("backgroundImageUrl", imageLink); |
|||
} |
|||
|
|||
JsonNode backgroundConfigNode = widgetSettings.get("background"); |
|||
if (backgroundConfigNode != null && backgroundConfigNode.isObject()) { |
|||
ObjectNode backgroundConfig = (ObjectNode) backgroundConfigNode; |
|||
if ("image".equals(getText(backgroundConfig, "type"))) { |
|||
String imageBase64 = getText(backgroundConfig, "imageBase64"); |
|||
if (imageBase64 != null) { |
|||
String imageName = imageNamePrefix + " - background image"; |
|||
String imageKeySuffix = ".background_image"; |
|||
String imageKey = imageKeyPrefix + imageKeySuffix; |
|||
String imageLink = saveImage(tenantId, imageName, imageKey, imageBase64, fqn + imageKeySuffix); |
|||
backgroundConfig.set("imageBase64", null); |
|||
backgroundConfig.put("imageUrl", imageLink); |
|||
backgroundConfig.put("type", "imageUrl"); |
|||
} |
|||
} |
|||
} |
|||
|
|||
return widgetConfig; |
|||
} |
|||
|
|||
private String getText(JsonNode jsonNode, String field) { |
|||
return Optional.ofNullable(jsonNode.get(field)) |
|||
.filter(JsonNode::isTextual) |
|||
.map(JsonNode::asText).orElse(null); |
|||
} |
|||
|
|||
private String saveImage(String title, String key, String data) { |
|||
return saveImage(TenantId.SYS_TENANT_ID, title, key, data, null); |
|||
} |
|||
|
|||
private String saveImage(TenantId tenantId, String title, String key, String data, |
|||
String existingImageQuery) { |
|||
if (data == null) { |
|||
return null; |
|||
} |
|||
String base64Data = StringUtils.substringAfter(data, "base64,"); |
|||
if (base64Data.isEmpty()) { |
|||
return data; |
|||
} |
|||
String imageMediaType = StringUtils.substringBetween(data, "data:", ";base64"); |
|||
String extension = MediaTypeUtils.getFileExtension(imageMediaType); |
|||
key += "." + extension; |
|||
|
|||
TbResourceInfo resourceInfo = resourceService.findResourceInfoByTenantIdAndKey(tenantId, ResourceType.IMAGE, key); |
|||
if (resourceInfo == null && !tenantId.isSysTenantId() && existingImageQuery != null) { |
|||
List<TbResourceInfo> existing = resourceService.findByTenantIdAndDataAndKeyStartingWith(TenantId.SYS_TENANT_ID, base64Data, existingImageQuery); |
|||
if (!existing.isEmpty()) { |
|||
resourceInfo = existing.get(0); |
|||
if (existing.size() > 1) { |
|||
log.warn("Found more than one system image resources for key {}", existingImageQuery); |
|||
} |
|||
log.info("Using system image {} for {}", resourceInfo.getLink(), key); |
|||
return resourceInfo.getLink(); |
|||
} |
|||
} |
|||
TbResource resource; |
|||
if (resourceInfo == null) { |
|||
resource = new TbResource(); |
|||
resource.setTenantId(tenantId); |
|||
resource.setResourceType(ResourceType.IMAGE); |
|||
resource.setResourceKey(key); |
|||
} else if (tenantId.isSysTenantId()) { |
|||
resource = new TbResource(resourceInfo); |
|||
} else { |
|||
return resourceInfo.getLink(); |
|||
} |
|||
resource.setTitle(title); |
|||
resource.setFileName(key); |
|||
resource.setMediaType(imageMediaType); |
|||
resource.setBase64Data(base64Data); |
|||
resource = resourceService.saveResource(resource); |
|||
log.info("[{}] {} image '{}' {} ({})", tenantId, resourceInfo == null ? "Created" : "Updated", |
|||
resource.getTitle(), resource.getResourceKey(), resource.getLink()); |
|||
return resource.getLink(); |
|||
|
|||
} |
|||
|
|||
} |
|||
@ -0,0 +1,22 @@ |
|||
package org.thingsboard.server.common.data.util; |
|||
|
|||
import lombok.AccessLevel; |
|||
import lombok.NoArgsConstructor; |
|||
import org.springframework.util.MimeTypeUtils; |
|||
|
|||
import java.util.Map; |
|||
|
|||
@NoArgsConstructor(access = AccessLevel.PRIVATE) |
|||
public class MediaTypeUtils { |
|||
|
|||
private static final Map<String, String> mappings = Map.of( |
|||
"jpeg", "jpg", |
|||
"svg+xml", "svg" |
|||
); |
|||
|
|||
public static String getFileExtension(String mimeType) { |
|||
String subtype = MimeTypeUtils.parseMimeType(mimeType).getSubtype(); |
|||
return mappings.getOrDefault(subtype, subtype); |
|||
} |
|||
|
|||
} |
|||
Loading…
Reference in new issue