package com.mossle.internal.whitelist.web;
import java.util.List;
import java.util.Map;
import javax.annotation.Resource;
import com.mossle.api.tenant.TenantHolder;
import com.mossle.core.export.Exportor;
import com.mossle.core.mapper.BeanMapper;
import com.mossle.core.page.Page;
import com.mossle.core.query.PropertyFilter;
import com.mossle.core.spring.MessageHelper;
import com.mossle.internal.whitelist.persistence.domain.WhitelistApp;
import com.mossle.internal.whitelist.persistence.manager.WhitelistAppManager;
import com.mossle.internal.whitelist.persistence.manager.WhitelistTypeManager;
import com.mossle.internal.whitelist.service.WhitelistService;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.ModelAttribute;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.servlet.mvc.support.RedirectAttributes;
@Controller
@RequestMapping("whitelist")
public class WhitelistAdminController {
private WhitelistAppManager whitelistAppManager;
private WhitelistTypeManager whitelistTypeManager;
private WhitelistService whitelistService;
private MessageHelper messageHelper;
private Exportor exportor;
private BeanMapper beanMapper = new BeanMapper();
private TenantHolder tenantHolder;
@RequestMapping("whitelist-admin-list")
public String list(@ModelAttribute Page page,
@RequestParam Map<String, Object> parameterMap, Model model) {
String tenantId = tenantHolder.getTenantId();
List<PropertyFilter> propertyFilters = PropertyFilter
.buildFromMap(parameterMap);
propertyFilters.add(new PropertyFilter("EQS_tenantId", tenantId));
page = whitelistAppManager.pagedQuery(page, propertyFilters);
model.addAttribute("page", page);
return "whitelist/whitelist-admin-list";
}
@RequestMapping("whitelist-admin-input")
public String input(
@RequestParam(value = "id", required = false) Long id,
@RequestParam(value = "type", required = false) String whitelistTypeCode,
Model model) {
String tenantId = tenantHolder.getTenantId();
WhitelistApp whitelistApp = null;
if (id != null) {
whitelistApp = whitelistAppManager.get(id);
model.addAttribute("model", whitelistApp);
}
if (whitelistApp != null) {
model.addAttribute("whitelistType", whitelistApp.getWhitelistType());
} else if (whitelistTypeCode != null) {
model.addAttribute("whitelistType", whitelistTypeManager
.findUnique(
"from WhitelistType where code=? and tenantId=?",
whitelistTypeCode, tenantId));
} else {
model.addAttribute("whitelistTypes",
whitelistTypeManager.findBy("tenantId", tenantId));
}
return "whitelist/whitelist-admin-input";
}
@RequestMapping("whitelist-admin-save")
public String save(@ModelAttribute WhitelistApp whitelistApp,
@RequestParam("typeId") Long whitelistTypeId,
@RequestParam("host") String hostContent,
@RequestParam("ip") String ipContent,
RedirectAttributes redirectAttributes) {
String tenantId = tenantHolder.getTenantId();
whitelistService.saveWhitelistApp(whitelistApp, whitelistTypeId,
hostContent, ipContent, whitelistApp.getUserId(), tenantId);
messageHelper.addFlashMessage(redirectAttributes, "core.success.save",
"保存成功");
return "redirect:/whitelist/whitelist-admin-list.do";
}
@RequestMapping("whitelist-admin-remove")
public String remove(@RequestParam("selectedItem") List<Long> selectedItem,
RedirectAttributes redirectAttributes) {
List<WhitelistApp> whitelistApps = whitelistAppManager
.findByIds(selectedItem);
for (WhitelistApp whitelistApp : whitelistApps) {
whitelistAppManager.removeAll(whitelistApp.getWhitelistHosts());
whitelistAppManager.removeAll(whitelistApp.getWhitelistIps());
whitelistAppManager.remove(whitelistApp);
}
messageHelper.addFlashMessage(redirectAttributes,
"core.success.delete", "删除成功");
return "redirect:/whitelist/whitelist-admin-list.do";
}
// ~ ======================================================================
@Resource
public void setWhitelistAppManager(WhitelistAppManager whitelistAppManager) {
this.whitelistAppManager = whitelistAppManager;
}
@Resource
public void setWhitelistTypeManager(
WhitelistTypeManager whitelistTypeManager) {
this.whitelistTypeManager = whitelistTypeManager;
}
@Resource
public void setWhitelistService(WhitelistService whitelistService) {
this.whitelistService = whitelistService;
}
@Resource
public void setMessageHelper(MessageHelper messageHelper) {
this.messageHelper = messageHelper;
}
@Resource
public void setExportor(Exportor exportor) {
this.exportor = exportor;
}
@Resource
public void setTenantHolder(TenantHolder tenantHolder) {
this.tenantHolder = tenantHolder;
}
}