com.ctb.service.task.AuthorizationService.java Source code

Introduction

Here is the source code for com.ctb.service.task.AuthorizationService.java

Source

/*******************************************************************************
 * Copyright (c) 2005, 2014 springside.github.io
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 *******************************************************************************/
package com.ctb.service.task;

import java.util.Collections;
import java.util.Set;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import org.springframework.transaction.annotation.Transactional;

import com.ctb.entity.Authorization;
import com.ctb.entity.User;
import com.ctb.repository.AuthorizationDao;
import com.ctb.service.account.AccountService;

// Spring Bean.
@Component
// public?.
@Transactional
public class AuthorizationService {

    @Autowired
    private AuthorizationDao authorizationDao;
    @Autowired
    private AccountService accountService;
    @Autowired
    private AppService appService;
    @Autowired
    private RoleService roleService;

    public Authorization createAuthorization(Authorization authorization) {
        return merge(authorization);
    }

    public Authorization updateAuthorization(Authorization authorization) {
        return merge(authorization);
    }

    public Authorization merge(Authorization authorization) {
        Authorization dbAuthorization = authorizationDao.findByAppIdAndUserId(authorization.getAppId(),
                authorization.getUserId());
        if (dbAuthorization == null) {// ?? 
            return authorizationDao.save(authorization);
        }

        if (dbAuthorization.equals(authorization)) {// ????
            return authorizationDao.save(authorization);
        }

        for (Long roleId : authorization.getRoleIds()) {// ??
            if (!dbAuthorization.getRoleIds().contains(roleId)) {
                dbAuthorization.getRoleIds().add(roleId);
            }
        }

        if (dbAuthorization.getRoleIds().isEmpty()) {//  ??
            authorizationDao.delete(dbAuthorization.getId());
            return dbAuthorization;
        }
        // ?
        return authorizationDao.save(dbAuthorization);
    }

    public void deleteAuthorization(Long authorizationId) {
        authorizationDao.delete(authorizationId);
    }

    public Authorization findOne(Long authorizationId) {
        return authorizationDao.findOne(authorizationId);
    }

    public Iterable<Authorization> findAll() {
        return authorizationDao.findAll();
    }

    /**
     * ???
     * 
     * @param username
     * @return
     */
    @SuppressWarnings("unchecked")
    public Set<String> findRoles(String appKey, String username) {
        User user = accountService.findUserByLoginName(username);
        if (user == null) {
            return Collections.EMPTY_SET;
        }
        Long appId = appService.findAppIdByAppKey(appKey);
        if (appId == null) {
            return Collections.EMPTY_SET;
        }
        Authorization authorization = authorizationDao.findByAppIdAndUserId(appId, user.getId());
        if (authorization == null) {
            return Collections.EMPTY_SET;
        }
        return roleService.findRoles(authorization.getRoleIds().toArray(new Long[0]));
    }

    /**
     * ?????
     * 
     * @param username
     * @return
     */
    @SuppressWarnings("unchecked")
    public Set<String> findPermissions(String appKey, String username) {
        User user = accountService.findUserByLoginName(username);
        if (user == null) {
            return Collections.EMPTY_SET;
        }
        Long appId = appService.findAppIdByAppKey(appKey);
        if (appId == null) {
            return Collections.EMPTY_SET;
        }
        Authorization authorization = authorizationDao.findByAppIdAndUserId(appId, user.getId());
        if (authorization == null) {
            return Collections.EMPTY_SET;
        }
        return roleService.findPermissions(authorization.getRoleIds().toArray(new Long[0]));
    }
}