// Copyright 2017 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 com.google.devtools.build.android.aapt2;

import com.google.devtools.build.android.AndroidResourceOutputs;
import com.google.devtools.build.android.ResourcesZip;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Path;
import javax.annotation.Nullable;

/** Represents the packaged, flattened resources. */
public class PackagedResources {

  private final Path apk;
  private final Path proto;
  private final Path rTxt;
  private final Path proguardConfig;
  private final Path mainDexProguard;
  private final Path javaSourceDirectory;
  private final Path resourceIds;
  private final Path attributes;
  private final Path packages;

  private PackagedResources(
      Path apk,
      Path proto,
      Path rTxt,
      Path proguardConfig,
      Path mainDexProguard,
      Path javaSourceDirectory,
      Path resourceIds,
      Path attributes,
      Path packages) {
    this.apk = apk;
    this.proto = proto;
    this.rTxt = rTxt;
    this.proguardConfig = proguardConfig;
    this.mainDexProguard = mainDexProguard;
    this.javaSourceDirectory = javaSourceDirectory;
    this.resourceIds = resourceIds;
    this.attributes = attributes;
    this.packages = packages;
  }

  public static PackagedResources of(
      Path outPath,
      Path protoPath,
      Path rTxt,
      Path proguardConfig,
      Path mainDexProguard,
      Path javaSourceDirectory,
      Path resourceIds,
      Path attributes,
      Path packages)
      throws IOException {
    return new PackagedResources(
        outPath,
        protoPath,
        rTxt,
        proguardConfig,
        mainDexProguard,
        javaSourceDirectory,
        resourceIds,
        attributes,
        packages);
  }

  public PackagedResources copyPackageTo(Path packagePath) throws IOException {
    return of(
        copy(apk, packagePath),
        proto,
        rTxt,
        proguardConfig,
        mainDexProguard,
        javaSourceDirectory,
        resourceIds,
        attributes,
        packages);
  }

  public PackagedResources copyRTxtTo(Path rOutput) throws IOException {
    if (rOutput == null) {
      return this;
    }
    return new PackagedResources(
        apk,
        proto,
        copy(rTxt, rOutput),
        proguardConfig,
        mainDexProguard,
        javaSourceDirectory,
        resourceIds,
        attributes,
        packages);
  }

  private Path copy(Path from, Path out) throws IOException {
    Files.createDirectories(out.getParent());
    Files.copy(from, out);
    return out;
  }

  public PackagedResources copyProguardTo(Path proguardOut) throws IOException {
    if (proguardOut == null) {
      return this;
    }
    return of(
        apk,
        proto,
        rTxt,
        copy(proguardConfig, proguardOut),
        mainDexProguard,
        javaSourceDirectory,
        resourceIds,
        attributes,
        packages);
  }

  public PackagedResources copyMainDexProguardTo(Path mainDexProguardOut) throws IOException {
    if (mainDexProguardOut == null) {
      return this;
    }
    return of(
        apk,
        proto,
        rTxt,
        proguardConfig,
        copy(mainDexProguard, mainDexProguardOut),
        javaSourceDirectory,
        resourceIds,
        attributes,
        packages);
  }

  public PackagedResources createSourceJar(@Nullable Path sourceJarPath) throws IOException {
    if (sourceJarPath == null) {
      return this;
    }
    AndroidResourceOutputs.createSrcJar(javaSourceDirectory, sourceJarPath, false);
    return of(
        apk,
        proto,
        rTxt,
        proguardConfig,
        mainDexProguard,
        sourceJarPath,
        resourceIds,
        attributes,
        packages);
  }

  public ResourcesZip asArchive() {
    return ResourcesZip.fromApkWithProto(proto, attributes, resourceIds, packages);
  }

  public Path getApk() {
    return apk;
  }
}