1 files changed, 230 insertions, 0 deletions
diff --git a/src/main/java/com/p4square/grow/frontend/GrowFrontend.java b/src/main/java/com/p4square/grow/frontend/GrowFrontend.java
new file mode 100644
index 0000000..b5f62fb
--- /dev/null
+++ b/src/main/java/com/p4square/grow/frontend/GrowFrontend.java
@@ -0,0 +1,230 @@
+/*
+ * Copyright 2013 Jesse Morgan <jesse@jesterpm.net>
+ */
+
+package com.p4square.grow.frontend;
+
+import java.io.File;
+import java.io.IOException;
+import java.lang.reflect.Constructor;
+
+import freemarker.template.Template;
+
+import org.restlet.Application;
+import org.restlet.Component;
+import org.restlet.Context;
+import org.restlet.Restlet;
+import org.restlet.data.Protocol;
+import org.restlet.resource.Directory;
+import org.restlet.routing.Redirector;
+import org.restlet.routing.Router;
+import org.restlet.security.Authenticator;
+
+import com.codahale.metrics.MetricRegistry;
+
+import org.apache.log4j.Logger;
+
+import com.p4square.fmfacade.FMFacade;
+import com.p4square.fmfacade.FreeMarkerPageResource;
+
+import com.p4square.grow.config.Config;
+
+import com.p4square.restlet.metrics.MetricRouter;
+
+import com.p4square.session.SessionCheckingAuthenticator;
+import com.p4square.session.SessionCreatingAuthenticator;
+import org.restlet.security.Verifier;
+
+/**
+ * This is the Restlet Application implementing the Grow project front-end.
+ * It's implemented as an extension of FMFacade that connects interactive pages
+ * with various ServerResources. This class provides a main method to start a
+ * Jetty instance for testing.
+ *
+ * @author Jesse Morgan <jesse@jesterpm.net>
+ */
+public class GrowFrontend extends FMFacade {
+    private static Logger LOG = Logger.getLogger(GrowFrontend.class);
+
+    private final Config mConfig;
+    private final MetricRegistry mMetricRegistry;
+
+    private IntegrationDriver mIntegrationFactory;
+
+    public GrowFrontend() {
+        this(new Config(), new MetricRegistry());
+    }
+
+    public GrowFrontend(Config config, MetricRegistry metricRegistry) {
+        mConfig = config;
+        mMetricRegistry = metricRegistry;
+    }
+
+    public Config getConfig() {
+        return mConfig;
+    }
+
+    public MetricRegistry getMetrics() {
+        return mMetricRegistry;
+    }
+
+    @Override
+    public synchronized void start() throws Exception {
+        Template errorTemplate = getTemplate("templates/error.ftl");
+        if (errorTemplate != null) {
+            ErrorPage.setTemplate(errorTemplate,
+                    FreeMarkerPageResource.baseRootObject(getContext(), this));
+        }
+
+        getContext().getAttributes().put("com.p4square.grow.config", mConfig);
+        getContext().getAttributes().put("com.p4square.grow.metrics", mMetricRegistry);
+
+        super.start();
+    }
+
+    public synchronized IntegrationDriver getThirdPartyIntegrationFactory() {
+        if (mIntegrationFactory == null) {
+            final String driverClassName = getConfig().getString("integrationDriver",
+                                                                 "com.p4square.f1oauth.FellowshipOneIntegrationDriver");
+            try {
+                Class<?> clazz = Class.forName(driverClassName);
+                Constructor<?> constructor = clazz.getConstructor(Context.class);
+                mIntegrationFactory = (IntegrationDriver) constructor.newInstance(getContext());
+            } catch (Exception e) {
+                LOG.error("Failed to instantiate IntegrationDriver " + driverClassName);
+            }
+        }
+
+        return mIntegrationFactory;
+    }
+
+    @Override
+    protected Router createRouter() {
+        Router router = new MetricRouter(getContext(), mMetricRegistry);
+
+        final Authenticator defaultGuard = new SessionCheckingAuthenticator(getContext(), true);
+        defaultGuard.setNext(FreeMarkerPageResource.class);
+        router.attachDefault(defaultGuard);
+        router.attach("/", new Redirector(getContext(), "index.html", Redirector.MODE_CLIENT_PERMANENT));
+        router.attach("/login.html", LoginPageResource.class);
+        router.attach("/newaccount.html", NewAccountResource.class);
+        router.attach("/newbeliever", NewBelieverResource.class);
+
+        final Router accountRouter = new MetricRouter(getContext(), mMetricRegistry);
+        accountRouter.attach("/authenticate", AuthenticatedResource.class);
+        accountRouter.attach("/logout", LogoutResource.class);
+
+        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);
+        accountRouter.attach("/training/{chapter}", TrainingPageResource.class);
+        accountRouter.attach("/training", TrainingPageResource.class);
+        accountRouter.attach("/feed/{topic}", FeedResource.class);
+        accountRouter.attach("/feed/{topic}/{thread}", FeedResource.class);
+
+        final Authenticator accountGuard = createAuthenticatorChain(accountRouter);
+        router.attach("/account", accountGuard);
+
+        return router;
+    }
+
+    private Authenticator createAuthenticatorChain(Restlet last) {
+        final Context context = getContext();
+        final String loginPage = getConfig().getString("dynamicRoot", "") + "/login.html";
+        final String loginPost = getConfig().getString("dynamicRoot", "") + "/account/authenticate";
+        final String defaultPage = getConfig().getString("dynamicRoot", "") + "/account";
+
+        // This is used to check for an existing session
+        SessionCheckingAuthenticator sessionChk = new SessionCheckingAuthenticator(context, true);
+
+        // This is used to authenticate the user
+        Verifier verifier = getThirdPartyIntegrationFactory().newUserAuthenticationVerifier();
+        LoginFormAuthenticator loginAuth = new LoginFormAuthenticator(context, false, verifier);
+        loginAuth.setLoginFormUrl(loginPage);
+        loginAuth.setLoginPostUrl(loginPost);
+        loginAuth.setDefaultPage(defaultPage);
+
+        // This is used to create a new session for a newly authenticated user.
+        SessionCreatingAuthenticator sessionCreate = new SessionCreatingAuthenticator(context);
+
+        sessionChk.setNext(loginAuth);
+        loginAuth.setNext(sessionCreate);
+
+        sessionCreate.setNext(last);
+
+        return sessionChk;
+    }
+
+    /**
+     * Stand-alone main for testing.
+     */
+    public static void main(String[] args) {
+        // Start the HTTP Server
+        final Component component = new Component();
+        component.getServers().add(Protocol.HTTP, 8085);
+        component.getClients().add(Protocol.HTTP);
+        component.getClients().add(Protocol.HTTPS);
+        component.getClients().add(Protocol.FILE);
+        //component.getClients().add(new Client(null, Arrays.asList(Protocol.HTTPS), "org.restlet.ext.httpclient.HttpClientHelper"));
+
+        // Static content
+        try {
+            component.getDefaultHost().attach("/images/", new FileServingApp("./build/root/images/"));
+            component.getDefaultHost().attach("/scripts", new FileServingApp("./build/root/scripts"));
+            component.getDefaultHost().attach("/style.css", new FileServingApp("./build/root/style.css"));
+            component.getDefaultHost().attach("/favicon.ico", new FileServingApp("./build/root/favicon.ico"));
+            component.getDefaultHost().attach("/notfound.html", new FileServingApp("./build/root/notfound.html"));
+            component.getDefaultHost().attach("/error.html", new FileServingApp("./build/root/error.html"));
+        } catch (IOException e) {
+            LOG.error("Could not create directory for static resources: "
+                    + e.getMessage(), e);
+        }
+
+        // Setup App
+        GrowFrontend app = new GrowFrontend();
+
+        // Load an optional config file from the first argument.
+        app.getConfig().setDomain("dev");
+        if (args.length == 1) {
+            app.getConfig().updateConfig(args[0]);
+        }
+
+        component.getDefaultHost().attach(app);
+
+        // Setup shutdown hook
+        Runtime.getRuntime().addShutdownHook(new Thread() {
+            public void run() {
+                try {
+                    component.stop();
+                } catch (Exception e) {
+                    LOG.error("Exception during cleanup", e);
+                }
+            }
+        });
+
+        LOG.info("Starting server...");
+
+        try {
+            component.start();
+        } catch (Exception e) {
+            LOG.fatal("Could not start: " + e.getMessage(), e);
+        }
+    }
+
+    private static class FileServingApp extends Application {
+        private final String mPath;
+
+        public FileServingApp(String path) throws IOException {
+            mPath = new File(path).getAbsolutePath();
+        }
+
+        @Override
+        public Restlet createInboundRoot() {
+            return new Directory(getContext(), "file://" + mPath);
+        }
+    }
+}