bindings/java/src/utility.rs

// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements.  See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership.  The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License.  You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing,
// software distributed under the License is distributed on an
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
// KIND, either express or implied.  See the License for the
// specific language governing permissions and limitations
// under the License.

use std::collections::HashSet;

use jni::JNIEnv;
use jni::objects::JClass;
use jni::objects::JObject;
use jni::sys::jobjectArray;
use jni::sys::jsize;

use crate::Result;
use crate::convert::string_to_jstring;

/// # Safety
///
/// This function should not be called before the Operator is ready.
#[unsafe(no_mangle)]
pub unsafe extern "system" fn Java_org_apache_opendal_OpenDAL_loadEnabledServices(
    mut env: JNIEnv,
    _: JClass,
) -> jobjectArray {
    intern_load_enabled_services(&mut env).unwrap_or_else(|e| {
        e.throw(&mut env);
        JObject::default().into_raw()
    })
}

fn intern_load_enabled_services(env: &mut JNIEnv) -> Result<jobjectArray> {
    let services = HashSet::from([
        opendal::services::ALIYUN_DRIVE_SCHEME,
        opendal::services::ALLUXIO_SCHEME,
        opendal::services::AZBLOB_SCHEME,
        opendal::services::AZDLS_SCHEME,
        opendal::services::AZFILE_SCHEME,
        opendal::services::B2_SCHEME,
        opendal::services::CACACHE_SCHEME,
        opendal::services::COS_SCHEME,
        opendal::services::DASHMAP_SCHEME,
        opendal::services::DROPBOX_SCHEME,
        opendal::services::ETCD_SCHEME,
        opendal::services::FS_SCHEME,
        opendal::services::GCS_SCHEME,
        opendal::services::GDRIVE_SCHEME,
        opendal::services::GHAC_SCHEME,
        opendal::services::GRIDFS_SCHEME,
        opendal::services::HTTP_SCHEME,
        opendal::services::HF_SCHEME,
        opendal::services::IPFS_SCHEME,
        opendal::services::IPMFS_SCHEME,
        opendal::services::KOOFR_SCHEME,
        opendal::services::MEMCACHED_SCHEME,
        opendal::services::MEMORY_SCHEME,
        opendal::services::MINI_MOKA_SCHEME,
        opendal::services::MOKA_SCHEME,
        opendal::services::MONGODB_SCHEME,
        opendal::services::MYSQL_SCHEME,
        opendal::services::OBS_SCHEME,
        opendal::services::ONEDRIVE_SCHEME,
        opendal::services::OSS_SCHEME,
        opendal::services::PERSY_SCHEME,
        opendal::services::POSTGRESQL_SCHEME,
        opendal::services::REDB_SCHEME,
        opendal::services::REDIS_SCHEME,
        opendal::services::S3_SCHEME,
        opendal::services::SEAFILE_SCHEME,
        #[cfg(unix)]
        opendal::services::SFTP_SCHEME,
        opendal::services::SLED_SCHEME,
        opendal::services::SQLITE_SCHEME,
        opendal::services::SWIFT_SCHEME,
        opendal::services::TIKV_SCHEME,
        opendal::services::UPYUN_SCHEME,
        opendal::services::VERCEL_ARTIFACTS_SCHEME,
        opendal::services::WEBDAV_SCHEME,
        opendal::services::WEBHDFS_SCHEME,
        opendal::services::YANDEX_DISK_SCHEME,
    ]);

    let res = env.new_object_array(services.len() as jsize, "java/lang/String", JObject::null())?;

    for (idx, service) in services.into_iter().enumerate() {
        let srv = string_to_jstring(env, Some(service))?;
        env.set_object_array_element(&res, idx as jsize, srv)?;
    }

    Ok(res.into_raw())
}