WindChat/windchat-admin/src/main/java/com/akaxin/site/web/admin/controller/PluginManageController.java

/**
 * Copyright 2018-2028 Akaxin Group
 * <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
 * <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.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.akaxin.site.web.admin.controller;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import javax.servlet.http.HttpServletRequest;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.servlet.ModelAndView;

import com.akaxin.common.utils.StringHelper;
import com.akaxin.proto.core.PluginProto;
import com.windchat.im.storage.bean.PluginBean;
import com.akaxin.site.web.admin.exception.UserPermissionException;
import com.akaxin.site.web.admin.service.IPluginService;
import com.google.protobuf.InvalidProtocolBufferException;

//插件扩展管理
@Controller
@RequestMapping("plugin")
public class PluginManageController extends AbstractController {
	private static final Logger logger = LoggerFactory.getLogger(UserManageController.class);

	@Autowired
	private IPluginService pluginService;

	@RequestMapping("/indexPage")
	public ModelAndView toPluginIndex(@RequestBody byte[] bodyParam) {
		ModelAndView modelAndView = new ModelAndView("plugin/index");
		PluginProto.ProxyPluginPackage pluginPackage = null;
		try {
			pluginPackage = PluginProto.ProxyPluginPackage.parseFrom(bodyParam);

			if (!isManager(getRequestSiteUserId(pluginPackage))) {
				throw new UserPermissionException("Current user is not a manager");
			}
			return modelAndView;

		} catch (InvalidProtocolBufferException e) {
			logger.error("to plugin  error", e);
		} catch (UserPermissionException e) {
			logger.error("to plugin  error : " + e.getMessage());
		}
		return new ModelAndView("error");
	}

	@RequestMapping("/addPage")
	public String toPluginAdd(@RequestBody byte[] bodyParam) {
		PluginProto.ProxyPluginPackage pluginPackage = null;
		try {
			pluginPackage = PluginProto.ProxyPluginPackage.parseFrom(bodyParam);

			if (!isManager(getRequestSiteUserId(pluginPackage))) {
				throw new UserPermissionException("Current user is not a manager");
			}
			return "plugin/add";

		} catch (InvalidProtocolBufferException e) {
			logger.error("to plugin add page error", e);
		} catch (UserPermissionException e) {
			logger.error("to plugin add page error : " + e.getMessage());
		}
		return "error";

	}

	@RequestMapping("/listPage")
	public String toPluginList(@RequestBody byte[] bodyParam) {
		PluginProto.ProxyPluginPackage pluginPackage = null;
		try {
			pluginPackage = PluginProto.ProxyPluginPackage.parseFrom(bodyParam);

			if (!isManager(getRequestSiteUserId(pluginPackage))) {
				throw new UserPermissionException("Current user is not a manager");
			}
			return "plugin/list";

		} catch (InvalidProtocolBufferException e) {
			logger.error("to plugin list error", e);
		} catch (UserPermissionException e) {
			logger.error("to plugin list error : " + e.getMessage());
		}
		return "error";
	}

	@RequestMapping("/editPage")
	public ModelAndView toEditPage(@RequestBody byte[] bodyParam) {
		ModelAndView modelAndView = new ModelAndView("plugin/update");
		Map<String, Object> model = modelAndView.getModel();
		PluginProto.ProxyPluginPackage pluginPackage = null;
		try {
			pluginPackage = PluginProto.ProxyPluginPackage.parseFrom(bodyParam);

			String siteUserId = getRequestSiteUserId(pluginPackage);
			if (!isManager(siteUserId)) {
				throw new UserPermissionException("Current user is not a manager");
			}
			// 解析Plugin_id
			int authKeyState = 1;
			String data = pluginPackage.getData();
			String[] split = data.split(":\"");
			String res = split[1].replaceAll("\"}", "");
			PluginBean plugin = pluginService.getPlugin(Integer.valueOf(res));
			model.put("name", plugin.getName());
			model.put("url_page", plugin.getUrlPage());
			model.put("api_url", plugin.getApiUrl());
			model.put("plugin_icon", plugin.getIcon());
			model.put("order", plugin.getSort());
			model.put("allow_ip", plugin.getAllowedIp());
			model.put("position", plugin.getPosition());
			model.put("per_status", plugin.getPermissionStatus());
			model.put("display_mode", plugin.getDisplayMode());
			model.put("plugin_id", plugin.getId());
			model.put("auth_key", plugin.getAuthKey());
			// 如果是默认添加的扩展则不提供修改authkey设置
			if (plugin.getId() == 1 || plugin.getId() == 2) {
				authKeyState = 0;
			}
			model.put("authKeyState", authKeyState);
			return modelAndView;
		} catch (InvalidProtocolBufferException e) {
			logger.error("to plugin  error", e);
		} catch (UserPermissionException e) {
			logger.error("to plugin  error : " + e.getMessage());
		}
		return new ModelAndView("error");
	}

	// 增加新扩展
	@RequestMapping(method = RequestMethod.POST, value = "/addPlugin")
	@ResponseBody
	public String addPlugin(HttpServletRequest request, @RequestBody byte[] bodyParam) {
		try {
			PluginProto.ProxyPluginPackage pluginPackage = PluginProto.ProxyPluginPackage.parseFrom(bodyParam);

			String siteUserId = getRequestSiteUserId(pluginPackage);
			if (!isManager(siteUserId)) {
				throw new UserPermissionException("Current user is not a manager");
			}
			Map<String, String> pluginData = getRequestDataMap(pluginPackage);
			PluginBean bean = new PluginBean();
			bean.setName(trim(pluginData.get("name")));
			bean.setIcon(pluginData.get("plugin_icon"));
			bean.setUrlPage(trim(pluginData.get("url_page")));
			bean.setApiUrl(trim(pluginData.get("api_url")));
			bean.setAllowedIp(trim(pluginData.get("allow_ip")));
			bean.setPosition(Integer.valueOf(pluginData.get("position")));
			bean.setSort(Integer.valueOf(pluginData.get("order")));
			bean.setDisplayMode(PluginProto.PluginDisplayMode.NEW_PAGE_VALUE);
			bean.setPermissionStatus(Integer.valueOf(pluginData.get("per_status")));
			bean.setDisplayMode(Integer.valueOf(pluginData.get("display_mode")));
			bean.setAddTime(System.currentTimeMillis());
			bean.setAuthKey(StringHelper.generateRandomString(16));// 随机生成

			logger.info("siteUserId={} add new plugin bean={}", siteUserId, bean);
			if (pluginService.addNewPlugin(bean)) {
				return SUCCESS;
			}
		} catch (InvalidProtocolBufferException e) {
			logger.error("add new plugin controller error", e);
		} catch (UserPermissionException e) {
			logger.error("add new plugin controller error : " + e.getMessage());
			return NO_PERMISSION;
		}

		return ERROR;
	}

	// 获取扩展列表
	@RequestMapping(method = RequestMethod.POST, value = "/pluginList")
	@ResponseBody
	public Map<String, Object> getPluginList(HttpServletRequest request, @RequestBody byte[] bodyParam) {
		Map<String, Object> result = new HashMap<String, Object>();
		boolean nodata = true;// 是还有更多数据
		try {
			PluginProto.ProxyPluginPackage pluginPackage = PluginProto.ProxyPluginPackage.parseFrom(bodyParam);
			String siteUserId = getRequestSiteUserId(pluginPackage);

			if (!isManager(siteUserId)) {
				throw new UserPermissionException("Current user is not a manager");
			}
			Map<String, String> dataMap = getRequestDataMap(pluginPackage);
			int pageNum = Integer.valueOf(dataMap.get("page"));
			List<PluginBean> pluginList = pluginService.getPluginList(pageNum, PAGE_SIZE);
			List<Map<String, Object>> data = new ArrayList<Map<String, Object>>();
			if (pluginList != null) {
				if (PAGE_SIZE == pluginList.size()) {
					nodata = false;
				}
				for (PluginBean bean : pluginList) {
					Map<String, Object> pluginMap = new HashMap<String, Object>();
					pluginMap.put("plugin_id", bean.getId());
					pluginMap.put("name", bean.getName());
					pluginMap.put("plugin_icon", bean.getIcon());
					pluginMap.put("url_page", bean.getUrlPage());
					pluginMap.put("api_url", bean.getApiUrl());
					pluginMap.put("position", bean.getPosition());
					pluginMap.put("order", bean.getSort());
					pluginMap.put("per_status", bean.getPermissionStatus());
					pluginMap.put("allow_ip", bean.getAllowedIp());
					// add to list
					data.add(pluginMap);
				}
			}
			result.put("pluginData", pluginList);
		} catch (InvalidProtocolBufferException e) {
			logger.error("get plugin list error", e);
		} catch (UserPermissionException e) {
			logger.error("get plugin list error : " + e.getMessage());
		}

		result.put("loading", nodata);
		return result;
	}

	// 编辑扩展
	@RequestMapping(method = RequestMethod.POST, value = "/editPlugin")
	@ResponseBody

	public String editPlugin(HttpServletRequest request, @RequestBody byte[] bodyParam) {
		try {
			PluginProto.ProxyPluginPackage pluginPackage = PluginProto.ProxyPluginPackage.parseFrom(bodyParam);
			String siteUserId = getRequestSiteUserId(pluginPackage);
			if (!isManager(siteUserId)) {
				throw new UserPermissionException("Current user is not a manager");
			}
			Map<String, String> pluginData = getRequestDataMap(pluginPackage);
			PluginBean bean = new PluginBean();
			bean.setId(Integer.valueOf(trim(pluginData.get("plugin_id"))));
			bean.setName(trim(pluginData.get("name")));
			bean.setIcon(pluginData.get("plugin_icon"));
			bean.setUrlPage(trim(pluginData.get("url_page")));
			bean.setApiUrl(trim(pluginData.get("api_url")));

			bean.setSort(Integer.valueOf(pluginData.get("order")));
			bean.setPosition(Integer.valueOf(pluginData.get("position")));
			bean.setDisplayMode(Integer.valueOf(pluginData.get("display_mode")));
			bean.setPermissionStatus(Integer.valueOf(pluginData.get("per_status")));

			bean.setAllowedIp(trim(pluginData.get("allow_ip")));
			if (pluginService.updatePlugin(bean)) {
				return SUCCESS;
			}
		} catch (InvalidProtocolBufferException e) {
			logger.error("edit plugin error", e);
		} catch (UserPermissionException e) {
			logger.error("edit plugin error : " + e.getMessage());
			return NO_PERMISSION;
		}

		return ERROR;
	}

	@RequestMapping(method = RequestMethod.POST, value = "/delPlugin")
	@ResponseBody
	public String deletePlugin(HttpServletRequest request, @RequestBody byte[] bodyParam) {
		try {
			PluginProto.ProxyPluginPackage pluginPackage = PluginProto.ProxyPluginPackage.parseFrom(bodyParam);
			String siteUserId = getRequestSiteUserId(pluginPackage);
			if (!isManager(siteUserId)) {
				throw new UserPermissionException("Current user is not a manager");
			}
			Map<String, String> dataMap = getRequestDataMap(pluginPackage);
			int pluginId = Integer.valueOf(dataMap.get("plugin_id"));

			if (pluginService.deletePlugin(pluginId)) {
				return SUCCESS;
			}
		} catch (InvalidProtocolBufferException e) {
			logger.error("edit plugin error", e);
		} catch (UserPermissionException e) {
			logger.error("edit plugin error : " + e.getMessage());
			return NO_PERMISSION;
		}
		return ERROR;
	}

	@RequestMapping(method = RequestMethod.POST, value = "/reSet")
	@ResponseBody
	public String reSetAuthKey(HttpServletRequest request, @RequestBody byte[] bodyParam) {
		try {
			PluginProto.ProxyPluginPackage pluginPackage = PluginProto.ProxyPluginPackage.parseFrom(bodyParam);
			String siteUserId = getRequestSiteUserId(pluginPackage);
			if (!isManager(siteUserId)) {
				throw new UserPermissionException("Current user is not a manager");
			}
			Map<String, String> dataMap = getRequestDataMap(pluginPackage);
			int pluginId = Integer.valueOf(dataMap.get("plugin_id"));
			String authKey = pluginService.reSetAuthKey(pluginId);
			return authKey;
		} catch (InvalidProtocolBufferException e) {
			logger.error("edit plugin error", e);
		} catch (UserPermissionException e) {
			logger.error("edit plugin error : " + e.getMessage());
			return NO_PERMISSION;
		}

		return ERROR;
	}
}