package org.apidesign.runtime.check;

 import java.io.IOException;

 import java.io.InputStream;

 import java.util.Map;

 final class VersionAwareClassLoader extends ClassLoader 

 implements RuntimeCheck.AwareLoader {

     private final Map<String,String> requestedVersions;

     public VersionAwareClassLoader(ClassLoader parent, Map<String,String> requestedVersions) {

         super(parent);

         this.requestedVersions = requestedVersions;

     }

     @Override

     protected synchronized Class<?> loadClass(String name, boolean resolve) throws ClassNotFoundException {

         if (name.endsWith("Test") && name.startsWith("api.")) {

             try {

                 InputStream is = getResourceAsStream(name.replace('.', '/').concat(".class"));

                 byte[] arr = new byte[is.available()];

                 int read = is.read(arr);

                 assert read >= 0;

                 is.close();

                 return defineClass(name, arr, 0, read);

             } catch (IOException ex) {

                 throw new ClassNotFoundException("Cannot load " + name, ex);

             }

         }

         return super.loadClass(name, resolve);

     }

     public String requestedVersion(String apiName) {

         return requestedVersions.get(apiName);

     }

 }