1 files changed, 138 insertions, 0 deletions
diff --git a/src/com/p4square/grow/backend/GrowBackend.java b/src/com/p4square/grow/backend/GrowBackend.java
new file mode 100644
index 0000000..515cd1b
--- /dev/null
+++ b/src/com/p4square/grow/backend/GrowBackend.java
@@ -0,0 +1,138 @@
+/*
+ * Copyright 2012 Jesse Morgan
+ */
+
+package com.p4square.grow.backend;
+
+import org.apache.log4j.Logger;
+
+import org.restlet.Application;
+import org.restlet.Component;
+import org.restlet.data.Protocol;
+import org.restlet.Restlet;
+import org.restlet.routing.Router;
+
+import com.p4square.grow.config.Config;
+
+import com.p4square.grow.backend.db.CassandraDatabase;
+import com.p4square.grow.backend.resources.SurveyResource;
+import com.p4square.grow.backend.resources.SurveyResultsResource;
+import com.p4square.grow.backend.resources.TrainingResource;
+import com.p4square.grow.backend.resources.TrainingRecordResource;
+
+/**
+ * Main class for the backend application.
+ *
+ * @author Jesse Morgan <jesse@jesterpm.net>
+ */
+public class GrowBackend extends Application {
+    private final static Logger cLog = Logger.getLogger(GrowBackend.class);
+
+    private final Config mConfig;
+    private final CassandraDatabase mDatabase;
+
+    public GrowBackend() {
+        mConfig = new Config();
+        mDatabase = new CassandraDatabase();
+    }
+
+    @Override
+    public Restlet createInboundRoot() {
+        Router router = new Router(getContext());
+
+        // Survey API
+        router.attach("/assessment/question/{questionId}", SurveyResource.class);
+        
+        router.attach("/accounts/{userId}/assessment", SurveyResultsResource.class);
+        router.attach("/accounts/{userId}/assessment/answers/{questionId}",
+                SurveyResultsResource.class);
+
+        // Training API
+        router.attach("/training/{level}", TrainingResource.class);
+        router.attach("/training/{level}/videos/{videoId}", TrainingResource.class);
+        
+        router.attach("/accounts/{userId}/training", TrainingRecordResource.class);
+        router.attach("/accounts/{userId}/training/videos/{videoId}",
+                TrainingRecordResource.class);
+
+
+        return router;
+    }
+    
+    /**
+     * Open the database.
+     */
+    @Override
+    public void start() throws Exception {
+        super.start();
+        
+        // Load config
+        final String configDomain =
+            getContext().getParameters().getFirstValue("configDomain");
+        if (configDomain != null) {
+            mConfig.setDomain(configDomain);
+        }
+
+        mConfig.updateConfig(this.getClass().getResourceAsStream("/grow.properties"));
+
+        final String configFilename =
+            getContext().getParameters().getFirstValue("configFile");
+
+        if (configFilename != null) {
+            mConfig.updateConfig(configFilename);
+        }
+
+        // Setup database
+        mDatabase.setClusterName(mConfig.getString("clusterName", "Dev Cluster"));
+        mDatabase.setKeyspaceName(mConfig.getString("keyspace", "GROW"));
+        mDatabase.init();
+    }
+    
+    /**
+     * Close the database.
+     */
+    @Override
+    public void stop() throws Exception {
+        cLog.info("Shutting down...");
+        mDatabase.close();
+
+        super.stop();
+    }
+
+    /**
+     * @return the current database.
+     */
+    public CassandraDatabase getDatabase() {
+        return mDatabase;
+    }
+
+    /**
+     * Stand-alone main for testing.
+     */
+    public static void main(String[] args) throws Exception {
+        // Start the HTTP Server
+        final Component component = new Component();
+        component.getServers().add(Protocol.HTTP, 9095);
+        component.getClients().add(Protocol.HTTP);
+        component.getDefaultHost().attach(new GrowBackend());
+        
+        // Setup shutdown hook
+        Runtime.getRuntime().addShutdownHook(new Thread() {
+            public void run() {
+                try {
+                    component.stop();
+                } catch (Exception e) {
+                    cLog.error("Exception during cleanup", e);
+                }
+            }
+        });
+
+        cLog.info("Starting server...");
+
+        try {
+            component.start();
+        } catch (Exception e) {
+            cLog.fatal("Could not start: " + e.getMessage(), e);
+        }
+    }
+}