loader.rs

use alloc::boxed::Box;

use jvm::{BootstrapClassLoader, ClassDefinition, Jvm, Result};

use crate::{RT_RUSTJAR, Runtime, RuntimeClassProto};

pub fn get_runtime_class_proto(name: &str) -> Option<RuntimeClassProto> {
    let protos = [
        crate::classes::java::io::BufferedReader::as_proto(),
        crate::classes::java::io::ByteArrayInputStream::as_proto(),
        crate::classes::java::io::ByteArrayOutputStream::as_proto(),
        crate::classes::java::io::DataInput::as_proto(),
        crate::classes::java::io::DataInputStream::as_proto(),
        crate::classes::java::io::DataOutput::as_proto(),
        crate::classes::java::io::DataOutputStream::as_proto(),
        crate::classes::java::io::EOFException::as_proto(),
        crate::classes::java::io::File::as_proto(),
        crate::classes::java::io::FileDescriptor::as_proto(),
        crate::classes::java::io::FileInputStream::as_proto(),
        crate::classes::java::io::FileNotFoundException::as_proto(),
        crate::classes::java::io::FileOutputStream::as_proto(),
        crate::classes::java::io::FilterInputStream::as_proto(),
        crate::classes::java::io::FilterOutputStream::as_proto(),
        crate::classes::java::io::InputStream::as_proto(),
        crate::classes::java::io::InputStreamReader::as_proto(),
        crate::classes::java::io::IOException::as_proto(),
        crate::classes::java::io::OutputStream::as_proto(),
        crate::classes::java::io::PrintStream::as_proto(),
        crate::classes::java::io::PrintWriter::as_proto(),
        crate::classes::java::io::RandomAccessFile::as_proto(),
        crate::classes::java::io::Reader::as_proto(),
        crate::classes::java::io::Serializable::as_proto(),
        crate::classes::java::io::StringWriter::as_proto(),
        crate::classes::java::io::Writer::as_proto(),
        crate::classes::java::lang::AbstractMethodError::as_proto(),
        crate::classes::java::lang::ArithmeticException::as_proto(),
        crate::classes::java::lang::ArrayIndexOutOfBoundsException::as_proto(),
        crate::classes::java::lang::ArrayStoreException::as_proto(),
        crate::classes::java::lang::Class::as_proto(),
        crate::classes::java::lang::ClassCastException::as_proto(),
        crate::classes::java::lang::ClassLoader::as_proto(),
        crate::classes::java::lang::Cloneable::as_proto(),
        crate::classes::java::lang::CloneNotSupportedException::as_proto(),
        crate::classes::java::lang::Comparable::as_proto(),
        crate::classes::java::lang::Error::as_proto(),
        crate::classes::java::lang::Exception::as_proto(),
        crate::classes::java::lang::ExceptionInInitializerError::as_proto(),
        crate::classes::java::lang::IllegalArgumentException::as_proto(),
        crate::classes::java::lang::InstantiationError::as_proto(),
        crate::classes::java::lang::IncompatibleClassChangeError::as_proto(),
        crate::classes::java::lang::IndexOutOfBoundsException::as_proto(),
        crate::classes::java::lang::Integer::as_proto(),
        crate::classes::java::lang::InterruptedException::as_proto(),
        crate::classes::java::lang::LinkageError::as_proto(),
        crate::classes::java::lang::Math::as_proto(),
        crate::classes::java::lang::NegativeArraySizeException::as_proto(),
        crate::classes::java::lang::NoClassDefFoundError::as_proto(),
        crate::classes::java::lang::NoSuchFieldError::as_proto(),
        crate::classes::java::lang::NoSuchMethodError::as_proto(),
        crate::classes::java::lang::NullPointerException::as_proto(),
        crate::classes::java::lang::NumberFormatException::as_proto(),
        crate::classes::java::lang::Object::as_proto(),
        crate::classes::java::lang::Runnable::as_proto(),
        crate::classes::java::lang::Runtime::as_proto(),
        crate::classes::java::lang::RuntimeException::as_proto(),
        crate::classes::java::lang::SecurityException::as_proto(),
        crate::classes::java::lang::String::as_proto(),
        crate::classes::java::lang::StringBuffer::as_proto(),
        crate::classes::java::lang::StringIndexOutOfBoundsException::as_proto(),
        crate::classes::java::lang::System::as_proto(),
        crate::classes::java::lang::Thread::as_proto(),
        crate::classes::java::lang::Throwable::as_proto(),
        crate::classes::java::lang::UnsupportedOperationException::as_proto(),
        crate::classes::java::net::JarURLConnection::as_proto(),
        crate::classes::java::net::MalformedURLException::as_proto(),
        crate::classes::java::net::UnknownServiceException::as_proto(),
        crate::classes::java::net::URL::as_proto(),
        crate::classes::java::net::URLClassLoader::as_proto(),
        crate::classes::java::net::URLConnection::as_proto(),
        crate::classes::java::net::URLStreamHandler::as_proto(),
        crate::classes::java::util::AbstractCollection::as_proto(),
        crate::classes::java::util::AbstractList::as_proto(),
        crate::classes::java::util::Calendar::as_proto(),
        crate::classes::java::util::Date::as_proto(),
        crate::classes::java::util::Dictionary::as_proto(),
        crate::classes::java::util::EmptyStackException::as_proto(),
        crate::classes::java::util::Enumeration::as_proto(),
        crate::classes::java::util::GregorianCalendar::as_proto(),
        crate::classes::java::util::Hashtable::as_proto(),
        crate::classes::java::util::HashtableEntry::as_proto(),
        crate::classes::java::util::NoSuchElementException::as_proto(),
        crate::classes::java::util::Properties::as_proto(),
        crate::classes::java::util::Random::as_proto(),
        crate::classes::java::util::SimpleTimeZone::as_proto(),
        crate::classes::java::util::Stack::as_proto(),
        crate::classes::java::util::Timer::as_proto(),
        crate::classes::java::util::TimerTask::as_proto(),
        crate::classes::java::util::TimerThread::as_proto(),
        crate::classes::java::util::TimeZone::as_proto(),
        crate::classes::java::util::Vector::as_proto(),
        crate::classes::java::util::jar::Attributes::as_proto(),
        crate::classes::java::util::jar::JarEntry::as_proto(),
        crate::classes::java::util::jar::JarFile::as_proto(),
        crate::classes::java::util::jar::JarFileEntries::as_proto(),
        crate::classes::java::util::jar::Manifest::as_proto(),
        crate::classes::java::util::zip::ZipEntry::as_proto(),
        crate::classes::java::util::zip::ZipFile::as_proto(),
        crate::classes::java::util::zip::ZipFileEntries::as_proto(),
        crate::classes::org::rustjava::net::FileURLConnection::as_proto(),
        crate::classes::org::rustjava::net::FileURLHandler::as_proto(),
        crate::classes::org::rustjava::net::JarURLConnection::as_proto(),
        crate::classes::org::rustjava::net::JarURLHandler::as_proto(),
    ];

    protos.into_iter().find(|proto| proto.name == name)
}

struct JavaRuntimeClassLoader {
    runtime: Box<dyn Runtime>,
}

#[async_trait::async_trait]
impl BootstrapClassLoader for JavaRuntimeClassLoader {
    async fn load_class(&self, jvm: &Jvm, name: &str) -> Result<Option<Box<dyn ClassDefinition>>> {
        if let Some(element_type_name) = name.strip_prefix('[') {
            return Ok(Some(self.runtime.define_array_class(jvm, element_type_name).await?));
        }

        self.runtime.find_rustjar_class(jvm, RT_RUSTJAR, name).await
    }
}

pub fn get_bootstrap_class_loader(runtime: Box<dyn Runtime>) -> impl BootstrapClassLoader {
    JavaRuntimeClassLoader { runtime }
}