//
// Copyright (C) 2025 The Android Open Source Project
//
// 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.
//

package rust

import (
 "android/soong/android"
)

// This module is used to compile the rust toolchain objects
// When RUST_PREBUILTS_VERSION is set, the object will generated
// from the given Rust version.
func init() {
 android.RegisterModuleType("rust_toolchain_object",
  rustToolchainObjectFactory)
 android.RegisterModuleType("rust_toolchain_object_host",
  rustToolchainObjectHostFactory)
}

type toolchainObjectProperties struct {
 // path to the toolchain object crate root, relative to the top of the toolchain source
 Toolchain_crate_root *string `android:"arch_variant"`
 // path to the rest of the toolchain srcs, relative to the top of the toolchain source
 Toolchain_srcs []string `android:"arch_variant"`
}

type toolchainObjectDecorator struct {
 *objectDecorator
 Properties toolchainObjectProperties
}

// toolchainCrateRoot implements toolchainCompiler.
func (t *toolchainObjectDecorator) toolchainCrateRoot() *string {
 return t.Properties.Toolchain_crate_root
}

// toolchainSrcs implements toolchainCompiler.
func (t *toolchainObjectDecorator) toolchainSrcs() []string {
 return t.Properties.Toolchain_srcs
}

// rust_toolchain_library produces all rust variants.
func rustToolchainObjectFactory() android.Module {
 module, object := NewRustObject(android.HostAndDeviceSupported)
 return initToolchainObject(module, object)
}

// rust_toolchain_library produces all rust variants.
func rustToolchainObjectHostFactory() android.Module {
 module, object := NewRustObject(android.HostSupported)
 return initToolchainObject(module, object)
}

func initToolchainObject(module *Module, object *objectDecorator) android.Module {
 toolchainObject := &toolchainObjectDecorator{
  objectDecorator: object,
 }
 module.compiler = toolchainObject
 module.AddProperties(&toolchainObject.Properties)

 android.AddLoadHook(module, rustSetToolchainSource)

 return module.Init()
}

func (t *toolchainObjectDecorator) compilerProps() []any {
 return append(t.objectDecorator.compilerProps(), &t.Properties)
}

func (t *toolchainObjectDecorator) crt() bool {
 return t.objectDecorator.crt()
}

func (t *toolchainObjectDecorator) install(ctx ModuleContext) {
 // Objects aren't installable, so do nothing.
}

func (t *toolchainObjectDecorator) everInstallable() bool {
 // Objects aren't installable.
 return false
}

var _ toolchainCompiler = (*toolchainObjectDecorator)(nil)