/*
 * Copyright (c) Facebook, Inc. and its affiliates.
 *
 * 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.facebook.buck.core.model.impl;

import com.facebook.buck.core.exceptions.BuckUncheckedExecutionException;
import com.facebook.buck.core.exceptions.HumanReadableException;
import com.facebook.buck.core.model.BuildTarget;
import com.facebook.buck.core.model.ConfigurationBuildTargets;
import com.facebook.buck.core.model.ConfigurationForConfigurationTargets;
import com.facebook.buck.core.model.RuleBasedTargetConfiguration;
import com.facebook.buck.core.model.TargetConfiguration;
import com.facebook.buck.core.model.TargetConfigurationSerializer;
import com.facebook.buck.core.model.UnconfiguredBuildTarget;
import com.facebook.buck.core.model.UnconfiguredTargetConfiguration;
import com.facebook.buck.util.json.ObjectMappers;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.ObjectReader;
import com.fasterxml.jackson.databind.ObjectWriter;
import com.fasterxml.jackson.databind.module.SimpleModule;
import com.google.common.base.Preconditions;
import com.google.common.collect.ImmutableMap;
import java.io.IOException;
import java.util.function.Function;

/** Serializer that uses JSON to represent {@link TargetConfiguration} in a text form. */
public class JsonTargetConfigurationSerializer implements TargetConfigurationSerializer {

  private static final ImmutableMap<String, Boolean>
      TARGET_CONFIGURATION_FOR_CONFIGURATION_ATTRIBUTES_FOR_SERIALIZATION =
          ImmutableMap.of("configuration", true);

  private final ObjectWriter objectWriter;
  private final ObjectReader objectReader;
  private final Function<String, UnconfiguredBuildTarget> buildTargetProvider;

  public JsonTargetConfigurationSerializer(
      Function<String, UnconfiguredBuildTarget> buildTargetProvider) {
    this.buildTargetProvider = buildTargetProvider;
    ObjectMapper objectMapper = ObjectMappers.createWithEmptyBeansPermitted();
    SimpleModule targetConfigurationModule = new SimpleModule();
    targetConfigurationModule.addSerializer(
        BuildTarget.class, new BuildTargetSimpleSerializer(BuildTarget.class));
    objectMapper.registerModule(targetConfigurationModule);

    objectReader = objectMapper.reader();
    objectWriter = objectMapper.writer();
  }

  @Override
  public String serialize(TargetConfiguration targetConfiguration) {
    try {
      if (targetConfiguration instanceof ConfigurationForConfigurationTargets) {
        return objectWriter.writeValueAsString(
            TARGET_CONFIGURATION_FOR_CONFIGURATION_ATTRIBUTES_FOR_SERIALIZATION);
      } else {
        return objectWriter.writeValueAsString(targetConfiguration);
      }
    } catch (JsonProcessingException e) {
      throw new BuckUncheckedExecutionException(
          e,
          String.format(
              "Cannot serialize target configuration %s (%s)",
              targetConfiguration, targetConfiguration.getClass().getName()));
    }
  }

  @Override
  public TargetConfiguration deserialize(String rawValue) {
    JsonNode node;
    try {
      node = objectReader.readTree(rawValue);
    } catch (IOException e) {
      throw new HumanReadableException(e, "Cannot deserialize target configuration %s", rawValue);
    }
    if (node.size() == 0) {
      return UnconfiguredTargetConfiguration.INSTANCE;
    }
    if (node.has("configuration")) {
      return ConfigurationForConfigurationTargets.INSTANCE;
    }
    JsonNode targetPlatformNode =
        Preconditions.checkNotNull(
            node.get("targetPlatform"), "Cannot find targetPlatform in %s", rawValue);
    BuildTarget platform =
        ConfigurationBuildTargets.convert(
            buildTargetProvider.apply(targetPlatformNode.textValue()));
    return RuleBasedTargetConfiguration.of(platform);
  }
}
