package cn.com.smartbi;


import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;


import org.apache.log4j.Logger;


import smartbi.SmartbiException;
import smartbi.config.ISystemConfig;
import smartbi.connectionpool.ConnectionPool;
import smartbi.framework.IModule;
import smartbi.freequery.FreeQueryErrorCode;
import smartbi.freequery.client.config.ConfigClientService;
import smartbi.olap.OlapErrorCode;
import smartbi.olap.OlapQueryService;
import smartbi.olap.query.OlapQueryBO;
import smartbi.state.IStateModule;
import smartbi.usermanager.IUserManagerModule;
import smartbi.usermanager.Role;
import smartbi.usermanager.User;
import smartbi.usermanager.UserBO;


public class ExportModule implements IModule{
 private static final Logger LOG = Logger.getLogger(ExportModule.class);
 private static ExportModule instance;
 
 public static ExportModule getInstance(){
  if (instance == null)
   instance = new ExportModule();
  return instance;
 }


 /**
  * 用户管理模块
  */
 private IUserManagerModule userManagerModule;
 
 public IUserManagerModule getUserManagerModule() {
  return userManagerModule;
 }


 public void setUserManagerModule(IUserManagerModule userManagerModule) {
  this.userManagerModule = userManagerModule;
 }
 
 /**
  * 状态模块
  */
 private IStateModule stateModule;
 
 public IStateModule getStateModule() {
  return stateModule;
 }


 public void setStateModule(IStateModule stateModule) {
  this.stateModule = stateModule;
 }
 
 public boolean isContainRole(String username, String password){
  User user = (User) userManagerModule.getUserByName(username);
  if(user==null){
   throw new SmartbiException(FreeQueryErrorCode.EXPORT_ERROR).setDetail("系统不存在该用户！");
  }else{
   String currentUsername = userManagerModule.getCurrentUser().getName();
   if(currentUsername.equals(user.getName())){
    throw new SmartbiException(FreeQueryErrorCode.EXPORT_ERROR).setDetail("不能给自己授权！");
   }
   String pwd = user.getPassword();
   UserBO ub = new UserBO(user);
   ub.setPasswordInner(password);
   String MD5Password = ub.getPassword();
   if(pwd.equals(MD5Password)){
    try{
     //String roleName = getValue("exportRoleName");
     String roleName = "";  //可通过系统选项添加
     ConfigClientService cs = ConfigClientService.getInstance();
     ISystemConfig scf = cs.getSystemConfig("ExportRoleName");
     if(scf != null){
      roleName = scf.getValue();
     }
     Role role = (Role) userManagerModule.getRoleByName(roleName);
     if(role==null){
      throw new SmartbiException(FreeQueryErrorCode.EXPORT_ERROR).setDetail("系统角色不存在！");
     }else{
      List<Role> roleList = user.getAssignedRoles();
      for(Role rl : roleList){
       if (rl.getId().equals(role.getId()))
        return true;
      }
      return false;
     }
    }catch(Exception e){
     throw new SmartbiException(FreeQueryErrorCode.EXPORT_ERROR,e);
    }
    
   }else{
    throw new SmartbiException(FreeQueryErrorCode.EXPORT_ERROR).setDetail("用户名或密码错误！");
   }
  }
 }
 
 //判断当前报表是否在权限控制表中
 public boolean isInReport(String resid){
  String sql = "select c_reportid from t_exportreport";
  ArrayList<String> list = new ArrayList<String>();
  Connection conn = null;
  PreparedStatement prep = null;
  ResultSet rs = null;
  try {
   conn = ConnectionPool.getInstance().getConnection("DS.SYSTEM知识库");
   prep = conn.prepareStatement(sql);
   rs = prep.executeQuery();
   while (rs.next()){
    String reportid = rs.getString("c_reportid");
    list.add(reportid);
   }
   rs.close();
   if(list.contains(resid)){
    return true;
   }
   return false;
  } catch (Exception e) {
   throw new SmartbiException(FreeQueryErrorCode.SQL_ERROR, e);
  } finally {
   try {
    prep.close();
    conn.close();
   } catch (SQLException e) {
    LOG.error(e.getMessage(), e);
   }
  }
 }
 
 public boolean olapIsInReport(String clientId){
  OlapQueryBO report = (OlapQueryBO) OlapQueryService.getInstance()
    .getStateModule().getSessionAttribute(clientId);
  if (report == null) {
   throw new SmartbiException(
     OlapErrorCode.EXPORT_REPORT_NOT_FOUND);
  }
  return isInReport(report.getId());
 }
  
 @Override
 public void activate() {
 }
}

package cn.com.smartbi;
import javax.servlet.Filter;
import javax.servlet.FilterChain;
import javax.servlet.FilterConfig;
import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServletRequest;
import org.apache.log4j.Logger;
import smartbi.SmartbiException;
import smartbi.decisionpanel.DecisionPanelModule;
import smartbi.decisionpanel.dashboard.DashboardBO;
import smartbi.freequery.FreeQueryErrorCode;
import smartbi.olap.OlapErrorCode;
import smartbi.olap.OlapQueryService;
import smartbi.olap.query.OlapQueryBO;
import smartbi.oltp.FreeQueryModule;
import smartbi.state.IStateModule;
public class ExportFilter implements Filter{
 private Logger log = Logger.getLogger(ExportFilter.class);
 @Override
 public void destroy() {
  // TODO Auto-generated method stub
  
 }
 @Override
 public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain){
  try{
   request.setCharacterEncoding("UTF-8");
   HttpServletRequest req = (HttpServletRequest)request;
   if(req.getRequestURI().indexOf("ExportServlet") != -1){
    String actionType = request.getParameter("actionType");
    if ("download".equals(actionType)) {
     chain.doFilter(request, response);
     return;
    }
    ExportModule em = ExportModule.getInstance();
    String clientId = request.getParameter("clientId");
    String resid = "";
    if(req.getRequestURI().indexOf("/ExportServlet") != -1){
     resid = request.getParameter("queryId");
    }else if(req.getRequestURI().indexOf("/InsightExportServlet") != -1){
     /*HttpSession session = req.getSession();
     InsightBO report = (InsightBO)  session.getAttribute(clientId);
     if (report == null)
      throw new SmartbiException(FreeQueryErrorCode.EXPORT_REPORT_NOT_FOUND);*/
     resid = request.getParameter("insightId");
    }else if(req.getRequestURI().indexOf("/DpExportServlet") != -1){
     IStateModule stateModule = DecisionPanelModule.getInstance().getStateModule();
     DashboardBO report = (DashboardBO) stateModule.getSessionAttribute(clientId);
     if (report == null)
      throw new SmartbiException(FreeQueryErrorCode.REPORT_CLIENT_ERROR);
     resid = report.getId();
    }else if(req.getRequestURI().indexOf("/OlapExportServlet") != -1){
     OlapQueryBO report = (OlapQueryBO) OlapQueryService.getInstance()
       .getStateModule().getSessionAttribute(clientId);
     if (report == null) {
      throw new SmartbiException(
        OlapErrorCode.EXPORT_REPORT_NOT_FOUND);
     }
     resid = report.getId();
    }else{
     chain.doFilter(request, response);
     return;
    }
    
    if(em.isInReport(resid)){
     String username = request.getParameter("uname");
     String password = request.getParameter("upsw");
     boolean rtn = ExportModule.getInstance().isContainRole(username, password);
     if(rtn){
      chain.doFilter(request, response);
     }else{
      throw new SmartbiException(FreeQueryErrorCode.EXPORT_ERROR).setDetail("权限验证不通过！");
     }
    }else{
     chain.doFilter(request, response);
    }
   }else if(req.getRequestURI().indexOf("/ssreportServlet") != -1){
    String refreshType = request.getParameter("refreshType");
    if("refresh".equals(refreshType)){
     chain.doFilter(request, response);
     
    }else{
     ExportModule em = ExportModule.getInstance();
     String resid = request.getParameter("resid");
     if(em.isInReport(resid)){
      String username = request.getParameter("uname");
      String password = request.getParameter("upsw");
      boolean rtn = ExportModule.getInstance().isContainRole(username, password);
      if(rtn){
       chain.doFilter(request, response);
      }else{
       throw new SmartbiException(FreeQueryErrorCode.EXPORT_ERROR).setDetail("权限验证不通过！");
      }
     }else{
      chain.doFilter(request, response);
     }
    }
   }else{
    chain.doFilter(request, response);
   }
   
  }catch(Exception e){
   log.error(e.getMessage(), e);
  }
  
 }
 
 protected IStateModule getStateModule() {
  return FreeQueryModule.getInstance().getStateModule();
 }
 @Override
 public void init(FilterConfig arg0) throws ServletException {
  
 }
}