com.h57.sample.controller.HomeController.java Source code

Introduction

Here is the source code for com.h57.sample.controller.HomeController.java

Source

/*
 * This file is licensed to you under the BSD License, (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.opensource.org/licenses/bsd-license.php
 * 
 * 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.h57.sample.controller;

import java.text.DateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Locale;

import javax.servlet.http.HttpServletRequest;
import javax.sql.DataSource;

import org.apache.shiro.SecurityUtils;
import org.apache.shiro.authc.AuthenticationException;
import org.apache.shiro.authc.UsernamePasswordToken;
import org.apache.shiro.subject.Subject;
import org.apache.tomcat.dbcp.dbcp.BasicDataSource;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.jdbc.datasource.SimpleDriverDataSource;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;

import com.h57.sample.domain.Identity;
import com.h57.sample.service.IdentityService;

/**
 * Handles requests for the application home page.
 * 
 * @author Bubba Hines <bubba@hines57.com>
 * @since 0.02
 */
@Controller
public class HomeController {
    @Autowired(required = true)
    IdentityService identityService;
    @Autowired(required = true)
    DataSource dataSource;

    private static final Logger logger = LoggerFactory.getLogger(HomeController.class);

    /**
     * Does some simple work to find the current shiro subject gets a list of
     * services, and the date.
     */
    @RequestMapping(method = RequestMethod.GET, value = { "/", "/index" })
    public String home(Locale locale, Model model, HttpServletRequest request) {
        logger.info("Welcome home! the client locale is " + locale.toString());

        // This gets the current subject from shiro
        Subject currentUser = SecurityUtils.getSubject();

        // I was going to have more services, who knows .. maybe we will add
        // more.
        List<String> services = new ArrayList<String>();
        // My SQL class org.apache.commons.dbcp.BasicDataSource
        if (dataSource instanceof BasicDataSource) {
            services.add("Data Source: " + ((BasicDataSource) dataSource).getUrl());
        } else if (dataSource instanceof SimpleDriverDataSource) {
            services.add("Data Source: " + ((SimpleDriverDataSource) dataSource).getUrl());
        }

        services.add("My SQL: " + dataSource.getClass());

        // Just to prove we can do it.
        Date date = new Date();
        DateFormat dateFormat = DateFormat.getDateTimeInstance(DateFormat.LONG, DateFormat.LONG, locale);

        String formattedDate = dateFormat.format(date);

        model.addAttribute("serverTime", formattedDate);

        // Lets get an identity object
        Identity thisIdentity = null;

        // Remembered (from cookie) is different from authenticated in Shiro
        if (currentUser.isRemembered()) {
            logger.info("Remembered PRINCIPAL: " + currentUser.getPrincipal());
            thisIdentity = identityService.getIdentity(currentUser.getPrincipal().toString());

            // Authenticated, we really do believe they are who they claim to
            // be!
        } else if (currentUser.isAuthenticated()) {
            logger.info("Authenticated PRINCIPAL: " + currentUser.getPrincipal());
            thisIdentity = identityService.getIdentity(currentUser.getPrincipal().toString());
        }

        // Pass this to the jsp.
        model.addAttribute("currentUser", currentUser);
        model.addAttribute("identity", thisIdentity);
        model.addAttribute("serverTime", formattedDate);
        model.addAttribute("services", services);
        return "home";
    }

    /**
     * Hard coded test against the default user that comes with the Schema.
     */
    @RequestMapping(value = "/smoketest", method = RequestMethod.GET)
    public String smoketest(Locale locale, Model model, HttpServletRequest request) {
        String salted = this.identityService.getApplicationSalt();
        logger.info("SALT = " + salted);
        String userid = "TestUser";
        String passphrase = "TestUserPassword";
        UsernamePasswordToken token = new UsernamePasswordToken(userid, passphrase);
        Subject currentUser = SecurityUtils.getSubject();

        try {
            currentUser.login(token);
            logger.info("AUTH SUCCESS");
        } catch (AuthenticationException ae) {
            logger.info("AUTH MSSG: " + ae.getMessage());
        }

        Identity thisIdentity = null;
        if (currentUser.isAuthenticated()) {
            logger.info("PRINCIPAL: " + currentUser.getPrincipal());
            thisIdentity = identityService.getIdentity(currentUser.getPrincipal().toString());
            salted = this.identityService.getCombinedSalt(thisIdentity.getSalt());
            logger.info("COMBINED SALT = " + salted);
        }
        model.addAttribute("identity", thisIdentity);
        return "smoketest/smoke";
    }
}