// Copyright 2019 Google Inc. All rights reserved.
//
// 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 java

import (
 "android/soong/android"
 "github.com/google/blueprint"
)

//go:generate go run ../../blueprint/gobtools/codegen

// @auto-generate: gob
type JavaPluginInfo struct {
 ProcessorClass *string
 GeneratesApi   bool
}

var JavaPluginInfoProvider = blueprint.NewProvider[JavaPluginInfo]()

// @auto-generate: gob
type KotlinPluginInfo struct {
}

var KotlinPluginInfoProvider = blueprint.NewProvider[KotlinPluginInfo]()

func init() {
 registerJavaPluginBuildComponents(android.InitRegistrationContext)
}

func registerJavaPluginBuildComponents(ctx android.RegistrationContext) {
 ctx.RegisterModuleType("java_plugin", PluginFactory)
 ctx.RegisterModuleType("kotlin_plugin", KotlinPluginFactory)
}

func PluginFactory() android.Module {
 module := &Plugin{}

 module.addHostProperties()
 module.AddProperties(&module.pluginProperties)

 InitJavaModule(module, android.HostSupported)

 return module
}

func KotlinPluginFactory() android.Module {
 module := &KotlinPlugin{}

 module.addHostProperties()

 InitJavaModule(module, android.HostSupported)

 return module
}

// Plugin describes a java_plugin module, a host java library that will be used by javac as an annotation processor.
type Plugin struct {
 Library

 pluginProperties PluginProperties
}

type PluginProperties struct {
 // The optional name of the class that javac will use to run the annotation processor.
 Processor_class *string

 // If true, assume the annotation processor will generate classes that are referenced from outside the module.
 // This necessitates disabling the turbine optimization on modules that use this plugin, which will reduce
 // parallelism and cause more recompilation for modules that depend on modules that use this plugin.
 Generates_api *bool
}

func (p *Plugin) GenerateAndroidBuildActions(ctx android.ModuleContext) {
 p.Library.GenerateAndroidBuildActions(ctx)

 android.SetProvider(ctx, JavaPluginInfoProvider, JavaPluginInfo{
  ProcessorClass: p.pluginProperties.Processor_class,
  GeneratesApi:   Bool(p.pluginProperties.Generates_api),
 })
}

// Plugin describes a kotlin_plugin module, a host java/kotlin library that will be used by kotlinc as a compiler plugin.
type KotlinPlugin struct {
 Library
}

func (p *KotlinPlugin) GenerateAndroidBuildActions(ctx android.ModuleContext) {
 p.Library.GenerateAndroidBuildActions(ctx)

 android.SetProvider(ctx, KotlinPluginInfoProvider, KotlinPluginInfo{})
}