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

import (
 "path"
 "path/filepath"
 "strings"

 "github.com/google/blueprint"
)

var (
 // Command line tool to generate SBOM in Soong
 genSbom = pctx.HostBinToolVariable("genSbom", "gen_sbom")

 // Command to generate SBOM in Soong.
 genSbomRule = pctx.AndroidStaticRule("genSbomRule", blueprint.RuleParams{
  Command:         "rm -rf $out && ${genSbom} --output_file ${out} --metadata ${in} --product_out ${productOut} --soong_out ${soongOut} --build_version \"$$(cat ${buildFingerprintFile})\" --product_mfr \"${productManufacturer}\" --json ${unbundledModule}",
  CommandDeps:     []string{"${genSbom}"},
  SandboxDisabled: true,
 }, "productOut", "soongOut", "buildFingerprintFile", "productManufacturer", "unbundledModule")
)

func init() {
 RegisterSbomSingleton(InitRegistrationContext)
}

func RegisterSbomSingleton(ctx RegistrationContext) {
 ctx.RegisterParallelSingletonType("sbom_singleton", sbomSingletonFactory)
}

// sbomSingleton is used to generate build actions of generating SBOM of products.
type sbomSingleton struct {
}

func sbomSingletonFactory() Singleton {
 return &sbomSingleton{}
}

// Generates SBOM of products
func (this *sbomSingleton) GenerateBuildActions(ctx SingletonContext) {
 if !ctx.Config().HasDeviceProduct() {
  return
 }
 buildFingerprintFile := ctx.Config().BuildFingerprintFile(ctx)
 metadataDb := PathForOutput(ctx, "compliance-metadata", ctx.Config().DeviceProduct(), "compliance-metadata.db")
 productOut := filepath.Join(ctx.Config().OutDir(), "target", "product", String(ctx.Config().productVariables.DeviceName))

 if ctx.Config().HasUnbundledBuildApps() {
  // In unbundled builds, the unbundled_builder module will find what modules to build
  // and then call BuildUnbundledSbom() on each of them, so this singleton doesn't need
  // to do anything.

  // Still create an sbom phony just so the phony exists even if there are no sbom-generating
  // unbundled apps.
  ctx.Phony("sbom")
 } else {
  // When building SBOM of products, phony rule "sbom" is for generating product SBOM in Soong.
  implicits := []Path{}
  implicits = append(implicits, buildFingerprintFile)

  // Add installed_files.stamp as implicit input, which depends on all installed files of the product.
  installedFilesStamp := PathForOutput(ctx, "compliance-metadata", ctx.Config().DeviceProduct(), "installed_files.stamp")
  implicits = append(implicits, installedFilesStamp)

  sbomFile := PathForOutput(ctx, "sbom", ctx.Config().DeviceProduct(), "sbom.spdx.json")
  ctx.Build(pctx, BuildParams{
   Rule:      genSbomRule,
   Input:     metadataDb,
   Implicits: implicits,
   Output:    sbomFile,
   Args: map[string]string{
    "productOut":           productOut,
    "soongOut":             ctx.Config().soongOutDir,
    "buildFingerprintFile": buildFingerprintFile.String(),
    "productManufacturer":  ctx.Config().ProductVariables().ProductManufacturer,
   },
  })

  ctx.Build(pctx, BuildParams{
   Rule:   blueprint.Phony,
   Inputs: []Path{sbomFile},
   Output: PathForPhony(ctx, "sbom"),
  })
  ctx.DistForGoalWithFilename("droid", sbomFile, "sbom/sbom.spdx.json")
 }
}

func BuildUnbundledSbom(ctx ModuleContext, module ModuleProxy) {
 if metadataInfo := GetComplianceMetadata(ctx, module); metadataInfo != nil && len(metadataInfo.filesContained) > 0 {
  buildFingerprintFile := ctx.Config().BuildFingerprintFile(ctx)
  metadataDb := PathForOutput(ctx, "compliance-metadata", ctx.Config().DeviceProduct(), "compliance-metadata.db")
  productOut := filepath.Join(ctx.Config().OutDir(), "target", "product", String(ctx.Config().productVariables.DeviceName))

  implicits := []Path{}
  implicits = append(implicits, buildFingerprintFile)
  installedFile := metadataInfo.filesContained[0]
  implicits = append(implicits, PathForArbitraryOutput(ctx, strings.TrimPrefix(installedFile, ctx.Config().OutDir()+"/")))

  sbomFile := PathForOutput(ctx, "sbom", ctx.Config().DeviceProduct(), module.Name(), path.Base(installedFile)+".spdx.json")
  ctx.Build(pctx, BuildParams{
   Rule:      genSbomRule,
   Input:     metadataDb,
   Implicits: implicits,
   Output:    sbomFile,
   Args: map[string]string{
    "productOut":           productOut,
    "soongOut":             ctx.Config().soongOutDir,
    "buildFingerprintFile": buildFingerprintFile.String(),
    "productManufacturer":  ctx.Config().ProductVariables().ProductManufacturer,
    "unbundledModule":      "--unbundled_module " + module.Name(),
   },
  })
  ctx.Phony("sbom", sbomFile)
  ctx.DistForGoalsWithFilename([]string{"apps_only", "sbom"}, sbomFile, "sbom/"+sbomFile.Base())
 }
}