From db13dbca958a78eed29227ddb1e1989aa36ce90e Mon Sep 17 00:00:00 2001
From: serge-rider <serge@jkiss.org>
Date: Thu, 24 Dec 2020 00:21:24 +0300
Subject: [PATCH] CB-506 Logs servlet

---
 .../service/admin/WebServiceBindingAdmin.java | 14 +++-
 .../admin/impl/WebAdminLogsServlet.java       | 74 +++++++++++++++++++
 2 files changed, 87 insertions(+), 1 deletion(-)
 create mode 100644 server/bundles/io.cloudbeaver.service.admin/src/io/cloudbeaver/service/admin/impl/WebAdminLogsServlet.java

diff --git a/server/bundles/io.cloudbeaver.service.admin/src/io/cloudbeaver/service/admin/WebServiceBindingAdmin.java b/server/bundles/io.cloudbeaver.service.admin/src/io/cloudbeaver/service/admin/WebServiceBindingAdmin.java
index 58bafed731..3c475a9f2a 100644
--- a/server/bundles/io.cloudbeaver.service.admin/src/io/cloudbeaver/service/admin/WebServiceBindingAdmin.java
+++ b/server/bundles/io.cloudbeaver.service.admin/src/io/cloudbeaver/service/admin/WebServiceBindingAdmin.java
@@ -19,14 +19,19 @@ package io.cloudbeaver.service.admin;
 import io.cloudbeaver.DBWebException;
 import io.cloudbeaver.WebServiceUtils;
 import io.cloudbeaver.model.WebConnectionConfig;
+import io.cloudbeaver.server.CBApplication;
 import io.cloudbeaver.service.DBWBindingContext;
+import io.cloudbeaver.service.DBWServiceBindingServlet;
 import io.cloudbeaver.service.WebServiceBindingBase;
+import io.cloudbeaver.service.admin.impl.WebAdminLogsServlet;
 import io.cloudbeaver.service.admin.impl.WebServiceAdmin;
+import org.eclipse.jetty.servlet.ServletContextHandler;
+import org.eclipse.jetty.servlet.ServletHolder;
 
 /**
  * Web service implementation
  */
-public class WebServiceBindingAdmin extends WebServiceBindingBase<DBWServiceAdmin> {
+public class WebServiceBindingAdmin extends WebServiceBindingBase<DBWServiceAdmin> implements DBWServiceBindingServlet {
 
     private static final String SCHEMA_FILE_NAME = "schema/service.admin.graphqls";
 
@@ -90,4 +95,11 @@ public class WebServiceBindingAdmin extends WebServiceBindingBase<DBWServiceAdmi
         ;
     }
 
+    @Override
+    public void addServlets(CBApplication application, ServletContextHandler servletContextHandler) {
+        servletContextHandler.addServlet(
+            new ServletHolder("adminLogs", new WebAdminLogsServlet(application)),
+            application.getServicesURI() + "logs/*");
+    }
+
 }
diff --git a/server/bundles/io.cloudbeaver.service.admin/src/io/cloudbeaver/service/admin/impl/WebAdminLogsServlet.java b/server/bundles/io.cloudbeaver.service.admin/src/io/cloudbeaver/service/admin/impl/WebAdminLogsServlet.java
new file mode 100644
index 0000000000..e2bc27b89c
--- /dev/null
+++ b/server/bundles/io.cloudbeaver.service.admin/src/io/cloudbeaver/service/admin/impl/WebAdminLogsServlet.java
@@ -0,0 +1,74 @@
+package io.cloudbeaver.service.admin.impl;
+
+import io.cloudbeaver.DBWConstants;
+import io.cloudbeaver.DBWebException;
+import io.cloudbeaver.model.session.WebSession;
+import io.cloudbeaver.server.CBApplication;
+import io.cloudbeaver.server.CBPlatform;
+import org.jkiss.dbeaver.Log;
+import org.jkiss.dbeaver.model.DBConstants;
+import org.jkiss.dbeaver.utils.GeneralUtils;
+import org.jkiss.dbeaver.utils.MimeTypes;
+import org.jkiss.utils.IOUtils;
+
+import javax.servlet.ServletException;
+import javax.servlet.http.HttpServlet;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.IOException;
+import java.io.InputStream;
+
+public class WebAdminLogsServlet extends HttpServlet {
+
+    private static final Log log = Log.getLog(WebAdminLogsServlet.class);
+
+    private final CBApplication application;
+
+    public WebAdminLogsServlet(CBApplication application) {
+        this.application = application;
+    }
+
+    @Override
+    protected void service(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
+        try {
+            WebSession webSession = CBPlatform.getInstance().getSessionManager().findWebSession(request);
+            if (webSession == null) {
+                throw new DBWebException("No active session");
+            }
+            if (!webSession.hasPermission(DBWConstants.PERMISSION_ADMIN)) {
+                response.sendError(HttpServletResponse.SC_FORBIDDEN, "Logs accessible for admins only");
+                return;
+            }
+
+            String logId = request.getPathInfo();
+            String logFileName = ".log";
+            if ("debug".equals(logId)) {
+                logFileName = DBConstants.DEBUG_LOG_FILE_NAME;
+            }
+            if (logFileName.contains("/") || logFileName.contains("\\")) {
+                throw new DBWebException("Bad log file name");
+            }
+            File logFile = new File(GeneralUtils.getMetadataFolder(), logFileName);
+            if (!logFile.exists()) {
+                throw new DBWebException("Log file '" + logFileName + "' not found");
+            }
+
+            response.setHeader("Content-Type", MimeTypes.TEXT_PLAIN);
+            if (logFileName.equals(".log")) {
+                logFileName = "server.log";
+            }
+            response.setHeader("Content-Disposition", "attachment; filename=\"" + logFileName + "\"");
+
+            try (InputStream is = new FileInputStream(logFile)) {
+                IOUtils.copyStream(is, response.getOutputStream());
+            }
+       } catch (Exception e) {
+            log.error(e);
+            response.sendError(HttpServletResponse.SC_BAD_REQUEST, "Error reading log: " + e.getMessage());
+            return;
+        }
+    }
+
+}
\ No newline at end of file