/*
 * Copyright 2016 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.idea.blaze.base.sync.aspects;

import com.google.idea.blaze.base.command.BlazeCommand;
import com.google.repackaged.devtools.build.lib.ideinfo.androidstudio.AndroidStudioIdeInfo;
import com.google.repackaged.protobuf.TextFormat;

import java.io.*;

/**
 * Indirection for our various ways of calling the aspect.
 */
public interface AspectStrategy {

  String getName();

  void modifyIdeInfoCommand(BlazeCommand.Builder blazeCommandBuilder);

  void modifyIdeResolveCommand(BlazeCommand.Builder blazeCommandBuilder);

  String getAspectOutputFileExtension();

  AndroidStudioIdeInfo.RuleIdeInfo readAspectFile(File file) throws IOException;

  AspectStrategy NATIVE_ASPECT = new AspectStrategy() {
    @Override
    public String getName() {
      return "NativeAspect";
    }

    @Override
    public void modifyIdeInfoCommand(BlazeCommand.Builder blazeCommandBuilder) {
      blazeCommandBuilder
        .addBlazeFlags("--aspects=AndroidStudioInfoAspect")
        .addBlazeFlags("--output_groups=ide-info");
    }

    @Override
    public void modifyIdeResolveCommand(BlazeCommand.Builder blazeCommandBuilder) {
      blazeCommandBuilder
        .addBlazeFlags("--aspects=AndroidStudioInfoAspect")
        .addBlazeFlags("--output_groups=ide-resolve");
    }

    @Override
    public String getAspectOutputFileExtension() {
      return ".aswb-build";
    }

    @Override
    public AndroidStudioIdeInfo.RuleIdeInfo readAspectFile(File file) throws IOException {
      try (InputStream inputStream = new FileInputStream(file)) {
        return AndroidStudioIdeInfo.RuleIdeInfo.parseFrom(inputStream);
      }
    }
  };

  AspectStrategy SKYLARK_ASPECT = new AspectStrategy() {
    @Override
    public String getName() {
      return "SkylarkAspect";
    }

    private void addAspectFlag(BlazeCommand.Builder blazeCommandBuilder) {
      blazeCommandBuilder.addBlazeFlags(
        "--aspects=//third_party/bazel/src/test/java/com/google/devtools/build/lib/ideinfo/intellij_info.bzl%intellij_info_aspect"
      );
    }

    @Override
    public void modifyIdeInfoCommand(BlazeCommand.Builder blazeCommandBuilder) {
      addAspectFlag(blazeCommandBuilder);
      blazeCommandBuilder.addBlazeFlags("--output_groups=ide-info-text");
    }

    @Override
    public void modifyIdeResolveCommand(BlazeCommand.Builder blazeCommandBuilder) {
      addAspectFlag(blazeCommandBuilder);
      blazeCommandBuilder.addBlazeFlags("--output_groups=ide-resolve");
    }

    @Override
    public String getAspectOutputFileExtension() {
      return ".intellij-build.txt";
    }

    @Override
    public AndroidStudioIdeInfo.RuleIdeInfo readAspectFile(File file) throws IOException {
      try (InputStream inputStream = new FileInputStream(file)) {
        AndroidStudioIdeInfo.RuleIdeInfo.Builder builder = AndroidStudioIdeInfo.RuleIdeInfo.newBuilder();
        TextFormat.Parser parser = TextFormat.Parser.newBuilder()
          .setAllowUnknownFields(true)
          .build();
        parser.merge(new InputStreamReader(inputStream), builder);
        return builder.build();
      }
    }
  };

}