#236 qr code generation for cloning via HTTP API

Author: doom369

integration-tests/pom.xml

@@ -69,7 +69,7 @@
         <dependency>
             <groupId>com.github.kenglxn.qrgen</groupId>
             <artifactId>javase</artifactId>
-            <version>2.1.0</version>
+            <version>${qrgen.version}</version>
             <scope>test</scope>
         </dependency>
     </dependencies>

pom.xml

@@ -102,6 +102,7 @@
         <commons-validator.version>1.5.0</commons-validator.version>
         <postgresql.version>9.4.1208</postgresql.version>
         <HikariCP.version>2.4.7</HikariCP.version>
+        <qrgen.version>2.1.0</qrgen.version>
 
         <!-- test versions -->
         <commons-lang3.version>3.4</commons-lang3.version>

server/core/src/main/java/cc/blynk/server/core/protocol/enums/Command.java

@@ -70,7 +70,8 @@ public final class Command {
     public static final short HTTP_NOTIFY = 46;
     public static final short HTTP_EMAIL = 47;
     public static final short HTTP_GET_PROJECT = 48;
-    public static final short HTTP_TOTAL = 49;
+    public static final short HTTP_QR = 49;
+    public static final short HTTP_TOTAL = 50;
 
 
     //all this code just to make logging more user-friendly

server/core/src/main/java/cc/blynk/server/core/stats/GlobalStats.java

@@ -9,7 +9,7 @@
  */
 public class GlobalStats {
 
-    private static final int LAST_COMMAND_INDEX = 50;
+    private static final int LAST_COMMAND_INDEX = 51;
     public final Meter incomeMessages;
     public final LongAdder[] specificCounters;
 

server/http-api/pom.xml

@@ -23,13 +23,18 @@
             <version>${project.version}</version>
         </dependency>
 
-
         <dependency>
             <groupId>commons-validator</groupId>
             <artifactId>commons-validator</artifactId>
             <version>${commons-validator.version}</version>
         </dependency>
 
+        <dependency>
+            <groupId>com.github.kenglxn.qrgen</groupId>
+            <artifactId>javase</artifactId>
+            <version>${qrgen.version}</version>
+        </dependency>
+
     </dependencies>
 
 </project>

server/http-api/src/main/java/cc/blynk/server/api/http/logic/HttpAPILogic.java

@@ -25,12 +25,17 @@
 import cc.blynk.server.notifications.push.GCMWrapper;
 import cc.blynk.server.notifications.push.android.AndroidGCMMessage;
 import cc.blynk.server.notifications.push.ios.IOSGCMMessage;
+import cc.blynk.utils.ByteUtils;
 import cc.blynk.utils.StringUtils;
+import net.glxn.qrgen.core.image.ImageType;
+import net.glxn.qrgen.javase.QRCode;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
 
 import javax.ws.rs.*;
 import javax.ws.rs.core.MediaType;
+import java.io.IOException;
+import java.util.Base64;
 
 import static cc.blynk.core.http.Response.*;
 import static cc.blynk.server.core.protocol.enums.Command.*;
@@ -186,6 +191,39 @@ public Response getWidgetPinData(@PathParam("token") String token,
         return ok(widget.getJsonValue());
     }
 
+    @GET
+    @Path("{token}/qr")
+    //todo cover with test
+    public Response updateWidgetPinData(@PathParam("token") String token) {
+        globalStats.mark(HTTP_QR);
+
+        User user = userDao.tokenManager.getUserByToken(token);
+
+        if (user == null) {
+            log.error("Requested token {} not found.", token);
+            return Response.badRequest("Invalid token.");
+        }
+
+        Integer dashId = user.getDashIdByToken(token);
+
+        if (dashId == null) {
+            log.error("Dash id for token {} not found. User {}", token, user.name);
+            return Response.badRequest("Didn't find dash id for token.");
+        }
+
+        DashBoard dashBoard = user.profile.getDashById(dashId);
+
+        try {
+            byte[] compressed = ByteUtils.compress(dashBoard.toString());
+            String qrData = "bp1" + Base64.getEncoder().encodeToString(compressed);
+            byte[] qrDataBinary = QRCode.from(qrData).to(ImageType.PNG).withSize(500, 500).stream().toByteArray();
+            return ok(qrDataBinary, "image/png");
+        } catch (IOException ioe) {
+            log.error("Error generating QR.", ioe);
+            return Response.badRequest("Error generating QR.");
+        }
+    }
+
     @PUT
     @Path("{token}/pin/{pin}")
     @Consumes(value = MediaType.APPLICATION_JSON)

server/http-core/src/main/java/cc/blynk/core/http/Response.java

@@ -36,6 +36,13 @@ public Response(HttpVersion version, HttpResponseStatus status, String content,
         headers().set(CONTENT_LENGTH, content().readableBytes());
     }
 
+    public Response(HttpVersion version, HttpResponseStatus status, byte[] content, String contentType) {
+        super(version, status, (content == null ? Unpooled.EMPTY_BUFFER : Unpooled.copiedBuffer(content)));
+        headers().set(CONNECTION, HttpHeaders.Values.KEEP_ALIVE);
+        headers().set(CONTENT_TYPE, contentType);
+        headers().set(CONTENT_LENGTH, content().readableBytes());
+    }
+
     public Response(HttpVersion version, HttpResponseStatus status) {
         super(version, status);
         headers().set(CONNECTION, HttpHeaders.Values.KEEP_ALIVE);
@@ -85,6 +92,11 @@ public static Response ok(String data, String contentType) {
         return new Response(HTTP_1_1, OK, data, contentType);
     }
 
+    public static Response ok(byte[] data, String contentType) {
+        return new Response(HTTP_1_1, OK, data, contentType);
+    }
+
+
     public static Response ok(boolean bool) {
         return new Response(HTTP_1_1, OK, String.valueOf(bool), JSON);
     }

server/http-core/src/main/java/cc/blynk/core/http/handlers/StaticFileHandler.java

@@ -1,5 +1,6 @@
 package cc.blynk.core.http.handlers;
 
+import cc.blynk.utils.ContentTypeUtil;
 import cc.blynk.utils.ServerProperties;
 import io.netty.buffer.Unpooled;
 import io.netty.channel.*;
@@ -106,20 +107,6 @@ private static void setDateAndCacheHeaders(io.netty.handler.codec.http.HttpRespo
                 LAST_MODIFIED, dateFormatter.format(new Date(fileToCache.lastModified())));
     }
 
-    private static String getContentType(String fileName) {
-        if (fileName.endsWith(".ico")) {
-            return "image/x-icon";
-        }
-        if (fileName.endsWith(".js")) {
-            return "application/javascript";
-        }
-        if (fileName.endsWith(".css")) {
-            return "text/css";
-        }
-
-        return "text/html";
-    }
-
     @Override
     public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
         if (!(msg instanceof FullHttpRequest)) {
@@ -211,7 +198,7 @@ private void serveStatic(ChannelHandlerContext ctx, FullHttpRequest request) thr
         HttpHeaders.setContentLength(response, fileLength);
 
         //setting content type
-        response.headers().set(CONTENT_TYPE, getContentType(file.getName()));
+        response.headers().set(CONTENT_TYPE, ContentTypeUtil.getContentType(file.getName()));
 
         setDateAndCacheHeaders(response, file);
         if (HttpHeaders.isKeepAlive(request)) {

server/http-core/src/main/java/cc/blynk/utils/ContentTypeUtil.java

@@ -0,0 +1,27 @@
+package cc.blynk.utils;
+
+/**
+ * The Blynk Project.
+ * Created by Dmitriy Dumanskiy.
+ * Created on 07.07.16.
+ */
+public class ContentTypeUtil {
+
+    public static String getContentType(String fileName) {
+        if (fileName.endsWith(".ico")) {
+            return "image/x-icon";
+        }
+        if (fileName.endsWith(".js")) {
+            return "application/javascript";
+        }
+        if (fileName.endsWith(".css")) {
+            return "text/css";
+        }
+        if (fileName.endsWith(".png")) {
+            return "image/png";
+        }
+
+        return "text/html";
+    }
+
+}