/*
 * Copyright 2015-present Facebook, Inc.
 *
 * 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.intellij.ideabuck.format;

import com.facebook.buck.intellij.ideabuck.lang.BuckLanguage;
import com.facebook.buck.intellij.ideabuck.lang.psi.BuckTypes;
import com.intellij.formatting.CustomFormattingModelBuilder;
import com.intellij.formatting.FormatTextRanges;
import com.intellij.formatting.FormattingMode;
import com.intellij.formatting.FormattingModel;
import com.intellij.formatting.FormattingModelBuilderEx;
import com.intellij.formatting.FormattingModelProvider;
import com.intellij.formatting.Indent;
import com.intellij.formatting.SpacingBuilder;
import com.intellij.lang.ASTNode;
import com.intellij.openapi.util.TextRange;
import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiFile;
import com.intellij.psi.codeStyle.CodeStyleSettings;
import com.intellij.psi.codeStyle.CommonCodeStyleSettings;
import org.jetbrains.annotations.Nullable;

/** Implements the code formatting for BUCK files. */
public class BuckFormattingModelBuilder
    implements FormattingModelBuilderEx, CustomFormattingModelBuilder {

  @Override
  public FormattingModel createModel(
      PsiElement element, CodeStyleSettings settings, FormattingMode mode) {
    final BuckBlock block =
        new BuckBlock(element.getNode(), settings, null, Indent.getNoneIndent(), null);
    return FormattingModelProvider.createFormattingModelForPsiFile(
        element.getContainingFile(), block, settings);
  }

  @Nullable
  @Override
  public CommonCodeStyleSettings.IndentOptions getIndentOptionsToUse(
      PsiFile file, FormatTextRanges ranges, CodeStyleSettings settings) {
    return null;
  }

  @Override
  public FormattingModel createModel(PsiElement element, CodeStyleSettings settings) {
    return createModel(element, settings, FormattingMode.REFORMAT);
  }

  @Nullable
  @Override
  public TextRange getRangeAffectingIndent(PsiFile file, int offset, ASTNode elementAtOffset) {
    final PsiElement element = elementAtOffset.getPsi();
    final PsiElement container = element.getParent();
    return container != null ? container.getTextRange() : null;
  }

  @Override
  public boolean isEngagedToFormat(PsiElement context) {
    PsiFile file = context.getContainingFile();
    return file != null && file.getLanguage() == BuckLanguage.INSTANCE;
  }

  protected static SpacingBuilder createSpacingBuilder(CodeStyleSettings settings) {
    return new SpacingBuilder(settings, BuckLanguage.INSTANCE)
        .between(BuckTypes.RULE_BLOCK, BuckTypes.RULE_BLOCK)
        .blankLines(1)
        .before(BuckTypes.L_BRACKET)
        .spacing(0, 0, 0, false, 0)
        .before(BuckTypes.L_PARENTHESES)
        .spacing(0, 0, 0, false, 0)
        .before(BuckTypes.EQUAL)
        .spacing(1, 1, 0, false, 0)
        .after(BuckTypes.EQUAL)
        .spacing(1, 1, 0, false, 0)
        .before(BuckTypes.COMMA)
        .spacing(0, 0, 0, false, 0)
        .after(BuckTypes.COMMA)
        .lineBreakInCode()
        .before(BuckTypes.R_PARENTHESES)
        .lineBreakInCode()
        .before(BuckTypes.R_BRACKET)
        .lineBreakInCode();
  }
}