// 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 filesystem

import (
	"fmt"
	"strconv"
	"strings"

	"github.com/google/blueprint"
	"github.com/google/blueprint/proptools"

	"android/soong/android"
)

type prebuiltDtboImg struct {
	android.ModuleBase
	properties prebuiltDtboImgProperties
}

type prebuiltDtboImgProperties struct {
	Src            *string `android:"path"`
	Partition_size *int64
	Stem           *string
	Use_avb        *bool
}

func PrebuiltDtboImgFactory() android.Module {
	module := &prebuiltDtboImg{}
	android.InitAndroidArchModule(module, android.DeviceSupported, android.MultilibFirst)
	module.AddProperties(&module.properties)
	return module
}

type DtboImgInfo struct {
	PropFileForMiscInfo android.Path
}

var DtboImgInfoProvider = blueprint.NewProvider[DtboImgInfo]()

func (p *prebuiltDtboImg) GenerateAndroidBuildActions(ctx android.ModuleContext) {
	input := android.PathForModuleSrc(ctx, proptools.String(p.properties.Src))
	output := p.avbAddHash(ctx, input)
	ctx.SetOutputFiles(android.Paths{output}, "")
	android.SetProvider(
		ctx,
		DtboImgInfoProvider,
		DtboImgInfo{
			PropFileForMiscInfo: p.buildPropFileForMiscInfo(ctx),
		},
	)
	android.SetProvider(ctx, vbmetaPartitionProvider, vbmetaPartitionInfo{
		Name:   "dtbo",
		Output: output,
	})
}

func (p *prebuiltDtboImg) avbAddHash(ctx android.ModuleContext, input android.Path) android.Path {
	if p.properties.Use_avb != nil && !*p.properties.Use_avb {
		// Do not sign if Use_avb is explicitly turned off.
		return input
	}
	builder := android.NewRuleBuilder(pctx, ctx)
	filename := proptools.StringDefault(p.properties.Stem, input.Base())
	output := android.PathForModuleOut(ctx, filename)
	builder.Command().Text("cp").Input(input).Output(output)
	builder.Command().Text("chmod").FlagWithArg("+w ", output.String())

	cmd := builder.Command().
		BuiltTool("avbtool").
		Text("add_hash_footer").
		FlagWithArg("--image ", output.String()).
		Flag("--partition_name dtbo").
		Textf(`--salt $(sha256sum "%s" "%s" | cut -d " " -f 1 | tr -d '\n')`, ctx.Config().BuildNumberFile(ctx), ctx.Config().Getenv("BUILD_DATETIME_FILE"))

	if p.properties.Partition_size == nil {
		cmd.Flag("--dynamic_partition_size")
	} else {
		cmd.FlagWithArg("--partition_size ", strconv.FormatInt(*p.properties.Partition_size, 10))
	}
	fingerprintFile := ctx.Config().BuildFingerprintFile(ctx)
	cmd.FlagWithArg("--prop ", fmt.Sprintf("com.android.build.dtbo.fingerprint:$(cat %s)", fingerprintFile.String())).Implicit(fingerprintFile)

	builder.Build("add_hash_footer", "add_hash_footer")

	return output
}

func (p *prebuiltDtboImg) buildPropFileForMiscInfo(ctx android.ModuleContext) android.Path {
	var sb strings.Builder
	addStr := func(name string, value string) {
		fmt.Fprintf(&sb, "%s=%s\n", name, value)
	}

	addStr("has_dtbo", "true")
	if p.properties.Partition_size != nil {
		addStr("dtbo_size", strconv.FormatInt(*p.properties.Partition_size, 10))
	}
	fingerprintFile := ctx.Config().BuildFingerprintFile(ctx)
	addStr("avb_dtbo_add_hash_footer_args", "--prop "+fmt.Sprintf("com.android.build.dtbo.fingerprint:{CONTENTS_OF:%s}", fingerprintFile.String()))

	propFilePreProcessing := android.PathForModuleOut(ctx, "prop_for_misc_info_pre_processing")
	android.WriteFileRuleVerbatim(ctx, propFilePreProcessing, sb.String())
	propFile := android.PathForModuleOut(ctx, "prop_file_for_misc_info")
	ctx.Build(pctx, android.BuildParams{
		Rule:     textFileProcessorRule,
		Input:    propFilePreProcessing,
		Output:   propFile,
		Implicit: fingerprintFile,
	})

	return propFile
}