[secure-mode] Refactor secure mode plugin (#2058)

Author: frankfliu

jacoco/build.gradle.kts

@@ -16,6 +16,7 @@ dependencies {
     jacocoAggregation(project(":plugins:kserve"))
     // jacocoAggregation(project(":plugins:management-console"))
     jacocoAggregation(project(":plugins:plugin-management-plugin"))
+    jacocoAggregation(project(":plugins:secure-mode"))
     jacocoAggregation(project(":plugins:static-file-plugin"))
     jacocoAggregation(project(":prometheus"))
     jacocoAggregation(project(":serving"))

plugins/secure-mode/src/main/java/ai/djl/serving/plugins/securemode/SecureModeModelServerListener.java

@@ -12,30 +12,21 @@
  */
 package ai.djl.serving.plugins.securemode;
 
-import ai.djl.Device;
-import ai.djl.ModelException;
+import ai.djl.serving.http.IllegalConfigurationException;
 import ai.djl.serving.wlm.ModelInfo;
 import ai.djl.serving.wlm.util.ModelServerListenerAdapter;
 
 import java.io.IOException;
-import java.net.URISyntaxException;
 
 class SecureModeModelServerListener extends ModelServerListenerAdapter {
 
+    /** {@inheritDoc} */
     @Override
-    public void onModelLoading(ModelInfo<?, ?> model, Device device) {
-        super.onModelLoading(model, device);
-
-        if (SecureModeUtils.isSecureMode()) {
-            try {
-                SecureModeUtils.validateSecurity();
-                SecureModeUtils.reconcileSources(model.getModelUrl());
-            } catch (ModelException e) {
-                throw new IllegalConfigurationException("Secure Mode check failed", e);
-            } catch (IOException | URISyntaxException e) {
-                throw new IllegalConfigurationException(
-                        "Error while running Secure Mode checks", e);
-            }
+    public void onModelConfigured(ModelInfo<?, ?> model) {
+        try {
+            SecureModeUtils.validateSecurity(model);
+        } catch (IOException e) {
+            throw new IllegalConfigurationException("Error while running Secure Mode checks", e);
         }
     }
 }