Add format=json support to collections

Difegue · Sep 14, 2024 · 2e8bdff · 2e8bdff
1 parent 1aa103e
commit 2e8bdff
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/src/com/difegue/doujinsoft/CollectionServlet.java b/src/com/difegue/doujinsoft/CollectionServlet.java
@@ -59,7 +59,11 @@ private Collection initCollection(HttpServletRequest request) throws FileNotFoun
 	protected void doGet(HttpServletRequest request, HttpServletResponse response)
 			throws ServletException, IOException {
 
-		response.setContentType("text/html; charset=UTF-8");
+		if (request.getParameterMap().containsKey("format") && request.getParameter("format").equals("json"))
+			response.setContentType("application/json; charset=UTF-8");
+		else 
+			response.setContentType("text/html; charset=UTF-8");
+
 		ServletContext application = getServletConfig().getServletContext();
 		String output = "Collection doesn't exist!";
 

diff --git a/src/com/difegue/doujinsoft/utils/TemplateBuilderCollection.java b/src/com/difegue/doujinsoft/utils/TemplateBuilderCollection.java
@@ -48,6 +48,13 @@ public String doStandardPageCollection(Collection c) throws Exception {
 		result.close();
 		statement.close();
 		connection.close();
+
+		// JSON hijack if specified in the parameters
+		if (request.getParameterMap().containsKey("format") && request.getParameter("format").equals("json")) {
+			Gson gson = new Gson();
+			return gson.toJson(context);
+		}
+
 		return writeToTemplate();
     }