Revize 47d0b110

Created save diagram servlet

sources/src/main/java/cz/zcu/kiv/offscreen/servlets/actions/SaveDiagram.java

@@ +1 @@
+package cz.zcu.kiv.offscreen.servlets.actions;
+
+import com.google.common.base.Strings;
+import cz.zcu.kiv.offscreen.user.DB;
+import cz.zcu.kiv.offscreen.user.Diagram;
+import org.apache.commons.lang.StringUtils;
+
+import javax.servlet.ServletException;
+import javax.servlet.http.HttpServlet;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+import java.io.IOException;
+import java.util.HashMap;
+import java.util.Map;
+
+public class SaveDiagram extends HttpServlet {
+
+    @Override
+    protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
+
+        // user is not logged in
+        if (request.getSession().getAttribute("logged_user_id") == null) {
+            response.sendError(response.SC_UNAUTHORIZED);
+            return;
+        }
+
+        Integer loggedUserId = Integer.parseInt(request.getSession().getAttribute("logged_user_id").toString());
+        String name = request.getParameter("name");
+        String graphJson = request.getParameter("graph_json");
+        String isPublic = StringUtils.defaultIfBlank(request.getParameter("public"), "0");
+
+        // input parameters are invalid
+        if (Strings.isNullOrEmpty(name) || Strings.isNullOrEmpty(graphJson)) {
+            response.sendError(response.SC_BAD_REQUEST);
+            return;
+        }
+
+        DB db = new DB(getServletContext());
+        Diagram diagram;
+
+        if (request.getParameter("diagram_id") == null) {
+            // new diagram
+            diagram = new Diagram(db);
+
+        } else {
+            // diagram exists
+            Integer diagramId = Integer.parseInt(request.getParameter("diagram_id"));
+            diagram = new Diagram(db, diagramId);
+
+            // user is not owner of the diagram
+            if (loggedUserId != diagram.getUserId()) {
+                response.sendError(response.SC_UNAUTHORIZED);
+                return;
+            }
+        }
+
+
+        Map<String, String> diagramParams = new HashMap<>();
+        diagramParams.put("name", name);
+        diagramParams.put("public", isPublic);
+        diagramParams.put("graph_json", graphJson);
+        diagramParams.put("user_id", loggedUserId.toString());
+
+        diagram.update(diagramParams);
+    }
+}

sources/src/main/webapp/WEB-INF/web.xml

@@ -64 +64 @@
 		<servlet-name>UploadFiles</servlet-name>
 		<servlet-class>cz.zcu.kiv.offscreen.servlets.UploadFiles</servlet-class>
 	</servlet>
+	<servlet>
+		<servlet-name>SaveDiagram</servlet-name>
+		<servlet-class>cz.zcu.kiv.offscreen.servlets.actions.SaveDiagram</servlet-class>
+	</servlet>
 	<servlet>
 		<servlet-name>DeleteAllComponents</servlet-name>
 		<servlet-class>cz.zcu.kiv.offscreen.servlets.actions.DeleteAllComponents</servlet-class>
@@ -104 +108 @@
 		<servlet-name>ShowGraph</servlet-name>
 		<url-pattern>/graph</url-pattern>
 	</servlet-mapping>
+	<servlet-mapping>
+		<servlet-name>SaveDiagram</servlet-name>
+		<url-pattern>/api/save-diagram</url-pattern>
+	</servlet-mapping>
+
 	<servlet-mapping>
 		<servlet-name>DeleteAllComponents</servlet-name>
 		<url-pattern>/delete-components</url-pattern>