com.sms.server.controller.UserController.java Source code

Introduction

Here is the source code for com.sms.server.controller.UserController.java

Source

/*
 * To change this license header, choose License Headers in Project Properties.
 * To change this template file, choose Tools | Templates
 * and open the template in the editor.
 */
package com.sms.server.controller;

import com.sms.server.dao.UserDao;
import com.sms.server.domain.User;
import com.sms.server.domain.UserStats;
import com.sms.server.domain.UserRole;
import com.sms.server.service.LogService;
import java.util.List;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
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.ResponseBody;
import org.springframework.web.bind.annotation.RestController;

/**
 *
 * @author scott2ware
 */

@RestController
@RequestMapping(value = "/user")
public class UserController {

    @Autowired
    private UserDao userDao;

    private static final String CLASS_NAME = "UserController";

    @RequestMapping(value = "/{username}", method = RequestMethod.PUT, headers = {
            "Content-type=application/json" })
    @ResponseBody
    public ResponseEntity<String> updateUser(@RequestBody User update, @PathVariable("username") String username) {
        User user = userDao.getUserByUsername(username);

        if (user == null) {
            return new ResponseEntity<String>("Username does not exist.", HttpStatus.BAD_REQUEST);
        }

        if (username.equals("admin")) {
            return new ResponseEntity<String>("You are not authenticated to perform this operation.",
                    HttpStatus.FORBIDDEN);
        }

        // Update user details
        if (update.getUsername() != null) {
            // Check username is available
            if (userDao.getUserByUsername(user.getUsername()) != null) {
                return new ResponseEntity<String>("Username already exists.", HttpStatus.NOT_ACCEPTABLE);
            } else {
                user.setUsername(update.getUsername());
            }
        }

        if (update.getPassword() != null) {
            user.setPassword(update.getPassword());
        }

        if (update.getEnabled() != null) {
            user.setEnabled(update.getEnabled());
        }

        // Update database
        if (!userDao.updateUser(user, username)) {
            LogService.getInstance().addLogEntry(LogService.Level.ERROR, CLASS_NAME,
                    "Error updating user '" + user.getUsername() + "'.", null);
            return new ResponseEntity<String>("Error updating user details.", HttpStatus.INTERNAL_SERVER_ERROR);
        }

        LogService.getInstance().addLogEntry(LogService.Level.INFO, CLASS_NAME,
                "User '" + user.getUsername() + "' updated successfully.", null);
        return new ResponseEntity<String>("User details updated successfully.", HttpStatus.ACCEPTED);
    }

    @RequestMapping(value = "/{username}/role", method = RequestMethod.GET)
    public ResponseEntity<List<UserRole>> getUserRoles(@PathVariable("username") String username) {
        List<UserRole> userRoles = userDao.getUserRolesByUsername(username);

        if (userRoles == null) {
            return new ResponseEntity<List<UserRole>>(HttpStatus.NOT_FOUND);
        }

        return new ResponseEntity<List<UserRole>>(userRoles, HttpStatus.OK);
    }

    @RequestMapping(value = "/{username}/stats", method = RequestMethod.GET)
    public ResponseEntity<UserStats> getUserStats(@PathVariable("username") String username) {
        UserStats userStats = userDao.getUserStatsByUsername(username);

        if (userStats == null) {
            return new ResponseEntity<UserStats>(HttpStatus.NOT_FOUND);
        }

        return new ResponseEntity<UserStats>(userStats, HttpStatus.OK);
    }
}