// Copyright 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 android

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

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

var (
 removeComments = pctx.AndroidStaticRule("remove_comments",
  blueprint.RuleParams{
   Command2: blueprint.NewCommand(Grep, " -v '#' $in > $out"),
  },
 )
 androidInfoTxtToProp = pctx.AndroidStaticRule("android_info_txt_to_prop",
  blueprint.RuleParams{
   Command2: blueprint.NewCommand(
    Grep, ` 'require version-' $in | `, Sed, ` -e 's/require version-/ro.build.expect./g' > $out`),
  },
 )
)

type AndroidInfoProperties struct {
 // Name of output file. Defaults to module name
 Stem *string

 // Paths of board-info.txt files.
 Board_info_files []string `android:"path"`

 // Name of bootloader board. If board_info_files is empty, `board={bootloader_board_name}` will
 // be printed to output. Ignored if board_info_files is not empty.
 Bootloader_board_name *string
}

type androidInfoModule struct {
 ModuleBase

 properties AndroidInfoProperties
}

// @auto-generate: gob
type AndroidInfo struct {
 // Path to the android info prop file which is used to generate the vendor/build.prop
 AndroidInfoProp Path
 // Path to the android info txt file, which is generated by merging the board info txt files
 // and removing the comments
 AndroidInfoTxt Path
 // Path to the merged board info txt file, with comments
 BoardInfoTxt Path
}

var AndroidInfoProvider = blueprint.NewProvider[AndroidInfo]()

func (p *androidInfoModule) GenerateAndroidBuildActions(ctx ModuleContext) {
 if len(p.properties.Board_info_files) > 0 && p.properties.Bootloader_board_name != nil {
  ctx.ModuleErrorf("Either Board_info_files or Bootloader_board_name should be set. Please remove one of them\n")
  return
 }
 androidInfoTxtName := proptools.StringDefault(p.properties.Stem, ctx.ModuleName()+".txt")
 androidInfoTxt := PathForModuleOut(ctx, androidInfoTxtName)
 androidInfoProp := androidInfoTxt.ReplaceExtension(ctx, "prop")
 mergedBoardInfoTxt := PathForModuleOut(ctx, "board-info.txt")

 androidInfo := AndroidInfo{
  AndroidInfoProp: androidInfoProp,
  AndroidInfoTxt:  androidInfoTxt,
 }

 if boardInfoFiles := PathsForModuleSrc(ctx, p.properties.Board_info_files); len(boardInfoFiles) > 0 {
  ctx.Build(pctx, BuildParams{
   Rule:   CatRule,
   Inputs: boardInfoFiles,
   Output: mergedBoardInfoTxt,
  })
  ctx.Build(pctx, BuildParams{
   Rule:   removeComments,
   Input:  mergedBoardInfoTxt,
   Output: androidInfoTxt,
  })
  androidInfo.BoardInfoTxt = mergedBoardInfoTxt
 } else if bootloaderBoardName := proptools.String(p.properties.Bootloader_board_name); bootloaderBoardName != "" {
  WriteFileRule(ctx, androidInfoTxt, "board="+bootloaderBoardName)
 } else {
  WriteFileRule(ctx, androidInfoTxt, "")
 }

 // Create android_info.prop
 ctx.Build(pctx, BuildParams{
  Rule:   androidInfoTxtToProp,
  Input:  androidInfoTxt,
  Output: androidInfoProp,
 })

 SetProvider(ctx, AndroidInfoProvider, androidInfo)

 ctx.SetOutputFiles(Paths{androidInfoProp}, "")
 ctx.SetOutputFiles(Paths{androidInfoTxt}, ".txt")
}

// android_info module generate a file named android-info.txt that contains various information
// about the device we're building for.  This file is typically packaged up with everything else.
func AndroidInfoFactory() Module {
 module := &androidInfoModule{}
 module.AddProperties(&module.properties)
 InitAndroidArchModule(module, DeviceSupported, MultilibCommon)
 return module
}