com.baidu.cc.common.SysUtils.java Source code

Introduction

Here is the source code for com.baidu.cc.common.SysUtils.java

Source

/*
 * Copyright 2014 the original author or authors.
 *
 * 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.baidu.cc.common;

import com.baidu.cc.configuration.bo.User;
import com.baidu.rigel.platform.util.Security;
import com.baidu.uc.util.CookieHelper;

import java.util.ArrayList;
import java.util.List;
import java.util.UUID;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import org.apache.commons.lang.StringUtils;
import org.apache.commons.lang.math.NumberUtils;

/**
 * 
 *  Created by zhenhui on 14-2-27.
 * 
 * @author zhenhui
 */
public class SysUtils {
    /** Character encode string. */
    public static final String UTF_8 = "utf-8";

    /** ???uidname?. */
    public static final String COOKIE_MD5_TOKEN = "configcenter_cookie_token_for_md5";

    /** ??. */
    public static final String AUTHCHECK_DES_KEY = "configcenter_authcheck_key_for_des";

    /** cookiepath. */
    public static final String COOKIE_PATH = "/";

    /** cookiemaxage. */
    public static final int COOKIE_MAXAGE = 1800;

    /** cookieuidkey??. */
    public static final String COOKIE_KEY_UID = "rcc_uid";

    /** cookienamekey??. */
    public static final String COOKIE_KEY_NAME = "rcc_name";

    /** cookietokenkey??. */
    public static final String COOKIE_KEY_TOKEN = "rcc_token";

    /**
     * access_setting.type 
     */
    public static final int ACCESS_SETTING_TYPE_PROJECT = 1;

    /**
     * access_setting.type 
     */
    public static final int ACCESS_SETTING_TYPE_ENV = 2;

    /** ??ID=1???. */
    public static final Long ADMIN_USERID = 1L;

    /** OP??ID=2??. */
    public static final Long OPADMIN_USERID = 2L;

    /**
     * UUID.
     * 
     * @return UUID 36?
     */
    public static String uuid() {
        UUID uuid = UUID.randomUUID();
        return uuid.toString();
    }

    /**
     * ?cookietoken.
     * 
     * @param userId
     *            userId
     * @param name
     *            name
     * @return MD5(COOKIE_MD5_TOKEN + userId + name)
     */
    public static String genCookieToken(long userId, String name) {
        return Security.MD5Encode(COOKIE_MD5_TOKEN + userId + name);
    }

    /**
     * ?cookie?.
     * 
     * @param request
     *            HttpServletRequest
     * @param response
     *            HttpServletResponse
     * @param user
     *            User
     */
    public static void addLoginCookie(HttpServletRequest request, HttpServletResponse response, User user) {
        CookieHelper cookieHelper = new CookieHelper(request, response);

        cookieHelper.setCookie(COOKIE_KEY_UID, user.getId().toString(), COOKIE_MAXAGE, COOKIE_PATH, null);
        cookieHelper.setCookie(COOKIE_KEY_NAME, user.getName(), COOKIE_MAXAGE, COOKIE_PATH, null);
        cookieHelper.setCookie(COOKIE_KEY_TOKEN, SysUtils.genCookieToken(user.getId(), user.getName()),
                COOKIE_MAXAGE, COOKIE_PATH, null);

    }

    /**
     * ?cookie?.
     * 
     * @param request
     *            HttpServletRequest
     * @param response
     *            HttpServletResponse
     */
    public static void deleteLoginCookie(HttpServletRequest request, HttpServletResponse response) {

        CookieHelper cookieHelper = new CookieHelper(request, response);
        cookieHelper.setCookie(COOKIE_KEY_UID, "", 0, COOKIE_PATH, null);
        cookieHelper.setCookie(COOKIE_KEY_NAME, "", 0, COOKIE_PATH, null);
        cookieHelper.setCookie(COOKIE_KEY_TOKEN, "", 0, COOKIE_PATH, null);
    }

    /**
     * ???.
     * 
     * @param userId
     *            ID
     * @param projectId
     *            ID
     * @return DES?
     */
    public static String encryptAuthCheck(Long userId, Long projectId) {
        return Security.DESEncrypt("userId=" + userId + ";projectId=" + projectId, AUTHCHECK_DES_KEY);
    }

    /**
     * ???.
     * 
     * @param request
     *            HttpServletRequest
     * @param projectId
     *            ID
     * @return DES?
     */
    public static String encryptAuthCheck(HttpServletRequest request, Long projectId) {
        return Security.DESEncrypt("userId=" + getUserIdFromCookie(request) + ";projectId=" + projectId,
                AUTHCHECK_DES_KEY);
    }

    /**
     * authCheck??userId.
     * 
     * @param authCheck
     *            authCheck
     * @return userId
     */
    public static Long getUserIdFromAuthcheck(String authCheck) {
        String value = Security.DESDecrypt(authCheck, AUTHCHECK_DES_KEY);
        return NumberUtils.toLong(StringUtils.substringBetween(value, "userId=", ";projectId"));
    }

    /**
     * cookie?userId.
     * 
     * @param request
     *            HttpServletRequest
     * @return userId
     */
    public static Long getUserIdFromCookie(HttpServletRequest request) {
        return NumberUtils.toLong(new CookieHelper(request, null).getCookieVal(COOKIE_KEY_UID));
    }

    /**
     * authCheck??projectId.
     * 
     * @param authCheck
     *            authCheck
     * @return userId
     */
    public static Long getProjectIdFromAuthcheck(String authCheck) {
        String value = Security.DESDecrypt(authCheck, AUTHCHECK_DES_KEY);
        return NumberUtils.toLong(StringUtils.substringAfter(value, "projectId="));
    }

    /**
     * ?ID??List<Long>.
     * 
     * @param ids
     *            the ids
     * @return the list
     * @returnList<Long>
     */
    public static List<Long> split2Ids(String ids) {

        ArrayList<Long> result = new ArrayList<Long>();
        if (StringUtils.isEmpty(ids)) {
            return result;
        }

        String[] idsArray = StringUtils.split(ids, ",");

        for (String id : idsArray) {
            result.add(NumberUtils.toLong(id));
        }
        return result;
    }
}