// Copyright 2024 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 codegen

import (
	"fmt"
	"maps"

	"android/soong/android"

	"github.com/google/blueprint"
)

type dependencyTag struct {
	blueprint.BaseDependencyTag
	name string
}

var (
	aconfigDeclarationsGroupTag = dependencyTag{name: "aconfigDeclarationsGroup"}
	javaAconfigLibraryTag       = dependencyTag{name: "javaAconfigLibrary"}
	ccAconfigLibraryTag         = dependencyTag{name: "ccAconfigLibrary"}
	rustAconfigLibraryTag       = dependencyTag{name: "rustAconfigLibrary"}
)

type AconfigDeclarationsGroup struct {
	android.ModuleBase
	android.DefaultableModuleBase

	properties AconfigDeclarationsGroupProperties
}

type AconfigDeclarationsGroupProperties struct {

	// Name of the aconfig_declarations_group modules
	Aconfig_declarations_groups []string

	// Name of the java_aconfig_library modules
	Java_aconfig_libraries []string

	// Name of the cc_aconfig_library modules
	Cc_aconfig_libraries []string

	// Name of the rust_aconfig_library modules
	Rust_aconfig_libraries []string
}

func AconfigDeclarationsGroupFactory() android.Module {
	module := &AconfigDeclarationsGroup{}
	module.AddProperties(&module.properties)
	android.InitAndroidArchModule(module, android.DeviceSupported, android.MultilibCommon)
	android.InitDefaultableModule(module)
	return module
}

func (adg *AconfigDeclarationsGroup) DepsMutator(ctx android.BottomUpMutatorContext) {
	ctx.AddDependency(ctx.Module(), aconfigDeclarationsGroupTag, adg.properties.Aconfig_declarations_groups...)
	ctx.AddDependency(ctx.Module(), javaAconfigLibraryTag, adg.properties.Java_aconfig_libraries...)
	ctx.AddDependency(ctx.Module(), ccAconfigLibraryTag, adg.properties.Cc_aconfig_libraries...)
	ctx.AddDependency(ctx.Module(), rustAconfigLibraryTag, adg.properties.Rust_aconfig_libraries...)
}

func (adg *AconfigDeclarationsGroup) GenerateAndroidBuildActions(ctx android.ModuleContext) {
	modeInfos := make(map[string]android.ModeInfo)
	var aconfigDeclarationNames []string
	var intermediateCacheOutputPaths android.Paths
	var javaSrcjars android.Paths
	ctx.VisitDirectDepsProxy(func(dep android.ModuleProxy) {
		tag := ctx.OtherModuleDependencyTag(dep)
		if provider, ok := android.OtherModuleProvider(ctx, dep, android.CodegenInfoProvider); ok {

			// aconfig declaration names and cache files are collected for all aconfig library dependencies
			aconfigDeclarationNames = append(aconfigDeclarationNames, provider.AconfigDeclarations...)
			intermediateCacheOutputPaths = append(intermediateCacheOutputPaths, provider.IntermediateCacheOutputPaths...)

			switch tag {
			case aconfigDeclarationsGroupTag:
				// Will retrieve outputs from another language codegen modules when support is added
				javaSrcjars = append(javaSrcjars, provider.Srcjars...)
				maps.Copy(modeInfos, provider.ModeInfos)
			case javaAconfigLibraryTag:
				javaSrcjars = append(javaSrcjars, provider.Srcjars...)
				maps.Copy(modeInfos, provider.ModeInfos)
			case ccAconfigLibraryTag:
				maps.Copy(modeInfos, provider.ModeInfos)
			case rustAconfigLibraryTag:
				maps.Copy(modeInfos, provider.ModeInfos)
			}
		}
	})

	aconfigDeclarationNames = android.FirstUniqueStrings(aconfigDeclarationNames)
	intermediateCacheOutputPaths = android.FirstUniquePaths(intermediateCacheOutputPaths)

	android.SetProvider(ctx, android.CodegenInfoProvider, android.CodegenInfo{
		AconfigDeclarations:          aconfigDeclarationNames,
		IntermediateCacheOutputPaths: intermediateCacheOutputPaths,
		Srcjars:                      javaSrcjars,
		ModeInfos:                    modeInfos,
	})

	combinedCacheOutputPath := android.PathForModuleOut(ctx, fmt.Sprintf("%s.pb", ctx.ModuleName()))
	rule := android.NewRuleBuilder(pctx, ctx)
	rule.Command().
		BuiltTool("aconfig").
		Text("dump-cache").
		Flag("--dedup").
		Flag("--format").
		Text("protobuf").
		FlagForEachInput("--cache ", intermediateCacheOutputPaths).
		FlagWithOutput("--out ", combinedCacheOutputPath)

	rule.Build("combine-proto", "combine proto files")

	ctx.SetOutputFiles(intermediateCacheOutputPaths, "")
	ctx.SetOutputFiles(javaSrcjars, ".srcjars")
	ctx.SetOutputFiles(android.Paths{combinedCacheOutputPath}, ".combined_proto")
}