WattDepotComponent.java

/**
 * WattDepotComponent.java This file is part of WattDepot.
 *
 * Copyright (C) 2013  Cam Moore
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
package org.wattdepot.server.http.api;


import com.fasterxml.jackson.databind.DeserializationFeature;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.SerializationFeature;
import org.restlet.Application;
import org.restlet.Component;
import org.restlet.data.Protocol;
import org.restlet.ext.jackson.JacksonRepresentation;
import org.restlet.representation.Representation;
import org.restlet.security.MemoryRealm;
import org.restlet.security.Role;
import org.restlet.security.User;
import org.wattdepot.common.domainmodel.Organization;
import org.wattdepot.common.domainmodel.UserInfo;
import org.wattdepot.common.domainmodel.UserPassword;
import org.wattdepot.common.exception.IdNotFoundException;
import org.wattdepot.server.StrongAES;
import org.wattdepot.server.WattDepotPersistence;
import org.wattdepot.server.depository.impl.hibernate.MeasurementImpl;

import javax.crypto.BadPaddingException;
import javax.crypto.IllegalBlockSizeException;
import java.io.IOException;
import java.security.InvalidKeyException;

/**
 * WattDepotComponent - Main class to start the WattDepot Http API component of
 * the WattDepotServer.
 * 
 * @author Cam Moore
 * 
 */
public class WattDepotComponent extends Component {

  /**
   * Sets up the WattDepotComponent with the given WattDepot.
   * 
   * @param depot The persitent store.
   * @throws javax.crypto.BadPaddingException if there is a problem with the encryption.
   * @throws java.security.InvalidKeyException if there is a problem with the encryption.
   * @throws javax.crypto.IllegalBlockSizeException if there is a problem with the encryption.
   * @throws java.io.IOException if there is a problem.
   */
  public WattDepotComponent(WattDepotPersistence depot) throws BadPaddingException, InvalidKeyException, IllegalBlockSizeException, IOException {
    setName("WattDepot HTTP API Server");
    setDescription("WattDepot RESTful server.");
    setAuthor("Cam Moore");
    getLogService().setLoggerName("org.wattdepot.server");
    // Add a CLAP client connector

    getClients().add(Protocol.CLAP);
    getClients().add(Protocol.FILE);
    getClients().add(Protocol.HTTP);


    getServers().getContext().getParameters().set("tracing", "true");

    WattDepotApplication app = new WattDepotApplication();
    app.setDepot(depot);
    // configure the JacksonRepresentation so that it uses ISO-8601 compliant
    // notation
    MeasurementImpl source = new MeasurementImpl();
    Representation rep = app.getConverterService().toRepresentation(source);
    @SuppressWarnings("rawtypes")
    ObjectMapper mapper = ((JacksonRepresentation) rep).getObjectMapper();
    mapper.disable(SerializationFeature.WRITE_DATES_AS_TIMESTAMPS);
//    mapper.enable(SerializationFeature.WRITE_DATES_AS_TIMESTAMPS);
    mapper.disable(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES);
    getDefaultHost().attachDefault(app);
    app.setComponent(this);

    // Set up the security realm
    MemoryRealm realm = new MemoryRealm();
    realm.setName("WattDepot Security");
    getRealms().add(realm);
    for (Organization group : app.getDepot().getOrganizations()) {
      app.getRoles().add(new Role((Application) app, group.getId()));
      if (Organization.ADMIN_GROUP.getName().equals(group.getName())) {
        // need to ensure the admin user is available.
        User user = new User(UserInfo.ROOT.getUid(), StrongAES.getInstance()
            .decrypt(UserPassword.ROOT.getEncryptedPassword()),
            UserInfo.ROOT.getFirstName(), UserInfo.ROOT.getLastName(),
            UserInfo.ROOT.getEmail());
        realm.getUsers().add(user);
        realm.map(user, app.getRole(group.getId()));
      }
      for (String userId : group.getUsers()) {
        UserPassword up = null;
        try {
          up = app.getDepot().getUserPassword(userId, group.getId(), true);
        }
        catch (IdNotFoundException e) {
          e.printStackTrace();
        }
        if (up != null) {
          User user = null;
          if (userId.equals(UserInfo.ROOT.getUid())) {
            // There isn't a UserInfo stored in persistence for ROOT.
            user = new User(UserInfo.ROOT.getUid(), StrongAES.getInstance()
                .decrypt(up.getEncryptedPassword()),
                UserInfo.ROOT.getFirstName(), UserInfo.ROOT.getLastName(),
                UserInfo.ROOT.getEmail());
          }
          else {
            UserInfo info;
            try {
              info = app.getDepot().getUser(userId, group.getId(), true);
              user = new User(info.getUid(), StrongAES.getInstance().decrypt(
                  up.getEncryptedPassword()), info.getFirstName(),
                  info.getLastName(), info.getEmail());
            }
            catch (IdNotFoundException e) {
              e.printStackTrace();
            }
          }
          realm.getUsers().add(user);
          realm.map(user, app.getRole(group.getId()));
        }
      }
    }

    // Set the realm's default enroler and verifier
    app.getContext().setDefaultEnroler(realm.getEnroler());
    app.getContext().setDefaultVerifier(realm.getVerifier());

    // Properties props = new Properties();
    // InputStream inputStream = this.getClass().getClassLoader()
    // .getResourceAsStream("log.properties");
    // if (inputStream != null) {
    // try {
    // props.load(inputStream);
    // }
    // catch (IOException e) {
    // // TODO Auto-generated catch block
    // e.printStackTrace();
    // }
    // }
    // // // // Configure the log service
    // getLogService().setLoggerName("WattDepot.AccessLog");
    // try {
    // LogManager.getLogManager().readConfiguration(inputStream);
    // }
    // catch (SecurityException e) {
    // // TODO Auto-generated catch block
    // e.printStackTrace();
    // }
    // catch (IOException e) {
    // // TODO Auto-generated catch block
    // e.printStackTrace();
    // }
    //
    // getLogService().setLogPropertiesRef("log.properties");

  }

}