// Copyright 2000-2017 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file.
package com.jetbrains.python.codeInsight;

import com.intellij.codeInsight.navigation.actions.GotoDeclarationHandlerBase;
import com.intellij.openapi.editor.Editor;
import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiFile;
import com.intellij.psi.PsiWhiteSpace;
import com.intellij.psi.util.PsiTreeUtil;
import com.jetbrains.python.PythonLanguage;
import com.jetbrains.python.psi.*;
import org.jetbrains.annotations.Nullable;

/**
 * Provides reaction on ctrl+click for {@code break} and {@code continue} statements.
 */
public class PyBreakContinueGotoProvider extends GotoDeclarationHandlerBase {
  @Override
  public PsiElement getGotoDeclarationTarget(@Nullable PsiElement source, Editor editor) {
    if (source != null && source.getLanguage() instanceof PythonLanguage) {
      final PsiElement breakOrContinue = source.getParent();
      final PyLoopStatement loop = PyUtil.getCorrespondingLoop(breakOrContinue);
      if (loop != null) {
        if (breakOrContinue instanceof PyContinueStatement) {
          return loop;
        }
        else {
          PsiElement outer_element = loop;
          PsiElement after_cycle;
          while (true) {
            after_cycle = outer_element.getNextSibling();
            if (after_cycle != null) {
              if (after_cycle instanceof PsiWhiteSpace) {
                after_cycle = after_cycle.getNextSibling();
              }
              if (after_cycle instanceof PyStatement) return after_cycle;
            }
            outer_element = outer_element.getParent();
            if (PsiTreeUtil.instanceOf(outer_element, PsiFile.class, PyFunction.class, PyClass.class)) {
              break;
            }
          }
          // cycle is the last statement in the flow of execution. its last element is our best bet.
          return PsiTreeUtil.getDeepestLast(loop);
        }
      }
    }

    return null;
  }
}
