// Copyright 2015 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 java

// This file generates the final rules for compiling all Java.  All properties related to
// compiling should have been translated into javaBuilderFlags or another argument to the Transform*
// functions.

import (
	"path/filepath"
	"strings"

	"github.com/google/blueprint"

	"android/soong/common"
)

var (
	aaptCreateResourceJavaFile = pctx.StaticRule("aaptCreateResourceJavaFile",
		blueprint.RuleParams{
			Command: `rm -rf "$javaDir" && mkdir -p "$javaDir" && ` +
				`$aaptCmd package -m $aaptFlags -P $publicResourcesFile -G $proguardOptionsFile ` +
				`-J $javaDir || ( rm -rf "$javaDir/*"; exit 41 ) && ` +
				`find $javaDir -name "*.java" > $javaFileList`,
			Description: "aapt create R.java $out",
		},
		"aaptFlags", "publicResourcesFile", "proguardOptionsFile", "javaDir", "javaFileList")

	aaptCreateAssetsPackage = pctx.StaticRule("aaptCreateAssetsPackage",
		blueprint.RuleParams{
			Command:     `rm -f $out && $aaptCmd package $aaptFlags -F $out`,
			Description: "aapt export package $out",
		},
		"aaptFlags", "publicResourcesFile", "proguardOptionsFile", "javaDir", "javaFileList")

	aaptAddResources = pctx.StaticRule("aaptAddResources",
		blueprint.RuleParams{
			// TODO: add-jni-shared-libs-to-package
			Command:     `cp -f $in $out.tmp && $aaptCmd package -u $aaptFlags -F $out.tmp && mv $out.tmp $out`,
			Description: "aapt package $out",
		},
		"aaptFlags")

	zipalign = pctx.StaticRule("zipalign",
		blueprint.RuleParams{
			Command:     `$zipalignCmd -f $zipalignFlags 4 $in $out`,
			Description: "zipalign $out",
		},
		"zipalignFlags")

	signapk = pctx.StaticRule("signapk",
		blueprint.RuleParams{
			Command:     `java -jar $signapkCmd $certificates $in $out`,
			Description: "signapk $out",
		},
		"certificates")

	androidManifestMerger = pctx.StaticRule("androidManifestMerger",
		blueprint.RuleParams{
			Command: "java -classpath $androidManifestMergerCmd com.android.manifmerger.Main merge " +
				"--main $in --libs $libsManifests --out $out",
			Description: "merge manifest files $out",
		},
		"libsManifests")
)

func init() {
	pctx.StaticVariable("androidManifestMergerCmd", "${srcDir}/prebuilts/devtools/tools/lib/manifest-merger.jar")
	pctx.VariableFunc("aaptCmd", func(c interface{}) (string, error) {
		return c.(common.Config).HostBinTool("aapt")
	})
	pctx.VariableFunc("zipalignCmd", func(c interface{}) (string, error) {
		return c.(common.Config).HostBinTool("zipalign")
	})
	pctx.VariableFunc("signapkCmd", func(c interface{}) (string, error) {
		return c.(common.Config).HostJavaTool("signapk.jar")
	})
}

func CreateResourceJavaFiles(ctx common.AndroidModuleContext, flags []string,
	deps []string) (string, string, string) {
	javaDir := filepath.Join(common.ModuleGenDir(ctx), "R")
	javaFileList := filepath.Join(common.ModuleOutDir(ctx), "R.filelist")
	publicResourcesFile := filepath.Join(common.ModuleOutDir(ctx), "public_resources.xml")
	proguardOptionsFile := filepath.Join(common.ModuleOutDir(ctx), "proguard.options")

	deps = append([]string{"$aaptCmd"}, deps...)

	ctx.Build(pctx, blueprint.BuildParams{
		Rule:      aaptCreateResourceJavaFile,
		Outputs:   []string{publicResourcesFile, proguardOptionsFile, javaFileList},
		Implicits: deps,
		Args: map[string]string{
			"aaptFlags":           strings.Join(flags, " "),
			"publicResourcesFile": publicResourcesFile,
			"proguardOptionsFile": proguardOptionsFile,
			"javaDir":             javaDir,
			"javaFileList":        javaFileList,
		},
	})

	return publicResourcesFile, proguardOptionsFile, javaFileList
}

func CreateExportPackage(ctx common.AndroidModuleContext, flags []string, deps []string) string {
	outputFile := filepath.Join(common.ModuleOutDir(ctx), "package-export.apk")

	deps = append([]string{"$aaptCmd"}, deps...)

	ctx.Build(pctx, blueprint.BuildParams{
		Rule:      aaptCreateAssetsPackage,
		Outputs:   []string{outputFile},
		Implicits: deps,
		Args: map[string]string{
			"aaptFlags": strings.Join(flags, " "),
		},
	})

	return outputFile
}

func CreateAppPackage(ctx common.AndroidModuleContext, flags []string, jarFile string,
	certificates []string) string {

	resourceApk := filepath.Join(common.ModuleOutDir(ctx), "resources.apk")

	ctx.Build(pctx, blueprint.BuildParams{
		Rule:      aaptAddResources,
		Outputs:   []string{resourceApk},
		Inputs:    []string{jarFile},
		Implicits: []string{"$aaptCmd"},
		Args: map[string]string{
			"aaptFlags": strings.Join(flags, " "),
		},
	})

	signedApk := filepath.Join(common.ModuleOutDir(ctx), "signed.apk")

	var certificateArgs []string
	for _, c := range certificates {
		certificateArgs = append(certificateArgs, c+".x509.pem", c+".pk8")
	}

	ctx.Build(pctx, blueprint.BuildParams{
		Rule:      signapk,
		Outputs:   []string{signedApk},
		Inputs:    []string{resourceApk},
		Implicits: []string{"$signapkCmd"},
		Args: map[string]string{
			"certificates": strings.Join(certificateArgs, " "),
		},
	})

	outputFile := filepath.Join(common.ModuleOutDir(ctx), "package.apk")

	ctx.Build(pctx, blueprint.BuildParams{
		Rule:      zipalign,
		Outputs:   []string{outputFile},
		Inputs:    []string{signedApk},
		Implicits: []string{"$zipalignCmd"},
		Args: map[string]string{
			"zipalignFlags": "",
		},
	})

	return outputFile
}