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

import (
 "android/soong/android"

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

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

func RegisterOtacertsZipBuildComponents(ctx android.RegistrationContext) {
 ctx.RegisterModuleType("otacerts_zip", otacertsZipFactory)
}

type otacertsZipProperties struct {
 // Make this module available when building for recovery.
 // Only the recovery partition is available.
 Recovery_available *bool

 // Optional subdirectory under which the zip file is installed into.
 Relative_install_path *string

 // Optional name for the installed file. If unspecified, otacerts.zip is used.
 Filename *string
}

type otacertsZipModule struct {
 android.ModuleBase

 properties otacertsZipProperties
 outputPath android.Path
}

// otacerts_zip collects key files defined in PRODUCT_DEFAULT_DEV_CERTIFICATE
// and PRODUCT_EXTRA_OTA_KEYS for system or PRODUCT_EXTRA_RECOVERY_KEYS for
// recovery image. The output file (otacerts.zip by default) is installed into
// the relative_install_path directory under the etc directory of the target
// partition.
func otacertsZipFactory() android.Module {
 module := &otacertsZipModule{}
 module.AddProperties(&module.properties)
 android.InitAndroidArchModule(module, android.DeviceSupported, android.MultilibCommon)
 return module
}

var _ android.ImageInterface = (*otacertsZipModule)(nil)

func (m *otacertsZipModule) ImageMutatorBegin(ctx android.ImageInterfaceContext) {}

func (m *otacertsZipModule) ImageMutatorSupported() bool {
 return true
}

func (m *otacertsZipModule) VendorVariantNeeded(ctx android.ImageInterfaceContext) bool {
 return false
}

func (m *otacertsZipModule) ProductVariantNeeded(ctx android.ImageInterfaceContext) bool {
 return false
}

func (m *otacertsZipModule) CoreVariantNeeded(ctx android.ImageInterfaceContext) bool {
 return !m.ModuleBase.InstallInRecovery()
}

func (m *otacertsZipModule) RamdiskVariantNeeded(ctx android.ImageInterfaceContext) bool {
 return false
}

func (m *otacertsZipModule) VendorRamdiskVariantNeeded(ctx android.ImageInterfaceContext) bool {
 return false
}

func (m *otacertsZipModule) DebugRamdiskVariantNeeded(ctx android.ImageInterfaceContext) bool {
 return false
}

func (m *otacertsZipModule) RecoveryVariantNeeded(ctx android.ImageInterfaceContext) bool {
 return proptools.Bool(m.properties.Recovery_available) || m.ModuleBase.InstallInRecovery()
}

func (m *otacertsZipModule) ExtraImageVariations(ctx android.ImageInterfaceContext) []string {
 return nil
}

func (m *otacertsZipModule) SetImageVariation(ctx android.ImageInterfaceContext, variation string) {
}

func (m *otacertsZipModule) InRecovery() bool {
 return m.ModuleBase.InRecovery() || m.ModuleBase.InstallInRecovery()
}

func (m *otacertsZipModule) InstallInRecovery() bool {
 return m.InRecovery()
}

func (m *otacertsZipModule) outputFileName() string {
 // Use otacerts.zip if not specified.
 return proptools.StringDefault(m.properties.Filename, "otacerts.zip")
}

func (m *otacertsZipModule) onlyInRecovery() bool {
 return m.ModuleBase.InstallInRecovery()
}

func (m *otacertsZipModule) DepsMutator(ctx android.BottomUpMutatorContext) {
 ctx.AddHostToolDependencies("soong_zip")
}

func (m *otacertsZipModule) GenerateAndroidBuildActions(ctx android.ModuleContext) {
 // Read .x509.pem file defined in PRODUCT_DEFAULT_DEV_CERTIFICATE or the default test key.
 pem, _ := ctx.Config().DefaultAppCertificate(ctx)
 // Read .x509.pem files listed  in PRODUCT_EXTRA_OTA_KEYS or PRODUCT_EXTRA_RECOVERY_KEYS.
 extras := ctx.Config().ExtraOtaKeys(ctx, m.InRecovery())
 srcPaths := append([]android.SourcePath{pem}, extras...)
 outputPath := android.PathForModuleOut(ctx, m.outputFileName())

 rule := android.NewRuleBuilder(pctx, ctx)
 cmd := rule.Command().BuiltTool("soong_zip").
  FlagWithOutput("-o ", outputPath).
  Flag("-j ").
  Flag("-symlinks=false ")
 for _, src := range srcPaths {
  cmd.FlagWithInput("-f ", src)
 }
 rule.Build(ctx.ModuleName(), "Generating the otacerts zip file")

 installPath := android.PathForModuleInstall(ctx, "etc", proptools.String(m.properties.Relative_install_path))
 ctx.InstallFile(installPath, m.outputFileName(), outputPath)
 m.outputPath = outputPath
}

func (m *otacertsZipModule) AndroidMkEntries() []android.AndroidMkEntries {
 nameSuffix := ""
 if m.InRecovery() && !m.onlyInRecovery() {
  nameSuffix = ".recovery"
 }
 return []android.AndroidMkEntries{android.AndroidMkEntries{
  Class:      "ETC",
  SubName:    nameSuffix,
  OutputFile: android.OptionalPathForPath(m.outputPath),
 }}
}