|
|
|
@ -1,12 +1,12 @@ |
|
|
|
/** |
|
|
|
* Copyright © 2016-2023 The Thingsboard Authors |
|
|
|
* |
|
|
|
* <p> |
|
|
|
* Licensed under the Apache License, Version 2.0 (the "License"); |
|
|
|
* you may not use this file except in compliance with the License. |
|
|
|
* You may obtain a copy of the License at |
|
|
|
* |
|
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
* |
|
|
|
* <p> |
|
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
* <p> |
|
|
|
* Unless required by applicable law or agreed to in writing, software |
|
|
|
* distributed under the License is distributed on an "AS IS" BASIS, |
|
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
|
|
|
@ -15,146 +15,49 @@ |
|
|
|
*/ |
|
|
|
package org.thingsboard.server.service.edge.rpc.fetch; |
|
|
|
|
|
|
|
import com.datastax.oss.driver.api.core.uuid.Uuids; |
|
|
|
import com.fasterxml.jackson.databind.JsonNode; |
|
|
|
import com.fasterxml.jackson.databind.node.ObjectNode; |
|
|
|
import freemarker.template.Configuration; |
|
|
|
import freemarker.template.Template; |
|
|
|
import lombok.AllArgsConstructor; |
|
|
|
import lombok.extern.slf4j.Slf4j; |
|
|
|
import org.apache.commons.lang3.text.WordUtils; |
|
|
|
import org.thingsboard.common.util.JacksonUtil; |
|
|
|
import org.thingsboard.server.common.data.AdminSettings; |
|
|
|
import org.thingsboard.server.common.data.EdgeUtils; |
|
|
|
import org.thingsboard.server.common.data.StringUtils; |
|
|
|
import org.thingsboard.server.common.data.edge.Edge; |
|
|
|
import org.thingsboard.server.common.data.edge.EdgeEvent; |
|
|
|
import org.thingsboard.server.common.data.edge.EdgeEventActionType; |
|
|
|
import org.thingsboard.server.common.data.edge.EdgeEventType; |
|
|
|
import org.thingsboard.server.common.data.id.AdminSettingsId; |
|
|
|
import org.thingsboard.server.common.data.id.EdgeId; |
|
|
|
import org.thingsboard.server.common.data.id.TenantId; |
|
|
|
import org.thingsboard.server.common.data.page.PageData; |
|
|
|
import org.thingsboard.server.common.data.page.PageLink; |
|
|
|
import org.thingsboard.server.dao.settings.AdminSettingsService; |
|
|
|
|
|
|
|
import java.util.ArrayList; |
|
|
|
import java.util.Arrays; |
|
|
|
import java.util.HashMap; |
|
|
|
import java.util.List; |
|
|
|
import java.util.Map; |
|
|
|
import java.util.regex.Matcher; |
|
|
|
import java.util.regex.Pattern; |
|
|
|
|
|
|
|
@AllArgsConstructor |
|
|
|
@Slf4j |
|
|
|
public class AdminSettingsEdgeEventFetcher implements EdgeEventFetcher { |
|
|
|
|
|
|
|
private final AdminSettingsService adminSettingsService; |
|
|
|
private final Configuration freemarkerConfig; |
|
|
|
|
|
|
|
private static final Pattern startPattern = Pattern.compile("<div class=\"content\".*?>"); |
|
|
|
private static final Pattern endPattern = Pattern.compile("<div class=\"footer\".*?>"); |
|
|
|
|
|
|
|
private static final List<String> templatesNames = Arrays.asList( |
|
|
|
"account.activated.ftl", |
|
|
|
"account.lockout.ftl", |
|
|
|
"activation.ftl", |
|
|
|
"password.was.reset.ftl", |
|
|
|
"reset.password.ftl", |
|
|
|
"test.ftl"); |
|
|
|
|
|
|
|
// TODO: @voba fix format of next templates
|
|
|
|
// "state.disabled.ftl",
|
|
|
|
// "state.enabled.ftl",
|
|
|
|
// "state.warning.ftl",
|
|
|
|
|
|
|
|
@Override |
|
|
|
public PageLink getPageLink(int pageSize) { |
|
|
|
return null; |
|
|
|
} |
|
|
|
|
|
|
|
@Override |
|
|
|
public PageData<EdgeEvent> fetchEdgeEvents(TenantId tenantId, Edge edge, PageLink pageLink) throws Exception { |
|
|
|
List<EdgeEvent> result = new ArrayList<>(); |
|
|
|
|
|
|
|
AdminSettings systemMailSettings = adminSettingsService.findAdminSettingsByKey(TenantId.SYS_TENANT_ID, "mail"); |
|
|
|
result.add(EdgeUtils.constructEdgeEvent(tenantId, edge.getId(), EdgeEventType.ADMIN_SETTINGS, |
|
|
|
EdgeEventActionType.UPDATED, null, JacksonUtil.valueToTree(systemMailSettings))); |
|
|
|
|
|
|
|
AdminSettings tenantMailSettings = convertToTenantAdminSettings(tenantId, systemMailSettings.getKey(), (ObjectNode) systemMailSettings.getJsonValue()); |
|
|
|
result.add(EdgeUtils.constructEdgeEvent(tenantId, edge.getId(), EdgeEventType.ADMIN_SETTINGS, |
|
|
|
EdgeEventActionType.UPDATED, null, JacksonUtil.valueToTree(tenantMailSettings))); |
|
|
|
|
|
|
|
AdminSettings systemMailTemplates = loadMailTemplates(tenantId); |
|
|
|
result.add(EdgeUtils.constructEdgeEvent(tenantId, edge.getId(), EdgeEventType.ADMIN_SETTINGS, |
|
|
|
EdgeEventActionType.UPDATED, null, JacksonUtil.valueToTree(systemMailTemplates))); |
|
|
|
|
|
|
|
AdminSettings tenantMailTemplates = convertToTenantAdminSettings(tenantId, systemMailTemplates.getKey(), (ObjectNode) systemMailTemplates.getJsonValue()); |
|
|
|
result.add(EdgeUtils.constructEdgeEvent(tenantId, edge.getId(), EdgeEventType.ADMIN_SETTINGS, |
|
|
|
EdgeEventActionType.UPDATED, null, JacksonUtil.valueToTree(tenantMailTemplates))); |
|
|
|
public PageData<EdgeEvent> fetchEdgeEvents(TenantId tenantId, Edge edge, PageLink pageLink) { |
|
|
|
List<EdgeEvent> result = fetchAdminSettingsForKeys(tenantId, edge.getId(), List.of("general", "mail", "connectivity", "jwt")); |
|
|
|
|
|
|
|
// return PageData object to be in sync with other fetchers
|
|
|
|
return new PageData<>(result, 1, result.size(), false); |
|
|
|
} |
|
|
|
|
|
|
|
private AdminSettings loadMailTemplates(TenantId tenantId) throws Exception { |
|
|
|
Map<String, Object> mailTemplates = new HashMap<>(); |
|
|
|
for (String templatesName : templatesNames) { |
|
|
|
Template template = freemarkerConfig.getTemplate(templatesName); |
|
|
|
if (template != null) { |
|
|
|
String name = validateName(template.getName()); |
|
|
|
Map<String, String> mailTemplate = getMailTemplateFromFile(template.toString()); |
|
|
|
if (mailTemplate != null) { |
|
|
|
mailTemplates.put(name, mailTemplate); |
|
|
|
} else { |
|
|
|
log.error("[{}] Can't load mail template from file {}", tenantId, template.getName()); |
|
|
|
} |
|
|
|
} |
|
|
|
} |
|
|
|
AdminSettings adminSettings = new AdminSettings(); |
|
|
|
adminSettings.setId(new AdminSettingsId(Uuids.timeBased())); |
|
|
|
adminSettings.setKey("mailTemplates"); |
|
|
|
adminSettings.setJsonValue(JacksonUtil.convertValue(mailTemplates, JsonNode.class)); |
|
|
|
return adminSettings; |
|
|
|
} |
|
|
|
|
|
|
|
private Map<String, String> getMailTemplateFromFile(String stringTemplate) { |
|
|
|
Map<String, String> mailTemplate = new HashMap<>(); |
|
|
|
Matcher start = startPattern.matcher(stringTemplate); |
|
|
|
Matcher end = endPattern.matcher(stringTemplate); |
|
|
|
if (start.find() && end.find()) { |
|
|
|
String body = StringUtils.substringBetween(stringTemplate, start.group(), end.group()).replaceAll("\t", ""); |
|
|
|
String subject = StringUtils.substringBetween(body, "<h2>", "</h2>"); |
|
|
|
mailTemplate.put("subject", subject); |
|
|
|
mailTemplate.put("body", body); |
|
|
|
} else { |
|
|
|
return null; |
|
|
|
} |
|
|
|
return mailTemplate; |
|
|
|
} |
|
|
|
|
|
|
|
private String validateName(String name) throws Exception { |
|
|
|
StringBuilder nameBuilder = new StringBuilder(); |
|
|
|
name = name.replace(".ftl", ""); |
|
|
|
String[] nameParts = name.split("\\."); |
|
|
|
if (nameParts.length >= 1) { |
|
|
|
nameBuilder.append(nameParts[0]); |
|
|
|
for (int i = 1; i < nameParts.length; i++) { |
|
|
|
String word = WordUtils.capitalize(nameParts[i]); |
|
|
|
nameBuilder.append(word); |
|
|
|
} |
|
|
|
return nameBuilder.toString(); |
|
|
|
} else { |
|
|
|
throw new Exception("Error during filename validation"); |
|
|
|
private List<EdgeEvent> fetchAdminSettingsForKeys(TenantId tenantId, EdgeId edgeId, List<String> keys) { |
|
|
|
List<EdgeEvent> result = new ArrayList<>(); |
|
|
|
for (String key : keys) { |
|
|
|
AdminSettings adminSettings = adminSettingsService.findAdminSettingsByKey(TenantId.SYS_TENANT_ID, key); |
|
|
|
result.add(EdgeUtils.constructEdgeEvent(tenantId, edgeId, EdgeEventType.ADMIN_SETTINGS, |
|
|
|
EdgeEventActionType.UPDATED, null, JacksonUtil.valueToTree(adminSettings))); |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
private AdminSettings convertToTenantAdminSettings(TenantId tenantId, String key, ObjectNode jsonValue) { |
|
|
|
AdminSettings tenantMailSettings = new AdminSettings(); |
|
|
|
tenantMailSettings.setTenantId(tenantId); |
|
|
|
jsonValue.put("useSystemMailSettings", true); |
|
|
|
tenantMailSettings.setJsonValue(jsonValue); |
|
|
|
tenantMailSettings.setKey(key); |
|
|
|
return tenantMailSettings; |
|
|
|
return result; |
|
|
|
} |
|
|
|
} |
|
|
|
|
Volodymyr Babak
3 years ago