Fix #3569

kogito-serverless-workflow/kogito-serverless-workflow-builder/src/main/java/org/kie/kogito/serverless/workflow/io/CachedContentLoader.java

@@ -20,7 +20,6 @@
 
 import java.io.ByteArrayInputStream;
 import java.io.InputStream;
-import java.net.URI;
 import java.nio.file.Path;
 import java.util.Optional;
 
@@ -48,10 +47,10 @@ public synchronized byte[] readAllBytes() {
         }
     }
 
-    private final URI uri;
+    protected final String uri;
     private URIContentLoader[] fallbackContentLoaders;
 
-    protected CachedContentLoader(URI uri, URIContentLoader... fallbackContentLoaders) {
+    protected CachedContentLoader(String uri, URIContentLoader... fallbackContentLoaders) {
         this.uri = uri;
         this.fallbackContentLoaders = fallbackContentLoaders;
     }
@@ -91,10 +90,20 @@ public InputStream getInputStream() {
         }
     }
 
-    protected abstract byte[] loadURI(URI uri);
+    protected static String trimScheme(String uri, String scheme) {
+        String str = uri;
+        if (str.toLowerCase().startsWith(scheme)) {
+            str = str.substring(scheme.length());
+        }
+        return str;
+    }
 
-    @Override
-    public URI uri() {
-        return uri;
+    protected static String removeSlash(String str) {
+        while (str.startsWith("/")) {
+            str = str.substring(1);
+        }
+        return str;
     }
+
+    protected abstract byte[] loadURI();
 }

kogito-serverless-workflow/kogito-serverless-workflow-builder/src/main/java/org/kie/kogito/serverless/workflow/io/ClassPathContentLoader.java

@@ -21,11 +21,8 @@
 import java.io.IOException;
 import java.io.InputStream;
 import java.io.UncheckedIOException;
-import java.net.URI;
 import java.net.URISyntaxException;
 import java.net.URL;
-import java.net.URLDecoder;
-import java.nio.charset.Charset;
 import java.nio.file.Path;
 import java.util.Optional;
 
@@ -34,24 +31,12 @@ public class ClassPathContentLoader extends CachedContentLoader {
     private final Optional<URL> resource;
     private final String classpath;
 
-    ClassPathContentLoader(URI uri, Optional<ClassLoader> cl, URIContentLoader... fallbackContentLoaders) {
+    ClassPathContentLoader(String uri, Optional<ClassLoader> cl, URIContentLoader... fallbackContentLoaders) {
         super(uri, fallbackContentLoaders);
-        this.classpath = getPath(uri);
+        this.classpath = uriToPath(uri);
         this.resource = Optional.ofNullable(cl.orElse(Thread.currentThread().getContextClassLoader()).getResource(classpath));
     }
 
-    static String getPath(URI uri) {
-        final String classPathPrefix = "classpath:";
-        String str = URLDecoder.decode(uri.toString(), Charset.defaultCharset());
-        if (str.toLowerCase().startsWith(classPathPrefix)) {
-            str = str.substring(classPathPrefix.length());
-            while (str.startsWith("/")) {
-                str = str.substring(1);
-            }
-        }
-        return str;
-    }
-
     public Optional<URL> getResource() {
         return resource;
     }
@@ -74,7 +59,7 @@ private static Path fromURL(URL url) {
     }
 
     @Override
-    protected byte[] loadURI(URI uri) {
+    protected byte[] loadURI() {
         return resource.map(this::loadBytes).orElseThrow(() -> new IllegalArgumentException("cannot find classpath resource " + classpath));
     }
 
@@ -86,6 +71,10 @@ private byte[] loadBytes(URL r) {
         }
     }
 
+    static String uriToPath(String uri) {
+        return removeSlash(trimScheme(uri, URIContentLoaderType.CLASSPATH.scheme()));
+    }
+
     @Override
     public URIContentLoaderType type() {
         return URIContentLoaderType.CLASSPATH;

kogito-serverless-workflow/kogito-serverless-workflow-builder/src/main/java/org/kie/kogito/serverless/workflow/io/FileContentLoader.java

@@ -25,35 +25,51 @@
 import java.nio.file.Path;
 import java.util.Optional;
 
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
 public class FileContentLoader extends CachedContentLoader {
 
     private final Path path;
 
-    FileContentLoader(URI uri, URIContentLoader... fallbackContentLoaders) {
+    private static final Logger logger = LoggerFactory.getLogger(FileContentLoader.class);
+
+    FileContentLoader(String uri, URIContentLoader... fallbackContentLoaders) {
         super(uri, fallbackContentLoaders);
-        this.path = Path.of(getPath(uri));
+        this.path = obtainPath(uri);
     }
 
     @Override
     public URIContentLoaderType type() {
         return URIContentLoaderType.FILE;
     }
 
+    private static Path obtainPath(String uri) {
+        if (uri.startsWith(URIContentLoaderType.FILE.scheme())) {
+            try {
+                return Path.of(URI.create(uri));
+            } catch (Exception ex) {
+                logger.info("URI {} is not valid one according to Java, trying alternative approach", uri, ex);
+            }
+        }
+        return Path.of(uriToPath(uri));
+    }
+
     @Override
     protected Optional<Path> internalGetPath() {
         return Files.exists(path) ? Optional.of(path) : Optional.empty();
     }
 
     @Override
-    protected byte[] loadURI(URI uri) {
+    protected byte[] loadURI() {
         try {
             return Files.readAllBytes(path);
         } catch (IOException io) {
             throw new UncheckedIOException(io);
         }
     }
 
-    static String getPath(URI uri) {
-        return uri.getPath();
+    static String uriToPath(String uri) {
+        return trimScheme(uri, URIContentLoaderType.FILE.scheme());
     }
 }

kogito-serverless-workflow/kogito-serverless-workflow-builder/src/main/java/org/kie/kogito/serverless/workflow/io/HttpContentLoader.java

@@ -22,6 +22,7 @@
 import java.io.InputStream;
 import java.net.HttpURLConnection;
 import java.net.URI;
+import java.net.URL;
 import java.util.Base64;
 import java.util.Collection;
 import java.util.HashMap;
@@ -50,17 +51,19 @@ class HttpContentLoader extends CachedContentLoader {
 
     private Optional<Workflow> workflow;
     private String authRef;
+    private final URIContentLoaderType type;
 
-    HttpContentLoader(URI uri, Optional<Workflow> workflow, String authRef) {
+    HttpContentLoader(String uri, Optional<Workflow> workflow, String authRef, URIContentLoaderType type) {
         super(uri);
         this.workflow = workflow;
         this.authRef = authRef;
+        this.type = type;
     }
 
     @Override
-    protected byte[] loadURI(URI u) {
+    protected byte[] loadURI() {
         try {
-            HttpURLConnection conn = (HttpURLConnection) u.toURL().openConnection();
+            HttpURLConnection conn = (HttpURLConnection) new URL(uri).openConnection();
             // some http servers required specific accept header (*/* is specified for those we do not care about accept) 
             conn.setRequestProperty("Accept", "application/json,application/yaml,application/yml,application/text,text/*,*/*");
             workflow.map(Workflow::getAuth).map(Auth::getAuthDefs).stream().flatMap(Collection::stream)
@@ -74,7 +77,7 @@ protected byte[] loadURI(URI u) {
             } else {
                 try (InputStream is = conn.getErrorStream()) {
                     throw new IllegalArgumentException(String.format(
-                            "Failed to fetch remote file: %s. Status code is %d and response: %n %s", u, code, is == null ? "" : new String(is.readAllBytes())));
+                            "Failed to fetch remote file: %s. Status code is %d and response: %n %s", uri, code, is == null ? "" : new String(is.readAllBytes())));
                 }
             }
         } catch (IOException io) {
@@ -147,6 +150,11 @@ private String encode(String str) {
 
     @Override
     public URIContentLoaderType type() {
-        return URIContentLoaderType.HTTP;
+        return type;
     }
+
+    static String uriToPath(String uri) {
+        return URI.create(uri).getPath();
+    }
+
 }

kogito-serverless-workflow/kogito-serverless-workflow-builder/src/main/java/org/kie/kogito/serverless/workflow/io/ResourceCache.java

@@ -18,9 +18,8 @@
  */
 package org.kie.kogito.serverless.workflow.io;
 
-import java.net.URI;
-import java.util.function.Function;
+import java.util.function.Supplier;
 
 public interface ResourceCache {
-    byte[] get(URI uri, Function<URI, byte[]> retrieveCall);
+    byte[] get(String uri, Supplier<byte[]> retrieveCall);
 }

kogito-serverless-workflow/kogito-serverless-workflow-builder/src/main/java/org/kie/kogito/serverless/workflow/io/ResourceCacheFactory.java

@@ -18,12 +18,11 @@
  */
 package org.kie.kogito.serverless.workflow.io;
 
-import java.net.URI;
 import java.util.Collections;
 import java.util.Map;
 import java.util.WeakHashMap;
 import java.util.concurrent.atomic.AtomicReference;
-import java.util.function.Function;
+import java.util.function.Supplier;
 
 public class ResourceCacheFactory {
     private static final AtomicReference<ResourceCache> cache = new AtomicReference<>(new LocalResourceCache());
@@ -33,11 +32,11 @@ public static ResourceCache getCache() {
     }
 
     private static class LocalResourceCache implements ResourceCache {
-        private final Map<URI, byte[]> map = Collections.synchronizedMap(new WeakHashMap<>());
+        private final Map<String, byte[]> map = Collections.synchronizedMap(new WeakHashMap<>());
 
         @Override
-        public byte[] get(URI uri, Function<URI, byte[]> retrieveCall) {
-            return map.computeIfAbsent(uri, retrieveCall);
+        public byte[] get(String uri, Supplier<byte[]> retrieveCall) {
+            return map.computeIfAbsent(uri, u -> retrieveCall.get());
         }
 
     }

kogito-serverless-workflow/kogito-serverless-workflow-builder/src/main/java/org/kie/kogito/serverless/workflow/io/URIContentLoader.java

@@ -19,14 +19,11 @@
 package org.kie.kogito.serverless.workflow.io;
 
 import java.io.InputStream;
-import java.net.URI;
 import java.nio.file.Path;
 import java.util.Optional;
 
 public interface URIContentLoader {
 
-    URI uri();
-
     InputStream getInputStream();
 
     URIContentLoaderType type();