Source/BladeX-Tool/blade-core-secure/src/main/java/org/springblade/core/secure/registry/SecureRegistry.java

¶Ô±ÈÐÂÎļþ
@@ -0,0 +1,196 @@
/*
 *      Copyright (c) 2018-2028, Chill Zhuang All rights reserved.
 *
 *  Redistribution and use in source and binary forms, with or without
 *  modification, are permitted provided that the following conditions are met:
 *
 *  Redistributions of source code must retain the above copyright notice,
 *  this list of conditions and the following disclaimer.
 *  Redistributions in binary form must reproduce the above copyright
 *  notice, this list of conditions and the following disclaimer in the
 *  documentation and/or other materials provided with the distribution.
 *  Neither the name of the dreamlu.net developer nor the names of its
 *  contributors may be used to endorse or promote products derived from
 *  this software without specific prior written permission.
 *  Author: Chill åº„骞 (smallchill@163.com)
 */
package org.springblade.core.secure.registry;

import lombok.Data;
import org.springblade.core.secure.props.AuthSecure;
import org.springblade.core.secure.props.BasicSecure;
import org.springblade.core.secure.props.SignSecure;
import org.springblade.core.secure.provider.HttpMethod;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * å®‰å…¨æ¡†æž¶ç»Ÿä¸€é…ç½®
 *
 * @author Chill
 */
@Data
public class SecureRegistry {

   /**
    * æ˜¯å¦å¼€å¯é‰´æƒ
    */
   private boolean enabled = false;

   /**
    * æ˜¯å¦å¼€å¯æŽˆæƒ
    */
   private boolean authEnabled = true;

   /**
    * æ˜¯å¦å¼€å¯åŸºç¡€è®¤è¯
    */
   private boolean basicEnabled = true;

   /**
    * æ˜¯å¦å¼€å¯ç­¾åè®¤è¯
    */
   private boolean signEnabled = true;

   /**
    * æ˜¯å¦å¼€å¯å®¢æˆ·ç«¯è®¤è¯
    */
   private boolean clientEnabled = true;

   /**
    * é»˜è®¤æ”¾è¡Œè§„则
    */
   private final List<String> defaultExcludePatterns = new ArrayList<>();

   /**
    * è‡ªå®šä¹‰æ”¾è¡Œè§„则
    */
   private final List<String> excludePatterns = new ArrayList<>();

   /**
    * è‡ªå®šä¹‰æŽˆæƒé›†åˆ
    */
   private final List<AuthSecure> authSecures = new ArrayList<>();

   /**
    * åŸºç¡€è®¤è¯é›†åˆ
    */
   private final List<BasicSecure> basicSecures = new ArrayList<>();

   /**
    * ç­¾åè®¤è¯é›†åˆ
    */
   private final List<SignSecure> signSecures = new ArrayList<>();

   public SecureRegistry() {
      this.defaultExcludePatterns.add("/actuator/health/**");
      this.defaultExcludePatterns.add("/v2/api-docs/**");
      this.defaultExcludePatterns.add("/auth/**");
      this.defaultExcludePatterns.add("/token/**");
      this.defaultExcludePatterns.add("/log/**");
      this.defaultExcludePatterns.add("/menu/routes");
      this.defaultExcludePatterns.add("/menu/auth-routes");
      this.defaultExcludePatterns.add("/menu/top-menu");
      this.defaultExcludePatterns.add("/process/resource-view");
      this.defaultExcludePatterns.add("/process/diagram-view");
      this.defaultExcludePatterns.add("/manager/check-upload");
      this.defaultExcludePatterns.add("/error/**");
      this.defaultExcludePatterns.add("/assets/**");
   }

   /**
    * è®¾ç½®å•ä¸ªæ”¾è¡Œapi
    */
   public SecureRegistry excludePathPattern(String pattern) {
      this.excludePatterns.add(pattern);
      return this;
   }

   /**
    * è®¾ç½®æ”¾è¡Œapi集合
    */
   public SecureRegistry excludePathPatterns(String... patterns) {
      this.excludePatterns.addAll(Arrays.asList(patterns));
      return this;
   }

   /**
    * è®¾ç½®æ”¾è¡Œapi集合
    */
   public SecureRegistry excludePathPatterns(List<String> patterns) {
      this.excludePatterns.addAll(patterns);
      return this;
   }

   /**
    * è®¾ç½®å•ä¸ªè‡ªå®šä¹‰æŽˆæƒ
    */
   public SecureRegistry addAuthPattern(HttpMethod method, String pattern, String expression) {
      this.authSecures.add(new AuthSecure(method, pattern, expression));
      return this;
   }

   /**
    * è®¾ç½®è‡ªå®šä¹‰æŽˆæƒé›†åˆ
    */
   public SecureRegistry addAuthPatterns(List<AuthSecure> authSecures) {
      this.authSecures.addAll(authSecures);
      return this;
   }

   /**
    * è¿”回自定义授权集合
    */
   public List<AuthSecure> getAuthSecures() {
      return this.authSecures;
   }

   /**
    * è®¾ç½®åŸºç¡€è®¤è¯
    */
   public SecureRegistry addBasicPattern(HttpMethod method, String pattern, String username, String password) {
      this.basicSecures.add(new BasicSecure(method, pattern, username, password));
      return this;
   }

   /**
    * è®¾ç½®åŸºç¡€è®¤è¯é›†åˆ
    */
   public SecureRegistry addBasicPatterns(List<BasicSecure> basicSecures) {
      this.basicSecures.addAll(basicSecures);
      return this;
   }

   /**
    * è¿”回基础认证集合
    */
   public List<BasicSecure> getBasicSecures() {
      return this.basicSecures;
   }

   /**
    * è®¾ç½®ç­¾åè®¤è¯
    */
   public SecureRegistry addSignPattern(HttpMethod method, String pattern, String crypto) {
      this.signSecures.add(new SignSecure(method, pattern, crypto));
      return this;
   }

   /**
    * è®¾ç½®ç­¾åè®¤è¯é›†åˆ
    */
   public SecureRegistry addSignPatterns(List<SignSecure> signSecures) {
      this.signSecures.addAll(signSecures);
      return this;
   }

   /**
    * è¿”回签名认证集合
    */
   public List<SignSecure> getSignSecures() {
      return this.signSecures;
   }

}