/*****************************************************************************
 * Copyright (c) 2026 Huawei Device Co., Ltd.
 * Licensed 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.
 *****************************************************************************/

import native from 'liblibrary.so';

import type { IOnnxRuntime } from 'liblibrary.so';
import type { IOnnxEnvironment } from 'liblibrary.so';

// 枚举值导出
export { OnnxLoggingLevel } from 'liblibrary.so';
export { OnnxExecutionMode } from 'liblibrary.so';
export { OnnxOptimizationLevel } from 'liblibrary.so';
export { OnnxValueType } from 'liblibrary.so';
export { OnnxArkTSType } from 'liblibrary.so';
export { OnnxTensorType } from 'liblibrary.so';
export { OnnxMapValueType } from 'liblibrary.so';
export { SparseTensorFormat } from 'liblibrary.so';

// 类和接口导出
export { IOnnxRuntime } from 'liblibrary.so';
export { OnnxRuntime } from 'liblibrary.so';
export { IOnnxEnvironment } from 'liblibrary.so';
export { OnnxEnvironment } from 'liblibrary.so';
export { OnnxThreadOptions } from 'liblibrary.so';
export { OnnxSessionOptions } from 'liblibrary.so';
export { OnnxRunOptions } from 'liblibrary.so';
export { OnnxLoraAdapter } from 'liblibrary.so';
export { OnnxSession } from 'liblibrary.so';
export { OnnxResult } from 'liblibrary.so';
export { OnnxModelMetadata } from 'liblibrary.so';
export { NodeInfo } from 'liblibrary.so';
export { ValueInfo } from 'liblibrary.so';
export { OnnxValue } from 'liblibrary.so';
export { TensorInfo } from 'liblibrary.so';
export { OnnxTensor } from 'liblibrary.so';
export { SparseTensor } from 'liblibrary.so';
export { COOTensor } from 'liblibrary.so';
export { CSRCTensor } from 'liblibrary.so';
export { BlockSparseTensor } from 'liblibrary.so';
export { OnnxSparseTensor } from 'liblibrary.so';
export { MapInfo } from 'liblibrary.so';
export { OnnxMap } from 'liblibrary.so';
export { SequenceInfo } from 'liblibrary.so';
export { OnnxSequence } from 'liblibrary.so';

// 默认导出
interface OnnxModule {
  OnnxRuntime: IOnnxRuntime;
  OnnxEnvironment: IOnnxEnvironment;
}

const onnx:OnnxModule = {
  OnnxRuntime: native.OnnxRuntime,
  OnnxEnvironment: native.OnnxEnvironment
};

export default onnx;