com.restfeel.controller.rest.EntitySessionController.java Source code

Introduction

Here is the source code for com.restfeel.controller.rest.EntitySessionController.java

Source

/*
 * Copyright 2015 Ranjan Kumar
 *
 * Licensed under the Apache License, Version 2.0 (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.apache.org/licenses/LICENSE-2.0
 *
 * 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.restfeel.controller.rest;

import java.util.Map;

import org.bson.types.ObjectId;
import org.json.JSONObject;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.mongodb.core.MongoTemplate;
import org.springframework.transaction.annotation.Propagation;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.bind.annotation.RestController;

import com.mongodb.BasicDBObject;
import com.mongodb.DBCollection;
import com.mongodb.DBObject;
import com.mongodb.DBRef;
import com.restfeel.service.auth.EntityAuthService;

@RestController
@Transactional(propagation = Propagation.REQUIRES_NEW)
public class EntitySessionController {
    Logger logger = LoggerFactory.getLogger(EntitySessionController.class);

    @Autowired
    private MongoTemplate mongoTemplate;

    @Autowired
    private EntityAuthService authService;

    @RequestMapping(value = "/api/{projectId}/entities/logout", method = RequestMethod.GET, headers = "Accept=application/json")
    public @ResponseBody String getEntityDataById(@PathVariable("projectId") String projectId,
            @RequestParam(value = "authToken", required = true) String llt) {

        boolean status = authService.logout(llt);

        JSONObject response = new JSONObject();

        if (status) {
            response.put("msg", "sucessfully logged out");
        } else {
            response.put("msg", "session expired");
        }

        return response.toString(4);
    }

    @RequestMapping(value = "/api/{projectId}/entities/login", method = RequestMethod.POST, headers = "Accept=application/json", consumes = "application/json")
    public @ResponseBody String createEntityData(@PathVariable("projectId") String projectId,
            @RequestBody Object userDTO) {
        DBObject data;
        if (!(userDTO instanceof Map)) {
            return null;
        }

        JSONObject response = new JSONObject();

        Map map = (Map) userDTO;
        JSONObject jsonObj = new JSONObject(map);
        try {
            data = authService.authenticate(jsonObj, projectId);
        } catch (Exception e) {
            response.put("msg", e.getMessage());
            return response.toString(4);
        }
        DBCollection userCollection = mongoTemplate.getCollection(projectId + "_User");

        DBObject user = userCollection.findOne(((DBRef) (data.get("user"))).getId());
        user.removeField("password");
        data.put("user", user);

        dbRefToRelation(data);
        data.put("authToken", data.get("_id"));
        data.removeField("_id");
        data.removeField("expireAt");
        String json = data.toString();

        // Indentation
        response = new JSONObject(json);
        return response.toString(4);
    }

    private void dbRefToRelation(DBObject dbObject) {
        if (dbObject == null) {
            return;
        }
        if (dbObject.containsField("_id"))
            dbObject.put("_id", ((ObjectId) dbObject.get("_id")).toHexString());
        for (String key : dbObject.keySet()) {
            Object obj = dbObject.get(key);
            if (obj instanceof DBRef) {
                DBRef ref = (DBRef) obj;
                dbObject.put(key, dbRefToRel(ref));
            } else if (obj instanceof DBObject) {
                dbRefToRelation((DBObject) obj);
            }
        }

    }

    private DBObject dbRefToRel(DBRef obj) {
        return new BasicDBObject().append("_rel",
                new BasicDBObject().append("entity", ((String) obj.getId()).split("_")[1]).append("_id",
                        ((ObjectId) obj.getId()).toHexString()));
    }

}