// Copyright (C) 2024 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 fsgen

import (
 "android/soong/android"
 "android/soong/etc"
 "android/soong/filesystem"
 "fmt"
 "strconv"
 "strings"

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

// Returns the appropriate dpi for recovery common resources selection. Replicates the logic in
// https://cs.android.com/android/platform/superproject/main/+/main:build/make/core/Makefile;l=2536;drc=a6af369e71ded123734523ea640b97b70a557cb9
func getDpi(ctx android.EarlyModuleContext) string {
 recoveryDensity := ctx.Config().ProductVariables().PartitionVarsForSoongMigrationOnlyDoNotUse.TargetScreenDensity
 if len(recoveryDensity) == 0 {
  aaptPreferredConfig := ctx.Config().ProductAAPTPreferredConfig()
  if len(aaptPreferredConfig) > 0 {
   recoveryDensity = aaptPreferredConfig
  } else {
   recoveryDensity = "mdpi"
  }
 }
 if !android.InList(recoveryDensity, []string{"xxxhdpi", "xxhdpi", "xhdpi", "hdpi", "mdpi"}) {
  recoveryDensity = strings.TrimSuffix(recoveryDensity, "dpi")
  dpiInt, err := strconv.ParseInt(recoveryDensity, 10, 64)
  if err != nil {
   panic(fmt.Sprintf("Error in parsing recoveryDensity: %s", err.Error()))
  }
  if dpiInt >= 560 {
   recoveryDensity = "xxxhdpi"
  } else if dpiInt >= 400 {
   recoveryDensity = "xxhdpi"
  } else if dpiInt >= 280 {
   recoveryDensity = "xhdpi"
  } else if dpiInt >= 200 {
   recoveryDensity = "hdpi"
  } else {
   recoveryDensity = "mdpi"
  }
 }

 if p := android.ExistentPathForSource(ctx, fmt.Sprintf("bootable/recovery/res-%s", recoveryDensity)); !p.Valid() {
  recoveryDensity = "xhdpi"
 }

 return recoveryDensity
}

// Returns the name of the appropriate prebuilt module for installing font.png file.
// https://cs.android.com/android/platform/superproject/main/+/main:build/make/core/Makefile;l=2536;drc=a6af369e71ded123734523ea640b97b70a557cb9
func getRecoveryFontModuleName(ctx android.LoadHookContext) string {
 if android.InList(getDpi(ctx), []string{"xxxhdpi", "xxhdpi", "xhdpi"}) {
  return "recovery-fonts-18"
 }
 return "recovery-fonts-12"
}

// Returns the module name and image width for recovery background image generation.
// https://source.corp.google.com/h/googleplex-android/platform/build/+/f1cb088a1446adc54950bd6b447eac9fc90831f6:core/Makefile;l=2539-2556;drc=acc8d6594498f50db20f04e0e05fff09750c412c;bpv=1;bpt=0
func getRecoveryBackgroundPicturesGeneratorModuleName(ctx android.EarlyModuleContext) (string, int64) {
 dpi := getDpi(ctx)
 if !android.InList(dpi, []string{"xxxhdpi", "xxhdpi"}) {
  return "", -1
 }
 var recoveryUiScreenWidth int64
 if dpi == "xxxhdpi" {
  recoveryUiScreenWidth = 1440 - 10
 } else { // xxhdpi
  recoveryUiScreenWidth = 1080 - 10
 }

 if marginWidth, exists := ctx.Config().ProductVariables().PartitionVarsForSoongMigrationOnlyDoNotUse.PrivateRecoveryUiProperties["margin_width"]; exists && marginWidth != "" {
  if marginWidth, err := strconv.ParseInt(marginWidth, 10, 64); err == nil {
   recoveryUiScreenWidth -= marginWidth
  } else {
   ctx.ModuleErrorf("Error parsing TARGET_RECOVERY_UI_MARGIN_WIDTH %s", err)
  }
 }

 return generatedModuleNameForPartition(ctx.Config(), "background_recovery_images"), recoveryUiScreenWidth
}

// Returns a new list of symlinks with prefix added to the dest directory for all symlinks
func symlinksWithNamePrefix(symlinks []filesystem.SymlinkDefinition, prefix string) []filesystem.SymlinkDefinition {
 ret := make([]filesystem.SymlinkDefinition, len(symlinks))
 for i, symlink := range symlinks {
  ret[i] = symlink.CopyWithNamePrefix(prefix)
 }
 return ret
}

// Creates a prebuilt_etc for recovery partition if TARGET_RECOVERY_WIPE is not empty.
// Returns the name of the autogenerated module.
func createTargetRecoveryWipeModuleName(ctx android.LoadHookContext) string {
 recoveryWipe := ctx.Config().ProductVariables().PartitionVarsForSoongMigrationOnlyDoNotUse.TargetRecoveryWipe
 if recoveryWipe == "" {
  return ""
 }
 name := generatedModuleName(ctx.Config(), "target_recovery_wipe")
 ctx.CreateModuleInDirectory(
  etc.PrebuiltEtcFactory,
  ".",
  &struct {
   Name     *string
   Srcs     []string
   Dsts     []string
   Recovery *bool
  }{
   Name:     proptools.StringPtr(name),
   Srcs:     []string{recoveryWipe},
   Dsts:     []string{"recovery.wipe"},
   Recovery: proptools.BoolPtr(true),
  },
 )
 return name
}

func getRecoveryFsTabSrc(config android.Config) (recoveryFstabSrc string) {
 partitionVars := config.ProductVariables().PartitionVarsForSoongMigrationOnlyDoNotUse
 if len(partitionVars.TargetRecoveryFstab) > 0 {
  recoveryFstabSrc = partitionVars.TargetRecoveryFstab
 } else if len(partitionVars.TargetRecoveryFstabGenrule) > 0 {
  recoveryFstabSrc = ":" + partitionVars.TargetRecoveryFstabGenrule

 } else {
  recoveryFstabSrc = partitionVars.TargetRecoveryFstabDefault
 }
 return
}

// Creates a prebuilt_etc module for recovery partition based on the values of
// TARGET_RECOVERY_FSTAB, TARGET_RECOVERY_FSTAB_GENRULE and TARGET_DEVICE_DIR.
// https://cs.android.com/android/platform/superproject/main/+/main:build/make/core/Makefile;l=2616-2622;drc=a951ebf0198006f7fd38073a05c442d0eb92f97b
func handleRecoveryFstab(ctx android.LoadHookContext) (moduleName string) {
 if recoveryFstabSrc := getRecoveryFsTabSrc(ctx.Config()); len(recoveryFstabSrc) > 0 {
  moduleName = generatedModuleName(ctx.Config(), "recovery_fstab")
  ctx.CreateModuleInDirectory(
   etc.PrebuiltEtcFactory,
   ".",
   &struct {
    Name     *string
    Dsts     []string
    Recovery *bool
   }{
    Name:     proptools.StringPtr(moduleName),
    Dsts:     []string{"recovery.fstab"},
    Recovery: proptools.BoolPtr(true),
   },
  )
 }
 return
}

func setRecoveryFstabSrcs(ctx android.BottomUpMutatorContext) {
 if !shouldEnableFilesystemCreator(ctx) {
  return
 }
 recoveryFstabModuleName := ctx.Config().Get(fsGenStateOnceKey).(*FsGenState).recoveryFstabModuleName
 if len(recoveryFstabModuleName) > 0 && ctx.ModuleName() == recoveryFstabModuleName {
  recoveryFstabSrc := getRecoveryFsTabSrc(ctx.Config())
  resolvedRecoveryFstabSrcName := recoveryFstabSrc
  // If the recovery fstab src start with ":", it's depending on a genrule
  // module. However the genrule may be under a soong namespace, thus this
  // needs to be resolved to a fully qualified module name.
  if strings.HasPrefix(recoveryFstabSrc, ":") {
   for _, namespace := range ctx.Config().ProductVariables().NamespacesToExport {
    if moduleName := fmt.Sprintf("//%s:%s", namespace, recoveryFstabSrc[1:]); ctx.OtherModuleExists(moduleName) {
     resolvedRecoveryFstabSrcName = moduleName
    }
   }
  }
  proptools.AppendMatchingProperties(
   ctx.Module().GetProperties(),
   &struct {
    Srcs []string
   }{
    Srcs: []string{resolvedRecoveryFstabSrcName},
   },
   nil,
  )
 }
}