Move dashboard to test/vti.

Copy code from test/vts/web to test/vti/.

Bug: 62339915
Test: none
Change-Id: I6e07fbaca77018f6d7bd24bd3da9e3bbbafab495

1 files changed, 138 insertions, 0 deletions

diff --git a/src/main/java/com/android/vts/api/UserFavoriteRestServlet.java b/src/main/java/com/android/vts/api/UserFavoriteRestServlet.java
new file mode 100644
index 0000000..f863051
--- /dev/null
+++ b/src/main/java/com/android/vts/api/UserFavoriteRestServlet.java
@@ -0,0 +1,138 @@
+/*
+ * Copyright (c) 2017 Google Inc. All Rights Reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License"); you
+ * may not use this file except in compliance with the License. You may
+ * obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
+ * implied. See the License for the specific language governing
+ * permissions and limitations under the License.
+ */
+
+package com.android.vts.api;
+
+import com.android.vts.entity.TestEntity;
+import com.android.vts.entity.UserFavoriteEntity;
+import com.google.appengine.api.datastore.DatastoreService;
+import com.google.appengine.api.datastore.DatastoreServiceFactory;
+import com.google.appengine.api.datastore.Entity;
+import com.google.appengine.api.datastore.EntityNotFoundException;
+import com.google.appengine.api.datastore.Key;
+import com.google.appengine.api.datastore.KeyFactory;
+import com.google.appengine.api.datastore.Query;
+import com.google.appengine.api.datastore.Transaction;
+import com.google.appengine.api.datastore.Query.CompositeFilterOperator;
+import com.google.appengine.api.datastore.Query.Filter;
+import com.google.appengine.api.datastore.Query.FilterOperator;
+import com.google.appengine.api.datastore.Query.FilterPredicate;
+import com.google.appengine.api.users.User;
+import com.google.appengine.api.users.UserService;
+import com.google.appengine.api.users.UserServiceFactory;
+import com.google.gson.Gson;
+import com.google.gson.JsonObject;
+import com.google.gson.JsonPrimitive;
+import java.io.IOException;
+import java.io.PrintWriter;
+import java.util.logging.Level;
+import java.util.logging.Logger;
+import javax.servlet.http.HttpServlet;
+import javax.servlet.http.HttpServletRequest;
+import javax.servlet.http.HttpServletResponse;
+
+/** Servlet for handling requests to add or remove subscriptions. */
+public class UserFavoriteRestServlet extends HttpServlet {
+    protected static final Logger logger =
+            Logger.getLogger(UserFavoriteRestServlet.class.getName());
+
+    /**
+     * Add a test to the user's favorites.
+     */
+    @Override
+    public void doPost(HttpServletRequest request, HttpServletResponse response)
+            throws IOException {
+        UserService userService = UserServiceFactory.getUserService();
+        User currentUser = userService.getCurrentUser();
+        DatastoreService datastore = DatastoreServiceFactory.getDatastoreService();
+
+        // Retrieve the added tests from the request.
+        String test = request.getPathInfo();
+        if (test == null) {
+            response.setStatus(HttpServletResponse.SC_BAD_REQUEST);
+            return;
+        }
+        if (test.startsWith("/")) {
+            test = test.substring(1);
+        }
+        Key addedTestKey = KeyFactory.createKey(TestEntity.KIND, test);
+        // Filter the tests that exist from the set of tests to add
+        try {
+            datastore.get(addedTestKey);
+        } catch (EntityNotFoundException e) {
+            response.setStatus(HttpServletResponse.SC_BAD_REQUEST);
+            return;
+        }
+
+        Filter userFilter =
+                new FilterPredicate(UserFavoriteEntity.USER, FilterOperator.EQUAL, currentUser);
+        Filter testFilter = new FilterPredicate(
+                UserFavoriteEntity.TEST_KEY, FilterOperator.EQUAL, addedTestKey);
+        Query q = new Query(UserFavoriteEntity.KIND)
+                          .setFilter(CompositeFilterOperator.and(userFilter, testFilter))
+                          .setKeysOnly();
+
+        Key favoriteKey = null;
+
+        Transaction txn = datastore.beginTransaction();
+        try {
+            for (Entity e : datastore.prepare(q).asIterable()) {
+                favoriteKey = e.getKey();
+                break;
+            }
+            if (favoriteKey == null) {
+                UserFavoriteEntity favorite = new UserFavoriteEntity(currentUser, addedTestKey);
+                Entity entity = favorite.toEntity();
+                datastore.put(entity);
+                favoriteKey = entity.getKey();
+            }
+            txn.commit();
+        } finally {
+            if (txn.isActive()) {
+                logger.log(Level.WARNING,
+                        "Transaction rollback forced for favorite creation: " + test);
+                txn.rollback();
+            }
+        }
+
+        response.setContentType("application/json");
+        PrintWriter writer = response.getWriter();
+        JsonObject json = new JsonObject();
+        json.add("key", new JsonPrimitive(KeyFactory.keyToString(favoriteKey)));
+        writer.print(new Gson().toJson(json));
+        writer.flush();
+        response.setStatus(HttpServletResponse.SC_OK);
+    }
+
+    /**
+     * Remove a test from the user's favorites.
+     */
+    @Override
+    public void doDelete(HttpServletRequest request, HttpServletResponse response)
+            throws IOException {
+        DatastoreService datastore = DatastoreServiceFactory.getDatastoreService();
+        String stringKey = request.getPathInfo();
+        if (stringKey == null) {
+            response.setStatus(HttpServletResponse.SC_BAD_REQUEST);
+            return;
+        }
+        if (stringKey.startsWith("/")) {
+            stringKey = stringKey.substring(1);
+        }
+        datastore.delete(KeyFactory.stringToKey(stringKey));
+        response.setStatus(HttpServletResponse.SC_OK);
+    }
+}