1 files changed, 123 insertions, 0 deletions
diff --git a/src/tools/ak/link/link.go b/src/tools/ak/link/link.go
new file mode 100644
index 0000000..4850ae7
--- /dev/null
+++ b/src/tools/ak/link/link.go
@@ -0,0 +1,123 @@
+// Copyright 2018 The Bazel Authors. 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 link is a thin wrapper around aapt2 to link android resources.
+package link
+
+import (
+	"flag"
+	"io/ioutil"
+	"log"
+	"os/exec"
+	"sync"
+
+	"src/common/golang/flags"
+	"src/common/golang/walk"
+	"src/common/golang/ziputils"
+	"src/tools/ak/types"
+)
+
+var (
+	// Cmd defines the command to run link.
+	Cmd = types.Command{
+		Init: Init,
+		Run:  Run,
+		Desc: desc,
+		Flags: []string{
+			"aapt2",
+			"sdk_jar",
+			"manifest",
+			"res_dirs",
+			"asset_dirs",
+			"pkg",
+			"src_jar",
+			"out",
+		},
+	}
+
+	aapt2     string
+	sdkJar    string
+	manifest  string
+	resDirs   flags.StringList
+	assetDirs flags.StringList
+	pkg       string
+	srcJar    string
+	out       string
+
+	initOnce sync.Once
+)
+
+// Init initializes link.
+func Init() {
+	initOnce.Do(func() {
+		flag.StringVar(&aapt2, "aapt2", "", "Path to the aapt2 binary.")
+		flag.StringVar(&sdkJar, "sdk_jar", "", "Path to the android jar.")
+		flag.StringVar(&manifest, "manifest", "", "Path to the application AndroidManifest.xml.")
+		flag.Var(&resDirs, "res_dirs", "List of resource archives to link.")
+		flag.Var(&assetDirs, "asset_dirs", "Paths to asset directories..")
+		flag.StringVar(&pkg, "pkg", "", "Package for R.java.")
+		flag.StringVar(&srcJar, "src_jar", "", "R java source jar path.")
+		flag.StringVar(&out, "out", "", "Output path for linked archive.")
+	})
+}
+
+func desc() string {
+	return "Link compiled Android resources."
+}
+
+// Run is the entry point for link.
+func Run() {
+	if aapt2 == "" ||
+		sdkJar == "" ||
+		manifest == "" ||
+		resDirs == nil ||
+		pkg == "" ||
+		srcJar == "" ||
+		out == "" {
+		log.Fatal("Flags -aapt2 -sdk_jar -manifest -res_dirs -pkg -src_jar and -out must be specified.")
+	}
+
+	// Note that relative order between directories needs to be respected by traversal function.
+	// I.e. all files in dir n most come before all files in directory n+1.
+	resArchives, err := walk.Files(resDirs)
+	if err != nil {
+		log.Fatalf("error getting resource archives: %v", err)
+	}
+
+	rjavaDir, err := ioutil.TempDir("", "rjava")
+	if err != nil {
+		log.Fatalf("error creating temp dir: %v", err)
+	}
+
+	args := []string{
+		"link", "--manifest", manifest, "--auto-add-overlay", "--no-static-lib-packages",
+		"--java", rjavaDir, "--custom-package", pkg, "-I", sdkJar}
+
+	for _, r := range resArchives {
+		args = append(args, "-R", r)
+	}
+
+	for _, a := range assetDirs {
+		args = append(args, "-A", a)
+	}
+
+	args = append(args, "-o", out)
+
+	if out, err := exec.Command(aapt2, args...).CombinedOutput(); err != nil {
+		log.Fatalf("error linking Android resources: %v\n %s", err, string(out))
+	}
+	if err := ziputils.Zip(rjavaDir, srcJar); err != nil {
+		log.Fatalf("error unable to create resources src jar: %v", err)
+	}
+}