org.apache.shiro.spring.boot.utils.NimbusdsUtils Maven / Gradle / Ivy

Go to download
/*
 * Copyright (c) 2018, vindell (https://github.com/vindell).
 *
 * 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 org.apache.shiro.spring.boot.utils;

import java.text.ParseException;
import java.util.Date;
import java.util.Iterator;
import java.util.Map;
import java.util.Map.Entry;

import org.apache.shiro.biz.utils.StringUtils;
import org.apache.shiro.spring.boot.jwt.JwtPayload;
import org.apache.shiro.util.CollectionUtils;

import com.google.common.collect.Lists;
import com.nimbusds.jwt.JWTClaimsSet;

/**
 * 基于Nimbusds组件的jwt工具对象
 * @author ： vindell
 */
public class NimbusdsUtils {

	public static JWTClaimsSet claimsSet(String jwtId, String subject, String issuer, String audience, Map claims,
			long period) {
		
		// Current TimeMillis
		long currentTimeMillis = System.currentTimeMillis();

		// Prepare JWT with claims set
		JWTClaimsSet.Builder builder = new JWTClaimsSet.Builder();

		// Jwt主键ID
		if (StringUtils.hasText(jwtId)) {
			builder.jwtID(jwtId);
		}
		// 用户名主题
		builder.subject(subject);
		// 接收对象
		if (StringUtils.hasText(audience)) {
			builder.audience(Lists.newArrayList(StringUtils.tokenizeToStringArray(audience)));
		}
		// 签发者
		if (StringUtils.hasText(issuer)) {
			builder.issuer(issuer);
		}
		// 声明信息
		if(!CollectionUtils.isEmpty(claims)) {
			Iterator> ite = claims.entrySet().iterator();
			while (ite.hasNext()) {
				Entry entry = ite.next();
				builder.claim(entry.getKey(), entry.getValue());
			}
		}
		
		// 签发时间
		builder.issueTime(new Date(currentTimeMillis));
		builder.notBeforeTime(new Date(currentTimeMillis));
		if (period >= 0) {
			// 有效时间
			Date expiration = new Date(currentTimeMillis + period);
			builder.expirationTime(expiration);
		}

		return builder.build();
	}
	
	
	public static JWTClaimsSet claimsSet(String jwtId, String subject, String issuer, String audience, String roles,
			String permissions, long period) {

		// Current TimeMillis
		long currentTimeMillis = System.currentTimeMillis();

		// Prepare JWT with claims set
		JWTClaimsSet.Builder builder = new JWTClaimsSet.Builder();

		// Jwt主键ID
		if (StringUtils.hasText(jwtId)) {
			builder.jwtID(jwtId);
		}
		// 用户名主题
		builder.subject(subject);
		// 接收对象
		if (StringUtils.hasText(audience)) {
			builder.audience(Lists.newArrayList(StringUtils.tokenizeToStringArray(audience)));
		}
		// 签发者
		if (StringUtils.hasText(issuer)) {
			builder.issuer(issuer);
		}
		// 角色
		if (StringUtils.hasText(roles)) {
			builder.claim("roles", roles);
		}
		// 权限
		if (StringUtils.hasText(permissions)) {
			builder.claim("perms", permissions);
		}
		// 签发时间
		builder.issueTime(new Date(currentTimeMillis));
		builder.notBeforeTime(new Date(currentTimeMillis));
		if (period >= 0) {
			// 有效时间
			Date expiration = new Date(currentTimeMillis + period);
			builder.expirationTime(expiration);
		}
		return builder.build();
	}

	public static JwtPayload payload(JWTClaimsSet jwtClaims) throws ParseException {

		JwtPayload payload = new JwtPayload();
		payload.setTokenId(jwtClaims.getJWTID());
		payload.setClientId(jwtClaims.getSubject());// 用户名
		payload.setIssuer(jwtClaims.getIssuer());// 签发者
		payload.setIssuedAt(jwtClaims.getIssueTime());// 签发时间
		payload.setExpiration(jwtClaims.getExpirationTime()); // 过期时间
		payload.setNotBefore(jwtClaims.getNotBeforeTime());
		payload.setAudience(jwtClaims.getAudience());// 接收方
		payload.setClaims(jwtClaims.getClaims()); // 访问主张
		
		return payload;
	}

}