// Copyright 2018 the V8 project authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef V8_AST_SOURCE_RANGE_AST_VISITOR_H_
#define V8_AST_SOURCE_RANGE_AST_VISITOR_H_

#include <unordered_set>

#include "src/ast/ast-traversal-visitor.h"

namespace v8 {
namespace internal {

class SourceRangeMap;

// Post-processes generated source ranges while the AST structure still exists.
//
// In particular, SourceRangeAstVisitor
//
// 1. deduplicates continuation source ranges, only keeping the outermost one.
// See also: https://crbug.com/v8/8539.
//
// 2. removes the source range associated with the final statement in a block
// or function body if the parent itself has a source range associated with it.
// See also: https://crbug.com/v8/8381.
class SourceRangeAstVisitor final
    : public AstTraversalVisitor<SourceRangeAstVisitor> {
 public:
  SourceRangeAstVisitor(uintptr_t stack_limit, Expression* root,
                        SourceRangeMap* source_range_map);

 private:
  friend class AstTraversalVisitor<SourceRangeAstVisitor>;

  void VisitBlock(Block* stmt);
  void VisitSwitchStatement(SwitchStatement* stmt);
  void VisitFunctionLiteral(FunctionLiteral* expr);
  bool VisitNode(AstNode* node);
  void VisitTryCatchStatement(TryCatchStatement* stmt);
  void VisitTryFinallyStatement(TryFinallyStatement* stmt);

  void MaybeRemoveContinuationRange(Statement* last_statement);
  void MaybeRemoveLastContinuationRange(ZonePtrList<Statement>* stmts);
  void MaybeRemoveContinuationRangeOfAsyncReturn(TryCatchStatement* stmt);

  SourceRangeMap* source_range_map_ = nullptr;
  std::unordered_set<int> continuation_positions_;
};

}  // namespace internal
}  // namespace v8

#endif  // V8_AST_SOURCE_RANGE_AST_VISITOR_H_