Adding a resource to delete an assessment.20131118a

4 files changed, 167 insertions, 0 deletions

diff --git a/src/com/p4square/grow/backend/db/CassandraDatabase.java b/src/com/p4square/grow/backend/db/CassandraDatabase.java
index e40c251..b8cb6df 100644
--- a/src/com/p4square/grow/backend/db/CassandraDatabase.java
+++ b/src/com/p4square/grow/backend/db/CassandraDatabase.java
@@ -16,6 +16,7 @@ import com.netflix.astyanax.ColumnMutation;
 import com.netflix.astyanax.model.Column;
 import com.netflix.astyanax.model.ColumnFamily;
 import com.netflix.astyanax.model.ColumnList;
+import com.netflix.astyanax.ColumnListMutation;
 import com.netflix.astyanax.MutationBatch;
 import com.netflix.astyanax.serializers.StringSerializer;
 import com.netflix.astyanax.thrift.ThriftFamilyFactory;
@@ -189,4 +190,23 @@ public class CassandraDatabase {
             throw new RuntimeException(e);
         }
     }
+
+    /**
+     * Remove a row
+     */
+    public void deleteRow(final String cfName, final String key) {
+        ColumnFamily<String, String> cf = new ColumnFamily(cfName,
+            StringSerializer.get(),
+            StringSerializer.get());
+
+        try {
+            MutationBatch batch = mKeyspace.prepareMutationBatch();
+            ColumnListMutation<String> cfm = batch.withRow(cf, key).delete();
+            batch.execute();
+
+        } catch (ConnectionException e) {
+            cLog.error("deleteRow failed due to Connection Exception", e);
+            throw new RuntimeException(e);
+        }
+    }
 }
diff --git a/src/com/p4square/grow/backend/resources/SurveyResultsResource.java b/src/com/p4square/grow/backend/resources/SurveyResultsResource.java
index e87126d..eea0fb1 100644
--- a/src/com/p4square/grow/backend/resources/SurveyResultsResource.java
+++ b/src/com/p4square/grow/backend/resources/SurveyResultsResource.java
@@ -117,6 +117,7 @@ public class SurveyResultsResource extends ServerResource {
 
             default:
                 setStatus(Status.CLIENT_ERROR_METHOD_NOT_ALLOWED);
+                return null;
         }
 
         if (success) {
@@ -130,6 +131,52 @@ public class SurveyResultsResource extends ServerResource {
     }
 
     /**
+     * Clear assessment results.
+     */
+    @Override
+    protected Representation delete() {
+        boolean success = false;
+
+        switch (mRequestType) {
+            case ANSWER:
+                try {
+                    mDb.deleteKey("assessments", mUserId, mQuestionId);
+                    mDb.deleteKey("assessments", mUserId, "summary");
+                    success = true;
+
+                } catch (Exception e) {
+                    LOG.warn("Caught exception putting answer: " + e.getMessage(), e);
+                }
+                break;
+
+            case ASSESSMENT:
+                try {
+                    mDb.deleteRow("assessments", mUserId);
+                    mDb.deleteKey("accounts", mUserId, "value"); // TODO: IMPORTANT! This needs to be narrower.
+                    success = true;
+
+                } catch (Exception e) {
+                    LOG.warn("Caught exception putting answer: " + e.getMessage(), e);
+                }
+                break;
+
+            default:
+                setStatus(Status.CLIENT_ERROR_METHOD_NOT_ALLOWED);
+                return null;
+        }
+
+        if (success) {
+            setStatus(Status.SUCCESS_NO_CONTENT);
+
+        } else {
+            setStatus(Status.SERVER_ERROR_INTERNAL);
+        }
+
+        return null;
+
+    }
+
+    /**
      * This method compiles assessment results.
      */
     private String buildAssessment() {
diff --git a/src/com/p4square/grow/frontend/AssessmentResetPage.java b/src/com/p4square/grow/frontend/AssessmentResetPage.java
new file mode 100644
index 0000000..95d5990
--- /dev/null
+++ b/src/com/p4square/grow/frontend/AssessmentResetPage.java
@@ -0,0 +1,99 @@
+/*
+ * Copyright 2013 Jesse Morgan
+ */
+
+package com.p4square.grow.frontend;
+
+import java.util.Map;
+
+import freemarker.template.Template;
+
+import org.restlet.data.MediaType;
+import org.restlet.data.Status;
+import org.restlet.representation.Representation;
+import org.restlet.representation.StringRepresentation;
+import org.restlet.ext.freemarker.TemplateRepresentation;
+
+import org.apache.log4j.Logger;
+
+import net.jesterpm.fmfacade.FreeMarkerPageResource;
+
+import net.jesterpm.fmfacade.json.JsonRequestClient;
+import net.jesterpm.fmfacade.json.JsonResponse;
+import net.jesterpm.fmfacade.json.ClientException;
+
+import com.p4square.grow.config.Config;
+
+/**
+ * This page delete's the current user's assessment.
+ *
+ * @author Jesse Morgan <jesse@jesterpm.net>
+ */
+public class AssessmentResetPage extends FreeMarkerPageResource {
+    private static final Logger LOG = Logger.getLogger(AssessmentResetPage.class);
+
+    private GrowFrontend mGrowFrontend;
+    private Config mConfig;
+    private JsonRequestClient mJsonClient;
+
+    private String mUserId;
+
+    @Override
+    public void doInit() {
+        super.doInit();
+
+        mGrowFrontend = (GrowFrontend) getApplication();
+        mConfig = mGrowFrontend.getConfig();
+
+        mJsonClient = new JsonRequestClient(getContext().getClientDispatcher());
+
+        mUserId = getRequest().getClientInfo().getUser().getIdentifier();
+    }
+
+    /**
+     * Return the login page.
+     */
+    @Override
+    protected Representation get() {
+        try {
+            // Get the assessment results
+            JsonResponse response = backendDelete("/accounts/" + mUserId + "/assessment");
+            if (!response.getStatus().isSuccess()) {
+                setStatus(Status.SERVER_ERROR_INTERNAL);
+                return ErrorPage.BACKEND_ERROR;
+            }
+
+            String nextPage = mConfig.getString("dynamicRoot", "")
+                    + "/account/assessment/question/first";
+            getResponse().redirectSeeOther(nextPage);
+            return new StringRepresentation("Redirecting to " + nextPage);
+
+        } catch (Exception e) {
+            LOG.fatal("Could not render page: " + e.getMessage(), e);
+            setStatus(Status.SERVER_ERROR_INTERNAL);
+            return ErrorPage.RENDER_ERROR;
+        }
+    }
+
+    /**
+     * @return The backend endpoint URI
+     */
+    private String getBackendEndpoint() {
+        return mConfig.getString("backendUri", "riap://component/backend");
+    }
+
+    /**
+     * Helper method to send a GET to the backend.
+     */
+    private JsonResponse backendDelete(final String uri) {
+        LOG.debug("Sending backend GET " + uri);
+
+        final JsonResponse response = mJsonClient.delete(getBackendEndpoint() + uri);
+        final Status status = response.getStatus();
+        if (!status.isSuccess() && !Status.CLIENT_ERROR_NOT_FOUND.equals(status)) {
+            LOG.warn("Error making backend request for '" + uri + "'. status = " + response.getStatus().toString());
+        }
+
+        return response;
+    }
+}
diff --git a/src/com/p4square/grow/frontend/GrowFrontend.java b/src/com/p4square/grow/frontend/GrowFrontend.java
index e9c2f5c..f6db9b8 100644
--- a/src/com/p4square/grow/frontend/GrowFrontend.java
+++ b/src/com/p4square/grow/frontend/GrowFrontend.java
@@ -103,6 +103,7 @@ public class GrowFrontend extends FMFacade {
         accountRouter.attach("", AccountRedirectResource.class);
         accountRouter.attach("/assessment/question/{questionId}", SurveyPageResource.class);
         accountRouter.attach("/assessment/results", AssessmentResultsPage.class);
+        accountRouter.attach("/assessment/reset", AssessmentResetPage.class);
         accountRouter.attach("/assessment", SurveyPageResource.class);
         accountRouter.attach("/training/{chapter}/completed", ChapterCompletePage.class);
         accountRouter.attach("/training/{chapter}/videos/{videoId}.json", VideosResource.class);